on master help sign in
CLAUDE.md+40−12
since splitsnapshots
unifiedsplit
@@ -1,22 +1,50 @@
1# UsageMenubar
1# UsageMenubar
2
2
3## Architecture
3## Architecture
4−macOS menu bar app (SwiftUI, macOS 14+) showing Hyper (Charm) credit balance, plus
4+macOS menu bar app (SwiftUI, macOS 14+) showing three LLM providers with equal
5+visual treatment:
6+- **Hyper** (Charm) — credit balance
5−Claude Code's Pro/Max usage limits when the CLI is signed in on the machine, plus
7+- **Claude Code** — Pro/Max usage limits (5-hour, 7-day, 7-day Opus)
6−z.ai Coding Plan usage limits when an API key is entered in settings.
8+- **z.ai Coding Plan** — usage limits (5-hour, weekly)
9+
10+Each configured provider gets the same popover treatment: a 26pt subhero headline
11+number, bars under every window, a sparkline of recent history, a trend arrow,
12+click-to-copy on the headline, and threshold-crossing notifications.
13+
7- `UsageMenubarApp.swift` — @main App, NSStatusBar, timer, sleep/wake, popover
14- `UsageMenubarApp.swift` — @main App, NSStatusBar, timer, sleep/wake, popover
8−- `ViewModel.swift` — Observable state: balance, Claude usage, z.ai usage, refresh, notifications, history
15+- `ViewModel.swift` — Observable state: balance, Claude usage, z.ai usage, refresh, notifications, per-provider history
9−- `MenuView.swift` — SwiftUI popover view (balance, sparkline, Claude limits, z.ai limits, settings)
16+- `MenuView.swift` — SwiftUI popover view (3 equal provider sections, settings)
10- `CreditsChecker.swift` — API client for `GET https://hyper.charm.land/v1/credits`
17- `CreditsChecker.swift` — API client for `GET https://hyper.charm.land/v1/credits`
11- `ClaudeUsageClient.swift` — read-only Claude OAuth client + models for `GET https://api.anthropic.com/api/oauth/usage`
18- `ClaudeUsageClient.swift` — read-only Claude OAuth client + models for `GET https://api.anthropic.com/api/oauth/usage`
12- `ZaiUsageClient.swift` — API client for z.ai coding plan usage (`GET https://api.z.ai/api/monitor/usage/quota/limit` + `GET https://api.z.ai/api/biz/subscription/list`)
19- `ZaiUsageClient.swift` — API client for z.ai coding plan usage (`GET https://api.z.ai/api/monitor/usage/quota/limit` + `GET https://api.z.ai/api/biz/subscription/list`)
13- `KeychainHelper.swift` — Keychain wrapper (KeychainStore + KeychainHelper + ZaiKeychainHelper)
20- `KeychainHelper.swift` — Keychain wrapper (KeychainStore + KeychainHelper + ZaiKeychainHelper)
14- `VersionFormatter.swift` — Version string formatting
21- `VersionFormatter.swift` — Version string formatting
15
22
16−The status bar title shows `⚡{balance}` (Hyper), `🕐{percent}%` (Claude 5-hour),
23+The status bar title shows `⚡{balance}` (Hyper), `⁕{percent}%` (Claude 5-hour),
17−`📅{percent}%` (Claude 7-day), and `🤖{percent}%` (z.ai 5-hour) — each omitted when
24+`⁕{percent}%` (Claude 7-day), `Z{percent}%` (z.ai 5-hour), and `Z{percent}%`
25+(z.ai weekly) — each omitted when absent or at 0%, and entirely absent when the
26+service is not configured (no API key or no credentials). An unconfigured service
18−absent or at 0%. The three fetches are independent — `ViewModel.refresh()` runs them
27+produces no placeholder in the menu bar. The three fetches are independent —
28+`ViewModel.refresh()` runs them with `async let`, and any can fail or be
19−with `async let`, and any can fail or be unconfigured without touching the others.
29+unconfigured without touching the others.
30+
31+## Equal provider treatment
32+All three providers share the same visual structure in the popover:
33+1. Section header with icon + plan label
34+2. Headline number (26pt subheroFont) with emoji, color-coded, click-to-copy, trend arrow
35+3. Reset countdown text (Claude/z.ai windows only — Hyper has no windows)
36+4. Bars under EVERY window (5-hour, 7-day, weekly — all get bars)
37+5. Sparkline (each provider tracks its own history via `MetricPoint` arrays)
38+6. Error row if applicable
39+
40+Notifications fire on threshold crossings:
41+- Hyper: balance drops below 10 credits
42+- Claude: 5-hour % crosses above 90%
43+- z.ai: 5-hour % crosses above 90%
44+
45+If a provider isn't configured, its section simply doesn't appear — no placeholder,
46+no error. This is already the case for Claude and z.ai; Hyper now behaves the same
47+way (gated on `hyperConfigured`).
20
48
21## Claude credentials (strictly read-only)
49## Claude credentials (strictly read-only)
22`ClaudeUsageClient` reads the credentials **Claude Code owns**: its `Claude Code-credentials`
50`ClaudeUsageClient` reads the credentials **Claude Code owns**: its `Claude Code-credentials`
@@ -47,9 +75,9 @@ Keychain item (via `/usr/bin/security`), falling back to `~/.claude/.credentials
47- Swift 5.9, macOS 14.0 deployment target
75- Swift 5.9, macOS 14.0 deployment target
48
76
49## Conventions
77## Conventions
50−- Clean minimal UI — big hero number, generous spacing, restrained color, no chrome
78+- Clean minimal UI — subhero numbers, generous spacing, restrained color, no chrome
51- No card-heavy/multi-material/gradient-rich designs
79- No card-heavy/multi-material/gradient-rich designs
52−- Rounded font family at one scale (hero/subhero/section/control/caption/footnote)
80+- Rounded font family at one scale (subhero/section/control/caption/footnote)
53- Monospaced digits for numbers that change
81- Monospaced digits for numbers that change
54- Tests use StubURLProtocol for network mocking, FakeKeychain for keychain mocking,
82- Tests use StubURLProtocol for network mocking, FakeKeychain for keychain mocking,
55 FakeClaudeCredentialStore for Claude credentials — never touch the real ones in a test
83 FakeClaudeCredentialStore for Claude credentials — never touch the real ones in a test
@@ -57,4 +85,4 @@ Keychain item (via `/usr/bin/security`), falling back to `~/.claude/.credentials
57
85
58## VCS
86## VCS
59- jj colocated with git
87- jj colocated with git
60−- Push directly to master — no PRs for this repo
88+- Push directly to master — no PRs for this repo
README.md+64−48
@@ -4,24 +4,33 @@
4
4
5# UsageMenubar
5# UsageMenubar
6
6
7−A macOS menu bar application that displays your [Hyper (Charm)](https://hyper.charm.land) credit balance — and, if you use the Claude Code CLI, your Claude subscription limits alongside it.
7+A macOS menu bar application that displays usage limits and credit balances for multiple LLM services: [Hyper (Charm)](https://hyper.charm.land), [Claude Code](https://claude.com/claude-code), and [z.ai](https://z.ai) Coding Plans. Each service is independent — configure the ones you use, ignore the ones you don't.
8
8
9## Features
9## Features
10
10
11−- ⚡ **Menu bar display** — shows `⚡{balance}` (e.g. `⚡42`) right in your menu bar
11+- ⚡ **Hyper credits** — balance from `hyper.charm.land`, with sparkline trend, click-to-copy, and low-balance notifications
12−- 🤖 **Claude Code limits** — your Pro/Max 5-hour and 7-day usage windows in the popover, with the percentage used and a countdown to the next reset. No setup: if you have signed in with the Claude Code CLI, it just appears. If you haven't, the section doesn't
12+- ⁕ **Claude Code limits** — Pro/Max 5-hour and 7-day usage windows, with countdown to next reset. No setup: if you've signed in with the Claude Code CLI, it just appears
13+- Z **z.ai Coding Plan** — 5-hour and weekly token quota percentages, with reset countdowns. Enter your API key in settings
13−- 🔄 **Auto-refresh** — fetches your balance on a configurable interval (1m / 5m / 15m / 30m), and again on wake from sleep
14+- 🔄 **Auto-refresh** — configurable interval (1m / 5m / 15m / 30m), plus refresh on wake from sleep
14−- 📊 **Balance sparkline** — a subtle mini-chart in the popover shows your balance trend over recent fetches
15−- 📋 **Click-to-copy** — click the balance number in the popover to copy it to the clipboard
16−- 🖱️ **Right-click menu** — right-click the menu bar icon for quick access to Refresh, Open hyper.charm.land, and Quit
15+- 🖱️ **Right-click menu** — quick access to Refresh and Quit
17−- 🔔 **Low-balance notifications** — alerts you when your balance drops below 10 credits, including on the very first fetch if you are already below the threshold
18−- 🔒 **Keychain storage** — your API key is stored securely in macOS Keychain
16+- 🔒 **Keychain storage** — API keys stored securely in macOS Keychain
19−- 🎨 **Color-coded balance** — green (≥100), yellow (10–99), red (<10)
20−- 🔐 **Secure input** — API key field is masked
21- 🚀 **Launch at login** — optional, using `SMAppService`
17- 🚀 **Launch at login** — optional, using `SMAppService`
22−- 🔑 **Get key link** — quick link to [hyper.charm.land](https://hyper.charm.land) right next to the API key field
23- 🫥 **Agent app** — no dock icon, no main window; lives entirely in the menu bar
18- 🫥 **Agent app** — no dock icon, no main window; lives entirely in the menu bar
24
19
20+## Menu bar
21+
22+Each service shows an emoji-prefixed segment when configured and above 0%:
23+
24+| Service | Emoji | Example |
25+|---------|-------|---------|
26+| Hyper balance | ⚡ | `⚡42` |
27+| Claude 5-hour | ⁕ | `⁕62%` |
28+| Claude 7-day | ⁕ | `⁕8%` |
29+| z.ai 5-hour | Z | `Z12%` |
30+| z.ai weekly | Z | `Z3%` |
31+
32+Unconfigured services don't appear at all — no placeholder, no icon.
33+
25## Installation
34## Installation
26
35
27### Homebrew (recommended)
36### Homebrew (recommended)
@@ -42,37 +51,47 @@ brew install --cask ilyakooo0/tap/usage-menubar
421. Go to the [Releases page](https://github.com/ilyakooo0/usage-menubar/releases)
511. Go to the [Releases page](https://github.com/ilyakooo0/usage-menubar/releases)
432. Download the latest `UsageMenubar-*.zip`
522. Download the latest `UsageMenubar-*.zip`
443. Unzip and move `UsageMenubar.app` to `/Applications`
533. Unzip and move `UsageMenubar.app` to `/Applications`
45−4. Launch the app — it will appear in your menu bar as ⚡
54+4. Launch the app — it will appear in your menu bar
46
55
47> **Note:** Releases are **ad-hoc signed** — the CI build applies `codesign -s -` to the app, which gives it a stable code identity but is *not* an Apple Developer ID signature. macOS will still warn you on first launch: right-click the app → **Open**, or allow it in **System Settings → Privacy & Security**. Removing that prompt entirely requires a paid Developer ID certificate and notarization.
56> **Note:** Releases are **ad-hoc signed** — the CI build applies `codesign -s -` to the app, which gives it a stable code identity but is *not* an Apple Developer ID signature. macOS will still warn you on first launch: right-click the app → **Open**, or allow it in **System Settings → Privacy & Security**. Removing that prompt entirely requires a paid Developer ID certificate and notarization.
48
57
49−### Set your API key
58+## Setup
59+
60+Click the menu bar icon to open the popover. Each service is configured independently:
50
61
51−1. Click the ⚡ icon in your menu bar
62+### Hyper credits
52−2. Enter your Hyper API key (starts with `sk-...`) in the **API Key** field
53−3. Click **Save**
54−4. Your balance will appear immediately
55
63
56−You can get your API key from [hyper.charm.land](https://hyper.charm.land).
64+1. Enter your Hyper API key (starts with `sk-...`) in the **Hyper API Key** field
65+2. Click **Save**
66+3. Your balance appears immediately
57
67
58−### Claude Code limits (optional, no setup)
68+Get your key from [hyper.charm.land](https://hyper.charm.land).
59
69
60−If you use the [Claude Code CLI](https://claude.com/claude-code), the popover also shows how much of your Pro/Max plan you have used: the limit currently in effect as a headline percentage, the 5-hour and 7-day windows underneath, and how long until the next reset.
70+### Claude Code limits (no setup)
61
71
62−There is nothing to configure. The app reads the OAuth credentials Claude Code already stores when you run `claude` and sign in — first from its login-Keychain item, then from `~/.claude/.credentials.json` if the Keychain has nothing. If neither exists, the Claude section simply doesn't appear; it is not treated as an error, and the Hyper balance is unaffected.
72+If you use the [Claude Code CLI](https://claude.com/claude-code), the popover shows how much of your Pro/Max plan you have used: the 5-hour and 7-day windows, and how long until the next reset.
63
73
64−Two things worth knowing:
74+There is nothing to configure. The app reads the OAuth credentials Claude Code already stores when you run `claude` and sign in — first from its login-Keychain item, then from `~/.claude/.credentials.json` if the Keychain has nothing. If neither exists, the Claude section doesn't appear.
65
75
66- **macOS will ask once.** Those credentials belong to Claude Code, not to this app, so the first read prompts you to allow access to the `Claude Code-credentials` Keychain item. Choose **Always Allow** and you won't be asked again.
76- **macOS will ask once.** Those credentials belong to Claude Code, not to this app, so the first read prompts you to allow access to the `Claude Code-credentials` Keychain item. Choose **Always Allow** and you won't be asked again.
67−- **The credentials are strictly read-only.** The app reads the access token Claude Code stored and spends it as-is. It never refreshes it, and never writes those credentials back — that is the CLI's job alone. (Refreshing rotates the refresh token, so an app doing it behind the CLI's back would race it and sign you out of your own terminal.) If the token has expired, the Claude section says so and asks you to run `claude`; once the CLI has refreshed on its own, the app picks the new token up on its next fetch.
77+- **The credentials are strictly read-only.** The app reads the access token Claude Code stored and spends it as-is. It never refreshes it, and never writes those credentials back — that is the CLI's job alone. If the token has expired, the Claude section says so and asks you to run `claude`.
78+
79+### z.ai Coding Plan
80+
81+1. Enter your z.ai API key in the **z.ai API Key** field
82+2. Click **Save**
83+3. Your 5-hour and weekly quota percentages appear immediately
84+
85+Get your key from [z.ai/manage-apikey](https://z.ai/manage-apikey/coding-plan/personal/my-plan).
68
86
69## Settings
87## Settings
70
88
71−Everything is configured from the popover — click the ⚡ icon in the menu bar.
89+Everything is configured from the popover — click the menu bar icon.
72
90
73| Setting | Where | Notes |
91| Setting | Where | Notes |
74| ------------------- | ------------------------------ | ------------------------------------------------------------------------------------ |
92| ------------------- | ------------------------------ | ------------------------------------------------------------------------------------ |
75−| **API key** | **API Key** field → **Save** | Stored in the macOS Keychain. Clearing the field and saving deletes the stored key. |
93+| **Hyper API key** | **Hyper API Key** field → **Save** | Stored in the macOS Keychain. Clearing the field and saving deletes the stored key. |
94+| **z.ai API key** | **z.ai API Key** field → **Save** | Stored separately from the Hyper key. |
76| **Launch at login** | **Launch at Login** toggle | Registers the app with `SMAppService`. |
95| **Launch at login** | **Launch at Login** toggle | Registers the app with `SMAppService`. |
77| **Notifications** | System permission prompt | macOS asks on first launch (only if not already decided). Required for low-balance alerts. |
96| **Notifications** | System permission prompt | macOS asks on first launch (only if not already decided). Required for low-balance alerts. |
78| **Refresh now** | **Refresh** button | Forces an immediate fetch, on top of the automatic refresh. |
97| **Refresh now** | **Refresh** button | Forces an immediate fetch, on top of the automatic refresh. |
@@ -138,18 +157,21 @@ usage-menubar/
138├── project.yml # xcodegen project definition
157├── project.yml # xcodegen project definition
139├── UsageMenubar/
158├── UsageMenubar/
140│ ├── UsageMenubarApp.swift # @main App, NSStatusBar, timer, sleep/wake
159│ ├── UsageMenubarApp.swift # @main App, NSStatusBar, timer, sleep/wake
141−│ ├── ViewModel.swift # Observable state: balance, Claude usage, refresh, notifications
160+│ ├── ViewModel.swift # Observable state: balance, Claude usage, z.ai usage, refresh
142│ ├── MenuView.swift # SwiftUI popover view
161│ ├── MenuView.swift # SwiftUI popover view
143−│ ├── CreditsChecker.swift # API client (GET /v1/credits)
162+│ ├── CreditsChecker.swift # Hyper API client (GET /v1/credits)
144│ ├── ClaudeUsageClient.swift # Claude OAuth client (GET /api/oauth/usage) + models
163│ ├── ClaudeUsageClient.swift # Claude OAuth client (GET /api/oauth/usage) + models
164+│ ├── ZaiUsageClient.swift # z.ai API client (GET /monitor/usage/quota/limit)
145−│ ├── KeychainHelper.swift # Keychain wrapper for API key
165+│ ├── KeychainHelper.swift # Keychain wrapper (Hyper + z.ai keys)
146│ ├── VersionFormatter.swift # YYYY.MM.DD.HHHH formatting
166│ ├── VersionFormatter.swift # YYYY.MM.DD.HHHH formatting
147│ ├── Info.plist # LSUIElement=true (agent app)
167│ ├── Info.plist # LSUIElement=true (agent app)
148−│ ├── UsageMenubar.entitlements # Network client; deliberately not sandboxed
168+│ ├── UsageMenubar.entitlements # Network client; deliberately not sandboxed
149│ └── Assets.xcassets/ # App icon, accent color
169│ └── Assets.xcassets/ # App icon, accent color
150├── UsageMenubarTests/
170├── UsageMenubarTests/
151−│ ├── CreditsCheckerTests.swift # API, retry, fractional balance, keychain tests
171+│ ├── CreditsCheckerTests.swift # Hyper API, retry, keychain, StubURLProtocol
152│ ├── ClaudeUsageClientTests.swift # Usage decoding, read-only credential store, retry
172│ ├── ClaudeUsageClientTests.swift # Usage decoding, read-only credential store, retry
173+│ ├── ZaiUsageClientTests.swift # z.ai API, retry, decoding
174+│ ├── StatusBarTextTests.swift # Menu bar title formatting
153│ └── VersionFormatterTests.swift # Version formatting tests
175│ └── VersionFormatterTests.swift # Version formatting tests
154├── .github/workflows/
176├── .github/workflows/
155│ └── release.yml # Build + release on push to master
177│ └── release.yml # Build + release on push to master
@@ -160,24 +182,18 @@ usage-menubar/
160
182
161## How it works
183## How it works
162
184
163−1. **Menu bar item**: The app creates an `NSStatusItem` with a variable-length button showing `⚡{balance}` or `⚡?`. Left-click toggles the popover; right-click opens a context menu (Refresh, Open hyper.charm.land, Quit).
185+1. **Menu bar item**: The app creates an `NSStatusItem` with a variable-length button. Left-click toggles the popover; right-click opens a context menu (Refresh, Quit).
164−2. **Popover**: Clicking the menu bar item shows an `NSPopover` containing a SwiftUI `MenuView` with the balance, sparkline, API key field, and settings.
186+2. **Popover**: Clicking the menu bar item shows an `NSPopover` containing a SwiftUI `MenuView` with each configured service's data and the settings.
187+3. **Independent fetches**: `ViewModel.refresh()` runs the Hyper, Claude, and z.ai requests concurrently with `async let`. Any can fail, stall, or be unconfigured without affecting the others.
165−3. **API client**: `CreditsChecker` performs `GET https://hyper.charm.land/v1/credits` with `Authorization: Bearer <key>`, retries transient failures (5xx, timeouts) with exponential backoff, and decodes `{"balance": <int>}` (tolerating fractional values by rounding).
188+4. **Hyper API client**: `CreditsChecker` performs `GET https://hyper.charm.land/v1/credits` with `Authorization: Bearer <key>`, retries transient failures with exponential backoff, and decodes `{"balance": <int>}`.
189+5. **Claude usage**: `ClaudeUsageClient` reads Claude Code's OAuth credentials (its `Claude Code-credentials` Keychain item via `/usr/bin/security`, falling back to `~/.claude/.credentials.json`) and fetches `GET https://api.anthropic.com/api/oauth/usage`. It is strictly read-only — it never refreshes the token and never writes credentials back. Missing credentials are reported as "not configured" rather than as an error, and the popover hides the section.
190+6. **z.ai usage**: `ZaiUsageClient` fetches `GET https://api.z.ai/api/monitor/usage/quota/limit` and `GET https://api.z.ai/api/biz/subscription/list` concurrently with an API key the user enters in settings. Same retry pattern as the other clients.
166−4. **Keychain**: The API key is stored/retrieved via `KeychainHelper` using the macOS Security framework. `save()` returns a `Bool` indicating success.
191+7. **Keychain**: API keys are stored/retrieved via `KeychainHelper` using the macOS Security framework. Hyper and z.ai keys are stored separately.
167−5. **Timer**: A `Timer` fires at the user-selected interval (1m / 5m / 15m / 30m, default 5m) to refresh the balance; the app also refreshes on wake from sleep. Changing the interval in the popover restarts the timer immediately.
192+8. **Timer**: A `Timer` fires at the user-selected interval (1m / 5m / 15m / 30m, default 5m); the app also refreshes on wake from sleep.
168−6. **State**: `ViewModel` owns the balance, loading/error state, balance history (for the sparkline), refresh task (cancellable to prevent races), and the refresh interval, and is shared between the menu bar item and the popover.
169−7. **Notifications**: When the balance crosses below 10 credits — or is already below 10 on the first successful fetch — `ViewModel` posts a local `UNUserNotification`. Permission is requested only once.
193+9. **Notifications**: When the Hyper balance crosses below 10 credits — or is already below 10 on the first successful fetch — `ViewModel` posts a local `UNUserNotification`. Permission is requested only once.
170−8. **Agent app**: `LSUIElement=true` in `Info.plist` makes the app a background agent — no dock icon, no main window.
194+10. **Agent app**: `LSUIElement=true` in `Info.plist` makes the app a background agent — no dock icon, no main window.
171−9. **Claude usage**: `ClaudeUsageClient` reads Claude Code's OAuth credentials (its `Claude Code-credentials` Keychain item via `/usr/bin/security`, falling back to `~/.claude/.credentials.json`) and fetches `GET https://api.anthropic.com/api/oauth/usage` with `Authorization: Bearer` and `anthropic-beta: oauth-2025-04-20`. It is strictly read-only — it never refreshes the token and never writes credentials back. The token endpoint rotates the refresh token, so an app that also refreshed would race the CLI and sign the user out of their own terminal. When the token is spent (401/403), the app says so and points at `claude`; the next fetch (every 1–5 min, credentials re-read each time) picks up whatever the CLI refreshed on its own. Missing credentials are reported as "not configured" rather than as an error, and the popover hides the section.
172−10. **Independent fetches**: `ViewModel.refresh()` runs the Hyper and Claude requests concurrently with `async let`. Neither can fail, stall, or be unconfigured in a way that affects the other.
17311. **No sandbox**: Reading credentials that belong to another app means spawning `/usr/bin/security` and reading a file in the user's home directory, neither of which the App Sandbox permits. The app is distributed ad-hoc signed via Homebrew rather than through the App Store, where the sandbox would be mandatory.
19511. **No sandbox**: Reading credentials that belong to another app means spawning `/usr/bin/security` and reading a file in the user's home directory, neither of which the App Sandbox permits. The app is distributed ad-hoc signed via Homebrew rather than through the App Store, where the sandbox would be mandatory.
174
196
175−## Credits
176−
177−- **Author**: Ilya Koo ([@ilyakooo0](https://github.com/ilyakooo0))
178−- **API**: [Hyper by Charm](https://hyper.charm.land)
179−- **Built with**: Swift 5.9, SwiftUI, macOS 14+
180−
181## License
197## License
182
198
183−MIT — see [LICENSE](LICENSE).
199+MIT — see [LICENSE](LICENSE).
UsageMenubar/Info.plist+1−1
@@ -7,7 +7,7 @@
7 <key>CFBundleDisplayName</key>
7 <key>CFBundleDisplayName</key>
8 <string>Usage Menubar</string>
8 <string>Usage Menubar</string>
9 <key>CFBundleIdentifier</key>
9 <key>CFBundleIdentifier</key>
10− <string>com.ilyakooo0.usage-menubar</string>
10+ <string>soy.iko.usage-menubar</string>
11 <key>CFBundleVersion</key>
11 <key>CFBundleVersion</key>
12 <string>1</string>
12 <string>1</string>
13 <key>CFBundleShortVersionString</key>
13 <key>CFBundleShortVersionString</key>
UsageMenubar/KeychainHelper.swift+2−2
@@ -82,7 +82,7 @@ struct KeychainStore {
82
82
83/// Simple wrapper around the macOS Keychain for storing the Hyper API key.
83/// Simple wrapper around the macOS Keychain for storing the Hyper API key.
84enum KeychainHelper {
84enum KeychainHelper {
85− private static let service = "com.ilyakooo0.usage-menubar"
85+ private static let service = "soy.iko.usage-menubar"
86 private static let account = "hyper-api-key"
86 private static let account = "hyper-api-key"
87
87
88 private static let store = KeychainStore(service: service, account: account)
88 private static let store = KeychainStore(service: service, account: account)
@@ -112,7 +112,7 @@ enum KeychainHelper {
112/// Uses the same keychain service as the Hyper key but a different account, so the
112/// Uses the same keychain service as the Hyper key but a different account, so the
113/// two keys coexist without clobbering each other.
113/// two keys coexist without clobbering each other.
114enum ZaiKeychainHelper {
114enum ZaiKeychainHelper {
115− private static let service = "com.ilyakooo0.usage-menubar"
115+ private static let service = "soy.iko.usage-menubar"
116 private static let account = "zai-api-key"
116 private static let account = "zai-api-key"
117
117
118 private static let store = KeychainStore(service: service, account: account)
118 private static let store = KeychainStore(service: service, account: account)
UsageMenubar/MenuView.swift+225−172
@@ -4,21 +4,15 @@ import SwiftUI
4struct MenuView: View {
4struct MenuView: View {
5 @ObservedObject var viewModel: ViewModel
5 @ObservedObject var viewModel: ViewModel
6
6
7− /// Dims the balance while the pointer is over it, hinting that it is clickable.
8− @State private var isHoveringBalance = false
9−
10 private static let websiteURL = URL(string: "https://hyper.charm.land")!
7 private static let websiteURL = URL(string: "https://hyper.charm.land")!
11 private static let zaiKeyURL = URL(string: "https://z.ai/manage-apikey/coding-plan/personal/my-plan")!
8 private static let zaiKeyURL = URL(string: "https://z.ai/manage-apikey/coding-plan/personal/my-plan")!
12
9
13 // MARK: - Type Scale
10 // MARK: - Type Scale
14
11
15− /// One rounded family, sized so that the balance is the only large element and
12+ /// One rounded family, sized so that every provider's headline number reads at the
16− /// everything else steps down from it — nothing competes with it for attention.
13+ /// same weight. With three providers in the popover, the 26pt subhero is the right
17− /// The Claude percentage is the one exception, and it is deliberately set well
14+ /// scale — a 46pt hero would make the popover too tall. Monospaced digits keep
18− /// below the balance so it reads as the second thing on the page, not the first.
19− /// Monospaced digits keep numbers from jittering as their digits change.
15+ /// numbers from jittering as their digits change.
20− private static let heroFont = Font.system(size: 46, weight: .semibold, design: .rounded)
21− .monospacedDigit()
22 private static let subheroFont = Font.system(size: 26, weight: .semibold, design: .rounded)
16 private static let subheroFont = Font.system(size: 26, weight: .semibold, design: .rounded)
23 .monospacedDigit()
17 .monospacedDigit()
24 private static let sectionFont = Font.system(size: 12, weight: .semibold, design: .rounded)
18 private static let sectionFont = Font.system(size: 12, weight: .semibold, design: .rounded)
@@ -28,7 +22,9 @@ struct MenuView: View {
28
22
29 var body: some View {
23 var body: some View {
30 VStack(alignment: .leading, spacing: 18) {
24 VStack(alignment: .leading, spacing: 18) {
25+ if viewModel.hyperConfigured {
31− hyper
26+ hyper
27+ }
32 if showsClaude {
28 if showsClaude {
33 serviceBreak
29 serviceBreak
34 claude
30 claude
@@ -51,10 +47,14 @@ struct MenuView: View {
51 .background(Color(nsColor: .windowBackgroundColor))
47 .background(Color(nsColor: .windowBackgroundColor))
52 .animation(.easeInOut(duration: 0.2), value: viewModel.savedConfirmation)
48 .animation(.easeInOut(duration: 0.2), value: viewModel.savedConfirmation)
53 .animation(.easeInOut(duration: 0.2), value: viewModel.copiedConfirmation)
49 .animation(.easeInOut(duration: 0.2), value: viewModel.copiedConfirmation)
50+ .animation(.easeInOut(duration: 0.2), value: viewModel.copiedClaudeConfirmation)
51+ .animation(.easeInOut(duration: 0.2), value: viewModel.copiedZaiConfirmation)
54 .animation(.easeInOut(duration: 0.2), value: viewModel.errorMessage)
52 .animation(.easeInOut(duration: 0.2), value: viewModel.errorMessage)
55 .animation(.easeInOut(duration: 0.25), value: viewModel.balance)
53 .animation(.easeInOut(duration: 0.25), value: viewModel.balance)
56 .animation(.easeInOut(duration: 0.25), value: viewModel.isLoading)
54 .animation(.easeInOut(duration: 0.25), value: viewModel.isLoading)
57− .animation(.easeInOut(duration: 0.3), value: historyValues.count)
55+ .animation(.easeInOut(duration: 0.3), value: viewModel.balanceSparklineValues.count)
56+ .animation(.easeInOut(duration: 0.3), value: viewModel.claudeSparklineValues.count)
57+ .animation(.easeInOut(duration: 0.3), value: viewModel.zaiSparklineValues.count)
58 .animation(.easeInOut(duration: 0.25), value: viewModel.claudeUsage)
58 .animation(.easeInOut(duration: 0.25), value: viewModel.claudeUsage)
59 .animation(.easeInOut(duration: 0.2), value: viewModel.claudeError)
59 .animation(.easeInOut(duration: 0.2), value: viewModel.claudeError)
60 .animation(.easeInOut(duration: 0.25), value: viewModel.zaiUsage)
60 .animation(.easeInOut(duration: 0.25), value: viewModel.zaiUsage)
@@ -66,15 +66,14 @@ struct MenuView: View {
66 Divider().opacity(0.3)
66 Divider().opacity(0.3)
67 }
67 }
68
68
69− /// The rule between the two services. Wider than the others on purpose: Hyper and
69+ /// The rule between services. Wider than the others on purpose: the providers are
70− /// Claude are two unrelated accounts, and the gap should say so before the headers do.
70+ /// unrelated accounts, and the gap should say so before the headers do.
71 private var serviceBreak: some View {
71 private var serviceBreak: some View {
72 hairline.padding(.vertical, 5)
72 hairline.padding(.vertical, 5)
73 }
73 }
74
74
75 /// A section header: an icon and a title, sized to match the "API Key" header below
75 /// A section header: an icon and a title, sized to match the "API Key" header below
76− /// so the popover reads as three sections rather than a hero with two appendices.
76+ /// so the popover reads as three equal sections rather than a hero with appendages.
77− /// The icon is what tells the two services apart at a glance.
78 private func sectionHeader(_ title: String, systemImage: String) -> some View {
77 private func sectionHeader(_ title: String, systemImage: String) -> some View {
79 HStack(spacing: 5) {
78 HStack(spacing: 5) {
80 Image(systemName: systemImage)
79 Image(systemName: systemImage)
@@ -85,113 +84,62 @@ struct MenuView: View {
85 }
84 }
86 }
85 }
87
86
88− // MARK: - Hyper
87+ /// A reusable headline number row: the provider's main metric at subhero scale,
89−
88+ /// with emoji, color, trend arrow, and click-to-copy. The `caption` doubles as the
90− /// The Hyper section: its header, then the balance as the popover's hero.
89+ /// copy-confirmation slot so the layout doesn't shift.
90+ private func headlineNumber(
91+ text: String,
92+ color: Color,
93+ trend: MetricTrend,
94+ copiedConfirmation: Bool,
95+ captionText: String,
96+ copyAction: @escaping () -> Void
91− private var hyper: some View {
97+ ) -> some View {
92− VStack(alignment: .leading, spacing: 12) {
98+ VStack(spacing: 6) {
93− sectionHeader("Hyper Credits", systemImage: "bolt.fill")
99+ Button(action: copyAction) {
94−
100+ HStack(alignment: .firstTextBaseline, spacing: 6) {
95− hero
96− }
97− }
98−
99− private var hero: some View {
100− VStack(spacing: 8) {
101− if viewModel.isLoading && viewModel.balance == nil {
102− PulsingBolt()
103− .transition(.opacity)
104− } else if viewModel.balance == nil {
105− VStack(spacing: 6) {
106− Text("⚡?")
101+ Text(text)
107− .font(Self.heroFont)
102+ .font(Self.subheroFont)
108− .foregroundColor(.secondary)
103+ .foregroundColor(color)
104+ .contentTransition(.opacity)
109
105
110− Text("Enter your API key below")
106+ if let symbol = trend.symbolName {
111− .font(Self.captionFont)
107+ Image(systemName: symbol)
108+ .font(.system(size: 12, weight: .semibold))
112− .foregroundColor(.secondary)
109+ .foregroundColor(.secondary)
113− }
114− .transition(.opacity)
110+ .transition(.opacity)
115− } else {
111+ }
116− VStack(spacing: 6) {
117− balanceNumber
118− caption
119− sparkline
120 }
112 }
121− .transition(.opacity)
113+ .contentShape(Rectangle())
122 }
114 }
123−
115+ .buttonStyle(.plain)
124− if let lastUpdatedText = viewModel.lastUpdatedText {
116+ .help("Click to copy")
125− Text(lastUpdatedText)
117+
126− .font(Self.footnoteFont)
118+ // Caption doubles as copy confirmation so the layout doesn't shift.
127− .foregroundColor(.secondary)
119+ Group {
128− .transition(.opacity)
120+ if copiedConfirmation {
129− }
121+ Text("✓ Copied")
130−
122+ .foregroundColor(.green)
131− if let error = viewModel.errorMessage {
123+ } else {
132− errorRow(error)
124+ Text(captionText)
133− .transition(.opacity)
134− }
135− }
136− .frame(maxWidth: .infinity)
137− }
138−
139− /// The balance itself. Clicking it copies the raw number to the clipboard.
140− private var balanceNumber: some View {
141− Button(action: { viewModel.copyBalanceToClipboard() }) {
142− HStack(alignment: .firstTextBaseline, spacing: 6) {
143− Text(viewModel.formattedBalance)
144− .font(Self.heroFont)
145− .foregroundColor(viewModel.balanceColor)
146− .contentTransition(.opacity)
147−
148− if let symbol = viewModel.balanceTrend.symbolName {
149− Image(systemName: symbol)
150− .font(.system(size: 12, weight: .semibold))
151 .foregroundColor(.secondary)
125 .foregroundColor(.secondary)
152− .transition(.opacity)
153 }
126 }
154 }
127 }
155− // Without this the hover area is only the glyphs themselves.
128+ .font(Self.captionFont)
156− .contentShape(Rectangle())
157− }
158− .buttonStyle(.plain)
159− .disabled(viewModel.balance == nil)
160− .help("Click to copy")
161− .opacity(isHoveringBalance && viewModel.balance != nil ? 0.65 : 1)
162− .onHover { isHoveringBalance = $0 }
163− .animation(.easeInOut(duration: 0.12), value: isHoveringBalance)
164− }
165−
166− /// Doubles as the copy confirmation so the layout doesn't shift.
167− private var caption: some View {
168− Group {
169− if viewModel.copiedConfirmation {
170− Text("✓ Copied")
171− .foregroundColor(.green)
172− } else {
173− Text("credits")
174− .foregroundColor(.secondary)
175− }
176 }
129 }
177− .font(Self.captionFont)
178 }
130 }
179
131
132+ /// A sparkline shown when there are at least 2 history points.
180 @ViewBuilder
133 @ViewBuilder
181− private var sparkline: some View {
134+ private func sparkline(values: [Int], color: Color) -> some View {
182− if historyValues.count >= 2 {
135+ if values.count >= 2 {
183− Sparkline(values: historyValues, color: viewModel.balanceColor)
136+ Sparkline(values: values, color: color)
184 .frame(height: 26)
137 .frame(height: 26)
185 .padding(.top, 4)
138 .padding(.top, 4)
186 .transition(.opacity)
139 .transition(.opacity)
187 }
140 }
188 }
141 }
189
142
190− /// The sparkline's data. Key paths can't index tuple elements, so this stays a closure.
191− private var historyValues: [Int] {
192− viewModel.balanceHistory.map { $0.balance }
193− }
194−
195 /// Errors read as a quiet inline notice rather than loose red text: an icon, the
143 /// Errors read as a quiet inline notice rather than loose red text: an icon, the
196 /// message, and a tinted rounded background with no border.
144 /// message, and a tinted rounded background with no border.
197 private func errorRow(_ message: String) -> some View {
145 private func errorRow(_ message: String) -> some View {
@@ -216,6 +164,56 @@ struct MenuView: View {
216 .padding(.top, 2)
164 .padding(.top, 2)
217 }
165 }
218
166
167+ // MARK: - Hyper
168+
169+ /// Whether there is anything to say about Hyper. Nothing at all when Hyper isn't
170+ /// configured — the section doesn't appear, same as Claude and z.ai.
171+ private var showsHyper: Bool {
172+ viewModel.hyperConfigured
173+ }
174+
175+ /// The Hyper section: header, balance as a subhero number, caption, sparkline.
176+ private var hyper: some View {
177+ VStack(alignment: .leading, spacing: 12) {
178+ sectionHeader("Hyper Credits", systemImage: "bolt.fill")
179+
180+ if viewModel.isLoading && viewModel.balance == nil {
181+ Text("…")
182+ .font(Self.subheroFont)
183+ .foregroundColor(.secondary)
184+ } else if let balance = viewModel.balance {
185+ VStack(spacing: 8) {
186+ headlineNumber(
187+ text: "⚡ \(viewModel.formattedBalance)",
188+ color: viewModel.balanceColor,
189+ trend: viewModel.balanceTrend,
190+ copiedConfirmation: viewModel.copiedConfirmation,
191+ captionText: "credits",
192+ copyAction: { viewModel.copyBalanceToClipboard() }
193+ )
194+
195+ sparkline(
196+ values: viewModel.balanceSparklineValues,
197+ color: viewModel.balanceColor
198+ )
199+ }
200+ .transition(.opacity)
201+ }
202+
203+ if let lastUpdatedText = viewModel.lastUpdatedText {
204+ Text(lastUpdatedText)
205+ .font(Self.footnoteFont)
206+ .foregroundColor(.secondary)
207+ .transition(.opacity)
208+ }
209+
210+ if let error = viewModel.errorMessage {
211+ errorRow(error)
212+ .transition(.opacity)
213+ }
214+ }
215+ }
216+
219 // MARK: - Claude
217 // MARK: - Claude
220
218
221 /// Whether there is anything to say about Claude. Nothing at all when Claude Code
219 /// Whether there is anything to say about Claude. Nothing at all when Claude Code
@@ -227,8 +225,8 @@ struct MenuView: View {
227 return false
225 return false
228 }
226 }
229
227
230− /// Claude Code's plan limits. The percentage leads, as the balance does above it,
228+ /// Claude Code's plan limits. Same visual structure as Hyper and z.ai: a subhero
231− /// with the individual windows underneath as quiet supporting detail.
229+ /// headline number, bars under every window, a sparkline, and error row.
232 private var claude: some View {
230 private var claude: some View {
233 VStack(alignment: .leading, spacing: 12) {
231 VStack(alignment: .leading, spacing: 12) {
234 HStack(spacing: 8) {
232 HStack(spacing: 8) {
@@ -255,23 +253,31 @@ struct MenuView: View {
255
253
256 private func usageDetail(_ usage: ClaudeUsage) -> some View {
254 private func usageDetail(_ usage: ClaudeUsage) -> some View {
257 VStack(alignment: .leading, spacing: 12) {
255 VStack(alignment: .leading, spacing: 12) {
258− // The 5-hour window leads as the hero number, since it's the one
256+ // The 5-hour window leads as the headline number.
259− // most likely to bind and the one shown in the menu bar title.
260 if let fiveHour = usage.fiveHour {
257 if let fiveHour = usage.fiveHour {
261 let percent = Int(fiveHour.utilization.rounded())
258 let percent = Int(fiveHour.utilization.rounded())
259+ VStack(spacing: 8) {
262− HStack(alignment: .firstTextBaseline, spacing: 8) {
260+ HStack(alignment: .firstTextBaseline, spacing: 8) {
261+ headlineNumber(
263− Text("🕐 \(percent)%")
262+ text: "⁕ \\(percent)%",
264− .font(Self.subheroFont)
265− .foregroundColor(Self.usageColor(percent))
263+ color: Self.usageColor(percent),
266− .contentTransition(.opacity)
264+ trend: viewModel.claudeTrend,
265+ copiedConfirmation: viewModel.copiedClaudeConfirmation,
266+ captionText: "5-hour usage",
267+ copyAction: { viewModel.copyClaudePercentToClipboard() }
268+ )
267
269
268− Spacer(minLength: 0)
270+ Spacer(minLength: 0)
269
271
270− if let resets = fiveHour.resetsInFormatted {
272+ if let resets = fiveHour.resetsInFormatted {
271− Text("resets in \(resets)")
273+ Text("resets in \(resets)")
272− .font(Self.footnoteFont)
274+ .font(Self.footnoteFont)
273− .foregroundColor(.secondary)
275+ .foregroundColor(.secondary)
276+ }
274 }
277 }
278+
279+ // Bar under the 5-hour window too, same as 7-day windows below.
280+ UsageBar(percent: percent, color: Self.usageColor(percent))
275 }
281 }
276 }
282 }
277
283
@@ -285,6 +291,12 @@ struct MenuView: View {
285 window("7-day Opus", opus)
291 window("7-day Opus", opus)
286 }
292 }
287 }
293 }
294+
295+ // Sparkline for the Claude 5-hour % history.
296+ sparkline(
297+ values: viewModel.claudeSparklineValues,
298+ color: Self.usageColor(claudeFiveHourPercent ?? 0)
299+ )
288 }
300 }
289 }
301 }
290
302
@@ -310,6 +322,11 @@ struct MenuView: View {
310 }
322 }
311 }
323 }
312
324
325+ /// The current Claude 5-hour percentage, or 0 if absent. Used for sparkline color.
326+ private var claudeFiveHourPercent: Int? {
327+ viewModel.claudeFiveHourPercent
328+ }
329+
313 /// The balance colors read the other way round here, because for a limit more is
330 /// The balance colors read the other way round here, because for a limit more is
314 /// worse: green with room to spare, yellow closing in, red nearly spent.
331 /// worse: green with room to spare, yellow closing in, red nearly spent.
315 private static func usageColor(_ percent: Int) -> Color {
332 private static func usageColor(_ percent: Int) -> Color {
@@ -329,8 +346,8 @@ struct MenuView: View {
329 return false
346 return false
330 }
347 }
331
348
332− /// z.ai Coding Plan's quota limits. Same visual structure as the Claude section:
349+ /// z.ai Coding Plan's quota limits. Same visual structure as Hyper and Claude:
333− /// the 5-hour percentage leads as a subhero, the weekly quota sits underneath.
350+ /// a subhero headline number, bars under every window, a sparkline, and error row.
334 private var zai: some View {
351 private var zai: some View {
335 VStack(alignment: .leading, spacing: 12) {
352 VStack(alignment: .leading, spacing: 12) {
336 HStack(spacing: 8) {
353 HStack(spacing: 8) {
@@ -357,21 +374,30 @@ struct MenuView: View {
357
374
358 private func zaiUsageDetail(_ usage: ZaiUsageReport) -> some View {
375 private func zaiUsageDetail(_ usage: ZaiUsageReport) -> some View {
359 VStack(alignment: .leading, spacing: 12) {
376 VStack(alignment: .leading, spacing: 12) {
360− // The 5-hour window leads as the hero number, same as Claude's.
377+ // The 5-hour window leads as the headline number.
361 if let fiveHour = usage.fiveHourPercent {
378 if let fiveHour = usage.fiveHourPercent {
379+ VStack(spacing: 8) {
362− HStack(alignment: .firstTextBaseline, spacing: 8) {
380+ HStack(alignment: .firstTextBaseline, spacing: 8) {
381+ headlineNumber(
363− Text("🤖 \(fiveHour)%")
382+ text: "Z \\(fiveHour)%",
364− .font(Self.subheroFont)
365− .foregroundColor(Self.usageColor(fiveHour))
383+ color: Self.usageColor(fiveHour),
366− .contentTransition(.opacity)
384+ trend: viewModel.zaiTrend,
385+ copiedConfirmation: viewModel.copiedZaiConfirmation,
386+ captionText: "5-hour usage",
387+ copyAction: { viewModel.copyZaiPercentToClipboard() }
388+ )
367
389
368− Spacer(minLength: 0)
390+ Spacer(minLength: 0)
369
391
370− if let resets = usage.fiveHourResetsAt {
392+ if let resets = usage.fiveHourResetsAt {
371− Text("resets in \(Self.zaiResetsIn(from: resets))")
393+ Text("resets in \(Self.zaiResetsIn(from: resets))")
372− .font(Self.footnoteFont)
394+ .font(Self.footnoteFont)
373− .foregroundColor(.secondary)
395+ .foregroundColor(.secondary)
396+ }
374 }
397 }
398+
399+ // Bar under the 5-hour window too.
400+ UsageBar(percent: fiveHour, color: Self.usageColor(fiveHour))
375 }
401 }
376 }
402 }
377
403
@@ -380,9 +406,35 @@ struct MenuView: View {
380 zaiWindow("Weekly", percent: weekly, resetsAt: usage.weeklyResetsAt)
406 zaiWindow("Weekly", percent: weekly, resetsAt: usage.weeklyResetsAt)
381 }
407 }
382 }
408 }
409+
410+ // Peak hours notice.
411+ zaiPeakHours
412+
413+ // Sparkline for the z.ai 5-hour % history, with an orange tint
414+ // when currently in peak hours.
415+ if viewModel.zaiSparklineValues.count >= 2 {
416+ Sparkline(
417+ values: viewModel.zaiSparklineValues,
418+ color: Self.usageColor(zaiFiveHourPercent ?? 0)
419+ )
420+ .frame(height: 26)
421+ .padding(.top, 4)
422+ .transition(.opacity)
423+ .background(
424+ ViewModel.zaiInPeakHours
425+ ? RoundedRectangle(cornerRadius: 4, style: .continuous)
426+ .fill(Color.orange.opacity(0.08))
427+ : nil
428+ )
429+ }
383 }
430 }
384 }
431 }
385
432
433+ /// The current z.ai 5-hour percentage, or 0 if absent. Used for sparkline color.
434+ private var zaiFiveHourPercent: Int? {
435+ viewModel.zaiFiveHourPercent
436+ }
437+
386 /// One window row for z.ai: label, percentage, and a hairline bar.
438 /// One window row for z.ai: label, percentage, and a hairline bar.
387 private func zaiWindow(_ label: String, percent: Int, resetsAt: Date?) -> some View {
439 private func zaiWindow(_ label: String, percent: Int, resetsAt: Date?) -> some View {
388 VStack(alignment: .leading, spacing: 5) {
440 VStack(alignment: .leading, spacing: 5) {
@@ -402,6 +454,42 @@ struct MenuView: View {
402 }
454 }
403 }
455 }
404
456
457+ /// z.ai peak hours: 14:00–18:00 CST (UTC+8), i.e. 06:00–10:00 UTC.
458+ /// Limits are most likely to bind during this window. Displayed in the
459+ /// user's local timezone so they know when to expect slowdowns.
460+ private var zaiPeakHours: some View {
461+ let inPeak = ViewModel.zaiInPeakHours
462+
463+ return HStack(spacing: 6) {
464+ Circle()
465+ .fill(inPeak ? Color.orange : Color.secondary.opacity(0.4))
466+ .frame(width: 6, height: 6)
467+
468+ Text(inPeak ? "Peak hours now" : "Peak \(Self.zaiPeakHoursLocal)")
469+ .font(Self.footnoteFont)
470+ .foregroundColor(inPeak ? .orange : .secondary)
471+ }
472+ }
473+
474+ /// z.ai peak hours converted to the user's local timezone, e.g. "09:00–13:00".
475+ private static var zaiPeakHoursLocal: String {
476+ let formatter = DateFormatter()
477+ formatter.timeZone = TimeZone.current
478+ formatter.dateFormat = "HH:mm"
479+
480+ var utc = DateComponents()
481+ utc.timeZone = TimeZone(identifier: "UTC")
482+ utc.hour = 6
483+ var calendar = Calendar(identifier: .gregorian)
484+ calendar.timeZone = TimeZone(identifier: "UTC")!
485+ let start = calendar.date(from: utc)!
486+
487+ utc.hour = 10
488+ let end = calendar.date(from: utc)!
489+
490+ return "\(formatter.string(from: start))–\(formatter.string(from: end))"
491+ }
492+
405 /// Formats the time until a z.ai window resets, e.g. "3h 20m".
493 /// Formats the time until a z.ai window resets, e.g. "3h 20m".
406 private static func zaiResetsIn(from date: Date) -> String {
494 private static func zaiResetsIn(from date: Date) -> String {
407 let interval = date.timeIntervalSince(Date())
495 let interval = date.timeIntervalSince(Date())
@@ -523,45 +611,10 @@ struct MenuView: View {
523 .keyboardShortcut("q")
611 .keyboardShortcut("q")
524 }
612 }
525 .controlSize(.small)
613 .controlSize(.small)
526−
527− Link("hyper.charm.land", destination: Self.websiteURL)
528− .font(Self.footnoteFont)
529− .foregroundColor(.secondary)
530− .opacity(0.7)
531− .frame(maxWidth: .infinity, alignment: .center)
532 }
614 }
533 }
615 }
534}
616}
535
617
536−// MARK: - Loading
537−
538−/// The loading state, before there has ever been a balance to show. Reuses the ⚡ from
539−/// the menu bar title rather than a generic spinner, so the two read as the same app.
540−private struct PulsingBolt: View {
541− @State private var isPulsing = false
542−
543− var body: some View {
544− VStack(spacing: 6) {
545− Text("⚡")
546− .font(.system(size: 34))
547− .opacity(isPulsing ? 0.35 : 1)
548− .scaleEffect(isPulsing ? 0.94 : 1)
549− .animation(
550− .easeInOut(duration: 0.9).repeatForever(autoreverses: true),
551− value: isPulsing
552− )
553− // Roughly the height of the balance it stands in for, so the hero
554− // doesn't jump when the first fetch lands.
555− .frame(height: 56)
556−
557− Text("Loading…")
558− .font(.system(size: 11, weight: .medium, design: .rounded))
559− .foregroundColor(.secondary)
560− }
561− .onAppear { isPulsing = true }
562− }
563−}
564−
565// MARK: - Usage Bar
618// MARK: - Usage Bar
566
619
567/// A hairline utilization bar: a track, a fill, and nothing else. Sized to read as a
620/// A hairline utilization bar: a track, a fill, and nothing else. Sized to read as a
@@ -592,8 +645,8 @@ private struct UsageBar: View {
592
645
593// MARK: - Sparkline
646// MARK: - Sparkline
594
647
595−/// A minimal line chart of recent balances: no axes, no labels, no grid. The stroke
648+/// A minimal line chart of recent values: no axes, no labels, no grid. The stroke
596−/// takes the balance's color so the trend reads as part of the number above it, with a
649+/// takes the metric's color so the trend reads as part of the number above it, with a
597/// faint fade underneath for weight and a dot marking the latest point.
650/// faint fade underneath for weight and a dot marking the latest point.
598///
651///
599/// Points are spaced evenly by index rather than by timestamp, which keeps the shape
652/// Points are spaced evenly by index rather than by timestamp, which keeps the shape
@@ -685,4 +738,4 @@ struct Sparkline: View {
685 path.closeSubpath()
738 path.closeSubpath()
686 }
739 }
687 }
740 }
688−}
741+}
UsageMenubar/UsageMenubarApp.swift+22−7
@@ -35,7 +35,7 @@ final class AppDelegate: NSObject, NSApplicationDelegate {
35
35
36 if let button = statusItem.button {
36 if let button = statusItem.button {
37 button.image = nil
37 button.image = nil
38− button.title = "⚡?"
38+ button.title = "…"
39 button.target = self
39 button.target = self
40 button.action = #selector(statusItemClicked(_:))
40 button.action = #selector(statusItemClicked(_:))
41 button.sendAction(on: [.leftMouseUp, .rightMouseUp])
41 button.sendAction(on: [.leftMouseUp, .rightMouseUp])
@@ -46,8 +46,9 @@ final class AppDelegate: NSObject, NSApplicationDelegate {
46 popover.behavior = .transient
46 popover.behavior = .transient
47 popover.contentViewController = NSHostingController(rootView: MenuView(viewModel: viewModel))
47 popover.contentViewController = NSHostingController(rootView: MenuView(viewModel: viewModel))
48
48
49− // Observe published changes (balance + isLoading + Claude usage + z.ai usage) to
49+ // Observe published changes (balance + isLoading + hyperConfigured +
50− // update the menu bar title. CombineLatest fires when any of them changes.
50+ // Claude usage + z.ai usage) to update the menu bar title. CombineLatest
51+ // fires when any of them changes.
51 //
52 //
52 // The title is built from the values the publisher hands us, not by reading the
53 // The title is built from the values the publisher hands us, not by reading the
53 // view model back: `@Published` emits in `willSet`, so the property still holds
54 // view model back: `@Published` emits in `willSet`, so the property still holds
@@ -56,13 +57,23 @@ final class AppDelegate: NSObject, NSApplicationDelegate {
56 // Delivery is hopped through `DispatchQueue.main` rather than `RunLoop.main`,
57 // Delivery is hopped through `DispatchQueue.main` rather than `RunLoop.main`,
57 // whose Combine scheduler only runs in the default run loop mode — the title
58 // whose Combine scheduler only runs in the default run loop mode — the title
58 // would freeze while a menu or a scroll was tracking.
59 // would freeze while a menu or a scroll was tracking.
60+ // Combine has CombineLatest4 but not 5, so nest: outer combines
61+ // balance + isLoading + hyperConfigured with an inner that merges
62+ // claudeUsage + zaiUsage. The inner Just(()) is a dummy third stream
63+ // to satisfy CombineLatest3 — it fires once and never again, so the
64+ // inner effectively republishes claudeUsage and zaiUsage changes.
59 statusTextCancellable = Publishers.CombineLatest4(
65 statusTextCancellable = Publishers.CombineLatest4(
60 viewModel.$balance,
66 viewModel.$balance,
61 viewModel.$isLoading,
67 viewModel.$isLoading,
68+ viewModel.$hyperConfigured,
69+ Publishers.CombineLatest3(
62− viewModel.$claudeUsage,
70+ viewModel.$claudeUsage,
63− viewModel.$zaiUsage
71+ viewModel.$zaiUsage,
72+ Just<Void>(())
73+ )
64 )
74 )
75+ .map { balance, isLoading, hyperConfigured, claudeAndZai in
65− .map { balance, isLoading, claudeUsage, zaiUsage in
76+ let (claudeUsage, zaiUsage, _) = claudeAndZai
66 let fiveHour = claudeUsage?.fiveHour
77 let fiveHour = claudeUsage?.fiveHour
67 .map { Int($0.utilization.rounded()) }
78 .map { Int($0.utilization.rounded()) }
68 let sevenDay = claudeUsage?.sevenDay
79 let sevenDay = claudeUsage?.sevenDay
@@ -70,12 +81,16 @@ final class AppDelegate: NSObject, NSApplicationDelegate {
70 ?? claudeUsage?.sevenDayOpus
81 ?? claudeUsage?.sevenDayOpus
71 .map { Int($0.utilization.rounded()) }
82 .map { Int($0.utilization.rounded()) }
72 let zaiFiveHour = zaiUsage?.fiveHourPercent
83 let zaiFiveHour = zaiUsage?.fiveHourPercent
84+ let zaiWeekly = zaiUsage?.weeklyPercent
73 return ViewModel.statusBarText(
85 return ViewModel.statusBarText(
74 balance: balance,
86 balance: balance,
75 isLoading: isLoading,
87 isLoading: isLoading,
88+ hyperConfigured: hyperConfigured,
76 claudeFiveHourPercent: fiveHour,
89 claudeFiveHourPercent: fiveHour,
77 claudeSevenDayPercent: sevenDay,
90 claudeSevenDayPercent: sevenDay,
78− zaiFiveHourPercent: zaiFiveHour
91+ zaiFiveHourPercent: zaiFiveHour,
92+ zaiWeeklyPercent: zaiWeekly,
93+ zaiInPeakHours: ViewModel.zaiInPeakHours
79 )
94 )
80 }
95 }
81 .removeDuplicates()
96 .removeDuplicates()
UsageMenubar/ViewModel.swift+240−43
@@ -7,11 +7,11 @@ import UserNotifications
7/// Posted when the user changes the refresh interval, so the app delegate can
7/// Posted when the user changes the refresh interval, so the app delegate can
8/// restart its timer with the new period.
8/// restart its timer with the new period.
9extension Notification.Name {
9extension Notification.Name {
10− static let refreshIntervalDidChange = Notification.Name("com.ilyakooo0.usage-menubar.refreshIntervalDidChange")
10+ static let refreshIntervalDidChange = Notification.Name("soy.iko.usage-menubar.refreshIntervalDidChange")
11}
11}
12
12
13−/// The direction the balance has moved over the most recent history points.
13+/// The direction a metric has moved over the most recent history points.
14−enum BalanceTrend {
14+enum MetricTrend {
15 case increasing
15 case increasing
16 case decreasing
16 case decreasing
17 case stable
17 case stable
@@ -29,10 +29,11 @@ enum BalanceTrend {
29 }
29 }
30}
30}
31
31
32−/// A single balance observation.
32+/// A single metric observation — used for Hyper balance, Claude 5-hour %,
33+/// and z.ai 5-hour % alike.
33−struct BalancePoint: Equatable {
34+struct MetricPoint: Equatable {
34 let date: Date
35 let date: Date
35− let balance: Int
36+ let value: Int
36}
37}
37
38
38/// How often the balance is refreshed. Lives outside `ViewModel` so the view and
39/// How often the balance is refreshed. Lives outside `ViewModel` so the view and
@@ -93,6 +94,11 @@ final class ViewModel: ObservableObject {
93 /// `activeZaiAPIKey`.
94 /// `activeZaiAPIKey`.
94 @Published var zaiAPIKeyInput: String = ""
95 @Published var zaiAPIKeyInput: String = ""
95
96
97+ /// Whether Hyper is configured — i.e. there is an API key in the Keychain.
98+ /// Drives whether the `⚡` segment appears in the menu bar at all: an
99+ /// unconfigured service should not take up space in the title.
100+ @Published private(set) var hyperConfigured = false
101+
96 /// The API key draft bound to the text field. Two-way bound to the `SecureField`,
102 /// The API key draft bound to the text field. Two-way bound to the `SecureField`,
97 /// so it changes on every keystroke — never fetch with it, use `activeAPIKey`.
103 /// so it changes on every keystroke — never fetch with it, use `activeAPIKey`.
98 @Published var apiKeyInput: String = ""
104 @Published var apiKeyInput: String = ""
@@ -106,8 +112,20 @@ final class ViewModel: ObservableObject {
106 /// Brief "✓ Copied" confirmation shown after copying the balance.
112 /// Brief "✓ Copied" confirmation shown after copying the balance.
107 @Published var copiedConfirmation: Bool = false
113 @Published var copiedConfirmation: Bool = false
108
114
115+ /// Brief "✓ Copied" confirmation shown after copying the Claude percentage.
116+ @Published var copiedClaudeConfirmation: Bool = false
117+
118+ /// Brief "✓ Copied" confirmation shown after copying the z.ai percentage.
119+ @Published var copiedZaiConfirmation: Bool = false
120+
109 /// Successful balance observations from the last 24 hours, oldest first.
121 /// Successful balance observations from the last 24 hours, oldest first.
110− @Published private(set) var history: [BalancePoint] = []
122+ @Published private(set) var history: [MetricPoint] = []
123+
124+ /// Successful Claude 5-hour % observations from the last 24 hours, oldest first.
125+ @Published private(set) var claudeHistory: [MetricPoint] = []
126+
127+ /// Successful z.ai 5-hour % observations from the last 24 hours, oldest first.
128+ @Published private(set) var zaiHistory: [MetricPoint] = []
111
129
112 /// How often the balance is refreshed, in minutes. Persisted to `UserDefaults`;
130 /// How often the balance is refreshed, in minutes. Persisted to `UserDefaults`;
113 /// changing it posts `.refreshIntervalDidChange` so the app delegate can
131 /// changing it posts `.refreshIntervalDidChange` so the app delegate can
@@ -158,6 +176,12 @@ final class ViewModel: ObservableObject {
158 /// Resets `savedConfirmation`; same rationale as `copiedResetTask`.
176 /// Resets `savedConfirmation`; same rationale as `copiedResetTask`.
159 private var savedResetTask: Task<Void, Never>?
177 private var savedResetTask: Task<Void, Never>?
160
178
179+ /// Resets `copiedClaudeConfirmation`; same rationale as `copiedResetTask`.
180+ private var copiedClaudeResetTask: Task<Void, Never>?
181+
182+ /// Resets `copiedZaiConfirmation`; same rationale as `copiedResetTask`.
183+ private var copiedZaiResetTask: Task<Void, Never>?
184+
161 // MARK: - History Configuration
185 // MARK: - History Configuration
162
186
163 /// 24 hours at the shortest (1 minute) interval would overflow this, so the
187 /// 24 hours at the shortest (1 minute) interval would overflow this, so the
@@ -165,6 +189,9 @@ final class ViewModel: ObservableObject {
165 private static let maxHistoryEntries = 300
189 private static let maxHistoryEntries = 300
166 private static let historyWindow: TimeInterval = 24 * 60 * 60
190 private static let historyWindow: TimeInterval = 24 * 60 * 60
167
191
192+ /// The threshold above which a usage percentage triggers a notification.
193+ private static let highUsageThreshold = 90
194+
168 init(
195 init(
169 checker: CreditsChecking = CreditsChecker(),
196 checker: CreditsChecking = CreditsChecker(),
170 claudeChecker: ClaudeUsageChecking = ClaudeUsageClient(),
197 claudeChecker: ClaudeUsageChecking = ClaudeUsageClient(),
@@ -178,6 +205,7 @@ final class ViewModel: ObservableObject {
178 let storedKey = KeychainHelper.load() ?? ""
205 let storedKey = KeychainHelper.load() ?? ""
179 apiKeyInput = storedKey
206 apiKeyInput = storedKey
180 activeAPIKey = storedKey
207 activeAPIKey = storedKey
208+ hyperConfigured = !storedKey.isEmpty
181 let storedZaiKey = ZaiKeychainHelper.load() ?? ""
209 let storedZaiKey = ZaiKeychainHelper.load() ?? ""
182 zaiAPIKeyInput = storedZaiKey
210 zaiAPIKeyInput = storedZaiKey
183 activeZaiAPIKey = storedZaiKey
211 activeZaiAPIKey = storedZaiKey
@@ -209,12 +237,13 @@ final class ViewModel: ObservableObject {
209 balanceFormatter.string(from: NSNumber(value: value)) ?? "\(value)"
237 balanceFormatter.string(from: NSNumber(value: value)) ?? "\(value)"
210 }
238 }
211
239
212− /// The text to show in the menu bar: `⚡{balance}` followed by Claude usage
240+ /// The text to show in the menu bar. Each configured provider contributes a
213− /// windows, each prefixed with an emoji so they are distinguishable at a glance:
241+ /// segment prefixed with a provider emoji; segments are separated by ` · `
214− /// 🕐 for the 5-hour window, 📅 for the 7-day window, 🤖 for the z.ai 5-hour
242+ /// (middle dot). Within a segment, windows are listed as bare percentages
215− /// window. Windows that are absent or at 0% are omitted; with neither, the bolt
243+ /// separated by spaces. An unconfigured service produces no segment.
216− /// stands in alone: `⚡…` while the first fetch is in flight, `⚡?` once it has
244+ /// When z.ai is in peak hours, a 🔶 marker is appended to the z.ai segment.
217− /// come back empty-handed.
245+ ///
246+ /// Example: `⚡42 · ⁕62% 8% · Z12% 3%🔶`
218 ///
247 ///
219 /// Takes its values as parameters rather than reading the properties: `@Published`
248 /// Takes its values as parameters rather than reading the properties: `@Published`
220 /// publishes in `willSet`, so a Combine subscriber that called back into the view
249 /// publishes in `willSet`, so a Combine subscriber that called back into the view
@@ -222,27 +251,54 @@ final class ViewModel: ObservableObject {
222 static func statusBarText(
251 static func statusBarText(
223 balance: Int?,
252 balance: Int?,
224 isLoading: Bool,
253 isLoading: Bool,
254+ hyperConfigured: Bool,
225 claudeFiveHourPercent: Int?,
255 claudeFiveHourPercent: Int?,
226 claudeSevenDayPercent: Int?,
256 claudeSevenDayPercent: Int?,
227− zaiFiveHourPercent: Int?
257+ zaiFiveHourPercent: Int?,
258+ zaiWeeklyPercent: Int?,
259+ zaiInPeakHours: Bool
228 ) -> String {
260 ) -> String {
229− var parts: [String] = []
261+ var segments: [String] = []
262+
263+ // Hyper
230− if let balance = balance {
264+ if hyperConfigured, let balance = balance {
231− parts.append("⚡\(formatBalance(balance))")
265+ segments.append("⚡\(formatBalance(balance))")
232 }
266 }
267+
268+ // Claude — provider emoji + bare percentages
269+ var claudeWindows: [String] = []
233− if let fiveHour = claudeFiveHourPercent, fiveHour > 0 {
270+ if let fiveHour = claudeFiveHourPercent {
234− parts.append("🕐\(fiveHour)%")
271+ claudeWindows.append("\(fiveHour)%")
272+ }
273+ if let sevenDay = claudeSevenDayPercent {
274+ claudeWindows.append("\(sevenDay)%")
235 }
275 }
236− if let sevenDay = claudeSevenDayPercent, sevenDay > 0 {
276+ if !claudeWindows.isEmpty {
237− parts.append("📅\(sevenDay)%")
277+ segments.append("⁕" + claudeWindows.joined(separator: " "))
238 }
278 }
279+
280+ // z.ai — provider emoji + bare percentages
281+ var zaiWindows: [String] = []
239− if let zaiFiveHour = zaiFiveHourPercent, zaiFiveHour > 0 {
282+ if let zaiFiveHour = zaiFiveHourPercent {
240− parts.append("🤖\(zaiFiveHour)%")
283+ zaiWindows.append("\(zaiFiveHour)%")
241 }
284 }
285+ if let zaiWeekly = zaiWeeklyPercent {
286+ zaiWindows.append("\(zaiWeekly)%")
287+ }
242− if parts.isEmpty {
288+ if !zaiWindows.isEmpty {
243− return isLoading ? "⚡…" : "⚡?"
289+ var segment = "Z" + zaiWindows.joined(separator: " ")
290+ if zaiInPeakHours {
291+ segment += "🔶"
292+ }
293+ segments.append(segment)
244 }
294 }
295+
296+ if segments.isEmpty {
297+ // No service is configured or has anything to show. While loading,
298+ // show an ellipsis; otherwise the title is empty.
299+ return isLoading ? "…" : ""
300+ }
245− return parts.joined(separator: " · ")
301+ return segments.joined(separator: " · ")
246 }
302 }
247
303
248 /// The text to show in the menu bar for the current state.
304 /// The text to show in the menu bar for the current state.
@@ -250,12 +306,21 @@ final class ViewModel: ObservableObject {
250 Self.statusBarText(
306 Self.statusBarText(
251 balance: balance,
307 balance: balance,
252 isLoading: isLoading,
308 isLoading: isLoading,
309+ hyperConfigured: hyperConfigured,
253 claudeFiveHourPercent: claudeFiveHourPercent,
310 claudeFiveHourPercent: claudeFiveHourPercent,
254 claudeSevenDayPercent: claudeSevenDayPercent,
311 claudeSevenDayPercent: claudeSevenDayPercent,
255− zaiFiveHourPercent: zaiFiveHourPercent
312+ zaiFiveHourPercent: zaiFiveHourPercent,
313+ zaiWeeklyPercent: zaiWeeklyPercent,
314+ zaiInPeakHours: Self.zaiInPeakHours
256 )
315 )
257 }
316 }
258
317
318+ /// Whether z.ai is currently in peak hours (06:00–10:00 UTC).
319+ static var zaiInPeakHours: Bool {
320+ let hour = Calendar(identifier: .gregorian).component(.hour, from: Date())
321+ return hour >= 6 && hour < 10
322+ }
323+
259 /// The formatted balance string for display in the popover.
324 /// The formatted balance string for display in the popover.
260 var formattedBalance: String {
325 var formattedBalance: String {
261 guard let balance = balance else { return "?" }
326 guard let balance = balance else { return "?" }
@@ -276,33 +341,69 @@ final class ViewModel: ObservableObject {
276 return "Updated \(Self.relativeFormatter.localizedString(for: lastUpdated, relativeTo: Date()))"
341 return "Updated \(Self.relativeFormatter.localizedString(for: lastUpdated, relativeTo: Date()))"
277 }
342 }
278
343
279− // MARK: - History
344+ // MARK: - History (Hyper)
280
345
281− /// The last 24 hours of successful fetches, oldest first.
346+ /// The last 24 hours of successful balance fetches, oldest first.
282 var balanceHistory: [(date: Date, balance: Int)] {
347 var balanceHistory: [(date: Date, balance: Int)] {
283− history.map { (date: $0.date, balance: $0.balance) }
348+ history.map { (date: $0.date, balance: $0.value) }
284 }
349 }
285
350
286 /// The direction of travel across the most recent handful of observations.
351 /// The direction of travel across the most recent handful of observations.
287 /// `.unknown` until there are at least two points to compare.
352 /// `.unknown` until there are at least two points to compare.
288− var balanceTrend: BalanceTrend {
353+ var balanceTrend: MetricTrend {
354+ metricTrend(history)
355+ }
356+
357+ /// The sparkline's data for the Hyper balance.
358+ var balanceSparklineValues: [Int] {
359+ history.map { $0.value }
360+ }
361+
362+ // MARK: - History (Claude)
363+
364+ /// The last 24 hours of successful Claude 5-hour % fetches, oldest first.
365+ var claudeSparklineValues: [Int] {
366+ claudeHistory.map { $0.value }
367+ }
368+
369+ /// The direction of travel for the Claude 5-hour %.
370+ var claudeTrend: MetricTrend {
371+ metricTrend(claudeHistory)
372+ }
373+
374+ // MARK: - History (z.ai)
375+
376+ /// The last 24 hours of successful z.ai 5-hour % fetches, oldest first.
377+ var zaiSparklineValues: [Int] {
378+ zaiHistory.map { $0.value }
379+ }
380+
381+ /// The direction of travel for the z.ai 5-hour %.
382+ var zaiTrend: MetricTrend {
383+ metricTrend(zaiHistory)
384+ }
385+
386+ // MARK: - History Helpers
387+
388+ /// Computes the trend from a generic history array.
389+ private func metricTrend(_ points: [MetricPoint]) -> MetricTrend {
289− guard history.count >= 2 else { return .unknown }
390+ guard points.count >= 2 else { return .unknown }
290− let recent = history.suffix(5)
391+ let recent = points.suffix(5)
291 guard let first = recent.first, let last = recent.last else { return .unknown }
392 guard let first = recent.first, let last = recent.last else { return .unknown }
292− if last.balance > first.balance { return .increasing }
393+ if last.value > first.value { return .increasing }
293− if last.balance < first.balance { return .decreasing }
394+ if last.value < first.value { return .decreasing }
294 return .stable
395 return .stable
295 }
396 }
296
397
297 /// Appends an observation and drops anything older than 24 hours or beyond the cap.
398 /// Appends an observation and drops anything older than 24 hours or beyond the cap.
298− private func recordHistory(balance: Int, at date: Date) {
399+ private func recordMetric(_ value: Int, into array: inout [MetricPoint], at date: Date) {
299− history.append(BalancePoint(date: date, balance: balance))
400+ array.append(MetricPoint(date: date, value: value))
300
401
301 let cutoff = date.addingTimeInterval(-Self.historyWindow)
402 let cutoff = date.addingTimeInterval(-Self.historyWindow)
302− history.removeAll { $0.date < cutoff }
403+ array.removeAll { $0.date < cutoff }
303
404
304− if history.count > Self.maxHistoryEntries {
405+ if array.count > Self.maxHistoryEntries {
305− history.removeFirst(history.count - Self.maxHistoryEntries)
406+ array.removeFirst(array.count - Self.maxHistoryEntries)
306 }
407 }
307 }
408 }
308
409
@@ -377,7 +478,7 @@ final class ViewModel: ObservableObject {
377 errorMessage = nil
478 errorMessage = nil
378 let now = Date()
479 let now = Date()
379 lastUpdated = now
480 lastUpdated = now
380− recordHistory(balance: newBalance, at: now)
481+ recordMetric(newBalance, into: &history, at: now)
381
482
382 // Low-balance notification: on threshold crossing (≥10 → <10)
483 // Low-balance notification: on threshold crossing (≥10 → <10)
383 // OR on first successful fetch if already below threshold.
484 // OR on first successful fetch if already below threshold.
@@ -422,12 +523,30 @@ final class ViewModel: ObservableObject {
422 claudeUsage = nil
523 claudeUsage = nil
423 claudePlan = nil
524 claudePlan = nil
424 claudeError = nil
525 claudeError = nil
526+ claudeHistory.removeAll()
425
527
426 case .usage(let report):
528 case .usage(let report):
529+ // Capture the previous 5-hour % for threshold-crossing notification.
530+ let previousPercent = claudeFiveHourPercent
427 claudeUsage = report.usage
531 claudeUsage = report.usage
428 claudePlan = report.subscriptionType
532 claudePlan = report.subscriptionType
429 claudeError = nil
533 claudeError = nil
430
534
535+ // Record history and fire high-usage notification based on the 5-hour %.
536+ if let percent = claudeFiveHourPercent {
537+ let now = Date()
538+ recordMetric(percent, into: &claudeHistory, at: now)
539+
540+ // High-usage notification: on threshold crossing (<90 → ≥90)
541+ // OR on first successful fetch if already above threshold.
542+ if let previous = previousPercent, previous < Self.highUsageThreshold,
543+ percent >= Self.highUsageThreshold {
544+ sendHighUsageNotification(provider: "Claude Code", percent: percent)
545+ } else if previousPercent == nil, percent >= Self.highUsageThreshold {
546+ sendHighUsageNotification(provider: "Claude Code", percent: percent)
547+ }
548+ }
549+
431 case .failure(let message):
550 case .failure(let message):
432 // As with the balance: keep the last good numbers on screen and say what
551 // As with the balance: keep the last good numbers on screen and say what
433 // went wrong, rather than blanking the section on a transient failure.
552 // went wrong, rather than blanking the section on a transient failure.
@@ -495,11 +614,29 @@ final class ViewModel: ObservableObject {
495 case .notConfigured:
614 case .notConfigured:
496 zaiUsage = nil
615 zaiUsage = nil
497 zaiError = nil
616 zaiError = nil
617+ zaiHistory.removeAll()
498
618
499 case .usage(let report):
619 case .usage(let report):
620+ // Capture the previous 5-hour % for threshold-crossing notification.
621+ let previousPercent = zaiFiveHourPercent
500 zaiUsage = report
622 zaiUsage = report
501 zaiError = nil
623 zaiError = nil
502
624
625+ // Record history and fire high-usage notification based on the 5-hour %.
626+ if let percent = zaiFiveHourPercent {
627+ let now = Date()
628+ recordMetric(percent, into: &zaiHistory, at: now)
629+
630+ // High-usage notification: on threshold crossing (<90 → ≥90)
631+ // OR on first successful fetch if already above threshold.
632+ if let previous = previousPercent, previous < Self.highUsageThreshold,
633+ percent >= Self.highUsageThreshold {
634+ sendHighUsageNotification(provider: "z.ai Coding", percent: percent)
635+ } else if previousPercent == nil, percent >= Self.highUsageThreshold {
636+ sendHighUsageNotification(provider: "z.ai Coding", percent: percent)
637+ }
638+ }
639+
503 case .failure(let message):
640 case .failure(let message):
504 // As with the balance and Claude: keep the last good numbers on screen
641 // As with the balance and Claude: keep the last good numbers on screen
505 // and say what went wrong, rather than blanking the section on a
642 // and say what went wrong, rather than blanking the section on a
@@ -553,7 +690,41 @@ final class ViewModel: ObservableObject {
553 }
690 }
554 }
691 }
555
692
556− // MARK: - Low Balance Notification
693+ /// Copies the current Claude 5-hour percentage to the clipboard.
694+ func copyClaudePercentToClipboard() {
695+ guard let percent = claudeFiveHourPercent else { return }
696+
697+ let pasteboard = NSPasteboard.general
698+ pasteboard.clearContents()
699+ pasteboard.setString("\(percent)%", forType: .string)
700+
701+ copiedClaudeConfirmation = true
702+ copiedClaudeResetTask?.cancel()
703+ copiedClaudeResetTask = Task { @MainActor in
704+ try? await Task.sleep(nanoseconds: 1_500_000_000)
705+ guard !Task.isCancelled else { return }
706+ copiedClaudeConfirmation = false
707+ }
708+ }
709+
710+ /// Copies the current z.ai 5-hour percentage to the clipboard.
711+ func copyZaiPercentToClipboard() {
712+ guard let percent = zaiFiveHourPercent else { return }
713+
714+ let pasteboard = NSPasteboard.general
715+ pasteboard.clearContents()
716+ pasteboard.setString("\(percent)%", forType: .string)
717+
718+ copiedZaiConfirmation = true
719+ copiedZaiResetTask?.cancel()
720+ copiedZaiResetTask = Task { @MainActor in
721+ try? await Task.sleep(nanoseconds: 1_500_000_000)
722+ guard !Task.isCancelled else { return }
723+ copiedZaiConfirmation = false
724+ }
725+ }
726+
727+ // MARK: - Notifications
557
728
558 private func sendLowBalanceNotification(balance: Int) {
729 private func sendLowBalanceNotification(balance: Int) {
559 // Build the body here, on the main actor. `getNotificationSettings` calls back on
730 // Build the body here, on the main actor. `getNotificationSettings` calls back on
@@ -582,6 +753,29 @@ final class ViewModel: ObservableObject {
582 }
753 }
583 }
754 }
584
755
756+ /// Sends a notification when a usage percentage crosses the high-usage threshold.
757+ private func sendHighUsageNotification(provider: String, percent: Int) {
758+ let body = "\(provider) 5-hour usage at \(percent)% — approaching limit"
759+
760+ let center = UNUserNotificationCenter.current()
761+ center.getNotificationSettings { settings in
762+ guard settings.authorizationStatus == .authorized || settings.authorizationStatus == .provisional else {
763+ return
764+ }
765+ let content = UNMutableNotificationContent()
766+ content.title = "High Usage Menubar"
767+ content.body = body
768+ content.sound = .default
769+
770+ let request = UNNotificationRequest(
771+ identifier: "high-usage-alert-\(provider)-\(UUID().uuidString)",
772+ content: content,
773+ trigger: nil
774+ )
775+ center.add(request)
776+ }
777+ }
778+
585 // MARK: - API Key
779 // MARK: - API Key
586
780
587 /// Saves the current `apiKeyInput` to the Keychain and triggers a refresh.
781 /// Saves the current `apiKeyInput` to the Keychain and triggers a refresh.
@@ -596,6 +790,7 @@ final class ViewModel: ObservableObject {
596 }
790 }
597 apiKeyInput = ""
791 apiKeyInput = ""
598 activeAPIKey = ""
792 activeAPIKey = ""
793+ hyperConfigured = false
599 // refresh() with an empty key clears balance, error, history, and
794 // refresh() with an empty key clears balance, error, history, and
600 // cancels any in-flight task from the previous key.
795 // cancels any in-flight task from the previous key.
601 refresh()
796 refresh()
@@ -619,6 +814,7 @@ final class ViewModel: ObservableObject {
619 // Show the trimmed key that was actually stored.
814 // Show the trimmed key that was actually stored.
620 apiKeyInput = key
815 apiKeyInput = key
621 activeAPIKey = key
816 activeAPIKey = key
817+ hyperConfigured = true
622 refresh()
818 refresh()
623 }
819 }
624
820
@@ -662,6 +858,7 @@ final class ViewModel: ObservableObject {
662 if key != activeZaiAPIKey {
858 if key != activeZaiAPIKey {
663 zaiUsage = nil
859 zaiUsage = nil
664 zaiError = nil
860 zaiError = nil
861+ zaiHistory.removeAll()
665 }
862 }
666 // Show the trimmed key that was actually stored.
863 // Show the trimmed key that was actually stored.
667 zaiAPIKeyInput = key
864 zaiAPIKeyInput = key
@@ -699,4 +896,4 @@ final class ViewModel: ObservableObject {
699 isInitializingLaunchAtLogin = false
896 isInitializingLaunchAtLogin = false
700 }
897 }
701 }
898 }
702−}
899+}
UsageMenubarTests/CreditsCheckerTests.swift+19
@@ -519,10 +519,22 @@ final class StubURLProtocol: URLProtocol {
519 Self.outcomes.append(contentsOf: outcomes)
519 Self.outcomes.append(contentsOf: outcomes)
520 }
520 }
521
521
522+ /// Queues an outcome for a specific URL path. URL-matched outcomes are consumed
523+ /// before the FIFO queue, so concurrent requests to different endpoints each get
524+ /// the right response regardless of arrival order.
525+ static func enqueue(_ outcome: Outcome, forURLPath path: String) {
526+ lock.lock()
527+ defer { lock.unlock() }
528+ urlMatchedOutcomes[path] = outcome
529+ }
530+
531+ private static var urlMatchedOutcomes: [String: Outcome] = [:]
532+
522 static func reset() {
533 static func reset() {
523 lock.lock()
534 lock.lock()
524 defer { lock.unlock() }
535 defer { lock.unlock() }
525 outcomes = []
536 outcomes = []
537+ urlMatchedOutcomes = [:]
526 lastOutcome = nil
538 lastOutcome = nil
527 requests = []
539 requests = []
528 bodies = []
540 bodies = []
@@ -548,12 +560,19 @@ final class StubURLProtocol: URLProtocol {
548 }
560 }
549
561
550 /// Records the request and pops the outcome that should answer it.
562 /// Records the request and pops the outcome that should answer it.
563+ /// URL-matched outcomes take priority over the FIFO queue, so concurrent
564+ /// requests to different endpoints each get the right response.
551 private static func nextOutcome(for request: URLRequest) -> Outcome {
565 private static func nextOutcome(for request: URLRequest) -> Outcome {
552 lock.lock()
566 lock.lock()
553 defer { lock.unlock() }
567 defer { lock.unlock() }
554 requests.append(request)
568 requests.append(request)
555 bodies.append(drainBody(from: request))
569 bodies.append(drainBody(from: request))
556
570
571+ if let path = request.url?.path,
572+ let matched = urlMatchedOutcomes[path] {
573+ return matched
574+ }
575+
557 if !outcomes.isEmpty {
576 if !outcomes.isEmpty {
558 lastOutcome = outcomes.removeFirst()
577 lastOutcome = outcomes.removeFirst()
559 }
578 }
UsageMenubarTests/StatusBarTextTests.swift+190−37
@@ -13,17 +13,21 @@ final class StatusBarTextTests: XCTestCase {
13 func testBothServices() {
13 func testBothServices() {
14 let title = ViewModel.statusBarText(
14 let title = ViewModel.statusBarText(
15 balance: 42, isLoading: false,
15 balance: 42, isLoading: false,
16+ hyperConfigured: true,
16 claudeFiveHourPercent: 62, claudeSevenDayPercent: 8,
17 claudeFiveHourPercent: 62, claudeSevenDayPercent: 8,
17− zaiFiveHourPercent: nil
18+ zaiFiveHourPercent: nil, zaiWeeklyPercent: nil,
19+ zaiInPeakHours: false
18 )
20 )
19− XCTAssertEqual(title, "⚡42 · 🕐62% · 📅8%")
21+ XCTAssertEqual(title, "⚡42 · \u{2055}62% 8%")
20 }
22 }
21
23
22 func testHyperOnly() {
24 func testHyperOnly() {
23 let title = ViewModel.statusBarText(
25 let title = ViewModel.statusBarText(
24 balance: 42, isLoading: false,
26 balance: 42, isLoading: false,
27+ hyperConfigured: true,
25 claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
28 claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
26− zaiFiveHourPercent: nil
29+ zaiFiveHourPercent: nil, zaiWeeklyPercent: nil,
30+ zaiInPeakHours: false
27 )
31 )
28 XCTAssertEqual(title, "⚡42")
32 XCTAssertEqual(title, "⚡42")
29 }
33 }
@@ -31,85 +35,140 @@ final class StatusBarTextTests: XCTestCase {
31 func testClaudeOnly() {
35 func testClaudeOnly() {
32 let title = ViewModel.statusBarText(
36 let title = ViewModel.statusBarText(
33 balance: nil, isLoading: false,
37 balance: nil, isLoading: false,
38+ hyperConfigured: false,
34 claudeFiveHourPercent: 62, claudeSevenDayPercent: 8,
39 claudeFiveHourPercent: 62, claudeSevenDayPercent: 8,
35− zaiFiveHourPercent: nil
40+ zaiFiveHourPercent: nil, zaiWeeklyPercent: nil,
41+ zaiInPeakHours: false
36 )
42 )
37− XCTAssertEqual(title, "🕐62% · 📅8%")
43+ XCTAssertEqual(title, "\u{2055}62% 8%")
38 }
44 }
39
45
40 func testLoadingWithNothingToShow() {
46 func testLoadingWithNothingToShow() {
41 let title = ViewModel.statusBarText(
47 let title = ViewModel.statusBarText(
42 balance: nil, isLoading: true,
48 balance: nil, isLoading: true,
49+ hyperConfigured: false,
43 claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
50 claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
44− zaiFiveHourPercent: nil
51+ zaiFiveHourPercent: nil, zaiWeeklyPercent: nil,
52+ zaiInPeakHours: false
45 )
53 )
46− XCTAssertEqual(title, "⚡…")
54+ XCTAssertEqual(title, "…")
47 }
55 }
48
56
49 func testNothingToShow() {
57 func testNothingToShow() {
50 let title = ViewModel.statusBarText(
58 let title = ViewModel.statusBarText(
51 balance: nil, isLoading: false,
59 balance: nil, isLoading: false,
60+ hyperConfigured: false,
52 claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
61 claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
53− zaiFiveHourPercent: nil
62+ zaiFiveHourPercent: nil, zaiWeeklyPercent: nil,
63+ zaiInPeakHours: false
54 )
64 )
55− XCTAssertEqual(title, "⚡?")
65+ XCTAssertEqual(title, "")
56 }
66 }
57
67
58 /// The placeholders only stand in for an *empty* title. Someone who runs Claude Code
68 /// The placeholders only stand in for an *empty* title. Someone who runs Claude Code
59− /// without a Hyper key refreshes every few minutes, and a title that fell back to `⚡…`
69+ /// without a Hyper key refreshes every few minutes, and a title that fell back to `…`
60 /// each time would blink their percentage away and back.
70 /// each time would blink their percentage away and back.
61 func testLoadingKeepsAClaudePercentOnScreen() {
71 func testLoadingKeepsAClaudePercentOnScreen() {
62 let title = ViewModel.statusBarText(
72 let title = ViewModel.statusBarText(
63 balance: nil, isLoading: true,
73 balance: nil, isLoading: true,
74+ hyperConfigured: false,
64 claudeFiveHourPercent: 62, claudeSevenDayPercent: nil,
75 claudeFiveHourPercent: 62, claudeSevenDayPercent: nil,
65− zaiFiveHourPercent: nil
76+ zaiFiveHourPercent: nil, zaiWeeklyPercent: nil,
77+ zaiInPeakHours: false
66 )
78 )
67− XCTAssertEqual(title, "🕐62%")
79+ XCTAssertEqual(title, "\u{2055}62%")
68 }
80 }
69
81
70 func testLoadingKeepsAStaleBalanceOnScreen() {
82 func testLoadingKeepsAStaleBalanceOnScreen() {
71 let title = ViewModel.statusBarText(
83 let title = ViewModel.statusBarText(
72 balance: 42, isLoading: true,
84 balance: 42, isLoading: true,
85+ hyperConfigured: true,
73 claudeFiveHourPercent: 62, claudeSevenDayPercent: 8,
86 claudeFiveHourPercent: 62, claudeSevenDayPercent: 8,
74− zaiFiveHourPercent: nil
87+ zaiFiveHourPercent: nil, zaiWeeklyPercent: nil,
88+ zaiInPeakHours: false
75 )
89 )
76− XCTAssertEqual(title, "⚡42 · 🕐62% · 📅8%")
90+ XCTAssertEqual(title, "⚡42 · \u{2055}62% 8%")
77 }
91 }
78
92
79− func testZeroPercentIsShownRatherThanOmitted() {
93+ func testZeroPercentIsShown() {
80 let title = ViewModel.statusBarText(
94 let title = ViewModel.statusBarText(
81 balance: 42, isLoading: false,
95 balance: 42, isLoading: false,
96+ hyperConfigured: true,
82 claudeFiveHourPercent: 0, claudeSevenDayPercent: 8,
97 claudeFiveHourPercent: 0, claudeSevenDayPercent: 8,
83− zaiFiveHourPercent: nil
98+ zaiFiveHourPercent: nil, zaiWeeklyPercent: nil,
99+ zaiInPeakHours: false
84 )
100 )
85− XCTAssertEqual(title, "⚡42 · 📅8%")
101+ XCTAssertEqual(title, "⚡42 · \u{2055}0% 8%")
86 }
102 }
87
103
88 func testBothClaudeWindowsZero() {
104 func testBothClaudeWindowsZero() {
89 let title = ViewModel.statusBarText(
105 let title = ViewModel.statusBarText(
90 balance: 42, isLoading: false,
106 balance: 42, isLoading: false,
107+ hyperConfigured: true,
91 claudeFiveHourPercent: 0, claudeSevenDayPercent: 0,
108 claudeFiveHourPercent: 0, claudeSevenDayPercent: 0,
92− zaiFiveHourPercent: nil
109+ zaiFiveHourPercent: nil, zaiWeeklyPercent: nil,
110+ zaiInPeakHours: false
93 )
111 )
94− XCTAssertEqual(title, "⚡42")
112+ XCTAssertEqual(title, "⚡42 · \u{2055}0% 0%")
95 }
113 }
96
114
97 func testFiveHourOnly() {
115 func testFiveHourOnly() {
98 let title = ViewModel.statusBarText(
116 let title = ViewModel.statusBarText(
99 balance: nil, isLoading: false,
117 balance: nil, isLoading: false,
118+ hyperConfigured: false,
100 claudeFiveHourPercent: 62, claudeSevenDayPercent: nil,
119 claudeFiveHourPercent: 62, claudeSevenDayPercent: nil,
101− zaiFiveHourPercent: nil
120+ zaiFiveHourPercent: nil, zaiWeeklyPercent: nil,
121+ zaiInPeakHours: false
102 )
122 )
103− XCTAssertEqual(title, "🕐62%")
123+ XCTAssertEqual(title, "\u{2055}62%")
104 }
124 }
105
125
106 func testSevenDayOnly() {
126 func testSevenDayOnly() {
107 let title = ViewModel.statusBarText(
127 let title = ViewModel.statusBarText(
108 balance: nil, isLoading: false,
128 balance: nil, isLoading: false,
129+ hyperConfigured: false,
109 claudeFiveHourPercent: nil, claudeSevenDayPercent: 8,
130 claudeFiveHourPercent: nil, claudeSevenDayPercent: 8,
110− zaiFiveHourPercent: nil
131+ zaiFiveHourPercent: nil, zaiWeeklyPercent: nil,
132+ zaiInPeakHours: false
111 )
133 )
112− XCTAssertEqual(title, "📅8%")
134+ XCTAssertEqual(title, "\u{2055}8%")
135+ }
136+
137+ // MARK: - Hyper not configured
138+
139+ func testHyperNotConfiguredHidesBoltEvenWithBalance() {
140+ // balance is non-nil but Hyper is not configured — should not appear.
141+ // This can happen briefly during the transition when a key is deleted.
142+ let title = ViewModel.statusBarText(
143+ balance: 42, isLoading: false,
144+ hyperConfigured: false,
145+ claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
146+ zaiFiveHourPercent: nil, zaiWeeklyPercent: nil,
147+ zaiInPeakHours: false
148+ )
149+ XCTAssertEqual(title, "")
150+ }
151+
152+ func testHyperNotConfiguredLoadingShowsJustDots() {
153+ let title = ViewModel.statusBarText(
154+ balance: nil, isLoading: true,
155+ hyperConfigured: false,
156+ claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
157+ zaiFiveHourPercent: nil, zaiWeeklyPercent: nil,
158+ zaiInPeakHours: false
159+ )
160+ XCTAssertEqual(title, "…")
161+ }
162+
163+ func testHyperNotConfiguredButClaudeStillShows() {
164+ let title = ViewModel.statusBarText(
165+ balance: nil, isLoading: false,
166+ hyperConfigured: false,
167+ claudeFiveHourPercent: 62, claudeSevenDayPercent: 8,
168+ zaiFiveHourPercent: nil, zaiWeeklyPercent: nil,
169+ zaiInPeakHours: false
170+ )
171+ XCTAssertEqual(title, "\u{2055}62% 8%")
113 }
172 }
114
173
115 // MARK: - z.ai in Title
174 // MARK: - z.ai in Title
@@ -117,55 +176,149 @@ final class StatusBarTextTests: XCTestCase {
117 func testZaiOnly() {
176 func testZaiOnly() {
118 let title = ViewModel.statusBarText(
177 let title = ViewModel.statusBarText(
119 balance: nil, isLoading: false,
178 balance: nil, isLoading: false,
179+ hyperConfigured: false,
120 claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
180 claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
121− zaiFiveHourPercent: 12
181+ zaiFiveHourPercent: 12, zaiWeeklyPercent: nil,
182+ zaiInPeakHours: false
122 )
183 )
123− XCTAssertEqual(title, "🤖12%")
184+ XCTAssertEqual(title, "\u{005A}12%")
124 }
185 }
125
186
126 func testZaiWithHyper() {
187 func testZaiWithHyper() {
127 let title = ViewModel.statusBarText(
188 let title = ViewModel.statusBarText(
128 balance: 42, isLoading: false,
189 balance: 42, isLoading: false,
190+ hyperConfigured: true,
129 claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
191 claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
130− zaiFiveHourPercent: 12
192+ zaiFiveHourPercent: 12, zaiWeeklyPercent: nil,
193+ zaiInPeakHours: false
131 )
194 )
132− XCTAssertEqual(title, "⚡42 · 🤖12%")
195+ XCTAssertEqual(title, "⚡42 · \u{005A}12%")
133 }
196 }
134
197
135 func testZaiWithClaude() {
198 func testZaiWithClaude() {
136 let title = ViewModel.statusBarText(
199 let title = ViewModel.statusBarText(
137 balance: nil, isLoading: false,
200 balance: nil, isLoading: false,
201+ hyperConfigured: false,
138 claudeFiveHourPercent: 62, claudeSevenDayPercent: 8,
202 claudeFiveHourPercent: 62, claudeSevenDayPercent: 8,
139− zaiFiveHourPercent: 12
203+ zaiFiveHourPercent: 12, zaiWeeklyPercent: nil,
204+ zaiInPeakHours: false
140 )
205 )
141− XCTAssertEqual(title, "🕐62% · 📅8% · 🤖12%")
206+ XCTAssertEqual(title, "\u{2055}62% 8% · \u{005A}12%")
142 }
207 }
143
208
144 func testAllThreeServices() {
209 func testAllThreeServices() {
145 let title = ViewModel.statusBarText(
210 let title = ViewModel.statusBarText(
146 balance: 42, isLoading: false,
211 balance: 42, isLoading: false,
212+ hyperConfigured: true,
147 claudeFiveHourPercent: 62, claudeSevenDayPercent: 8,
213 claudeFiveHourPercent: 62, claudeSevenDayPercent: 8,
148− zaiFiveHourPercent: 12
214+ zaiFiveHourPercent: 12, zaiWeeklyPercent: 3,
215+ zaiInPeakHours: false
149 )
216 )
150− XCTAssertEqual(title, "⚡42 · 🕐62% · 📅8% · 🤖12%")
217+ XCTAssertEqual(title, "⚡42 · \u{2055}62% 8% · \u{005A}12% 3%")
151 }
218 }
152
219
153− func testZaiZeroIsOmitted() {
220+ func testZaiZeroIsShown() {
154 let title = ViewModel.statusBarText(
221 let title = ViewModel.statusBarText(
155 balance: 42, isLoading: false,
222 balance: 42, isLoading: false,
223+ hyperConfigured: true,
156 claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
224 claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
157− zaiFiveHourPercent: 0
225+ zaiFiveHourPercent: 0, zaiWeeklyPercent: 0,
226+ zaiInPeakHours: false
158 )
227 )
159− XCTAssertEqual(title, "⚡42")
228+ XCTAssertEqual(title, "⚡42 · \u{005A}0% 0%")
160 }
229 }
161
230
162 func testLoadingKeepsZaiPercentOnScreen() {
231 func testLoadingKeepsZaiPercentOnScreen() {
163 let title = ViewModel.statusBarText(
232 let title = ViewModel.statusBarText(
164 balance: nil, isLoading: true,
233 balance: nil, isLoading: true,
234+ hyperConfigured: false,
235+ claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
236+ zaiFiveHourPercent: 12, zaiWeeklyPercent: nil,
237+ zaiInPeakHours: false
238+ )
239+ XCTAssertEqual(title, "\u{005A}12%")
240+ }
241+
242+ // MARK: - z.ai Weekly
243+
244+ func testZaiWeeklyOnly() {
245+ let title = ViewModel.statusBarText(
246+ balance: nil, isLoading: false,
247+ hyperConfigured: false,
165 claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
248 claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
166− zaiFiveHourPercent: 12
249+ zaiFiveHourPercent: nil, zaiWeeklyPercent: 3,
250+ zaiInPeakHours: false
251+ )
252+ XCTAssertEqual(title, "\u{005A}3%")
253+ }
254+
255+ func testZaiWeeklyWithFiveHour() {
256+ let title = ViewModel.statusBarText(
257+ balance: nil, isLoading: false,
258+ hyperConfigured: false,
259+ claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
260+ zaiFiveHourPercent: 12, zaiWeeklyPercent: 3,
261+ zaiInPeakHours: false
262+ )
263+ XCTAssertEqual(title, "\u{005A}12% 3%")
264+ }
265+
266+ func testZaiWeeklyZeroIsShown() {
267+ let title = ViewModel.statusBarText(
268+ balance: nil, isLoading: false,
269+ hyperConfigured: false,
270+ claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
271+ zaiFiveHourPercent: 12, zaiWeeklyPercent: 0,
272+ zaiInPeakHours: false
273+ )
274+ XCTAssertEqual(title, "\u{005A}12% 0%")
275+ }
276+
277+ func testLoadingKeepsZaiWeeklyOnScreen() {
278+ let title = ViewModel.statusBarText(
279+ balance: nil, isLoading: true,
280+ hyperConfigured: false,
281+ claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
282+ zaiFiveHourPercent: nil, zaiWeeklyPercent: 3,
283+ zaiInPeakHours: false
284+ )
285+ XCTAssertEqual(title, "\u{005A}3%")
286+ }
287+
288+ // MARK: - z.ai Peak Hours
289+
290+ func testPeakHoursAddsDiamondToZaiSegment() {
291+ let title = ViewModel.statusBarText(
292+ balance: 42, isLoading: false,
293+ hyperConfigured: true,
294+ claudeFiveHourPercent: 62, claudeSevenDayPercent: 8,
295+ zaiFiveHourPercent: 12, zaiWeeklyPercent: 3,
296+ zaiInPeakHours: true
297+ )
298+ XCTAssertEqual(title, "⚡42 · \u{2055}62% 8% · \u{005A}12% 3%\u{1F536}")
299+ }
300+
301+ func testPeakHoursWithZaiOnly() {
302+ let title = ViewModel.statusBarText(
303+ balance: nil, isLoading: false,
304+ hyperConfigured: false,
305+ claudeFiveHourPercent: nil, claudeSevenDayPercent: nil,
306+ zaiFiveHourPercent: 12, zaiWeeklyPercent: nil,
307+ zaiInPeakHours: true
308+ )
309+ XCTAssertEqual(title, "\u{005A}12%\u{1F536}")
310+ }
311+
312+ func testPeakHoursIgnoredWhenZaiNotConfigured() {
313+ let title = ViewModel.statusBarText(
314+ balance: 42, isLoading: false,
315+ hyperConfigured: true,
316+ claudeFiveHourPercent: 62, claudeSevenDayPercent: 8,
317+ zaiFiveHourPercent: nil, zaiWeeklyPercent: nil,
318+ zaiInPeakHours: true
167 )
319 )
320+ // No z.ai segment, so peak marker has nothing to attach to.
168− XCTAssertEqual(title, "🤖12%")
321+ XCTAssertEqual(title, "⚡42 · \u{2055}62% 8%")
169 }
322 }
170
323
171 // MARK: - Claude Window Percentages
324 // MARK: - Claude Window Percentages
@@ -222,4 +375,4 @@ final class StatusBarTextTests: XCTestCase {
222 XCTAssertNil(vm.claudeFiveHourPercent)
375 XCTAssertNil(vm.claudeFiveHourPercent)
223 XCTAssertNil(vm.claudeSevenDayPercent)
376 XCTAssertNil(vm.claudeSevenDayPercent)
224 }
377 }
225−}
378+}
UsageMenubarTests/ZaiUsageClientTests.swift+23−28
@@ -22,6 +22,18 @@ final class ZaiUsageClientTests: XCTestCase {
22 return ZaiUsageClient(session: URLSession(configuration: config), retryBaseDelay: retryBaseDelay)
22 return ZaiUsageClient(session: URLSession(configuration: config), retryBaseDelay: retryBaseDelay)
23 }
23 }
24
24
25+ /// Enqueues quota and subscription responses matched to their URL paths, so the
26+ /// order of arrival doesn't matter — each endpoint gets its own response.
27+ private func enqueueQuotaAndSubscription(
28+ quotaStatus: Int = 200,
29+ quotaJSON: Data,
30+ subscriptionStatus: Int = 200,
31+ subscriptionJSON: Data
32+ ) {
33+ StubURLProtocol.enqueue(.status(quotaStatus, quotaJSON), forURLPath: "/api/monitor/usage/quota/limit")
34+ StubURLProtocol.enqueue(.status(subscriptionStatus, subscriptionJSON), forURLPath: "/api/biz/subscription/list")
35+ }
36+
25 // MARK: - Decoding
37 // MARK: - Decoding
26
38
27 func testDecodeQuotaResponse() throws {
39 func testDecodeQuotaResponse() throws {
@@ -78,21 +90,6 @@ final class ZaiUsageClientTests: XCTestCase {
78
90
79 // MARK: - Fetch
91 // MARK: - Fetch
80
92
81− /// Helper: enqueues a quota response and a subscription response. The two endpoints
82− /// are fetched concurrently, so the order they're enqueued in is the order they'll
83− /// be dequeued — but which request gets which response depends on which arrives first
84− /// at the stub. Since the stub dequeues in FIFO order, we enqueue quota first (it's
85− /// the one whose data matters), then subscription.
86− private func enqueueQuotaAndSubscription(
87− quotaStatus: Int = 200,
88− quotaJSON: Data,
89− subscriptionStatus: Int = 200,
90− subscriptionJSON: Data
91− ) {
92− StubURLProtocol.enqueue(.status(quotaStatus, quotaJSON))
93− StubURLProtocol.enqueue(.status(subscriptionStatus, subscriptionJSON))
94− }
95−
96 func testFetchUsageSuccess() async throws {
93 func testFetchUsageSuccess() async throws {
97 let quotaJSON = """
94 let quotaJSON = """
98 {"code":200,"msg":"Operation successful","success":true,"data":{"limits":[
95 {"code":200,"msg":"Operation successful","success":true,"data":{"limits":[
@@ -130,12 +127,11 @@ final class ZaiUsageClientTests: XCTestCase {
130 }
127 }
131
128
132 func testFetchUsageInvalidKey() async {
129 func testFetchUsageInvalidKey() async {
133− // Both endpoints will 401 — the quota endpoint's 401 is what surfaces.
134 let errorJSON = """
130 let errorJSON = """
135 {"code":401,"msg":"token expired or incorrect","success":false}
131 {"code":401,"msg":"token expired or incorrect","success":false}
136 """.data(using: .utf8)!
132 """.data(using: .utf8)!
137− StubURLProtocol.enqueue(.status(401, errorJSON))
133+ StubURLProtocol.enqueue(.status(401, errorJSON), forURLPath: "/api/monitor/usage/quota/limit")
138− StubURLProtocol.enqueue(.status(401, errorJSON))
134+ StubURLProtocol.enqueue(.status(401, errorJSON), forURLPath: "/api/biz/subscription/list")
139
135
140 let client = makeStubbedClient()
136 let client = makeStubbedClient()
141 do {
137 do {
@@ -160,11 +156,10 @@ final class ZaiUsageClientTests: XCTestCase {
160 {"code":200,"data":[],"success":true}
156 {"code":200,"data":[],"success":true}
161 """.data(using: .utf8)!
157 """.data(using: .utf8)!
162
158
163− // Quota: 500 then 200 (retry). Subscription: 200 (first try).
159+ // Subscription succeeds on first try. Quota fails with 500, then succeeds.
164− // The order is: quota(500), subscription(200), quota(200 on retry).
160+ StubURLProtocol.enqueue(.status(200, subscriptionJSON), forURLPath: "/api/biz/subscription/list")
165− StubURLProtocol.enqueue(.status(500, Data()))
161+ StubURLProtocol.enqueue(.status(500, Data()), forURLPath: "/api/monitor/usage/quota/limit")
166− StubURLProtocol.enqueue(.status(200, subscriptionJSON))
167− StubURLProtocol.enqueue(.status(200, quotaJSON))
162+ StubURLProtocol.enqueue(.status(200, quotaJSON), forURLPath: "/api/monitor/usage/quota/limit")
168
163
169 let client = makeStubbedClient()
164 let client = makeStubbedClient()
170 let report = try await client.fetchUsage(apiKey: "test-key")
165 let report = try await client.fetchUsage(apiKey: "test-key")
@@ -181,9 +176,9 @@ final class ZaiUsageClientTests: XCTestCase {
181 {"code":200,"data":[],"success":true}
176 {"code":200,"data":[],"success":true}
182 """.data(using: .utf8)!
177 """.data(using: .utf8)!
183
178
184− StubURLProtocol.enqueue(.status(429, Data()))
179+ StubURLProtocol.enqueue(.status(200, subscriptionJSON), forURLPath: "/api/biz/subscription/list")
185− StubURLProtocol.enqueue(.status(200, subscriptionJSON))
180+ StubURLProtocol.enqueue(.status(429, Data()), forURLPath: "/api/monitor/usage/quota/limit")
186− StubURLProtocol.enqueue(.status(200, quotaJSON))
181+ StubURLProtocol.enqueue(.status(200, quotaJSON), forURLPath: "/api/monitor/usage/quota/limit")
187
182
188 let client = makeStubbedClient()
183 let client = makeStubbedClient()
189 let report = try await client.fetchUsage(apiKey: "test-key")
184 let report = try await client.fetchUsage(apiKey: "test-key")
@@ -194,8 +189,8 @@ final class ZaiUsageClientTests: XCTestCase {
194 let errorJSON = """
189 let errorJSON = """
195 {"code":401,"msg":"token expired or incorrect","success":false}
190 {"code":401,"msg":"token expired or incorrect","success":false}
196 """.data(using: .utf8)!
191 """.data(using: .utf8)!
197− StubURLProtocol.enqueue(.status(401, errorJSON))
192+ StubURLProtocol.enqueue(.status(401, errorJSON), forURLPath: "/api/monitor/usage/quota/limit")
198− StubURLProtocol.enqueue(.status(401, errorJSON))
193+ StubURLProtocol.enqueue(.status(401, errorJSON), forURLPath: "/api/biz/subscription/list")
199
194
200 let client = makeStubbedClient()
195 let client = makeStubbedClient()
201 do {
196 do {
project.yml+4−4
@@ -6,7 +6,7 @@ name: UsageMenubar
6options:
6options:
7 # Reverse-DNS prefix xcodegen uses when it needs to synthesize a bundle id.
7 # Reverse-DNS prefix xcodegen uses when it needs to synthesize a bundle id.
8 # The app target pins its own id explicitly below, so this is just a default.
8 # The app target pins its own id explicitly below, so this is just a default.
9− bundleIdPrefix: com.ilyakooo0
9+ bundleIdPrefix: soy.iko
10 # Oldest macOS we support. SMAppService (launch at login) requires 13.0+;
10 # Oldest macOS we support. SMAppService (launch at login) requires 13.0+;
11 # we target 14.0 (Sonoma). Keep in sync with MACOSX_DEPLOYMENT_TARGET below.
11 # we target 14.0 (Sonoma). Keep in sync with MACOSX_DEPLOYMENT_TARGET below.
12 deploymentTarget:
12 deploymentTarget:
@@ -47,7 +47,7 @@ targets:
47 properties:
47 properties:
48 CFBundleName: UsageMenubar
48 CFBundleName: UsageMenubar
49 CFBundleDisplayName: Usage Menubar
49 CFBundleDisplayName: Usage Menubar
50− CFBundleIdentifier: com.ilyakooo0.usage-menubar
50+ CFBundleIdentifier: soy.iko.usage-menubar
51 # Overwritten by CI — see the note on MARKETING_VERSION above.
51 # Overwritten by CI — see the note on MARKETING_VERSION above.
52 CFBundleVersion: "1"
52 CFBundleVersion: "1"
53 CFBundleShortVersionString: "1.0.0"
53 CFBundleShortVersionString: "1.0.0"
@@ -57,7 +57,7 @@ targets:
57 NSHighResolutionCapable: true
57 NSHighResolutionCapable: true
58 settings:
58 settings:
59 base:
59 base:
60− PRODUCT_BUNDLE_IDENTIFIER: com.ilyakooo0.usage-menubar
60+ PRODUCT_BUNDLE_IDENTIFIER: soy.iko.usage-menubar
61 PRODUCT_NAME: UsageMenubar
61 PRODUCT_NAME: UsageMenubar
62 INFOPLIST_FILE: UsageMenubar/Info.plist
62 INFOPLIST_FILE: UsageMenubar/Info.plist
63 # Grants the outgoing-network-connection entitlement needed to reach
63 # Grants the outgoing-network-connection entitlement needed to reach
@@ -84,7 +84,7 @@ targets:
84 - target: UsageMenubar
84 - target: UsageMenubar
85 settings:
85 settings:
86 base:
86 base:
87− PRODUCT_BUNDLE_IDENTIFIER: com.ilyakooo0.usage-menubar.tests
87+ PRODUCT_BUNDLE_IDENTIFIER: soy.iko.usage-menubar.tests
88 PRODUCT_NAME: UsageMenubarTests
88 PRODUCT_NAME: UsageMenubarTests
89 BUNDLE_LOADER: "$(TEST_HOST)"
89 BUNDLE_LOADER: "$(TEST_HOST)"
90 TEST_HOST: "$(BUILT_PRODUCTS_DIR)/UsageMenubar.app/Contents/MacOS/UsageMenubar"
90 TEST_HOST: "$(BUILT_PRODUCTS_DIR)/UsageMenubar.app/Contents/MacOS/UsageMenubar"