import CodexBarCoreimport Foundation enum MenuBarMetricWindowResolver { private enum Lane { case primary case secondary case tertiary } static func rateWindow( preference: MenuBarMetricPreference, provider: UsageProvider, snapshot: UsageSnapshot?, supportsAverage: Bool) -> RateWindow? { guard let snapshot else { return nil } switch preference { case .monthlyPlan: return snapshot.extraRateWindows?.first { $0.id == "mistral-monthly-plan" }?.window case .extraUsage: return Self.extraUsageWindow(snapshot: snapshot) case .tertiary: return Self.requestedWindow( provider: provider, snapshot: snapshot, lanes: Self.tertiaryOrder(for: provider)) case .primary: return Self.requestedWindow( provider: provider, snapshot: snapshot, lanes: Self.primaryOrder(for: provider)) case .secondary: return Self.requestedWindow( provider: provider, snapshot: snapshot, lanes: Self.secondaryOrder(for: provider)) case .average: return Self.averageWindow(provider: provider, snapshot: snapshot, supportsAverage: supportsAverage) case .automatic: return Self.automaticWindow(provider: provider, snapshot: snapshot) } } private static func tertiaryOrder(for provider: UsageProvider) -> [Lane] { if provider == .zai { return [.tertiary, .primary, .secondary] } if provider == .perplexity || provider == .cursor || provider == .antigravity { return [.tertiary, .secondary, .primary] } return [.primary, .secondary] } private static func primaryOrder(for provider: UsageProvider) -> [Lane] { if provider == .zai { return [.primary, .tertiary, .secondary] } if provider == .perplexity || provider == .antigravity { return [.primary, .secondary, .tertiary] } return [.primary, .secondary] } private static func secondaryOrder(for provider: UsageProvider) -> [Lane] { if provider == .zai || provider == .antigravity { return [.secondary, .primary, .tertiary] } if provider == .perplexity { return [.secondary, .tertiary, .primary] } return [.secondary, .primary] } private static func averageWindow( provider: UsageProvider, snapshot: UsageSnapshot, supportsAverage: Bool) -> RateWindow? { guard supportsAverage, let primary = snapshot.primary, let secondary = snapshot.secondary else { if provider == .antigravity { return self.window(in: snapshot, following: [.primary, .secondary, .tertiary]) } return snapshot.primary ?? snapshot.secondary } let usedPercent = (primary.usedPercent + secondary.usedPercent) / 2 return RateWindow(usedPercent: usedPercent, windowMinutes: nil, resetsAt: nil, resetDescription: nil) } private static func automaticWindow(provider: UsageProvider, snapshot: UsageSnapshot) -> RateWindow? { if provider == .antigravity { if let window = mostConstrainedAntigravityQuotaSummaryWindow(snapshot: snapshot) { return window } return self.mostConstrainedWindow( primary: snapshot.primary, secondary: snapshot.secondary, tertiary: snapshot.tertiary) ?? self.mostConstrainedAntigravityLegacyExtraWindow(snapshot: snapshot) } if provider == .perplexity { return snapshot.automaticPerplexityWindow() } if provider == .zai { return self.mostConstrainedWindow( primary: snapshot.primary, secondary: snapshot.tertiary, tertiary: nil) ?? snapshot.secondary } if provider == .factory || provider == .kimi { return snapshot.secondary ?? snapshot.primary } if provider == .litellm { return snapshot.secondary ?? snapshot.primary } if provider == .copilot, let primary = snapshot.primary, let secondary = snapshot.secondary { return primary.usedPercent >= secondary.usedPercent ? primary : secondary } if provider == .cursor { return Self.mostConstrainedCursorWindow( total: snapshot.primary, auto: snapshot.secondary, api: snapshot.tertiary) } if provider == .minimax { return Self.mostConstrainedWindow( primary: snapshot.primary, secondary: snapshot.secondary, tertiary: snapshot.tertiary) } if provider == .claude, Self.shouldUseClaudeSpendLimit(providerCost: snapshot.providerCost, snapshot: snapshot), let extraUsage = Self.extraUsageWindow(snapshot: snapshot) { return extraUsage } return snapshot.primary ?? snapshot.secondary } private static let antigravityQuotaSummaryWindowIDPrefix = "antigravity-quota-summary-" private static let antigravityCompactFallbackWindowIDPrefix = "antigravity-compact-fallback-" private static func mostConstrainedAntigravityQuotaSummaryWindow(snapshot: UsageSnapshot) -> RateWindow? { let windows = snapshot.extraRateWindows? .filter { $0.usageKnown && $0.id.hasPrefix(Self.antigravityQuotaSummaryWindowIDPrefix) } .map(\.window) ?? [] guard !windows.isEmpty else { return nil } let usableWindows = windows.filter { $0.usedPercent < 100 } if let maxUsable = usableWindows.max(by: { $0.usedPercent < $1.usedPercent }) { return maxUsable } return windows.max(by: { $0.usedPercent < $1.usedPercent }) } private static func mostConstrainedAntigravityLegacyExtraWindow(snapshot: UsageSnapshot) -> RateWindow? { let windows = snapshot.extraRateWindows? .filter { $0.usageKnown && $0.id.hasPrefix(Self.antigravityCompactFallbackWindowIDPrefix) } .map(\.window) ?? [] guard !windows.isEmpty else { return nil } let usableWindows = windows.filter { $0.usedPercent < 100 } if let maxUsable = usableWindows.max(by: { $0.usedPercent < $1.usedPercent }) { return maxUsable } return windows.max(by: { $0.usedPercent < $1.usedPercent }) } private static func requestedWindow( provider: UsageProvider, snapshot: UsageSnapshot, lanes: [Lane]) -> RateWindow? { self.window(in: snapshot, following: lanes) ?? (provider == .antigravity ? self.mostConstrainedAntigravityLegacyExtraWindow(snapshot: snapshot) : nil) } private static func window(in snapshot: UsageSnapshot, following lanes: [Lane]) -> RateWindow? { for lane in lanes { if let window = self.window(in: snapshot, lane: lane) { return window } } return nil } private static func window(in snapshot: UsageSnapshot, lane: Lane) -> RateWindow? { switch lane { case .primary: snapshot.primary case .secondary: snapshot.secondary case .tertiary: snapshot.tertiary } } private static func mostConstrainedWindow( primary: RateWindow?, secondary: RateWindow?, tertiary: RateWindow?) -> RateWindow? { let windows = [primary, secondary, tertiary].compactMap(\.self) guard !windows.isEmpty else { return nil } return windows.max(by: { $0.usedPercent < $1.usedPercent }) } private static func mostConstrainedCursorWindow( total: RateWindow?, auto: RateWindow?, api: RateWindow?) -> RateWindow? { if let total, total.usedPercent >= 100 { return total } let subquotaWindows = [auto, api].compactMap(\.self) let usableSubquotaWindows = subquotaWindows.filter { $0.usedPercent < 100 } if !subquotaWindows.isEmpty, usableSubquotaWindows.isEmpty { return subquotaWindows.max(by: { $0.usedPercent < $1.usedPercent }) } return ([total].compactMap(\.self) + usableSubquotaWindows) .max(by: { $0.usedPercent < $1.usedPercent }) } private static func shouldUseClaudeSpendLimit( providerCost: ProviderCostSnapshot?, snapshot: UsageSnapshot) -> Bool { guard providerCost?.limit ?? 0 > 0, snapshot.secondary == nil, snapshot.tertiary == nil else { return false } guard let primary = snapshot.primary else { return true } return primary.usedPercent == 0 && primary.windowMinutes == 5 * 60 && primary.resetsAt == nil && primary.resetDescription == nil } private static func extraUsageWindow(snapshot: UsageSnapshot?) -> RateWindow? { guard let cost = snapshot?.providerCost, cost.limit > 0 else { return nil } let usedPercent = max(0, min(100, (cost.used / cost.limit) * 100)) return RateWindow( usedPercent: usedPercent, windowMinutes: nil, resetsAt: cost.resetsAt, resetDescription: nil) }}
import CodexBarCore
import Foundation
enum MenuBarMetricWindowResolver {
private enum Lane {
case primary
case secondary
case tertiary
}
static func rateWindow(
preference: MenuBarMetricPreference,
provider: UsageProvider,
snapshot: UsageSnapshot?,
supportsAverage: Bool)
-> RateWindow?
{
guard let snapshot else { return nil }
switch preference {
case .monthlyPlan:
return snapshot.extraRateWindows?.first { $0.id == "mistral-monthly-plan" }?.window
case .extraUsage:
return Self.extraUsageWindow(snapshot: snapshot)
case .tertiary:
return Self.requestedWindow(
provider: provider,
snapshot: snapshot,
lanes: Self.tertiaryOrder(for: provider))
case .primary:
return Self.requestedWindow(
provider: provider,
snapshot: snapshot,
lanes: Self.primaryOrder(for: provider))
case .secondary:
return Self.requestedWindow(
provider: provider,
snapshot: snapshot,
lanes: Self.secondaryOrder(for: provider))
case .average:
return Self.averageWindow(provider: provider, snapshot: snapshot, supportsAverage: supportsAverage)
case .automatic:
return Self.automaticWindow(provider: provider, snapshot: snapshot)
}
}
private static func tertiaryOrder(for provider: UsageProvider) -> [Lane] {
if provider == .zai {
return [.tertiary, .primary, .secondary]
}
if provider == .perplexity || provider == .cursor || provider == .antigravity {
return [.tertiary, .secondary, .primary]
}
return [.primary, .secondary]
}
private static func primaryOrder(for provider: UsageProvider) -> [Lane] {
if provider == .zai {
return [.primary, .tertiary, .secondary]
}
if provider == .perplexity || provider == .antigravity {
return [.primary, .secondary, .tertiary]
}
return [.primary, .secondary]
}
private static func secondaryOrder(for provider: UsageProvider) -> [Lane] {
if provider == .zai || provider == .antigravity {
return [.secondary, .primary, .tertiary]
}
if provider == .perplexity {
return [.secondary, .tertiary, .primary]
}
return [.secondary, .primary]
}
private static func averageWindow(
provider: UsageProvider,
snapshot: UsageSnapshot,
supportsAverage: Bool)
-> RateWindow?
{
guard supportsAverage,
let primary = snapshot.primary,
let secondary = snapshot.secondary
else {
if provider == .antigravity {
return self.window(in: snapshot, following: [.primary, .secondary, .tertiary])
}
return snapshot.primary ?? snapshot.secondary
}
let usedPercent = (primary.usedPercent + secondary.usedPercent) / 2
return RateWindow(usedPercent: usedPercent, windowMinutes: nil, resetsAt: nil, resetDescription: nil)
}
private static func automaticWindow(provider: UsageProvider, snapshot: UsageSnapshot) -> RateWindow? {
if provider == .antigravity {
if let window = mostConstrainedAntigravityQuotaSummaryWindow(snapshot: snapshot) {
return window
}
return self.mostConstrainedWindow(
primary: snapshot.primary,
secondary: snapshot.secondary,
tertiary: snapshot.tertiary)
?? self.mostConstrainedAntigravityLegacyExtraWindow(snapshot: snapshot)
}
if provider == .perplexity {
return snapshot.automaticPerplexityWindow()
}
if provider == .zai {
return self.mostConstrainedWindow(
primary: snapshot.primary,
secondary: snapshot.tertiary,
tertiary: nil) ?? snapshot.secondary
}
if provider == .factory || provider == .kimi {
return snapshot.secondary ?? snapshot.primary
}
if provider == .litellm {
return snapshot.secondary ?? snapshot.primary
}
if provider == .copilot,
let primary = snapshot.primary,
let secondary = snapshot.secondary
{
return primary.usedPercent >= secondary.usedPercent ? primary : secondary
}
if provider == .cursor {
return Self.mostConstrainedCursorWindow(
total: snapshot.primary,
auto: snapshot.secondary,
api: snapshot.tertiary)
}
if provider == .minimax {
return Self.mostConstrainedWindow(
primary: snapshot.primary,
secondary: snapshot.secondary,
tertiary: snapshot.tertiary)
}
if provider == .claude,
Self.shouldUseClaudeSpendLimit(providerCost: snapshot.providerCost, snapshot: snapshot),
let extraUsage = Self.extraUsageWindow(snapshot: snapshot)
{
return extraUsage
}
return snapshot.primary ?? snapshot.secondary
}
private static let antigravityQuotaSummaryWindowIDPrefix = "antigravity-quota-summary-"
private static let antigravityCompactFallbackWindowIDPrefix = "antigravity-compact-fallback-"
private static func mostConstrainedAntigravityQuotaSummaryWindow(snapshot: UsageSnapshot) -> RateWindow? {
let windows = snapshot.extraRateWindows?
.filter { $0.usageKnown && $0.id.hasPrefix(Self.antigravityQuotaSummaryWindowIDPrefix) }
.map(\.window) ?? []
guard !windows.isEmpty else { return nil }
let usableWindows = windows.filter { $0.usedPercent < 100 }
if let maxUsable = usableWindows.max(by: { $0.usedPercent < $1.usedPercent }) {
return maxUsable
}
return windows.max(by: { $0.usedPercent < $1.usedPercent })
}
private static func mostConstrainedAntigravityLegacyExtraWindow(snapshot: UsageSnapshot) -> RateWindow? {
let windows = snapshot.extraRateWindows?
.filter {
$0.usageKnown && $0.id.hasPrefix(Self.antigravityCompactFallbackWindowIDPrefix)
}
.map(\.window) ?? []
guard !windows.isEmpty else { return nil }
let usableWindows = windows.filter { $0.usedPercent < 100 }
if let maxUsable = usableWindows.max(by: { $0.usedPercent < $1.usedPercent }) {
return maxUsable
}
return windows.max(by: { $0.usedPercent < $1.usedPercent })
}
private static func requestedWindow(
provider: UsageProvider,
snapshot: UsageSnapshot,
lanes: [Lane]) -> RateWindow?
{
self.window(in: snapshot, following: lanes)
?? (provider == .antigravity
? self.mostConstrainedAntigravityLegacyExtraWindow(snapshot: snapshot)
: nil)
}
private static func window(in snapshot: UsageSnapshot, following lanes: [Lane]) -> RateWindow? {
for lane in lanes {
if let window = self.window(in: snapshot, lane: lane) {
return window
}
}
return nil
}
private static func window(in snapshot: UsageSnapshot, lane: Lane) -> RateWindow? {
switch lane {
case .primary:
snapshot.primary
case .secondary:
snapshot.secondary
case .tertiary:
snapshot.tertiary
}
}
private static func mostConstrainedWindow(
primary: RateWindow?,
secondary: RateWindow?,
tertiary: RateWindow?)
-> RateWindow?
{
let windows = [primary, secondary, tertiary].compactMap(\.self)
guard !windows.isEmpty else { return nil }
return windows.max(by: { $0.usedPercent < $1.usedPercent })
}
private static func mostConstrainedCursorWindow(
total: RateWindow?,
auto: RateWindow?,
api: RateWindow?)
-> RateWindow?
{
if let total, total.usedPercent >= 100 {
return total
}
let subquotaWindows = [auto, api].compactMap(\.self)
let usableSubquotaWindows = subquotaWindows.filter { $0.usedPercent < 100 }
if !subquotaWindows.isEmpty, usableSubquotaWindows.isEmpty {
return subquotaWindows.max(by: { $0.usedPercent < $1.usedPercent })
}
return ([total].compactMap(\.self) + usableSubquotaWindows)
.max(by: { $0.usedPercent < $1.usedPercent })
}
private static func shouldUseClaudeSpendLimit(
providerCost: ProviderCostSnapshot?,
snapshot: UsageSnapshot)
-> Bool
{
guard providerCost?.limit ?? 0 > 0,
snapshot.secondary == nil,
snapshot.tertiary == nil
else { return false }
guard let primary = snapshot.primary else { return true }
return primary.usedPercent == 0
&& primary.windowMinutes == 5 * 60
&& primary.resetsAt == nil
&& primary.resetDescription == nil
}
private static func extraUsageWindow(snapshot: UsageSnapshot?) -> RateWindow? {
guard let cost = snapshot?.providerCost, cost.limit > 0 else { return nil }
let usedPercent = max(0, min(100, (cost.used / cost.limit) * 100))
return RateWindow(
usedPercent: usedPercent,
windowMinutes: nil,
resetsAt: cost.resetsAt,
resetDescription: nil)
}
}