import CodexBarCoreimport Foundation extension SettingsStore {    func menuBarMetricPreference(for provider: UsageProvider) -> MenuBarMetricPreference {        if Self.isBalanceOnlyProvider(provider), provider != .mistral {            return .automatic        }        if provider == .mistral {            let raw = self.menuBarMetricPreferencesRaw[provider.rawValue] ?? ""            let preference = MenuBarMetricPreference(rawValue: raw) ?? .automatic            switch preference {            case .automatic, .monthlyPlan:                return preference            case .primary, .secondary, .tertiary, .extraUsage, .average:                return .automatic            }        }        if provider == .openrouter {            let raw = self.menuBarMetricPreferencesRaw[provider.rawValue] ?? ""            let preference = MenuBarMetricPreference(rawValue: raw) ?? .automatic            switch preference {            case .automatic, .primary:                return preference            case .secondary, .average, .tertiary, .extraUsage, .monthlyPlan:                return .automatic            }        }        let raw = self.menuBarMetricPreferencesRaw[provider.rawValue] ?? ""        let preference = MenuBarMetricPreference(rawValue: raw) ?? .automatic        if preference == .average, !self.menuBarMetricSupportsAverage(for: provider) {            return .automatic        }        if preference == .tertiary, !self.menuBarMetricSupportsTertiary(for: provider) {            return .automatic        }        if preference == .extraUsage, !self.menuBarMetricSupportsExtraUsage(for: provider) {            return .automatic        }        if preference == .monthlyPlan {            return .automatic        }        return preference    }     func setMenuBarMetricPreference(_ preference: MenuBarMetricPreference, for provider: UsageProvider) {        if Self.isBalanceOnlyProvider(provider), provider != .mistral {            self.menuBarMetricPreferencesRaw[provider.rawValue] = MenuBarMetricPreference.automatic.rawValue            return        }        if provider == .mistral {            switch preference {            case .automatic, .monthlyPlan:                self.menuBarMetricPreferencesRaw[provider.rawValue] = preference.rawValue            case .primary, .secondary, .tertiary, .extraUsage, .average:                self.menuBarMetricPreferencesRaw[provider.rawValue] = MenuBarMetricPreference.automatic.rawValue            }            return        }        if provider == .openrouter {            switch preference {            case .automatic, .primary:                self.menuBarMetricPreferencesRaw[provider.rawValue] = preference.rawValue            case .secondary, .average, .tertiary, .extraUsage, .monthlyPlan:                self.menuBarMetricPreferencesRaw[provider.rawValue] = MenuBarMetricPreference.automatic.rawValue            }            return        }        if preference == .tertiary, !self.menuBarMetricSupportsTertiary(for: provider) {            self.menuBarMetricPreferencesRaw[provider.rawValue] = MenuBarMetricPreference.automatic.rawValue            return        }        if preference == .extraUsage, !self.menuBarMetricSupportsExtraUsage(for: provider) {            self.menuBarMetricPreferencesRaw[provider.rawValue] = MenuBarMetricPreference.automatic.rawValue            return        }        if preference == .monthlyPlan {            self.menuBarMetricPreferencesRaw[provider.rawValue] = MenuBarMetricPreference.automatic.rawValue            return        }        self.menuBarMetricPreferencesRaw[provider.rawValue] = preference.rawValue    }     func menuBarMetricSupportsAverage(for provider: UsageProvider) -> Bool {        provider == .gemini    }     func menuBarMetricSupportsTertiary(for provider: UsageProvider) -> Bool {        provider == .cursor || provider == .perplexity || provider == .zai    }     func menuBarMetricSupportsTertiary(for provider: UsageProvider, snapshot: UsageSnapshot?) -> Bool {        if provider == .cursor || provider == .zai {            return snapshot?.tertiary != nil        }        return self.menuBarMetricSupportsTertiary(for: provider)    }     func menuBarMetricSupportsExtraUsage(for provider: UsageProvider) -> Bool {        provider == .cursor || provider == .claude    }     func menuBarMetricSupportsExtraUsage(for provider: UsageProvider, snapshot: UsageSnapshot?) -> Bool {        guard self.menuBarMetricSupportsExtraUsage(for: provider) else { return false }        guard let cost = snapshot?.providerCost else { return false }        return cost.limit > 0    }     func menuBarMetricPreference(for provider: UsageProvider, snapshot: UsageSnapshot?) -> MenuBarMetricPreference {        let preference = self.menuBarMetricPreference(for: provider)        if preference == .tertiary,           !self.menuBarMetricSupportsTertiary(for: provider, snapshot: snapshot)        {            return .automatic        }        if preference == .extraUsage,           !self.menuBarMetricSupportsExtraUsage(for: provider, snapshot: snapshot)        {            return .automatic        }        return preference    }     func isCostUsageEffectivelyEnabled(for provider: UsageProvider) -> Bool {        self.costUsageEnabled            && ProviderDescriptorRegistry.descriptor(for: provider).tokenCost.supportsTokenCost    }     var resetTimeDisplayStyle: ResetTimeDisplayStyle {        self.resetTimesShowAbsolute ? .absolute : .countdown    }     static func isBalanceOnlyProvider(_ provider: UsageProvider) -> Bool {        switch provider {        case .deepseek, .mistral, .kimik2, .moonshot, .poe:            true        default:            false        }    }}