chore: import upstream snapshot with attribution

2026-07-13 12:22:33 +08:00
commit fc4fcbab58
1848 changed files with 472303 additions and 0 deletions
@@ -0,0 +1,124 @@
+import Foundation
+
+/// Parsed view of CommandCode `/internal/billing/credits` + `/internal/billing/subscriptions`.
+public struct CommandCodeUsageSnapshot: Sendable {
+    /// USD remaining in the current monthly grant (`credits.monthlyCredits`).
+    public let monthlyCreditsRemaining: Double
+    /// USD top-up balance carried over (`credits.purchasedCredits`).
+    public let purchasedCredits: Double
+    /// USD remaining in the premium monthly grant (`credits.premiumMonthlyCredits`).
+    public let premiumMonthlyCredits: Double
+    /// USD remaining in the open-source monthly grant (`credits.opensourceMonthlyCredits`).
+    public let opensourceMonthlyCredits: Double
+    /// Subscription plan, or nil when the user is on the free tier.
+    public let plan: CommandCodePlanCatalog.Plan?
+    /// `currentPeriodEnd` from the active subscription.
+    public let billingPeriodEnd: Date?
+    /// Subscription status (e.g. `active`, `canceled`).
+    public let subscriptionStatus: String?
+    /// The optional subscription request timed out or failed for this refresh.
+    public let subscriptionEnrichmentUnavailable: Bool
+    public let updatedAt: Date
+
+    public init(
+        monthlyCreditsRemaining: Double,
+        purchasedCredits: Double,
+        premiumMonthlyCredits: Double,
+        opensourceMonthlyCredits: Double,
+        plan: CommandCodePlanCatalog.Plan?,
+        billingPeriodEnd: Date?,
+        subscriptionStatus: String?,
+        subscriptionEnrichmentUnavailable: Bool = false,
+        updatedAt: Date = Date())
+    {
+        self.monthlyCreditsRemaining = monthlyCreditsRemaining
+        self.purchasedCredits = purchasedCredits
+        self.premiumMonthlyCredits = premiumMonthlyCredits
+        self.opensourceMonthlyCredits = opensourceMonthlyCredits
+        self.plan = plan
+        self.billingPeriodEnd = billingPeriodEnd
+        self.subscriptionStatus = subscriptionStatus
+        self.subscriptionEnrichmentUnavailable = subscriptionEnrichmentUnavailable
+        self.updatedAt = updatedAt
+    }
+
+    /// USD allocation for the active monthly grant (from the catalog).
+    public var monthlyCreditsTotal: Double? {
+        self.plan?.monthlyCreditsUSD
+    }
+
+    /// USD spent in the current monthly grant (total – remaining), clamped to [0, total].
+    public var monthlyCreditsUsed: Double? {
+        guard let total = self.monthlyCreditsTotal else { return nil }
+        return max(0, min(total, total - self.monthlyCreditsRemaining))
+    }
+
+    public func toUsageSnapshot() -> UsageSnapshot {
+        let primary = self.makePrimaryWindow()
+
+        let identity = ProviderIdentitySnapshot(
+            providerID: .commandcode,
+            accountEmail: nil,
+            accountOrganization: nil,
+            loginMethod: self.makeLoginMethod())
+
+        return UsageSnapshot(
+            primary: primary,
+            secondary: nil,
+            tertiary: nil,
+            providerCost: nil,
+            commandCodeSubscriptionEnrichmentUnavailable: self.subscriptionEnrichmentUnavailable,
+            commandCodeHasSubscriptionPlan: self.plan != nil,
+            commandCodeMonthlyGrantDepleted: self.monthlyCreditsRemaining <= 0,
+            updatedAt: self.updatedAt,
+            identity: identity)
+    }
+
+    private func makePrimaryWindow() -> RateWindow? {
+        guard let total = self.monthlyCreditsTotal, total > 0 else {
+            // Free / unknown plan with no allowance — surface 100% so the bar renders empty.
+            if self.monthlyCreditsRemaining > 0 || self.purchasedCredits > 0 {
+                return RateWindow(
+                    usedPercent: 0,
+                    windowMinutes: nil,
+                    resetsAt: self.billingPeriodEnd,
+                    resetDescription: nil)
+            }
+            return nil
+        }
+        let used = self.monthlyCreditsUsed ?? 0
+        let percent = min(100, max(0, (used / total) * 100))
+        return RateWindow(
+            usedPercent: percent,
+            windowMinutes: nil,
+            resetsAt: self.billingPeriodEnd,
+            resetDescription: nil)
+    }
+
+    private func makeLoginMethod() -> String? {
+        var parts: [String] = []
+        if let name = self.plan?.displayName, !name.isEmpty {
+            parts.append(name)
+        }
+        if let total = self.monthlyCreditsTotal {
+            let used = self.monthlyCreditsUsed ?? 0
+            parts.append("\(Self.formatUSD(used)) of \(Self.formatUSD(total))")
+        } else if self.monthlyCreditsRemaining > 0 {
+            parts.append("\(Self.formatUSD(self.monthlyCreditsRemaining)) remaining")
+        }
+        if self.purchasedCredits > 0 {
+            parts.append("+ \(Self.formatUSD(self.purchasedCredits)) credits")
+        }
+        return parts.isEmpty ? nil : parts.joined(separator: " · ")
+    }
+
+    static func formatUSD(_ value: Double) -> String {
+        let formatter = NumberFormatter()
+        formatter.numberStyle = .currency
+        formatter.currencyCode = "USD"
+        formatter.locale = Locale(identifier: "en_US")
+        formatter.maximumFractionDigits = value < 100 ? 2 : 0
+        formatter.minimumFractionDigits = value < 100 ? 2 : 0
+        return formatter.string(from: NSNumber(value: value)) ?? "$\(value)"
+    }
+}