import CodexBarCoreimport Foundation enum MenuBarDisplayText { static func percentText(window: RateWindow?, showUsed: Bool) -> String? { guard let window else { return nil } let percent = showUsed ? window.usedPercent : window.remainingPercent return UsageFormatter.percentString(percent) } static func paceText(pace: UsagePace?) -> String? { guard let pace else { return nil } let deltaValue = Int(abs(pace.deltaPercent).rounded()) if deltaValue == 0 { return "0%" } let sign = pace.deltaPercent >= 0 ? "+" : "-" return "\(sign)\(deltaValue)%" } /// Combined "session · weekly" menu-bar text shared by providers that expose both a /// session (5h) and weekly (7d) lane, e.g. Codex and Claude. static func combinedSessionWeeklyPercentText( sessionWindow: RateWindow?, weeklyWindow: RateWindow?, showUsed: Bool, resetTimeDisplayStyle: ResetTimeDisplayStyle = .countdown, showsResetTimeWhenExhausted: Bool = false, now: Date = .init()) -> String? { var parts: [String] = [] if let sessionWindow, let session = self.laneValueText( window: sessionWindow, showUsed: showUsed, resetTimeDisplayStyle: resetTimeDisplayStyle, showsResetTimeWhenExhausted: showsResetTimeWhenExhausted, now: now) { parts.append("\(self.sessionWindowLabel(window: sessionWindow)) \(session)") } if let weeklyWindow, let weekly = self.laneValueText( window: weeklyWindow, showUsed: showUsed, resetTimeDisplayStyle: resetTimeDisplayStyle, showsResetTimeWhenExhausted: showsResetTimeWhenExhausted, now: now) { parts.append("W \(weekly)") } return parts.isEmpty ? nil : parts.joined(separator: " · ") } private static func laneValueText( window: RateWindow, showUsed: Bool, resetTimeDisplayStyle: ResetTimeDisplayStyle, showsResetTimeWhenExhausted: Bool, now: Date) -> String? { if let resetText = self.exhaustedResetText( window: window, enabled: showsResetTimeWhenExhausted, style: resetTimeDisplayStyle, now: now) { return resetText } return self.percentText(window: window, showUsed: showUsed) } private static func sessionWindowLabel(window: RateWindow) -> String { guard let minutes = window.windowMinutes, minutes > 0 else { return "S" } guard minutes.isMultiple(of: 60) else { return "\(minutes)m" } return "\(minutes / 60)h" } static func displayText( mode: MenuBarDisplayMode, percentWindow: RateWindow?, pace: UsagePace? = nil, showUsed: Bool, resetTimeDisplayStyle: ResetTimeDisplayStyle = .countdown, showsResetTimeWhenExhausted: Bool = false, now: Date = .init()) -> String? { if mode != .resetTime, showsResetTimeWhenExhausted, let percentWindow, percentWindow.remainingPercent <= 0 { if let resetText = self.exhaustedResetText( window: percentWindow, enabled: true, style: resetTimeDisplayStyle, now: now) { return resetText } // Smart mode cannot replace an exhausted percentage unless the reset is concrete, future, // and schedulable. Preserve the quota signal in pace/both modes too; a pace from another // combined lane must not hide that this displayed lane is already exhausted. return self.percentText(window: percentWindow, showUsed: showUsed) } switch mode { case .percent: return self.percentText(window: percentWindow, showUsed: showUsed) case .pace: // Pace can be temporarily unavailable near a reset or when a provider omits window metadata. // Keep the selected quota visible instead of collapsing the status item to an icon-only state. return self.paceText(pace: pace) ?? self.percentText(window: percentWindow, showUsed: showUsed) case .both: guard let percent = percentText(window: percentWindow, showUsed: showUsed) else { return nil } // Fall back to percent-only when pace is unavailable (e.g. Copilot) guard let paceText = Self.paceText(pace: pace) else { return percent } return "\(percent) · \(paceText)" case .resetTime: guard let percentWindow else { return nil } return self.resetTimeText(window: percentWindow, style: resetTimeDisplayStyle, now: now) ?? self.percentText(window: percentWindow, showUsed: showUsed) } } /// "↻ …" reset text for a window, or nil when it carries no usable reset metadata. static func resetTimeText( window: RateWindow, style: ResetTimeDisplayStyle, now: Date) -> String? { if let resetsAt = window.resetsAt { let description = switch style { case .countdown: UsageFormatter.resetCountdownDescription(from: resetsAt, now: now) case .absolute: UsageFormatter.resetDescription(from: resetsAt, now: now) } return "↻ \(description)" } if let resetDescription = self.resetMetadataText(window.resetDescription) { return "↻ \(resetDescription)" } return nil } /// Smart-mode replacement: when enabled and the quota is exhausted (0% remaining, regardless of /// whether the display shows used or remaining), surface the reset time instead of a dead percent. /// /// Requires a concrete, still-future `resetsAt`. The smart option only replaces the percent when it /// has a reset time it can both render as a live countdown/clock AND hand to the refresh scheduler, /// so the lane keeps ticking and flips back to the percentage once the reset passes. Windows with /// only textual reset metadata (`resetDescription`, no `resetsAt`) or an already-elapsed reset can't /// be scheduled, so they keep showing the percent instead of freezing on stale reset text. private static func exhaustedResetText( window: RateWindow?, enabled: Bool, style: ResetTimeDisplayStyle, now: Date) -> String? { guard enabled, let window, window.remainingPercent <= 0 else { return nil } guard let resetsAt = window.resetsAt, resetsAt > now else { return nil } return self.resetTimeText(window: window, style: style, now: now) } private static func resetMetadataText(_ description: String?) -> String? { guard let description else { return nil } let trimmed = description.trimmingCharacters(in: .whitespacesAndNewlines) guard !trimmed.isEmpty else { return nil } // RateWindow.resetDescription predates provider-specific detail fields and is also used for // request/token summaries. Only trust phrases that explicitly describe reset timing. let normalized = trimmed.lowercased() let resetPrefixes = [ "reset ", "resets ", "in ", "today ", "today,", "tomorrow ", "tomorrow,", "next ", "expire ", "expires ", "refill ", "refills ", ] let exactResetDescriptions = ["today", "tomorrow", "expired", "now", "soon"] return exactResetDescriptions.contains(normalized) || resetPrefixes.contains(where: normalized.hasPrefix) ? trimmed : nil }}
import CodexBarCore
import Foundation
enum MenuBarDisplayText {
static func percentText(window: RateWindow?, showUsed: Bool) -> String? {
guard let window else { return nil }
let percent = showUsed ? window.usedPercent : window.remainingPercent
return UsageFormatter.percentString(percent)
}
static func paceText(pace: UsagePace?) -> String? {
guard let pace else { return nil }
let deltaValue = Int(abs(pace.deltaPercent).rounded())
if deltaValue == 0 { return "0%" }
let sign = pace.deltaPercent >= 0 ? "+" : "-"
return "\(sign)\(deltaValue)%"
}
/// Combined "session · weekly" menu-bar text shared by providers that expose both a
/// session (5h) and weekly (7d) lane, e.g. Codex and Claude.
static func combinedSessionWeeklyPercentText(
sessionWindow: RateWindow?,
weeklyWindow: RateWindow?,
showUsed: Bool,
resetTimeDisplayStyle: ResetTimeDisplayStyle = .countdown,
showsResetTimeWhenExhausted: Bool = false,
now: Date = .init())
-> String?
{
var parts: [String] = []
if let sessionWindow,
let session = self.laneValueText(
window: sessionWindow,
showUsed: showUsed,
resetTimeDisplayStyle: resetTimeDisplayStyle,
showsResetTimeWhenExhausted: showsResetTimeWhenExhausted,
now: now)
{
parts.append("\(self.sessionWindowLabel(window: sessionWindow)) \(session)")
}
if let weeklyWindow,
let weekly = self.laneValueText(
window: weeklyWindow,
showUsed: showUsed,
resetTimeDisplayStyle: resetTimeDisplayStyle,
showsResetTimeWhenExhausted: showsResetTimeWhenExhausted,
now: now)
{
parts.append("W \(weekly)")
}
return parts.isEmpty ? nil : parts.joined(separator: " · ")
}
private static func laneValueText(
window: RateWindow,
showUsed: Bool,
resetTimeDisplayStyle: ResetTimeDisplayStyle,
showsResetTimeWhenExhausted: Bool,
now: Date) -> String?
{
if let resetText = self.exhaustedResetText(
window: window,
enabled: showsResetTimeWhenExhausted,
style: resetTimeDisplayStyle,
now: now)
{
return resetText
}
return self.percentText(window: window, showUsed: showUsed)
}
private static func sessionWindowLabel(window: RateWindow) -> String {
guard let minutes = window.windowMinutes, minutes > 0 else { return "S" }
guard minutes.isMultiple(of: 60) else { return "\(minutes)m" }
return "\(minutes / 60)h"
}
static func displayText(
mode: MenuBarDisplayMode,
percentWindow: RateWindow?,
pace: UsagePace? = nil,
showUsed: Bool,
resetTimeDisplayStyle: ResetTimeDisplayStyle = .countdown,
showsResetTimeWhenExhausted: Bool = false,
now: Date = .init()) -> String?
{
if mode != .resetTime,
showsResetTimeWhenExhausted,
let percentWindow,
percentWindow.remainingPercent <= 0
{
if let resetText = self.exhaustedResetText(
window: percentWindow,
enabled: true,
style: resetTimeDisplayStyle,
now: now)
{
return resetText
}
// Smart mode cannot replace an exhausted percentage unless the reset is concrete, future,
// and schedulable. Preserve the quota signal in pace/both modes too; a pace from another
// combined lane must not hide that this displayed lane is already exhausted.
return self.percentText(window: percentWindow, showUsed: showUsed)
}
switch mode {
case .percent:
return self.percentText(window: percentWindow, showUsed: showUsed)
case .pace:
// Pace can be temporarily unavailable near a reset or when a provider omits window metadata.
// Keep the selected quota visible instead of collapsing the status item to an icon-only state.
return self.paceText(pace: pace)
?? self.percentText(window: percentWindow, showUsed: showUsed)
case .both:
guard let percent = percentText(window: percentWindow, showUsed: showUsed) else { return nil }
// Fall back to percent-only when pace is unavailable (e.g. Copilot)
guard let paceText = Self.paceText(pace: pace) else { return percent }
return "\(percent) · \(paceText)"
case .resetTime:
guard let percentWindow else { return nil }
return self.resetTimeText(window: percentWindow, style: resetTimeDisplayStyle, now: now)
?? self.percentText(window: percentWindow, showUsed: showUsed)
}
}
/// "↻ …" reset text for a window, or nil when it carries no usable reset metadata.
static func resetTimeText(
window: RateWindow,
style: ResetTimeDisplayStyle,
now: Date) -> String?
{
if let resetsAt = window.resetsAt {
let description = switch style {
case .countdown:
UsageFormatter.resetCountdownDescription(from: resetsAt, now: now)
case .absolute:
UsageFormatter.resetDescription(from: resetsAt, now: now)
}
return "↻ \(description)"
}
if let resetDescription = self.resetMetadataText(window.resetDescription) {
return "↻ \(resetDescription)"
}
return nil
}
/// Smart-mode replacement: when enabled and the quota is exhausted (0% remaining, regardless of
/// whether the display shows used or remaining), surface the reset time instead of a dead percent.
///
/// Requires a concrete, still-future `resetsAt`. The smart option only replaces the percent when it
/// has a reset time it can both render as a live countdown/clock AND hand to the refresh scheduler,
/// so the lane keeps ticking and flips back to the percentage once the reset passes. Windows with
/// only textual reset metadata (`resetDescription`, no `resetsAt`) or an already-elapsed reset can't
/// be scheduled, so they keep showing the percent instead of freezing on stale reset text.
private static func exhaustedResetText(
window: RateWindow?,
enabled: Bool,
style: ResetTimeDisplayStyle,
now: Date) -> String?
{
guard enabled, let window, window.remainingPercent <= 0 else { return nil }
guard let resetsAt = window.resetsAt, resetsAt > now else { return nil }
return self.resetTimeText(window: window, style: style, now: now)
}
private static func resetMetadataText(_ description: String?) -> String? {
guard let description else { return nil }
let trimmed = description.trimmingCharacters(in: .whitespacesAndNewlines)
guard !trimmed.isEmpty else { return nil }
// RateWindow.resetDescription predates provider-specific detail fields and is also used for
// request/token summaries. Only trust phrases that explicitly describe reset timing.
let normalized = trimmed.lowercased()
let resetPrefixes = [
"reset ", "resets ", "in ", "today ", "today,", "tomorrow ", "tomorrow,", "next ",
"expire ", "expires ", "refill ", "refills ",
]
let exactResetDescriptions = ["today", "tomorrow", "expired", "now", "soon"]
return exactResetDescriptions.contains(normalized) || resetPrefixes.contains(where: normalized.hasPrefix)
? trimmed
: nil
}
}