From a290985eea4188ddc6a754fbd00998e5e476318d Mon Sep 17 00:00:00 2001
From: Vincent Pochet <vincent@getlago.com>
Date: Mon, 2 Sep 2024 17:54:25 +0200
Subject: [PATCH] feat(ProgressiveBilling): Expose lifetime usage GET and PUT

---
 lifetime_usage.go | 90 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 90 insertions(+)
 create mode 100644 lifetime_usage.go

diff --git a/lifetime_usage.go b/lifetime_usage.go
new file mode 100644
index 0000000..0643d73
--- /dev/null
+++ b/lifetime_usage.go
@@ -0,0 +1,90 @@
+package lago
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"github.com/google/uuid"
+)
+
+type LifetimeUsage struct {
+	LagoID                 uuid.UUID `json:"lago_id"`
+	LagoSubscriptionID     uuid.UUID `json:"lago_subscription_id"`
+	ExternalSubscriptionID string    `json:"external_subscription_id"`
+
+	ExternalHistoricalUsageAmountCents *int `json:"external_historical_usage_amount_cents,omitempty"`
+	InvoicedUsageAmountCents           *int `json:"invoiced_usage_amount_cents,omitempty"`
+	CurrentUsageAmountCents            *int `json:"current_usage_amount_cents,omitempty"`
+
+	FromDatetime time.Time `json:"from_datetime"`
+	ToDatetime   time.Time `json:"to_datetime"`
+
+	UsageThresholds []LifetimeUsageThreshold `json:"usage_thresholds,omitempty"`
+}
+
+type LifetimeUsageThreshold struct {
+	AmountCents     int        `json:"amount_cents"`
+	CompletionRatio float32    `json:"completion_ratio"`
+	ReachedAt       *time.Time `json:"reached_at"`
+}
+
+type LifetimeUsageResult struct {
+	LifetimeUsage *LifetimeUsage `json:"lifetime_usage"`
+}
+
+type LifetimeUsageParams struct {
+	LifetimeUsage *LifetimeUsageInput `json:"lifetime_usage"`
+}
+
+type LifetimeUsageInput struct {
+	ExternalSubscriptionID             string `json:"external_subscription_id"`
+	ExternalHistoricalUsageAmountCents int    `json:"external_historical_usage_amount_cents"`
+}
+
+func (sr *SubscriptionRequest) GetLifetimeUsage(ctx context.Context, externalSubscriptionID string) (*LifetimeUsage, *Error) {
+	subPath := fmt.Sprintf("%s/%s/%s", "subscriptions", externalSubscriptionID, "lifetime_usage")
+
+	clientRequest := &ClientRequest{
+		Path:   subPath,
+		Result: &LifetimeUsageResult{},
+	}
+
+	result, clientErr := sr.client.Get(ctx, clientRequest)
+	if clientErr != nil {
+		return nil, clientErr
+	}
+
+	lifetimeUsageResult, ok := result.(*LifetimeUsageResult)
+	if !ok {
+		return nil, &ErrorTypeAssert
+	}
+
+	return lifetimeUsageResult.LifetimeUsage, nil
+}
+
+func (sr *SubscriptionRequest) UpdateLifetimeUsage(ctx context.Context, lifetimeUsageInput *LifetimeUsageInput) (*LifetimeUsage, *Error) {
+	subPath := fmt.Sprintf("%s/%s/%s", "subscriptions", lifetimeUsageInput.ExternalSubscriptionID, "lifetime_usage")
+
+	lifetimeUsageParams := &LifetimeUsageParams{
+		LifetimeUsage: lifetimeUsageInput,
+	}
+
+	clientRequest := &ClientRequest{
+		Path:   subPath,
+		Result: &LifetimeUsageResult{},
+		Body:   lifetimeUsageParams,
+	}
+
+	result, clientErr := sr.client.Put(ctx, clientRequest)
+	if clientErr != nil {
+		return nil, clientErr
+	}
+
+	lifetimeUsageResult, ok := result.(*LifetimeUsageResult)
+	if !ok {
+		return nil, &ErrorTypeAssert
+	}
+
+	return lifetimeUsageResult.LifetimeUsage, nil
+}