feat: harden prepaid billing and commercial operations

1 files changed, 13 insertions, 3 deletions

diff --git a/internal/controlplane/usage.go b/internal/controlplane/usage.go
index 6c69a9f..b436d8c 100644
--- a/internal/controlplane/usage.go
+++ b/internal/controlplane/usage.go
@@ -29,13 +29,13 @@ func (s *Store) RecordUsage(ctx context.Context, event domain.UsageEvent) error
 			request_id, tenant_id, project_id, key_id, public_model, provider_id, upstream_model,
 			protocol, stream, status_code, success, error_type, attempts, started_at, duration_ms,
 			input_tokens, output_tokens, total_tokens, cache_creation_input_tokens, cache_read_input_tokens,
-			cost_micros, charged_micros, uncollected_micros)
-		VALUES ($1,$2,$3,$4,$5,NULLIF($6,''),NULLIF($7,''),$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18,$19,$20,0,0,0)
+			cost_micros, charged_micros, uncollected_micros, usage_reported, metering_status)
+		VALUES ($1,$2,$3,$4,$5,NULLIF($6,''),NULLIF($7,''),$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18,$19,$20,0,0,0,$21,$22)
 		ON CONFLICT (request_id) DO NOTHING`, event.RequestID, event.TenantID, event.ProjectID, event.KeyID,
 		event.PublicModel, event.ProviderID, event.UpstreamModel, string(event.Protocol), event.Stream,
 		event.StatusCode, event.Success, event.ErrorType, event.Attempts, event.StartedAt, event.DurationMS,
 		event.Usage.InputTokens, event.Usage.OutputTokens, event.Usage.TotalTokens,
-		event.Usage.CacheCreationInputTokens, event.Usage.CacheReadInputTokens)
+		event.Usage.CacheCreationInputTokens, event.Usage.CacheReadInputTokens, event.UsageReported, usageMeteringStatus(event))
 	if err != nil {
 		return fmt.Errorf("persist usage event: %w", err)
 	}
@@ -51,6 +51,16 @@ func (s *Store) RecordUsage(ctx context.Context, event domain.UsageEvent) error
 	return nil
 }
 
+func usageMeteringStatus(event domain.UsageEvent) string {
+	if event.StatusCode < 200 || event.StatusCode >= 300 || !event.Success {
+		return "upstream_failed"
+	}
+	if event.UsageReported {
+		return "reported"
+	}
+	return "missing"
+}
+
 func upsertUsageRollup(ctx context.Context, tx pgx.Tx, event domain.UsageEvent, cost, charged, uncollected int64) error {
 	period := time.Date(event.StartedAt.UTC().Year(), event.StartedAt.UTC().Month(), 1, 0, 0, 0, 0, time.UTC)
 	_, err := tx.Exec(ctx, `