[ { "path": ".github/workflows/ci.yml", "content": "name: CI\n\non:\n push:\n branches: [\"*\"]\n pull_request:\n\njobs:\n lint-build-test:\n runs-on: macos-latest\n steps:\n - uses: actions/checkout@v6\n\n - name: Select Xcode 26.1.1 (if present) or fallback to default\n run: |\n set -euo pipefail\n for candidate in /Applications/Xcode_26.1.1.app /Applications/Xcode_26.1.app /Applications/Xcode.app; do\n if [[ -d \"$candidate\" ]]; then\n sudo xcode-select -s \"${candidate}/Contents/Developer\"\n echo \"DEVELOPER_DIR=${candidate}/Contents/Developer\" >> \"$GITHUB_ENV\"\n break\n fi\n done\n /usr/bin/xcodebuild -version\n\n - name: Swift toolchain version\n run: |\n set -euo pipefail\n swift --version\n swift package --version\n\n - name: Install lint tools\n run: ./Scripts/install_lint_tools.sh\n\n - name: Lint\n run: ./Scripts/lint.sh lint\n\n - name: Swift Test\n run: swift test --no-parallel\n\n build-linux-cli:\n strategy:\n fail-fast: false\n matrix:\n include:\n - name: linux-x64\n runs-on: ubuntu-24.04\n - name: linux-arm64\n runs-on: ubuntu-24.04-arm\n runs-on: ${{ matrix.runs-on }}\n steps:\n - uses: actions/checkout@v6\n\n - name: Runner info\n run: |\n set -euo pipefail\n uname -a\n uname -m\n\n - name: Setup Swift 6.2.1\n uses: swift-actions/setup-swift@v3\n with:\n swift-version: \"6.2.1\"\n skip-verify-signature: true\n\n - name: Build CodexBarCLI (release, static Swift stdlib)\n run: swift build -c release --product CodexBarCLI --static-swift-stdlib\n\n - name: Swift Test (Linux only)\n run: swift test --parallel\n\n - name: Smoke test CodexBarCLI\n shell: bash\n run: |\n set -euo pipefail\n BIN_DIR=\"$(swift build -c release --product CodexBarCLI --static-swift-stdlib --show-bin-path)\"\n BIN=\"$BIN_DIR/CodexBarCLI\"\n \"$BIN\" --help >/dev/null\n \"$BIN\" --version >/dev/null\n if \"$BIN\" usage --provider codex --web >/dev/null 2>&1; then\n echo \"Expected --web to fail on Linux\"\n exit 1\n fi\n \"$BIN\" usage --provider codex --web 2>&1 | tee /tmp/codexbarcli-stderr.txt >/dev/null || true\n grep -q \"macOS\" /tmp/codexbarcli-stderr.txt\n" }, { "path": ".github/workflows/release-cli.yml", "content": "name: Release Linux CLI\n\non:\n release:\n types: [published]\n workflow_dispatch:\n\npermissions:\n contents: write\n\njobs:\n build-linux-cli:\n strategy:\n fail-fast: false\n matrix:\n include:\n - name: linux-x64\n runs-on: ubuntu-24.04\n - name: linux-arm64\n runs-on: ubuntu-24.04-arm\n runs-on: ${{ matrix.runs-on }}\n steps:\n - uses: actions/checkout@v6\n\n - name: Runner info\n run: |\n set -euo pipefail\n uname -a\n uname -m\n\n - name: Setup Swift 6.2.1\n uses: swift-actions/setup-swift@v3\n with:\n swift-version: \"6.2.1\"\n skip-verify-signature: true\n\n - name: Build CodexBarCLI (release)\n run: swift build -c release --product CodexBarCLI --static-swift-stdlib\n\n - name: Package\n id: pkg\n shell: bash\n run: |\n set -euo pipefail\n\n TAG=\"${GITHUB_REF_NAME}\"\n if [[ -z \"$TAG\" ]]; then\n echo \"Missing tag (GITHUB_REF_NAME).\" >&2\n exit 1\n fi\n\n ARCH=\"$(uname -m)\"\n case \"$ARCH\" in\n x86_64) ARCH=\"x86_64\" ;;\n aarch64|arm64) ARCH=\"aarch64\" ;;\n esac\n\n BIN_DIR=\"$(swift build -c release --product CodexBarCLI --static-swift-stdlib --show-bin-path)\"\n OUT_DIR=\"$(mktemp -d)\"\n install -m 0755 \"$BIN_DIR/CodexBarCLI\" \"$OUT_DIR/CodexBarCLI\"\n ln -s \"CodexBarCLI\" \"$OUT_DIR/codexbar\"\n\n ASSET=\"CodexBarCLI-${TAG}-linux-${ARCH}.tar.gz\"\n (cd \"$OUT_DIR\" && tar czf \"$ASSET\" CodexBarCLI codexbar)\n sha256sum \"$OUT_DIR/$ASSET\" > \"$OUT_DIR/$ASSET.sha256\"\n\n echo \"out_dir=$OUT_DIR\" >> \"$GITHUB_OUTPUT\"\n echo \"asset=$ASSET\" >> \"$GITHUB_OUTPUT\"\n\n - name: Upload release assets\n if: github.event_name == 'release'\n env:\n GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}\n shell: bash\n run: |\n set -euo pipefail\n TAG=\"${GITHUB_REF_NAME}\"\n OUT_DIR=\"${{ steps.pkg.outputs.out_dir }}\"\n ASSET=\"${{ steps.pkg.outputs.asset }}\"\n gh release upload \"$TAG\" \"$OUT_DIR/$ASSET\" \"$OUT_DIR/$ASSET.sha256\" --clobber\n\n - name: Upload workflow artifact (manual runs)\n if: github.event_name != 'release'\n uses: actions/upload-artifact@v6\n with:\n name: codexbar-linux-cli-${{ matrix.name }}\n path: |\n ${{ steps.pkg.outputs.out_dir }}/${{ steps.pkg.outputs.asset }}\n ${{ steps.pkg.outputs.out_dir }}/${{ steps.pkg.outputs.asset }}.sha256\n" }, { "path": ".github/workflows/upstream-monitor.yml", "content": "name: Monitor Upstream Changes\n\non:\n schedule:\n # Run Monday and Thursday at 9 AM UTC\n - cron: '0 9 * * 1,4'\n workflow_dispatch:\n inputs:\n target:\n description: 'Which upstream to check'\n required: false\n default: 'all'\n type: choice\n options:\n - all\n - upstream\n - quotio\n\njobs:\n check-upstreams:\n runs-on: ubuntu-latest\n permissions:\n issues: write\n contents: read\n \n steps:\n - name: Checkout repository\n uses: actions/checkout@v6\n with:\n fetch-depth: 0\n \n - name: Configure git\n run: |\n git config --global user.name 'github-actions[bot]'\n git config --global user.email 'github-actions[bot]@users.noreply.github.com'\n \n - name: Add upstream remotes\n run: |\n git remote add upstream https://github.com/steipete/CodexBar.git || true\n git remote add quotio https://github.com/nguyenphutrong/quotio.git || true\n git fetch upstream\n git fetch quotio\n \n - name: Check for new commits\n id: check\n run: |\n # Count new commits in upstream\n UPSTREAM_NEW=$(git log --oneline main..upstream/main --no-merges 2>/dev/null | wc -l | tr -d ' ')\n echo \"upstream_commits=$UPSTREAM_NEW\" >> $GITHUB_OUTPUT\n \n # Count new commits in quotio (last 7 days)\n QUOTIO_NEW=$(git log --oneline --all --remotes=quotio/main --since=\"7 days ago\" 2>/dev/null | wc -l | tr -d ' ')\n echo \"quotio_commits=$QUOTIO_NEW\" >> $GITHUB_OUTPUT\n \n # Get commit summaries\n echo \"upstream_summary<> $GITHUB_OUTPUT\n git log --oneline main..upstream/main --no-merges 2>/dev/null | head -10 >> $GITHUB_OUTPUT || echo \"No commits\" >> $GITHUB_OUTPUT\n echo \"EOF\" >> $GITHUB_OUTPUT\n \n echo \"quotio_summary<> $GITHUB_OUTPUT\n git log --oneline --remotes=quotio/main --since=\"7 days ago\" 2>/dev/null | head -10 >> $GITHUB_OUTPUT || echo \"No commits\" >> $GITHUB_OUTPUT\n echo \"EOF\" >> $GITHUB_OUTPUT\n \n - name: Create or update issue\n if: steps.check.outputs.upstream_commits > 0 || steps.check.outputs.quotio_commits > 0\n uses: actions/github-script@v8\n with:\n script: |\n const upstreamCommits = '${{ steps.check.outputs.upstream_commits }}';\n const quotioCommits = '${{ steps.check.outputs.quotio_commits }}';\n const upstreamSummary = `${{ steps.check.outputs.upstream_summary }}`;\n const quotioSummary = `${{ steps.check.outputs.quotio_summary }}`;\n \n const body = `## 🔄 Upstream Changes Detected\n \n **steipete/CodexBar:** ${upstreamCommits} new commits\n **quotio:** ${quotioCommits} new commits (last 7 days)\n \n ### steipete/CodexBar Recent Commits\n \\`\\`\\`\n ${upstreamSummary}\n \\`\\`\\`\n \n ### quotio Recent Commits\n \\`\\`\\`\n ${quotioSummary}\n \\`\\`\\`\n \n ### 📋 Review Actions\n \n **Review upstream changes:**\n \\`\\`\\`bash\n ./Scripts/review_upstream.sh upstream\n \\`\\`\\`\n \n **Review quotio changes:**\n \\`\\`\\`bash\n ./Scripts/analyze_quotio.sh\n \\`\\`\\`\n \n **View detailed diffs:**\n \\`\\`\\`bash\n git diff main..upstream/main\n git log -p quotio/main --since='7 days ago'\n \\`\\`\\`\n \n ### 🔗 Links\n - [steipete commits](https://github.com/steipete/CodexBar/compare/${context.sha}...steipete:CodexBar:main)\n - [quotio commits](https://github.com/nguyenphutrong/quotio/commits/main)\n \n ---\n *Auto-generated by upstream-monitor workflow*\n *Last checked: ${new Date().toISOString()}*`;\n \n // Check for existing open issue\n const issues = await github.rest.issues.listForRepo({\n owner: context.repo.owner,\n repo: context.repo.repo,\n labels: 'upstream-sync',\n state: 'open'\n });\n \n if (issues.data.length > 0) {\n // Update existing issue\n await github.rest.issues.update({\n owner: context.repo.owner,\n repo: context.repo.repo,\n issue_number: issues.data[0].number,\n body: body\n });\n \n // Add comment\n await github.rest.issues.createComment({\n owner: context.repo.owner,\n repo: context.repo.repo,\n issue_number: issues.data[0].number,\n body: `🔄 Updated with latest changes (${upstreamCommits} upstream, ${quotioCommits} quotio)`\n });\n } else {\n // Create new issue\n await github.rest.issues.create({\n owner: context.repo.owner,\n repo: context.repo.repo,\n title: '🔄 Upstream Changes Available for Review',\n body: body,\n labels: ['upstream-sync', 'needs-review']\n });\n }\n \n - name: No changes detected\n if: steps.check.outputs.upstream_commits == 0 && steps.check.outputs.quotio_commits == 0\n run: |\n echo \"✅ No new upstream changes detected\"\n echo \"steipete/CodexBar: up to date\"\n echo \"quotio: no commits in last 7 days\"\n\n" }, { "path": ".gitignore", "content": "# Xcode user/session\nxcuserdata/\n.swiftpm/xcode/xcshareddata/\n.codexbar/config.json\n*.env\n*.local\n\n# Build products\n.build/\nDerivedData\n\n# Bundles / artifacts\n# Main app bundle (any variation)\nCodexBar.app/\nCodexBar *.app/\nCodexBar_*.app/\nCodexbar.app/\n# Release artifacts\n*.ipa\n*.dSYM*\n*.zip\n*.delta\n*.dmg\n*.pkg\n*.tar.gz\n*.tgz\nIcon.iconset\ndebug_*.swift\n\n# Misc\n.DS_Store\n.vscode/\n.codex/environments/\n.swiftpm-cache/\n\n# Debug/analysis docs\ndocs/*-analysis.md\ndocs/.astro/\n\n# Swift Package Manager metadata (leave sources tracked)\n# Packages/\n# Package.resolved\n" }, { "path": ".swiftformat", "content": "# SwiftFormat configuration for Peekaboo project\n# Compatible with Swift 6 strict concurrency mode\n\n# IMPORTANT: Don't remove self where it's required for Swift 6 concurrency\n--self insert # Insert self for member references (required for Swift 6)\n--selfrequired # List of functions that require explicit self\n--importgrouping testable-bottom # Group @testable imports at the bottom\n--extensionacl on-declarations # Set ACL on extension members\n\n# Indentation\n--indent 4\n--indentcase false\n--ifdef no-indent\n--xcodeindentation enabled\n\n# Line breaks\n--linebreaks lf\n--maxwidth 120\n\n# Whitespace\n--trimwhitespace always\n--emptybraces no-space\n--nospaceoperators ...,..<\n--ranges no-space\n--someAny true\n\n# Wrapping\n--wraparguments before-first\n--wrapparameters before-first\n--wrapcollections before-first\n--closingparen same-line\n\n# Organization\n--organizetypes class,struct,enum,extension\n--extensionmark \"MARK: - %t + %p\"\n--marktypes always\n--markextensions always\n--structthreshold 0\n--enumthreshold 0\n\n# Swift 6 specific\n--swiftversion 6.2\n\n# Other\n--stripunusedargs closure-only\n--header ignore\n--allman false\n\n# Exclusions\n--exclude .build,.swiftpm,DerivedData,node_modules,dist,coverage,xcuserdata,Core/PeekabooCore/Sources/PeekabooCore/Extensions/NSArray+Extensions.swift\n" }, { "path": ".swiftlint.yml", "content": "# SwiftLint configuration for Peekaboo - Swift 6 compatible\n\n# Paths to include\nincluded:\n - Sources\n - Tests\n\n# Paths to exclude\nexcluded:\n - .build\n - DerivedData\n - \"**/Generated\"\n - \"**/Resources\"\n - \"**/.build\"\n - \"**/Package.swift\"\n - \"**/Tests/Resources\"\n - \"Apps/CLI/.build\"\n - \"**/DerivedData\"\n - \"**/.swiftpm\"\n - Pods\n - Carthage\n - fastlane\n - vendor\n - \"*.playground\"\n # Exclude specific files that should not be linted/formatted\n - \"Core/PeekabooCore/Sources/PeekabooCore/Extensions/NSArray+Extensions.swift\"\n\n# Analyzer rules (require compilation)\nanalyzer_rules:\n - unused_declaration\n - unused_import\n\n# Enable specific rules\nopt_in_rules:\n - array_init\n - closure_spacing\n - contains_over_first_not_nil\n - empty_count\n - empty_string\n - explicit_init\n - fallthrough\n - fatal_error_message\n - first_where\n - joined_default_parameter\n - last_where\n - literal_expression_end_indentation\n - multiline_arguments\n - multiline_parameters\n - operator_usage_whitespace\n - overridden_super_call\n - pattern_matching_keywords\n - private_outlet\n - prohibited_super_call\n - redundant_nil_coalescing\n - sorted_first_last\n - switch_case_alignment\n - unneeded_parentheses_in_closure_argument\n - vertical_parameter_alignment_on_call\n\n# Disable rules that conflict with Swift 6 or our coding style\ndisabled_rules:\n # Swift 6 requires explicit self - disable explicit_self rule\n - explicit_self\n \n # SwiftFormat handles these\n - trailing_whitespace\n - trailing_newline\n - trailing_comma\n - vertical_whitespace\n - indentation_width\n \n # Too restrictive or not applicable\n - identifier_name # Single letter names are fine in many contexts\n - file_header\n - explicit_top_level_acl\n - explicit_acl\n - explicit_type_interface\n - missing_docs\n - required_deinit\n - prefer_nimble\n - quick_discouraged_call\n - quick_discouraged_focused_test\n - quick_discouraged_pending_test\n - anonymous_argument_in_multiline_closure\n - no_extension_access_modifier\n - no_grouping_extension\n - switch_case_on_newline\n - strict_fileprivate\n - extension_access_modifier\n - convenience_type\n - no_magic_numbers\n - one_declaration_per_file\n - vertical_whitespace_between_cases\n - vertical_whitespace_closing_braces\n - superfluous_else\n - number_separator\n - prefixed_toplevel_constant\n - opening_brace\n - trailing_closure\n - contrasted_opening_brace\n - sorted_imports\n - redundant_type_annotation\n - shorthand_optional_binding\n - untyped_error_in_catch\n - file_name\n - todo\n \n# Rule configurations\nforce_cast: warning\nforce_try: warning\n\n# identifier_name rule disabled - see disabled_rules section\n\ntype_name:\n min_length:\n warning: 2\n error: 1\n max_length:\n warning: 60\n error: 80\n\nfunction_body_length:\n warning: 150\n error: 300\n\nfile_length:\n warning: 1500\n error: 2500\n ignore_comment_only_lines: true\n\ntype_body_length:\n warning: 800\n error: 1200\n\ncyclomatic_complexity:\n warning: 20\n error: 120\n\nlarge_tuple:\n warning: 4\n error: 5\n\nnesting:\n type_level:\n warning: 4\n error: 6\n function_level:\n warning: 5\n error: 7\n\nline_length:\n warning: 120\n error: 250\n ignores_comments: true\n ignores_urls: true\n\n# Custom rules can be added here if needed\n\n# Reporter type\nreporter: \"xcode\"\n" }, { "path": ".swiftpm/xcode/package.xcworkspace/contents.xcworkspacedata", "content": "\n\n \n \n\n" }, { "path": "AGENTS.md", "content": "# Repository Guidelines\n\n## Project Structure & Modules\n- `Sources/CodexBar`: Swift 6 menu bar app (usage/credits probes, icon renderer, settings). Keep changes small and reuse existing helpers.\n- `Tests/CodexBarTests`: XCTest coverage for usage parsing, status probes, icon patterns; mirror new logic with focused tests.\n- `Scripts`: build/package helpers (`package_app.sh`, `sign-and-notarize.sh`, `make_appcast.sh`, `build_icon.sh`, `compile_and_run.sh`).\n- `docs`: release notes and process (`docs/RELEASING.md`, screenshots). Root-level zips/appcast are generated artifacts—avoid editing except during releases.\n\n## Build, Test, Run\n- Dev loop: `./Scripts/compile_and_run.sh` kills old instances, runs `swift build` + `swift test`, packages, relaunches `CodexBar.app`, and confirms it stays running.\n- Quick build/test: `swift build` (debug) or `swift build -c release`; `swift test` for the full XCTest suite.\n- Package locally: `./Scripts/package_app.sh` to refresh `CodexBar.app`, then restart with `pkill -x CodexBar || pkill -f CodexBar.app || true; cd /Users/steipete/Projects/codexbar && open -n /Users/steipete/Projects/codexbar/CodexBar.app`.\n- Release flow: `./Scripts/sign-and-notarize.sh` (arm64 notarized zip) and `./Scripts/make_appcast.sh `; follow validation steps in `docs/RELEASING.md`.\n\n## Coding Style & Naming\n- Enforce SwiftFormat/SwiftLint: run `swiftformat Sources Tests` and `swiftlint --strict`. 4-space indent, 120-char lines, explicit `self` is intentional—do not remove.\n- Favor small, typed structs/enums; maintain existing `MARK` organization. Use descriptive symbols; match current commit tone.\n\n## Testing Guidelines\n- Add/extend XCTest cases under `Tests/CodexBarTests/*Tests.swift` (`FeatureNameTests` with `test_caseDescription` methods).\n- Always run `swift test` (or `./Scripts/compile_and_run.sh`) before handoff; add fixtures for new parsing/formatting scenarios.\n- After any code change, run `pnpm check` and fix all reported format/lint issues before handoff.\n\n## Commit & PR Guidelines\n- Commit messages: short imperative clauses (e.g., “Improve usage probe”, “Fix icon dimming”); keep commits scoped.\n- PRs/patches should list summary, commands run, screenshots/GIFs for UI changes, and linked issue/reference when relevant.\n\n## Agent Notes\n- Use the provided scripts and package manager (SwiftPM); avoid adding dependencies or tooling without confirmation.\n- Validate behavior against the freshly built bundle; restart via the pkill+open command above to avoid running stale binaries.\n- To guarantee the right bundle is running after a rebuild, use: `pkill -x CodexBar || pkill -f CodexBar.app || true; cd /Users/steipete/Projects/codexbar && open -n /Users/steipete/Projects/codexbar/CodexBar.app`.\n- After any code change that affects the app, always rebuild with `Scripts/package_app.sh` and restart the app using the command above before validating behavior.\n- If you edited code, run `scripts/compile_and_run.sh` before handoff; it kills old instances, builds, tests, packages, relaunches, and verifies the app stays running.\n- Per user request: after every edit (code or docs), rebuild and restart using `./Scripts/compile_and_run.sh` so the running app reflects the latest changes.\n- Release script: keep it in the foreground; do not background it—wait until it finishes.\n- Release keys: find in `~/.profile` if missing (Sparkle + App Store Connect).\n- Prefer modern SwiftUI/Observation macros: use `@Observable` models with `@State` ownership and `@Bindable` in views; avoid `ObservableObject`, `@ObservedObject`, and `@StateObject`.\n- Favor modern macOS 15+ APIs over legacy/deprecated counterparts when refactoring (Observation, new display link APIs, updated menu item styling, etc.).\n- Keep provider data siloed: when rendering usage or account info for a provider (Claude vs Codex), never display identity/plan fields sourced from a different provider.***\n- Claude CLI status line is custom + user-configurable; never rely on it for usage parsing.\n- Cookie imports: default Chrome-only when possible to avoid other browser prompts; override via browser list when needed.\n" }, { "path": "CHANGELOG.md", "content": "# Changelog\n\n## Unreleased\n### Highlights\n- Add Alibaba Coding Plan provider with region-aware quota fetching, widget integration, and browser-cookie import defaults (#574).\n- Add GPT-5.4 mini and nano pricing (#561). Thanks @iam-brain!\n- Add per-model token counts to cost history (#546). Thanks @iam-brain!\n- Refactor the Claude provider end to end into clearer, better-tested components while preserving behavior (#494).\n\n### Providers & Usage\n- Alibaba: add Coding Plan provider support with region-aware web/API quota fetching, widget integration, and browser-cookie import defaults (#574).\n- Claude: refactor the provider end to end into clearer components, with baseline docs and expanded tests to lock down behavior (#494).\n- Codex: add GPT-5.4 mini and nano pricing (#561). Thanks @iam-brain!\n- Cost history: add per-model token counts so token usage is broken out by model (#546). Thanks @iam-brain!\n\n### Menu & Settings\n- Menu: wrap long status blurbs and preserve wrapped titles for multiline entries (#543). Thanks @zkforge!\n\n## 0.18.0 — 2026-03-15\n### Highlights\n- Add Kilo provider support with API/CLI source modes, widget integration, and pass/credit handling (#454). Built on work by @coreh.\n- Add Ollama provider, including token-account support in Settings and CLI (#380). Thanks @CryptoSageSnr!\n- Add OpenRouter provider for credit-based usage tracking (#396). Thanks @chountalas!\n- Add Codex historical pace with risk forecasting, backfill, and zero-usage-day handling (#482, supersedes #438). Thanks @tristanmanchester!\n- Add a merged-menu Overview tab with configurable providers and row-to-provider navigation (#416). @ratulsarna\n- Add an experimental option to suppress Claude Keychain prompts (#388).\n- Reduce CPU/energy regressions and JSONL scanner overhead in Codex/web usage paths (#402, #392). Thanks @bald-ai and @asonawalla!\n\n### Providers & Usage\n- Codex: add historical pace risk forecasting and backfill, gate pace computation by display mode, and handle zero-usage days in historical data (#482, supersedes #438). Thanks @tristanmanchester!\n- Kilo: add provider support with source-mode fallback, clearer credential/login guidance, auto top-up activity labeling, zero-balance credit handling, and pass parsing/menu rendering (#454). Thanks @coreh!\n- Ollama: add provider support with token-account support in app/CLI, Chrome-default auto cookie import, and manual-cookie mode (#380). Thanks @CryptoSageSnr!\n- OpenRouter: add provider support with credit tracking, key-quota popup support, token-account labels, fallback status icons, and updated icon/color (#396). Thanks @chountalas!\n- Gemini: show separate Pro, Flash, and Flash Lite meters by splitting Gemini CLI quota buckets for `gemini-2.5-flash` and `gemini-2.5-flash-lite` (#496). Thanks @aladh\n- Codex: in percent display mode with \"show remaining,\" show remaining credits in the menu bar when session or weekly usage is exhausted (#336). Thanks @teron131!\n- Claude: surface rate-limit errors from the CLI `/usage` probe with a user-friendly message, and harden \"Failed to load usage data\" matching against whitespace-collapsed output.\n- Claude: restore weekly/Sonnet reset parsing from whitespace-collapsed CLI `/usage` output so reset times and pace details still appear after CLI fallback.\n- Claude: fix extra-usage double conversion so OAuth/Web values stay on a single normalization path (#472, supersedes #463). Thanks @Priyans-hu!\n- Claude: remove root-directory mtime short-circuiting in cost scanning so new session logs inside existing `~/.claude/projects/*` folders are discovered reliably (#462, fixes #411). Thanks @Priyans-hu!\n- Copilot: harden free-plan quota parsing and fallback behavior by treating underdetermined values as unknown, preserving missing metadata as nil (#432, supersedes #393). Thanks @emanuelst!\n- OpenCode: treat explicit `null` subscription responses as missing usage data, skip POST fallback, and return a clearer workspace-specific error (#412).\n- OpenCode: surface clearer HTTP errors. Thanks @SalimBinYousuf1!\n- Codex: preserve exact GPT-5 model IDs in local cost history, add GPT-5.4 pricing, and label zero-cost `gpt-5.3-codex-spark` sessions as \"Research Preview\" in cost breakdowns (#511). Thanks @iam-brain!\n- Augment: prevent refresh stalls when `auggie account status` hangs by replacing unbounded CLI waits with timed subprocess execution and fallback handling (#481). Thanks @bryant24hao!\n- Update Kiro parsing for `kiro-cli` 1.24+ / Q Developer formats and non-managed plan handling (#288). Thanks @kilhyeonjun!\n- Kimi: in automatic metric mode, prioritize the 5-hour rate-limit window for menu bar and merged highest-usage calculations (#390). Thanks @ajaxjiang96!\n- Browser cookie import: match Gecko `*.default*` profile directories case-insensitively so Firefox/Zen cookie detection works with uppercase `.Default` directories (#422). Thanks @bald-ai!\n- MiniMax: make both Settings \"Open Coding Plan\" actions region-aware so China mainland selection opens `platform.minimaxi.com` instead of the global domain (#426, fixes #378). Thanks @bald-ai!\n- Menu: rebuild the merged provider switcher when “Show usage as used” changes so switcher progress updates immediately (#306). Thanks @Flohhhhh!\n- Warp: update API key setup guidance.\n- Claude: update the \"not installed\" help link to the current Claude Code documentation URL (#431). Thanks @skebby11!\n- Fix Claude setup message package name (#376). Thanks @daegwang!\n\n### Menu & Settings\n- Merged menu: keep Merge Icons, the switcher, and Overview tied to user-enabled providers even when some providers are temporarily unavailable, while defaulting menu content and icon state to an available provider when possible (#525). Thanks @Astro-Han!\n- Merged menu: add an Overview switcher tab that shows up to three provider usage rows in provider order (#416).\n- Settings: add \"Overview tab providers\" controls to choose/deselect Overview providers, with persisted selection reconciliation as enabled providers change (#416).\n- Menu: hide contextual provider actions while Overview is selected and rebuild switcher state when overview availability changes (#416).\n\n### Claude OAuth & Keychain\n- Add an experimental Claude OAuth Security-CLI reader path and option in settings.\n- Apply stored prompt mode and fallback policy to silent/noninteractive keychain probes.\n- Add cooldown for background OAuth keychain retries.\n- Disable experimental toggle when keychain access is disabled.\n- Use a `claude-code/` User-Agent for OAuth usage requests instead of a generic identifier.\n\n### Performance & Reliability\n- Codex/OpenAI web: reduce CPU and energy overhead by shortening failed CLI probe windows, capping web retry timeouts, and using adaptive idle blink scheduling (#402). Thanks @bald-ai!\n- Cost usage scanner: optimize JSONL chunk parsing to avoid buffer-front removal overhead on large logs (#392). Thanks @asonawalla!\n- TTY runner: fence shutdown registration to avoid launch/shutdown races, isolate process groups before shutdown rejection, and ensure lingering CLI descendants are cleaned up on app termination (#429). Thanks @uraimo!\n\n\n## 0.18.0-beta.3 — 2026-02-13\n### Highlights\n- Claude OAuth/keychain flows were reworked across a series of follow-up PRs to reduce prompt storms, stabilize background behavior, surface a setting to control prompt policy and make failure modes deterministic (#245, #305, #308, #309, #364). Thanks @manikv12!\n- Claude: harden Claude Code PTY capture for `/usage` and `/status` (prompt automation, safer command palette confirmation, partial UTF-8 handling, and parsing guards against status-bar context meters) (#320).\n- New provider: Warp (credits + add-on credits) (#352). Thanks @Kathie-yu!\n- Provider correctness fixes landed for Cursor plan parsing and MiniMax region routing (#240, #234, #344). Thanks @robinebers and @theglove44!\n- Menu bar animation behavior was hardened in merged mode and fallback mode (#283, #291). Thanks @vignesh07 and @Ilakiancs!\n- CI/tooling reliability improved via pinned lint tools, deterministic macOS test execution, and PTY timing test stabilization plus Node 24-ready GitHub Actions upgrades (#292, #312, #290).\n\n### Claude OAuth & Keychain\n- Claude OAuth creds are cached in CodexBar Keychain to reduce repeated prompts.\n- Prompts can still appear when Claude OAuth credentials are expired, invalid, or missing and re-auth is required.\n- In Auto mode, background refresh keeps prompts suppressed; interactive prompts are limited to user actions (menu open or manual refresh).\n- OAuth-only mode remains strict (no silent Web/CLI fallback); Auto mode may do one delegated CLI refresh + one OAuth retry before falling back.\n- Preferences now expose a Claude Keychain prompt policy (Never / Only on user action / Always allow prompts) under Providers → Claude; if global Keychain access is disabled in Advanced, this control remains visible but inactive.\n\n### Provider & Usage Fixes\n- Warp: add Warp provider support (credits + add-on credits), configurable via Settings or `WARP_API_KEY`/`WARP_TOKEN` (#352). Thanks @Kathie-yu!\n- Cursor: compute usage against `plan.limit` rather than `breakdown.total` to avoid incorrect limit interpretation (#240). Thanks @robinebers!\n- MiniMax: correct API region URL selection to route requests to the expected regional endpoint (#234). Thanks @theglove44!\n- MiniMax: always show the API region picker and retry the China endpoint when the global host rejects the token to avoid upgrade regressions for users without a persisted region (#344). Thanks @apoorvdarshan!\n- Claude: add Opus 4.6 pricing so token cost scanning tracks USD consumed correctly (#348). Thanks @arandaschimpf!\n- z.ai: handle quota responses with missing token-limit fields, avoid incorrect used-percent calculations, and harden empty-response behavior with safer logging (#346). Thanks @MohamedMohana and @halilertekin!\n- z.ai: fix provider visibility in the menu when enabled with token-account credentials (availability now considers the effective fetch environment).\n- Amp: detect login redirects during usage fetch and fail fast when the session is invalid (#339). Thanks @JosephDoUrden!\n- Resource loading: fix app bundle lookup path to avoid \"could not load resource bundle\" startup failures (#223). Thanks @validatedev!\n- OpenAI Web dashboard: keep WebView instances cached for reuse to reduce repeated network fetch overhead; tests were updated to avoid network-dependent flakes (#284). Thanks @vignesh07!\n- Token-account precedence: selected token account env injection now correctly overrides provider config `apiKey` values in app and CLI environments. Thanks @arvindcr4!\n- Claude: make Claude CLI probing more resilient by scoping auto-input to the active subcommand and trimming to the latest Usage panel before parsing to avoid false matches from earlier screen fragments (#320).\n\n### Menu Bar & UI Behavior\n- Prevent fallback-provider loading animation loops (battery/CPU drain when no providers are enabled) (#283). Thanks @vignesh07!\n- Prevent status overlay rendering for disabled providers while in merged mode (#291). Thanks @Ilakiancs!\n\n### CI, Tooling & Test Stability\n- Pin SwiftFormat/SwiftLint versions and harden lint installer behavior (version drift + temp-file leak fixes) (#292).\n- Use more deterministic macOS CI test settings (including non-parallel paths where needed) and align runner/toolchain behavior for stability (#292).\n- Stabilize PTY command timing tests to reduce CI flakiness (#312).\n- Upgrade `actions/checkout` to v6 and `actions/github-script` to v8 for Node 24 compatibility in `upstream-monitor.yml` (#290). Thanks @salmanmkc!\n- Tests: add TaskLocal-based keychain/cache overrides so keychain gating and KeychainCacheStore test stores do not leak across concurrent test execution (#320).\n\n### Docs & Maintenance\n- Update docs for Claude data fetch behavior and keychain troubleshooting notes.\n- Update MIT license year.\n\n## 0.18.0-beta.2 — 2026-01-21\n### Highlights\n- OpenAI web dashboard refresh cadence now follows 5× the base refresh interval.\n- OpenAI web dashboard WebView is kept warm between scrapes to avoid repeated SPA downloads while idle CPU stays low (#284). Thanks @vignesh07!\n- Menu bar: avoid fallback animation loop when all providers are disabled (#283). Thanks @vignesh07!\n- Codex settings now include a toggle to disable OpenAI web extras.\n\n### Providers\n- Providers: add Dia browser support across cookie import and profile detection (#209). Thanks @validatedev!\n- Codex: include archived session logs in local token cost scanning and dedupe by session id.\n- Claude: harden CLI /usage parsing and avoid ANTHROPIC_* env interference during probes.\n\n### Menu & Menu Bar\n- Menu: opening OpenAI web submenus triggers a refresh when the data is stale.\n- Menu: fix usage line labels to honor “Show usage as used”.\n- Debug: add a toggle to keep Codex/Claude CLI sessions alive between probes.\n- Debug: add a button to reset CLI probe sessions.\n- App icon: use the classic icon on macOS 15 and earlier while keeping Liquid Glass for macOS 26+ (#178). Thanks @zerone0x!\n\n## 0.18.0-beta.1 — 2026-01-18\n### Highlights\n- New providers: OpenCode (web usage), Vertex AI, Kiro, Kimi, Kimi K2, Augment, Amp, Synthetic.\n- Provider source controls: usage source pickers for Codex/Claude, manual cookie headers, cookie caching with source/timestamp.\n- Menu bar upgrades: display mode picker (percent/pace/both), auto-select near limit, absolute reset times, pace summary line.\n- CLI/config revamp: config-backed provider settings, JSON-only errors, config validate/dump.\n\n### Providers\n- OpenCode: add web usage provider with workspace override + Chrome-first cookie import (#188). Thanks @anthnykr!\n- OpenCode: refresh provider logo (#190). Thanks @anthnykr!\n- Vertex AI: add provider with quota-based usage from gcloud ADC. Thanks @bahag-chaurasiak!\n- Vertex AI: token costs are shown via the Claude provider (same local logs).\n- Vertex AI: harden quota usage parsing for edge-case responses.\n- Kiro: add CLI-based usage provider via kiro-cli. Thanks @neror!\n- Kiro: clean up provider wiring and show plan name in the menu.\n- Kiro: harden CLI idle handling to avoid partial usage snapshots (#145). Thanks @chadneal!\n- Kimi: add usage provider with cookie-based API token stored in Keychain (#146). Thanks @rehanchrl!\n- Kimi K2: add API-key usage provider for credit totals (#147). Thanks @0-CYBERDYNE-SYSTEMS-0!\n- Augment: add provider with browser-cookie usage tracking.\n- Augment: prefer Auggie CLI usage with web fallback, plus session refresh + recovery tools (#142). Thanks @bcharleson!\n- Amp: add provider with Amp Free usage tracking (#167). Thanks @duailibe!\n- Synthetic: add API-key usage provider with quota snapshots (#171). Thanks @monotykamary!\n- JetBrains AI: include IDEs missing quota files, expand custom paths, and add Android Studio base paths (#194). Thanks @steipete!\n- JetBrains AI: detect IDE directories case-insensitively (#200). Thanks @zerone0x!\n- Cursor: support legacy request-based plans and show individual on-demand usage (#125) — thanks @vltansky\n- Cursor: avoid Intel crash when opening login and harden WebKit teardown. Thanks @meghanto!\n- Cursor: load stored session cookies before reads to make relaunches deterministic.\n- z.ai: add BigModel CN region option for API endpoint selection (#140). Thanks @nailuoGG!\n- MiniMax: add China mainland region option + host overrides (#143). Thanks @nailuoGG!\n- MiniMax: support API token or cookie auth; API token takes precedence and hides cookie UI (#149). Thanks @aonsyed!\n- Gemini: prefer loadCodeAssist project IDs for quota fetches (#172). Thanks @lolwierd!\n- Gemini: honor loadCodeAssist project IDs for quota + support Nix CLI layout (#184). Thanks @HaukeSchnau!\n- Claude: fix OAuth “Extra usage” spend/limit units when the API returns minor currency units (#97).\n- Claude: rescale extra usage costs when plan hints are missing and prefer web plan hints for extras (#181). Thanks @jorda0mega!\n- Usage formatting: fix currency parsing/formatting on non-US locales (e.g., pt-BR). Thanks @mneves75!\n\n### Provider Sources & Security\n- Providers: cache browser cookies in Keychain (per provider) and show cached source/time in settings.\n- Codex/Claude/Cursor/Factory/MiniMax: cookie sources now include Manual (paste a Cookie header) in addition to Automatic.\n- Codex/Claude/Cursor/Factory/MiniMax: skip cookie imports from browsers without usable cookie stores (profile/cookie DB) to avoid unnecessary Keychain prompts.\n- Providers: suppress repeated Chromium Keychain prompts after access denied and honor disabled Keychain access.\n\n### Preferences & Settings\n- Preferences: swap provider refresh button and enable toggle order.\n- Preferences: animate settings width and widen Providers on selection.\n- Preferences: shrink default settings size and reduce overall height.\n- Preferences: move “Hide personal information” to Advanced.\n- Providers: shorten fetch subtitle to relative time only.\n- Preferences: soften provider sidebar background and stabilize drag reordering.\n- Preferences: restrict provider drag handle to handle-only.\n- Preferences: move provider refresh timing to a dedicated second line.\n- Preferences: tighten provider usage metrics spacing.\n- Preferences: show refresh timing inline in provider detail subtitle.\n- Preferences: move “Access OpenAI via web” into Providers → Codex.\n- Preferences: add usage source pickers for Codex + Claude with auto fallback.\n- Preferences: add cookie source pickers with contextual helper text for the selected mode.\n- Preferences: move “Disable Keychain access” to Advanced and require manual cookies when enabled.\n- Preferences: add per-provider menu bar metric picker (#185) — thanks @HaukeSchnau\n- Preferences: tighten provider rows (inline pickers, compact layout, inline refresh + auto-source status).\n- Preferences: remove the “experimental” label from Antigravity.\n\n### Menu & Menu Bar\n- Menu: add a toggle to show reset times as absolute clock values (instead of countdowns).\n- Menu: show an “Open Terminal” action when Claude OAuth fails.\n- Menu: add “Hide personal information” toggle and redact emails in menu UI (#137). Thanks @t3dotgg!\n- Menu: keep a pace summary line alongside the visual marker (#155). Thanks @antons!\n- Menu: reduce provider-switch flicker and avoid redundant menu card sizing for faster opens (#132). Thanks @ibehnam!\n- Menu: keep background refresh on open without forcing token usage (#158). Thanks @weequan93!\n- Menu: Cursor switcher shows On-Demand remaining when Plan is exhausted in show-remaining mode (#193). Thanks @vltansky!\n- Menu: avoid single-letter wraps in provider switcher titles.\n- Menu: widen provider switcher buttons to avoid clipped titles.\n- Menu bar: rebuild provider status items on reorder so icons update correctly.\n- Menu bar: optional auto-select provider closest to its rate limit and keep switcher progress visible (#159). Thanks @phillco!\n- Menu bar: add display mode picker for percent/pace/both in the menu bar icon (#169). Thanks @PhilETaylor!\n- Menu bar: fix combined loading indicator flicker during loading animation (incl. debug replay).\n- Menu bar: prevent blink updates from clobbering the loading animation.\n\n### CLI & Config\n- CLI: respect the reset time display setting.\n- CLI: add pink accents, usage bars, and weekly pace lines to text output.\n- CLI: add config-backed provider settings, `--json-only`, and `--source api` for key-based providers.\n- CLI: add `config validate`/`config dump` commands and per-provider JSON error payloads.\n- CLI/App: move provider secrets + ordering to `~/.codexbar/config.json` (no Keychain persistence).\n- Providers: resolve API tokens from config/env only (no Keychain fallback).\n\n### Dev & Tests\n- Dev: move Chromium profile discovery into SweetCookieKit (adds Helium net.imput.helium). Thanks @hhushhas!\n- Dev: bump SweetCookieKit to 0.2.0.\n- Dev: migrate stored Keychain items to reduce rebuild prompts.\n- Dev: move path debug snapshot off the main thread and debounce refreshes to avoid startup hitches (#131). Thanks @ibehnam!\n- Tests: expand Kiro CLI coverage.\n- Tests: stabilize Claude PTY integration cleanup and reset CLI sessions after probes.\n- Tests: kill leaked codex app-server after tests.\n- Tests: add regression coverage for merged loading icon layout stability.\n- Tests: cover config validation and JSON-only CLI errors.\n- Build: stabilize Swift test runtime.\n\n## 0.17.0 — 2025-12-31\n- New providers: MiniMax.\n- Keychain: show a preflight explanation before macOS prompts for OAuth tokens or cookie decryption.\n- Providers: defer z.ai + Copilot Keychain reads until the user interacts with the token field.\n- Menu bar: avoid status item menu reattachment and layout flips during refresh to reduce icon flicker.\n- Dev: align SweetCookieKit local-storage tests with Swift Testing.\n- Charts: align hover selection bands with visible bars in credits + usage breakdown history.\n- About: fix website link in the About panel. Thanks @felipeorlando!\n\n## 0.16.1 — 2025-12-29\n- Menu: reduce layout thrash when opening menus and sizing charts. Thanks @ibehnam!\n- Packaging: default release notarization builds universal (arm64 + x86_64) zip.\n- OpenAI web: reduce idle CPU by suspending cached WebViews when not scraping. Thanks @douglascamata!\n- Icons: switch provider brand icons to SVGs for sharper rendering. Thanks @vandamd!\n\n## 0.16.0 — 2025-12-29\n- Menu bar: optional “percent mode” (provider brand icons + percentage labels) via Advanced toggle.\n- CLI: add `codexbar cost` to print local cost usage (text/JSON) for Codex + Claude.\n- Cost: align local cost scanner with ccusage; stabilize parsing/decoding and handle large JSONL lines.\n- Claude: skip pricing for unknown models (tokens still tracked) to avoid hard-coded legacy prices.\n- Performance: reduce menu bar CPU usage by caching morph icons, skipping redundant status-item updates, and caching provider enablement/order during animations.\n- Menu: improve provider switcher hover contrast in light mode.\n- Icons: refresh Droid + Claude brand assets to better match menu sizing.\n- CI: avoid interactive login-shell probes to reduce noisy “CLI missing” errors.\n\n## 0.15.3 — 2025-12-28\n- Codex: default to OAuth usage API (ChatGPT backend) with CLI-only override in Debug.\n- Codex: map OAuth credits balance directly, avoiding web fallback for credits.\n- Preferences: add optional “Access OpenAI via web” toggle and show blended source labels when web extras are active.\n- Copilot: replace blocking auth wait dialog with a non-modal sheet to avoid stuck login.\n\n## 0.15.2 — 2025-12-28\n- Copilot: fix device-flow waiting modal to close reliably after auth (and avoid stuck waits).\n- Packaging: include the KeyboardShortcuts resource bundle to prevent Settings → Keyboard shortcut crashes in packaged builds.\n\n## 0.15.1 — 2025-12-28\n- Preferences: fix provider API key fields reusing the wrong input when switching rows.\n- Preferences: avoid Advanced tab crash when opening settings.\n\n## 0.15.0 — 2025-12-28\n- New providers: Droid (Factory), Cursor, z.ai, Copilot.\n- macOS: CodexBar now supports Intel Macs (x86_64 builds + Sonoma fallbacks). Thanks @epoyraz!\n- Droid (Factory): new provider with Standard + Premium usage via browser cookies, plus dashboard + status links. Thanks @shashank-factory!\n- Menu: allow multi-line error messages in the provider subtitle (up to 4 lines).\n- Menu: fix subtitle sizing for multi-line error states.\n- Menu: avoid clipping on multi-line error subtitles.\n- Menu: widen the menu card when 7+ providers are enabled.\n- Providers: Codex, Claude Code, Cursor, Gemini, Antigravity, z.ai.\n- Gemini: switch plan detection to loadCodeAssist tier lookup (Paid/Workspace/Free/Legacy). Thanks @381181295!\n- Codex: OpenAI web dashboard is now the primary source for usage + credits; CLI fallback only when no matching cookies exist.\n- Claude: prefer OAuth when credentials exist; fall back to web cookies or CLI (thanks @ibehnam).\n- CLI: replace `--web`/`--claude-source` with `--source` (auto/web/cli/oauth); auto falls back only when cookies are missing.\n- Homebrew: cask now installs the `codexbar` CLI symlink. Thanks @dalisoft!\n- Cursor: add new usage provider with browser cookie auth (cursor.com + cursor.sh), on-demand bar support, and dashboard access.\n- Cursor: keep stored sessions on transient failures; clear only on invalid auth.\n- z.ai: new provider support with Tokens + MCP usage bars and MCP details submenu; API token now lives in Preferences (stored in Keychain); usage bars respect the show-used toggle. Thanks @uwe-schwarz for the initial work!\n- Copilot: new GitHub Copilot provider with device flow login plus Premium + Chat usage bars (including CLI support). Thanks @roshan-c!\n- Preferences: fix Advanced Display checkboxes and move the Quit button to the bottom of General.\n- Preferences: hide “Augment Claude via web” unless Claude usage source is CLI; rename the cost toggle to “Show cost summary”.\n- Preferences: add an Advanced toggle to show/hide optional Codex Credits + Claude Extra usage sections (on by default).\n- Widgets: add a new “CodexBar Switcher” widget that lets you switch providers and remember the selection.\n- Menu: provider switcher now uses crisp brand icons with equal-width segments and a per-provider usage indicator.\n- Menu: tighten provider switcher sizing and increase spacing between label and weekly indicator bar.\n- Menu: provider switcher no longer forces a wider menu when many providers are enabled; segments clamp to the menu width.\n- Menu: provider switcher now aligns to the same horizontal padding grid as the menu cards when space allows.\n- Dev: `compile_and_run.sh` now force-kills old instances to avoid launching duplicates.\n- Dev: `compile_and_run.sh` now waits for slow launches (polling for the process).\n- Dev: `compile_and_run.sh` now launches a single app instance (no more extra windows).\n- CI: build/test Linux `CodexBarCLI` (x86_64 + aarch64) and publish release assets as `CodexBarCLI--linux-.tar.gz` (+ `.sha256`).\n- CLI: add alias fallback for Codex/Claude detection when PATH lookups fail.\n- Providers: support Arc browser cookies for Factory/Droid (and other Chromium-based cookie imports).\n- Providers: support ChatGPT Atlas browser data for Chromium cookie imports.\n- Providers: accept Auth.js secure session cookies for Factory/Droid login detection.\n- Providers: accept Factory auth session cookies (session/access-token) for Droid.\n- Droid: surface Factory API errors instead of masking them as missing sessions.\n- Droid: retry auth without access-token cookies when Factory flags a stale token.\n- Droid: try all detected browser profiles before giving up.\n- Droid: fall back to auth.factory.ai endpoints when cookies live on the auth host.\n- Droid: use WorkOS refresh tokens from browser local storage when cookies fail.\n- Droid: read WorkOS refresh tokens from Safari local storage.\n- Droid: try stored/WorkOS tokens before Chrome cookies to reduce Chrome Safe Storage prompts.\n- Menu: provider switcher bars now track primary quotas (Plan/Tokens/Pro), with Premium shown for Droid.\n- Menu: avoid duplicate summary blocks when a provider has no action rows.\n- OpenAI web: ignore cookie sets without session tokens to avoid false-positive dashboard fetches.\n- Providers: hide z.ai in the menu until an API key is set.\n- Menu: refresh runs automatically when opening the menu with a short retry (refresh row removed).\n- Menu: hide the Status Page row when a provider has no status URL.\n- Menu: align switcher bar with the “show usage as used” toggle.\n- Antigravity: fix lsof port filtering by ANDing listen + pid conditions. Thanks @shaw-baobao!\n- Claude: default to Claude Code OAuth usage API (credentials from Keychain or `~/.claude/.credentials.json`), with Debug selector + `--claude-source` CLI override (OAuth/Web/CLI).\n- OpenAI web: allow importing any signed-in browser session when Codex email is unknown (first-run friendly).\n- Core: Linux CLI builds now compile (mac-only WebKit/logging gated; FoundationNetworking imports where needed).\n- Core: fix CI flake for Claude trust prompts by making PTY writes fully reliable.\n- Core: Cursor provider is macOS-only (Linux CLI builds stub it).\n- Core: make `RateWindow` equatable (used by OpenAI dashboard snapshots and tests).\n- Tests: cover alias fallback resolution for Codex/Claude and add Linux platform gating coverage (run in CI).\n- Tests: cover hiding Codex Credits + Claude Extra usage via the Advanced toggle.\n- Docs: expand CLI docs for Linux install + flags.\n\n## 0.14.0 — 2025-12-25\n- New providers: Antigravity.\n- Antigravity: new local provider for the Antigravity language server (Claude + Gemini quotas) with an experimental toggle; improved plan display + debug output; clearer not-running/port errors; hide account switch.\n- Status: poll Google Workspace incidents for Gemini + Antigravity; Status Page opens the Workspace status page.\n- Settings: add Providers tab; move ccusage + status toggles to General; keep display controls in Advanced.\n- Menu/UI: widen the menu for four providers; cards/charts adapt to menu width; tighten provider switcher/toggle spacing; keep menus refreshed while open.\n- Gemini: hide the dashboard action when unsupported.\n- Claude: fix Extra usage spend/limit units (cents); improve CLI probe stability; surface web session info in Debug.\n- OpenAI web: fix dashboard ghost overlay on desktop (WebKit keepalive window).\n- Debug: add a debug-lldb build mode for troubleshooting.\n\n## 0.13.0 — 2025-12-24\n- Claude: add optional web-first usage via Safari/Chrome cookies (no CLI fallback) including “Extra usage” budget bar.\n- Claude: web identity now uses `/api/account` for email + plan (via rate_limit_tier).\n- Settings: standardize “Augment … via web” copy for Codex + Claude web cookie features.\n- Debug: Claude dump now shows web strategy, cookie discovery, HTTP status codes, and parsed summary.\n- Dev: add Claude web probe CLI to enumerate endpoints/fields using browser cookies.\n- Tests: add unit coverage for Claude web API usage, overage, and account parsing.\n- Menu: custom menu items now use the native selection highlight color (plus matching selection text/track colors).\n- Charts: boost hover highlight contrast for credits/usage history bands.\n- Menu: reorder Codex blocks to show credits before cost.\n- Menu: split Claude “Extra usage” (no submenu) from “Cost” (history submenu) and trim redundant extra-usage subtext.\n\n## 0.12.0 — 2025-12-23\n- Widgets: add WidgetKit extension backed by a shared app‑group usage snapshot.\n- New local cost usage tracking (Codex + Claude) via a lightweight scanner — inspired by ccusage (MIT). Computes cost from local JSONL logs without Node CLIs. Thanks @ryoppippi!\n- Cost summary now includes last‑30‑days tokens; weekly pace indicators (with runout copy) hide when usage is fully depleted. Thanks @Remedy92!\n- Claude: PTY probes now stop after idle, auto‑clean on restart, and run under a watchdog to avoid runaway CLI processes.\n- Menu polish: group history under card sections, simplify history labels, and refresh menus live while open.\n- Performance: faster usage log scanning + cost parsing; cache menu icons and speed up OpenAI dashboard parsing.\n- Sparkle: auto-download updates when auto-check is enabled, and only show the restart menu entry once an update is ready.\n- Widgets: experimental WidgetKit extension (may require restarting the widget gallery/Dock to appear).\n- Credits: show credits as a progress bar and add a credits history chart when OpenAI web data is available.\n- Credits: move “Buy Credits…” into its own menu item and improve auto-start checkout flow.\n\n## 0.11.2 — 2025-12-21\n- ccusage-codex cost fetch is faster and more reliable by limiting the session scan window.\n- Fix ccusage cost fetch hanging for large Codex histories by draining subprocess output while commands run.\n- Fix merged-icon loading animation when another provider is fetching (only the selected provider animates).\n- CLI PATH capture now uses an interactive login shell and merges with the app PATH, fixing missing Node/Codex/Claude/Gemini resolution for NVM-style installs.\n\n## 0.11.1 — 2025-12-21\n- Gemini OAuth token refresh now supports Bun/npm installations. Thanks @ben-vargas!\n\n## 0.11.0 — 2025-12-21\n- New optional cost display in the menu (session + last 30 days), powered by ccusage. Thanks @Xuanwo!\n- Fix loading-state card spacing to avoid double separators.\n\n## 0.10.0 — 2025-12-20\n- Gemini provider support (usage, plan detection, login flow). Thanks @381181295!\n- Unified menu bar icon mode with a provider switcher and Merge Icons toggle (default on when multiple providers are enabled). Thanks @ibehnam!\n- Fix regression from 0.9.1 where CLI detection failed for some installs by restoring interactive login-shell PATH loading.\n\n## 0.9.1 — 2025-12-19\n- CLI resolution now uses the login shell PATH directly (no more heuristic path scanning), so Codex/Claude match your shell config reliably.\n\n## 0.9.0 — 2025-12-19\n- New optional OpenAI web access: reuses your signed-in Safari/Chrome session to show **Code review remaining**, **Usage breakdown**, and **Credits usage history** in the menu (no credentials stored).\n- Credits still come from the Codex CLI; OpenAI web access is only used for the dashboard extras above.\n- OpenAI web sessions auto-sync to the Codex CLI email, support multiple accounts, and reset/re-import cookies on account switches to avoid stale cross-account data.\n- Fix Chrome cookie import (macOS 10): signed-in Chrome sessions are detected reliably (thanks @tobihagemann!).\n- Usage breakdown submenu: compact chart with hover details for day/service totals.\n- New “Show usage as used” toggle to invert progress bars (default remains “% left”, now in Advanced).\n- Session (5-hour) reset now shows a relative countdown (“Resets in 3h 31m”) in the menu card for Codex and Claude.\n- Claude: fix reset parsing so “Resets …” can’t be mis-attributed to the wrong window (session vs weekly).\n\n## 0.8.1 — 2025-12-17\n- Claude trust prompts (“Do you trust the files in this folder?”) are now auto-accepted during probes to prevent stuck refreshes. Thanks @tobihagemann!\n\n## 0.8.0 — 2025-12-17\n- CodexBar is now available via Homebrew: `brew install --cask steipete/tap/codexbar` (updates via `brew upgrade --cask steipete/tap/codexbar`).\n- Added session quota notifications for the sliding 5-hour window (Codex + Claude): notifies when it hits 0% and when it’s available again, based only on observed refresh data (including startup when already depleted). Thanks @GKannanDev!\n\n## 0.7.3 — 2025-12-17\n- Claude Enterprise accounts whose Claude Code `/usage` panel only shows “Current session” no longer fail parsing; weekly usage is treated as unavailable (fixes #19).\n\n## 0.7.2 — 2025-12-13\n- Claude “Open Dashboard” now routes subscription accounts (Max/Pro/Ultra/Team) to the usage page instead of the API console billing page. Thanks @auroraflux!\n- Codex/Claude binary resolution now detects mise/rtx installs (shims and newest installed tool version), fixing missing CLI detection for mise users. Thanks @philipp-spiess!\n- Claude usage/status probes now auto-accept the first-run “Ready to code here?” permission prompt (when launched from Finder), preventing timeouts and parse errors. Thanks @alexissan!\n- General preferences now surface full Codex/Claude fetch errors with one-click copy and expandable details, reducing first-run confusion when a CLI is missing.\n- Polished the menu bar “critter” icons: Claude is now a crisper, blockier pixel crab, and Codex has punchier eyes with reduced blurring in SwiftUI/menu rendering.\n\n## 0.7.1 — 2025-12-09\n- Menu bar icons now render on a true 18 pt/2× backing with pixel-aligned bars and overlays for noticeably crisper edges.\n- PTY runner now preserves the caller’s environment (HOME/TERM/bun installs) while enriching PATH, preventing Codex/Claude\n probes from failing when CLIs are installed via bun/nvm or need their auth/config paths.\n- Added regression tests to lock in the enriched environment behavior.\n- Fixed a first-launch crash on macOS 26 caused by the 1×1 keepalive window triggering endless constraint updates; the hidden\n window now uses a safe size and no longer spams SwiftUI state warnings.\n- Menu action rows now ship with SF Symbol icons (refresh, dashboard, status, settings, about, quit, copy error) for clearer at-a-glance affordances.\n- When the Codex CLI is missing, menu and CLI now surface an actionable install hint (`npm i -g @openai/codex` / bun) instead of a generic PATH error.\n- Node manager (nvm/fnm) resolution corrected so codex/claude binaries — and their `node` — are found reliably even when installed via fnm aliases or nvm defaults. Thanks @aliceisjustplaying for surfacing the gaps.\n- Login menu now shows phase-specific subtitles and disables interaction while running: “Requesting login…” while starting the CLI, then “Waiting in browser…” once the auth URL is printed; success still triggers the macOS notification.\n- Login state is tracked per provider so Codex and Claude icons/menus no longer share the same in-flight status when switching accounts.\n- Claude login PTY runner detects the auth URL without clearing buffers, keeps the session alive until confirmation, and exposes a Sendable phase callback used by the menu.\n- Claude CLI detection now includes Claude Code’s self-updating paths (`~/.claude/local/claude`, `~/.claude/bin/claude`) so PTY probes work even when only the bundled installer is used.\n\n## 0.7.0 — 2025-12-07\n- ✨ New rich menu card with inline progress bars and reset times for each provider, giving the menu a beautiful, at-a-glance dashboard feel (credit: Anton Sotkov @antons).\n\n## 0.6.1 — 2025-12-07\n- Claude CLI probes stop passing `--dangerously-skip-permissions`, aligning with the default permission prompt and avoiding hidden first-run failures.\n\n## 0.6.0 — 2025-12-04\n- New bundled CLI (`codexbar`) with single `usage` command, `--format text|json`, `--status`, and fast `-h/-V`.\n- CLI output now shows consistent headers (`Codex 0.x.y (codex-cli)`, `Claude Code (claude)`) and JSON includes `source` + `status`.\n- Advanced prefs install button symlinks `codexbar` into /usr/local/bin and /opt/homebrew/bin; docs refreshed.\n\n## 0.5.7 — 2025-11-26\n- Status Page and Usage Dashboard menu actions now honor the icon you click; Codex menus no longer open the Claude status site.\n\n## 0.5.6 — 2025-11-25\n- New playful “Surprise me” option adds occasional blinks/tilts/wiggles to the menu bar icons (one random effect at a time) plus a Debug “Blink now” trigger.\n- Preferences now include an Advanced tab (refresh cadence, Surprise me toggle, Debug visibility); window height trimmed ~20% for a tighter fit.\n- Motion timing eased and lengthened so blinks/wiggles feel smoother and less twitchy.\n\n## 0.5.5 — 2025-11-25\n- Claude usage scrape now recognizes the new “Current week (Sonnet only)” bar while keeping the legacy Opus label as a fallback.\n- Menu and docs now label the Claude tertiary limit as Sonnet to match the latest CLI wording.\n- PATH seeding now uses a deterministic binary locator plus a one-shot login-shell capture at startup (no globbed nvm paths); the Debug tab shows the resolved Codex binary and effective PATH layers.\n\n## 0.5.4 — 2025-11-24\n- Status blurb under “Status Page” no longer prefixes the text with “Status:”, keeping the incident description concise.\n- PTY runner now registers cleanup before launch so both ends of the TTY and the process group are torn down even when `Process.run()` throws (no leaked fds when spawn fails).\n\n## 0.5.3 — 2025-11-22\n- Added a per-provider “Status Page” menu item beneath Usage that opens the provider’s live status page (OpenAI or Claude).\n- Status API now refreshes alongside usage; incident states show a dot/! overlay on the status icon plus a status blurb under the menu item.\n- General preferences now include a default-on “Check provider status” toggle above refresh cadence.\n\n## 0.5.2 — 2025-11-22\n- Release packaging now includes uploading the dSYM archive alongside the app zip to aid crash symbolication (policy documented in the shared mac release guide).\n- Claude PTY fallback removed: Claude probes now rely solely on `script` stdout parsing, and the generic TTY runner is trimmed to Codex `/status` handling.\n- Fixed a busy-loop on the codex RPC stderr pipe (handler now detaches on EOF), eliminating the long-running high-CPU spin reported in issue #9.\n\n## 0.5.1 — 2025-11-22\n- Debug pane now exposes the Claude parse dump toggle, keeping the captured raw scrape in memory for inspection.\n- Claude About/debug views embed the current git hash so builds can be identified precisely.\n- Minor runtime robustness tweaks in the PTY runner and usage fetcher.\n\n## 0.5.0 — 2025-11-22\n- Codex usage/credits now use the codex app-server RPC by default (with PTY `/status` fallback when RPC is unavailable), reducing flakiness and speeding refreshes.\n- Codex CLI launches seed PATH with Homebrew/bun/npm/nvm/fnm defaults to avoid ENOENT in hardened/release builds; TTY probes reuse the same PATH.\n- Claude CLI probe now runs `/usage` and `/status` in parallel (no simulated typing), captures reset strings, and uses a resilient parser (label-first with ordered fallback) while keeping org/email separate by provider.\n- TTY runner now always tears down the spawned process group (even on early Claude login prompts) to avoid leaking CLI processes.\n- Default refresh cadence is now 5 minutes, and a 15-minute option was added to the settings picker.\n- Claude probes/version detection now start with `--allowed-tools \"\"` (tool access disabled) while keeping interactive PTY mode working.\n- Codex probes and version detection now launch the CLI with `-s read-only -a untrusted` to keep PTY runs sandboxed.\n- Codex warm-up screens (“data not available yet”) are handled gracefully: cached credits stay visible and the menu skips the scary parse error.\n- Codex reset times are shown for both RPC and TTY fallback, and plan labels are capitalized while emails stay verbatim.\n\n## 0.4.3 — 2025-11-21\n- Fix status item creation timing on macOS 15 by deferring NSStatusItem setup to after launch; adds a regression test for the path.\n- Menu bar icon with unknown usage now draws empty tracks (instead of a full bar when decorations are shown) by treating nil values as 0%.\n\n## 0.4.2 — 2025-11-21\n- Sparkle updates re-enabled in release builds (disabled only for the debug bundle ID).\n\n## 0.4.1 — 2025-11-21\n- Both Codex and Claude probes now run off the main thread (background PTY), avoiding menu/UI stalls during `/status` or `/usage` fetches.\n- Codex credits stay available even when `/status` times out: cached values are kept and errors are surfaced separately.\n- Claude/Codex provider autodetect runs on first launch (defaults to Codex if neither is installed) with a debug reset button.\n- Sparkle updates re-enabled in release builds (disabled only for debug bundle ID).\n- Claude probe now issues the `/usage` slash command directly to land on the Usage tab reliably and avoid palette misfires.\n\n## 0.4.0 — 2025-11-21\n- Claude Code support: dedicated Claude menu/icon plus dual-wired menus when both providers are enabled; shows email/org/plan and Sonnet usage with clickable errors.\n- New Preferences window: General/About tabs with provider toggles, refresh cadence, start-at-login, and always-on Quit.\n- Codex credits without web login: we now read `codex /status` in a PTY, auto-skip the update prompt, and parse session/weekly/credits; cached credits stay visible on transient timeouts.\n- Resilience: longer PTY timeouts, cached-credit fallback, one-line menu errors, and clearer parse/update messages.\n\n## 0.3.0 — 2025-11-18\n- Credits support: reads Codex CLI `/status` via PTY (no browser login), shows remaining credits inline, and moves history to a submenu.\n- Sign-in window with cookie reuse and a logout/clear-cookies action; waits out workspace picker and auto-navigates to usage page.\n- Menu: credits line bolded; login prompt hides once credits load; debug toggle always visible (HTML dump).\n- Icon: when weekly is empty, top bar becomes a thick credits bar (capped at 1k); otherwise bars stay 5h/weekly.\n\n## 0.2.2 — 2025-11-17\n- Menu bar icon stays static when no account/usage is present; loading animation only runs while fetching (12 fps) to keep idle CPU low.\n- Usage refresh first tails the newest session log (512 KB window) before scanning everything, reducing IO on large Codex logs.\n- Packaging/signing hardened: strip extended attributes, delete AppleDouble (`._*`) files, and re-sign Sparkle + app bundle to satisfy Gatekeeper.\n\n## 0.2.1 — 2025-11-17\n- Patch bump for refactor/relative-time changes; packaging scripts set to 0.2.1 (5).\n- Streamlined Codex usage parsing: modern rate-limit handling, flexible reset time parsing, and account rate-limit updates (thanks @jazzyalex and https://jazzyalex.github.io/agent-sessions/).\n\n## 0.2.0 — 2025-11-16\n- CADisplayLink-based loading animations (macOS 15 displayLink API) with randomized patterns (Knight Rider, Cylon, outside-in, race, pulse) and debug replay cycling through all.\n- Debug replay toggle (`defaults write com.steipete.codexbar debugMenuEnabled -bool YES`) to view every pattern.\n- Usage Dashboard link in menu; menu layout tweaked.\n- Updated time now shows relative formatting when fresher than 24h; refactored sources into smaller files for maintainability.\n- Version bumped to 0.2.0 (4).\n\n## 0.1.2 — 2025-11-16\n- Animated loading icon (dual bars sweep until usage arrives); always uses rendered template icon.\n- Sparkle embedding/signing fixed with deep+timestamp; notarization pipeline solid.\n- Icon conversion scripted via ictool with docs.\n- Menu: settings submenu, no GitHub item; About link clickable.\n\n## 0.1.1 — 2025-11-16\n- Launch-at-login toggle (SMAppService) and saved preference applied at startup.\n- Sparkle auto-update wiring (SUFeedURL to GitHub, SUPublicEDKey set); Settings submenu with auto-update toggle + Check for Updates.\n- Menu cleanup: settings grouped, GitHub menu removed, About link clickable.\n- Usage parser scans newest session logs until it finds `token_count` events.\n- Icon pipeline fixed: regenerated `.icns` via ictool with proper transparency (docs in docs/icon.md).\n- Added lint/format configs, Swift Testing, strict concurrency, and usage parser tests.\n- Notarized release build \"CodexBar-0.1.0.zip\" remains current artifact; app version 0.1.1.\n\n## 0.1.0 — 2025-11-16\n- Initial CodexBar release: macOS 15+ menu bar app, no Dock icon.\n- Reads latest Codex CLI `token_count` events from session logs (5h + weekly usage, reset times); no extra login or browser scraping.\n- Shows account email/plan decoded locally from `auth.json`.\n- Horizontal dual-bar icon (top = 5h, bottom = weekly); dims on errors.\n- Configurable refresh cadence, manual refresh, and About links.\n- Async off-main log parsing for responsiveness; strict-concurrency build flags enabled.\n- Packaging + signing/notarization scripts (arm64); build scripts convert `.icon` bundle to `.icns`.\n" }, { "path": "FORK_STATUS.md", "content": "# CodexBar Fork - Current Status\n\n**Last Updated:** January 4, 2026\n**Fork Maintainer:** Brandon Charleson\n**Branch:** `feature/augment-integration`\n\n---\n\n## ✅ Completed Work\n\n### Phase 1: Fork Identity & Credits ✓\n\n**Commits:**\n1. `da3d13e` - \"feat: establish fork identity with dual attribution\"\n2. `745293e` - \"docs: add fork roadmap and quick start guide\"\n3. `8a87473` - \"docs: add fork status tracking document\"\n4. `df75ae2` - \"feat: comprehensive multi-upstream fork management system\"\n\n**Changes:**\n- ✅ Updated About section with dual attribution (original + fork)\n- ✅ Updated PreferencesAboutPane with organized sections\n- ✅ Changed app icon click to open fork repository\n- ✅ Updated README with fork notice and enhancements section\n- ✅ Created comprehensive `docs/augment.md` documentation\n- ✅ Created `docs/FORK_ROADMAP.md` with 5-phase plan\n- ✅ Created `docs/FORK_QUICK_START.md` developer guide\n- ✅ Created `FORK_STATUS.md` tracking document\n- ✅ **Implemented complete multi-upstream management system**\n\n**Build Status:** ✅ App builds and runs successfully\n\n### Multi-Upstream Management System ✓\n\n**Automation Scripts:**\n- ✅ `Scripts/check_upstreams.sh` - Monitor both upstreams\n- ✅ `Scripts/review_upstream.sh` - Create review branches\n- ✅ `Scripts/prepare_upstream_pr.sh` - Prepare upstream PRs\n- ✅ `Scripts/analyze_quotio.sh` - Analyze quotio patterns\n\n**GitHub Actions:**\n- ✅ `.github/workflows/upstream-monitor.yml` - Automated monitoring\n\n**Documentation:**\n- ✅ `docs/UPSTREAM_STRATEGY.md` - Complete management guide\n- ✅ `docs/QUOTIO_ANALYSIS.md` - Pattern analysis framework\n- ✅ `docs/FORK_SETUP.md` - One-time setup guide\n\n---\n\n## 🎯 Current State\n\n### What Works\n- ✅ Fork identity clearly established\n- ✅ Dual attribution in place (original + fork)\n- ✅ Comprehensive documentation\n- ✅ Clear development roadmap\n- ✅ App builds without errors\n- ✅ All existing functionality preserved\n- ✅ **Multi-upstream management system operational**\n- ✅ **Automated upstream monitoring configured**\n- ✅ **Quotio analysis framework ready**\n\n### Critical Discovery\n- ⚠️ **Upstream (steipete) has REMOVED Augment provider**\n - 627 lines deleted from `AugmentStatusProbe.swift`\n - 88 lines deleted from `AugmentStatusProbeTests.swift`\n - **This validates our fork strategy!**\n - We preserve Augment support for our users\n - We can selectively sync other improvements\n\n### Known Issues\n- ⚠️ Augment cookie disconnection (Phase 2 will address)\n- ⚠️ Debug print statements in AugmentStatusProbe.swift (needs proper logging)\n\n### Uncommitted Changes\n- `Sources/CodexBarCore/Providers/Augment/AugmentStatusProbe.swift` has debug print statements\n - These should be replaced with proper `CodexBarLog` logging in Phase 2\n - Currently unstaged to keep commits clean\n\n---\n\n## 📋 Next Steps\n\n### URGENT: Upstream Sync Decision\n**Before proceeding with Phase 2, decide on upstream sync strategy:**\n\n1. **Review upstream changes:**\n ```bash\n ./Scripts/check_upstreams.sh upstream\n ./Scripts/review_upstream.sh upstream\n ```\n\n2. **Decide what to sync:**\n - ✅ Vertex AI improvements (5 commits)\n - ✅ SwiftFormat/SwiftLint fixes\n - ❌ Augment provider removal (SKIP!)\n\n3. **Cherry-pick valuable commits:**\n ```bash\n git checkout -b upstream-sync/vertex-improvements\n git cherry-pick 001019c # style fixes\n git cherry-pick e4f1e4c # vertex token cost\n git cherry-pick 202efde # vertex fix\n git cherry-pick 0c2f888 # vertex docs\n git cherry-pick 3c4ca30 # vertex tracking\n # Skip Augment removal commits!\n ```\n\n### Immediate (Phase 2)\n1. **Replace debug prints with proper logging**\n - Use `CodexBarLog.logger(\"augment\")` pattern\n - Add structured metadata\n - Follow Claude/Cursor provider patterns\n\n2. **Enhanced cookie diagnostics**\n - Log cookie expiration times\n - Track refresh attempts\n - Add domain filtering diagnostics\n\n3. **Session keepalive monitoring**\n - Add keepalive status to debug pane\n - Log refresh attempts\n - Add manual \"Force Refresh\" button\n\n### Short Term (Phases 3-4)\n- **Analyze Quotio features** using `./Scripts/analyze_quotio.sh`\n- **Regular upstream monitoring** (automated via GitHub Actions)\n- **Weekly sync routine** (Monday: upstream, Thursday: quotio)\n\n### Medium Term (Phase 5)\n- Implement multi-account management (inspired by quotio)\n- Start with Augment provider\n- Extend to other providers\n\n---\n\n## 📁 Key Files Modified\n\n### Source Code\n- `Sources/CodexBar/About.swift` - Dual attribution\n- `Sources/CodexBar/PreferencesAboutPane.swift` - Organized sections\n- `Sources/CodexBarCore/Providers/Augment/AugmentStatusProbe.swift` - Debug prints (unstaged)\n\n### Documentation\n- `README.md` - Fork notice and enhancements\n- `docs/augment.md` - Augment provider guide (NEW)\n- `docs/FORK_ROADMAP.md` - Development roadmap (NEW)\n- `docs/FORK_QUICK_START.md` - Quick reference (NEW)\n\n---\n\n## 🔄 Git Status\n\n```bash\n# Current branch\nfeature/augment-integration\n\n# Commits ahead of main\n4 commits:\n- da3d13e: Fork identity with dual attribution\n- 745293e: Roadmap and quick start guide\n- 8a87473: Fork status tracking\n- df75ae2: Multi-upstream management system\n\n# Uncommitted changes\nM Sources/CodexBarCore/Providers/Augment/AugmentStatusProbe.swift (debug prints)\n\n# Git remotes configured\norigin git@github.com:topoffunnel/CodexBar.git\nupstream https://github.com/steipete/CodexBar.git (needs to be added)\nquotio https://github.com/nguyenphutrong/quotio.git (needs to be added)\n```\n\n---\n\n## 🚀 How to Continue\n\n### RECOMMENDED: Setup Multi-Upstream System First\n\n```bash\n# 1. Configure git remotes\ngit remote add upstream https://github.com/steipete/CodexBar.git\ngit remote add quotio https://github.com/nguyenphutrong/quotio.git\ngit fetch --all\n\n# 2. Test automation scripts\n./Scripts/check_upstreams.sh\n\n# 3. Review upstream changes (IMPORTANT!)\n./Scripts/review_upstream.sh upstream\n\n# 4. Decide what to sync\n# See \"URGENT: Upstream Sync Decision\" section above\n\n# 5. Analyze quotio\n./Scripts/analyze_quotio.sh\n```\n\n### Option 1: Sync Upstream First, Then Phase 2\n```bash\n# Discard debug prints (will redo in Phase 2)\ngit checkout Sources/CodexBarCore/Providers/Augment/AugmentStatusProbe.swift\n\n# Sync valuable upstream changes\ngit checkout -b upstream-sync/vertex-improvements\n# Cherry-pick commits (see URGENT section)\n\n# Merge to main\ngit checkout main\ngit merge feature/augment-integration\ngit merge upstream-sync/vertex-improvements\n\n# Then start Phase 2\ngit checkout -b feature/augment-diagnostics\n```\n\n### Option 2: Phase 2 First, Sync Later\n```bash\n# Keep debug prints and enhance them\ngit add Sources/CodexBarCore/Providers/Augment/AugmentStatusProbe.swift\n\n# Continue on current branch\n# Replace print() with CodexBarLog.logger(\"augment\")\n# Complete Phase 2\n# Then sync upstream\n```\n\n### Option 3: Merge Current Work, Setup System\n```bash\n# Discard debug prints\ngit checkout Sources/CodexBarCore/Providers/Augment/AugmentStatusProbe.swift\n\n# Merge to main\ngit checkout main\ngit merge feature/augment-integration\n\n# Setup remotes\ngit remote add upstream https://github.com/steipete/CodexBar.git\ngit remote add quotio https://github.com/nguyenphutrong/quotio.git\n\n# Start using the system\n./Scripts/check_upstreams.sh\n```\n\n---\n\n## 📊 Progress Tracking\n\n### Phase 1: Fork Identity ✅ COMPLETE\n- [x] Dual attribution in About\n- [x] Fork notice in README\n- [x] Augment documentation\n- [x] Development roadmap\n- [x] Quick start guide\n\n### Phase 2: Enhanced Diagnostics 🔄 READY TO START\n- [ ] Replace print() with CodexBarLog\n- [ ] Enhanced cookie diagnostics\n- [ ] Session keepalive monitoring\n- [ ] Debug pane improvements\n\n### Phase 3: Quotio Analysis 📋 PLANNED\n- [ ] Feature comparison matrix\n- [ ] Implementation recommendations\n- [ ] Priority ranking\n\n### Phase 4: Upstream Sync 📋 PLANNED\n- [ ] Sync script\n- [ ] Conflict resolution guide\n- [ ] Automated checks\n\n### Phase 5: Multi-Account 📋 PLANNED\n- [ ] Account management UI\n- [ ] Account storage\n- [ ] Account switching\n- [ ] UI enhancements\n\n---\n\n## 🎯 Success Criteria\n\n### Phase 1 (Current) ✅\n- [x] Fork identity clearly established\n- [x] Original author properly credited\n- [x] Comprehensive documentation\n- [x] App builds and runs\n- [x] No regressions\n\n### Phase 2 (Next)\n- [ ] Zero cookie disconnection issues\n- [ ] Proper structured logging\n- [ ] Enhanced debug diagnostics\n- [ ] Manual refresh capability\n- [ ] All tests passing\n\n---\n\n## 📞 Questions & Decisions Needed\n\n### Before Starting Phase 2\n1. **Logging approach:** Keep debug prints and enhance, or start fresh?\n2. **Branch strategy:** Continue on `feature/augment-integration` or create new branch?\n3. **Merge timing:** Merge Phase 1 to main first, or continue with all phases?\n\n### For Phase 3\n1. **Quotio access:** Do you have access to Quotio source code?\n2. **Feature priority:** Which Quotio features are most important?\n3. **Timeline:** How much time to allocate for analysis?\n\n### For Phase 5\n1. **Account limit:** How many accounts per provider?\n2. **UI design:** Menu bar dropdown or separate window?\n3. **Storage:** Keychain per account or shared?\n\n---\n\n## 🔗 Quick Links\n\n- **Roadmap:** `docs/FORK_ROADMAP.md`\n- **Quick Start:** `docs/FORK_QUICK_START.md`\n- **Augment Docs:** `docs/augment.md`\n- **Original Repo:** https://github.com/steipete/CodexBar\n- **Fork Repo:** https://github.com/topoffunnel/CodexBar\n\n---\n\n## 💡 Recommendations\n\n1. **Merge Phase 1 to main** - Establish fork identity as baseline\n2. **Create Phase 2 branch** - `feature/augment-diagnostics`\n3. **Start with logging** - Replace prints with proper CodexBarLog\n4. **Test thoroughly** - Ensure no regressions\n5. **Document as you go** - Update docs with findings\n\n---\n\n**Ready to proceed with Phase 2?** See `docs/FORK_ROADMAP.md` for detailed tasks.\n\n" }, { "path": "IMPLEMENTATION_SUMMARY.md", "content": "# CodexBar Fork - Implementation Summary\n\n**Date:** January 4, 2026 \n**Implementer:** Augment AI Assistant \n**For:** Brandon Charleson (topoffunnel.com)\n\n---\n\n## 🎉 What Was Accomplished\n\n### Phase 1: Fork Identity & Credits ✅ COMPLETE\n\n**Objective:** Establish clear fork identity while properly crediting original author\n\n**Deliverables:**\n1. **Dual Attribution System**\n - Updated `About.swift` with original author + fork maintainer\n - Updated `PreferencesAboutPane.swift` with organized sections\n - App icon click now opens fork repository\n - Clear separation of original vs fork contributions\n\n2. **Documentation Suite**\n - `docs/augment.md` - Comprehensive Augment provider guide (150+ lines)\n - `docs/FORK_ROADMAP.md` - 5-phase development plan\n - `docs/FORK_QUICK_START.md` - Developer quick reference\n - `FORK_STATUS.md` - Living status tracker\n\n3. **README Updates**\n - Fork notice at top with link to original\n - \"Fork Enhancements\" section documenting improvements\n - Updated credits with dual attribution\n - Clear differentiation from original\n\n**Result:** Fork has professional identity, ready for distribution via topoffunnel.com\n\n---\n\n### Multi-Upstream Management System ✅ COMPLETE\n\n**Objective:** Monitor and selectively incorporate changes from two upstream repositories\n\n**Deliverables:**\n\n#### 1. Automation Scripts (4 scripts, all executable)\n\n**`Scripts/check_upstreams.sh`**\n- Monitors both upstream and quotio for new commits\n- Shows commit summaries and file changes\n- Color-coded output for easy scanning\n- Usage: `./Scripts/check_upstreams.sh [upstream|quotio|all]`\n\n**`Scripts/review_upstream.sh`**\n- Creates review branch for upstream changes\n- Shows detailed commit log and diffs\n- Generates review log file\n- Usage: `./Scripts/review_upstream.sh [upstream|quotio]`\n\n**`Scripts/prepare_upstream_pr.sh`**\n- Creates clean branch from upstream/main for PR submission\n- Provides guidelines for what to include/exclude\n- Prevents fork branding from going upstream\n- Usage: `./Scripts/prepare_upstream_pr.sh `\n\n**`Scripts/analyze_quotio.sh`**\n- Analyzes quotio repository structure and recent changes\n- Generates analysis report with action items\n- Helps identify patterns to adapt (not copy)\n- Usage: `./Scripts/analyze_quotio.sh [feature-area]`\n\n#### 2. GitHub Actions Workflow\n\n**`.github/workflows/upstream-monitor.yml`**\n- Runs Monday and Thursday at 9 AM UTC\n- Checks both upstreams for new commits\n- Creates/updates GitHub issue with summaries\n- Provides links to review changes\n- Can be triggered manually\n\n#### 3. Comprehensive Documentation (3 guides)\n\n**`docs/UPSTREAM_STRATEGY.md`** (630+ lines)\n- Complete multi-upstream management guide\n- Git repository structure and remote configuration\n- Workflows for monitoring, reviewing, incorporating changes\n- Decision matrix: what to contribute upstream vs keep in fork\n- Commit message strategies and attribution\n- Practical examples and troubleshooting\n- Best practices and success metrics\n\n**`docs/QUOTIO_ANALYSIS.md`** (150+ lines)\n- Framework for learning from quotio patterns\n- Ethical guidelines (adapt patterns, don't copy code)\n- Analysis process and templates\n- Feature comparison matrix\n- Implementation planning\n- Legal and attribution considerations\n\n**`docs/FORK_SETUP.md`** (150+ lines)\n- One-time setup guide for git remotes\n- Script testing and verification\n- Critical discovery documentation\n- Selective sync strategy\n- Regular workflow recommendations\n\n---\n\n## 🚨 Critical Discovery\n\n**Upstream (steipete) has REMOVED the Augment provider!**\n\n**Evidence:**\n```\nFiles changed in upstream:\n .../Providers/Augment/AugmentStatusProbe.swift | 627 deletions\n Tests/CodexBarTests/AugmentStatusProbeTests.swift | 88 deletions\n```\n\n**Impact:**\n- ✅ **Validates fork strategy** - We preserve features important to our users\n- ✅ **Justifies independent development** - Can't rely on upstream for Augment\n- ✅ **Enables selective sync** - Cherry-pick valuable changes, skip Augment removal\n- ✅ **Protects user experience** - Fork users keep Augment functionality\n\n**Action Required:**\nWhen syncing with upstream, must cherry-pick commits selectively to avoid losing Augment support.\n\n---\n\n## 📊 Commits Summary\n\n**Total Commits:** 5\n\n1. `da3d13e` - Fork identity with dual attribution\n2. `745293e` - Roadmap and quick start guide\n3. `8a87473` - Fork status tracking document\n4. `df75ae2` - Multi-upstream management system\n5. `158d00c` - Updated fork status\n\n**Lines Added:** ~2,500+ lines of documentation and automation\n**Files Created:** 11 new files\n**Scripts Created:** 4 executable automation scripts\n**Workflows Created:** 1 GitHub Actions workflow\n\n---\n\n## 🎯 Strategic Benefits\n\n### For Fork Development\n1. **Independence** - Can develop features without upstream dependency\n2. **Selective Sync** - Cherry-pick valuable improvements, skip unwanted changes\n3. **Attribution Protection** - Fork-specific commits stay separate\n4. **User Focus** - Preserve features important to your users (Augment)\n\n### For Upstream Relationship\n1. **Contribution Ready** - Clean PR branches for upstream submissions\n2. **Good Citizenship** - Can contribute bug fixes and improvements\n3. **Proper Credit** - Attribution system respects original author\n4. **Flexibility** - Option to contribute or keep changes in fork\n\n### For Learning from Quotio\n1. **Ethical Framework** - Clear guidelines for pattern analysis\n2. **Legal Protection** - Adapt patterns, don't copy code\n3. **Innovation** - Learn from their solutions, implement independently\n4. **Attribution** - Credit inspiration appropriately\n\n---\n\n## 📋 Current State\n\n### What's Ready\n- ✅ Fork identity established\n- ✅ Comprehensive documentation\n- ✅ Automation scripts tested and working\n- ✅ GitHub Actions workflow configured\n- ✅ Git remotes documented (need to be added)\n- ✅ Selective sync strategy defined\n- ✅ App builds and runs successfully\n\n### What's Pending\n- ⏳ Git remotes need to be added (one-time setup)\n- ⏳ Upstream sync decision needed (5 new commits available)\n- ⏳ Quotio analysis to be performed\n- ⏳ Phase 2 (Enhanced Augment diagnostics)\n\n### Known Issues\n- ⚠️ Augment cookie disconnection (Phase 2 will address)\n- ⚠️ Debug print statements in AugmentStatusProbe.swift (unstaged)\n\n---\n\n## 🚀 Next Steps for You\n\n### Immediate (Before Phase 2)\n\n**1. Setup Git Remotes**\n```bash\ngit remote add upstream https://github.com/steipete/CodexBar.git\ngit remote add quotio https://github.com/nguyenphutrong/quotio.git\ngit fetch --all\n```\n\n**2. Test Automation**\n```bash\n./Scripts/check_upstreams.sh\n./Scripts/review_upstream.sh upstream\n./Scripts/analyze_quotio.sh\n```\n\n**3. Decide on Upstream Sync**\n- Review 5 new upstream commits\n- Cherry-pick valuable changes (Vertex AI improvements)\n- Skip Augment removal commits\n- See `FORK_STATUS.md` for detailed instructions\n\n### Short Term (This Week)\n\n**4. Merge to Main**\n```bash\ngit checkout main\ngit merge feature/augment-integration\n```\n\n**5. Enable GitHub Actions**\n- Push to your fork\n- Enable Actions in repository settings\n- Verify workflow runs\n\n**6. Start Regular Monitoring**\n- Monday: Check upstream (`./Scripts/check_upstreams.sh upstream`)\n- Thursday: Analyze quotio (`./Scripts/analyze_quotio.sh`)\n\n### Medium Term (Next 2 Weeks)\n\n**7. Complete Phase 2**\n- Enhanced Augment diagnostics\n- Proper logging with CodexBarLog\n- Session keepalive monitoring\n\n**8. Quotio Analysis**\n- Document multi-account patterns\n- Plan implementation\n- Prioritize features\n\n---\n\n## 📖 Documentation Index\n\n### Core Documents\n- `README.md` - Main documentation with fork notice\n- `FORK_STATUS.md` - Current status and next steps\n- `IMPLEMENTATION_SUMMARY.md` - This document\n\n### Setup & Strategy\n- `docs/FORK_SETUP.md` - One-time setup guide\n- `docs/FORK_QUICK_START.md` - Developer quick reference\n- `docs/UPSTREAM_STRATEGY.md` - Multi-upstream management\n- `docs/FORK_ROADMAP.md` - 5-phase development plan\n\n### Provider & Analysis\n- `docs/augment.md` - Augment provider guide\n- `docs/QUOTIO_ANALYSIS.md` - Quotio pattern analysis framework\n\n### Scripts\n- `Scripts/check_upstreams.sh` - Monitor upstreams\n- `Scripts/review_upstream.sh` - Review changes\n- `Scripts/prepare_upstream_pr.sh` - Prepare PRs\n- `Scripts/analyze_quotio.sh` - Analyze quotio\n\n---\n\n## 💡 Key Insights\n\n1. **Fork Validation** - Upstream removing Augment proves fork was necessary\n2. **Best of Both Worlds** - Can learn from two sources while maintaining independence\n3. **Selective Sync** - Cherry-picking gives control over what changes to adopt\n4. **Attribution Matters** - Separate commits protect your contributions\n5. **Automation Wins** - Scripts and workflows reduce manual effort\n\n---\n\n## ✅ Success Criteria Met\n\n- [x] Fork identity clearly established\n- [x] Original author properly credited\n- [x] Comprehensive documentation\n- [x] Multi-upstream monitoring system\n- [x] Automation scripts working\n- [x] GitHub Actions configured\n- [x] Selective sync strategy defined\n- [x] App builds and runs\n- [x] No regressions\n\n---\n\n**Status:** Phase 1 COMPLETE + Multi-Upstream System OPERATIONAL \n**Ready for:** Upstream sync decision + Phase 2 development \n**Recommendation:** Setup remotes, sync upstream, then proceed to Phase 2\n\n" }, { "path": "Icon.icon/icon.json", "content": "{\n \"fill\" : {\n \"automatic-gradient\" : \"extended-srgb:0.00000,0.53333,1.00000,1.00000\"\n },\n \"groups\" : [\n {\n \"layers\" : [\n {\n \"image-name\" : \"codexbar.png\",\n \"name\" : \"codexbar\",\n \"position\" : {\n \"scale\" : 1.4,\n \"translation-in-points\" : [\n 0,\n 0\n ]\n }\n }\n ],\n \"shadow\" : {\n \"kind\" : \"neutral\",\n \"opacity\" : 0.5\n },\n \"translucency\" : {\n \"enabled\" : true,\n \"value\" : 0.5\n }\n }\n ],\n \"supported-platforms\" : {\n \"circles\" : [\n \"watchOS\"\n ],\n \"squares\" : \"shared\"\n }\n}" }, { "path": "LICENSE", "content": "MIT License\n\nCopyright (c) 2026 Peter Steinberger\n\nPermission is hereby granted, free of charge, to any person obtaining a copy\nof this software and associated documentation files (the \"Software\"), to deal\nin the Software without restriction, including without limitation the rights\nto use, copy, modify, merge, publish, distribute, sublicense, and/or sell\ncopies of the Software, and to permit persons to whom the Software is\nfurnished to do so, subject to the following conditions:\n\nThe above copyright notice and this permission notice shall be included in all\ncopies or substantial portions of the Software.\n\nTHE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\nIMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\nFITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\nAUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\nLIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\nOUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\nSOFTWARE.\n" }, { "path": "Package.resolved", "content": "{\n \"originHash\" : \"74bd6f3ab6e0b0cb0c2cddb00f2167c2ab0a1c00cd54ffc1a2899c7ef8c56367\",\n \"pins\" : [\n {\n \"identity\" : \"commander\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/steipete/Commander\",\n \"state\" : {\n \"revision\" : \"9e349575c8e3c6745e81fe19e5bb5efa01b078ce\",\n \"version\" : \"0.2.1\"\n }\n },\n {\n \"identity\" : \"keyboardshortcuts\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/sindresorhus/KeyboardShortcuts\",\n \"state\" : {\n \"revision\" : \"1aef85578fdd4f9eaeeb8d53b7b4fc31bf08fe27\",\n \"version\" : \"2.4.0\"\n }\n },\n {\n \"identity\" : \"sparkle\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/sparkle-project/Sparkle\",\n \"state\" : {\n \"revision\" : \"5581748cef2bae787496fe6d61139aebe0a451f6\",\n \"version\" : \"2.8.1\"\n }\n },\n {\n \"identity\" : \"sweetcookiekit\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/steipete/SweetCookieKit\",\n \"state\" : {\n \"revision\" : \"4d5b71ffbb296937dc5ee8472f64721bca771cf0\",\n \"version\" : \"0.4.0\"\n }\n },\n {\n \"identity\" : \"swift-log\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/apple/swift-log\",\n \"state\" : {\n \"revision\" : \"2778fd4e5a12a8aaa30a3ee8285f4ce54c5f3181\",\n \"version\" : \"1.9.1\"\n }\n },\n {\n \"identity\" : \"swift-syntax\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/apple/swift-syntax\",\n \"state\" : {\n \"revision\" : \"0687f71944021d616d34d922343dcef086855920\",\n \"version\" : \"600.0.1\"\n }\n }\n ],\n \"version\" : 3\n}\n" }, { "path": "Package.swift", "content": "// swift-tools-version: 6.2\nimport CompilerPluginSupport\nimport Foundation\nimport PackageDescription\n\nlet sweetCookieKitPath = \"../SweetCookieKit\"\nlet useLocalSweetCookieKit =\n ProcessInfo.processInfo.environment[\"CODEXBAR_USE_LOCAL_SWEETCOOKIEKIT\"] == \"1\"\nlet sweetCookieKitDependency: Package.Dependency =\n useLocalSweetCookieKit && FileManager.default.fileExists(atPath: sweetCookieKitPath)\n ? .package(path: sweetCookieKitPath)\n : .package(url: \"https://github.com/steipete/SweetCookieKit\", from: \"0.4.0\")\n\nlet package = Package(\n name: \"CodexBar\",\n platforms: [\n .macOS(.v14),\n ],\n dependencies: [\n .package(url: \"https://github.com/sparkle-project/Sparkle\", from: \"2.8.1\"),\n .package(url: \"https://github.com/steipete/Commander\", from: \"0.2.1\"),\n .package(url: \"https://github.com/apple/swift-log\", from: \"1.9.1\"),\n .package(url: \"https://github.com/apple/swift-syntax\", from: \"600.0.1\"),\n .package(url: \"https://github.com/sindresorhus/KeyboardShortcuts\", from: \"2.4.0\"),\n sweetCookieKitDependency,\n ],\n targets: {\n var targets: [Target] = [\n .target(\n name: \"CodexBarCore\",\n dependencies: [\n \"CodexBarMacroSupport\",\n .product(name: \"Logging\", package: \"swift-log\"),\n .product(name: \"SweetCookieKit\", package: \"SweetCookieKit\"),\n ],\n swiftSettings: [\n .enableUpcomingFeature(\"StrictConcurrency\"),\n ]),\n .macro(\n name: \"CodexBarMacros\",\n dependencies: [\n .product(name: \"SwiftCompilerPlugin\", package: \"swift-syntax\"),\n .product(name: \"SwiftSyntaxBuilder\", package: \"swift-syntax\"),\n .product(name: \"SwiftSyntaxMacros\", package: \"swift-syntax\"),\n ]),\n .target(\n name: \"CodexBarMacroSupport\",\n dependencies: [\n \"CodexBarMacros\",\n ]),\n .executableTarget(\n name: \"CodexBarCLI\",\n dependencies: [\n \"CodexBarCore\",\n .product(name: \"Commander\", package: \"Commander\"),\n ],\n path: \"Sources/CodexBarCLI\",\n swiftSettings: [\n .enableUpcomingFeature(\"StrictConcurrency\"),\n ]),\n .testTarget(\n name: \"CodexBarLinuxTests\",\n dependencies: [\"CodexBarCore\", \"CodexBarCLI\"],\n path: \"TestsLinux\",\n swiftSettings: [\n .enableUpcomingFeature(\"StrictConcurrency\"),\n .enableExperimentalFeature(\"SwiftTesting\"),\n ]),\n ]\n\n #if os(macOS)\n targets.append(contentsOf: [\n .executableTarget(\n name: \"CodexBarClaudeWatchdog\",\n dependencies: [],\n path: \"Sources/CodexBarClaudeWatchdog\",\n swiftSettings: [\n .enableUpcomingFeature(\"StrictConcurrency\"),\n ]),\n .executableTarget(\n name: \"CodexBar\",\n dependencies: [\n .product(name: \"Sparkle\", package: \"Sparkle\"),\n .product(name: \"KeyboardShortcuts\", package: \"KeyboardShortcuts\"),\n \"CodexBarMacroSupport\",\n \"CodexBarCore\",\n ],\n path: \"Sources/CodexBar\",\n resources: [\n .process(\"Resources\"),\n ],\n swiftSettings: [\n // Opt into Swift 6 strict concurrency (approachable migration path).\n .enableUpcomingFeature(\"StrictConcurrency\"),\n .define(\"ENABLE_SPARKLE\"),\n ]),\n .executableTarget(\n name: \"CodexBarWidget\",\n dependencies: [\"CodexBarCore\"],\n path: \"Sources/CodexBarWidget\",\n swiftSettings: [\n .enableUpcomingFeature(\"StrictConcurrency\"),\n ]),\n .executableTarget(\n name: \"CodexBarClaudeWebProbe\",\n dependencies: [\"CodexBarCore\"],\n path: \"Sources/CodexBarClaudeWebProbe\",\n swiftSettings: [\n .enableUpcomingFeature(\"StrictConcurrency\"),\n ]),\n ])\n\n targets.append(.testTarget(\n name: \"CodexBarTests\",\n dependencies: [\"CodexBar\", \"CodexBarCore\", \"CodexBarCLI\", \"CodexBarWidget\"],\n path: \"Tests\",\n swiftSettings: [\n .enableUpcomingFeature(\"StrictConcurrency\"),\n .enableExperimentalFeature(\"SwiftTesting\"),\n ]))\n #endif\n\n return targets\n }())\n" }, { "path": "README.md", "content": "# CodexBar 🎚️ - May your tokens never run out.\n\nTiny macOS 14+ menu bar app that keeps your Codex, Claude, Cursor, Gemini, Antigravity, Droid (Factory), Copilot, z.ai, Kiro, Vertex AI, Augment, Amp, JetBrains AI, and OpenRouter limits visible (session + weekly where available) and shows when each window resets. One status item per provider (or Merge Icons mode with a provider switcher and optional Overview tab); enable what you use from Settings. No Dock icon, minimal UI, dynamic bar icons in the menu bar.\n\n\"CodexBar\n\n## Install\n\n### Requirements\n- macOS 14+ (Sonoma)\n\n### GitHub Releases\nDownload: \n\n### Homebrew\n```bash\nbrew install --cask steipete/tap/codexbar\n```\n\n### Linux (CLI only)\n```bash\nbrew install steipete/tap/codexbar\n```\nOr download `CodexBarCLI-v-linux-.tar.gz` from GitHub Releases.\nLinux support via Omarchy: community Waybar module and TUI, driven by the `codexbar` executable.\n\n### First run\n- Open Settings → Providers and enable what you use.\n- Install/sign in to the provider sources you rely on (e.g. `codex`, `claude`, `gemini`, browser cookies, or OAuth; Antigravity requires the Antigravity app running).\n- Optional: Settings → Providers → Codex → OpenAI cookies (Automatic or Manual) to add dashboard extras.\n\n## Providers\n\n- [Codex](docs/codex.md) — Local Codex CLI RPC (+ PTY fallback) and optional OpenAI web dashboard extras.\n- [Claude](docs/claude.md) — OAuth API or browser cookies (+ CLI PTY fallback); session + weekly usage.\n- [Cursor](docs/cursor.md) — Browser session cookies for plan + usage + billing resets.\n- [Gemini](docs/gemini.md) — OAuth-backed quota API using Gemini CLI credentials (no browser cookies).\n- [Antigravity](docs/antigravity.md) — Local language server probe (experimental); no external auth.\n- [Droid](docs/factory.md) — Browser cookies + WorkOS token flows for Factory usage + billing.\n- [Copilot](docs/copilot.md) — GitHub device flow + Copilot internal usage API.\n- [z.ai](docs/zai.md) — API token (Keychain) for quota + MCP windows.\n- [Kimi](docs/kimi.md) — Auth token (JWT from `kimi-auth` cookie) for weekly quota + 5‑hour rate limit.\n- [Kimi K2](docs/kimi-k2.md) — API key for credit-based usage totals.\n- [Kiro](docs/kiro.md) — CLI-based usage via `kiro-cli /usage` command; monthly credits + bonus credits.\n- [Vertex AI](docs/vertexai.md) — Google Cloud gcloud OAuth with token cost tracking from local Claude logs.\n- [Augment](docs/augment.md) — Browser cookie-based authentication with automatic session keepalive; credits tracking and usage monitoring.\n- [Amp](docs/amp.md) — Browser cookie-based authentication with Amp Free usage tracking.\n- [JetBrains AI](docs/jetbrains.md) — Local XML-based quota from JetBrains IDE configuration; monthly credits tracking.\n- [OpenRouter](docs/openrouter.md) — API token for credit-based usage tracking across multiple AI providers.\n- Open to new providers: [provider authoring guide](docs/provider.md).\n\n## Icon & Screenshot\nThe menu bar icon is a tiny two-bar meter:\n- Top bar: 5‑hour/session window. If weekly is missing/exhausted and credits are available, it becomes a thicker credits bar.\n- Bottom bar: weekly window (hairline).\n- Errors/stale data dim the icon; status overlays indicate incidents.\n\n## Features\n- Multi-provider menu bar with per-provider toggles (Settings → Providers).\n- Session + weekly meters with reset countdowns.\n- Optional Codex web dashboard enrichments (code review remaining, usage breakdown, credits history).\n- Local cost-usage scan for Codex + Claude (last 30 days).\n- Provider status polling with incident badges in the menu and icon overlay.\n- Merge Icons mode to combine providers into one status item + switcher, with an optional Overview tab for up to three providers.\n- Refresh cadence presets (manual, 1m, 2m, 5m, 15m).\n- Bundled CLI (`codexbar`) for scripts and CI (including `codexbar cost --provider codex|claude` for local cost usage); Linux CLI builds available.\n- WidgetKit widget mirrors the menu card snapshot.\n- Privacy-first: on-device parsing by default; browser cookies are opt-in and reused (no passwords stored).\n\n## Privacy note\nWondering if CodexBar scans your disk? It doesn’t crawl your filesystem; it reads a small set of known locations (browser cookies/local storage, local JSONL logs) when the related features are enabled. See the discussion and audit notes in [issue #12](https://github.com/steipete/CodexBar/issues/12).\n\n## macOS permissions (why they’re needed)\n- **Full Disk Access (optional)**: only required to read Safari cookies/local storage for web-based providers (Codex web, Claude web, Cursor, Droid/Factory). If you don’t grant it, use Chrome/Firefox cookies or CLI-only sources instead.\n- **Keychain access (prompted by macOS)**:\n - Chrome cookie import needs the “Chrome Safe Storage” key to decrypt cookies.\n - Claude OAuth credentials (written by the Claude CLI) are read from Keychain when present.\n - z.ai API token is stored in Keychain from Preferences → Providers; Copilot stores its API token in Keychain during device flow.\n - **How do I prevent those keychain alerts?**\n - Open **Keychain Access.app** → login keychain → search the item (e.g., “Claude Code-credentials”).\n - Open the item → **Access Control** → add `CodexBar.app` under “Always allow access by these applications”.\n - Prefer adding just CodexBar (avoid “Allow all applications” unless you want it wide open).\n - Relaunch CodexBar after saving.\n - Reference screenshot: ![Keychain access control](docs/keychain-allow.png)\n - **How to do the same for the browser?**\n - Find the browser’s “Safe Storage” key (e.g., “Chrome Safe Storage”, “Brave Safe Storage”, “Firefox”, “Microsoft Edge Safe Storage”).\n - Open the item → **Access Control** → add `CodexBar.app` under “Always allow access by these applications”.\n - This removes the prompt when CodexBar decrypts cookies for that browser.\n- **Files & Folders prompts (folder/volume access)**: CodexBar launches provider CLIs (codex/claude/gemini/antigravity). If those CLIs read a project directory or external drive, macOS may ask CodexBar for that folder/volume (e.g., Desktop or an external volume). This is driven by the CLI’s working directory, not background disk scanning.\n- **What we do not request**: no Screen Recording, Accessibility, or Automation permissions; no passwords are stored (browser cookies are reused when you opt in).\n\n## Docs\n- Providers overview: [docs/providers.md](docs/providers.md)\n- Provider authoring: [docs/provider.md](docs/provider.md)\n- UI & icon notes: [docs/ui.md](docs/ui.md)\n- CLI reference: [docs/cli.md](docs/cli.md)\n- Architecture: [docs/architecture.md](docs/architecture.md)\n- Refresh loop: [docs/refresh-loop.md](docs/refresh-loop.md)\n- Status polling: [docs/status.md](docs/status.md)\n- Sparkle updates: [docs/sparkle.md](docs/sparkle.md)\n- Release checklist: [docs/RELEASING.md](docs/RELEASING.md)\n\n## Getting started (dev)\n- Clone the repo and open it in Xcode or run the scripts directly.\n- Launch once, then toggle providers in Settings → Providers.\n- Install/sign in to provider sources you rely on (CLIs, browser cookies, or OAuth).\n- Optional: set OpenAI cookies (Automatic or Manual) for Codex dashboard extras.\n\n## Build from source\n```bash\nswift build -c release # or debug for development\n./Scripts/package_app.sh # builds CodexBar.app in-place\nCODEXBAR_SIGNING=adhoc ./Scripts/package_app.sh # ad-hoc signing (no Apple Developer account)\nopen CodexBar.app\n```\n\nDev loop:\n```bash\n./Scripts/compile_and_run.sh\n```\n\n## Related\n- ✂️ [Trimmy](https://github.com/steipete/Trimmy) — “Paste once, run once.” Flatten multi-line shell snippets so they paste and run.\n- 🧳 [MCPorter](https://mcporter.dev) — TypeScript toolkit + CLI for Model Context Protocol servers.\n- 🧿 [oracle](https://askoracle.dev) — Ask the oracle when you're stuck. Invoke GPT-5 Pro with a custom context and files.\n\n## Looking for a Windows version?\n- [Win-CodexBar](https://github.com/Finesssee/Win-CodexBar)\n\n## Credits\nInspired by [ccusage](https://github.com/ryoppippi/ccusage) (MIT), specifically the cost usage tracking.\n\n## License\nMIT • Peter Steinberger ([steipete](https://twitter.com/steipete))\n" }, { "path": "Scripts/analyze_quotio.sh", "content": "#!/bin/bash\n# Analyze quotio repository for interesting patterns and features\n# Usage: ./Scripts/analyze_quotio.sh [feature-area]\n\nset -e\n\nAREA=${1:-all}\n\n# Colors\nRED='\\033[0;31m'\nGREEN='\\033[0;32m'\nYELLOW='\\033[1;33m'\nBLUE='\\033[0;34m'\nNC='\\033[0m'\n\necho -e \"${BLUE}==> Fetching latest quotio...${NC}\"\ngit fetch quotio 2>/dev/null || {\n echo -e \"${YELLOW}Adding quotio remote...${NC}\"\n git remote add quotio https://github.com/nguyenphutrong/quotio.git\n git fetch quotio\n}\n\necho \"\"\necho -e \"${GREEN}==> Quotio Repository Analysis${NC}\"\necho \"\"\n\n# Show recent activity\necho -e \"${BLUE}Recent Activity (last 30 days):${NC}\"\ngit log --oneline --graph --remotes=quotio/main --since=\"30 days ago\" | head -20\necho \"\"\n\n# Analyze file structure\necho -e \"${BLUE}File Structure:${NC}\"\ngit ls-tree -r --name-only quotio/main | grep -E '\\.(swift|md)$' | head -30\necho \"\"\n\n# Find interesting patterns based on area\ncase $AREA in\n \"providers\"|\"all\")\n echo -e \"${BLUE}Provider Implementations:${NC}\"\n git ls-tree -r --name-only quotio/main | grep -i provider | head -20\n echo \"\"\n ;;\nesac\n\ncase $AREA in\n \"ui\"|\"all\")\n echo -e \"${BLUE}UI Components:${NC}\"\n git ls-tree -r --name-only quotio/main | grep -iE '(view|ui|menu)' | head -20\n echo \"\"\n ;;\nesac\n\ncase $AREA in\n \"auth\"|\"all\")\n echo -e \"${BLUE}Authentication/Session:${NC}\"\n git ls-tree -r --name-only quotio/main | grep -iE '(auth|session|cookie|login)' | head -20\n echo \"\"\n ;;\nesac\n\n# Show commit messages for pattern analysis\necho -e \"${BLUE}Recent Commit Messages (for pattern analysis):${NC}\"\ngit log --oneline quotio/main --since=\"60 days ago\" | head -30\necho \"\"\n\n# Create analysis report\nREPORT_FILE=\"quotio-analysis-$(date +%Y%m%d).md\"\ncat > \"$REPORT_FILE\" << EOF\n# Quotio Analysis Report\n**Date:** $(date +%Y-%m-%d)\n**Purpose:** Identify patterns and features for CodexBar fork inspiration\n\n## Recent Activity\n\\`\\`\\`\n$(git log --oneline --graph --remotes=quotio/main --since=\"30 days ago\" | head -20)\n\\`\\`\\`\n\n## File Structure\n\\`\\`\\`\n$(git ls-tree -r --name-only quotio/main | grep -E '\\.(swift|md)$' | head -50)\n\\`\\`\\`\n\n## Recent Commits\n\\`\\`\\`\n$(git log --oneline quotio/main --since=\"60 days ago\" | head -30)\n\\`\\`\\`\n\n## Areas of Interest\n\n### Providers\n- [ ] Review provider implementations\n- [ ] Compare with CodexBar approach\n- [ ] Identify improvements\n\n### UI/UX\n- [ ] Menu bar organization\n- [ ] Settings layout\n- [ ] Status indicators\n\n### Authentication\n- [ ] Session management\n- [ ] Cookie handling\n- [ ] OAuth flows\n\n### Multi-Account\n- [ ] Account switching\n- [ ] Account storage\n- [ ] UI patterns\n\n## Action Items\n- [ ] Review specific files of interest\n- [ ] Document patterns (not code)\n- [ ] Create implementation plan\n- [ ] Implement independently\n\n## Notes\nRemember: We're looking for PATTERNS and IDEAS, not copying code.\nAll implementations must be original and follow CodexBar conventions.\nEOF\n\necho -e \"${GREEN}Analysis report saved to: $REPORT_FILE${NC}\"\necho \"\"\necho -e \"${YELLOW}Next steps:${NC}\"\necho \"\"\necho \"1. View specific files:\"\necho \" ${GREEN}git show quotio/main:path/to/file${NC}\"\necho \"\"\necho \"2. Compare implementations:\"\necho \" ${GREEN}git diff main quotio/main -- path/to/similar/file${NC}\"\necho \"\"\necho \"3. Review commit details:\"\necho \" ${GREEN}git log -p quotio/main --since='30 days ago'${NC}\"\necho \"\"\necho \"4. Document patterns in:\"\necho \" ${GREEN}docs/QUOTIO_ANALYSIS.md${NC}\"\necho \"\"\necho -e \"${BLUE}Remember: Adapt patterns, don't copy code!${NC}\"\n\n" }, { "path": "Scripts/build_icon.sh", "content": "#!/usr/bin/env bash\nset -euo pipefail\n\nICON_FILE=${1:-Icon.icon}\nBASENAME=${2:-Icon}\nOUT_ROOT=${3:-build/icon}\nXCODE_APP=${XCODE_APP:-/Applications/Xcode.app}\n\nICTOOL=\"$XCODE_APP/Contents/Applications/Icon Composer.app/Contents/Executables/ictool\"\nif [[ ! -x \"$ICTOOL\" ]]; then\n ICTOOL=\"$XCODE_APP/Contents/Applications/Icon Composer.app/Contents/Executables/icontool\"\nfi\nif [[ ! -x \"$ICTOOL\" ]]; then\n echo \"ictool/icontool not found. Set XCODE_APP if Xcode is elsewhere.\" >&2\n exit 1\nfi\n\nICONSET_DIR=\"$OUT_ROOT/${BASENAME}.iconset\"\nTMP_DIR=\"$OUT_ROOT/tmp\"\nmkdir -p \"$ICONSET_DIR\" \"$TMP_DIR\"\n\nMASTER_ART=\"$TMP_DIR/icon_art_824.png\"\nMASTER_1024=\"$TMP_DIR/icon_1024.png\"\n\n# Render inner art (no margin) with macOS Default appearance\n\"$ICTOOL\" \"$ICON_FILE\" \\\n --export-preview macOS Default 824 824 1 -45 \"$MASTER_ART\"\n\n# Pad to 1024x1024 with transparent border\nsips --padToHeightWidth 1024 1024 \"$MASTER_ART\" --out \"$MASTER_1024\" >/dev/null\n\n# Generate required sizes\nsizes=(16 32 64 128 256 512 1024)\nfor sz in \"${sizes[@]}\"; do\n out=\"$ICONSET_DIR/icon_${sz}x${sz}.png\"\n sips -z \"$sz\" \"$sz\" \"$MASTER_1024\" --out \"$out\" >/dev/null\n if [[ \"$sz\" -ne 1024 ]]; then\n dbl=$((sz*2))\n out2=\"$ICONSET_DIR/icon_${sz}x${sz}@2x.png\"\n sips -z \"$dbl\" \"$dbl\" \"$MASTER_1024\" --out \"$out2\" >/dev/null\n fi\ndone\n\n# 512x512@2x already covered by 1024; ensure it exists\ncp \"$MASTER_1024\" \"$ICONSET_DIR/icon_512x512@2x.png\"\n\niconutil -c icns \"$ICONSET_DIR\" -o Icon.icns\n\necho \"Icon.icns generated at $(pwd)/Icon.icns\"\n" }, { "path": "Scripts/changelog-to-html.sh", "content": "#!/usr/bin/env bash\nset -euo pipefail\n\nVERSION=${1:-}\nCHANGELOG_FILE=${2:-}\n\nif [[ -z \"$VERSION\" ]]; then\n echo \"Usage: $0 [changelog_file]\" >&2\n exit 1\nfi\n\nSCRIPT_DIR=$(cd \"$(dirname \"${BASH_SOURCE[0]}\")\" && pwd)\nif [[ -z \"$CHANGELOG_FILE\" ]]; then\n if [[ -f \"$SCRIPT_DIR/../CHANGELOG.md\" ]]; then\n CHANGELOG_FILE=\"$SCRIPT_DIR/../CHANGELOG.md\"\n elif [[ -f \"CHANGELOG.md\" ]]; then\n CHANGELOG_FILE=\"CHANGELOG.md\"\n elif [[ -f \"../CHANGELOG.md\" ]]; then\n CHANGELOG_FILE=\"../CHANGELOG.md\"\n else\n echo \"Error: Could not find CHANGELOG.md\" >&2\n exit 1\n fi\nfi\n\nif [[ ! -f \"$CHANGELOG_FILE\" ]]; then\n echo \"Error: Changelog file '$CHANGELOG_FILE' not found\" >&2\n exit 1\nfi\n\nextract_version_section() {\n local version=$1\n local file=$2\n awk -v version=\"$version\" '\n BEGIN { found=0 }\n /^## / {\n if ($0 ~ \"^##[[:space:]]+\" version \"([[:space:]].*|$)\") { found=1; next }\n if (found) { exit }\n }\n found { print }\n ' \"$file\"\n}\n\nmarkdown_to_html() {\n local text=$1\n text=$(echo \"$text\" | sed 's/^### \\(.*\\)$/

\\1<\\/h3>/')\n text=$(echo \"$text\" | sed 's/^## \\(.*\\)$/

\\1<\\/h2>/')\n text=$(echo \"$text\" | sed 's/^- \\*\\*\\([^*]*\\)\\*\\*\\(.*\\)$/
  • \\1<\\/strong>\\2<\\/li>/')\n text=$(echo \"$text\" | sed 's/^- \\([^*].*\\)$/
  • \\1<\\/li>/')\n text=$(echo \"$text\" | sed 's/\\*\\*\\([^*]*\\)\\*\\*/\\1<\\/strong>/g')\n text=$(echo \"$text\" | sed 's/`\\([^`]*\\)`/\\1<\\/code>/g')\n text=$(echo \"$text\" | sed 's/\\[\\([^]]*\\)\\](\\([^)]*\\))/\\1<\\/a>/g')\n echo \"$text\"\n}\n\nversion_content=$(extract_version_section \"$VERSION\" \"$CHANGELOG_FILE\")\nif [[ -z \"$version_content\" ]]; then\n echo \"

    CodexBar $VERSION

    \"\n echo \"

    Latest CodexBar update.

    \"\n echo \"

    View full changelog

    \"\n exit 0\nfi\n\necho \"

    CodexBar $VERSION

    \"\n\nin_list=false\nwhile IFS= read -r line; do\n if [[ \"$line\" =~ ^- ]]; then\n if [[ \"$in_list\" == false ]]; then\n echo \"
      \"\n in_list=true\n fi\n markdown_to_html \"$line\"\n else\n if [[ \"$in_list\" == true ]]; then\n echo \"
    \"\n in_list=false\n fi\n if [[ -n \"$line\" ]]; then\n markdown_to_html \"$line\"\n fi\n fi\ndone <<< \"$version_content\"\n\nif [[ \"$in_list\" == true ]]; then\n echo \"\"\nfi\n\necho \"

    View full changelog

    \"\n" }, { "path": "Scripts/check-release-assets.sh", "content": "#!/usr/bin/env bash\nset -euo pipefail\n\nROOT=$(cd \"$(dirname \"$0\")/..\" && pwd)\nsource \"$HOME/Projects/agent-scripts/release/sparkle_lib.sh\"\n\nTAG=${1:-$(git describe --tags --abbrev=0)}\nARTIFACT_PREFIX=\"CodexBar-\"\n\ncheck_assets \"$TAG\" \"$ARTIFACT_PREFIX\"\n" }, { "path": "Scripts/check_upstreams.sh", "content": "#!/bin/bash\n# Check for new changes in upstream repositories\n# Usage: ./Scripts/check_upstreams.sh [upstream|quotio|all]\n\nset -e\n\nTARGET=${1:-all}\nDAYS=${2:-7}\n\n# Colors for output\nRED='\\033[0;31m'\nGREEN='\\033[0;32m'\nYELLOW='\\033[1;33m'\nBLUE='\\033[0;34m'\nNC='\\033[0m' # No Color\n\necho -e \"${BLUE}==> Fetching upstream changes...${NC}\"\nif [ \"$TARGET\" = \"all\" ] || [ \"$TARGET\" = \"upstream\" ]; then\n git fetch upstream 2>/dev/null || {\n echo -e \"${YELLOW}Adding upstream remote...${NC}\"\n git remote add upstream https://github.com/steipete/CodexBar.git\n git fetch upstream\n }\nfi\n\nif [ \"$TARGET\" = \"all\" ] || [ \"$TARGET\" = \"quotio\" ]; then\n git fetch quotio 2>/dev/null || {\n echo -e \"${YELLOW}Adding quotio remote...${NC}\"\n git remote add quotio https://github.com/nguyenphutrong/quotio.git\n git fetch quotio\n }\nfi\n\necho \"\"\n\n# Check upstream (steipete)\nif [ \"$TARGET\" = \"all\" ] || [ \"$TARGET\" = \"upstream\" ]; then\n echo -e \"${BLUE}==> Upstream (steipete/CodexBar) changes:${NC}\"\n \n UPSTREAM_COUNT=$(git log --oneline main..upstream/main --no-merges 2>/dev/null | wc -l | tr -d ' ')\n \n if [ \"$UPSTREAM_COUNT\" -gt 0 ]; then\n echo -e \"${GREEN}Found $UPSTREAM_COUNT new commits${NC}\"\n echo \"\"\n git log --oneline --graph main..upstream/main --no-merges | head -20\n echo \"\"\n echo -e \"${YELLOW}Files changed:${NC}\"\n git diff --stat main..upstream/main | tail -20\n else\n echo -e \"${GREEN}No new commits (up to date)${NC}\"\n fi\n echo \"\"\nfi\n\n# Check quotio\nif [ \"$TARGET\" = \"all\" ] || [ \"$TARGET\" = \"quotio\" ]; then\n echo -e \"${BLUE}==> Quotio changes (last $DAYS days):${NC}\"\n \n QUOTIO_COUNT=$(git log --oneline --all --remotes=quotio/main --since=\"$DAYS days ago\" 2>/dev/null | wc -l | tr -d ' ')\n \n if [ \"$QUOTIO_COUNT\" -gt 0 ]; then\n echo -e \"${GREEN}Found $QUOTIO_COUNT commits in last $DAYS days${NC}\"\n echo \"\"\n git log --oneline --graph --remotes=quotio/main --since=\"$DAYS days ago\" | head -20\n echo \"\"\n echo -e \"${YELLOW}Recent file changes:${NC}\"\n # Show changes from last 10 commits\n git diff --stat quotio/main~10..quotio/main 2>/dev/null | tail -20 || echo \"Unable to show diff\"\n else\n echo -e \"${GREEN}No new commits in last $DAYS days${NC}\"\n fi\n echo \"\"\nfi\n\n# Summary\necho -e \"${BLUE}==> Summary${NC}\"\nif [ \"$TARGET\" = \"all\" ] || [ \"$TARGET\" = \"upstream\" ]; then\n echo \"Upstream commits: $UPSTREAM_COUNT\"\nfi\nif [ \"$TARGET\" = \"all\" ] || [ \"$TARGET\" = \"quotio\" ]; then\n echo \"Quotio commits (${DAYS}d): $QUOTIO_COUNT\"\nfi\n\necho \"\"\necho -e \"${YELLOW}Next steps:${NC}\"\necho \" Review upstream: ./Scripts/review_upstream.sh upstream\"\necho \" Review quotio: ./Scripts/review_upstream.sh quotio\"\necho \" Detailed diff: git diff main..upstream/main\"\necho \" View quotio: git log -p quotio/main~10..quotio/main\"\n\n" }, { "path": "Scripts/compile_and_run.sh", "content": "#!/usr/bin/env bash\n# Reset CodexBar: kill running instances, build, package, relaunch, verify.\n\nset -euo pipefail\n\nROOT_DIR=\"$(cd \"$(dirname \"${BASH_SOURCE[0]}\")/..\" && pwd)\"\nAPP_BUNDLE=\"${ROOT_DIR}/CodexBar.app\"\nAPP_PROCESS_PATTERN=\"CodexBar.app/Contents/MacOS/CodexBar\"\nDEBUG_PROCESS_PATTERN=\"${ROOT_DIR}/.build/debug/CodexBar\"\nRELEASE_PROCESS_PATTERN=\"${ROOT_DIR}/.build/release/CodexBar\"\nLOCK_KEY=\"$(printf '%s' \"${ROOT_DIR}\" | shasum -a 256 | cut -c1-8)\"\nLOCK_DIR=\"${TMPDIR:-/tmp}/codexbar-compile-and-run-${LOCK_KEY}\"\nLOCK_PID_FILE=\"${LOCK_DIR}/pid\"\nWAIT_FOR_LOCK=0\nRUN_TESTS=0\nDEBUG_LLDB=0\nRELEASE_ARCHES=\"\"\nSIGNING_MODE=\"${CODEXBAR_SIGNING:-}\"\n\nlog() { printf '%s\\n' \"$*\"; }\nfail() { printf 'ERROR: %s\\n' \"$*\" >&2; exit 1; }\n\nhas_signing_identity() {\n local identity=\"${1:-}\"\n if [[ -z \"${identity}\" ]]; then\n return 1\n fi\n security find-identity -p codesigning -v 2>/dev/null | grep -F \"${identity}\" >/dev/null 2>&1\n}\n\nresolve_signing_mode() {\n if [[ -n \"${SIGNING_MODE}\" ]]; then\n return\n fi\n\n if [[ -n \"${APP_IDENTITY:-}\" ]]; then\n if has_signing_identity \"${APP_IDENTITY}\"; then\n SIGNING_MODE=\"identity\"\n return\n fi\n log \"WARN: APP_IDENTITY not found in Keychain; falling back to adhoc signing.\"\n SIGNING_MODE=\"adhoc\"\n return\n fi\n\n local candidate=\"\"\n for candidate in \\\n \"Developer ID Application: Peter Steinberger (Y5PE65HELJ)\" \\\n \"CodexBar Development\"\n do\n if has_signing_identity \"${candidate}\"; then\n APP_IDENTITY=\"${candidate}\"\n export APP_IDENTITY\n SIGNING_MODE=\"identity\"\n return\n fi\n done\n\n SIGNING_MODE=\"adhoc\"\n}\n\nrun_step() {\n local label=\"$1\"; shift\n log \"==> ${label}\"\n if ! \"$@\"; then\n fail \"${label} failed\"\n fi\n}\n\ncleanup() {\n if [[ -d \"${LOCK_DIR}\" ]]; then\n rm -rf \"${LOCK_DIR}\"\n fi\n}\n\nacquire_lock() {\n while true; do\n if mkdir \"${LOCK_DIR}\" 2>/dev/null; then\n echo \"$$\" > \"${LOCK_PID_FILE}\"\n return 0\n fi\n\n local existing_pid=\"\"\n if [[ -f \"${LOCK_PID_FILE}\" ]]; then\n existing_pid=\"$(cat \"${LOCK_PID_FILE}\" 2>/dev/null || true)\"\n fi\n\n if [[ -n \"${existing_pid}\" ]] && kill -0 \"${existing_pid}\" 2>/dev/null; then\n if [[ \"${WAIT_FOR_LOCK}\" == \"1\" ]]; then\n log \"==> Another agent is compiling (pid ${existing_pid}); waiting...\"\n while kill -0 \"${existing_pid}\" 2>/dev/null; do\n sleep 1\n done\n continue\n fi\n log \"==> Another agent is compiling (pid ${existing_pid}); re-run with --wait.\"\n exit 0\n fi\n\n rm -rf \"${LOCK_DIR}\"\n done\n}\n\ntrap cleanup EXIT INT TERM\n\nkill_claude_probes() {\n # CodexBar spawns `claude /usage` + `/status` in a PTY; if we kill the app mid-probe we can orphan them.\n pkill -f \"claude (/status|/usage) --allowed-tools\" 2>/dev/null || true\n sleep 0.2\n pkill -9 -f \"claude (/status|/usage) --allowed-tools\" 2>/dev/null || true\n}\n\nkill_all_codexbar() {\n is_running() {\n pgrep -f \"${APP_PROCESS_PATTERN}\" >/dev/null 2>&1 \\\n || pgrep -f \"${DEBUG_PROCESS_PATTERN}\" >/dev/null 2>&1 \\\n || pgrep -f \"${RELEASE_PROCESS_PATTERN}\" >/dev/null 2>&1 \\\n || pgrep -x \"CodexBar\" >/dev/null 2>&1\n }\n\n # Phase 1: request termination (give the app time to exit cleanly).\n for _ in {1..25}; do\n pkill -f \"${APP_PROCESS_PATTERN}\" 2>/dev/null || true\n pkill -f \"${DEBUG_PROCESS_PATTERN}\" 2>/dev/null || true\n pkill -f \"${RELEASE_PROCESS_PATTERN}\" 2>/dev/null || true\n pkill -x \"CodexBar\" 2>/dev/null || true\n if ! is_running; then\n return 0\n fi\n sleep 0.2\n done\n\n # Phase 2: force kill any stragglers (avoids `open -n` creating multiple instances).\n pkill -9 -f \"${APP_PROCESS_PATTERN}\" 2>/dev/null || true\n pkill -9 -f \"${DEBUG_PROCESS_PATTERN}\" 2>/dev/null || true\n pkill -9 -f \"${RELEASE_PROCESS_PATTERN}\" 2>/dev/null || true\n pkill -9 -x \"CodexBar\" 2>/dev/null || true\n\n for _ in {1..25}; do\n if ! is_running; then\n return 0\n fi\n sleep 0.2\n done\n\n fail \"Failed to kill all CodexBar instances.\"\n}\n\n# 1) Ensure a single runner instance.\nfor arg in \"$@\"; do\n case \"${arg}\" in\n --wait|-w) WAIT_FOR_LOCK=1 ;;\n --test|-t) RUN_TESTS=1 ;;\n --debug-lldb) DEBUG_LLDB=1 ;;\n --release-universal) RELEASE_ARCHES=\"arm64 x86_64\" ;;\n --release-arches=*) RELEASE_ARCHES=\"${arg#*=}\" ;;\n --help|-h)\n log \"Usage: $(basename \"$0\") [--wait] [--test] [--debug-lldb] [--release-universal] [--release-arches=\\\"arm64 x86_64\\\"]\"\n exit 0\n ;;\n *)\n ;;\n esac\ndone\n\nresolve_signing_mode\nif [[ \"${SIGNING_MODE}\" == \"adhoc\" ]]; then\n log \"==> Signing: adhoc (set APP_IDENTITY or install a dev cert to avoid keychain prompts)\"\nelse\n log \"==> Signing: ${APP_IDENTITY:-Developer ID Application}\"\nfi\n\nacquire_lock\n\n# 2) Kill all running CodexBar instances (debug, release, bundled).\nlog \"==> Killing existing CodexBar instances\"\nkill_all_codexbar\nkill_claude_probes\n\n# 2.5) Delete keychain entries to avoid permission prompts with adhoc signing\n# (adhoc signature changes on every build, making old keychain entries inaccessible)\nif [[ \"${SIGNING_MODE:-adhoc}\" == \"adhoc\" ]]; then\n log \"==> Clearing keychain entries (adhoc signing)\"\n security delete-generic-password -s \"com.steipete.CodexBar\" 2>/dev/null || true\n # Clear all keychain items for the app to avoid multiple prompts\n while security delete-generic-password -s \"com.steipete.CodexBar\" 2>/dev/null; do\n :\n done\nfi\n\n# 3) Package (release build happens inside package_app.sh).\nif [[ \"${RUN_TESTS}\" == \"1\" ]]; then\n run_step \"swift test\" swift test -q\nfi\nif [[ \"${DEBUG_LLDB}\" == \"1\" && -n \"${RELEASE_ARCHES}\" ]]; then\n fail \"--release-arches is only supported for release packaging\"\nfi\nHOST_ARCH=\"$(uname -m)\"\nARCHES_VALUE=\"${HOST_ARCH}\"\nif [[ -n \"${RELEASE_ARCHES}\" ]]; then\n ARCHES_VALUE=\"${RELEASE_ARCHES}\"\nfi\nif [[ \"${DEBUG_LLDB}\" == \"1\" ]]; then\n run_step \"package app\" env CODEXBAR_ALLOW_LLDB=1 ARCHES=\"${ARCHES_VALUE}\" \"${ROOT_DIR}/Scripts/package_app.sh\" debug\nelse\n if [[ -n \"${SIGNING_MODE}\" ]]; then\n run_step \"package app\" env CODEXBAR_SIGNING=\"${SIGNING_MODE}\" ARCHES=\"${ARCHES_VALUE}\" \"${ROOT_DIR}/Scripts/package_app.sh\"\n else\n run_step \"package app\" env ARCHES=\"${ARCHES_VALUE}\" \"${ROOT_DIR}/Scripts/package_app.sh\"\n fi\nfi\n\n# 4) Launch the packaged app.\nlog \"==> launch app\"\nif ! open \"${APP_BUNDLE}\"; then\n log \"WARN: launch app returned non-zero; falling back to direct binary launch.\"\n \"${APP_BUNDLE}/Contents/MacOS/CodexBar\" >/dev/null 2>&1 &\n disown\nfi\n\n# 5) Verify the app stays up for at least a moment (launch can be >1s on some systems).\nfor _ in {1..10}; do\n if pgrep -f \"${APP_PROCESS_PATTERN}\" >/dev/null 2>&1; then\n log \"OK: CodexBar is running.\"\n exit 0\n fi\n sleep 0.4\ndone\nfail \"App exited immediately. Check crash logs in Console.app (User Reports).\"\n" }, { "path": "Scripts/docs-list.mjs", "content": "#!/usr/bin/env node\nimport { readdirSync, readFileSync } from 'node:fs';\nimport { dirname, join, relative } from 'node:path';\nimport { fileURLToPath } from 'node:url';\n\nconst DOCS_DIR = join(dirname(fileURLToPath(import.meta.url)), '..', 'docs');\nconst EXCLUDED_DIRS = new Set(['archive', 'research']);\n\nfunction walkMarkdownFiles(dir, base = dir) {\n const entries = readdirSync(dir, { withFileTypes: true });\n const files = [];\n\n for (const entry of entries) {\n if (entry.name.startsWith('.')) continue;\n const fullPath = join(dir, entry.name);\n\n if (entry.isDirectory()) {\n if (EXCLUDED_DIRS.has(entry.name)) continue;\n files.push(...walkMarkdownFiles(fullPath, base));\n } else if (entry.isFile() && entry.name.endsWith('.md')) {\n files.push(relative(base, fullPath));\n }\n }\n\n return files.sort((a, b) => a.localeCompare(b));\n}\n\nfunction extractMetadata(fullPath) {\n const content = readFileSync(fullPath, 'utf8');\n\n if (!content.startsWith('---')) {\n return { summary: null, readWhen: [], error: 'missing front matter' };\n }\n\n const endIndex = content.indexOf('\\n---', 3);\n if (endIndex === -1) {\n return { summary: null, readWhen: [], error: 'unterminated front matter' };\n }\n\n const frontMatter = content.slice(3, endIndex).trim();\n const lines = frontMatter.split('\\n');\n\n let summaryLine = null;\n const readWhen = [];\n let collectingReadWhen = false;\n\n for (const rawLine of lines) {\n const line = rawLine.trim();\n\n if (line.startsWith('summary:')) {\n summaryLine = line;\n collectingReadWhen = false;\n continue;\n }\n\n if (line.startsWith('read_when:')) {\n collectingReadWhen = true;\n const inline = line.slice('read_when:'.length).trim();\n if (inline.startsWith('[') && inline.endsWith(']')) {\n try {\n const parsed = JSON.parse(inline.replace(/'/g, '\"'));\n if (Array.isArray(parsed)) {\n parsed\n .map((v) => String(v).trim())\n .filter(Boolean)\n .forEach((v) => readWhen.push(v));\n }\n } catch {\n /* ignore malformed inline */\n }\n }\n continue;\n }\n\n if (collectingReadWhen) {\n if (line.startsWith('- ')) {\n const hint = line.slice(2).trim();\n if (hint) readWhen.push(hint);\n } else if (line === '') {\n // allow blank spacer lines inside list\n } else {\n collectingReadWhen = false;\n }\n }\n }\n\n if (!summaryLine) {\n return { summary: null, readWhen, error: 'summary key missing' };\n }\n\n const summaryValue = summaryLine.slice('summary:'.length).trim();\n const normalized = summaryValue.replace(/^['\"]|['\"]$/g, '').replace(/\\s+/g, ' ').trim();\n\n if (!normalized) {\n return { summary: null, readWhen, error: 'summary is empty' };\n }\n\n return { summary: normalized, readWhen };\n}\n\nconsole.log('Listing all markdown files in docs folder:');\n\nconst markdownFiles = walkMarkdownFiles(DOCS_DIR);\n\nfor (const relativePath of markdownFiles) {\n const fullPath = join(DOCS_DIR, relativePath);\n const { summary, readWhen, error } = extractMetadata(fullPath);\n if (summary) {\n console.log(`${relativePath} - ${summary}`);\n if (readWhen.length > 0) {\n console.log(` Read when: ${readWhen.join('; ')}`);\n }\n } else {\n const reason = error ? ` - [${error}]` : '';\n console.log(`${relativePath}${reason}`);\n }\n}\n\nconsole.log('\\nReminder: keep docs up to date as behavior changes. When your task matches any \"Read when\" hint above (React hooks, cache directives, database work, tests, etc.), read that doc before coding, and suggest new coverage when it is missing.');\n" }, { "path": "Scripts/install_lint_tools.sh", "content": "#!/usr/bin/env bash\n\nset -euo pipefail\n\nROOT_DIR=\"$(cd \"$(dirname \"${BASH_SOURCE[0]}\")/..\" && pwd)\"\nTOOLS_DIR=\"${ROOT_DIR}/.build/lint-tools\"\nBIN_DIR=\"${TOOLS_DIR}/bin\"\n\nSWIFTFORMAT_VERSION=\"0.59.1\"\nSWIFTLINT_VERSION=\"0.63.2\"\n\nSWIFTFORMAT_SHA256_DARWIN=\"8b6289b608a44e73cd3851c3589dbd7c553f32cc805aa54b3a496ce2b90febe7\"\nSWIFTLINT_SHA256_DARWIN=\"c59a405c85f95b92ced677a500804e081596a4cae4a6a485af76065557d6ed29\"\n\nlog() { printf '%s\\n' \"$*\"; }\nfail() { printf 'ERROR: %s\\n' \"$*\" >&2; exit 1; }\n\nsha256_value() {\n local path=\"$1\"\n if command -v shasum >/dev/null 2>&1; then\n shasum -a 256 \"$path\" | awk '{print $1}'\n return 0\n fi\n if command -v sha256sum >/dev/null 2>&1; then\n sha256sum \"$path\" | awk '{print $1}'\n return 0\n fi\n fail \"Missing shasum/sha256sum.\"\n}\n\ndownload_file() {\n local url=\"$1\"\n local out=\"$2\"\n curl -fL --retry 3 --retry-connrefused --retry-delay 2 -o \"$out\" \"$url\"\n}\n\ninstall_zip_binary() {\n local label=\"$1\"\n local url=\"$2\"\n local expected_sha=\"$3\"\n local binary_name=\"$4\"\n\n local tmp_zip\n tmp_zip=\"$(mktemp -t \"${label}.XXXX\")\"\n local tmp_dir\n tmp_dir=\"$(mktemp -d -t \"${label}.XXXX\")\"\n\n log \"==> Downloading ${label}\"\n download_file \"$url\" \"$tmp_zip\"\n\n local actual_sha\n actual_sha=\"$(sha256_value \"$tmp_zip\")\"\n if [[ -n \"$expected_sha\" && \"$actual_sha\" != \"$expected_sha\" ]]; then\n rm -f \"$tmp_zip\"\n rm -rf \"$tmp_dir\"\n fail \"${label} SHA256 mismatch (expected ${expected_sha}, got ${actual_sha})\"\n fi\n\n unzip -q \"$tmp_zip\" -d \"$tmp_dir\"\n\n local extracted_path=\"\"\n if [[ -f \"${tmp_dir}/${binary_name}\" ]]; then\n extracted_path=\"${tmp_dir}/${binary_name}\"\n else\n extracted_path=\"$(find \"$tmp_dir\" -type f -name \"$binary_name\" | head -n 1 || true)\"\n fi\n\n if [[ -z \"$extracted_path\" || ! -f \"$extracted_path\" ]]; then\n rm -f \"$tmp_zip\"\n rm -rf \"$tmp_dir\"\n fail \"${label} binary '${binary_name}' not found in archive\"\n fi\n\n install -m 0755 \"$extracted_path\" \"${BIN_DIR}/${binary_name}\"\n\n rm -f \"$tmp_zip\"\n rm -rf \"$tmp_dir\"\n}\n\nmkdir -p \"$BIN_DIR\"\n\nif [[ -x \"${BIN_DIR}/swiftformat\" && -x \"${BIN_DIR}/swiftlint\" ]]; then\n if [[ \"$(\"${BIN_DIR}/swiftformat\" --version 2>/dev/null || true)\" == \"${SWIFTFORMAT_VERSION}\" ]] \\\n && [[ \"$(\"${BIN_DIR}/swiftlint\" version 2>/dev/null || true)\" == \"${SWIFTLINT_VERSION}\" ]]\n then\n log \"==> Lint tools already installed (${SWIFTFORMAT_VERSION}, ${SWIFTLINT_VERSION})\"\n exit 0\n fi\nfi\n\nOS=\"$(uname -s)\"\nARCH=\"$(uname -m)\"\n\ncase \"$OS\" in\n Darwin)\n SWIFTFORMAT_URL=\"https://github.com/nicklockwood/SwiftFormat/releases/download/${SWIFTFORMAT_VERSION}/swiftformat.zip\"\n SWIFTLINT_URL=\"https://github.com/realm/SwiftLint/releases/download/${SWIFTLINT_VERSION}/portable_swiftlint.zip\"\n\n install_zip_binary \"SwiftFormat ${SWIFTFORMAT_VERSION}\" \"$SWIFTFORMAT_URL\" \"$SWIFTFORMAT_SHA256_DARWIN\" \"swiftformat\"\n install_zip_binary \"SwiftLint ${SWIFTLINT_VERSION}\" \"$SWIFTLINT_URL\" \"$SWIFTLINT_SHA256_DARWIN\" \"swiftlint\"\n ;;\n Linux)\n case \"$ARCH\" in\n x86_64)\n SWIFTFORMAT_URL=\"https://github.com/nicklockwood/SwiftFormat/releases/download/${SWIFTFORMAT_VERSION}/swiftformat_linux.zip\"\n SWIFTLINT_URL=\"https://github.com/realm/SwiftLint/releases/download/${SWIFTLINT_VERSION}/swiftlint_linux_amd64.zip\"\n ;;\n aarch64|arm64)\n SWIFTFORMAT_URL=\"https://github.com/nicklockwood/SwiftFormat/releases/download/${SWIFTFORMAT_VERSION}/swiftformat_linux_aarch64.zip\"\n SWIFTLINT_URL=\"https://github.com/realm/SwiftLint/releases/download/${SWIFTLINT_VERSION}/swiftlint_linux_arm64.zip\"\n ;;\n *)\n fail \"Unsupported Linux arch: ${ARCH}\"\n ;;\n esac\n\n # SHA256 is intentionally only enforced for the macOS CI path.\n # If we later run lint on Linux CI, add pinned SHAs here as well.\n log \"WARN: Linux SHA256 verification not configured for ${ARCH}; installing anyway.\"\n install_zip_binary \"SwiftFormat ${SWIFTFORMAT_VERSION}\" \"$SWIFTFORMAT_URL\" \"\" \"swiftformat\"\n install_zip_binary \"SwiftLint ${SWIFTLINT_VERSION}\" \"$SWIFTLINT_URL\" \"\" \"swiftlint\"\n ;;\n *)\n fail \"Unsupported OS: ${OS}\"\n ;;\nesac\n\nlog \"==> Installed lint tools to ${BIN_DIR}\"\n\"${BIN_DIR}/swiftformat\" --version\n\"${BIN_DIR}/swiftlint\" version\n" }, { "path": "Scripts/launch.sh", "content": "#!/bin/bash\nset -euo pipefail\n\n# Simple script to launch CodexBar (kills existing instance first)\n# Usage: ./Scripts/launch.sh\n\nSCRIPT_DIR=\"$(cd \"$(dirname \"${BASH_SOURCE[0]}\")\" && pwd)\"\nPROJECT_ROOT=\"$(cd \"$SCRIPT_DIR/..\" && pwd)\"\nAPP_PATH=\"$PROJECT_ROOT/CodexBar.app\"\n\necho \"==> Killing existing CodexBar instances\"\npkill -x CodexBar || pkill -f CodexBar.app || true\nsleep 0.5\n\nif [[ ! -d \"$APP_PATH\" ]]; then\n echo \"ERROR: CodexBar.app not found at $APP_PATH\"\n echo \"Run ./Scripts/package_app.sh first to build the app\"\n exit 1\nfi\n\necho \"==> Launching CodexBar from $APP_PATH\"\nopen -n \"$APP_PATH\"\n\n# Wait a moment and check if it's running\nsleep 1\nif pgrep -x CodexBar > /dev/null; then\n echo \"OK: CodexBar is running.\"\nelse\n echo \"ERROR: App exited immediately. Check crash logs in Console.app (User Reports).\"\n exit 1\nfi\n\n" }, { "path": "Scripts/lint.sh", "content": "#!/usr/bin/env bash\n\nset -euo pipefail\n\nROOT_DIR=\"$(cd \"$(dirname \"${BASH_SOURCE[0]}\")/..\" && pwd)\"\nBIN_DIR=\"${ROOT_DIR}/.build/lint-tools/bin\"\n\nensure_tools() {\n # Always delegate to the installer so pinned versions are enforced.\n # The installer is idempotent and exits early when the expected versions are already present.\n \"${ROOT_DIR}/Scripts/install_lint_tools.sh\"\n}\n\ncmd=\"${1:-lint}\"\n\ncase \"$cmd\" in\n lint)\n ensure_tools\n \"${BIN_DIR}/swiftformat\" Sources Tests --lint\n \"${BIN_DIR}/swiftlint\" --strict\n ;;\n format)\n ensure_tools\n \"${BIN_DIR}/swiftformat\" Sources Tests\n ;;\n *)\n printf 'Usage: %s [lint|format]\\n' \"$(basename \"$0\")\" >&2\n exit 2\n ;;\nesac\n" }, { "path": "Scripts/make_appcast.sh", "content": "#!/usr/bin/env bash\nset -euo pipefail\n\nROOT=$(cd \"$(dirname \"$0\")/..\" && pwd)\nZIP=${1:?\n\"Usage: $0 CodexBar-.zip\"}\nFEED_URL=${2:-\"https://raw.githubusercontent.com/steipete/CodexBar/main/appcast.xml\"}\nPRIVATE_KEY_FILE=${SPARKLE_PRIVATE_KEY_FILE:-}\nSPARKLE_CHANNEL=${SPARKLE_CHANNEL:-}\nif [[ -z \"$PRIVATE_KEY_FILE\" ]]; then\n echo \"Set SPARKLE_PRIVATE_KEY_FILE to your ed25519 private key (Sparkle).\" >&2\n exit 1\nfi\nif [[ ! -f \"$ZIP\" ]]; then\n echo \"Zip not found: $ZIP\" >&2\n exit 1\nfi\n\nZIP_DIR=$(cd \"$(dirname \"$ZIP\")\" && pwd)\nZIP_NAME=$(basename \"$ZIP\")\nZIP_BASE=\"${ZIP_NAME%.zip}\"\nVERSION=${SPARKLE_RELEASE_VERSION:-}\nif [[ -z \"$VERSION\" ]]; then\n if [[ \"$ZIP_NAME\" =~ ^CodexBar-([0-9]+(\\.[0-9]+){1,2}([-.][^.]*)?)\\.zip$ ]]; then\n VERSION=\"${BASH_REMATCH[1]}\"\n else\n echo \"Could not infer version from $ZIP_NAME; set SPARKLE_RELEASE_VERSION.\" >&2\n exit 1\n fi\nfi\n\nNOTES_HTML=\"${ZIP_DIR}/${ZIP_BASE}.html\"\nKEEP_NOTES=${KEEP_SPARKLE_NOTES:-0}\nif [[ -x \"$ROOT/Scripts/changelog-to-html.sh\" ]]; then\n \"$ROOT/Scripts/changelog-to-html.sh\" \"$VERSION\" >\"$NOTES_HTML\"\nelse\n echo \"Missing Scripts/changelog-to-html.sh; cannot generate HTML release notes.\" >&2\n exit 1\nfi\ncleanup() {\n if [[ -n \"${WORK_DIR:-}\" ]]; then\n rm -rf \"$WORK_DIR\"\n fi\n if [[ \"$KEEP_NOTES\" != \"1\" ]]; then\n rm -f \"$NOTES_HTML\"\n fi\n}\ntrap cleanup EXIT\n\nDOWNLOAD_URL_PREFIX=${SPARKLE_DOWNLOAD_URL_PREFIX:-\"https://github.com/steipete/CodexBar/releases/download/v${VERSION}/\"}\n\n# Sparkle provides generate_appcast; ensure it's on PATH (via SwiftPM build of Sparkle's bin) or Xcode dmg\nif ! command -v generate_appcast >/dev/null; then\n echo \"generate_appcast not found in PATH. Install Sparkle tools (see Sparkle docs).\" >&2\n exit 1\nfi\n\nWORK_DIR=$(mktemp -d /tmp/codexbar-appcast.XXXXXX)\n\ncp \"$ROOT/appcast.xml\" \"$WORK_DIR/appcast.xml\"\ncp \"$ZIP\" \"$WORK_DIR/$ZIP_NAME\"\ncp \"$NOTES_HTML\" \"$WORK_DIR/$ZIP_BASE.html\"\n\npushd \"$WORK_DIR\" >/dev/null\ngenerate_appcast \\\n --ed-key-file \"$PRIVATE_KEY_FILE\" \\\n --download-url-prefix \"$DOWNLOAD_URL_PREFIX\" \\\n --embed-release-notes \\\n --link \"$FEED_URL\" \\\n \"$WORK_DIR\"\npopd >/dev/null\n\nif [[ -n \"$SPARKLE_CHANNEL\" ]]; then\n python3 - \"$WORK_DIR/appcast.xml\" \"$VERSION\" \"$SPARKLE_CHANNEL\" <<'PY'\nimport re\nimport sys\n\npath, version, channel = sys.argv[1], sys.argv[2], sys.argv[3]\nwith open(path, \"r\", encoding=\"utf-8\") as handle:\n lines = handle.read().splitlines()\n\ntarget = f\"{version}\"\ntry:\n index = next(i for i, line in enumerate(lines) if target in line)\nexcept StopIteration as exc:\n raise SystemExit(f\"Could not find {target} in {path}\") from exc\n\nfor j in range(index, -1, -1):\n if \" for version {version} in {path}\")\n\nwith open(path, \"w\", encoding=\"utf-8\") as handle:\n handle.write(\"\\n\".join(lines) + \"\\n\")\nPY\n echo \"Tagged ${VERSION} with sparkle:channel=\\\"${SPARKLE_CHANNEL}\\\"\"\nfi\n\ncp \"$WORK_DIR/appcast.xml\" \"$ROOT/appcast.xml\"\n\necho \"Appcast generated (appcast.xml). Upload alongside $ZIP at $FEED_URL\"\n" }, { "path": "Scripts/package_app.sh", "content": "#!/usr/bin/env bash\nset -euo pipefail\nCONF=${1:-release}\nALLOW_LLDB=${CODEXBAR_ALLOW_LLDB:-0}\nSIGNING_MODE=${CODEXBAR_SIGNING:-}\nROOT=$(cd \"$(dirname \"$0\")/..\" && pwd)\ncd \"$ROOT\"\n\n# Load version info\nsource \"$ROOT/version.env\"\n\n# Clean build only when explicitly requested (slower).\nif [[ \"${CODEXBAR_FORCE_CLEAN:-0}\" == \"1\" ]]; then\n if [[ -d \"$ROOT/.build\" ]]; then\n if command -v trash >/dev/null 2>&1; then\n if ! trash \"$ROOT/.build\"; then\n echo \"WARN: trash .build failed; continuing with swift package clean.\" >&2\n fi\n else\n rm -rf \"$ROOT/.build\" || echo \"WARN: rm -rf .build failed; continuing with swift package clean.\" >&2\n fi\n fi\n swift package clean >/dev/null 2>&1 || true\nfi\n\n# Build for host architecture by default; allow overriding via ARCHES (e.g., \"arm64 x86_64\" for universal).\nARCH_LIST=( ${ARCHES:-} )\nif [[ ${#ARCH_LIST[@]} -eq 0 ]]; then\n HOST_ARCH=$(uname -m)\n case \"$HOST_ARCH\" in\n arm64) ARCH_LIST=(arm64) ;;\n x86_64) ARCH_LIST=(x86_64) ;;\n *) ARCH_LIST=(\"$HOST_ARCH\") ;;\n esac\nfi\n\npatch_keyboard_shortcuts() {\n local util_path=\"$ROOT/.build/checkouts/KeyboardShortcuts/Sources/KeyboardShortcuts/Utilities.swift\"\n if [[ ! -f \"$util_path\" ]]; then\n return 0\n fi\n if grep -q \"keyboardShortcutsSafeBundle\" \"$util_path\"; then\n return 0\n fi\n\n chmod +w \"$util_path\" || true\n python3 - \"$util_path\" <<'PY'\nimport sys\nfrom pathlib import Path\n\npath = Path(sys.argv[1])\ntext = path.read_text()\nif \".keyboardShortcutsSafeBundle\" in text:\n sys.exit(0)\n\ntext = text.replace(\n 'NSLocalizedString(self, bundle: .module, comment: self)',\n 'NSLocalizedString(self, bundle: .keyboardShortcutsSafeBundle, comment: self)',\n)\n\ninject = \"\"\"\nprivate extension Bundle {\n /// Safe lookup that avoids the fatal trap in the autogenerated `Bundle.module`\n /// when the resource bundle is not placed at the bundle root.\n static let keyboardShortcutsSafeBundle: Bundle = {\n #if os(macOS)\n if let url = Bundle.main.url(forResource: \"KeyboardShortcuts_KeyboardShortcuts\", withExtension: \"bundle\"),\n let bundle = Bundle(url: url) {\n return bundle\n }\n\n let rootURL = Bundle.main.bundleURL.appendingPathComponent(\"KeyboardShortcuts_KeyboardShortcuts.bundle\")\n if let bundle = Bundle(url: rootURL) {\n return bundle\n }\n #endif\n\n let devURL = URL(fileURLWithPath: #file)\n .deletingLastPathComponent() // Utilities.swift\n .deletingLastPathComponent() // KeyboardShortcuts\n .deletingLastPathComponent() // Sources\n .appendingPathComponent(\"KeyboardShortcuts_KeyboardShortcuts.bundle\")\n if let bundle = Bundle(url: devURL) {\n return bundle\n }\n\n return Bundle.main\n }()\n}\n\"\"\"\n\nmarker = \"}\\n\\n\\nextension Data {\"\nif marker not in text:\n raise SystemExit(\"Marker not found in Utilities.swift; patch failed.\")\n\ntext = text.replace(marker, \"}\\n\\n\" + inject + \"\\n\\nextension Data {\")\npath.write_text(text)\nPY\n}\n\nKEYBOARD_SHORTCUTS_UTIL=\"$ROOT/.build/checkouts/KeyboardShortcuts/Sources/KeyboardShortcuts/Utilities.swift\"\nif [[ ! -f \"$KEYBOARD_SHORTCUTS_UTIL\" ]]; then\n swift build -c \"$CONF\" --arch \"${ARCH_LIST[0]}\"\nfi\npatch_keyboard_shortcuts\n\nfor ARCH in \"${ARCH_LIST[@]}\"; do\n swift build -c \"$CONF\" --arch \"$ARCH\"\ndone\n\nAPP=\"$ROOT/CodexBar.app\"\nrm -rf \"$APP\"\nmkdir -p \"$APP/Contents/MacOS\" \"$APP/Contents/Resources\" \"$APP/Contents/Frameworks\"\nmkdir -p \"$APP/Contents/Helpers\" \"$APP/Contents/PlugIns\"\n\n# Convert new .icon bundle to .icns if present (macOS 14+/IconStudio export)\nICON_SOURCE=\"$ROOT/Icon.icon\"\nICON_TARGET=\"$ROOT/Icon.icns\"\nif [[ -f \"$ICON_SOURCE\" ]]; then\n iconutil --convert icns --output \"$ICON_TARGET\" \"$ICON_SOURCE\"\nfi\n\nBUNDLE_ID=\"com.steipete.codexbar\"\nFEED_URL=\"https://raw.githubusercontent.com/steipete/CodexBar/main/appcast.xml\"\nAUTO_CHECKS=true\nLOWER_CONF=$(printf \"%s\" \"$CONF\" | tr '[:upper:]' '[:lower:]')\nif [[ \"$LOWER_CONF\" == \"debug\" ]]; then\n BUNDLE_ID=\"com.steipete.codexbar.debug\"\n FEED_URL=\"\"\n AUTO_CHECKS=false\nfi\nif [[ \"$SIGNING_MODE\" == \"adhoc\" ]]; then\n FEED_URL=\"\"\n AUTO_CHECKS=false\nfi\nWIDGET_BUNDLE_ID=\"${BUNDLE_ID}.widget\"\nAPP_GROUP_ID=\"group.com.steipete.codexbar\"\nif [[ \"$BUNDLE_ID\" == *\".debug\"* ]]; then\n APP_GROUP_ID=\"group.com.steipete.codexbar.debug\"\nfi\nENTITLEMENTS_DIR=\"$ROOT/.build/entitlements\"\nAPP_ENTITLEMENTS=\"${ENTITLEMENTS_DIR}/CodexBar.entitlements\"\nWIDGET_ENTITLEMENTS=\"${ENTITLEMENTS_DIR}/CodexBarWidget.entitlements\"\nmkdir -p \"$ENTITLEMENTS_DIR\"\nif [[ \"$ALLOW_LLDB\" == \"1\" && \"$LOWER_CONF\" != \"debug\" ]]; then\n echo \"ERROR: CODEXBAR_ALLOW_LLDB requires debug configuration\" >&2\n exit 1\nfi\ncat > \"$APP_ENTITLEMENTS\" <\n\n\n\n com.apple.security.application-groups\n \n ${APP_GROUP_ID}\n \n $(if [[ \"$ALLOW_LLDB\" == \"1\" ]]; then echo \" com.apple.security.get-task-allow\"; fi)\n\n\nPLIST\ncat > \"$WIDGET_ENTITLEMENTS\" <\n\n\n\n com.apple.security.app-sandbox\n \n com.apple.security.application-groups\n \n ${APP_GROUP_ID}\n \n\n\nPLIST\nBUILD_TIMESTAMP=$(date -u +\"%Y-%m-%dT%H:%M:%SZ\")\nGIT_COMMIT=$(git rev-parse --short HEAD 2>/dev/null || echo \"unknown\")\n\ncat > \"$APP/Contents/Info.plist\" <\n\n\n\n CFBundleNameCodexBar\n CFBundleDisplayNameCodexBar\n CFBundleIdentifier${BUNDLE_ID}\n CFBundleExecutableCodexBar\n CFBundlePackageTypeAPPL\n CFBundleShortVersionString${MARKETING_VERSION}\n CFBundleVersion${BUILD_NUMBER}\n LSMinimumSystemVersion14.0\n LSUIElement\n CFBundleIconFileIcon\n NSHumanReadableCopyright© 2025 Peter Steinberger. MIT License.\n SUFeedURL${FEED_URL}\n SUPublicEDKeyAGCY8w5vHirVfGGDGc8Szc5iuOqupZSh9pMj/Qs67XI=\n SUEnableAutomaticChecks<${AUTO_CHECKS}/>\n CodexBuildTimestamp${BUILD_TIMESTAMP}\n CodexGitCommit${GIT_COMMIT}\n\n\nPLIST\n\nbuild_product_path() {\n local name=\"$1\"\n local arch=\"$2\"\n case \"$arch\" in\n arm64|x86_64) echo \".build/${arch}-apple-macosx/$CONF/$name\" ;;\n *) echo \".build/$CONF/$name\" ;;\n esac\n}\n\n# Resolve path to built binary; some SwiftPM versions use .build/$CONF/ when building for host only.\nresolve_binary_path() {\n local name=\"$1\"\n local arch=\"$2\"\n local candidate\n candidate=$(build_product_path \"$name\" \"$arch\")\n if [[ -f \"$candidate\" ]]; then\n echo \"$candidate\"\n return\n fi\n if [[ \"$arch\" == \"arm64\" || \"$arch\" == \"x86_64\" ]] && [[ -f \".build/$CONF/$name\" ]]; then\n echo \".build/$CONF/$name\"\n fi\n}\n\nverify_binary_arches() {\n local binary=\"$1\"; shift\n local expected=(\"$@\")\n local actual\n actual=$(lipo -archs \"$binary\")\n local actual_count expected_count\n actual_count=$(wc -w <<<\"$actual\" | tr -d ' ')\n expected_count=${#expected[@]}\n if [[ \"$actual_count\" -ne \"$expected_count\" ]]; then\n echo \"ERROR: $binary arch mismatch (expected: ${expected[*]}, actual: ${actual})\" >&2\n exit 1\n fi\n for arch in \"${expected[@]}\"; do\n if [[ \"$actual\" != *\"$arch\"* ]]; then\n echo \"ERROR: $binary missing arch $arch (have: ${actual})\" >&2\n exit 1\n fi\n done\n}\n\ninstall_binary() {\n local name=\"$1\"\n local dest=\"$2\"\n local binaries=()\n for arch in \"${ARCH_LIST[@]}\"; do\n local src\n src=$(resolve_binary_path \"$name\" \"$arch\")\n if [[ -z \"$src\" || ! -f \"$src\" ]]; then\n echo \"ERROR: Missing ${name} build for ${arch} at $(build_product_path \"$name\" \"$arch\")\" >&2\n exit 1\n fi\n binaries+=(\"$src\")\n done\n if [[ ${#ARCH_LIST[@]} -gt 1 ]]; then\n lipo -create \"${binaries[@]}\" -output \"$dest\"\n else\n cp \"${binaries[0]}\" \"$dest\"\n fi\n chmod +x \"$dest\"\n verify_binary_arches \"$dest\" \"${ARCH_LIST[@]}\"\n}\n\ninstall_binary \"CodexBar\" \"$APP/Contents/MacOS/CodexBar\"\n# Ship CodexBarCLI alongside the app for easy symlinking.\nif [[ -n \"$(resolve_binary_path \"CodexBarCLI\" \"${ARCH_LIST[0]}\")\" ]]; then\n install_binary \"CodexBarCLI\" \"$APP/Contents/Helpers/CodexBarCLI\"\nfi\n# Watchdog helper: ensures `claude` probes die when CodexBar crashes/gets killed.\nif [[ -n \"$(resolve_binary_path \"CodexBarClaudeWatchdog\" \"${ARCH_LIST[0]}\")\" ]]; then\n install_binary \"CodexBarClaudeWatchdog\" \"$APP/Contents/Helpers/CodexBarClaudeWatchdog\"\nfi\nif [[ -n \"$(resolve_binary_path \"CodexBarWidget\" \"${ARCH_LIST[0]}\")\" ]]; then\n WIDGET_APP=\"$APP/Contents/PlugIns/CodexBarWidget.appex\"\n mkdir -p \"$WIDGET_APP/Contents/MacOS\" \"$WIDGET_APP/Contents/Resources\"\n cat > \"$WIDGET_APP/Contents/Info.plist\" <\n\n\n\n CFBundleNameCodexBarWidget\n CFBundleDisplayNameCodexBar\n CFBundleIdentifier${WIDGET_BUNDLE_ID}\n CFBundleExecutableCodexBarWidget\n CFBundlePackageTypeXPC!\n CFBundleShortVersionString${MARKETING_VERSION}\n CFBundleVersion${BUILD_NUMBER}\n LSMinimumSystemVersion14.0\n NSExtension\n \n NSExtensionPointIdentifiercom.apple.widgetkit-extension\n NSExtensionPrincipalClassCodexBarWidget.CodexBarWidgetBundle\n \n\n\nPLIST\n install_binary \"CodexBarWidget\" \"$WIDGET_APP/Contents/MacOS/CodexBarWidget\"\nfi\n# Embed Sparkle.framework\nif [[ -d \".build/$CONF/Sparkle.framework\" ]]; then\n cp -R \".build/$CONF/Sparkle.framework\" \"$APP/Contents/Frameworks/\"\n chmod -R a+rX \"$APP/Contents/Frameworks/Sparkle.framework\"\n install_name_tool -add_rpath \"@executable_path/../Frameworks\" \"$APP/Contents/MacOS/CodexBar\"\n # Re-sign Sparkle and all nested components with Developer ID + timestamp\n SPARKLE=\"$APP/Contents/Frameworks/Sparkle.framework\"\nif [[ \"$SIGNING_MODE\" == \"adhoc\" ]]; then\n CODESIGN_ID=\"-\"\n CODESIGN_ARGS=(--force --sign \"$CODESIGN_ID\")\nelif [[ \"$ALLOW_LLDB\" == \"1\" ]]; then\n CODESIGN_ID=\"-\"\n CODESIGN_ARGS=(--force --sign \"$CODESIGN_ID\")\nelse\n CODESIGN_ID=\"${APP_IDENTITY:-Developer ID Application: Peter Steinberger (Y5PE65HELJ)}\"\n CODESIGN_ARGS=(--force --timestamp --options runtime --sign \"$CODESIGN_ID\")\nfi\nfunction resign() { codesign \"${CODESIGN_ARGS[@]}\" \"$1\"; }\n # Sign innermost binaries first, then the framework root to seal resources\n resign \"$SPARKLE\"\n resign \"$SPARKLE/Versions/B/Sparkle\"\n resign \"$SPARKLE/Versions/B/Autoupdate\"\n resign \"$SPARKLE/Versions/B/Updater.app\"\n resign \"$SPARKLE/Versions/B/Updater.app/Contents/MacOS/Updater\"\n resign \"$SPARKLE/Versions/B/XPCServices/Downloader.xpc\"\n resign \"$SPARKLE/Versions/B/XPCServices/Downloader.xpc/Contents/MacOS/Downloader\"\n resign \"$SPARKLE/Versions/B/XPCServices/Installer.xpc\"\n resign \"$SPARKLE/Versions/B/XPCServices/Installer.xpc/Contents/MacOS/Installer\"\n resign \"$SPARKLE/Versions/B\"\n resign \"$SPARKLE\"\nfi\n\nif [[ -f \"$ICON_TARGET\" ]]; then\n cp \"$ICON_TARGET\" \"$APP/Contents/Resources/Icon.icns\"\nfi\n\n# Bundle app resources (provider icons, etc.).\nAPP_RESOURCES_DIR=\"$ROOT/Sources/CodexBar/Resources\"\nif [[ -d \"$APP_RESOURCES_DIR\" ]]; then\n cp -R \"$APP_RESOURCES_DIR/.\" \"$APP/Contents/Resources/\"\nfi\nif [[ ! -f \"$APP/Contents/Resources/Icon-classic.icns\" ]]; then\n echo \"ERROR: Missing Icon-classic.icns in app bundle resources.\" >&2\n exit 1\nfi\n\n# SwiftPM resource bundles (e.g. KeyboardShortcuts) are emitted next to the built binary.\nCODEXBAR_BINARY=\"$(resolve_binary_path \"CodexBar\" \"${ARCH_LIST[0]}\")\"\nPREFERRED_BUILD_DIR=\"$(dirname \"${CODEXBAR_BINARY:-$(build_product_path \"CodexBar\" \"${ARCH_LIST[0]}\")}\")\"\nshopt -s nullglob\nSWIFTPM_BUNDLES=(\"${PREFERRED_BUILD_DIR}/\"*.bundle)\nshopt -u nullglob\nif [[ ${#SWIFTPM_BUNDLES[@]} -gt 0 ]]; then\n for bundle in \"${SWIFTPM_BUNDLES[@]}\"; do\n bundle_name=\"$(basename \"$bundle\")\"\n cp -R \"$bundle\" \"$APP/Contents/Resources/\"\n done\nfi\nif [[ ! -d \"$APP/Contents/Resources/KeyboardShortcuts_KeyboardShortcuts.bundle\" ]]; then\n echo \"ERROR: Missing KeyboardShortcuts SwiftPM resource bundle (Settings → Keyboard shortcut will crash).\" >&2\n echo \"Expected: ${PREFERRED_BUILD_DIR}/KeyboardShortcuts_KeyboardShortcuts.bundle\" >&2\n exit 1\nfi\n\n# Ensure contents are writable before stripping attributes and signing.\nchmod -R u+w \"$APP\"\n\n# Strip extended attributes to prevent AppleDouble (._*) files that break code sealing\nxattr -cr \"$APP\"\nfind \"$APP\" -name '._*' -delete\n\n# Sign helper binaries if present\nif [[ -f \"${APP}/Contents/Helpers/CodexBarCLI\" ]]; then\n codesign \"${CODESIGN_ARGS[@]}\" \"${APP}/Contents/Helpers/CodexBarCLI\"\nfi\nif [[ -f \"${APP}/Contents/Helpers/CodexBarClaudeWatchdog\" ]]; then\n codesign \"${CODESIGN_ARGS[@]}\" \"${APP}/Contents/Helpers/CodexBarClaudeWatchdog\"\nfi\n\n# Sign widget extension if present\nif [[ -d \"${APP}/Contents/PlugIns/CodexBarWidget.appex\" ]]; then\n codesign \"${CODESIGN_ARGS[@]}\" \\\n --entitlements \"$WIDGET_ENTITLEMENTS\" \\\n \"$APP/Contents/PlugIns/CodexBarWidget.appex/Contents/MacOS/CodexBarWidget\"\n codesign \"${CODESIGN_ARGS[@]}\" \\\n --entitlements \"$WIDGET_ENTITLEMENTS\" \\\n \"$APP/Contents/PlugIns/CodexBarWidget.appex\"\nfi\n\n# Finally sign the app bundle itself\ncodesign \"${CODESIGN_ARGS[@]}\" \\\n --entitlements \"$APP_ENTITLEMENTS\" \\\n \"$APP\"\n\necho \"Created $APP\"\n" }, { "path": "Scripts/prepare_upstream_pr.sh", "content": "#!/bin/bash\n# Prepare a clean branch for upstream PR submission\n# Usage: ./Scripts/prepare_upstream_pr.sh \n\nset -e\n\nFEATURE_NAME=$1\n\n# Colors\nRED='\\033[0;31m'\nGREEN='\\033[0;32m'\nYELLOW='\\033[1;33m'\nBLUE='\\033[0;34m'\nNC='\\033[0m'\n\nif [ -z \"$FEATURE_NAME\" ]; then\n echo -e \"${RED}Error: Feature name required${NC}\"\n echo \"Usage: ./Scripts/prepare_upstream_pr.sh \"\n echo \"\"\n echo \"Examples:\"\n echo \" ./Scripts/prepare_upstream_pr.sh fix-cursor-bonus\"\n echo \" ./Scripts/prepare_upstream_pr.sh improve-cookie-handling\"\n exit 1\nfi\n\nBRANCH_NAME=\"upstream-pr/$FEATURE_NAME\"\n\necho -e \"${BLUE}==> Fetching latest upstream...${NC}\"\ngit fetch upstream\n\necho -e \"${BLUE}==> Creating upstream PR branch from upstream/main...${NC}\"\ngit checkout upstream/main\ngit checkout -b \"$BRANCH_NAME\"\n\necho \"\"\necho -e \"${GREEN}==> Branch created: $BRANCH_NAME${NC}\"\necho \"\"\necho -e \"${YELLOW}⚠️ IMPORTANT: This branch is for UPSTREAM submission${NC}\"\necho \"\"\necho -e \"${BLUE}Guidelines for upstream PRs:${NC}\"\necho \"\"\necho \"✅ DO include:\"\necho \" - Bug fixes that affect all users\"\necho \" - Performance improvements\"\necho \" - Provider enhancements (generic)\"\necho \" - Documentation improvements\"\necho \" - Test coverage\"\necho \"\"\necho \"❌ DO NOT include:\"\necho \" - Fork branding (About.swift, PreferencesAboutPane.swift)\"\necho \" - Fork-specific features (multi-account, etc.)\"\necho \" - References to topoffunnel.com\"\necho \" - Experimental features\"\necho \"\"\necho -e \"${BLUE}Next steps:${NC}\"\necho \"\"\necho \"1. Cherry-pick your commits (clean, no fork branding):\"\necho \" ${GREEN}git cherry-pick ${NC}\"\necho \"\"\necho \"2. Or manually apply changes:\"\necho \" ${GREEN}# Edit files${NC}\"\necho \" ${GREEN}git add ${NC}\"\necho \" ${GREEN}git commit -m 'fix: description'${NC}\"\necho \"\"\necho \"3. Ensure tests pass:\"\necho \" ${GREEN}swift test${NC}\"\necho \"\"\necho \"4. Review changes:\"\necho \" ${GREEN}git diff upstream/main${NC}\"\necho \"\"\necho \"5. Push to your fork:\"\necho \" ${GREEN}git push origin $BRANCH_NAME${NC}\"\necho \"\"\necho \"6. Create PR on GitHub:\"\necho \" ${GREEN}https://github.com/steipete/CodexBar/compare/main...topoffunnel:$BRANCH_NAME${NC}\"\necho \"\"\necho -e \"${YELLOW}Remember: Keep PRs small and focused for better merge chances!${NC}\"\n\n" }, { "path": "Scripts/release.sh", "content": "#!/usr/bin/env bash\nset -euo pipefail\n\nROOT=$(cd \"$(dirname \"$0\")/..\" && pwd)\ncd \"$ROOT\"\n\nsource \"$ROOT/version.env\"\nsource \"$HOME/Projects/agent-scripts/release/sparkle_lib.sh\"\n\nAPPCAST=\"$ROOT/appcast.xml\"\nAPP_NAME=\"CodexBar\"\nARTIFACT_PREFIX=\"CodexBar-\"\nBUNDLE_ID=\"com.steipete.codexbar\"\nTAG=\"v${MARKETING_VERSION}\"\n\nerr() { echo \"ERROR: $*\" >&2; exit 1; }\n\nrequire_clean_worktree\nensure_changelog_finalized \"$MARKETING_VERSION\"\nensure_appcast_monotonic \"$APPCAST\" \"$MARKETING_VERSION\" \"$BUILD_NUMBER\"\n\nswiftformat Sources Tests >/dev/null\nswiftlint --strict\nswift test\n\n# Note: run this script in the foreground; do not background it so it waits to completion.\n\"$ROOT/Scripts/sign-and-notarize.sh\"\n\nKEY_FILE=$(clean_key \"$SPARKLE_PRIVATE_KEY_FILE\")\ntrap 'rm -f \"$KEY_FILE\"' EXIT\n\nprobe_sparkle_key \"$KEY_FILE\"\n\nclear_sparkle_caches \"$BUNDLE_ID\"\n\nNOTES_FILE=$(mktemp /tmp/codexbar-notes.XXXXXX.md)\nextract_notes_from_changelog \"$MARKETING_VERSION\" \"$NOTES_FILE\"\ntrap 'rm -f \"$KEY_FILE\" \"$NOTES_FILE\"' EXIT\n\ngit tag -s -f -m \"${APP_NAME} ${MARKETING_VERSION}\" \"$TAG\"\ngit push -f origin \"$TAG\"\n\ngh release create \"$TAG\" ${APP_NAME}-${MARKETING_VERSION}.zip ${APP_NAME}-${MARKETING_VERSION}.dSYM.zip \\\n --title \"${APP_NAME} ${MARKETING_VERSION}\" \\\n --notes-file \"$NOTES_FILE\"\n\nSPARKLE_PRIVATE_KEY_FILE=\"$KEY_FILE\" \\\n \"$ROOT/Scripts/make_appcast.sh\" \\\n \"${APP_NAME}-${MARKETING_VERSION}.zip\" \\\n \"https://raw.githubusercontent.com/steipete/CodexBar/main/appcast.xml\"\n\nverify_appcast_entry \"$APPCAST\" \"$MARKETING_VERSION\" \"$KEY_FILE\"\n\ngit add \"$APPCAST\"\ngit commit -m \"docs: update appcast for ${MARKETING_VERSION}\"\ngit push origin main\n\nif [[ \"${RUN_SPARKLE_UPDATE_TEST:-0}\" == \"1\" ]]; then\n PREV_TAG=$(git tag --sort=-v:refname | sed -n '2p')\n [[ -z \"$PREV_TAG\" ]] && err \"RUN_SPARKLE_UPDATE_TEST=1 set but no previous tag found\"\n \"$ROOT/Scripts/test_live_update.sh\" \"$PREV_TAG\" \"v${MARKETING_VERSION}\"\nfi\n\ncheck_assets \"$TAG\" \"$ARTIFACT_PREFIX\"\n\ngit push origin --tags\n\necho \"Release ${MARKETING_VERSION} complete.\"\n" }, { "path": "Scripts/review_upstream.sh", "content": "#!/bin/bash\n# Create a review branch for upstream changes\n# Usage: ./Scripts/review_upstream.sh [upstream|quotio]\n\nset -e\n\nUPSTREAM=${1:-upstream}\nDATE=$(date +%Y%m%d)\nBRANCH_NAME=\"upstream-sync/${UPSTREAM}-${DATE}\"\n\n# Colors\nRED='\\033[0;31m'\nGREEN='\\033[0;32m'\nYELLOW='\\033[1;33m'\nBLUE='\\033[0;34m'\nNC='\\033[0m'\n\nif [ \"$UPSTREAM\" != \"upstream\" ] && [ \"$UPSTREAM\" != \"quotio\" ]; then\n echo -e \"${RED}Error: Must specify 'upstream' or 'quotio'${NC}\"\n echo \"Usage: ./Scripts/review_upstream.sh [upstream|quotio]\"\n exit 1\nfi\n\necho -e \"${BLUE}==> Creating review branch for $UPSTREAM...${NC}\"\ngit checkout main\ngit checkout -b \"$BRANCH_NAME\"\n\necho -e \"${BLUE}==> Fetching latest from $UPSTREAM...${NC}\"\ngit fetch \"$UPSTREAM\"\n\necho \"\"\necho -e \"${GREEN}==> Commits to review:${NC}\"\ngit log --oneline --graph main..\"$UPSTREAM\"/main | head -30\n\necho \"\"\necho -e \"${GREEN}==> File changes summary:${NC}\"\ngit diff --stat main..\"$UPSTREAM\"/main\n\necho \"\"\necho -e \"${YELLOW}==> Review branch created: $BRANCH_NAME${NC}\"\necho \"\"\necho -e \"${BLUE}Next steps:${NC}\"\necho \"\"\necho \"1. Review commits in detail:\"\necho \" ${GREEN}git log -p main..$UPSTREAM/main${NC}\"\necho \"\"\necho \"2. View specific files:\"\necho \" ${GREEN}git show $UPSTREAM/main:path/to/file${NC}\"\necho \"\"\necho \"3. Cherry-pick specific commits:\"\necho \" ${GREEN}git cherry-pick ${NC}\"\necho \"\"\necho \"4. Or merge all changes:\"\necho \" ${GREEN}git merge $UPSTREAM/main${NC}\"\necho \"\"\necho \"5. Test thoroughly:\"\necho \" ${GREEN}./Scripts/compile_and_run.sh${NC}\"\necho \"\"\necho \"6. If satisfied, merge to main:\"\necho \" ${GREEN}git checkout main && git merge $BRANCH_NAME${NC}\"\necho \"\"\necho \"7. Or discard review branch:\"\necho \" ${GREEN}git checkout main && git branch -D $BRANCH_NAME${NC}\"\necho \"\"\n\n# Create a review log file\nLOG_FILE=\"upstream-review-${UPSTREAM}-${DATE}.txt\"\necho \"=== Upstream Review: $UPSTREAM @ $DATE ===\" > \"$LOG_FILE\"\necho \"\" >> \"$LOG_FILE\"\necho \"Commits:\" >> \"$LOG_FILE\"\ngit log --oneline main..\"$UPSTREAM\"/main >> \"$LOG_FILE\"\necho \"\" >> \"$LOG_FILE\"\necho \"File changes:\" >> \"$LOG_FILE\"\ngit diff --stat main..\"$UPSTREAM\"/main >> \"$LOG_FILE\"\n\necho -e \"${GREEN}Review log saved to: $LOG_FILE${NC}\"\n\n" }, { "path": "Scripts/setup_dev_signing.sh", "content": "#!/usr/bin/env bash\n# Setup stable development code signing to reduce keychain prompts\nset -euo pipefail\n\necho \"🔐 Setting up stable development code signing...\"\necho \"\"\necho \"This will create a self-signed certificate that stays consistent across rebuilds,\"\necho \"reducing keychain permission prompts.\"\necho \"\"\n\n# Check if we already have a CodexBar development certificate\nCERT_NAME=\"CodexBar Development\"\nif security find-certificate -c \"$CERT_NAME\" >/dev/null 2>&1; then\n echo \"✅ Certificate '$CERT_NAME' already exists!\"\n echo \"\"\n echo \"To use it, add this to your shell profile (~/.zshrc or ~/.bashrc):\"\n echo \"\"\n echo \" export APP_IDENTITY='$CERT_NAME'\"\n echo \"\"\n echo \"Then restart your terminal and rebuild with ./Scripts/compile_and_run.sh\"\n exit 0\nfi\n\necho \"Creating self-signed certificate '$CERT_NAME'...\"\necho \"\"\n\n# Create a temporary config file for the certificate\nTEMP_CONFIG=$(mktemp)\ntrap \"rm -f $TEMP_CONFIG\" EXIT\n\ncat > \"$TEMP_CONFIG\" </dev/null\n\n# Convert to PKCS12 format\nopenssl pkcs12 -export -out /tmp/codexbar-dev.p12 \\\n -inkey /tmp/codexbar-dev.key -in /tmp/codexbar-dev.crt \\\n -passout pass: 2>/dev/null\n\n# Import into keychain\nsecurity import /tmp/codexbar-dev.p12 -k ~/Library/Keychains/login.keychain-db -T /usr/bin/codesign -T /usr/bin/security\n\n# Clean up temporary files\nrm -f /tmp/codexbar-dev.{key,crt,p12}\n\necho \"\"\necho \"✅ Certificate created successfully!\"\necho \"\"\necho \"⚠️ IMPORTANT: You need to trust this certificate for code signing:\"\necho \"\"\necho \"1. Open Keychain Access.app\"\necho \"2. Find '$CERT_NAME' in the 'login' keychain\"\necho \"3. Double-click it\"\necho \"4. Expand 'Trust' section\"\necho \"5. Set 'Code Signing' to 'Always Trust'\"\necho \"6. Close the window (enter your password when prompted)\"\necho \"\"\necho \"Then add this to your shell profile (~/.zshrc or ~/.bashrc):\"\necho \"\"\necho \" export APP_IDENTITY='$CERT_NAME'\"\necho \"\"\necho \"Restart your terminal and rebuild with ./Scripts/compile_and_run.sh\"\n\n" }, { "path": "Scripts/sign-and-notarize.sh", "content": "#!/usr/bin/env bash\nset -euo pipefail\n\nAPP_NAME=\"CodexBar\"\nAPP_IDENTITY=\"Developer ID Application: Peter Steinberger (Y5PE65HELJ)\"\nAPP_BUNDLE=\"CodexBar.app\"\nROOT=$(cd \"$(dirname \"$0\")/..\" && pwd)\nsource \"$ROOT/version.env\"\nZIP_NAME=\"${APP_NAME}-${MARKETING_VERSION}.zip\"\nDSYM_ZIP=\"${APP_NAME}-${MARKETING_VERSION}.dSYM.zip\"\n\nif [[ -z \"${APP_STORE_CONNECT_API_KEY_P8:-}\" || -z \"${APP_STORE_CONNECT_KEY_ID:-}\" || -z \"${APP_STORE_CONNECT_ISSUER_ID:-}\" ]]; then\n echo \"Missing APP_STORE_CONNECT_* env vars (API key, key id, issuer id).\" >&2\n exit 1\nfi\nif [[ -z \"${SPARKLE_PRIVATE_KEY_FILE:-}\" ]]; then\n echo \"SPARKLE_PRIVATE_KEY_FILE is required for release signing/verification.\" >&2\n exit 1\nfi\nif [[ ! -f \"$SPARKLE_PRIVATE_KEY_FILE\" ]]; then\n echo \"Sparkle key file not found: $SPARKLE_PRIVATE_KEY_FILE\" >&2\n exit 1\nfi\nkey_lines=$(grep -v '^[[:space:]]*#' \"$SPARKLE_PRIVATE_KEY_FILE\" | sed '/^[[:space:]]*$/d')\nif [[ $(printf \"%s\\n\" \"$key_lines\" | wc -l) -ne 1 ]]; then\n echo \"Sparkle key file must contain exactly one base64 line (no comments/blank lines).\" >&2\n exit 1\nfi\n\necho \"$APP_STORE_CONNECT_API_KEY_P8\" | sed 's/\\\\n/\\n/g' > /tmp/codexbar-api-key.p8\ntrap 'rm -f /tmp/codexbar-api-key.p8 /tmp/${APP_NAME}Notarize.zip' EXIT\n\n# Allow building a universal binary if ARCHES is provided; default to universal (arm64 + x86_64).\nARCHES_VALUE=${ARCHES:-\"arm64 x86_64\"}\nARCH_LIST=( ${ARCHES_VALUE} )\nfor ARCH in \"${ARCH_LIST[@]}\"; do\n swift build -c release --arch \"$ARCH\"\ndone\nARCHES=\"${ARCHES_VALUE}\" ./Scripts/package_app.sh release\n\nENTITLEMENTS_DIR=\"$ROOT/.build/entitlements\"\nAPP_ENTITLEMENTS=\"${ENTITLEMENTS_DIR}/CodexBar.entitlements\"\nWIDGET_ENTITLEMENTS=\"${ENTITLEMENTS_DIR}/CodexBarWidget.entitlements\"\n\necho \"Signing with $APP_IDENTITY\"\nif [[ -f \"$APP_BUNDLE/Contents/Helpers/CodexBarCLI\" ]]; then\n codesign --force --timestamp --options runtime --sign \"$APP_IDENTITY\" \\\n \"$APP_BUNDLE/Contents/Helpers/CodexBarCLI\"\nfi\nif [[ -f \"$APP_BUNDLE/Contents/Helpers/CodexBarClaudeWatchdog\" ]]; then\n codesign --force --timestamp --options runtime --sign \"$APP_IDENTITY\" \\\n \"$APP_BUNDLE/Contents/Helpers/CodexBarClaudeWatchdog\"\nfi\nif [[ -d \"$APP_BUNDLE/Contents/PlugIns/CodexBarWidget.appex\" ]]; then\n codesign --force --timestamp --options runtime --sign \"$APP_IDENTITY\" \\\n --entitlements \"$WIDGET_ENTITLEMENTS\" \\\n \"$APP_BUNDLE/Contents/PlugIns/CodexBarWidget.appex/Contents/MacOS/CodexBarWidget\"\n codesign --force --timestamp --options runtime --sign \"$APP_IDENTITY\" \\\n --entitlements \"$WIDGET_ENTITLEMENTS\" \\\n \"$APP_BUNDLE/Contents/PlugIns/CodexBarWidget.appex\"\nfi\ncodesign --force --timestamp --options runtime --sign \"$APP_IDENTITY\" \\\n --entitlements \"$APP_ENTITLEMENTS\" \\\n \"$APP_BUNDLE\"\n\nDITTO_BIN=${DITTO_BIN:-/usr/bin/ditto}\n\"$DITTO_BIN\" --norsrc -c -k --keepParent \"$APP_BUNDLE\" \"/tmp/${APP_NAME}Notarize.zip\"\n\necho \"Submitting for notarization\"\nxcrun notarytool submit \"/tmp/${APP_NAME}Notarize.zip\" \\\n --key /tmp/codexbar-api-key.p8 \\\n --key-id \"$APP_STORE_CONNECT_KEY_ID\" \\\n --issuer \"$APP_STORE_CONNECT_ISSUER_ID\" \\\n --wait\n\necho \"Stapling ticket\"\nxcrun stapler staple \"$APP_BUNDLE\"\n\n# Strip any extended attributes that would create AppleDouble files when zipping\nxattr -cr \"$APP_BUNDLE\"\nfind \"$APP_BUNDLE\" -name '._*' -delete\n\n\"$DITTO_BIN\" --norsrc -c -k --keepParent \"$APP_BUNDLE\" \"$ZIP_NAME\"\n\nspctl -a -t exec -vv \"$APP_BUNDLE\"\nstapler validate \"$APP_BUNDLE\"\n\necho \"Packaging dSYM\"\nFIRST_ARCH=\"${ARCH_LIST[0]}\"\nPREFERRED_ARCH_DIR=\".build/${FIRST_ARCH}-apple-macosx/release\"\nDSYM_PATH=\"${PREFERRED_ARCH_DIR}/${APP_NAME}.dSYM\"\nif [[ ! -d \"$DSYM_PATH\" ]]; then\n echo \"Missing dSYM at $DSYM_PATH\" >&2\n exit 1\nfi\nif [[ ${#ARCH_LIST[@]} -gt 1 ]]; then\n MERGED_DSYM=\"${PREFERRED_ARCH_DIR}/${APP_NAME}.dSYM-universal\"\n rm -rf \"$MERGED_DSYM\"\n cp -R \"$DSYM_PATH\" \"$MERGED_DSYM\"\n DWARF_PATH=\"${MERGED_DSYM}/Contents/Resources/DWARF/${APP_NAME}\"\n BINARIES=()\n for ARCH in \"${ARCH_LIST[@]}\"; do\n ARCH_DSYM=\".build/${ARCH}-apple-macosx/release/${APP_NAME}.dSYM/Contents/Resources/DWARF/${APP_NAME}\"\n if [[ ! -f \"$ARCH_DSYM\" ]]; then\n echo \"Missing dSYM for ${ARCH} at $ARCH_DSYM\" >&2\n exit 1\n fi\n BINARIES+=(\"$ARCH_DSYM\")\n done\n lipo -create \"${BINARIES[@]}\" -output \"$DWARF_PATH\"\n DSYM_PATH=\"$MERGED_DSYM\"\nfi\n\"$DITTO_BIN\" --norsrc -c -k --keepParent \"$DSYM_PATH\" \"$DSYM_ZIP\"\n\necho \"Done: $ZIP_NAME\"\n" }, { "path": "Scripts/test_live_update.sh", "content": "#!/usr/bin/env bash\nset -euo pipefail\n\nPREV_TAG=${1:?\"pass previous release tag (e.g. v0.1.0)\"}\nCUR_TAG=${2:?\"pass current release tag (e.g. v0.1.1)\"}\n\nROOT=$(cd \"$(dirname \"$0\")/..\" && pwd)\nPREV_VER=${PREV_TAG#v}\nAPP_NAME=\"CodexBar\"\n\nZIP_URL=\"https://github.com/steipete/CodexBar/releases/download/${PREV_TAG}/${APP_NAME}-${PREV_VER}.zip\"\nTMP_DIR=$(mktemp -d /tmp/codexbar-live.XXXX)\ntrap 'rm -rf \"$TMP_DIR\"' EXIT\n\necho \"Downloading previous release $PREV_TAG from $ZIP_URL\"\ncurl -L -o \"$TMP_DIR/prev.zip\" \"$ZIP_URL\"\n\necho \"Installing previous release to /Applications/${APP_NAME}.app\"\nrm -rf /Applications/${APP_NAME}.app\nditto -x -k \"$TMP_DIR/prev.zip\" \"$TMP_DIR\"\nditto \"$TMP_DIR/${APP_NAME}.app\" /Applications/${APP_NAME}.app\n\necho \"Launching previous build…\"\nopen -n /Applications/${APP_NAME}.app\nsleep 4\n\ncat <<'MSG'\nManual step: trigger \"Check for Updates…\" in the app and install the update.\nExpect to land on the newly released version. When done, confirm below.\nMSG\n\nread -rp \"Did the update succeed from ${PREV_TAG} to ${CUR_TAG}? (y/N) \" answer\nif [[ ! \"$answer\" =~ ^[Yy]$ ]]; then\n echo \"Live update test NOT confirmed; failing per RUN_SPARKLE_UPDATE_TEST.\" >&2\n exit 1\nfi\n\necho \"Live update test confirmed.\"\n" }, { "path": "Scripts/validate_changelog.sh", "content": "#!/usr/bin/env bash\nset -euo pipefail\nVERSION=${1:?\"usage: $0 \"}\nROOT=$(cd \"$(dirname \"$0\")/..\" && pwd)\ncd \"$ROOT\"\n\nfirst_line=$(grep -m1 '^## ' CHANGELOG.md | sed 's/^## //')\nif [[ \"$first_line\" != ${VERSION}* ]]; then\n echo \"ERROR: Top CHANGELOG section is '$first_line' but expected '${VERSION} — …'\" >&2\n exit 1\nfi\n\ngrep -q \"^## ${VERSION} \" CHANGELOG.md || {\n echo \"ERROR: No section for version ${VERSION} in CHANGELOG.md\" >&2\n exit 1\n}\n\ngrep -q '^## [0-9]\\+\\.[0-9]\\+\\.[0-9].*Unreleased' CHANGELOG.md && {\n echo \"ERROR: Top section still labeled Unreleased; finalize changelog first.\" >&2\n exit 1\n}\n\necho \"Changelog OK for ${VERSION}\"\n" }, { "path": "Scripts/verify_appcast.sh", "content": "#!/usr/bin/env bash\nset -euo pipefail\n\n# Verifies that the appcast entry for the given version has a valid ed25519 signature\n# and that the enclosure length matches the downloaded archive.\n#\n# Usage: SPARKLE_PRIVATE_KEY_FILE=/path/to/key ./Scripts/verify_appcast.sh [version]\n\nROOT=$(cd \"$(dirname \"$0\")/..\" && pwd)\nVERSION=${1:-$(source \"$ROOT/version.env\" && echo \"$MARKETING_VERSION\")}\nAPPCAST=\"${ROOT}/appcast.xml\"\n\nif [[ -z \"${SPARKLE_PRIVATE_KEY_FILE:-}\" ]]; then\n echo \"SPARKLE_PRIVATE_KEY_FILE is required\" >&2\n exit 1\nfi\nif [[ ! -f \"$SPARKLE_PRIVATE_KEY_FILE\" ]]; then\n echo \"Sparkle key file not found: $SPARKLE_PRIVATE_KEY_FILE\" >&2\n exit 1\nfi\nif [[ ! -f \"$APPCAST\" ]]; then\n echo \"appcast.xml not found at $APPCAST\" >&2\n exit 1\nfi\n\n# Clean the key file: strip comments/blank lines and require exactly one line of base64.\nfunction cleaned_key_path() {\n local tmp key_lines\n key_lines=$(grep -v '^[[:space:]]*#' \"$SPARKLE_PRIVATE_KEY_FILE\" | sed '/^[[:space:]]*$/d')\n if [[ $(printf \"%s\\n\" \"$key_lines\" | wc -l) -ne 1 ]]; then\n echo \"Sparkle key file must contain exactly one base64 line (no comments/blank lines).\" >&2\n exit 1\n fi\n tmp=$(mktemp)\n printf \"%s\" \"$key_lines\" > \"$tmp\"\n echo \"$tmp\"\n}\n\nKEY_FILE=$(cleaned_key_path)\ntrap 'rm -f \"$KEY_FILE\" \"$TMP_ZIP\"' EXIT\n\nTMP_ZIP=$(mktemp /tmp/codexbar-enclosure.XXXX.zip)\n\npython3 - \"$APPCAST\" \"$VERSION\" >\"$TMP_ZIP.meta\" <<'PY'\nimport sys, xml.etree.ElementTree as ET\n\nappcast = sys.argv[1]\nversion = sys.argv[2]\ntree = ET.parse(appcast)\nroot = tree.getroot()\nns = {\"sparkle\": \"http://www.andymatuschak.org/xml-namespaces/sparkle\"}\n\nentry = None\nfor item in root.findall(\"./channel/item\"):\n sv = item.findtext(\"sparkle:shortVersionString\", default=\"\", namespaces=ns)\n if sv == version:\n entry = item\n break\n\nif entry is None:\n sys.exit(\"No appcast entry found for version {}\".format(version))\n\nenclosure = entry.find(\"enclosure\")\nurl = enclosure.get(\"url\")\nsig = enclosure.get(\"{http://www.andymatuschak.org/xml-namespaces/sparkle}edSignature\")\nlength = enclosure.get(\"length\")\n\nif not all([url, sig, length]):\n sys.exit(\"Missing url/signature/length in appcast for version {}\".format(version))\n\nprint(url)\nprint(sig)\nprint(length)\nPY\n\nreadarray -t META <\"$TMP_ZIP.meta\"\nURL=\"${META[0]}\"\nSIG=\"${META[1]}\"\nLEN_EXPECTED=\"${META[2]}\"\n\necho \"Downloading enclosure: $URL\"\ncurl -L -o \"$TMP_ZIP\" \"$URL\"\n\nLEN_ACTUAL=$(stat -f%z \"$TMP_ZIP\")\nif [[ \"$LEN_ACTUAL\" != \"$LEN_EXPECTED\" ]]; then\n echo \"Length mismatch: expected $LEN_EXPECTED, got $LEN_ACTUAL\" >&2\n exit 1\nfi\n\necho \"Verifying Sparkle signature…\"\nsign_update --verify \"$TMP_ZIP\" \"$SIG\" --ed-key-file \"$KEY_FILE\"\necho \"Appcast entry for $VERSION verified (signature and length match).\"\n" }, { "path": "Sources/CodexBar/About.swift", "content": "import AppKit\n\n@MainActor\nfunc showAbout() {\n NSApp.activate(ignoringOtherApps: true)\n\n let version = Bundle.main.object(forInfoDictionaryKey: \"CFBundleShortVersionString\") as? String ?? \"–\"\n let build = Bundle.main.object(forInfoDictionaryKey: \"CFBundleVersion\") as? String ?? \"\"\n let versionString = build.isEmpty ? version : \"\\(version) (\\(build))\"\n let buildTimestamp = Bundle.main.object(forInfoDictionaryKey: \"CodexBuildTimestamp\") as? String\n let gitCommit = Bundle.main.object(forInfoDictionaryKey: \"CodexGitCommit\") as? String\n\n let separator = NSAttributedString(string: \" · \", attributes: [\n .font: NSFont.systemFont(ofSize: NSFont.smallSystemFontSize),\n ])\n\n func makeLink(_ title: String, urlString: String) -> NSAttributedString {\n NSAttributedString(string: title, attributes: [\n .link: URL(string: urlString) as Any,\n .font: NSFont.systemFont(ofSize: NSFont.smallSystemFontSize),\n ])\n }\n\n let credits = NSMutableAttributedString(string: \"Peter Steinberger — MIT License\\n\")\n credits.append(makeLink(\"GitHub\", urlString: \"https://github.com/steipete/CodexBar\"))\n credits.append(separator)\n credits.append(makeLink(\"Website\", urlString: \"https://codexbar.app\"))\n credits.append(separator)\n credits.append(makeLink(\"Twitter\", urlString: \"https://twitter.com/steipete\"))\n credits.append(separator)\n credits.append(makeLink(\"Email\", urlString: \"mailto:peter@steipete.me\"))\n if let buildTimestamp, let formatted = formattedBuildTimestamp(buildTimestamp) {\n var builtLine = \"Built \\(formatted)\"\n if let gitCommit, !gitCommit.isEmpty, gitCommit != \"unknown\" {\n builtLine += \" (\\(gitCommit)\"\n #if DEBUG\n builtLine += \" DEBUG BUILD\"\n #endif\n builtLine += \")\"\n }\n credits.append(NSAttributedString(string: \"\\n\\(builtLine)\", attributes: [\n .font: NSFont.systemFont(ofSize: NSFont.smallSystemFontSize),\n .foregroundColor: NSColor.secondaryLabelColor,\n ]))\n }\n\n let options: [NSApplication.AboutPanelOptionKey: Any] = [\n .applicationName: \"CodexBar\",\n .applicationVersion: versionString,\n .version: versionString,\n .credits: credits,\n .applicationIcon: (NSApplication.shared.applicationIconImage ?? NSImage()) as Any,\n ]\n\n NSApp.orderFrontStandardAboutPanel(options: options)\n\n // Remove the focus ring around the app icon in the standard About panel for a cleaner look.\n if let aboutPanel = NSApp.windows.first(where: { $0.className.contains(\"About\") }) {\n removeFocusRings(in: aboutPanel.contentView)\n }\n}\n\nprivate func formattedBuildTimestamp(_ timestamp: String) -> String? {\n let parser = ISO8601DateFormatter()\n parser.formatOptions = [.withInternetDateTime]\n guard let date = parser.date(from: timestamp) else { return timestamp }\n\n let formatter = DateFormatter()\n formatter.dateStyle = .medium\n formatter.timeStyle = .short\n formatter.locale = .current\n return formatter.string(from: date)\n}\n\n@MainActor\nprivate func removeFocusRings(in view: NSView?) {\n guard let view else { return }\n if let imageView = view as? NSImageView {\n imageView.focusRingType = .none\n }\n for subview in view.subviews {\n removeFocusRings(in: subview)\n }\n}\n" }, { "path": "Sources/CodexBar/AppNotifications.swift", "content": "import CodexBarCore\nimport Foundation\n@preconcurrency import UserNotifications\n\n@MainActor\nfinal class AppNotifications {\n static let shared = AppNotifications()\n\n private let centerProvider: @Sendable () -> UNUserNotificationCenter\n private let logger = CodexBarLog.logger(LogCategories.notifications)\n private var authorizationTask: Task?\n\n init(centerProvider: @escaping @Sendable () -> UNUserNotificationCenter = { UNUserNotificationCenter.current() }) {\n self.centerProvider = centerProvider\n }\n\n func requestAuthorizationOnStartup() {\n guard !Self.isRunningUnderTests else { return }\n _ = self.ensureAuthorizationTask()\n }\n\n func post(idPrefix: String, title: String, body: String, badge: NSNumber? = nil) {\n guard !Self.isRunningUnderTests else { return }\n let center = self.centerProvider()\n let logger = self.logger\n\n Task { @MainActor in\n let granted = await self.ensureAuthorized()\n guard granted else {\n logger.debug(\"not authorized; skipping post\", metadata: [\"prefix\": idPrefix])\n return\n }\n\n let content = UNMutableNotificationContent()\n content.title = title\n content.body = body\n content.sound = .default\n content.badge = badge\n\n let request = UNNotificationRequest(\n identifier: \"codexbar-\\(idPrefix)-\\(UUID().uuidString)\",\n content: content,\n trigger: nil)\n\n logger.info(\"posting\", metadata: [\"prefix\": idPrefix])\n do {\n try await center.add(request)\n } catch {\n let errorText = String(describing: error)\n logger.error(\"failed to post\", metadata: [\"prefix\": idPrefix, \"error\": errorText])\n }\n }\n }\n\n // MARK: - Private\n\n private func ensureAuthorizationTask() -> Task {\n if let authorizationTask { return authorizationTask }\n let task = Task { @MainActor in\n await self.requestAuthorization()\n }\n self.authorizationTask = task\n return task\n }\n\n private func ensureAuthorized() async -> Bool {\n await self.ensureAuthorizationTask().value\n }\n\n private func requestAuthorization() async -> Bool {\n if let existing = await self.notificationAuthorizationStatus() {\n if existing == .authorized || existing == .provisional {\n return true\n }\n if existing == .denied {\n return false\n }\n }\n\n let center = self.centerProvider()\n return await withCheckedContinuation { continuation in\n center.requestAuthorization(options: [.alert, .sound, .badge]) { granted, _ in\n continuation.resume(returning: granted)\n }\n }\n }\n\n private func notificationAuthorizationStatus() async -> UNAuthorizationStatus? {\n let center = self.centerProvider()\n return await withCheckedContinuation { continuation in\n center.getNotificationSettings { settings in\n continuation.resume(returning: settings.authorizationStatus)\n }\n }\n }\n\n private static var isRunningUnderTests: Bool {\n // Swift Testing doesn't always set XCTest env vars, and removing XCTest imports from\n // the test target can make NSClassFromString(\"XCTestCase\") return nil. If we're not\n // running inside an app bundle, treat it as \"tests/headless\" to avoid crashes when\n // accessing UNUserNotificationCenter.\n if Bundle.main.bundleURL.pathExtension != \"app\" { return true }\n let env = ProcessInfo.processInfo.environment\n if env[\"XCTestConfigurationFilePath\"] != nil { return true }\n if env[\"TESTING_LIBRARY_VERSION\"] != nil { return true }\n if env[\"SWIFT_TESTING\"] != nil { return true }\n return NSClassFromString(\"XCTestCase\") != nil\n }\n}\n" }, { "path": "Sources/CodexBar/ClaudeLoginRunner.swift", "content": "import CodexBarCore\nimport Darwin\nimport Foundation\n\nstruct ClaudeLoginRunner {\n enum Phase {\n case requesting\n case waitingBrowser\n }\n\n struct Result {\n enum Outcome {\n case success\n case timedOut\n case failed(status: Int32)\n case missingBinary\n case launchFailed(String)\n }\n\n let outcome: Outcome\n let output: String\n let authLink: String?\n }\n\n static func run(timeout: TimeInterval = 120, onPhaseChange: @escaping @Sendable (Phase) -> Void) async -> Result {\n await Task(priority: .userInitiated) {\n onPhaseChange(.requesting)\n do {\n let runResult = try self.runPTY(timeout: timeout, onPhaseChange: onPhaseChange)\n let link = self.firstLink(in: runResult.output)\n if let link {\n return Result(outcome: .success, output: runResult.output, authLink: link)\n }\n return Result(outcome: .timedOut, output: runResult.output, authLink: nil)\n } catch LoginError.binaryNotFound {\n return Result(outcome: .missingBinary, output: \"\", authLink: nil)\n } catch let LoginError.timedOut(text) {\n return Result(outcome: .timedOut, output: text, authLink: self.firstLink(in: text))\n } catch let LoginError.failed(status, text) {\n return Result(outcome: .failed(status: status), output: text, authLink: self.firstLink(in: text))\n } catch {\n return Result(outcome: .launchFailed(error.localizedDescription), output: \"\", authLink: nil)\n }\n }.value\n }\n\n // MARK: - PTY runner\n\n private enum LoginError: Error {\n case binaryNotFound\n case timedOut(text: String)\n case failed(status: Int32, text: String)\n case launchFailed(String)\n }\n\n private struct PTYRunResult {\n let output: String\n }\n\n private static func runPTY(\n timeout: TimeInterval,\n onPhaseChange: @escaping @Sendable (Phase) -> Void) throws -> PTYRunResult\n {\n let runner = TTYCommandRunner()\n var options = TTYCommandRunner.Options(rows: 50, cols: 160, timeout: timeout)\n options.extraArgs = [\"/login\"]\n options.stopOnURL = false // keep running until CLI confirms\n options.stopOnSubstrings = [\"Successfully logged in\", \"Login successful\", \"Logged in successfully\"]\n options.sendEnterEvery = 1.0\n options.settleAfterStop = 0.35\n do {\n let result = try runner.run(\n binary: \"claude\",\n send: \"\",\n options: options,\n onURLDetected: { onPhaseChange(.waitingBrowser) })\n return PTYRunResult(output: result.text)\n } catch TTYCommandRunner.Error.binaryNotFound {\n throw LoginError.binaryNotFound\n } catch TTYCommandRunner.Error.timedOut {\n throw LoginError.timedOut(text: \"\")\n } catch let TTYCommandRunner.Error.launchFailed(msg) {\n throw LoginError.launchFailed(msg)\n } catch {\n throw LoginError.launchFailed(error.localizedDescription)\n }\n }\n\n private static func firstLink(in text: String) -> String? {\n let pattern = #\"https?://[A-Za-z0-9._~:/?#\\[\\]@!$&'()*+,;=%-]+\"#\n guard let regex = try? NSRegularExpression(pattern: pattern) else { return nil }\n let nsRange = NSRange(text.startIndex..\\\"'\").contains(last)\n {\n url.unicodeScalars.removeLast()\n }\n return url\n }\n}\n" }, { "path": "Sources/CodexBar/CodexLoginRunner.swift", "content": "import CodexBarCore\nimport Darwin\nimport Foundation\n\nstruct CodexLoginRunner {\n struct Result {\n enum Outcome {\n case success\n case timedOut\n case failed(status: Int32)\n case missingBinary\n case launchFailed(String)\n }\n\n let outcome: Outcome\n let output: String\n }\n\n static func run(timeout: TimeInterval = 120) async -> Result {\n await Task(priority: .userInitiated) {\n var env = ProcessInfo.processInfo.environment\n env[\"PATH\"] = PathBuilder.effectivePATH(\n purposes: [.rpc, .tty, .nodeTooling],\n env: env,\n loginPATH: LoginShellPathCache.shared.current)\n\n guard let executable = BinaryLocator.resolveCodexBinary(\n env: env,\n loginPATH: LoginShellPathCache.shared.current)\n else {\n return Result(outcome: .missingBinary, output: \"\")\n }\n\n let process = Process()\n process.executableURL = URL(fileURLWithPath: \"/usr/bin/env\")\n process.arguments = [executable, \"login\"]\n process.environment = env\n\n let stdout = Pipe()\n let stderr = Pipe()\n process.standardOutput = stdout\n process.standardError = stderr\n\n var processGroup: pid_t?\n do {\n try process.run()\n processGroup = self.attachProcessGroup(process)\n } catch {\n return Result(outcome: .launchFailed(error.localizedDescription), output: \"\")\n }\n\n let timedOut = await self.wait(for: process, timeout: timeout)\n if timedOut {\n self.terminate(process, processGroup: processGroup)\n }\n\n let output = await self.combinedOutput(stdout: stdout, stderr: stderr)\n if timedOut {\n return Result(outcome: .timedOut, output: output)\n }\n\n let status = process.terminationStatus\n if status == 0 {\n return Result(outcome: .success, output: output)\n }\n return Result(outcome: .failed(status: status), output: output)\n }.value\n }\n\n private static func wait(for process: Process, timeout: TimeInterval) async -> Bool {\n await withTaskGroup(of: Bool.self) { group -> Bool in\n group.addTask {\n process.waitUntilExit()\n return false\n }\n group.addTask {\n let nanos = UInt64(max(0, timeout) * 1_000_000_000)\n try? await Task.sleep(nanoseconds: nanos)\n return true\n }\n let result = await group.next() ?? false\n group.cancelAll()\n return result\n }\n }\n\n private static func terminate(_ process: Process, processGroup: pid_t?) {\n if let pgid = processGroup {\n kill(-pgid, SIGTERM)\n }\n if process.isRunning {\n process.terminate()\n }\n\n let deadline = Date().addingTimeInterval(2.0)\n while process.isRunning, Date() < deadline {\n usleep(100_000)\n }\n\n if process.isRunning {\n if let pgid = processGroup {\n kill(-pgid, SIGKILL)\n }\n kill(process.processIdentifier, SIGKILL)\n }\n }\n\n private static func attachProcessGroup(_ process: Process) -> pid_t? {\n let pid = process.processIdentifier\n return setpgid(pid, pid) == 0 ? pid : nil\n }\n\n private static func combinedOutput(stdout: Pipe, stderr: Pipe) async -> String {\n async let out = self.readToEnd(stdout)\n async let err = self.readToEnd(stderr)\n let stdoutText = await out\n let stderrText = await err\n\n let merged: String = if !stdoutText.isEmpty, !stderrText.isEmpty {\n [stdoutText, stderrText].joined(separator: \"\\n\")\n } else {\n stdoutText + stderrText\n }\n let trimmed = merged.trimmingCharacters(in: .whitespacesAndNewlines)\n let limited = trimmed.prefix(4000)\n return limited.isEmpty ? \"No output captured.\" : String(limited)\n }\n\n private static func readToEnd(_ pipe: Pipe, timeout: TimeInterval = 3.0) async -> String {\n await withTaskGroup(of: String?.self) { group -> String in\n group.addTask {\n if #available(macOS 13.0, *) {\n if let data = try? pipe.fileHandleForReading.readToEnd() { return self.decode(data) }\n }\n let data = pipe.fileHandleForReading.readDataToEndOfFile()\n return Self.decode(data)\n }\n group.addTask {\n let nanos = UInt64(max(0, timeout) * 1_000_000_000)\n try? await Task.sleep(nanoseconds: nanos)\n return nil\n }\n let result = await group.next()\n group.cancelAll()\n if let result, let text = result { return text }\n return \"\"\n }\n }\n\n private static func decode(_ data: Data) -> String {\n guard let text = String(data: data, encoding: .utf8) else { return \"\" }\n return text\n }\n}\n" }, { "path": "Sources/CodexBar/CodexbarApp.swift", "content": "import AppKit\nimport CodexBarCore\nimport KeyboardShortcuts\nimport Observation\nimport QuartzCore\nimport Security\nimport SwiftUI\n\n@main\nstruct CodexBarApp: App {\n @NSApplicationDelegateAdaptor(AppDelegate.self) private var appDelegate\n @State private var settings: SettingsStore\n @State private var store: UsageStore\n private let preferencesSelection: PreferencesSelection\n private let account: AccountInfo\n\n init() {\n let env = ProcessInfo.processInfo.environment\n let storedLevel = CodexBarLog.parseLevel(UserDefaults.standard.string(forKey: \"debugLogLevel\")) ?? .verbose\n let level = CodexBarLog.parseLevel(env[\"CODEXBAR_LOG_LEVEL\"]) ?? storedLevel\n CodexBarLog.bootstrapIfNeeded(.init(\n destination: .oslog(subsystem: \"com.steipete.codexbar\"),\n level: level,\n json: false))\n\n let version = Bundle.main.object(forInfoDictionaryKey: \"CFBundleShortVersionString\") as? String ?? \"unknown\"\n let build = Bundle.main.object(forInfoDictionaryKey: \"CFBundleVersion\") as? String ?? \"unknown\"\n let gitCommit = Bundle.main.object(forInfoDictionaryKey: \"CodexGitCommit\") as? String ?? \"unknown\"\n let buildTimestamp = Bundle.main.object(forInfoDictionaryKey: \"CodexBuildTimestamp\") as? String ?? \"unknown\"\n CodexBarLog.logger(LogCategories.app).info(\n \"CodexBar starting\",\n metadata: [\n \"version\": version,\n \"build\": build,\n \"git\": gitCommit,\n \"built\": buildTimestamp,\n ])\n\n KeychainAccessGate.isDisabled = UserDefaults.standard.bool(forKey: \"debugDisableKeychainAccess\")\n KeychainPromptCoordinator.install()\n\n let preferencesSelection = PreferencesSelection()\n let settings = SettingsStore()\n let fetcher = UsageFetcher()\n let browserDetection = BrowserDetection(cacheTTL: BrowserDetection.defaultCacheTTL)\n let account = fetcher.loadAccountInfo()\n let store = UsageStore(fetcher: fetcher, browserDetection: browserDetection, settings: settings)\n self.preferencesSelection = preferencesSelection\n _settings = State(wrappedValue: settings)\n _store = State(wrappedValue: store)\n self.account = account\n CodexBarLog.setLogLevel(settings.debugLogLevel)\n self.appDelegate.configure(\n store: store,\n settings: settings,\n account: account,\n selection: preferencesSelection)\n }\n\n @SceneBuilder\n var body: some Scene {\n // Hidden 1×1 window to keep SwiftUI's lifecycle alive so `Settings` scene\n // shows the native toolbar tabs even though the UI is AppKit-based.\n WindowGroup(\"CodexBarLifecycleKeepalive\") {\n HiddenWindowView()\n }\n .defaultSize(width: 20, height: 20)\n .windowStyle(.hiddenTitleBar)\n\n Settings {\n PreferencesView(\n settings: self.settings,\n store: self.store,\n updater: self.appDelegate.updaterController,\n selection: self.preferencesSelection)\n }\n .defaultSize(width: PreferencesTab.general.preferredWidth, height: PreferencesTab.general.preferredHeight)\n .windowResizability(.contentSize)\n }\n\n private func openSettings(tab: PreferencesTab) {\n self.preferencesSelection.tab = tab\n NSApp.activate(ignoringOtherApps: true)\n _ = NSApp.sendAction(Selector((\"showPreferencesWindow:\")), to: nil, from: nil)\n }\n}\n\n// MARK: - Updater abstraction\n\n@MainActor\nprotocol UpdaterProviding: AnyObject {\n var automaticallyChecksForUpdates: Bool { get set }\n var automaticallyDownloadsUpdates: Bool { get set }\n var isAvailable: Bool { get }\n var unavailableReason: String? { get }\n var updateStatus: UpdateStatus { get }\n func checkForUpdates(_ sender: Any?)\n}\n\n/// No-op updater used for debug builds and non-bundled runs to suppress Sparkle dialogs.\nfinal class DisabledUpdaterController: UpdaterProviding {\n var automaticallyChecksForUpdates: Bool = false\n var automaticallyDownloadsUpdates: Bool = false\n let isAvailable: Bool = false\n let unavailableReason: String?\n let updateStatus = UpdateStatus()\n\n init(unavailableReason: String? = nil) {\n self.unavailableReason = unavailableReason\n }\n\n func checkForUpdates(_ sender: Any?) {}\n}\n\n@MainActor\n@Observable\nfinal class UpdateStatus {\n static let disabled = UpdateStatus()\n var isUpdateReady: Bool\n\n init(isUpdateReady: Bool = false) {\n self.isUpdateReady = isUpdateReady\n }\n}\n\n#if canImport(Sparkle) && ENABLE_SPARKLE\nimport Sparkle\n\n@MainActor\nfinal class SparkleUpdaterController: NSObject, UpdaterProviding, SPUUpdaterDelegate {\n private lazy var controller = SPUStandardUpdaterController(\n startingUpdater: false,\n updaterDelegate: self,\n userDriverDelegate: nil)\n let updateStatus = UpdateStatus()\n let unavailableReason: String? = nil\n\n init(savedAutoUpdate: Bool) {\n super.init()\n let updater = self.controller.updater\n updater.automaticallyChecksForUpdates = savedAutoUpdate\n updater.automaticallyDownloadsUpdates = savedAutoUpdate\n self.controller.startUpdater()\n }\n\n var automaticallyChecksForUpdates: Bool {\n get { self.controller.updater.automaticallyChecksForUpdates }\n set { self.controller.updater.automaticallyChecksForUpdates = newValue }\n }\n\n var automaticallyDownloadsUpdates: Bool {\n get { self.controller.updater.automaticallyDownloadsUpdates }\n set { self.controller.updater.automaticallyDownloadsUpdates = newValue }\n }\n\n var isAvailable: Bool {\n true\n }\n\n func checkForUpdates(_ sender: Any?) {\n self.controller.checkForUpdates(sender)\n }\n\n nonisolated func updater(_ updater: SPUUpdater, didDownloadUpdate item: SUAppcastItem) {\n Task { @MainActor in\n self.updateStatus.isUpdateReady = true\n }\n }\n\n nonisolated func updater(_ updater: SPUUpdater, failedToDownloadUpdate item: SUAppcastItem, error: Error) {\n Task { @MainActor in\n self.updateStatus.isUpdateReady = false\n }\n }\n\n nonisolated func userDidCancelDownload(_ updater: SPUUpdater) {\n Task { @MainActor in\n self.updateStatus.isUpdateReady = false\n }\n }\n\n nonisolated func updater(\n _ updater: SPUUpdater,\n userDidMake choice: SPUUserUpdateChoice,\n forUpdate updateItem: SUAppcastItem,\n state: SPUUserUpdateState)\n {\n let downloaded = state.stage == .downloaded\n Task { @MainActor in\n switch choice {\n case .install, .skip:\n self.updateStatus.isUpdateReady = false\n case .dismiss:\n self.updateStatus.isUpdateReady = downloaded\n @unknown default:\n self.updateStatus.isUpdateReady = false\n }\n }\n }\n\n nonisolated func allowedChannels(for updater: SPUUpdater) -> Set {\n UpdateChannel.current.allowedSparkleChannels\n }\n}\n\nprivate func isDeveloperIDSigned(bundleURL: URL) -> Bool {\n var staticCode: SecStaticCode?\n guard SecStaticCodeCreateWithPath(bundleURL as CFURL, SecCSFlags(), &staticCode) == errSecSuccess,\n let code = staticCode else { return false }\n\n var infoCF: CFDictionary?\n guard SecCodeCopySigningInformation(code, SecCSFlags(rawValue: kSecCSSigningInformation), &infoCF) == errSecSuccess,\n let info = infoCF as? [String: Any],\n let certs = info[kSecCodeInfoCertificates as String] as? [SecCertificate],\n let leaf = certs.first else { return false }\n\n if let summary = SecCertificateCopySubjectSummary(leaf) as String? {\n return summary.hasPrefix(\"Developer ID Application:\")\n }\n return false\n}\n\n@MainActor\nprivate func makeUpdaterController() -> UpdaterProviding {\n let bundleURL = Bundle.main.bundleURL\n let isBundledApp = bundleURL.pathExtension == \"app\"\n guard isBundledApp else {\n return DisabledUpdaterController(unavailableReason: \"Updates unavailable in this build.\")\n }\n\n if InstallOrigin.isHomebrewCask(appBundleURL: bundleURL) {\n return DisabledUpdaterController(\n unavailableReason: \"Updates managed by Homebrew. Run: brew upgrade --cask steipete/tap/codexbar\")\n }\n\n guard isDeveloperIDSigned(bundleURL: bundleURL) else {\n return DisabledUpdaterController(unavailableReason: \"Updates unavailable in this build.\")\n }\n\n let defaults = UserDefaults.standard\n let autoUpdateKey = \"autoUpdateEnabled\"\n // Default to true for first launch; fall back to saved preference thereafter.\n let savedAutoUpdate = (defaults.object(forKey: autoUpdateKey) as? Bool) ?? true\n return SparkleUpdaterController(savedAutoUpdate: savedAutoUpdate)\n}\n#else\nprivate func makeUpdaterController() -> UpdaterProviding {\n DisabledUpdaterController()\n}\n#endif\n\n@MainActor\nfinal class AppDelegate: NSObject, NSApplicationDelegate {\n let updaterController: UpdaterProviding = makeUpdaterController()\n private var statusController: StatusItemControlling?\n private var store: UsageStore?\n private var settings: SettingsStore?\n private var account: AccountInfo?\n private var preferencesSelection: PreferencesSelection?\n\n func configure(store: UsageStore, settings: SettingsStore, account: AccountInfo, selection: PreferencesSelection) {\n self.store = store\n self.settings = settings\n self.account = account\n self.preferencesSelection = selection\n }\n\n func applicationWillFinishLaunching(_ notification: Notification) {\n self.configureAppIconForMacOSVersion()\n }\n\n func applicationDidFinishLaunching(_ notification: Notification) {\n AppNotifications.shared.requestAuthorizationOnStartup()\n self.ensureStatusController()\n KeyboardShortcuts.onKeyUp(for: .openMenu) { [weak self] in\n Task { @MainActor [weak self] in\n self?.statusController?.openMenuFromShortcut()\n }\n }\n }\n\n func applicationWillTerminate(_ notification: Notification) {\n TTYCommandRunner.terminateActiveProcessesForAppShutdown()\n }\n\n /// Use the classic (non-Liquid Glass) app icon on macOS versions before 26.\n private func configureAppIconForMacOSVersion() {\n if #unavailable(macOS 26) {\n self.applyClassicAppIcon()\n }\n }\n\n private func applyClassicAppIcon() {\n guard let classicIcon = Self.loadClassicIcon() else { return }\n NSApp.applicationIconImage = classicIcon\n }\n\n private static func loadClassicIcon() -> NSImage? {\n guard let url = self.classicIconURL(),\n let image = NSImage(contentsOf: url)\n else {\n return nil\n }\n return image\n }\n\n private static func classicIconURL() -> URL? {\n Bundle.main.url(forResource: \"Icon-classic\", withExtension: \"icns\")\n }\n\n private func ensureStatusController() {\n if self.statusController != nil { return }\n\n if let store, let settings, let account, let selection = self.preferencesSelection {\n self.statusController = StatusItemController.factory(\n store,\n settings,\n account,\n self.updaterController,\n selection)\n return\n }\n\n // Defensive fallback: this should not be hit in normal app lifecycle.\n CodexBarLog.logger(LogCategories.app)\n .error(\"StatusItemController fallback path used; settings/store mismatch likely.\")\n assertionFailure(\"StatusItemController fallback path used; check app lifecycle wiring.\")\n let fallbackSettings = SettingsStore()\n let fetcher = UsageFetcher()\n let browserDetection = BrowserDetection(cacheTTL: BrowserDetection.defaultCacheTTL)\n let fallbackAccount = fetcher.loadAccountInfo()\n let fallbackStore = UsageStore(fetcher: fetcher, browserDetection: browserDetection, settings: fallbackSettings)\n self.statusController = StatusItemController.factory(\n fallbackStore,\n fallbackSettings,\n fallbackAccount,\n self.updaterController,\n PreferencesSelection())\n }\n}\n" }, { "path": "Sources/CodexBar/Config/CodexBarConfigMigrator.swift", "content": "import CodexBarCore\nimport Foundation\n\nstruct CodexBarConfigMigrator {\n struct LegacyStores {\n let zaiTokenStore: any ZaiTokenStoring\n let syntheticTokenStore: any SyntheticTokenStoring\n let codexCookieStore: any CookieHeaderStoring\n let claudeCookieStore: any CookieHeaderStoring\n let cursorCookieStore: any CookieHeaderStoring\n let opencodeCookieStore: any CookieHeaderStoring\n let factoryCookieStore: any CookieHeaderStoring\n let minimaxCookieStore: any MiniMaxCookieStoring\n let minimaxAPITokenStore: any MiniMaxAPITokenStoring\n let kimiTokenStore: any KimiTokenStoring\n let kimiK2TokenStore: any KimiK2TokenStoring\n let augmentCookieStore: any CookieHeaderStoring\n let ampCookieStore: any CookieHeaderStoring\n let copilotTokenStore: any CopilotTokenStoring\n let tokenAccountStore: any ProviderTokenAccountStoring\n }\n\n private struct MigrationState {\n var didUpdate = false\n var sawLegacySecrets = false\n var sawLegacyAccounts = false\n }\n\n static func loadOrMigrate(\n configStore: CodexBarConfigStore,\n userDefaults: UserDefaults,\n stores: LegacyStores) -> CodexBarConfig\n {\n let log = CodexBarLog.logger(LogCategories.configMigration)\n let existing = try? configStore.load()\n var config = (existing ?? CodexBarConfig.makeDefault()).normalized()\n var state = MigrationState()\n\n if existing == nil {\n self.applyLegacyOrderAndToggles(userDefaults: userDefaults, config: &config, state: &state)\n }\n\n self.applyLegacyCookieSources(userDefaults: userDefaults, config: &config, state: &state)\n self.migrateLegacySecrets(userDefaults: userDefaults, stores: stores, config: &config, state: &state)\n self.migrateLegacyAccounts(stores: stores, config: &config, state: &state)\n\n if state.didUpdate {\n do {\n try configStore.save(config)\n } catch {\n log.error(\"Failed to persist config: \\(error)\")\n }\n }\n\n if state.sawLegacySecrets || state.sawLegacyAccounts {\n self.clearLegacyStores(stores: stores, sawAccounts: state.sawLegacyAccounts, log: log)\n }\n\n return config.normalized()\n }\n\n private static func applyLegacyOrderAndToggles(\n userDefaults: UserDefaults,\n config: inout CodexBarConfig,\n state: inout MigrationState)\n {\n if let order = userDefaults.stringArray(forKey: \"providerOrder\"), !order.isEmpty {\n config = self.applyProviderOrder(order, config: config)\n state.didUpdate = true\n }\n let toggles = userDefaults.dictionary(forKey: \"providerToggles\") as? [String: Bool] ?? [:]\n if !toggles.isEmpty {\n config = self.applyProviderToggles(toggles, config: config)\n state.didUpdate = true\n }\n }\n\n private static func migrateLegacySecrets(\n userDefaults: UserDefaults,\n stores: LegacyStores,\n config: inout CodexBarConfig,\n state: inout MigrationState)\n {\n self.migrateTokenProviders(\n [\n (.zai, stores.zaiTokenStore.loadToken),\n (.synthetic, stores.syntheticTokenStore.loadToken),\n (.copilot, stores.copilotTokenStore.loadToken),\n (.kimik2, stores.kimiK2TokenStore.loadToken),\n ],\n config: &config,\n state: &state)\n\n self.migrateCookieProviders(\n [\n (.codex, stores.codexCookieStore.loadCookieHeader),\n (.claude, stores.claudeCookieStore.loadCookieHeader),\n (.cursor, stores.cursorCookieStore.loadCookieHeader),\n (.factory, stores.factoryCookieStore.loadCookieHeader),\n (.augment, stores.augmentCookieStore.loadCookieHeader),\n (.amp, stores.ampCookieStore.loadCookieHeader),\n ],\n config: &config,\n state: &state)\n\n self.migrateMiniMax(userDefaults: userDefaults, stores: stores, config: &config, state: &state)\n self.migrateKimi(userDefaults: userDefaults, stores: stores, config: &config, state: &state)\n self.migrateOpenCode(userDefaults: userDefaults, stores: stores, config: &config, state: &state)\n }\n\n private static func applyLegacyCookieSources(\n userDefaults: UserDefaults,\n config: inout CodexBarConfig,\n state: inout MigrationState)\n {\n let sources: [(UsageProvider, String)] = [\n (.codex, \"codexCookieSource\"),\n (.claude, \"claudeCookieSource\"),\n (.cursor, \"cursorCookieSource\"),\n (.opencode, \"opencodeCookieSource\"),\n (.factory, \"factoryCookieSource\"),\n (.minimax, \"minimaxCookieSource\"),\n (.kimi, \"kimiCookieSource\"),\n (.augment, \"augmentCookieSource\"),\n (.amp, \"ampCookieSource\"),\n ]\n\n for (provider, key) in sources {\n guard let raw = userDefaults.string(forKey: key),\n let source = ProviderCookieSource(rawValue: raw)\n else { continue }\n self.updateProvider(provider, config: &config, state: &state) { entry in\n guard entry.cookieSource == nil else { return false }\n entry.cookieSource = source\n return true\n }\n }\n\n if userDefaults.object(forKey: \"openAIWebAccessEnabled\") as? Bool == false {\n self.updateProvider(.codex, config: &config, state: &state) { entry in\n guard entry.cookieSource == nil else { return false }\n entry.cookieSource = .off\n return true\n }\n }\n }\n\n private static func migrateTokenProviders(\n _ providers: [(UsageProvider, () throws -> String?)],\n config: inout CodexBarConfig,\n state: inout MigrationState)\n {\n for (provider, loader) in providers {\n let token = try? loader()\n if token != nil { state.sawLegacySecrets = true }\n self.updateProvider(provider, config: &config, state: &state) { entry in\n self.setIfEmpty(&entry.apiKey, token)\n }\n }\n }\n\n private static func migrateCookieProviders(\n _ providers: [(UsageProvider, () throws -> String?)],\n config: inout CodexBarConfig,\n state: inout MigrationState)\n {\n for (provider, loader) in providers {\n let header = try? loader()\n if header != nil { state.sawLegacySecrets = true }\n self.updateProvider(provider, config: &config, state: &state) { entry in\n self.setIfEmpty(&entry.cookieHeader, header)\n }\n }\n }\n\n private static func migrateMiniMax(\n userDefaults: UserDefaults,\n stores: LegacyStores,\n config: inout CodexBarConfig,\n state: inout MigrationState)\n {\n let token = try? stores.minimaxAPITokenStore.loadToken()\n let header = try? stores.minimaxCookieStore.loadCookieHeader()\n if token != nil || header != nil {\n state.sawLegacySecrets = true\n }\n let regionRaw = userDefaults.string(forKey: \"minimaxAPIRegion\")\n self.updateProvider(.minimax, config: &config, state: &state) { entry in\n var changed = false\n changed = self.setIfEmpty(&entry.apiKey, token) || changed\n if let regionRaw, !regionRaw.isEmpty, entry.region == nil {\n entry.region = regionRaw\n changed = true\n }\n changed = self.setIfEmpty(&entry.cookieHeader, header) || changed\n return changed\n }\n }\n\n private static func migrateKimi(\n userDefaults: UserDefaults,\n stores: LegacyStores,\n config: inout CodexBarConfig,\n state: inout MigrationState)\n {\n var token = try? stores.kimiTokenStore.loadToken()\n if token?.isEmpty ?? true {\n token = userDefaults.string(forKey: \"kimiManualCookieHeader\")\n }\n if token != nil { state.sawLegacySecrets = true }\n self.updateProvider(.kimi, config: &config, state: &state) { entry in\n self.setIfEmpty(&entry.cookieHeader, token)\n }\n }\n\n private static func migrateOpenCode(\n userDefaults: UserDefaults,\n stores: LegacyStores,\n config: inout CodexBarConfig,\n state: inout MigrationState)\n {\n let header = try? stores.opencodeCookieStore.loadCookieHeader()\n if header != nil { state.sawLegacySecrets = true }\n let workspaceID = userDefaults.string(forKey: \"opencodeWorkspaceID\")\n self.updateProvider(.opencode, config: &config, state: &state) { entry in\n var changed = false\n changed = self.setIfEmpty(&entry.cookieHeader, header) || changed\n if let workspaceID, !workspaceID.isEmpty, entry.workspaceID == nil {\n entry.workspaceID = workspaceID\n changed = true\n }\n return changed\n }\n }\n\n private static func migrateLegacyAccounts(\n stores: LegacyStores,\n config: inout CodexBarConfig,\n state: inout MigrationState)\n {\n guard let accounts = try? stores.tokenAccountStore.loadAccounts(), !accounts.isEmpty else { return }\n state.sawLegacyAccounts = true\n for (provider, data) in accounts where !data.accounts.isEmpty {\n self.updateProvider(provider, config: &config, state: &state) { entry in\n guard entry.tokenAccounts == nil else { return false }\n entry.tokenAccounts = data\n return true\n }\n }\n }\n\n private static func updateProvider(\n _ provider: UsageProvider,\n config: inout CodexBarConfig,\n state: inout MigrationState,\n mutate: (inout ProviderConfig) -> Bool)\n {\n guard let index = config.providers.firstIndex(where: { $0.id == provider }) else { return }\n var entry = config.providers[index]\n let changed = mutate(&entry)\n if changed {\n config.providers[index] = entry\n state.didUpdate = true\n }\n }\n\n private static func setIfEmpty(_ value: inout String?, _ replacement: String?) -> Bool {\n let cleaned = replacement?.trimmingCharacters(in: .whitespacesAndNewlines)\n guard let cleaned, !cleaned.isEmpty else { return false }\n if value == nil || value?.isEmpty == true {\n value = cleaned\n return true\n }\n return false\n }\n\n private static func clearLegacyStores(\n stores: LegacyStores,\n sawAccounts: Bool,\n log: CodexBarLogger)\n {\n do {\n try stores.zaiTokenStore.storeToken(nil)\n try stores.syntheticTokenStore.storeToken(nil)\n try stores.copilotTokenStore.storeToken(nil)\n try stores.minimaxAPITokenStore.storeToken(nil)\n try stores.kimiTokenStore.storeToken(nil)\n try stores.kimiK2TokenStore.storeToken(nil)\n try stores.codexCookieStore.storeCookieHeader(nil)\n try stores.claudeCookieStore.storeCookieHeader(nil)\n try stores.cursorCookieStore.storeCookieHeader(nil)\n try stores.opencodeCookieStore.storeCookieHeader(nil)\n try stores.factoryCookieStore.storeCookieHeader(nil)\n try stores.minimaxCookieStore.storeCookieHeader(nil)\n try stores.augmentCookieStore.storeCookieHeader(nil)\n try stores.ampCookieStore.storeCookieHeader(nil)\n } catch {\n log.error(\"Failed to clear legacy secrets: \\(error)\")\n }\n\n if sawAccounts {\n let legacyURL = FileTokenAccountStore.defaultURL()\n if FileManager.default.fileExists(atPath: legacyURL.path) {\n try? FileManager.default.removeItem(at: legacyURL)\n }\n }\n }\n\n private static func applyProviderOrder(_ raw: [String], config: CodexBarConfig) -> CodexBarConfig {\n let configsByID = Dictionary(uniqueKeysWithValues: config.providers.map { ($0.id, $0) })\n var seen: Set = []\n var ordered: [ProviderConfig] = []\n ordered.reserveCapacity(config.providers.count)\n\n for rawValue in raw {\n guard let provider = UsageProvider(rawValue: rawValue),\n let entry = configsByID[provider],\n !seen.contains(provider)\n else { continue }\n seen.insert(provider)\n ordered.append(entry)\n }\n\n for provider in UsageProvider.allCases where !seen.contains(provider) {\n ordered.append(configsByID[provider] ?? ProviderConfig(id: provider))\n }\n\n var updated = config\n updated.providers = ordered\n return updated\n }\n\n private static func applyProviderToggles(\n _ toggles: [String: Bool],\n config: CodexBarConfig) -> CodexBarConfig\n {\n var updated = config\n for index in updated.providers.indices {\n let provider = updated.providers[index].id\n let meta = ProviderDescriptorRegistry.descriptor(for: provider).metadata\n if let value = toggles[meta.cliName] {\n updated.providers[index].enabled = value\n }\n }\n return updated\n }\n}\n" }, { "path": "Sources/CodexBar/CookieHeaderStore.swift", "content": "import CodexBarCore\nimport Foundation\nimport Security\n\nprotocol CookieHeaderStoring: Sendable {\n func loadCookieHeader() throws -> String?\n func storeCookieHeader(_ header: String?) throws\n}\n\nenum CookieHeaderStoreError: LocalizedError {\n case keychainStatus(OSStatus)\n case invalidData\n\n var errorDescription: String? {\n switch self {\n case let .keychainStatus(status):\n \"Keychain error: \\(status)\"\n case .invalidData:\n \"Keychain returned invalid data.\"\n }\n }\n}\n\nstruct KeychainCookieHeaderStore: CookieHeaderStoring {\n private static let log = CodexBarLog.logger(LogCategories.cookieHeaderStore)\n\n private let service = \"com.steipete.CodexBar\"\n private let account: String\n private let promptKind: KeychainPromptContext.Kind\n\n // Cache to reduce keychain access frequency\n private nonisolated(unsafe) static var cache: [String: CachedValue] = [:]\n private static let cacheLock = NSLock()\n private static let cacheTTL: TimeInterval = 1800 // 30 minutes\n\n private struct CachedValue {\n let value: String?\n let timestamp: Date\n\n var isExpired: Bool {\n Date().timeIntervalSince(self.timestamp) > KeychainCookieHeaderStore.cacheTTL\n }\n }\n\n init(account: String, promptKind: KeychainPromptContext.Kind) {\n self.account = account\n self.promptKind = promptKind\n }\n\n func loadCookieHeader() throws -> String? {\n guard !KeychainAccessGate.isDisabled else {\n Self.log.debug(\"Keychain access disabled; skipping cookie load\")\n return nil\n }\n // Check cache first\n Self.cacheLock.lock()\n if let cached = Self.cache[self.account], !cached.isExpired {\n Self.cacheLock.unlock()\n Self.log.debug(\"Using cached cookie header for \\(self.account)\")\n return cached.value\n }\n Self.cacheLock.unlock()\n var result: CFTypeRef?\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n kSecMatchLimit as String: kSecMatchLimitOne,\n kSecReturnData as String: true,\n ]\n\n if case .interactionRequired = KeychainAccessPreflight\n .checkGenericPassword(service: self.service, account: self.account)\n {\n KeychainPromptHandler.handler?(KeychainPromptContext(\n kind: self.promptKind,\n service: self.service,\n account: self.account))\n }\n\n let status = SecItemCopyMatching(query as CFDictionary, &result)\n if status == errSecItemNotFound {\n // Cache the nil result\n Self.cacheLock.lock()\n Self.cache[self.account] = CachedValue(value: nil, timestamp: Date())\n Self.cacheLock.unlock()\n return nil\n }\n guard status == errSecSuccess else {\n Self.log.error(\"Keychain read failed: \\(status)\")\n throw CookieHeaderStoreError.keychainStatus(status)\n }\n\n guard let data = result as? Data else {\n throw CookieHeaderStoreError.invalidData\n }\n let header = String(data: data, encoding: .utf8)?.trimmingCharacters(in: .whitespacesAndNewlines)\n let finalValue = (header?.isEmpty == false) ? header : nil\n\n // Cache the result\n Self.cacheLock.lock()\n Self.cache[self.account] = CachedValue(value: finalValue, timestamp: Date())\n Self.cacheLock.unlock()\n\n return finalValue\n }\n\n func storeCookieHeader(_ header: String?) throws {\n guard !KeychainAccessGate.isDisabled else {\n Self.log.debug(\"Keychain access disabled; skipping cookie store\")\n return\n }\n guard let raw = header?.trimmingCharacters(in: .whitespacesAndNewlines),\n !raw.isEmpty\n else {\n try self.deleteIfPresent()\n // Invalidate cache\n Self.cacheLock.lock()\n Self.cache.removeValue(forKey: self.account)\n Self.cacheLock.unlock()\n return\n }\n guard CookieHeaderNormalizer.normalize(raw) != nil else {\n try self.deleteIfPresent()\n // Invalidate cache\n Self.cacheLock.lock()\n Self.cache.removeValue(forKey: self.account)\n Self.cacheLock.unlock()\n return\n }\n\n let data = raw.data(using: .utf8)!\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n ]\n let attributes: [String: Any] = [\n kSecValueData as String: data,\n kSecAttrAccessible as String: kSecAttrAccessibleAfterFirstUnlockThisDeviceOnly,\n ]\n\n let updateStatus = SecItemUpdate(query as CFDictionary, attributes as CFDictionary)\n if updateStatus == errSecSuccess {\n // Update cache\n Self.cacheLock.lock()\n Self.cache[self.account] = CachedValue(value: raw, timestamp: Date())\n Self.cacheLock.unlock()\n return\n }\n if updateStatus != errSecItemNotFound {\n Self.log.error(\"Keychain update failed: \\(updateStatus)\")\n throw CookieHeaderStoreError.keychainStatus(updateStatus)\n }\n\n var addQuery = query\n for (key, value) in attributes {\n addQuery[key] = value\n }\n let addStatus = SecItemAdd(addQuery as CFDictionary, nil)\n guard addStatus == errSecSuccess else {\n Self.log.error(\"Keychain add failed: \\(addStatus)\")\n throw CookieHeaderStoreError.keychainStatus(addStatus)\n }\n\n // Update cache\n Self.cacheLock.lock()\n Self.cache[self.account] = CachedValue(value: raw, timestamp: Date())\n Self.cacheLock.unlock()\n }\n\n private func deleteIfPresent() throws {\n guard !KeychainAccessGate.isDisabled else { return }\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n ]\n let status = SecItemDelete(query as CFDictionary)\n if status == errSecSuccess || status == errSecItemNotFound {\n // Invalidate cache\n Self.cacheLock.lock()\n Self.cache.removeValue(forKey: self.account)\n Self.cacheLock.unlock()\n return\n }\n Self.log.error(\"Keychain delete failed: \\(status)\")\n throw CookieHeaderStoreError.keychainStatus(status)\n }\n}\n" }, { "path": "Sources/CodexBar/CopilotTokenStore.swift", "content": "import CodexBarCore\nimport Foundation\nimport Security\n\nprotocol CopilotTokenStoring: Sendable {\n func loadToken() throws -> String?\n func storeToken(_ token: String?) throws\n}\n\nenum CopilotTokenStoreError: LocalizedError {\n case keychainStatus(OSStatus)\n case invalidData\n\n var errorDescription: String? {\n switch self {\n case let .keychainStatus(status):\n \"Keychain error: \\(status)\"\n case .invalidData:\n \"Keychain returned invalid data.\"\n }\n }\n}\n\nstruct KeychainCopilotTokenStore: CopilotTokenStoring {\n private static let log = CodexBarLog.logger(LogCategories.copilotTokenStore)\n\n private let service = \"com.steipete.CodexBar\"\n private let account = \"copilot-api-token\"\n\n func loadToken() throws -> String? {\n guard !KeychainAccessGate.isDisabled else {\n Self.log.debug(\"Keychain access disabled; skipping token load\")\n return nil\n }\n var result: CFTypeRef?\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n kSecMatchLimit as String: kSecMatchLimitOne,\n kSecReturnData as String: true,\n ]\n\n if case .interactionRequired = KeychainAccessPreflight\n .checkGenericPassword(service: self.service, account: self.account)\n {\n KeychainPromptHandler.handler?(KeychainPromptContext(\n kind: .copilotToken,\n service: self.service,\n account: self.account))\n }\n\n let status = SecItemCopyMatching(query as CFDictionary, &result)\n if status == errSecItemNotFound {\n return nil\n }\n guard status == errSecSuccess else {\n Self.log.error(\"Keychain read failed: \\(status)\")\n throw CopilotTokenStoreError.keychainStatus(status)\n }\n\n guard let data = result as? Data else {\n throw CopilotTokenStoreError.invalidData\n }\n let token = String(data: data, encoding: .utf8)?.trimmingCharacters(in: .whitespacesAndNewlines)\n if let token, !token.isEmpty {\n return token\n }\n return nil\n }\n\n func storeToken(_ token: String?) throws {\n guard !KeychainAccessGate.isDisabled else {\n Self.log.debug(\"Keychain access disabled; skipping token store\")\n return\n }\n let cleaned = token?.trimmingCharacters(in: .whitespacesAndNewlines)\n if cleaned == nil || cleaned?.isEmpty == true {\n try self.deleteTokenIfPresent()\n return\n }\n\n let data = cleaned!.data(using: .utf8)!\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n ]\n let attributes: [String: Any] = [\n kSecValueData as String: data,\n kSecAttrAccessible as String: kSecAttrAccessibleAfterFirstUnlockThisDeviceOnly,\n ]\n\n let updateStatus = SecItemUpdate(query as CFDictionary, attributes as CFDictionary)\n if updateStatus == errSecSuccess {\n return\n }\n if updateStatus != errSecItemNotFound {\n Self.log.error(\"Keychain update failed: \\(updateStatus)\")\n throw CopilotTokenStoreError.keychainStatus(updateStatus)\n }\n\n var addQuery = query\n for (key, value) in attributes {\n addQuery[key] = value\n }\n let addStatus = SecItemAdd(addQuery as CFDictionary, nil)\n guard addStatus == errSecSuccess else {\n Self.log.error(\"Keychain add failed: \\(addStatus)\")\n throw CopilotTokenStoreError.keychainStatus(addStatus)\n }\n }\n\n private func deleteTokenIfPresent() throws {\n guard !KeychainAccessGate.isDisabled else { return }\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n ]\n let status = SecItemDelete(query as CFDictionary)\n if status == errSecSuccess || status == errSecItemNotFound {\n return\n }\n Self.log.error(\"Keychain delete failed: \\(status)\")\n throw CopilotTokenStoreError.keychainStatus(status)\n }\n}\n" }, { "path": "Sources/CodexBar/CostHistoryChartMenuView.swift", "content": "import Charts\nimport CodexBarCore\nimport SwiftUI\n\n@MainActor\nstruct CostHistoryChartMenuView: View {\n typealias DailyEntry = CostUsageDailyReport.Entry\n\n private struct Point: Identifiable {\n let id: String\n let date: Date\n let costUSD: Double\n let totalTokens: Int?\n\n init(date: Date, costUSD: Double, totalTokens: Int?) {\n self.date = date\n self.costUSD = costUSD\n self.totalTokens = totalTokens\n self.id = \"\\(Int(date.timeIntervalSince1970))-\\(costUSD)\"\n }\n }\n\n private let provider: UsageProvider\n private let daily: [DailyEntry]\n private let totalCostUSD: Double?\n private let width: CGFloat\n @State private var selectedDateKey: String?\n\n init(provider: UsageProvider, daily: [DailyEntry], totalCostUSD: Double?, width: CGFloat) {\n self.provider = provider\n self.daily = daily\n self.totalCostUSD = totalCostUSD\n self.width = width\n }\n\n var body: some View {\n let model = Self.makeModel(provider: self.provider, daily: self.daily)\n VStack(alignment: .leading, spacing: 10) {\n if model.points.isEmpty {\n Text(\"No cost history data.\")\n .font(.footnote)\n .foregroundStyle(.secondary)\n } else {\n Chart {\n ForEach(model.points) { point in\n BarMark(\n x: .value(\"Day\", point.date, unit: .day),\n y: .value(\"Cost\", point.costUSD))\n .foregroundStyle(model.barColor)\n }\n if let peak = Self.peakPoint(model: model) {\n let capStart = max(peak.costUSD - Self.capHeight(maxValue: model.maxCostUSD), 0)\n BarMark(\n x: .value(\"Day\", peak.date, unit: .day),\n yStart: .value(\"Cap start\", capStart),\n yEnd: .value(\"Cap end\", peak.costUSD))\n .foregroundStyle(Color(nsColor: .systemYellow))\n }\n }\n .chartYAxis(.hidden)\n .chartXAxis {\n AxisMarks(values: model.axisDates) { _ in\n AxisGridLine().foregroundStyle(Color.clear)\n AxisTick().foregroundStyle(Color.clear)\n AxisValueLabel(format: .dateTime.month(.abbreviated).day())\n .font(.caption2)\n .foregroundStyle(Color(nsColor: .tertiaryLabelColor))\n }\n }\n .chartLegend(.hidden)\n .frame(height: 130)\n .chartOverlay { proxy in\n GeometryReader { geo in\n ZStack(alignment: .topLeading) {\n if let rect = self.selectionBandRect(model: model, proxy: proxy, geo: geo) {\n Rectangle()\n .fill(Self.selectionBandColor)\n .frame(width: rect.width, height: rect.height)\n .position(x: rect.midX, y: rect.midY)\n .allowsHitTesting(false)\n }\n MouseLocationReader { location in\n self.updateSelection(location: location, model: model, proxy: proxy, geo: geo)\n }\n .frame(maxWidth: .infinity, maxHeight: .infinity)\n .contentShape(Rectangle())\n }\n }\n }\n\n let detail = self.detailLines(model: model)\n VStack(alignment: .leading, spacing: 0) {\n Text(detail.primary)\n .font(.caption)\n .foregroundStyle(.secondary)\n .lineLimit(1)\n .truncationMode(.tail)\n .frame(height: 16, alignment: .leading)\n Text(detail.secondary ?? \" \")\n .font(.caption)\n .foregroundStyle(.secondary)\n .lineLimit(1)\n .truncationMode(.tail)\n .frame(height: 16, alignment: .leading)\n .opacity(detail.secondary == nil ? 0 : 1)\n }\n }\n\n if let total = self.totalCostUSD {\n Text(\"Total (30d): \\(UsageFormatter.usdString(total))\")\n .font(.caption)\n .foregroundStyle(.secondary)\n }\n }\n .padding(.horizontal, 16)\n .padding(.vertical, 10)\n .frame(minWidth: self.width, maxWidth: .infinity, alignment: .leading)\n }\n\n private struct Model {\n let points: [Point]\n let pointsByDateKey: [String: Point]\n let entriesByDateKey: [String: DailyEntry]\n let dateKeys: [(key: String, date: Date)]\n let axisDates: [Date]\n let barColor: Color\n let peakKey: String?\n let maxCostUSD: Double\n }\n\n private static let selectionBandColor = Color(nsColor: .labelColor).opacity(0.1)\n\n private static func capHeight(maxValue: Double) -> Double {\n maxValue * 0.05\n }\n\n private static func makeModel(provider: UsageProvider, daily: [DailyEntry]) -> Model {\n let sorted = daily.sorted { lhs, rhs in lhs.date < rhs.date }\n var points: [Point] = []\n points.reserveCapacity(sorted.count)\n\n var pointsByKey: [String: Point] = [:]\n pointsByKey.reserveCapacity(sorted.count)\n\n var entriesByKey: [String: DailyEntry] = [:]\n entriesByKey.reserveCapacity(sorted.count)\n\n var dateKeys: [(key: String, date: Date)] = []\n dateKeys.reserveCapacity(sorted.count)\n\n var peak: (key: String, costUSD: Double)?\n var maxCostUSD: Double = 0\n for entry in sorted {\n guard let costUSD = entry.costUSD, costUSD >= 0 else { continue }\n guard let date = self.dateFromDayKey(entry.date) else { continue }\n let point = Point(date: date, costUSD: costUSD, totalTokens: entry.totalTokens)\n points.append(point)\n pointsByKey[entry.date] = point\n entriesByKey[entry.date] = entry\n dateKeys.append((entry.date, date))\n if let cur = peak {\n if costUSD > cur.costUSD { peak = (entry.date, costUSD) }\n } else {\n peak = (entry.date, costUSD)\n }\n maxCostUSD = max(maxCostUSD, costUSD)\n }\n\n let axisDates: [Date] = {\n guard let first = dateKeys.first?.date, let last = dateKeys.last?.date else { return [] }\n if Calendar.current.isDate(first, inSameDayAs: last) { return [first] }\n return [first, last]\n }()\n\n let barColor = Self.barColor(for: provider)\n return Model(\n points: points,\n pointsByDateKey: pointsByKey,\n entriesByDateKey: entriesByKey,\n dateKeys: dateKeys,\n axisDates: axisDates,\n barColor: barColor,\n peakKey: maxCostUSD > 0 ? peak?.key : nil,\n maxCostUSD: maxCostUSD)\n }\n\n private static func barColor(for provider: UsageProvider) -> Color {\n let color = ProviderDescriptorRegistry.descriptor(for: provider).branding.color\n return Color(red: color.red, green: color.green, blue: color.blue)\n }\n\n private static func dateFromDayKey(_ key: String) -> Date? {\n let parts = key.split(separator: \"-\")\n guard parts.count == 3,\n let year = Int(parts[0]),\n let month = Int(parts[1]),\n let day = Int(parts[2]) else { return nil }\n\n var comps = DateComponents()\n comps.calendar = Calendar.current\n comps.timeZone = TimeZone.current\n comps.year = year\n comps.month = month\n comps.day = day\n comps.hour = 12\n return comps.date\n }\n\n private static func peakPoint(model: Model) -> Point? {\n guard let key = model.peakKey else { return nil }\n return model.pointsByDateKey[key]\n }\n\n private func selectionBandRect(model: Model, proxy: ChartProxy, geo: GeometryProxy) -> CGRect? {\n guard let key = self.selectedDateKey else { return nil }\n guard let plotAnchor = proxy.plotFrame else { return nil }\n let plotFrame = geo[plotAnchor]\n guard let index = model.dateKeys.firstIndex(where: { $0.key == key }) else { return nil }\n let date = model.dateKeys[index].date\n guard let x = proxy.position(forX: date) else { return nil }\n\n func xForIndex(_ idx: Int) -> CGFloat? {\n guard idx >= 0, idx < model.dateKeys.count else { return nil }\n return proxy.position(forX: model.dateKeys[idx].date)\n }\n\n let xPrev = xForIndex(index - 1)\n let xNext = xForIndex(index + 1)\n\n let leftInPlot: CGFloat = if let xPrev {\n (xPrev + x) / 2\n } else if let xNext {\n x - (xNext - x) / 2\n } else {\n x - 8\n }\n\n let rightInPlot: CGFloat = if let xNext {\n (xNext + x) / 2\n } else if let xPrev {\n x + (x - xPrev) / 2\n } else {\n x + 8\n }\n\n let left = plotFrame.origin.x + min(leftInPlot, rightInPlot)\n let right = plotFrame.origin.x + max(leftInPlot, rightInPlot)\n return CGRect(x: left, y: plotFrame.origin.y, width: right - left, height: plotFrame.height)\n }\n\n private func updateSelection(\n location: CGPoint?,\n model: Model,\n proxy: ChartProxy,\n geo: GeometryProxy)\n {\n guard let location else {\n if self.selectedDateKey != nil { self.selectedDateKey = nil }\n return\n }\n\n guard let plotAnchor = proxy.plotFrame else { return }\n let plotFrame = geo[plotAnchor]\n guard plotFrame.contains(location) else { return }\n\n let xInPlot = location.x - plotFrame.origin.x\n guard let date: Date = proxy.value(atX: xInPlot) else { return }\n guard let nearest = self.nearestDateKey(to: date, model: model) else { return }\n\n if self.selectedDateKey != nearest {\n self.selectedDateKey = nearest\n }\n }\n\n private func nearestDateKey(to date: Date, model: Model) -> String? {\n guard !model.dateKeys.isEmpty else { return nil }\n var best: (key: String, distance: TimeInterval)?\n for entry in model.dateKeys {\n let dist = abs(entry.date.timeIntervalSince(date))\n if let cur = best {\n if dist < cur.distance { best = (entry.key, dist) }\n } else {\n best = (entry.key, dist)\n }\n }\n return best?.key\n }\n\n private func detailLines(model: Model) -> (primary: String, secondary: String?) {\n guard let key = self.selectedDateKey,\n let point = model.pointsByDateKey[key],\n let date = Self.dateFromDayKey(key)\n else {\n return (\"Hover a bar for details\", nil)\n }\n\n let dayLabel = date.formatted(.dateTime.month(.abbreviated).day())\n let cost = UsageFormatter.usdString(point.costUSD)\n if let tokens = point.totalTokens {\n let primary = \"\\(dayLabel): \\(cost) · \\(UsageFormatter.tokenCountString(tokens)) tokens\"\n let secondary = self.topModelsText(key: key, model: model)\n return (primary, secondary)\n }\n let primary = \"\\(dayLabel): \\(cost)\"\n let secondary = self.topModelsText(key: key, model: model)\n return (primary, secondary)\n }\n\n private func topModelsText(key: String, model: Model) -> String? {\n guard let entry = model.entriesByDateKey[key] else { return nil }\n guard let breakdown = entry.modelBreakdowns, !breakdown.isEmpty else { return nil }\n let parts = breakdown\n .compactMap { item -> String? in\n let name = UsageFormatter.modelDisplayName(item.modelName)\n guard let detail = UsageFormatter.modelCostDetail(\n item.modelName,\n costUSD: item.costUSD,\n totalTokens: item.totalTokens)\n else { return nil }\n return \"\\(name) \\(detail)\"\n }\n .prefix(3)\n guard !parts.isEmpty else { return nil }\n return \"Top: \\(parts.joined(separator: \" · \"))\"\n }\n}\n" }, { "path": "Sources/CodexBar/CreditsHistoryChartMenuView.swift", "content": "import Charts\nimport CodexBarCore\nimport SwiftUI\n\n@MainActor\nstruct CreditsHistoryChartMenuView: View {\n private struct Point: Identifiable {\n let id: String\n let date: Date\n let creditsUsed: Double\n\n init(date: Date, creditsUsed: Double) {\n self.date = date\n self.creditsUsed = creditsUsed\n self.id = \"\\(Int(date.timeIntervalSince1970))-\\(creditsUsed)\"\n }\n }\n\n private let breakdown: [OpenAIDashboardDailyBreakdown]\n private let width: CGFloat\n @State private var selectedDayKey: String?\n\n init(breakdown: [OpenAIDashboardDailyBreakdown], width: CGFloat) {\n self.breakdown = breakdown\n self.width = width\n }\n\n var body: some View {\n let model = Self.makeModel(from: self.breakdown)\n VStack(alignment: .leading, spacing: 10) {\n if model.points.isEmpty {\n Text(\"No credits history data.\")\n .font(.footnote)\n .foregroundStyle(.secondary)\n } else {\n Chart {\n ForEach(model.points) { point in\n BarMark(\n x: .value(\"Day\", point.date, unit: .day),\n y: .value(\"Credits used\", point.creditsUsed))\n .foregroundStyle(Self.barColor)\n }\n if let peak = Self.peakPoint(model: model) {\n let capStart = max(peak.creditsUsed - Self.capHeight(maxValue: model.maxCreditsUsed), 0)\n BarMark(\n x: .value(\"Day\", peak.date, unit: .day),\n yStart: .value(\"Cap start\", capStart),\n yEnd: .value(\"Cap end\", peak.creditsUsed))\n .foregroundStyle(Color(nsColor: .systemYellow))\n }\n }\n .chartYAxis(.hidden)\n .chartXAxis {\n AxisMarks(values: model.axisDates) { _ in\n AxisGridLine().foregroundStyle(Color.clear)\n AxisTick().foregroundStyle(Color.clear)\n AxisValueLabel(format: .dateTime.month(.abbreviated).day())\n .font(.caption2)\n .foregroundStyle(Color(nsColor: .tertiaryLabelColor))\n }\n }\n .chartLegend(.hidden)\n .frame(height: 130)\n .chartOverlay { proxy in\n GeometryReader { geo in\n ZStack(alignment: .topLeading) {\n if let rect = self.selectionBandRect(model: model, proxy: proxy, geo: geo) {\n Rectangle()\n .fill(Self.selectionBandColor)\n .frame(width: rect.width, height: rect.height)\n .position(x: rect.midX, y: rect.midY)\n .allowsHitTesting(false)\n }\n MouseLocationReader { location in\n self.updateSelection(location: location, model: model, proxy: proxy, geo: geo)\n }\n .frame(maxWidth: .infinity, maxHeight: .infinity)\n .contentShape(Rectangle())\n }\n }\n }\n\n let detail = self.detailLines(model: model)\n VStack(alignment: .leading, spacing: 0) {\n Text(detail.primary)\n .font(.caption)\n .foregroundStyle(.secondary)\n .lineLimit(1)\n .truncationMode(.tail)\n .frame(height: 16, alignment: .leading)\n Text(detail.secondary ?? \" \")\n .font(.caption)\n .foregroundStyle(.secondary)\n .lineLimit(1)\n .truncationMode(.tail)\n .frame(height: 16, alignment: .leading)\n .opacity(detail.secondary == nil ? 0 : 1)\n }\n\n if let total = model.totalCreditsUsed {\n Text(\"Total (30d): \\(total.formatted(.number.precision(.fractionLength(0...2)))) credits\")\n .font(.caption)\n .foregroundStyle(.secondary)\n }\n }\n }\n .padding(.horizontal, 16)\n .padding(.vertical, 10)\n .frame(minWidth: self.width, maxWidth: .infinity, alignment: .leading)\n }\n\n private struct Model {\n let points: [Point]\n let breakdownByDayKey: [String: OpenAIDashboardDailyBreakdown]\n let pointsByDayKey: [String: Point]\n let dayDates: [(dayKey: String, date: Date)]\n let selectableDayDates: [(dayKey: String, date: Date)]\n let axisDates: [Date]\n let peakKey: String?\n let totalCreditsUsed: Double?\n let maxCreditsUsed: Double\n }\n\n private static let barColor = Color(red: 73 / 255, green: 163 / 255, blue: 176 / 255)\n private static let selectionBandColor = Color(nsColor: .labelColor).opacity(0.1)\n private static func capHeight(maxValue: Double) -> Double {\n maxValue * 0.05\n }\n\n private static func makeModel(from breakdown: [OpenAIDashboardDailyBreakdown]) -> Model {\n let sorted = breakdown.sorted { lhs, rhs in lhs.day < rhs.day }\n\n var points: [Point] = []\n points.reserveCapacity(sorted.count)\n\n var breakdownByDayKey: [String: OpenAIDashboardDailyBreakdown] = [:]\n breakdownByDayKey.reserveCapacity(sorted.count)\n\n var pointsByDayKey: [String: Point] = [:]\n pointsByDayKey.reserveCapacity(sorted.count)\n\n var dayDates: [(dayKey: String, date: Date)] = []\n dayDates.reserveCapacity(sorted.count)\n\n var selectableDayDates: [(dayKey: String, date: Date)] = []\n selectableDayDates.reserveCapacity(sorted.count)\n\n var totalCreditsUsed: Double = 0\n var peak: (key: String, creditsUsed: Double)?\n var maxCreditsUsed: Double = 0\n\n for day in sorted {\n guard let date = self.dateFromDayKey(day.day) else { continue }\n breakdownByDayKey[day.day] = day\n dayDates.append((dayKey: day.day, date: date))\n totalCreditsUsed += day.totalCreditsUsed\n if day.totalCreditsUsed > 0 {\n let point = Point(date: date, creditsUsed: day.totalCreditsUsed)\n points.append(point)\n pointsByDayKey[day.day] = point\n selectableDayDates.append((dayKey: day.day, date: date))\n if let cur = peak {\n if day.totalCreditsUsed > cur.creditsUsed { peak = (day.day, day.totalCreditsUsed) }\n } else {\n peak = (day.day, day.totalCreditsUsed)\n }\n maxCreditsUsed = max(maxCreditsUsed, day.totalCreditsUsed)\n }\n }\n\n let axisDates: [Date] = {\n guard let first = dayDates.first?.date, let last = dayDates.last?.date else { return [] }\n if Calendar.current.isDate(first, inSameDayAs: last) { return [first] }\n return [first, last]\n }()\n\n return Model(\n points: points,\n breakdownByDayKey: breakdownByDayKey,\n pointsByDayKey: pointsByDayKey,\n dayDates: dayDates,\n selectableDayDates: selectableDayDates,\n axisDates: axisDates,\n peakKey: peak?.key,\n totalCreditsUsed: totalCreditsUsed > 0 ? totalCreditsUsed : nil,\n maxCreditsUsed: maxCreditsUsed)\n }\n\n private static func dateFromDayKey(_ key: String) -> Date? {\n let parts = key.split(separator: \"-\")\n guard parts.count == 3,\n let year = Int(parts[0]),\n let month = Int(parts[1]),\n let day = Int(parts[2])\n else {\n return nil\n }\n\n var comps = DateComponents()\n comps.calendar = Calendar.current\n comps.timeZone = TimeZone.current\n comps.year = year\n comps.month = month\n comps.day = day\n comps.hour = 12\n return comps.date\n }\n\n private static func peakPoint(model: Model) -> Point? {\n guard let key = model.peakKey else { return nil }\n return model.pointsByDayKey[key]\n }\n\n private func selectionBandRect(model: Model, proxy: ChartProxy, geo: GeometryProxy) -> CGRect? {\n guard let key = self.selectedDayKey else { return nil }\n guard let plotAnchor = proxy.plotFrame else { return nil }\n let plotFrame = geo[plotAnchor]\n guard let index = model.dayDates.firstIndex(where: { $0.dayKey == key }) else { return nil }\n let date = model.dayDates[index].date\n guard let x = proxy.position(forX: date) else { return nil }\n\n func xForIndex(_ idx: Int) -> CGFloat? {\n guard idx >= 0, idx < model.dayDates.count else { return nil }\n return proxy.position(forX: model.dayDates[idx].date)\n }\n\n let xPrev = xForIndex(index - 1)\n let xNext = xForIndex(index + 1)\n\n if model.dayDates.count <= 1 {\n return CGRect(\n x: plotFrame.origin.x,\n y: plotFrame.origin.y,\n width: plotFrame.width,\n height: plotFrame.height)\n }\n\n let leftInPlot: CGFloat = if let xPrev {\n (xPrev + x) / 2\n } else if let xNext {\n x - (xNext - x) / 2\n } else {\n x - 8\n }\n\n let rightInPlot: CGFloat = if let xNext {\n (xNext + x) / 2\n } else if let xPrev {\n x + (x - xPrev) / 2\n } else {\n x + 8\n }\n\n let left = plotFrame.origin.x + min(leftInPlot, rightInPlot)\n let right = plotFrame.origin.x + max(leftInPlot, rightInPlot)\n return CGRect(x: left, y: plotFrame.origin.y, width: right - left, height: plotFrame.height)\n }\n\n private func updateSelection(\n location: CGPoint?,\n model: Model,\n proxy: ChartProxy,\n geo: GeometryProxy)\n {\n guard let location else {\n if self.selectedDayKey != nil { self.selectedDayKey = nil }\n return\n }\n\n guard let plotAnchor = proxy.plotFrame else { return }\n let plotFrame = geo[plotAnchor]\n guard plotFrame.contains(location) else { return }\n\n let xInPlot = location.x - plotFrame.origin.x\n guard let date: Date = proxy.value(atX: xInPlot) else { return }\n guard let nearest = self.nearestDayKey(to: date, model: model) else { return }\n\n if self.selectedDayKey != nearest {\n self.selectedDayKey = nearest\n }\n }\n\n private func nearestDayKey(to date: Date, model: Model) -> String? {\n guard !model.selectableDayDates.isEmpty else { return nil }\n var best: (key: String, distance: TimeInterval)?\n for entry in model.selectableDayDates {\n let dist = abs(entry.date.timeIntervalSince(date))\n if let cur = best {\n if dist < cur.distance { best = (entry.dayKey, dist) }\n } else {\n best = (entry.dayKey, dist)\n }\n }\n return best?.key\n }\n\n private func detailLines(model: Model) -> (primary: String, secondary: String?) {\n guard let key = self.selectedDayKey,\n let day = model.breakdownByDayKey[key],\n let date = Self.dateFromDayKey(key)\n else {\n return (\"Hover a bar for details\", nil)\n }\n\n let dayLabel = date.formatted(.dateTime.month(.abbreviated).day())\n let total = day.totalCreditsUsed.formatted(.number.precision(.fractionLength(0...2)))\n if day.services.isEmpty {\n return (\"\\(dayLabel): \\(total) credits\", nil)\n }\n if day.services.count <= 1, let first = day.services.first {\n let used = first.creditsUsed.formatted(.number.precision(.fractionLength(0...2)))\n return (\"\\(dayLabel): \\(used) credits\", first.service)\n }\n\n let services = day.services\n .sorted { lhs, rhs in\n if lhs.creditsUsed == rhs.creditsUsed { return lhs.service < rhs.service }\n return lhs.creditsUsed > rhs.creditsUsed\n }\n .prefix(3)\n .map { \"\\($0.service) \\($0.creditsUsed.formatted(.number.precision(.fractionLength(0...2))))\" }\n .joined(separator: \" · \")\n\n return (\"\\(dayLabel): \\(total) credits\", services)\n }\n}\n" }, { "path": "Sources/CodexBar/CursorLoginRunner.swift", "content": "import AppKit\nimport CodexBarCore\nimport Foundation\nimport WebKit\n\n/// Handles Cursor login flow using a WebKit-based browser window.\n/// Captures session cookies after successful authentication.\n@MainActor\nfinal class CursorLoginRunner: NSObject {\n enum Phase {\n case loading\n case waitingLogin\n case success\n case failed(String)\n }\n\n struct Result {\n enum Outcome {\n case success\n case cancelled\n case failed(String)\n }\n\n let outcome: Outcome\n let email: String?\n }\n\n private let browserDetection: BrowserDetection\n private var webView: WKWebView?\n private var window: NSWindow?\n private var continuation: CheckedContinuation?\n private var phaseCallback: ((Phase) -> Void)?\n private var hasCompletedLogin = false\n private let logger = CodexBarLog.logger(LogCategories.cursorLogin)\n\n private static let dashboardURL = URL(string: \"https://cursor.com/dashboard\")!\n private static let loginURLPattern = \"authenticator.cursor.sh\"\n\n init(browserDetection: BrowserDetection) {\n self.browserDetection = browserDetection\n super.init()\n }\n\n /// Runs the Cursor login flow in a browser window.\n /// Returns the result after the user completes login or cancels.\n func run(onPhaseChange: @escaping @Sendable (Phase) -> Void) async -> Result {\n // Keep this instance alive during the flow.\n WebKitTeardown.retain(self)\n self.phaseCallback = onPhaseChange\n onPhaseChange(.loading)\n self.logger.info(\"Cursor login started\")\n\n return await withCheckedContinuation { continuation in\n self.continuation = continuation\n self.setupWindow()\n }\n }\n\n private func setupWindow() {\n // Use a non-persistent store for the login flow; cookies are persisted explicitly.\n let config = WKWebViewConfiguration()\n config.websiteDataStore = .nonPersistent()\n\n let webView = WKWebView(frame: NSRect(x: 0, y: 0, width: 480, height: 640), configuration: config)\n webView.navigationDelegate = self\n self.webView = webView\n\n // Create window\n let window = NSWindow(\n contentRect: NSRect(x: 0, y: 0, width: 480, height: 640),\n styleMask: [.titled, .closable, .resizable],\n backing: .buffered,\n defer: false)\n window.isReleasedWhenClosed = false\n window.title = \"Cursor Login\"\n window.contentView = webView\n window.center()\n window.delegate = self\n window.makeKeyAndOrderFront(nil)\n self.window = window\n self.logger.info(\"Cursor login window opened\")\n\n // Navigate to dashboard (will redirect to login if not authenticated)\n let request = URLRequest(url: Self.dashboardURL)\n webView.load(request)\n }\n\n private func complete(with result: Result) {\n guard let continuation = self.continuation else { return }\n self.continuation = nil\n self.logger.info(\"Cursor login completed\", metadata: [\"outcome\": \"\\(result.outcome)\"])\n self.scheduleCleanup()\n continuation.resume(returning: result)\n }\n\n private func scheduleCleanup() {\n self.logger.info(\"Cursor login window closing\")\n WebKitTeardown.scheduleCleanup(owner: self, window: self.window, webView: self.webView)\n }\n\n private func captureSessionCookies() async {\n guard let webView = self.webView else { return }\n\n let dataStore = webView.configuration.websiteDataStore\n let cookies = await dataStore.httpCookieStore.allCookies()\n\n // Filter for cursor.com cookies\n let cursorCookies = cookies.filter { cookie in\n cookie.domain.contains(\"cursor.com\") || cookie.domain.contains(\"cursor.sh\")\n }\n\n guard !cursorCookies.isEmpty else {\n self.phaseCallback?(.failed(\"No session cookies found\"))\n self.logger.warning(\"Cursor login failed: no session cookies found\")\n self.complete(with: Result(outcome: .failed(\"No session cookies found\"), email: nil))\n return\n }\n\n // Save cookies to the session store\n await CursorSessionStore.shared.setCookies(cursorCookies)\n self.logger.info(\"Cursor session cookies captured\", metadata: [\"count\": \"\\(cursorCookies.count)\"])\n\n // Try to get user email\n let email = await self.fetchUserEmail()\n\n self.hasCompletedLogin = true\n self.phaseCallback?(.success)\n self.complete(with: Result(outcome: .success, email: email))\n }\n\n private func fetchUserEmail() async -> String? {\n do {\n let probe = CursorStatusProbe(browserDetection: self.browserDetection)\n let snapshot = try await probe.fetch()\n return snapshot.accountEmail\n } catch {\n return nil\n }\n }\n}\n\n// MARK: - WKNavigationDelegate\n\nextension CursorLoginRunner: WKNavigationDelegate {\n nonisolated func webView(_ webView: WKWebView, didFinish navigation: WKNavigation!) {\n Task { @MainActor in\n guard let url = webView.url else { return }\n\n let urlString = url.absoluteString\n\n // Check if on login page\n if urlString.contains(Self.loginURLPattern) {\n self.phaseCallback?(.waitingLogin)\n return\n }\n\n // Check if on dashboard (login successful)\n if urlString.contains(\"cursor.com/dashboard\"), !self.hasCompletedLogin {\n await self.captureSessionCookies()\n }\n }\n }\n\n nonisolated func webView(\n _ webView: WKWebView,\n didReceiveServerRedirectForProvisionalNavigation navigation: WKNavigation!)\n {\n Task { @MainActor in\n guard let url = webView.url else { return }\n let urlString = url.absoluteString\n\n // Detect redirect to dashboard after login\n if urlString.contains(\"cursor.com/dashboard\"), !self.hasCompletedLogin {\n // Wait a moment for cookies to be set, then capture\n try? await Task.sleep(nanoseconds: 500_000_000)\n await self.captureSessionCookies()\n }\n }\n }\n\n nonisolated func webView(\n _ webView: WKWebView,\n didFail navigation: WKNavigation!,\n withError error: Error)\n {\n Task { @MainActor in\n self.phaseCallback?(.failed(error.localizedDescription))\n self.logger.error(\"Cursor login navigation failed\", metadata: [\"error\": error.localizedDescription])\n self.complete(with: Result(outcome: .failed(error.localizedDescription), email: nil))\n }\n }\n\n nonisolated func webView(\n _ webView: WKWebView,\n didFailProvisionalNavigation navigation: WKNavigation!,\n withError error: Error)\n {\n Task { @MainActor in\n // Ignore cancelled navigations (common during redirects)\n let nsError = error as NSError\n if nsError.domain == NSURLErrorDomain, nsError.code == NSURLErrorCancelled {\n return\n }\n self.phaseCallback?(.failed(error.localizedDescription))\n self.logger.error(\"Cursor login navigation failed\", metadata: [\"error\": error.localizedDescription])\n self.complete(with: Result(outcome: .failed(error.localizedDescription), email: nil))\n }\n }\n}\n\n// MARK: - NSWindowDelegate\n\nextension CursorLoginRunner: NSWindowDelegate {\n nonisolated func windowWillClose(_ notification: Notification) {\n Task { @MainActor in\n if !self.hasCompletedLogin {\n self.logger.info(\"Cursor login cancelled\")\n self.complete(with: Result(outcome: .cancelled, email: nil))\n }\n }\n }\n}\n" }, { "path": "Sources/CodexBar/Date+RelativeDescription.swift", "content": "import Foundation\n\nenum RelativeTimeFormatters {\n @MainActor\n static let full: RelativeDateTimeFormatter = {\n let formatter = RelativeDateTimeFormatter()\n formatter.unitsStyle = .full\n return formatter\n }()\n}\n\nextension Date {\n @MainActor\n func relativeDescription(now: Date = .now) -> String {\n let seconds = abs(now.timeIntervalSince(self))\n if seconds < 15 {\n return \"just now\"\n }\n return RelativeTimeFormatters.full.localizedString(for: self, relativeTo: now)\n }\n}\n" }, { "path": "Sources/CodexBar/DisplayLink.swift", "content": "import AppKit\nimport CoreVideo\nimport Observation\nimport QuartzCore\n\n/// Minimal display link driver using NSScreen.displayLink on macOS 15+,\n/// and CVDisplayLink on macOS 14.\n/// Publishes ticks on the main thread at the requested frame rate.\n@MainActor\n@Observable\nfinal class DisplayLinkDriver {\n // Published counter used to drive SwiftUI updates.\n var tick: Int = 0\n private var displayLink: CADisplayLink?\n private var cvDisplayLink: CVDisplayLink?\n private var targetInterval: CFTimeInterval = 1.0 / 60.0\n private var lastTickTimestamp: CFTimeInterval = 0\n private let onTick: (() -> Void)?\n\n init(onTick: (() -> Void)? = nil) {\n self.onTick = onTick\n }\n\n func start(fps: Double = 12) {\n guard self.displayLink == nil, self.cvDisplayLink == nil else { return }\n let clampedFps = max(fps, 1)\n self.targetInterval = 1.0 / clampedFps\n self.lastTickTimestamp = 0\n if #available(macOS 15, *), let screen = NSScreen.main {\n // NSScreen.displayLink is macOS 15+ only.\n let displayLink = screen.displayLink(target: self, selector: #selector(self.step))\n let rate = Float(clampedFps)\n displayLink.preferredFrameRateRange = CAFrameRateRange(\n minimum: rate,\n maximum: rate,\n preferred: rate)\n displayLink.add(to: .main, forMode: .common)\n self.displayLink = displayLink\n } else {\n self.startCVDisplayLink()\n }\n }\n\n func stop() {\n self.displayLink?.invalidate()\n self.displayLink = nil\n if let cvDisplayLink = self.cvDisplayLink {\n CVDisplayLinkStop(cvDisplayLink)\n }\n self.cvDisplayLink = nil\n }\n\n @objc private func step(_: AnyObject) {\n self.handleTick()\n }\n\n private func handleTick() {\n let now = CACurrentMediaTime()\n if self.lastTickTimestamp > 0, now - self.lastTickTimestamp < self.targetInterval {\n return\n }\n self.lastTickTimestamp = now\n // Safe on main runloop; drives SwiftUI updates.\n self.tick &+= 1\n self.onTick?()\n }\n\n private func startCVDisplayLink() {\n var link: CVDisplayLink?\n if CVDisplayLinkCreateWithActiveCGDisplays(&link) != kCVReturnSuccess {\n return\n }\n guard let link else { return }\n let callback: CVDisplayLinkOutputCallback = { _, _, _, _, _, userInfo in\n guard let userInfo else { return kCVReturnSuccess }\n let driver = Unmanaged.fromOpaque(userInfo).takeUnretainedValue()\n driver.scheduleTick()\n return kCVReturnSuccess\n }\n CVDisplayLinkSetOutputCallback(link, callback, Unmanaged.passUnretained(self).toOpaque())\n CVDisplayLinkStart(link)\n self.cvDisplayLink = link\n }\n\n private nonisolated func scheduleTick() {\n Task { @MainActor [weak self] in\n self?.handleTick()\n }\n }\n\n deinit {\n Task { @MainActor [weak self] in\n self?.stop()\n }\n }\n}\n" }, { "path": "Sources/CodexBar/GeminiLoginRunner.swift", "content": "import AppKit\nimport CodexBarCore\nimport Foundation\n\nenum GeminiLoginRunner {\n private static let geminiConfigDir = FileManager.default.homeDirectoryForCurrentUser\n .appendingPathComponent(\".gemini\")\n private static let credentialsFile = \"oauth_creds.json\"\n\n private static func clearCredentials() {\n let fm = FileManager.default\n let filesToDelete = [credentialsFile, \"google_accounts.json\"]\n for file in filesToDelete {\n let path = self.geminiConfigDir.appendingPathComponent(file)\n try? fm.removeItem(at: path)\n }\n }\n\n struct Result {\n enum Outcome {\n case success\n case missingBinary\n case launchFailed(String)\n }\n\n let outcome: Outcome\n }\n\n static func run(onCredentialsCreated: (@Sendable () -> Void)? = nil) async -> Result {\n await Task(priority: .userInitiated) {\n let env = ProcessInfo.processInfo.environment\n guard let binary = BinaryLocator.resolveGeminiBinary(\n env: env,\n loginPATH: LoginShellPathCache.shared.current)\n else {\n return Result(outcome: .missingBinary)\n }\n\n // Clear existing credentials before auth (enables clean account switch)\n Self.clearCredentials()\n\n // Start watching for credentials file to be created\n if let callback = onCredentialsCreated {\n Self.watchForCredentials(callback: callback)\n }\n\n // Create a temporary shell script that runs gemini (auto-prompts for auth when no creds)\n let scriptContent = \"\"\"\n #!/bin/bash\n cd ~\n \"\\(binary)\"\n \"\"\"\n\n let tempDir = FileManager.default.temporaryDirectory\n let scriptURL = tempDir.appendingPathComponent(\"gemini_login_\\(UUID().uuidString).command\")\n\n do {\n try scriptContent.write(to: scriptURL, atomically: true, encoding: .utf8)\n try FileManager.default.setAttributes([.posixPermissions: 0o755], ofItemAtPath: scriptURL.path)\n\n let config = NSWorkspace.OpenConfiguration()\n config.activates = true\n try await NSWorkspace.shared.open(scriptURL, configuration: config)\n\n // Clean up script after Terminal has time to read it\n let scriptPath = scriptURL.path\n DispatchQueue.global().asyncAfter(deadline: .now() + 10) {\n try? FileManager.default.removeItem(atPath: scriptPath)\n }\n\n return Result(outcome: .success)\n } catch {\n return Result(outcome: .launchFailed(error.localizedDescription))\n }\n }.value\n }\n\n /// Watch for credentials file to be created, then call callback once\n private static func watchForCredentials(callback: @escaping @Sendable () -> Void, timeout: TimeInterval = 300) {\n let credsPath = self.geminiConfigDir.appendingPathComponent(self.credentialsFile).path\n\n DispatchQueue.global(qos: .utility).async {\n let startTime = Date()\n while Date().timeIntervalSince(startTime) < timeout {\n if FileManager.default.fileExists(atPath: credsPath) {\n // Small delay to ensure file is fully written\n Thread.sleep(forTimeInterval: 0.5)\n callback()\n return\n }\n Thread.sleep(forTimeInterval: 1.0)\n }\n }\n }\n}\n" }, { "path": "Sources/CodexBar/HiddenWindowView.swift", "content": "import SwiftUI\n\nstruct HiddenWindowView: View {\n @Environment(\\.openSettings) private var openSettings\n\n var body: some View {\n Color.clear\n .frame(width: 20, height: 20)\n .onReceive(NotificationCenter.default.publisher(for: .codexbarOpenSettings)) { _ in\n Task { @MainActor in\n self.openSettings()\n }\n }\n .task {\n // Migrate keychain items to reduce permission prompts during development (runs off main thread)\n await Task.detached(priority: .userInitiated) {\n KeychainMigration.migrateIfNeeded()\n }.value\n }\n .onAppear {\n if let window = NSApp.windows.first(where: { $0.title == \"CodexBarLifecycleKeepalive\" }) {\n // Make the keepalive window truly invisible and non-interactive.\n window.styleMask = [.borderless]\n window.collectionBehavior = [.auxiliary, .ignoresCycle, .transient, .canJoinAllSpaces]\n window.isExcludedFromWindowsMenu = true\n window.level = .floating\n window.isOpaque = false\n window.alphaValue = 0\n window.backgroundColor = .clear\n window.hasShadow = false\n window.ignoresMouseEvents = true\n window.canHide = false\n window.setContentSize(NSSize(width: 1, height: 1))\n window.setFrameOrigin(NSPoint(x: -5000, y: -5000))\n }\n }\n }\n}\n" }, { "path": "Sources/CodexBar/HistoricalUsagePace.swift", "content": "import CodexBarCore\nimport Foundation\n\nenum HistoricalUsageWindowKind: String, Codable {\n case secondary\n}\n\nenum HistoricalUsageRecordSource: String, Codable {\n case live\n case backfill\n}\n\nstruct HistoricalUsageRecord: Codable {\n let v: Int\n let provider: UsageProvider\n let windowKind: HistoricalUsageWindowKind\n let source: HistoricalUsageRecordSource\n let accountKey: String?\n let sampledAt: Date\n let usedPercent: Double\n let resetsAt: Date\n let windowMinutes: Int\n\n init(\n v: Int,\n provider: UsageProvider,\n windowKind: HistoricalUsageWindowKind,\n source: HistoricalUsageRecordSource,\n accountKey: String?,\n sampledAt: Date,\n usedPercent: Double,\n resetsAt: Date,\n windowMinutes: Int)\n {\n self.v = v\n self.provider = provider\n self.windowKind = windowKind\n self.source = source\n self.accountKey = accountKey\n self.sampledAt = sampledAt\n self.usedPercent = usedPercent\n self.resetsAt = resetsAt\n self.windowMinutes = windowMinutes\n }\n\n init(from decoder: Decoder) throws {\n let container = try decoder.container(keyedBy: CodingKeys.self)\n self.v = try container.decodeIfPresent(Int.self, forKey: .v) ?? 1\n self.provider = try container.decode(UsageProvider.self, forKey: .provider)\n self.windowKind = try container.decode(HistoricalUsageWindowKind.self, forKey: .windowKind)\n self.source = try container.decodeIfPresent(HistoricalUsageRecordSource.self, forKey: .source) ?? .live\n self.accountKey = try container.decodeIfPresent(String.self, forKey: .accountKey)\n self.sampledAt = try container.decode(Date.self, forKey: .sampledAt)\n self.usedPercent = try container.decode(Double.self, forKey: .usedPercent)\n self.resetsAt = try container.decode(Date.self, forKey: .resetsAt)\n self.windowMinutes = try container.decode(Int.self, forKey: .windowMinutes)\n }\n}\n\nstruct HistoricalWeekProfile {\n let resetsAt: Date\n let windowMinutes: Int\n let curve: [Double]\n}\n\nstruct CodexHistoricalDataset {\n static let gridPointCount = 169\n let weeks: [HistoricalWeekProfile]\n}\n\nactor HistoricalUsageHistoryStore {\n private static let schemaVersion = 1\n private static let writeInterval: TimeInterval = 30 * 60\n private static let writeDeltaThreshold: Double = 1\n private static let retentionDays: TimeInterval = 56 * 24 * 60 * 60\n private static let minimumWeekSamples = 6\n private static let boundaryCoverageWindow: TimeInterval = 24 * 60 * 60\n private static let backfillWindowCapWeeks = 8\n private static let backfillCalibrationMinimumUsedPercent = 1.0\n private static let backfillCalibrationMinimumCredits = 0.001\n private static let backfillSampleFractions: [Double] = (0...14).map { Double($0) / 14.0 }\n private static let coverageTolerance: TimeInterval = 16 * 60 * 60\n private static let resetBucketSeconds: TimeInterval = 60\n\n private let fileURL: URL\n private var records: [HistoricalUsageRecord] = []\n private var loaded = false\n\n init(fileURL: URL? = nil) {\n self.fileURL = fileURL ?? HistoricalUsageHistoryStore.defaultFileURL()\n }\n\n func loadCodexDataset(accountKey: String?) -> CodexHistoricalDataset? {\n self.ensureLoaded()\n return self.buildDataset(accountKey: accountKey)\n }\n\n func recordCodexWeekly(\n window: RateWindow,\n sampledAt: Date = .init(),\n accountKey: String?) -> CodexHistoricalDataset?\n {\n guard let rawResetsAt = window.resetsAt else { return self.loadCodexDataset(accountKey: accountKey) }\n guard let windowMinutes = window.windowMinutes, windowMinutes > 0 else {\n return self.loadCodexDataset(accountKey: accountKey)\n }\n self.ensureLoaded()\n let resetsAt = Self.normalizeReset(rawResetsAt)\n\n let sample = HistoricalUsageRecord(\n v: Self.schemaVersion,\n provider: .codex,\n windowKind: .secondary,\n source: .live,\n accountKey: accountKey,\n sampledAt: sampledAt,\n usedPercent: Self.clamp(window.usedPercent, lower: 0, upper: 100),\n resetsAt: resetsAt,\n windowMinutes: windowMinutes)\n\n if !self.shouldAccept(sample) {\n return self.buildDataset(accountKey: accountKey)\n }\n\n self.records.append(sample)\n self.pruneOldRecords(now: sampledAt)\n self.records.sort { lhs, rhs in\n if lhs.sampledAt == rhs.sampledAt {\n if lhs.resetsAt == rhs.resetsAt {\n return lhs.usedPercent < rhs.usedPercent\n }\n return lhs.resetsAt < rhs.resetsAt\n }\n return lhs.sampledAt < rhs.sampledAt\n }\n self.persist()\n return self.buildDataset(accountKey: accountKey)\n }\n\n func backfillCodexWeeklyFromUsageBreakdown(\n _ breakdown: [OpenAIDashboardDailyBreakdown],\n referenceWindow: RateWindow,\n now: Date = .init(),\n accountKey: String?) -> CodexHistoricalDataset?\n {\n self.ensureLoaded()\n let existingDataset = self.buildDataset(accountKey: accountKey)\n\n guard let rawResetsAt = referenceWindow.resetsAt else { return existingDataset }\n guard let windowMinutes = referenceWindow.windowMinutes, windowMinutes > 0 else { return existingDataset }\n let resetsAt = Self.normalizeReset(rawResetsAt)\n\n let duration = TimeInterval(windowMinutes) * 60\n guard duration > 0 else { return existingDataset }\n\n let windowStart = resetsAt.addingTimeInterval(-duration)\n let calibrationEnd = Self.clampDate(now, lower: windowStart, upper: resetsAt)\n let dayUsages = Self.parseDayUsages(\n from: breakdown,\n asOf: calibrationEnd,\n fillingFrom: windowStart)\n guard !dayUsages.isEmpty else { return existingDataset }\n guard let coverageStart = dayUsages.first?.start, let coverageEnd = dayUsages.last?.end else {\n return existingDataset\n }\n guard coverageStart <= windowStart.addingTimeInterval(Self.coverageTolerance) else {\n return existingDataset\n }\n guard coverageEnd >= calibrationEnd.addingTimeInterval(-Self.coverageTolerance) else {\n return existingDataset\n }\n\n let currentUsedPercent = Self.clamp(referenceWindow.usedPercent, lower: 0, upper: 100)\n guard currentUsedPercent >= Self.backfillCalibrationMinimumUsedPercent else { return existingDataset }\n\n let currentCredits = Self.creditsUsed(\n from: dayUsages,\n between: windowStart,\n and: calibrationEnd)\n guard currentCredits > Self.backfillCalibrationMinimumCredits else { return existingDataset }\n\n let estimatedCreditsAtLimit = currentCredits / (currentUsedPercent / 100)\n guard estimatedCreditsAtLimit.isFinite, estimatedCreditsAtLimit > Self.backfillCalibrationMinimumCredits else {\n return existingDataset\n }\n\n struct RecordKey: Hashable {\n let resetsAt: Date\n let sampledAt: Date\n let windowMinutes: Int\n let accountKey: String?\n }\n\n var synthesized: [HistoricalUsageRecord] = []\n synthesized.reserveCapacity(Self.backfillWindowCapWeeks * Self.backfillSampleFractions.count)\n\n for weeksBack in 1...Self.backfillWindowCapWeeks {\n let reset = Self.normalizeReset(resetsAt.addingTimeInterval(-duration * Double(weeksBack)))\n let start = reset.addingTimeInterval(-duration)\n guard start >= coverageStart.addingTimeInterval(-Self.coverageTolerance),\n reset <= coverageEnd.addingTimeInterval(Self.coverageTolerance)\n else {\n continue\n }\n\n let existingForWeek = self.records.filter {\n $0.provider == .codex &&\n $0.windowKind == .secondary &&\n $0.windowMinutes == windowMinutes &&\n $0.accountKey == accountKey &&\n $0.resetsAt == reset\n }\n if Self.isCompleteWeek(samples: existingForWeek, windowStart: start, resetsAt: reset) {\n continue\n }\n var existingRecordKeys = Set(existingForWeek.map {\n RecordKey(\n resetsAt: $0.resetsAt,\n sampledAt: $0.sampledAt,\n windowMinutes: $0.windowMinutes,\n accountKey: $0.accountKey)\n })\n\n let weekCredits = Self.creditsUsed(from: dayUsages, between: start, and: reset)\n guard weekCredits > Self.backfillCalibrationMinimumCredits else { continue }\n\n for fraction in Self.backfillSampleFractions {\n let sampledAt = start.addingTimeInterval(duration * fraction)\n let recordKey = RecordKey(\n resetsAt: reset,\n sampledAt: sampledAt,\n windowMinutes: windowMinutes,\n accountKey: accountKey)\n guard !existingRecordKeys.contains(recordKey) else { continue }\n let cumulativeCredits = Self.creditsUsed(from: dayUsages, between: start, and: sampledAt)\n let usedPercent = Self.clamp((cumulativeCredits / estimatedCreditsAtLimit) * 100, lower: 0, upper: 100)\n synthesized.append(HistoricalUsageRecord(\n v: Self.schemaVersion,\n provider: .codex,\n windowKind: .secondary,\n source: .backfill,\n accountKey: accountKey,\n sampledAt: sampledAt,\n usedPercent: usedPercent,\n resetsAt: reset,\n windowMinutes: windowMinutes))\n existingRecordKeys.insert(recordKey)\n }\n }\n\n guard !synthesized.isEmpty else { return existingDataset }\n self.records.append(contentsOf: synthesized)\n self.pruneOldRecords(now: now)\n self.records.sort { lhs, rhs in\n if lhs.sampledAt == rhs.sampledAt {\n if lhs.resetsAt == rhs.resetsAt {\n return lhs.usedPercent < rhs.usedPercent\n }\n return lhs.resetsAt < rhs.resetsAt\n }\n return lhs.sampledAt < rhs.sampledAt\n }\n self.persist()\n return self.buildDataset(accountKey: accountKey)\n }\n\n private func shouldAccept(_ sample: HistoricalUsageRecord) -> Bool {\n guard let prior = self.records\n .last(where: {\n $0.provider == sample.provider &&\n $0.windowKind == sample.windowKind &&\n $0.accountKey == sample.accountKey &&\n $0.windowMinutes == sample.windowMinutes\n })\n else {\n return true\n }\n\n if prior.resetsAt != sample.resetsAt { return true }\n if sample.sampledAt.timeIntervalSince(prior.sampledAt) >= Self.writeInterval { return true }\n if abs(sample.usedPercent - prior.usedPercent) >= Self.writeDeltaThreshold { return true }\n return false\n }\n\n private func pruneOldRecords(now: Date) {\n let cutoff = now.addingTimeInterval(-Self.retentionDays)\n self.records.removeAll { $0.sampledAt < cutoff }\n }\n\n private func ensureLoaded() {\n guard !self.loaded else { return }\n self.loaded = true\n self.records = self.readRecordsFromDisk()\n self.pruneOldRecords(now: .init())\n }\n\n private func readRecordsFromDisk() -> [HistoricalUsageRecord] {\n guard let data = try? Data(contentsOf: self.fileURL), !data.isEmpty else { return [] }\n guard let text = String(data: data, encoding: .utf8) else { return [] }\n\n let decoder = JSONDecoder()\n decoder.dateDecodingStrategy = .iso8601\n var decoded: [HistoricalUsageRecord] = []\n decoded.reserveCapacity(text.count / 80)\n\n for rawLine in text.split(separator: \"\\n\", omittingEmptySubsequences: true) {\n let line = String(rawLine).trimmingCharacters(in: .whitespacesAndNewlines)\n guard !line.isEmpty, let lineData = line.data(using: .utf8) else { continue }\n guard var record = try? decoder.decode(HistoricalUsageRecord.self, from: lineData) else { continue }\n record = HistoricalUsageRecord(\n v: record.v,\n provider: record.provider,\n windowKind: record.windowKind,\n source: record.source,\n accountKey: record.accountKey?.isEmpty == false ? record.accountKey : nil,\n sampledAt: record.sampledAt,\n usedPercent: Self.clamp(record.usedPercent, lower: 0, upper: 100),\n resetsAt: Self.normalizeReset(record.resetsAt),\n windowMinutes: record.windowMinutes)\n decoded.append(record)\n }\n return decoded\n }\n\n private func persist() {\n let encoder = JSONEncoder()\n encoder.dateEncodingStrategy = .iso8601\n encoder.outputFormatting = [.sortedKeys]\n\n var lines: [String] = []\n lines.reserveCapacity(self.records.count)\n for record in self.records {\n guard let data = try? encoder.encode(record),\n let line = String(data: data, encoding: .utf8)\n else {\n continue\n }\n lines.append(line)\n }\n\n let payload = (lines.joined(separator: \"\\n\") + \"\\n\").data(using: .utf8) ?? Data()\n let directory = self.fileURL.deletingLastPathComponent()\n do {\n try FileManager.default.createDirectory(at: directory, withIntermediateDirectories: true)\n try payload.write(to: self.fileURL, options: [.atomic])\n } catch {\n // Best-effort cache file; ignore write failures.\n }\n }\n\n private func buildDataset(accountKey: String?) -> CodexHistoricalDataset? {\n struct WeekKey: Hashable {\n let resetsAt: Date\n let windowMinutes: Int\n }\n\n let scoped = self.records\n .filter { record in\n guard record.provider == .codex, record.windowKind == .secondary, record.windowMinutes > 0 else {\n return false\n }\n if let accountKey {\n return record.accountKey == accountKey\n }\n return record.accountKey == nil\n }\n if scoped.isEmpty { return nil }\n\n let grouped = Dictionary(grouping: scoped) {\n WeekKey(resetsAt: $0.resetsAt, windowMinutes: $0.windowMinutes)\n }\n\n var weeks: [HistoricalWeekProfile] = []\n weeks.reserveCapacity(grouped.count)\n\n for (key, samples) in grouped {\n let duration = TimeInterval(key.windowMinutes) * 60\n guard duration > 0 else { continue }\n let windowStart = key.resetsAt.addingTimeInterval(-duration)\n guard Self.isCompleteWeek(samples: samples, windowStart: windowStart, resetsAt: key.resetsAt) else {\n continue\n }\n\n guard let curve = Self.reconstructWeekCurve(\n samples: samples,\n windowStart: windowStart,\n windowDuration: duration,\n gridPointCount: CodexHistoricalDataset.gridPointCount)\n else {\n continue\n }\n\n weeks.append(HistoricalWeekProfile(\n resetsAt: key.resetsAt,\n windowMinutes: key.windowMinutes,\n curve: curve))\n }\n\n weeks.sort { $0.resetsAt < $1.resetsAt }\n if weeks.isEmpty { return nil }\n return CodexHistoricalDataset(weeks: weeks)\n }\n\n private static func reconstructWeekCurve(\n samples: [HistoricalUsageRecord],\n windowStart: Date,\n windowDuration: TimeInterval,\n gridPointCount: Int) -> [Double]?\n {\n guard gridPointCount >= 2 else { return nil }\n\n var points = samples.map { sample -> (u: Double, value: Double) in\n let offset = sample.sampledAt.timeIntervalSince(windowStart)\n let u = Self.clamp(offset / windowDuration, lower: 0, upper: 1)\n return (u: u, value: Self.clamp(sample.usedPercent, lower: 0, upper: 100))\n }\n points.sort { lhs, rhs in\n if lhs.u == rhs.u {\n return lhs.value < rhs.value\n }\n return lhs.u < rhs.u\n }\n\n guard !points.isEmpty else { return nil }\n\n // Enforce monotonicity on observed samples before interpolation.\n var monotonePoints: [(u: Double, value: Double)] = []\n monotonePoints.reserveCapacity(points.count)\n var runningMax = 0.0\n for point in points {\n runningMax = max(runningMax, point.value)\n monotonePoints.append((u: point.u, value: runningMax))\n }\n\n // Anchor reconstructed curves to reset start and end-of-week plateau.\n let endValue = monotonePoints.last?.value ?? 0\n monotonePoints.append((u: 0, value: 0))\n monotonePoints.append((u: 1, value: endValue))\n monotonePoints.sort { lhs, rhs in\n if lhs.u == rhs.u {\n return lhs.value < rhs.value\n }\n return lhs.u < rhs.u\n }\n runningMax = 0\n for index in monotonePoints.indices {\n runningMax = max(runningMax, monotonePoints[index].value)\n monotonePoints[index].value = runningMax\n }\n\n var curve = Array(repeating: 0.0, count: gridPointCount)\n let first = monotonePoints[0]\n let last = monotonePoints[monotonePoints.count - 1]\n\n var upperIndex = 1\n let denominator = Double(gridPointCount - 1)\n\n for index in 0..= last.u {\n curve[index] = last.value\n continue\n }\n\n while upperIndex < monotonePoints.count, monotonePoints[upperIndex].u < u {\n upperIndex += 1\n }\n\n let hi = monotonePoints[min(upperIndex, monotonePoints.count - 1)]\n let lo = monotonePoints[max(0, upperIndex - 1)]\n if hi.u <= lo.u {\n curve[index] = max(lo.value, hi.value)\n continue\n }\n\n let ratio = Self.clamp((u - lo.u) / (hi.u - lo.u), lower: 0, upper: 1)\n curve[index] = lo.value + (hi.value - lo.value) * ratio\n }\n\n // Re-enforce monotonicity on reconstructed grid.\n var curveMax = 0.0\n for index in curve.indices {\n curve[index] = Self.clamp(curve[index], lower: 0, upper: 100)\n curveMax = max(curveMax, curve[index])\n curve[index] = curveMax\n }\n return curve\n }\n\n private static func isCompleteWeek(samples: [HistoricalUsageRecord], windowStart: Date, resetsAt: Date) -> Bool {\n guard samples.count >= self.minimumWeekSamples else { return false }\n let startBoundary = windowStart.addingTimeInterval(Self.boundaryCoverageWindow)\n let endBoundary = resetsAt.addingTimeInterval(-Self.boundaryCoverageWindow)\n let hasStartCoverage = samples.contains { sample in\n sample.sampledAt >= windowStart && sample.sampledAt <= startBoundary\n }\n let hasEndCoverage = samples.contains { sample in\n sample.sampledAt >= endBoundary && sample.sampledAt <= resetsAt\n }\n return hasStartCoverage && hasEndCoverage\n }\n\n private struct DayUsage {\n let start: Date\n let end: Date\n let creditsUsed: Double\n }\n\n private static func parseDayUsages(\n from breakdown: [OpenAIDashboardDailyBreakdown],\n asOf: Date,\n fillingFrom expectedCoverageStart: Date? = nil) -> [DayUsage]\n {\n var creditsByStart: [Date: Double] = [:]\n creditsByStart.reserveCapacity(breakdown.count)\n\n for day in breakdown {\n guard let dayStart = Self.dayStart(for: day.day) else { continue }\n creditsByStart[dayStart, default: 0] += max(0, day.totalCreditsUsed)\n }\n\n let calendar = Self.gregorianCalendar()\n var dayUsages: [DayUsage] = []\n dayUsages.reserveCapacity(creditsByStart.count)\n for (dayStart, credits) in creditsByStart {\n guard let nominalEnd = calendar.date(byAdding: .day, value: 1, to: dayStart) else { continue }\n let effectiveEnd: Date = if dayStart <= asOf, asOf < nominalEnd {\n asOf\n } else {\n nominalEnd\n }\n guard effectiveEnd > dayStart else { continue }\n dayUsages.append(DayUsage(start: dayStart, end: effectiveEnd, creditsUsed: credits))\n }\n\n dayUsages.sort { lhs, rhs in lhs.start < rhs.start }\n return Self.fillMissingZeroUsageDays(\n in: dayUsages,\n through: asOf,\n fillingFrom: expectedCoverageStart)\n }\n\n private static func fillMissingZeroUsageDays(\n in dayUsages: [DayUsage],\n through asOf: Date,\n fillingFrom expectedCoverageStart: Date? = nil) -> [DayUsage]\n {\n guard let firstStart = dayUsages.first?.start else { return [] }\n\n let calendar = Self.gregorianCalendar()\n let fillStart: Date = if let expectedCoverageStart {\n min(firstStart, calendar.startOfDay(for: expectedCoverageStart))\n } else {\n firstStart\n }\n let finalDayStart = calendar.startOfDay(for: asOf)\n guard fillStart <= finalDayStart else { return dayUsages }\n\n let creditsByStart = Dictionary(uniqueKeysWithValues: dayUsages.map { ($0.start, $0.creditsUsed) })\n let daySpan = max(0, calendar.dateComponents([.day], from: fillStart, to: finalDayStart).day ?? 0)\n var filled: [DayUsage] = []\n filled.reserveCapacity(daySpan + 1)\n\n var cursor = fillStart\n while cursor <= finalDayStart {\n guard let nominalEnd = calendar.date(byAdding: .day, value: 1, to: cursor) else { break }\n let effectiveEnd: Date = if cursor <= asOf, asOf < nominalEnd {\n asOf\n } else {\n nominalEnd\n }\n guard effectiveEnd > cursor else { break }\n filled.append(DayUsage(\n start: cursor,\n end: effectiveEnd,\n creditsUsed: creditsByStart[cursor] ?? 0))\n guard let next = calendar.date(byAdding: .day, value: 1, to: cursor) else { break }\n cursor = next\n }\n\n return filled\n }\n\n private static func dayStart(for key: String) -> Date? {\n let components = key.split(separator: \"-\", omittingEmptySubsequences: true)\n guard components.count == 3,\n let year = Int(components[0]),\n let month = Int(components[1]),\n let day = Int(components[2])\n else {\n return nil\n }\n\n let calendar = Self.gregorianCalendar()\n var dateComponents = DateComponents()\n dateComponents.calendar = calendar\n dateComponents.timeZone = calendar.timeZone\n dateComponents.year = year\n dateComponents.month = month\n dateComponents.day = day\n dateComponents.hour = 0\n dateComponents.minute = 0\n dateComponents.second = 0\n return dateComponents.date\n }\n\n private static func creditsUsed(from dayUsages: [DayUsage], between start: Date, and end: Date) -> Double {\n guard end > start else { return 0 }\n var total = 0.0\n for day in dayUsages {\n if day.end <= start { continue }\n if day.start >= end { break }\n let overlapStart = max(day.start, start)\n let overlapEnd = min(day.end, end)\n guard overlapEnd > overlapStart else { continue }\n\n let dayDuration = day.end.timeIntervalSince(day.start)\n guard dayDuration > 0 else { continue }\n let overlap = overlapEnd.timeIntervalSince(overlapStart)\n total += day.creditsUsed * (overlap / dayDuration)\n }\n return max(0, total)\n }\n\n nonisolated static func defaultFileURL() -> URL {\n let root = FileManager.default.urls(for: .applicationSupportDirectory, in: .userDomainMask).first\n ?? FileManager.default.homeDirectoryForCurrentUser\n return root\n .appendingPathComponent(\"CodexBar\", isDirectory: true)\n .appendingPathComponent(\"usage-history.jsonl\", isDirectory: false)\n }\n\n private nonisolated static func clamp(_ value: Double, lower: Double, upper: Double) -> Double {\n min(upper, max(lower, value))\n }\n\n private nonisolated static func clampDate(_ value: Date, lower: Date, upper: Date) -> Date {\n min(upper, max(lower, value))\n }\n\n private nonisolated static func normalizeReset(_ value: Date) -> Date {\n let bucket = Self.resetBucketSeconds\n guard bucket > 0 else { return value }\n let rounded = (value.timeIntervalSinceReferenceDate / bucket).rounded() * bucket\n return Date(timeIntervalSinceReferenceDate: rounded)\n }\n\n private nonisolated static func gregorianCalendar() -> Calendar {\n var calendar = Calendar(identifier: .gregorian)\n calendar.timeZone = TimeZone.current\n return calendar\n }\n\n #if DEBUG\n nonisolated static func _dayStartForTesting(_ key: String) -> Date? {\n self.dayStart(for: key)\n }\n\n nonisolated static func _creditsUsedForTesting(\n breakdown: [OpenAIDashboardDailyBreakdown],\n asOf: Date,\n start: Date,\n end: Date) -> Double\n {\n let dayUsages = Self.parseDayUsages(from: breakdown, asOf: asOf)\n return Self.creditsUsed(from: dayUsages, between: start, and: end)\n }\n #endif\n}\n\nenum CodexHistoricalPaceEvaluator {\n static let minimumCompleteWeeksForHistorical = 3\n static let minimumWeeksForRisk = 5\n private static let recencyTauWeeks: Double = 3\n private static let epsilon: Double = 1e-9\n private static let resetBucketSeconds: TimeInterval = 60\n\n static func evaluate(window: RateWindow, now: Date, dataset: CodexHistoricalDataset?) -> UsagePace? {\n guard let dataset else { return nil }\n guard let resetsAt = window.resetsAt else { return nil }\n let minutes = window.windowMinutes ?? 10080\n guard minutes > 0 else { return nil }\n\n let duration = TimeInterval(minutes) * 60\n guard duration > 0 else { return nil }\n\n let timeUntilReset = resetsAt.timeIntervalSince(now)\n guard timeUntilReset > 0, timeUntilReset <= duration else { return nil }\n let normalizedResetsAt = Self.normalizeReset(resetsAt)\n\n let elapsed = Self.clamp(duration - timeUntilReset, lower: 0, upper: duration)\n let actual = Self.clamp(window.usedPercent, lower: 0, upper: 100)\n if elapsed == 0, actual > 0 { return nil }\n\n let uNow = Self.clamp(elapsed / duration, lower: 0, upper: 1)\n let scopedWeeks = dataset.weeks.filter { week in\n week.windowMinutes == minutes && week.resetsAt < normalizedResetsAt\n }\n guard scopedWeeks.count >= Self.minimumCompleteWeeksForHistorical else { return nil }\n\n let weightedWeeks = scopedWeeks.map { week in\n let ageWeeks = Self.clamp(\n normalizedResetsAt.timeIntervalSince(week.resetsAt) / duration,\n lower: 0,\n upper: Double.greatestFiniteMagnitude)\n let weight = exp(-ageWeeks / Self.recencyTauWeeks)\n return (week: week, weight: weight)\n }\n let totalWeight = weightedWeeks.reduce(0.0) { $0 + $1.weight }\n guard totalWeight > Self.epsilon else { return nil }\n\n let totalWeightSquared = weightedWeeks.reduce(0.0) { $0 + ($1.weight * $1.weight) }\n let nEff = totalWeightSquared > Self.epsilon ? (totalWeight * totalWeight) / totalWeightSquared : 0\n let lambda = Self.clamp((nEff - 2) / 6, lower: 0, upper: 1)\n\n let gridCount = CodexHistoricalDataset.gridPointCount\n let denominator = Double(gridCount - 1)\n var expectedCurve = Array(repeating: 0.0, count: gridCount)\n for index in 0..= 100 - Self.epsilon\n if runOut {\n weightedRunOutMass += weight\n if let crossingU = Self.firstCrossing(\n after: uNow,\n curve: week.curve,\n shift: shift,\n actualAtNow: actual)\n {\n let etaSeconds = max(0, (crossingU - uNow) * duration)\n crossingCandidates.append((etaSeconds: etaSeconds, weight: weight))\n }\n }\n }\n\n let smoothedProbability = Self.clamp(\n (weightedRunOutMass + 0.5) / (totalWeight + 1),\n lower: 0,\n upper: 1)\n let runOutProbability: Double? = scopedWeeks.count >= Self.minimumWeeksForRisk ? smoothedProbability : nil\n\n var willLastToReset = smoothedProbability < 0.5\n var etaSeconds: TimeInterval?\n\n if !willLastToReset {\n let values = crossingCandidates.map(\\.etaSeconds)\n let weights = crossingCandidates.map(\\.weight)\n if values.isEmpty {\n willLastToReset = true\n } else {\n etaSeconds = max(0, Self.weightedMedian(values: values, weights: weights))\n }\n }\n\n return UsagePace.historical(\n expectedUsedPercent: expectedNow,\n actualUsedPercent: actual,\n etaSeconds: etaSeconds,\n willLastToReset: willLastToReset,\n runOutProbability: runOutProbability)\n }\n\n private static func firstCrossing(\n after uNow: Double,\n curve: [Double],\n shift: Double,\n actualAtNow: Double) -> Double?\n {\n let gridCount = curve.count\n guard gridCount >= 2 else { return nil }\n\n let denominator = Double(gridCount - 1)\n var previousU = uNow\n var previousValue = actualAtNow\n\n let startIndex = min(gridCount - 1, max(1, Int(floor(uNow * denominator)) + 1))\n for index in startIndex..= 100 - Self.epsilon {\n let delta = value - previousValue\n if abs(delta) <= Self.epsilon { return u }\n let ratio = Self.clamp((100 - previousValue) / delta, lower: 0, upper: 1)\n return Self.clamp(previousU + ratio * (u - previousU), lower: uNow, upper: 1)\n }\n previousU = u\n previousValue = value\n }\n return nil\n }\n\n private static func interpolate(curve: [Double], at u: Double) -> Double {\n guard !curve.isEmpty else { return 0 }\n if curve.count == 1 { return curve[0] }\n\n let clipped = Self.clamp(u, lower: 0, upper: 1)\n let scaled = clipped * Double(curve.count - 1)\n let lower = Int(floor(scaled))\n let upper = min(curve.count - 1, lower + 1)\n if lower == upper { return curve[lower] }\n let ratio = scaled - Double(lower)\n return curve[lower] + ((curve[upper] - curve[lower]) * ratio)\n }\n\n private static func weightedMedian(values: [Double], weights: [Double]) -> Double {\n guard values.count == weights.count, !values.isEmpty else { return 0 }\n let pairs = zip(values, weights)\n .map { (value: $0, weight: max(0, $1)) }\n .sorted { lhs, rhs in lhs.value < rhs.value }\n let totalWeight = pairs.reduce(0.0) { $0 + $1.weight }\n if totalWeight <= Self.epsilon {\n let sortedValues = values.sorted()\n return sortedValues[sortedValues.count / 2]\n }\n\n let threshold = totalWeight / 2\n var cumulative = 0.0\n for pair in pairs {\n cumulative += pair.weight\n if cumulative >= threshold {\n return pair.value\n }\n }\n return pairs.last?.value ?? 0\n }\n\n private static func clamp(_ value: Double, lower: Double, upper: Double) -> Double {\n min(upper, max(lower, value))\n }\n\n private static func normalizeReset(_ value: Date) -> Date {\n let bucket = Self.resetBucketSeconds\n guard bucket > 0 else { return value }\n let rounded = (value.timeIntervalSinceReferenceDate / bucket).rounded() * bucket\n return Date(timeIntervalSinceReferenceDate: rounded)\n }\n}\n" }, { "path": "Sources/CodexBar/IconRenderer.swift", "content": "import AppKit\nimport CodexBarCore\n\n// swiftlint:disable:next type_body_length\nenum IconRenderer {\n private static let creditsCap: Double = 1000\n private static let baseSize = NSSize(width: 18, height: 18)\n // Render to an 18×18 pt template (36×36 px at 2×) to match the system menu bar size.\n private static let outputSize = NSSize(width: 18, height: 18)\n private static let outputScale: CGFloat = 2\n private static let canvasPx = Int(outputSize.width * outputScale)\n\n private struct PixelGrid {\n let scale: CGFloat\n\n func pt(_ px: Int) -> CGFloat {\n CGFloat(px) / self.scale\n }\n\n func rect(x: Int, y: Int, w: Int, h: Int) -> CGRect {\n CGRect(x: self.pt(x), y: self.pt(y), width: self.pt(w), height: self.pt(h))\n }\n\n func snapDelta(_ value: CGFloat) -> CGFloat {\n (value * self.scale).rounded() / self.scale\n }\n }\n\n private static let grid = PixelGrid(scale: outputScale)\n\n private struct IconCacheKey: Hashable {\n let primary: Int\n let weekly: Int\n let credits: Int\n let stale: Bool\n let style: Int\n let indicator: Int\n }\n\n private final class IconCacheStore: @unchecked Sendable {\n private var cache: [IconCacheKey: NSImage] = [:]\n private var order: [IconCacheKey] = []\n private let lock = NSLock()\n\n func cachedIcon(for key: IconCacheKey) -> NSImage? {\n self.lock.lock()\n defer { self.lock.unlock() }\n guard let image = self.cache[key] else { return nil }\n if let idx = self.order.firstIndex(of: key) {\n self.order.remove(at: idx)\n self.order.append(key)\n }\n return image\n }\n\n func storeIcon(_ image: NSImage, for key: IconCacheKey, limit: Int) {\n self.lock.lock()\n defer { self.lock.unlock() }\n self.cache[key] = image\n self.order.removeAll { $0 == key }\n self.order.append(key)\n while self.order.count > limit {\n let oldest = self.order.removeFirst()\n self.cache.removeValue(forKey: oldest)\n }\n }\n }\n\n private static let iconCacheStore = IconCacheStore()\n private static let iconCacheLimit = 64\n private static let morphBucketCount = 200\n private static let morphCache = MorphCache(limit: 512)\n\n private final class MorphCache: @unchecked Sendable {\n private let cache = NSCache()\n\n init(limit: Int) {\n self.cache.countLimit = limit\n }\n\n func image(for key: NSNumber) -> NSImage? {\n self.cache.object(forKey: key)\n }\n\n func set(_ image: NSImage, for key: NSNumber) {\n self.cache.setObject(image, forKey: key)\n }\n }\n\n private struct RectPx: Hashable {\n let x: Int\n let y: Int\n let w: Int\n let h: Int\n\n var midXPx: Int {\n self.x + self.w / 2\n }\n\n var midYPx: Int {\n self.y + self.h / 2\n }\n\n func rect() -> CGRect {\n Self.grid.rect(x: self.x, y: self.y, w: self.w, h: self.h)\n }\n\n private static let grid = IconRenderer.grid\n }\n\n // swiftlint:disable function_body_length\n static func makeIcon(\n primaryRemaining: Double?,\n weeklyRemaining: Double?,\n creditsRemaining: Double?,\n stale: Bool,\n style: IconStyle,\n blink: CGFloat = 0,\n wiggle: CGFloat = 0,\n tilt: CGFloat = 0,\n statusIndicator: ProviderStatusIndicator = .none) -> NSImage\n {\n let shouldCache = blink <= 0.0001 && wiggle <= 0.0001 && tilt <= 0.0001\n let render = {\n self.renderImage {\n // Keep monochrome template icons; Claude uses subtle shape cues only.\n let baseFill = NSColor.labelColor\n let trackFillAlpha: CGFloat = stale ? 0.18 : 0.28\n let trackStrokeAlpha: CGFloat = stale ? 0.28 : 0.44\n let fillColor = baseFill.withAlphaComponent(stale ? 0.55 : 1.0)\n\n let barWidthPx = 30 // 15 pt at 2×, uses the slot better without touching edges.\n let barXPx = (Self.canvasPx - barWidthPx) / 2\n\n func drawBar(\n rectPx: RectPx,\n remaining: Double?,\n alpha: CGFloat = 1.0,\n addNotches: Bool = false,\n addFace: Bool = false,\n addGeminiTwist: Bool = false,\n addAntigravityTwist: Bool = false,\n addFactoryTwist: Bool = false,\n addWarpTwist: Bool = false,\n blink: CGFloat = 0,\n drawTrackFill: Bool = true,\n warpEyesFilled: Bool = false)\n {\n let rect = rectPx.rect()\n // Claude reads better as a blockier critter; Codex stays as a capsule.\n // Warp uses small corner radius for rounded rectangle (matching logo style)\n let cornerRadiusPx = addNotches ? 0 : (addWarpTwist ? 3 : rectPx.h / 2)\n let radius = Self.grid.pt(cornerRadiusPx)\n\n let trackPath = NSBezierPath(roundedRect: rect, xRadius: radius, yRadius: radius)\n if drawTrackFill {\n baseFill.withAlphaComponent(trackFillAlpha * alpha).setFill()\n trackPath.fill()\n }\n\n // Crisp outline: stroke an inset path so the stroke stays within pixel bounds.\n let strokeWidthPx = 2 // 1 pt == 2 px at 2×\n let insetPx = strokeWidthPx / 2\n let strokeRect = Self.grid.rect(\n x: rectPx.x + insetPx,\n y: rectPx.y + insetPx,\n w: max(0, rectPx.w - insetPx * 2),\n h: max(0, rectPx.h - insetPx * 2))\n let strokePath = NSBezierPath(\n roundedRect: strokeRect,\n xRadius: Self.grid.pt(max(0, cornerRadiusPx - insetPx)),\n yRadius: Self.grid.pt(max(0, cornerRadiusPx - insetPx)))\n strokePath.lineWidth = CGFloat(strokeWidthPx) / Self.outputScale\n baseFill.withAlphaComponent(trackStrokeAlpha * alpha).setStroke()\n strokePath.stroke()\n\n // Fill: clip to the capsule and paint a left-to-right rect so the progress edge is straight.\n if let remaining {\n let clamped = max(0, min(remaining / 100, 1))\n let fillWidthPx = max(0, min(rectPx.w, Int((CGFloat(rectPx.w) * CGFloat(clamped)).rounded())))\n if fillWidthPx > 0 {\n NSGraphicsContext.current?.cgContext.saveGState()\n trackPath.addClip()\n fillColor.withAlphaComponent(alpha).setFill()\n NSBezierPath(\n rect: Self.grid.rect(\n x: rectPx.x,\n y: rectPx.y,\n w: fillWidthPx,\n h: rectPx.h)).fill()\n NSGraphicsContext.current?.cgContext.restoreGState()\n }\n }\n\n // Codex face: eye cutouts plus faint eyelids to give the prompt some personality.\n if addFace {\n let ctx = NSGraphicsContext.current?.cgContext\n let eyeSizePx = 4\n let eyeOffsetPx = 7\n let eyeCenterYPx = rectPx.y + rectPx.h / 2\n let centerXPx = rectPx.midXPx\n\n ctx?.saveGState()\n ctx?.setShouldAntialias(false)\n ctx?.clear(Self.grid.rect(\n x: centerXPx - eyeOffsetPx - eyeSizePx / 2,\n y: eyeCenterYPx - eyeSizePx / 2,\n w: eyeSizePx,\n h: eyeSizePx))\n ctx?.clear(Self.grid.rect(\n x: centerXPx + eyeOffsetPx - eyeSizePx / 2,\n y: eyeCenterYPx - eyeSizePx / 2,\n w: eyeSizePx,\n h: eyeSizePx))\n ctx?.restoreGState()\n\n // Blink: refill eyes from the top down using the bar fill color.\n if blink > 0.001 {\n let clamped = max(0, min(blink, 1))\n let blinkHeightPx = Int((CGFloat(eyeSizePx) * clamped).rounded())\n fillColor.withAlphaComponent(alpha).setFill()\n let blinkRectLeft = Self.grid.rect(\n x: centerXPx - eyeOffsetPx - eyeSizePx / 2,\n y: eyeCenterYPx + eyeSizePx / 2 - blinkHeightPx,\n w: eyeSizePx,\n h: blinkHeightPx)\n let blinkRectRight = Self.grid.rect(\n x: centerXPx + eyeOffsetPx - eyeSizePx / 2,\n y: eyeCenterYPx + eyeSizePx / 2 - blinkHeightPx,\n w: eyeSizePx,\n h: blinkHeightPx)\n NSBezierPath(rect: blinkRectLeft).fill()\n NSBezierPath(rect: blinkRectRight).fill()\n }\n\n // Hat: a tiny cap hovering above the eyes to give the face more character.\n let hatWidthPx = 18\n let hatHeightPx = 4\n let hatRect = Self.grid.rect(\n x: centerXPx - hatWidthPx / 2,\n y: rectPx.y + rectPx.h - hatHeightPx,\n w: hatWidthPx,\n h: hatHeightPx)\n ctx?.saveGState()\n if abs(tilt) > 0.0001 {\n // Tilt only the hat; keep eyes pixel-crisp and axis-aligned.\n let faceCenter = CGPoint(x: Self.grid.pt(centerXPx), y: Self.grid.pt(eyeCenterYPx))\n ctx?.translateBy(x: faceCenter.x, y: faceCenter.y)\n ctx?.rotate(by: tilt)\n ctx?.translateBy(x: -faceCenter.x, y: -faceCenter.y - abs(tilt) * 1.2)\n }\n fillColor.withAlphaComponent(alpha).setFill()\n NSBezierPath(rect: hatRect).fill()\n ctx?.restoreGState()\n }\n\n // Claude twist: blocky crab-style critter (arms + legs + vertical eyes).\n if addNotches {\n let ctx = NSGraphicsContext.current?.cgContext\n let wiggleOffset = Self.grid.snapDelta(wiggle * 0.6)\n let wigglePx = Int((wiggleOffset * Self.outputScale).rounded())\n\n fillColor.withAlphaComponent(alpha).setFill()\n\n // Arms/claws: mid-height side protrusions.\n // Keep within the 18×18pt canvas: barX is 3px, so 3px arms reach the edge without clipping.\n let armWidthPx = 3\n let armHeightPx = max(0, rectPx.h - 6)\n let armYPx = rectPx.y + 3 + wigglePx / 6\n let leftArm = Self.grid.rect(\n x: rectPx.x - armWidthPx,\n y: armYPx,\n w: armWidthPx,\n h: armHeightPx)\n let rightArm = Self.grid.rect(\n x: rectPx.x + rectPx.w,\n y: armYPx,\n w: armWidthPx,\n h: armHeightPx)\n NSBezierPath(rect: leftArm).fill()\n NSBezierPath(rect: rightArm).fill()\n\n // Legs: 4 little pixels underneath, like a tiny crab.\n let legCount = 4\n let legWidthPx = 2\n let legHeightPx = 3\n let legYPx = rectPx.y - legHeightPx + wigglePx / 6\n let stepPx = max(1, rectPx.w / (legCount + 1))\n for idx in 0.. 0.001 {\n let clamped = max(0, min(blink, 1))\n let blinkHeightPx = Int((CGFloat(eyeHeightPx) * clamped).rounded())\n fillColor.withAlphaComponent(alpha).setFill()\n let leftBlink = Self.grid.rect(\n x: rectPx.midXPx - eyeOffsetPx - eyeWidthPx / 2,\n y: eyeYPx + eyeHeightPx - blinkHeightPx,\n w: eyeWidthPx,\n h: blinkHeightPx)\n let rightBlink = Self.grid.rect(\n x: rectPx.midXPx + eyeOffsetPx - eyeWidthPx / 2,\n y: eyeYPx + eyeHeightPx - blinkHeightPx,\n w: eyeWidthPx,\n h: blinkHeightPx)\n NSBezierPath(rect: leftBlink).fill()\n NSBezierPath(rect: rightBlink).fill()\n }\n }\n\n // Gemini twist: sparkle-inspired design with prominent 4-pointed stars as eyes\n // and decorative points extending from the bar.\n if addGeminiTwist {\n let ctx = NSGraphicsContext.current?.cgContext\n let centerXPx = rectPx.midXPx\n let eyeCenterYPx = rectPx.y + rectPx.h / 2\n\n ctx?.saveGState()\n ctx?.setShouldAntialias(true)\n\n // 4-pointed star cutouts (Gemini sparkle eyes) - BIGGER\n let starSizePx = 8\n let eyeOffsetPx = 8\n let sr = Self.grid.pt(starSizePx / 2)\n let innerR = sr * 0.25\n\n func drawStarCutout(cx: CGFloat, cy: CGFloat) {\n let path = NSBezierPath()\n for i in 0..<8 {\n let angle = CGFloat(i) * .pi / 4 - .pi / 2\n let radius = (i % 2 == 0) ? sr : innerR\n let px = cx + cos(angle) * radius\n let py = cy + sin(angle) * radius\n if i == 0 {\n path.move(to: NSPoint(x: px, y: py))\n } else {\n path.line(to: NSPoint(x: px, y: py))\n }\n }\n path.close()\n path.fill()\n }\n\n let ldCx = Self.grid.pt(centerXPx - eyeOffsetPx)\n let rdCx = Self.grid.pt(centerXPx + eyeOffsetPx)\n let yCy = Self.grid.pt(eyeCenterYPx)\n\n // Clear star shapes for eyes\n ctx?.setBlendMode(.clear)\n drawStarCutout(cx: ldCx, cy: yCy)\n drawStarCutout(cx: rdCx, cy: yCy)\n ctx?.setBlendMode(.normal)\n\n // Decorative sparkle points extending from bar (sized to stay within 36px canvas)\n fillColor.withAlphaComponent(alpha).setFill()\n let pointHeightPx = 4\n let pointWidthPx = 4\n\n // Top center point (like a crown/sparkle)\n let topPointPath = NSBezierPath()\n let topCx = Self.grid.pt(centerXPx)\n let topBaseY = Self.grid.pt(rectPx.y + rectPx.h)\n let topPeakY = Self.grid.pt(rectPx.y + rectPx.h + pointHeightPx)\n let halfW = Self.grid.pt(pointWidthPx / 2)\n topPointPath.move(to: NSPoint(x: topCx - halfW, y: topBaseY))\n topPointPath.line(to: NSPoint(x: topCx, y: topPeakY))\n topPointPath.line(to: NSPoint(x: topCx + halfW, y: topBaseY))\n topPointPath.close()\n topPointPath.fill()\n\n // Bottom center point\n let bottomPointPath = NSBezierPath()\n let bottomBaseY = Self.grid.pt(rectPx.y)\n let bottomPeakY = Self.grid.pt(rectPx.y - pointHeightPx)\n bottomPointPath.move(to: NSPoint(x: topCx - halfW, y: bottomBaseY))\n bottomPointPath.line(to: NSPoint(x: topCx, y: bottomPeakY))\n bottomPointPath.line(to: NSPoint(x: topCx + halfW, y: bottomBaseY))\n bottomPointPath.close()\n bottomPointPath.fill()\n\n // Side points (max 3px to stay within canvas edge)\n let sidePointH = 3\n let sidePointW = 3\n let sideHalfW = Self.grid.pt(sidePointW / 2)\n let barMidY = Self.grid.pt(eyeCenterYPx)\n\n // Left side point\n let leftSidePath = NSBezierPath()\n let leftBaseX = Self.grid.pt(rectPx.x)\n let leftPeakX = Self.grid.pt(rectPx.x - sidePointH)\n leftSidePath.move(to: NSPoint(x: leftBaseX, y: barMidY - sideHalfW))\n leftSidePath.line(to: NSPoint(x: leftPeakX, y: barMidY))\n leftSidePath.line(to: NSPoint(x: leftBaseX, y: barMidY + sideHalfW))\n leftSidePath.close()\n leftSidePath.fill()\n\n // Right side point\n let rightSidePath = NSBezierPath()\n let rightBaseX = Self.grid.pt(rectPx.x + rectPx.w)\n let rightPeakX = Self.grid.pt(rectPx.x + rectPx.w + sidePointH)\n rightSidePath.move(to: NSPoint(x: rightBaseX, y: barMidY - sideHalfW))\n rightSidePath.line(to: NSPoint(x: rightPeakX, y: barMidY))\n rightSidePath.line(to: NSPoint(x: rightBaseX, y: barMidY + sideHalfW))\n rightSidePath.close()\n rightSidePath.fill()\n\n ctx?.restoreGState()\n\n // Blink: fill star eyes\n if blink > 0.001 {\n let clamped = max(0, min(blink, 1))\n fillColor.withAlphaComponent(alpha).setFill()\n let blinkR = sr * clamped\n let blinkInnerR = blinkR * 0.25\n\n func drawBlinkStar(cx: CGFloat, cy: CGFloat) {\n let path = NSBezierPath()\n for i in 0..<8 {\n let angle = CGFloat(i) * .pi / 4 - .pi / 2\n let radius = (i % 2 == 0) ? blinkR : blinkInnerR\n let px = cx + cos(angle) * radius\n let py = cy + sin(angle) * radius\n if i == 0 {\n path.move(to: NSPoint(x: px, y: py))\n } else {\n path.line(to: NSPoint(x: px, y: py))\n }\n }\n path.close()\n path.fill()\n }\n\n drawBlinkStar(cx: ldCx, cy: yCy)\n drawBlinkStar(cx: rdCx, cy: yCy)\n }\n }\n\n if addAntigravityTwist {\n let dotSizePx = 3\n let dotOffsetXPx = rectPx.x + rectPx.w + 2\n let dotOffsetYPx = rectPx.y + rectPx.h - 2\n fillColor.withAlphaComponent(alpha).setFill()\n let dotRect = Self.grid.rect(\n x: dotOffsetXPx - dotSizePx / 2,\n y: dotOffsetYPx - dotSizePx / 2,\n w: dotSizePx,\n h: dotSizePx)\n NSBezierPath(ovalIn: dotRect).fill()\n }\n\n // Factory twist: 8-pointed asterisk/gear-like eyes with cog teeth accents\n if addFactoryTwist {\n let ctx = NSGraphicsContext.current?.cgContext\n let centerXPx = rectPx.midXPx\n let eyeCenterYPx = rectPx.y + rectPx.h / 2\n\n ctx?.saveGState()\n ctx?.setShouldAntialias(true)\n\n // 8-pointed asterisk cutouts (Factory gear-like eyes)\n let starSizePx = 7\n let eyeOffsetPx = 8\n let sr = Self.grid.pt(starSizePx / 2)\n let innerR = sr * 0.3\n\n func drawAsteriskCutout(cx: CGFloat, cy: CGFloat) {\n let path = NSBezierPath()\n // 8 points for the asterisk\n for i in 0..<16 {\n let angle = CGFloat(i) * .pi / 8 - .pi / 2\n let radius = (i % 2 == 0) ? sr : innerR\n let px = cx + cos(angle) * radius\n let py = cy + sin(angle) * radius\n if i == 0 {\n path.move(to: NSPoint(x: px, y: py))\n } else {\n path.line(to: NSPoint(x: px, y: py))\n }\n }\n path.close()\n path.fill()\n }\n\n let ldCx = Self.grid.pt(centerXPx - eyeOffsetPx)\n let rdCx = Self.grid.pt(centerXPx + eyeOffsetPx)\n let yCy = Self.grid.pt(eyeCenterYPx)\n\n // Clear asterisk shapes for eyes\n ctx?.setBlendMode(.clear)\n drawAsteriskCutout(cx: ldCx, cy: yCy)\n drawAsteriskCutout(cx: rdCx, cy: yCy)\n ctx?.setBlendMode(.normal)\n\n // Small gear teeth on top and bottom edges\n fillColor.withAlphaComponent(alpha).setFill()\n let toothWidthPx = 3\n let toothHeightPx = 2\n\n // Top teeth (2 small rectangles)\n let topY = Self.grid.pt(rectPx.y + rectPx.h)\n let tooth1X = Self.grid.pt(centerXPx - 5 - toothWidthPx / 2)\n let tooth2X = Self.grid.pt(centerXPx + 5 - toothWidthPx / 2)\n NSBezierPath(rect: CGRect(\n x: tooth1X,\n y: topY,\n width: Self.grid.pt(toothWidthPx),\n height: Self.grid.pt(toothHeightPx))).fill()\n NSBezierPath(rect: CGRect(\n x: tooth2X,\n y: topY,\n width: Self.grid.pt(toothWidthPx),\n height: Self.grid.pt(toothHeightPx))).fill()\n\n // Bottom teeth\n let bottomY = Self.grid.pt(rectPx.y - toothHeightPx)\n NSBezierPath(rect: CGRect(\n x: tooth1X,\n y: bottomY,\n width: Self.grid.pt(toothWidthPx),\n height: Self.grid.pt(toothHeightPx))).fill()\n NSBezierPath(rect: CGRect(\n x: tooth2X,\n y: bottomY,\n width: Self.grid.pt(toothWidthPx),\n height: Self.grid.pt(toothHeightPx))).fill()\n\n ctx?.restoreGState()\n\n // Blink: fill asterisk eyes\n if blink > 0.001 {\n let clamped = max(0, min(blink, 1))\n fillColor.withAlphaComponent(alpha).setFill()\n let blinkR = sr * clamped\n let blinkInnerR = blinkR * 0.3\n\n func drawBlinkAsterisk(cx: CGFloat, cy: CGFloat) {\n let path = NSBezierPath()\n for i in 0..<16 {\n let angle = CGFloat(i) * .pi / 8 - .pi / 2\n let radius = (i % 2 == 0) ? blinkR : blinkInnerR\n let px = cx + cos(angle) * radius\n let py = cy + sin(angle) * radius\n if i == 0 {\n path.move(to: NSPoint(x: px, y: py))\n } else {\n path.line(to: NSPoint(x: px, y: py))\n }\n }\n path.close()\n path.fill()\n }\n\n drawBlinkAsterisk(cx: ldCx, cy: yCy)\n drawBlinkAsterisk(cx: rdCx, cy: yCy)\n }\n }\n\n // Warp twist: \"Warp\" style face with tilted-eye cutouts.\n if addWarpTwist {\n let ctx = NSGraphicsContext.current?.cgContext\n let centerXPx = rectPx.midXPx\n let eyeCenterYPx = rectPx.y + rectPx.h / 2\n\n ctx?.saveGState()\n ctx?.setShouldAntialias(true) // Smooth edges for tilted ellipse eyes\n\n // 1. Draw Eyes (Tilted ellipse cutouts - \"fox eye\" / \"cat eye\" style)\n // Keep sizes in integer pixels so grid conversion stays exact.\n let eyeWidthPx = 5\n let eyeHeightPx = 8\n let eyeOffsetPx = 7\n let eyeTiltAngle: CGFloat = .pi / 3 // 60 degrees tilt\n\n let leftEyeCx = Self.grid.pt(centerXPx) - Self.grid.pt(eyeOffsetPx)\n let rightEyeCx = Self.grid.pt(centerXPx) + Self.grid.pt(eyeOffsetPx)\n let eyeCy = Self.grid.pt(eyeCenterYPx)\n let eyeW = Self.grid.pt(eyeWidthPx)\n let eyeH = Self.grid.pt(eyeHeightPx)\n\n /// Draw a tilted ellipse eye at the given center.\n func drawTiltedEyeCutout(cx: CGFloat, cy: CGFloat, tiltAngle: CGFloat) {\n guard let ctx else { return }\n let eyeRect = CGRect(x: -eyeW / 2, y: -eyeH / 2, width: eyeW, height: eyeH)\n\n // Use CGContext transforms instead of AffineTransform-on-path so the rotation origin\n // is unambiguous and the current blend mode is consistently respected.\n ctx.saveGState()\n ctx.translateBy(x: cx, y: cy)\n ctx.rotate(by: tiltAngle)\n ctx.addEllipse(in: eyeRect)\n ctx.fillPath()\n ctx.restoreGState()\n }\n\n if warpEyesFilled {\n fillColor.withAlphaComponent(alpha).setFill()\n drawTiltedEyeCutout(cx: leftEyeCx, cy: eyeCy, tiltAngle: eyeTiltAngle)\n drawTiltedEyeCutout(cx: rightEyeCx, cy: eyeCy, tiltAngle: -eyeTiltAngle)\n } else {\n // Clear eyes using blend mode\n ctx?.setBlendMode(.clear)\n drawTiltedEyeCutout(cx: leftEyeCx, cy: eyeCy, tiltAngle: eyeTiltAngle)\n drawTiltedEyeCutout(cx: rightEyeCx, cy: eyeCy, tiltAngle: -eyeTiltAngle)\n ctx?.setBlendMode(.normal)\n }\n ctx?.restoreGState() // Restore graphics state\n }\n }\n\n let effectiveWeeklyRemaining: Double? = {\n if style == .warp, let weeklyRemaining, weeklyRemaining <= 0 {\n return nil\n }\n return weeklyRemaining\n }()\n let topValue = primaryRemaining\n let bottomValue = effectiveWeeklyRemaining\n let creditsRatio = creditsRemaining.map { min($0 / Self.creditsCap * 100, 100) }\n\n let hasWeekly = (bottomValue != nil)\n let weeklyAvailable = hasWeekly && (bottomValue ?? 0) > 0\n let creditsAlpha: CGFloat = 1.0\n let topRectPx = RectPx(x: barXPx, y: 19, w: barWidthPx, h: 12)\n let bottomRectPx = RectPx(x: barXPx, y: 5, w: barWidthPx, h: 8)\n let creditsRectPx = RectPx(x: barXPx, y: 14, w: barWidthPx, h: 16)\n let creditsBottomRectPx = RectPx(x: barXPx, y: 4, w: barWidthPx, h: 6)\n\n // Warp special case: when no bonus or bonus exhausted, show \"top monthly, bottom dimmed\"\n let warpNoBonus = style == .warp && !weeklyAvailable\n\n if weeklyAvailable {\n // Normal: top=primary, bottom=secondary (bonus/weekly).\n drawBar(\n rectPx: topRectPx,\n remaining: topValue,\n addNotches: style == .claude,\n addFace: style == .codex,\n addGeminiTwist: style == .gemini || style == .antigravity,\n addAntigravityTwist: style == .antigravity,\n addFactoryTwist: style == .factory,\n addWarpTwist: style == .warp,\n blink: blink)\n drawBar(rectPx: bottomRectPx, remaining: bottomValue)\n } else if !hasWeekly || warpNoBonus {\n if style == .warp {\n // Warp: no bonus or bonus exhausted -> top=monthly credits, bottom=dimmed track\n drawBar(\n rectPx: topRectPx,\n remaining: topValue,\n addWarpTwist: true,\n blink: blink)\n drawBar(rectPx: bottomRectPx, remaining: nil, alpha: 0.45)\n } else {\n // Weekly missing (e.g. Claude enterprise): keep normal layout but\n // dim the bottom track to indicate N/A.\n if topValue == nil, let ratio = creditsRatio {\n // Credits-only: show credits prominently (e.g. credits loaded before usage).\n drawBar(\n rectPx: creditsRectPx,\n remaining: ratio,\n alpha: creditsAlpha,\n addNotches: style == .claude,\n addFace: style == .codex,\n addGeminiTwist: style == .gemini || style == .antigravity,\n addAntigravityTwist: style == .antigravity,\n addFactoryTwist: style == .factory,\n addWarpTwist: style == .warp,\n blink: blink)\n drawBar(rectPx: creditsBottomRectPx, remaining: nil, alpha: 0.45)\n } else {\n drawBar(\n rectPx: topRectPx,\n remaining: topValue,\n addNotches: style == .claude,\n addFace: style == .codex,\n addGeminiTwist: style == .gemini || style == .antigravity,\n addAntigravityTwist: style == .antigravity,\n addFactoryTwist: style == .factory,\n addWarpTwist: style == .warp,\n blink: blink)\n drawBar(rectPx: bottomRectPx, remaining: nil, alpha: 0.45)\n }\n }\n } else {\n // Weekly exhausted/missing: show credits on top (thicker), weekly (likely 0) on bottom.\n if let ratio = creditsRatio {\n drawBar(\n rectPx: creditsRectPx,\n remaining: ratio,\n alpha: creditsAlpha,\n addNotches: style == .claude,\n addFace: style == .codex,\n addGeminiTwist: style == .gemini || style == .antigravity,\n addAntigravityTwist: style == .antigravity,\n addFactoryTwist: style == .factory,\n addWarpTwist: style == .warp,\n blink: blink)\n } else {\n // No credits available; fall back to 5h if present.\n drawBar(\n rectPx: topRectPx,\n remaining: topValue,\n addNotches: style == .claude,\n addFace: style == .codex,\n addGeminiTwist: style == .gemini || style == .antigravity,\n addAntigravityTwist: style == .antigravity,\n addFactoryTwist: style == .factory,\n addWarpTwist: style == .warp,\n blink: blink)\n }\n drawBar(rectPx: creditsBottomRectPx, remaining: bottomValue)\n }\n\n Self.drawStatusOverlay(indicator: statusIndicator)\n }\n }\n\n if shouldCache {\n let key = IconCacheKey(\n primary: self.quantizedPercent(primaryRemaining),\n weekly: self.quantizedPercent(weeklyRemaining),\n credits: self.quantizedCredits(creditsRemaining),\n stale: stale,\n style: self.styleKey(style),\n indicator: self.indicatorKey(statusIndicator))\n if let cached = self.cachedIcon(for: key) {\n return cached\n }\n let image = render()\n self.storeIcon(image, for: key)\n return image\n }\n\n return render()\n }\n\n // swiftlint:enable function_body_length\n\n /// Morph helper: unbraids a simplified knot into our bar icon.\n static func makeMorphIcon(progress: Double, style: IconStyle) -> NSImage {\n let clamped = max(0, min(progress, 1))\n let key = self.morphCacheKey(progress: clamped, style: style)\n if let cached = self.morphCache.image(for: key) {\n return cached\n }\n let image = self.renderImage {\n self.drawUnbraidMorph(t: clamped, style: style)\n }\n self.morphCache.set(image, for: key)\n return image\n }\n\n private static func quantizedPercent(_ value: Double?) -> Int {\n guard let value else { return -1 }\n return Int((value * 10).rounded())\n }\n\n private static func quantizedCredits(_ value: Double?) -> Int {\n guard let value else { return -1 }\n let clamped = max(0, min(value, self.creditsCap))\n return Int((clamped * 10).rounded())\n }\n\n private static let styleKeyLookup: [IconStyle: Int] = {\n var lookup: [IconStyle: Int] = [:]\n for (index, style) in IconStyle.allCases.enumerated() {\n lookup[style] = index\n }\n return lookup\n }()\n\n private static func styleKey(_ style: IconStyle) -> Int {\n self.styleKeyLookup[style] ?? 0\n }\n\n private static func indicatorKey(_ indicator: ProviderStatusIndicator) -> Int {\n switch indicator {\n case .none: 0\n case .minor: 1\n case .major: 2\n case .critical: 3\n case .maintenance: 4\n case .unknown: 5\n }\n }\n\n private static func morphCacheKey(progress: Double, style: IconStyle) -> NSNumber {\n let bucket = Int((progress * Double(self.morphBucketCount)).rounded())\n let key = self.styleKey(style) * 1000 + bucket\n return NSNumber(value: key)\n }\n\n private static func cachedIcon(for key: IconCacheKey) -> NSImage? {\n self.iconCacheStore.cachedIcon(for: key)\n }\n\n private static func storeIcon(_ image: NSImage, for key: IconCacheKey) {\n self.iconCacheStore.storeIcon(image, for: key, limit: self.iconCacheLimit)\n }\n\n private static func drawUnbraidMorph(t: Double, style: IconStyle) {\n let t = CGFloat(max(0, min(t, 1)))\n let size = Self.baseSize\n let center = CGPoint(x: size.width / 2, y: size.height / 2)\n let baseColor = NSColor.labelColor\n\n struct Segment {\n let startCenter: CGPoint\n let endCenter: CGPoint\n let startAngle: CGFloat\n let endAngle: CGFloat\n let startLength: CGFloat\n let endLength: CGFloat\n let startThickness: CGFloat\n let endThickness: CGFloat\n let fadeOut: Bool\n }\n\n let segments: [Segment] = [\n // Upper ribbon -> top bar\n .init(\n startCenter: center.offset(dx: 0, dy: 2),\n endCenter: CGPoint(x: center.x, y: 9.0),\n startAngle: -30,\n endAngle: 0,\n startLength: 16,\n endLength: 14,\n startThickness: 3.4,\n endThickness: 3.0,\n fadeOut: false),\n // Lower ribbon -> bottom bar\n .init(\n startCenter: center.offset(dx: 0, dy: -2),\n endCenter: CGPoint(x: center.x, y: 4.0),\n startAngle: 210,\n endAngle: 0,\n startLength: 16,\n endLength: 12,\n startThickness: 3.4,\n endThickness: 2.4,\n fadeOut: false),\n // Side ribbon fades away\n .init(\n startCenter: center,\n endCenter: center.offset(dx: 0, dy: 6),\n startAngle: 90,\n endAngle: 0,\n startLength: 16,\n endLength: 8,\n startThickness: 3.4,\n endThickness: 1.8,\n fadeOut: true),\n ]\n\n for seg in segments {\n let p = seg.fadeOut ? t * 1.1 : t\n let c = seg.startCenter.lerp(to: seg.endCenter, p: p)\n let angle = seg.startAngle.lerp(to: seg.endAngle, p: p)\n let length = seg.startLength.lerp(to: seg.endLength, p: p)\n let thickness = seg.startThickness.lerp(to: seg.endThickness, p: p)\n let alpha = seg.fadeOut ? (1 - p) : 1\n\n self.drawRoundedRibbon(\n center: c,\n length: length,\n thickness: thickness,\n angle: angle,\n color: baseColor.withAlphaComponent(alpha))\n }\n\n // Cross-fade in bar fill emphasis near the end of the morph.\n if t > 0.55 {\n let barT = (t - 0.55) / 0.45\n let bars = self.makeIcon(\n primaryRemaining: 100,\n weeklyRemaining: 100,\n creditsRemaining: nil,\n stale: false,\n style: style)\n bars.draw(in: CGRect(origin: .zero, size: size), from: .zero, operation: .sourceOver, fraction: barT)\n }\n }\n\n private static func drawRoundedRibbon(\n center: CGPoint,\n length: CGFloat,\n thickness: CGFloat,\n angle: CGFloat,\n color: NSColor)\n {\n var transform = AffineTransform.identity\n transform.translate(x: center.x, y: center.y)\n transform.rotate(byDegrees: angle)\n transform.translate(x: -center.x, y: -center.y)\n\n let rect = CGRect(\n x: center.x - length / 2,\n y: center.y - thickness / 2,\n width: length,\n height: thickness)\n\n let path = NSBezierPath(roundedRect: rect, xRadius: thickness / 2, yRadius: thickness / 2)\n path.transform(using: transform)\n color.setFill()\n path.fill()\n }\n\n private static func drawStatusOverlay(indicator: ProviderStatusIndicator) {\n guard indicator.hasIssue else { return }\n let color = NSColor.labelColor\n\n switch indicator {\n case .minor, .maintenance:\n let size: CGFloat = 4\n let rect = Self.snapRect(\n x: Self.baseSize.width - size - 2,\n y: 2,\n width: size,\n height: size)\n let path = NSBezierPath(ovalIn: rect)\n color.setFill()\n path.fill()\n case .major, .critical, .unknown:\n let lineRect = Self.snapRect(\n x: Self.baseSize.width - 6,\n y: 4,\n width: 2.0,\n height: 6)\n let linePath = NSBezierPath(roundedRect: lineRect, xRadius: 1, yRadius: 1)\n color.setFill()\n linePath.fill()\n\n let dotRect = Self.snapRect(\n x: Self.baseSize.width - 6,\n y: 2,\n width: 2.0,\n height: 2.0)\n NSBezierPath(ovalIn: dotRect).fill()\n case .none:\n break\n }\n }\n\n private static func withScaledContext(_ draw: () -> Void) {\n guard let ctx = NSGraphicsContext.current?.cgContext else {\n draw()\n return\n }\n ctx.saveGState()\n ctx.setShouldAntialias(true)\n ctx.interpolationQuality = .none\n draw()\n ctx.restoreGState()\n }\n\n private static func snap(_ value: CGFloat) -> CGFloat {\n (value * self.outputScale).rounded() / self.outputScale\n }\n\n private static func snapRect(x: CGFloat, y: CGFloat, width: CGFloat, height: CGFloat) -> CGRect {\n CGRect(x: self.snap(x), y: self.snap(y), width: self.snap(width), height: self.snap(height))\n }\n\n private static func renderImage(_ draw: () -> Void) -> NSImage {\n let image = NSImage(size: Self.outputSize)\n\n if let rep = NSBitmapImageRep(\n bitmapDataPlanes: nil,\n pixelsWide: Int(Self.outputSize.width * Self.outputScale),\n pixelsHigh: Int(Self.outputSize.height * Self.outputScale),\n bitsPerSample: 8,\n samplesPerPixel: 4,\n hasAlpha: true,\n isPlanar: false,\n colorSpaceName: .deviceRGB,\n bytesPerRow: 0,\n bitsPerPixel: 0)\n {\n rep.size = Self.outputSize // points\n image.addRepresentation(rep)\n\n NSGraphicsContext.saveGraphicsState()\n if let ctx = NSGraphicsContext(bitmapImageRep: rep) {\n NSGraphicsContext.current = ctx\n Self.withScaledContext(draw)\n }\n NSGraphicsContext.restoreGraphicsState()\n } else {\n // Fallback to legacy focus if the bitmap rep fails for any reason.\n image.lockFocus()\n Self.withScaledContext(draw)\n image.unlockFocus()\n }\n\n image.isTemplate = true\n return image\n }\n}\n\nextension CGPoint {\n fileprivate func lerp(to other: CGPoint, p: CGFloat) -> CGPoint {\n CGPoint(x: self.x + (other.x - self.x) * p, y: self.y + (other.y - self.y) * p)\n }\n\n fileprivate func offset(dx: CGFloat, dy: CGFloat) -> CGPoint {\n CGPoint(x: self.x + dx, y: self.y + dy)\n }\n}\n\nextension CGFloat {\n fileprivate func lerp(to other: CGFloat, p: CGFloat) -> CGFloat {\n self + (other - self) * p\n }\n}\n" }, { "path": "Sources/CodexBar/IconView.swift", "content": "import CodexBarCore\nimport SwiftUI\n\n@MainActor\nstruct IconView: View {\n let snapshot: UsageSnapshot?\n let creditsRemaining: Double?\n let isStale: Bool\n let showLoadingAnimation: Bool\n let style: IconStyle\n @State private var phase: CGFloat = 0\n @State private var displayLink = DisplayLinkDriver()\n @State private var pattern: LoadingPattern = .knightRider\n @State private var debugCycle = false\n @State private var cycleIndex = 0\n @State private var cycleCounter = 0\n private let loadingFPS: Double = 12\n // Advance to next pattern every N ticks when debug cycling.\n private let cycleIntervalTicks = 20\n private let patterns = LoadingPattern.allCases\n\n private var isLoading: Bool {\n self.showLoadingAnimation && self.snapshot == nil\n }\n\n var body: some View {\n Group {\n if let snapshot {\n Image(nsImage: IconRenderer.makeIcon(\n primaryRemaining: snapshot.primary?.remainingPercent,\n weeklyRemaining: snapshot.secondary?.remainingPercent,\n creditsRemaining: self.creditsRemaining,\n stale: self.isStale,\n style: self.style))\n .renderingMode(.original)\n .interpolation(.none)\n .frame(width: 20, height: 18, alignment: .center)\n .padding(.horizontal, 2)\n } else if self.showLoadingAnimation {\n // Loading: animate bars with the current pattern until data arrives.\n Image(nsImage: self.loadingImage)\n .renderingMode(.original)\n .interpolation(.none)\n .frame(width: 20, height: 18, alignment: .center)\n .padding(.horizontal, 2)\n .onChange(of: self.displayLink.tick) { _, _ in\n self.phase += 0.09 // half-speed animation\n if self.debugCycle {\n self.cycleCounter += 1\n if self.cycleCounter >= self.cycleIntervalTicks {\n self.cycleCounter = 0\n self.cycleIndex = (self.cycleIndex + 1) % self.patterns.count\n self.pattern = self.patterns[self.cycleIndex]\n }\n }\n }\n } else {\n // No animation when usage/account is unavailable; show empty tracks.\n Image(nsImage: IconRenderer.makeIcon(\n primaryRemaining: nil,\n weeklyRemaining: nil,\n creditsRemaining: self.creditsRemaining,\n stale: self.isStale,\n style: self.style))\n .renderingMode(.original)\n .interpolation(.none)\n .frame(width: 20, height: 18, alignment: .center)\n .padding(.horizontal, 2)\n }\n }\n .onChange(of: self.isLoading, initial: true) { _, isLoading in\n if isLoading {\n self.displayLink.start(fps: self.loadingFPS)\n if !self.debugCycle {\n self.pattern = self.patterns.randomElement() ?? .knightRider\n }\n } else {\n self.displayLink.stop()\n self.debugCycle = false\n self.phase = 0\n }\n }\n .onDisappear { self.displayLink.stop() }\n .onReceive(NotificationCenter.default.publisher(for: .codexbarDebugReplayAllAnimations)) { notification in\n if let raw = notification.userInfo?[\"pattern\"] as? String,\n let selected = LoadingPattern(rawValue: raw)\n {\n self.debugCycle = false\n self.pattern = selected\n self.cycleIndex = self.patterns.firstIndex(of: selected) ?? 0\n } else {\n self.debugCycle = true\n self.cycleIndex = 0\n self.pattern = self.patterns.first ?? .knightRider\n }\n self.cycleCounter = 0\n self.phase = 0\n }\n }\n\n private var loadingPrimary: Double {\n self.pattern.value(phase: Double(self.phase))\n }\n\n private var loadingSecondary: Double {\n self.pattern.value(phase: Double(self.phase + self.pattern.secondaryOffset))\n }\n\n private var loadingImage: NSImage {\n if self.pattern == .unbraid {\n let progress = self.loadingPrimary / 100\n return IconRenderer.makeMorphIcon(progress: progress, style: self.style)\n } else {\n return IconRenderer.makeIcon(\n primaryRemaining: self.loadingPrimary,\n weeklyRemaining: self.loadingSecondary,\n creditsRemaining: nil,\n stale: false,\n style: self.style)\n }\n }\n}\n" }, { "path": "Sources/CodexBar/InstallOrigin.swift", "content": "import Foundation\n\nenum InstallOrigin {\n static func isHomebrewCask(appBundleURL: URL) -> Bool {\n let resolved = appBundleURL.resolvingSymlinksInPath()\n let path = resolved.path\n return path.contains(\"/Caskroom/\") || path.contains(\"/Homebrew/Caskroom/\")\n }\n}\n" }, { "path": "Sources/CodexBar/KeyboardShortcuts+Names.swift", "content": "import KeyboardShortcuts\n\n@MainActor\nextension KeyboardShortcuts.Name {\n static let openMenu = Self(\"openMenu\")\n}\n" }, { "path": "Sources/CodexBar/KeychainMigration.swift", "content": "import CodexBarCore\nimport Foundation\nimport Security\n\n/// Migrates keychain items to use kSecAttrAccessibleAfterFirstUnlockThisDeviceOnly\n/// to prevent permission prompts on every rebuild during development.\nenum KeychainMigration {\n private static let log = CodexBarLog.logger(LogCategories.keychainMigration)\n private static let migrationKey = \"KeychainMigrationV1Completed\"\n\n struct MigrationItem: Hashable {\n let service: String\n let account: String?\n\n var label: String {\n let accountLabel = self.account ?? \"\"\n return \"\\(self.service):\\(accountLabel)\"\n }\n }\n\n static let itemsToMigrate: [MigrationItem] = [\n MigrationItem(service: \"com.steipete.CodexBar\", account: \"codex-cookie\"),\n MigrationItem(service: \"com.steipete.CodexBar\", account: \"claude-cookie\"),\n MigrationItem(service: \"com.steipete.CodexBar\", account: \"cursor-cookie\"),\n MigrationItem(service: \"com.steipete.CodexBar\", account: \"factory-cookie\"),\n MigrationItem(service: \"com.steipete.CodexBar\", account: \"minimax-cookie\"),\n MigrationItem(service: \"com.steipete.CodexBar\", account: \"minimax-api-token\"),\n MigrationItem(service: \"com.steipete.CodexBar\", account: \"augment-cookie\"),\n MigrationItem(service: \"com.steipete.CodexBar\", account: \"copilot-api-token\"),\n MigrationItem(service: \"com.steipete.CodexBar\", account: \"zai-api-token\"),\n MigrationItem(service: \"com.steipete.CodexBar\", account: \"synthetic-api-key\"),\n ]\n\n /// Run migration once per installation\n static func migrateIfNeeded() {\n guard !KeychainAccessGate.isDisabled else {\n self.log.info(\"Keychain access disabled; skipping migration\")\n return\n }\n\n if !UserDefaults.standard.bool(forKey: self.migrationKey) {\n self.log.info(\"Starting keychain migration to reduce permission prompts\")\n\n var migratedCount = 0\n var errorCount = 0\n\n for item in self.itemsToMigrate {\n do {\n if try self.migrateItem(item) {\n migratedCount += 1\n }\n } catch {\n errorCount += 1\n self.log.error(\"Failed to migrate \\(item.label): \\(String(describing: error))\")\n }\n }\n\n self.log.info(\"Keychain migration complete: \\(migratedCount) migrated, \\(errorCount) errors\")\n UserDefaults.standard.set(true, forKey: self.migrationKey)\n\n if migratedCount > 0 {\n self.log.info(\"✅ Future rebuilds will not prompt for keychain access\")\n }\n } else {\n self.log.debug(\"Keychain migration already completed, skipping\")\n }\n }\n\n /// Migrate a single keychain item to the new accessibility level\n /// Returns true if item was migrated, false if item didn't exist\n private static func migrateItem(_ item: MigrationItem) throws -> Bool {\n // First, try to read the existing item\n var result: CFTypeRef?\n var query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: item.service,\n kSecMatchLimit as String: kSecMatchLimitOne,\n kSecReturnData as String: true,\n kSecReturnAttributes as String: true,\n ]\n if let account = item.account {\n query[kSecAttrAccount as String] = account\n }\n\n let status = SecItemCopyMatching(query as CFDictionary, &result)\n\n if status == errSecItemNotFound {\n // Item doesn't exist, nothing to migrate\n return false\n }\n\n guard status == errSecSuccess else {\n throw KeychainMigrationError.readFailed(status)\n }\n\n guard let rawItem = result as? [String: Any],\n let data = rawItem[kSecValueData as String] as? Data,\n let accessible = rawItem[kSecAttrAccessible as String] as? String\n else {\n throw KeychainMigrationError.invalidItemFormat\n }\n\n // Check if already using the correct accessibility\n if accessible == (kSecAttrAccessibleAfterFirstUnlockThisDeviceOnly as String) {\n self.log.debug(\"\\(item.label) already using correct accessibility\")\n return false\n }\n\n // Delete the old item\n var deleteQuery: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: item.service,\n ]\n if let account = item.account {\n deleteQuery[kSecAttrAccount as String] = account\n }\n\n let deleteStatus = SecItemDelete(deleteQuery as CFDictionary)\n guard deleteStatus == errSecSuccess else {\n throw KeychainMigrationError.deleteFailed(deleteStatus)\n }\n\n // Add it back with the new accessibility\n var addQuery: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: item.service,\n kSecValueData as String: data,\n kSecAttrAccessible as String: kSecAttrAccessibleAfterFirstUnlockThisDeviceOnly,\n ]\n if let account = item.account {\n addQuery[kSecAttrAccount as String] = account\n }\n\n let addStatus = SecItemAdd(addQuery as CFDictionary, nil)\n guard addStatus == errSecSuccess else {\n throw KeychainMigrationError.addFailed(addStatus)\n }\n\n self.log.info(\"Migrated \\(item.label) to new accessibility level\")\n return true\n }\n\n /// Reset migration flag (for testing)\n static func resetMigrationFlag() {\n UserDefaults.standard.removeObject(forKey: self.migrationKey)\n }\n}\n\nenum KeychainMigrationError: Error {\n case readFailed(OSStatus)\n case deleteFailed(OSStatus)\n case addFailed(OSStatus)\n case invalidItemFormat\n}\n" }, { "path": "Sources/CodexBar/KeychainPromptCoordinator.swift", "content": "import AppKit\nimport CodexBarCore\nimport SweetCookieKit\n\nenum KeychainPromptCoordinator {\n private static let promptLock = NSLock()\n private static let log = CodexBarLog.logger(LogCategories.keychainPrompt)\n\n static func install() {\n KeychainPromptHandler.handler = { context in\n self.presentKeychainPrompt(context)\n }\n BrowserCookieKeychainPromptHandler.handler = { context in\n self.presentBrowserCookiePrompt(context)\n }\n }\n\n private static func presentKeychainPrompt(_ context: KeychainPromptContext) {\n let (title, message) = self.keychainCopy(for: context)\n self.log.info(\"Keychain prompt requested\", metadata: [\"kind\": \"\\(context.kind)\"])\n self.presentAlert(title: title, message: message)\n }\n\n private static func presentBrowserCookiePrompt(_ context: BrowserCookieKeychainPromptContext) {\n let title = \"Keychain Access Required\"\n let message = [\n \"CodexBar will ask macOS Keychain for “\\(context.label)” so it can decrypt browser cookies\",\n \"and authenticate your account. Click OK to continue.\",\n ].joined(separator: \" \")\n self.log.info(\"Browser cookie keychain prompt requested\", metadata: [\"label\": context.label])\n self.presentAlert(title: title, message: message)\n }\n\n private static func keychainCopy(for context: KeychainPromptContext) -> (title: String, message: String) {\n let title = \"Keychain Access Required\"\n switch context.kind {\n case .claudeOAuth:\n return (title, [\n \"CodexBar will ask macOS Keychain for the Claude Code OAuth token\",\n \"so it can fetch your Claude usage. Click OK to continue.\",\n ].joined(separator: \" \"))\n case .codexCookie:\n return (title, [\n \"CodexBar will ask macOS Keychain for your OpenAI cookie header\",\n \"so it can fetch Codex dashboard extras. Click OK to continue.\",\n ].joined(separator: \" \"))\n case .claudeCookie:\n return (title, [\n \"CodexBar will ask macOS Keychain for your Claude cookie header\",\n \"so it can fetch Claude web usage. Click OK to continue.\",\n ].joined(separator: \" \"))\n case .cursorCookie:\n return (title, [\n \"CodexBar will ask macOS Keychain for your Cursor cookie header\",\n \"so it can fetch usage. Click OK to continue.\",\n ].joined(separator: \" \"))\n case .opencodeCookie:\n return (title, [\n \"CodexBar will ask macOS Keychain for your OpenCode cookie header\",\n \"so it can fetch usage. Click OK to continue.\",\n ].joined(separator: \" \"))\n case .factoryCookie:\n return (title, [\n \"CodexBar will ask macOS Keychain for your Factory cookie header\",\n \"so it can fetch usage. Click OK to continue.\",\n ].joined(separator: \" \"))\n case .zaiToken:\n return (title, [\n \"CodexBar will ask macOS Keychain for your z.ai API token\",\n \"so it can fetch usage. Click OK to continue.\",\n ].joined(separator: \" \"))\n case .syntheticToken:\n return (title, [\n \"CodexBar will ask macOS Keychain for your Synthetic API key\",\n \"so it can fetch usage. Click OK to continue.\",\n ].joined(separator: \" \"))\n case .copilotToken:\n return (title, [\n \"CodexBar will ask macOS Keychain for your GitHub Copilot token\",\n \"so it can fetch usage. Click OK to continue.\",\n ].joined(separator: \" \"))\n case .kimiToken:\n return (title, [\n \"CodexBar will ask macOS Keychain for your Kimi auth token\",\n \"so it can fetch usage. Click OK to continue.\",\n ].joined(separator: \" \"))\n case .kimiK2Token:\n return (title, [\n \"CodexBar will ask macOS Keychain for your Kimi K2 API key\",\n \"so it can fetch usage. Click OK to continue.\",\n ].joined(separator: \" \"))\n case .minimaxCookie:\n return (title, [\n \"CodexBar will ask macOS Keychain for your MiniMax cookie header\",\n \"so it can fetch usage. Click OK to continue.\",\n ].joined(separator: \" \"))\n case .minimaxToken:\n return (title, [\n \"CodexBar will ask macOS Keychain for your MiniMax API token\",\n \"so it can fetch usage. Click OK to continue.\",\n ].joined(separator: \" \"))\n case .augmentCookie:\n return (title, [\n \"CodexBar will ask macOS Keychain for your Augment cookie header\",\n \"so it can fetch usage. Click OK to continue.\",\n ].joined(separator: \" \"))\n case .ampCookie:\n return (title, [\n \"CodexBar will ask macOS Keychain for your Amp cookie header\",\n \"so it can fetch usage. Click OK to continue.\",\n ].joined(separator: \" \"))\n }\n }\n\n private static func presentAlert(title: String, message: String) {\n self.promptLock.lock()\n defer { self.promptLock.unlock() }\n\n if Thread.isMainThread {\n MainActor.assumeIsolated {\n self.showAlert(title: title, message: message)\n }\n return\n }\n DispatchQueue.main.sync {\n MainActor.assumeIsolated {\n self.showAlert(title: title, message: message)\n }\n }\n }\n\n @MainActor\n private static func showAlert(title: String, message: String) {\n let alert = NSAlert()\n alert.messageText = title\n alert.informativeText = message\n alert.addButton(withTitle: \"OK\")\n _ = alert.runModal()\n }\n}\n" }, { "path": "Sources/CodexBar/KimiK2TokenStore.swift", "content": "import CodexBarCore\nimport Foundation\nimport Security\n\nprotocol KimiK2TokenStoring: Sendable {\n func loadToken() throws -> String?\n func storeToken(_ token: String?) throws\n}\n\nenum KimiK2TokenStoreError: LocalizedError {\n case keychainStatus(OSStatus)\n case invalidData\n\n var errorDescription: String? {\n switch self {\n case let .keychainStatus(status):\n \"Keychain error: \\(status)\"\n case .invalidData:\n \"Keychain returned invalid data.\"\n }\n }\n}\n\nstruct KeychainKimiK2TokenStore: KimiK2TokenStoring {\n private static let log = CodexBarLog.logger(LogCategories.kimiK2TokenStore)\n\n private let service = \"com.steipete.CodexBar\"\n private let account = \"kimi-k2-api-token\"\n\n func loadToken() throws -> String? {\n var result: CFTypeRef?\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n kSecMatchLimit as String: kSecMatchLimitOne,\n kSecReturnData as String: true,\n ]\n\n if case .interactionRequired = KeychainAccessPreflight\n .checkGenericPassword(service: self.service, account: self.account)\n {\n KeychainPromptHandler.handler?(KeychainPromptContext(\n kind: .kimiK2Token,\n service: self.service,\n account: self.account))\n }\n\n let status = SecItemCopyMatching(query as CFDictionary, &result)\n if status == errSecItemNotFound {\n return nil\n }\n guard status == errSecSuccess else {\n Self.log.error(\"Keychain read failed: \\(status)\")\n throw KimiK2TokenStoreError.keychainStatus(status)\n }\n\n guard let data = result as? Data else {\n throw KimiK2TokenStoreError.invalidData\n }\n let token = String(data: data, encoding: .utf8)?.trimmingCharacters(in: .whitespacesAndNewlines)\n if let token, !token.isEmpty {\n return token\n }\n return nil\n }\n\n func storeToken(_ token: String?) throws {\n let cleaned = token?.trimmingCharacters(in: .whitespacesAndNewlines)\n if cleaned == nil || cleaned?.isEmpty == true {\n try self.deleteTokenIfPresent()\n return\n }\n\n let data = cleaned!.data(using: .utf8)!\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n ]\n let attributes: [String: Any] = [\n kSecValueData as String: data,\n kSecAttrAccessible as String: kSecAttrAccessibleAfterFirstUnlockThisDeviceOnly,\n ]\n\n let updateStatus = SecItemUpdate(query as CFDictionary, attributes as CFDictionary)\n if updateStatus == errSecSuccess {\n return\n }\n if updateStatus != errSecItemNotFound {\n Self.log.error(\"Keychain update failed: \\(updateStatus)\")\n throw KimiK2TokenStoreError.keychainStatus(updateStatus)\n }\n\n var addQuery = query\n for (key, value) in attributes {\n addQuery[key] = value\n }\n let addStatus = SecItemAdd(addQuery as CFDictionary, nil)\n guard addStatus == errSecSuccess else {\n Self.log.error(\"Keychain add failed: \\(addStatus)\")\n throw KimiK2TokenStoreError.keychainStatus(addStatus)\n }\n }\n\n private func deleteTokenIfPresent() throws {\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n ]\n let status = SecItemDelete(query as CFDictionary)\n if status == errSecSuccess || status == errSecItemNotFound {\n return\n }\n Self.log.error(\"Keychain delete failed: \\(status)\")\n throw KimiK2TokenStoreError.keychainStatus(status)\n }\n}\n" }, { "path": "Sources/CodexBar/KimiTokenStore.swift", "content": "import CodexBarCore\nimport Foundation\nimport Security\n\nprotocol KimiTokenStoring: Sendable {\n func loadToken() throws -> String?\n func storeToken(_ token: String?) throws\n}\n\nenum KimiTokenStoreError: LocalizedError {\n case keychainStatus(OSStatus)\n case invalidData\n\n var errorDescription: String? {\n switch self {\n case let .keychainStatus(status):\n \"Keychain error: \\(status)\"\n case .invalidData:\n \"Keychain returned invalid data.\"\n }\n }\n}\n\nstruct KeychainKimiTokenStore: KimiTokenStoring {\n private static let log = CodexBarLog.logger(LogCategories.kimiTokenStore)\n\n private let service = \"com.steipete.CodexBar\"\n private let account = \"kimi-auth-token\"\n\n func loadToken() throws -> String? {\n guard !KeychainAccessGate.isDisabled else {\n Self.log.debug(\"Keychain access disabled; skipping token load\")\n return nil\n }\n var result: CFTypeRef?\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n kSecMatchLimit as String: kSecMatchLimitOne,\n kSecReturnData as String: true,\n ]\n\n if case .interactionRequired = KeychainAccessPreflight\n .checkGenericPassword(service: self.service, account: self.account)\n {\n KeychainPromptHandler.handler?(KeychainPromptContext(\n kind: .kimiToken,\n service: self.service,\n account: self.account))\n }\n\n let status = SecItemCopyMatching(query as CFDictionary, &result)\n if status == errSecItemNotFound {\n return nil\n }\n guard status == errSecSuccess else {\n Self.log.error(\"Keychain read failed: \\(status)\")\n throw KimiTokenStoreError.keychainStatus(status)\n }\n\n guard let data = result as? Data else {\n throw KimiTokenStoreError.invalidData\n }\n let token = String(data: data, encoding: .utf8)?.trimmingCharacters(in: .whitespacesAndNewlines)\n if let token, !token.isEmpty {\n return token\n }\n return nil\n }\n\n func storeToken(_ token: String?) throws {\n guard !KeychainAccessGate.isDisabled else {\n Self.log.debug(\"Keychain access disabled; skipping token store\")\n return\n }\n let cleaned = token?.trimmingCharacters(in: .whitespacesAndNewlines)\n if cleaned == nil || cleaned?.isEmpty == true {\n try self.deleteTokenIfPresent()\n return\n }\n\n let data = cleaned!.data(using: .utf8)!\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n ]\n let attributes: [String: Any] = [\n kSecValueData as String: data,\n kSecAttrAccessible as String: kSecAttrAccessibleAfterFirstUnlockThisDeviceOnly,\n ]\n\n let updateStatus = SecItemUpdate(query as CFDictionary, attributes as CFDictionary)\n if updateStatus == errSecSuccess {\n return\n }\n if updateStatus != errSecItemNotFound {\n Self.log.error(\"Keychain update failed: \\(updateStatus)\")\n throw KimiTokenStoreError.keychainStatus(updateStatus)\n }\n\n var addQuery = query\n for (key, value) in attributes {\n addQuery[key] = value\n }\n let addStatus = SecItemAdd(addQuery as CFDictionary, nil)\n guard addStatus == errSecSuccess else {\n Self.log.error(\"Keychain add failed: \\(addStatus)\")\n throw KimiTokenStoreError.keychainStatus(addStatus)\n }\n }\n\n private func deleteTokenIfPresent() throws {\n guard !KeychainAccessGate.isDisabled else { return }\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n ]\n let status = SecItemDelete(query as CFDictionary)\n if status == errSecSuccess || status == errSecItemNotFound {\n return\n }\n Self.log.error(\"Keychain delete failed: \\(status)\")\n throw KimiTokenStoreError.keychainStatus(status)\n }\n}\n" }, { "path": "Sources/CodexBar/LaunchAtLoginManager.swift", "content": "import CodexBarCore\nimport ServiceManagement\n\nenum LaunchAtLoginManager {\n private static let isRunningTests: Bool = {\n let env = ProcessInfo.processInfo.environment\n if env[\"XCTestConfigurationFilePath\"] != nil { return true }\n if env[\"TESTING_LIBRARY_VERSION\"] != nil { return true }\n if env[\"SWIFT_TESTING\"] != nil { return true }\n return NSClassFromString(\"XCTestCase\") != nil\n }()\n\n static func setEnabled(_ enabled: Bool) {\n if self.isRunningTests { return }\n let service = SMAppService.mainApp\n do {\n if enabled {\n try service.register()\n } else {\n try service.unregister()\n }\n } catch {\n CodexBarLog.logger(LogCategories.launchAtLogin).error(\"Failed to update login item: \\(error)\")\n }\n }\n}\n" }, { "path": "Sources/CodexBar/LoadingPattern.swift", "content": "import Foundation\n\nenum LoadingPattern: String, CaseIterable, Identifiable {\n case knightRider\n case cylon\n case outsideIn\n case race\n case pulse\n case unbraid\n\n var id: String {\n self.rawValue\n }\n\n var displayName: String {\n switch self {\n case .knightRider: \"Knight Rider\"\n case .cylon: \"Cylon\"\n case .outsideIn: \"Outside-In\"\n case .race: \"Race\"\n case .pulse: \"Pulse\"\n case .unbraid: \"Unbraid (logo → bars)\"\n }\n }\n\n /// Secondary offset so the lower bar moves differently.\n var secondaryOffset: Double {\n switch self {\n case .knightRider: .pi\n case .cylon: .pi / 2\n case .outsideIn: .pi\n case .race: .pi / 3\n case .pulse: .pi / 2\n case .unbraid: .pi / 2\n }\n }\n\n func value(phase: Double) -> Double {\n let v: Double\n switch self {\n case .knightRider:\n v = 0.5 + 0.5 * sin(phase) // ping-pong\n case .cylon:\n let t = phase.truncatingRemainder(dividingBy: .pi * 2) / (.pi * 2)\n v = t // sawtooth 0→1\n case .outsideIn:\n v = abs(cos(phase)) // high at edges, dip center\n case .race:\n let t = (phase * 1.2).truncatingRemainder(dividingBy: .pi * 2) / (.pi * 2)\n v = t\n case .pulse:\n v = 0.4 + 0.6 * (0.5 + 0.5 * sin(phase)) // 40–100%\n case .unbraid:\n v = 0.5 + 0.5 * sin(phase) // smooth 0→1 for morph\n }\n return max(0, min(v * 100, 100))\n }\n}\n\nextension Notification.Name {\n static let codexbarDebugReplayAllAnimations = Notification.Name(\"codexbarDebugReplayAllAnimations\")\n}\n" }, { "path": "Sources/CodexBar/MenuBarDisplayMode.swift", "content": "import Foundation\n\n/// Controls what the menu bar displays when brand icon mode is enabled.\nenum MenuBarDisplayMode: String, CaseIterable, Identifiable {\n case percent\n case pace\n case both\n\n var id: String {\n self.rawValue\n }\n\n var label: String {\n switch self {\n case .percent: \"Percent\"\n case .pace: \"Pace\"\n case .both: \"Both\"\n }\n }\n\n var description: String {\n switch self {\n case .percent: \"Show remaining/used percentage (e.g. 45%)\"\n case .pace: \"Show pace indicator (e.g. +5%)\"\n case .both: \"Show both percentage and pace (e.g. 45% · +5%)\"\n }\n }\n}\n" }, { "path": "Sources/CodexBar/MenuBarDisplayText.swift", "content": "import CodexBarCore\nimport Foundation\n\nenum MenuBarDisplayText {\n static func percentText(window: RateWindow?, showUsed: Bool) -> String? {\n guard let window else { return nil }\n let percent = showUsed ? window.usedPercent : window.remainingPercent\n let clamped = min(100, max(0, percent))\n return String(format: \"%.0f%%\", clamped)\n }\n\n static func paceText(pace: UsagePace?) -> String? {\n guard let pace else { return nil }\n let deltaValue = Int(abs(pace.deltaPercent).rounded())\n let sign = pace.deltaPercent >= 0 ? \"+\" : \"-\"\n return \"\\(sign)\\(deltaValue)%\"\n }\n\n static func displayText(\n mode: MenuBarDisplayMode,\n percentWindow: RateWindow?,\n pace: UsagePace? = nil,\n showUsed: Bool) -> String?\n {\n switch mode {\n case .percent:\n return self.percentText(window: percentWindow, showUsed: showUsed)\n case .pace:\n return self.paceText(pace: pace)\n case .both:\n guard let percent = percentText(window: percentWindow, showUsed: showUsed) else { return nil }\n let paceText: String? = Self.paceText(pace: pace)\n guard let paceText else { return nil }\n return \"\\(percent) · \\(paceText)\"\n }\n }\n}\n" }, { "path": "Sources/CodexBar/MenuCardView.swift", "content": "import AppKit\nimport CodexBarCore\nimport SwiftUI\n\n/// SwiftUI card used inside the NSMenu to mirror Apple's rich menu panels.\nstruct UsageMenuCardView: View {\n struct Model {\n enum PercentStyle: String {\n case left\n case used\n\n var labelSuffix: String {\n switch self {\n case .left: \"left\"\n case .used: \"used\"\n }\n }\n\n var accessibilityLabel: String {\n switch self {\n case .left: \"Usage remaining\"\n case .used: \"Usage used\"\n }\n }\n }\n\n struct Metric: Identifiable {\n let id: String\n let title: String\n let percent: Double\n let percentStyle: PercentStyle\n let resetText: String?\n let detailText: String?\n let detailLeftText: String?\n let detailRightText: String?\n let pacePercent: Double?\n let paceOnTop: Bool\n\n var percentLabel: String {\n String(format: \"%.0f%% %@\", self.percent, self.percentStyle.labelSuffix)\n }\n }\n\n enum SubtitleStyle {\n case info\n case loading\n case error\n }\n\n struct TokenUsageSection {\n let sessionLine: String\n let monthLine: String\n let hintLine: String?\n let errorLine: String?\n let errorCopyText: String?\n }\n\n struct ProviderCostSection {\n let title: String\n let percentUsed: Double\n let spendLine: String\n }\n\n let provider: UsageProvider\n let providerName: String\n let email: String\n let subtitleText: String\n let subtitleStyle: SubtitleStyle\n let planText: String?\n let metrics: [Metric]\n let usageNotes: [String]\n let creditsText: String?\n let creditsRemaining: Double?\n let creditsHintText: String?\n let creditsHintCopyText: String?\n let providerCost: ProviderCostSection?\n let tokenUsage: TokenUsageSection?\n let placeholder: String?\n let progressColor: Color\n }\n\n let model: Model\n let width: CGFloat\n @Environment(\\.menuItemHighlighted) private var isHighlighted\n\n static func popupMetricTitle(provider: UsageProvider, metric: Model.Metric) -> String {\n if provider == .openrouter, metric.id == \"primary\" {\n return \"API key limit\"\n }\n return metric.title\n }\n\n var body: some View {\n VStack(alignment: .leading, spacing: 6) {\n UsageMenuCardHeaderView(model: self.model)\n\n if self.hasDetails {\n Divider()\n }\n\n if self.model.metrics.isEmpty {\n if !self.model.usageNotes.isEmpty {\n UsageNotesContent(notes: self.model.usageNotes)\n } else if let placeholder = self.model.placeholder {\n Text(placeholder)\n .foregroundStyle(MenuHighlightStyle.secondary(self.isHighlighted))\n .font(.subheadline)\n }\n } else {\n let hasUsage = !self.model.metrics.isEmpty || !self.model.usageNotes.isEmpty\n let hasCredits = self.model.creditsText != nil\n let hasProviderCost = self.model.providerCost != nil\n let hasCost = self.model.tokenUsage != nil || hasProviderCost\n\n VStack(alignment: .leading, spacing: 12) {\n if hasUsage {\n VStack(alignment: .leading, spacing: 12) {\n ForEach(self.model.metrics, id: \\.id) { metric in\n MetricRow(\n metric: metric,\n title: Self.popupMetricTitle(provider: self.model.provider, metric: metric),\n progressColor: self.model.progressColor)\n }\n if !self.model.usageNotes.isEmpty {\n UsageNotesContent(notes: self.model.usageNotes)\n }\n }\n }\n if hasUsage, hasCredits || hasCost {\n Divider()\n }\n if let credits = self.model.creditsText {\n CreditsBarContent(\n creditsText: credits,\n creditsRemaining: self.model.creditsRemaining,\n hintText: self.model.creditsHintText,\n hintCopyText: self.model.creditsHintCopyText,\n progressColor: self.model.progressColor)\n }\n if hasCredits, hasCost {\n Divider()\n }\n if let providerCost = self.model.providerCost {\n ProviderCostContent(\n section: providerCost,\n progressColor: self.model.progressColor)\n }\n if hasProviderCost, self.model.tokenUsage != nil {\n Divider()\n }\n if let tokenUsage = self.model.tokenUsage {\n VStack(alignment: .leading, spacing: 6) {\n Text(\"Cost\")\n .font(.body)\n .fontWeight(.medium)\n Text(tokenUsage.sessionLine)\n .font(.footnote)\n Text(tokenUsage.monthLine)\n .font(.footnote)\n if let hint = tokenUsage.hintLine, !hint.isEmpty {\n Text(hint)\n .font(.footnote)\n .foregroundStyle(MenuHighlightStyle.secondary(self.isHighlighted))\n .lineLimit(4)\n .fixedSize(horizontal: false, vertical: true)\n }\n if let error = tokenUsage.errorLine, !error.isEmpty {\n Text(error)\n .font(.footnote)\n .foregroundStyle(MenuHighlightStyle.error(self.isHighlighted))\n .lineLimit(4)\n .fixedSize(horizontal: false, vertical: true)\n .overlay {\n ClickToCopyOverlay(copyText: tokenUsage.errorCopyText ?? error)\n }\n }\n }\n }\n }\n .padding(.bottom, self.model.creditsText == nil ? 6 : 0)\n }\n }\n .padding(.horizontal, 16)\n .padding(.top, 2)\n .padding(.bottom, 2)\n .frame(width: self.width, alignment: .leading)\n }\n\n private var hasDetails: Bool {\n !self.model.metrics.isEmpty || !self.model.usageNotes.isEmpty || self.model.placeholder != nil ||\n self.model.tokenUsage != nil ||\n self.model.providerCost != nil\n }\n}\n\nprivate struct UsageMenuCardHeaderView: View {\n let model: UsageMenuCardView.Model\n @Environment(\\.menuItemHighlighted) private var isHighlighted\n\n var body: some View {\n VStack(alignment: .leading, spacing: 3) {\n HStack(alignment: .firstTextBaseline) {\n Text(self.model.providerName)\n .font(.headline)\n .fontWeight(.semibold)\n Spacer()\n Text(self.model.email)\n .font(.subheadline)\n .foregroundStyle(MenuHighlightStyle.secondary(self.isHighlighted))\n }\n let subtitleAlignment: VerticalAlignment = self.model.subtitleStyle == .error ? .top : .firstTextBaseline\n HStack(alignment: subtitleAlignment) {\n Text(self.model.subtitleText)\n .font(.footnote)\n .foregroundStyle(self.subtitleColor)\n .lineLimit(self.model.subtitleStyle == .error ? 4 : 1)\n .multilineTextAlignment(.leading)\n .fixedSize(horizontal: false, vertical: true)\n .layoutPriority(1)\n .padding(.bottom, self.model.subtitleStyle == .error ? 4 : 0)\n Spacer()\n if self.model.subtitleStyle == .error, !self.model.subtitleText.isEmpty {\n CopyIconButton(copyText: self.model.subtitleText, isHighlighted: self.isHighlighted)\n }\n if let plan = self.model.planText {\n Text(plan)\n .font(.footnote)\n .foregroundStyle(MenuHighlightStyle.secondary(self.isHighlighted))\n .lineLimit(1)\n }\n }\n }\n }\n\n private var subtitleColor: Color {\n switch self.model.subtitleStyle {\n case .info: MenuHighlightStyle.secondary(self.isHighlighted)\n case .loading: MenuHighlightStyle.secondary(self.isHighlighted)\n case .error: MenuHighlightStyle.error(self.isHighlighted)\n }\n }\n}\n\nprivate struct CopyIconButtonStyle: ButtonStyle {\n let isHighlighted: Bool\n\n func makeBody(configuration: Configuration) -> some View {\n configuration.label\n .padding(4)\n .background {\n RoundedRectangle(cornerRadius: 4, style: .continuous)\n .fill(MenuHighlightStyle.secondary(self.isHighlighted).opacity(configuration.isPressed ? 0.18 : 0))\n }\n .scaleEffect(configuration.isPressed ? 0.94 : 1)\n .animation(.easeOut(duration: 0.12), value: configuration.isPressed)\n }\n}\n\nprivate struct CopyIconButton: View {\n let copyText: String\n let isHighlighted: Bool\n\n @State private var didCopy = false\n @State private var resetTask: Task?\n\n var body: some View {\n Button {\n self.copyToPasteboard()\n withAnimation(.easeOut(duration: 0.12)) {\n self.didCopy = true\n }\n self.resetTask?.cancel()\n self.resetTask = Task { @MainActor in\n try? await Task.sleep(for: .seconds(0.9))\n withAnimation(.easeOut(duration: 0.2)) {\n self.didCopy = false\n }\n }\n } label: {\n Image(systemName: self.didCopy ? \"checkmark\" : \"doc.on.doc\")\n .font(.caption2.weight(.semibold))\n .foregroundStyle(MenuHighlightStyle.secondary(self.isHighlighted))\n .frame(width: 18, height: 18)\n }\n .buttonStyle(CopyIconButtonStyle(isHighlighted: self.isHighlighted))\n .accessibilityLabel(self.didCopy ? \"Copied\" : \"Copy error\")\n }\n\n private func copyToPasteboard() {\n let pb = NSPasteboard.general\n pb.clearContents()\n pb.setString(self.copyText, forType: .string)\n }\n}\n\nprivate struct ProviderCostContent: View {\n let section: UsageMenuCardView.Model.ProviderCostSection\n let progressColor: Color\n @Environment(\\.menuItemHighlighted) private var isHighlighted\n\n var body: some View {\n VStack(alignment: .leading, spacing: 6) {\n Text(self.section.title)\n .font(.body)\n .fontWeight(.medium)\n UsageProgressBar(\n percent: self.section.percentUsed,\n tint: self.progressColor,\n accessibilityLabel: \"Extra usage spent\")\n HStack(alignment: .firstTextBaseline) {\n Text(self.section.spendLine)\n .font(.footnote)\n Spacer()\n Text(String(format: \"%.0f%% used\", min(100, max(0, self.section.percentUsed))))\n .font(.footnote)\n .foregroundStyle(MenuHighlightStyle.secondary(self.isHighlighted))\n }\n }\n }\n}\n\nprivate struct MetricRow: View {\n let metric: UsageMenuCardView.Model.Metric\n let title: String\n let progressColor: Color\n @Environment(\\.menuItemHighlighted) private var isHighlighted\n\n var body: some View {\n VStack(alignment: .leading, spacing: 6) {\n Text(self.title)\n .font(.body)\n .fontWeight(.medium)\n UsageProgressBar(\n percent: self.metric.percent,\n tint: self.progressColor,\n accessibilityLabel: self.metric.percentStyle.accessibilityLabel,\n pacePercent: self.metric.pacePercent,\n paceOnTop: self.metric.paceOnTop)\n VStack(alignment: .leading, spacing: 2) {\n HStack(alignment: .firstTextBaseline) {\n Text(self.metric.percentLabel)\n .font(.footnote)\n .lineLimit(1)\n Spacer()\n if let rightLabel = self.metric.resetText {\n Text(rightLabel)\n .font(.footnote)\n .foregroundStyle(MenuHighlightStyle.secondary(self.isHighlighted))\n .lineLimit(1)\n }\n }\n if self.metric.detailLeftText != nil || self.metric.detailRightText != nil {\n HStack(alignment: .firstTextBaseline) {\n if let detailLeft = self.metric.detailLeftText {\n Text(detailLeft)\n .font(.footnote)\n .foregroundStyle(MenuHighlightStyle.primary(self.isHighlighted))\n .lineLimit(1)\n }\n Spacer()\n if let detailRight = self.metric.detailRightText {\n Text(detailRight)\n .font(.footnote)\n .foregroundStyle(MenuHighlightStyle.secondary(self.isHighlighted))\n .lineLimit(1)\n }\n }\n }\n }\n .frame(maxWidth: .infinity, alignment: .leading)\n if let detail = self.metric.detailText {\n Text(detail)\n .font(.footnote)\n .foregroundStyle(MenuHighlightStyle.secondary(self.isHighlighted))\n .lineLimit(1)\n }\n }\n .frame(maxWidth: .infinity, alignment: .leading)\n }\n}\n\nprivate struct UsageNotesContent: View {\n let notes: [String]\n @Environment(\\.menuItemHighlighted) private var isHighlighted\n\n var body: some View {\n VStack(alignment: .leading, spacing: 4) {\n ForEach(Array(self.notes.enumerated()), id: \\.offset) { _, note in\n Text(note)\n .font(.footnote)\n .foregroundStyle(MenuHighlightStyle.secondary(self.isHighlighted))\n .lineLimit(2)\n .fixedSize(horizontal: false, vertical: true)\n }\n }\n .frame(maxWidth: .infinity, alignment: .leading)\n }\n}\n\nstruct UsageMenuCardHeaderSectionView: View {\n let model: UsageMenuCardView.Model\n let showDivider: Bool\n let width: CGFloat\n\n var body: some View {\n VStack(alignment: .leading, spacing: 6) {\n UsageMenuCardHeaderView(model: self.model)\n\n if self.showDivider {\n Divider()\n }\n }\n .padding(.horizontal, 16)\n .padding(.top, 2)\n .padding(.bottom, self.model.subtitleStyle == .error ? 2 : 0)\n .frame(width: self.width, alignment: .leading)\n }\n}\n\nstruct UsageMenuCardUsageSectionView: View {\n let model: UsageMenuCardView.Model\n let showBottomDivider: Bool\n let bottomPadding: CGFloat\n let width: CGFloat\n @Environment(\\.menuItemHighlighted) private var isHighlighted\n\n var body: some View {\n VStack(alignment: .leading, spacing: 12) {\n if self.model.metrics.isEmpty {\n if !self.model.usageNotes.isEmpty {\n UsageNotesContent(notes: self.model.usageNotes)\n } else if let placeholder = self.model.placeholder {\n Text(placeholder)\n .foregroundStyle(MenuHighlightStyle.secondary(self.isHighlighted))\n .font(.subheadline)\n }\n } else {\n ForEach(self.model.metrics, id: \\.id) { metric in\n MetricRow(\n metric: metric,\n title: UsageMenuCardView.popupMetricTitle(provider: self.model.provider, metric: metric),\n progressColor: self.model.progressColor)\n }\n if !self.model.usageNotes.isEmpty {\n UsageNotesContent(notes: self.model.usageNotes)\n }\n }\n if self.showBottomDivider {\n Divider()\n }\n }\n .padding(.horizontal, 16)\n .padding(.top, 10)\n .padding(.bottom, self.bottomPadding)\n .frame(width: self.width, alignment: .leading)\n }\n}\n\nstruct UsageMenuCardCreditsSectionView: View {\n let model: UsageMenuCardView.Model\n let showBottomDivider: Bool\n let topPadding: CGFloat\n let bottomPadding: CGFloat\n let width: CGFloat\n\n var body: some View {\n if let credits = self.model.creditsText {\n VStack(alignment: .leading, spacing: 6) {\n CreditsBarContent(\n creditsText: credits,\n creditsRemaining: self.model.creditsRemaining,\n hintText: self.model.creditsHintText,\n hintCopyText: self.model.creditsHintCopyText,\n progressColor: self.model.progressColor)\n if self.showBottomDivider {\n Divider()\n }\n }\n .padding(.horizontal, 16)\n .padding(.top, self.topPadding)\n .padding(.bottom, self.bottomPadding)\n .frame(width: self.width, alignment: .leading)\n }\n }\n}\n\nprivate struct CreditsBarContent: View {\n private static let fullScaleTokens: Double = 1000\n\n let creditsText: String\n let creditsRemaining: Double?\n let hintText: String?\n let hintCopyText: String?\n let progressColor: Color\n @Environment(\\.menuItemHighlighted) private var isHighlighted\n\n private var percentLeft: Double? {\n guard let creditsRemaining else { return nil }\n let percent = (creditsRemaining / Self.fullScaleTokens) * 100\n return min(100, max(0, percent))\n }\n\n private var scaleText: String {\n let scale = UsageFormatter.tokenCountString(Int(Self.fullScaleTokens))\n return \"\\(scale) tokens\"\n }\n\n var body: some View {\n VStack(alignment: .leading, spacing: 6) {\n Text(\"Credits\")\n .font(.body)\n .fontWeight(.medium)\n if let percentLeft {\n UsageProgressBar(\n percent: percentLeft,\n tint: self.progressColor,\n accessibilityLabel: \"Credits remaining\")\n HStack(alignment: .firstTextBaseline) {\n Text(self.creditsText)\n .font(.caption)\n Spacer()\n Text(self.scaleText)\n .font(.caption)\n .foregroundStyle(MenuHighlightStyle.secondary(self.isHighlighted))\n }\n } else {\n Text(self.creditsText)\n .font(.caption)\n }\n if let hintText, !hintText.isEmpty {\n Text(hintText)\n .font(.footnote)\n .foregroundStyle(MenuHighlightStyle.secondary(self.isHighlighted))\n .lineLimit(4)\n .fixedSize(horizontal: false, vertical: true)\n .overlay {\n ClickToCopyOverlay(copyText: self.hintCopyText ?? hintText)\n }\n }\n }\n }\n}\n\nstruct UsageMenuCardCostSectionView: View {\n let model: UsageMenuCardView.Model\n let topPadding: CGFloat\n let bottomPadding: CGFloat\n let width: CGFloat\n @Environment(\\.menuItemHighlighted) private var isHighlighted\n\n var body: some View {\n let hasTokenCost = self.model.tokenUsage != nil\n return Group {\n if hasTokenCost {\n VStack(alignment: .leading, spacing: 10) {\n if let tokenUsage = self.model.tokenUsage {\n VStack(alignment: .leading, spacing: 6) {\n Text(\"Cost\")\n .font(.body)\n .fontWeight(.medium)\n Text(tokenUsage.sessionLine)\n .font(.caption)\n Text(tokenUsage.monthLine)\n .font(.caption)\n if let hint = tokenUsage.hintLine, !hint.isEmpty {\n Text(hint)\n .font(.footnote)\n .foregroundStyle(MenuHighlightStyle.secondary(self.isHighlighted))\n .lineLimit(4)\n .fixedSize(horizontal: false, vertical: true)\n }\n if let error = tokenUsage.errorLine, !error.isEmpty {\n Text(error)\n .font(.footnote)\n .foregroundStyle(MenuHighlightStyle.error(self.isHighlighted))\n .lineLimit(4)\n .fixedSize(horizontal: false, vertical: true)\n .overlay {\n ClickToCopyOverlay(copyText: tokenUsage.errorCopyText ?? error)\n }\n }\n }\n }\n }\n .padding(.horizontal, 16)\n .padding(.top, self.topPadding)\n .padding(.bottom, self.bottomPadding)\n .frame(width: self.width, alignment: .leading)\n }\n }\n }\n}\n\nstruct UsageMenuCardExtraUsageSectionView: View {\n let model: UsageMenuCardView.Model\n let topPadding: CGFloat\n let bottomPadding: CGFloat\n let width: CGFloat\n\n var body: some View {\n Group {\n if let providerCost = self.model.providerCost {\n ProviderCostContent(\n section: providerCost,\n progressColor: self.model.progressColor)\n .padding(.horizontal, 16)\n .padding(.top, self.topPadding)\n .padding(.bottom, self.bottomPadding)\n .frame(width: self.width, alignment: .leading)\n }\n }\n }\n}\n\n// MARK: - Model factory\n\nextension UsageMenuCardView.Model {\n struct Input {\n let provider: UsageProvider\n let metadata: ProviderMetadata\n let snapshot: UsageSnapshot?\n let credits: CreditsSnapshot?\n let creditsError: String?\n let dashboard: OpenAIDashboardSnapshot?\n let dashboardError: String?\n let tokenSnapshot: CostUsageTokenSnapshot?\n let tokenError: String?\n let account: AccountInfo\n let isRefreshing: Bool\n let lastError: String?\n let usageBarsShowUsed: Bool\n let resetTimeDisplayStyle: ResetTimeDisplayStyle\n let tokenCostUsageEnabled: Bool\n let showOptionalCreditsAndExtraUsage: Bool\n let sourceLabel: String?\n let kiloAutoMode: Bool\n let hidePersonalInfo: Bool\n let weeklyPace: UsagePace?\n let now: Date\n\n init(\n provider: UsageProvider,\n metadata: ProviderMetadata,\n snapshot: UsageSnapshot?,\n credits: CreditsSnapshot?,\n creditsError: String?,\n dashboard: OpenAIDashboardSnapshot?,\n dashboardError: String?,\n tokenSnapshot: CostUsageTokenSnapshot?,\n tokenError: String?,\n account: AccountInfo,\n isRefreshing: Bool,\n lastError: String?,\n usageBarsShowUsed: Bool,\n resetTimeDisplayStyle: ResetTimeDisplayStyle,\n tokenCostUsageEnabled: Bool,\n showOptionalCreditsAndExtraUsage: Bool,\n sourceLabel: String? = nil,\n kiloAutoMode: Bool = false,\n hidePersonalInfo: Bool,\n weeklyPace: UsagePace? = nil,\n now: Date)\n {\n self.provider = provider\n self.metadata = metadata\n self.snapshot = snapshot\n self.credits = credits\n self.creditsError = creditsError\n self.dashboard = dashboard\n self.dashboardError = dashboardError\n self.tokenSnapshot = tokenSnapshot\n self.tokenError = tokenError\n self.account = account\n self.isRefreshing = isRefreshing\n self.lastError = lastError\n self.usageBarsShowUsed = usageBarsShowUsed\n self.resetTimeDisplayStyle = resetTimeDisplayStyle\n self.tokenCostUsageEnabled = tokenCostUsageEnabled\n self.showOptionalCreditsAndExtraUsage = showOptionalCreditsAndExtraUsage\n self.sourceLabel = sourceLabel\n self.kiloAutoMode = kiloAutoMode\n self.hidePersonalInfo = hidePersonalInfo\n self.weeklyPace = weeklyPace\n self.now = now\n }\n }\n\n static func make(_ input: Input) -> UsageMenuCardView.Model {\n let planText = Self.plan(\n for: input.provider,\n snapshot: input.snapshot,\n account: input.account,\n metadata: input.metadata)\n let metrics = Self.metrics(input: input)\n let usageNotes = Self.usageNotes(input: input)\n let creditsText: String? = if input.provider == .openrouter {\n nil\n } else if input.provider == .codex, !input.showOptionalCreditsAndExtraUsage {\n nil\n } else {\n Self.creditsLine(metadata: input.metadata, credits: input.credits, error: input.creditsError)\n }\n let providerCost: ProviderCostSection? = if input.provider == .claude, !input.showOptionalCreditsAndExtraUsage {\n nil\n } else {\n Self.providerCostSection(provider: input.provider, cost: input.snapshot?.providerCost)\n }\n let tokenUsage = Self.tokenUsageSection(\n provider: input.provider,\n enabled: input.tokenCostUsageEnabled,\n snapshot: input.tokenSnapshot,\n error: input.tokenError)\n let subtitle = Self.subtitle(\n snapshot: input.snapshot,\n isRefreshing: input.isRefreshing,\n lastError: input.lastError)\n let redacted = Self.redactedText(input: input, subtitle: subtitle)\n let placeholder = input.snapshot == nil && !input.isRefreshing && input.lastError == nil ? \"No usage yet\" : nil\n\n return UsageMenuCardView.Model(\n provider: input.provider,\n providerName: input.metadata.displayName,\n email: redacted.email,\n subtitleText: redacted.subtitleText,\n subtitleStyle: subtitle.style,\n planText: planText,\n metrics: metrics,\n usageNotes: usageNotes,\n creditsText: creditsText,\n creditsRemaining: input.credits?.remaining,\n creditsHintText: redacted.creditsHintText,\n creditsHintCopyText: redacted.creditsHintCopyText,\n providerCost: providerCost,\n tokenUsage: tokenUsage,\n placeholder: placeholder,\n progressColor: Self.progressColor(for: input.provider))\n }\n\n private static func usageNotes(input: Input) -> [String] {\n if input.provider == .kilo {\n var notes = Self.kiloLoginDetails(snapshot: input.snapshot)\n let resolvedSource = input.sourceLabel?\n .trimmingCharacters(in: .whitespacesAndNewlines)\n .lowercased()\n if input.kiloAutoMode,\n resolvedSource == \"cli\",\n !notes.contains(where: { $0.caseInsensitiveCompare(\"Using CLI fallback\") == .orderedSame })\n {\n notes.append(\"Using CLI fallback\")\n }\n return notes\n }\n\n guard input.provider == .openrouter,\n let openRouter = input.snapshot?.openRouterUsage\n else {\n return []\n }\n\n return switch openRouter.keyQuotaStatus {\n case .available: []\n case .noLimitConfigured: [\"No limit set for the API key\"]\n case .unavailable: [\"API key limit unavailable right now\"]\n }\n }\n\n private static func email(\n for provider: UsageProvider,\n snapshot: UsageSnapshot?,\n account: AccountInfo,\n metadata: ProviderMetadata) -> String\n {\n if let email = snapshot?.accountEmail(for: provider), !email.isEmpty { return email }\n if metadata.usesAccountFallback,\n let email = account.email, !email.isEmpty\n {\n return email\n }\n return \"\"\n }\n\n private static func plan(\n for provider: UsageProvider,\n snapshot: UsageSnapshot?,\n account: AccountInfo,\n metadata: ProviderMetadata) -> String?\n {\n if provider == .kilo {\n guard let pass = self.kiloLoginPass(snapshot: snapshot) else {\n return nil\n }\n return self.planDisplay(pass)\n }\n if let plan = snapshot?.loginMethod(for: provider), !plan.isEmpty {\n return self.planDisplay(plan)\n }\n if metadata.usesAccountFallback,\n let plan = account.plan, !plan.isEmpty\n {\n return Self.planDisplay(plan)\n }\n return nil\n }\n\n private static func planDisplay(_ text: String) -> String {\n let cleaned = UsageFormatter.cleanPlanName(text)\n return cleaned.isEmpty ? text : cleaned\n }\n\n private static func kiloLoginPass(snapshot: UsageSnapshot?) -> String? {\n self.kiloLoginParts(snapshot: snapshot).pass\n }\n\n private static func kiloLoginDetails(snapshot: UsageSnapshot?) -> [String] {\n self.kiloLoginParts(snapshot: snapshot).details\n }\n\n private static func kiloLoginParts(snapshot: UsageSnapshot?) -> (pass: String?, details: [String]) {\n guard let loginMethod = snapshot?.loginMethod(for: .kilo) else {\n return (nil, [])\n }\n let parts = loginMethod\n .components(separatedBy: \"·\")\n .map { $0.trimmingCharacters(in: .whitespacesAndNewlines) }\n .filter { !$0.isEmpty }\n guard !parts.isEmpty else {\n return (nil, [])\n }\n let first = parts[0]\n if self.isKiloActivitySegment(first) {\n return (nil, parts)\n }\n return (first, Array(parts.dropFirst()))\n }\n\n private static func isKiloActivitySegment(_ text: String) -> Bool {\n let normalized = text.trimmingCharacters(in: .whitespacesAndNewlines).lowercased()\n return normalized.hasPrefix(\"auto top-up:\")\n }\n\n private static func subtitle(\n snapshot: UsageSnapshot?,\n isRefreshing: Bool,\n lastError: String?) -> (text: String, style: SubtitleStyle)\n {\n if let lastError, !lastError.isEmpty {\n return (lastError.trimmingCharacters(in: .whitespacesAndNewlines), .error)\n }\n\n if isRefreshing, snapshot == nil {\n return (\"Refreshing...\", .loading)\n }\n\n if let updated = snapshot?.updatedAt {\n return (UsageFormatter.updatedString(from: updated), .info)\n }\n\n return (\"Not fetched yet\", .info)\n }\n\n private struct RedactedText {\n let email: String\n let subtitleText: String\n let creditsHintText: String?\n let creditsHintCopyText: String?\n }\n\n private static func redactedText(\n input: Input,\n subtitle: (text: String, style: SubtitleStyle)) -> RedactedText\n {\n let email = PersonalInfoRedactor.redactEmail(\n Self.email(\n for: input.provider,\n snapshot: input.snapshot,\n account: input.account,\n metadata: input.metadata),\n isEnabled: input.hidePersonalInfo)\n let subtitleText = PersonalInfoRedactor.redactEmails(in: subtitle.text, isEnabled: input.hidePersonalInfo)\n ?? subtitle.text\n let creditsHintText = PersonalInfoRedactor.redactEmails(\n in: Self.dashboardHint(provider: input.provider, error: input.dashboardError),\n isEnabled: input.hidePersonalInfo)\n let creditsHintCopyText = Self.creditsHintCopyText(\n dashboardError: input.dashboardError,\n hidePersonalInfo: input.hidePersonalInfo)\n return RedactedText(\n email: email,\n subtitleText: subtitleText,\n creditsHintText: creditsHintText,\n creditsHintCopyText: creditsHintCopyText)\n }\n\n private static func creditsHintCopyText(dashboardError: String?, hidePersonalInfo: Bool) -> String? {\n guard let dashboardError, !dashboardError.isEmpty else { return nil }\n return hidePersonalInfo ? \"\" : dashboardError\n }\n\n private static func metrics(input: Input) -> [Metric] {\n guard let snapshot = input.snapshot else { return [] }\n var metrics: [Metric] = []\n let percentStyle: PercentStyle = input.usageBarsShowUsed ? .used : .left\n let zaiUsage = input.provider == .zai ? snapshot.zaiUsage : nil\n let zaiTokenDetail = Self.zaiLimitDetailText(limit: zaiUsage?.tokenLimit)\n let zaiTimeDetail = Self.zaiLimitDetailText(limit: zaiUsage?.timeLimit)\n let openRouterQuotaDetail = Self.openRouterQuotaDetail(provider: input.provider, snapshot: snapshot)\n if let primary = snapshot.primary {\n var primaryDetailText: String? = input.provider == .zai ? zaiTokenDetail : nil\n var primaryResetText = Self.resetText(for: primary, style: input.resetTimeDisplayStyle, now: input.now)\n if input.provider == .openrouter,\n let openRouterQuotaDetail\n {\n primaryResetText = openRouterQuotaDetail\n }\n if input.provider == .warp || input.provider == .kilo,\n let detail = primary.resetDescription,\n !detail.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty\n {\n primaryDetailText = detail\n }\n if input.provider == .alibaba,\n let detail = primary.resetDescription,\n !detail.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty\n {\n primaryDetailText = detail\n }\n if input.provider == .warp || input.provider == .kilo, primary.resetsAt == nil {\n primaryResetText = nil\n }\n metrics.append(Metric(\n id: \"primary\",\n title: input.metadata.sessionLabel,\n percent: Self.clamped(\n input.usageBarsShowUsed ? primary.usedPercent : primary.remainingPercent),\n percentStyle: percentStyle,\n resetText: primaryResetText,\n detailText: primaryDetailText,\n detailLeftText: nil,\n detailRightText: nil,\n pacePercent: nil,\n paceOnTop: true))\n }\n if let weekly = snapshot.secondary {\n let paceDetail = Self.weeklyPaceDetail(\n window: weekly,\n now: input.now,\n pace: input.weeklyPace,\n showUsed: input.usageBarsShowUsed)\n var weeklyResetText = Self.resetText(for: weekly, style: input.resetTimeDisplayStyle, now: input.now)\n var weeklyDetailText: String? = input.provider == .zai ? zaiTimeDetail : nil\n if input.provider == .warp,\n let detail = weekly.resetDescription,\n !detail.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty\n {\n weeklyResetText = nil\n weeklyDetailText = detail\n }\n if input.provider == .kilo,\n let detail = weekly.resetDescription,\n !detail.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty\n {\n weeklyDetailText = detail\n if weekly.resetsAt == nil {\n weeklyResetText = nil\n }\n }\n if input.provider == .alibaba,\n let detail = weekly.resetDescription,\n !detail.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty\n {\n weeklyDetailText = detail\n }\n metrics.append(Metric(\n id: \"secondary\",\n title: input.metadata.weeklyLabel,\n percent: Self.clamped(input.usageBarsShowUsed ? weekly.usedPercent : weekly.remainingPercent),\n percentStyle: percentStyle,\n resetText: weeklyResetText,\n detailText: weeklyDetailText,\n detailLeftText: paceDetail?.leftLabel,\n detailRightText: paceDetail?.rightLabel,\n pacePercent: paceDetail?.pacePercent,\n paceOnTop: paceDetail?.paceOnTop ?? true))\n }\n if input.provider == .kilo,\n metrics.contains(where: { $0.id == \"primary\" }),\n metrics.contains(where: { $0.id == \"secondary\" })\n {\n metrics.sort { lhs, rhs in\n let kiloOrder: [String: Int] = [\n \"secondary\": 0,\n \"primary\": 1,\n ]\n return (kiloOrder[lhs.id] ?? Int.max) < (kiloOrder[rhs.id] ?? Int.max)\n }\n }\n if input.metadata.supportsOpus, let opus = snapshot.tertiary {\n var tertiaryDetailText: String?\n if input.provider == .alibaba,\n let detail = opus.resetDescription,\n !detail.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty\n {\n tertiaryDetailText = detail\n }\n metrics.append(Metric(\n id: \"tertiary\",\n title: input.metadata.opusLabel ?? \"Sonnet\",\n percent: Self.clamped(input.usageBarsShowUsed ? opus.usedPercent : opus.remainingPercent),\n percentStyle: percentStyle,\n resetText: Self.resetText(for: opus, style: input.resetTimeDisplayStyle, now: input.now),\n detailText: tertiaryDetailText,\n detailLeftText: nil,\n detailRightText: nil,\n pacePercent: nil,\n paceOnTop: true))\n }\n\n if input.provider == .codex, let remaining = input.dashboard?.codeReviewRemainingPercent {\n let percent = input.usageBarsShowUsed ? (100 - remaining) : remaining\n metrics.append(Metric(\n id: \"code-review\",\n title: \"Code review\",\n percent: Self.clamped(percent),\n percentStyle: percentStyle,\n resetText: nil,\n detailText: nil,\n detailLeftText: nil,\n detailRightText: nil,\n pacePercent: nil,\n paceOnTop: true))\n }\n return metrics\n }\n\n private static func zaiLimitDetailText(limit: ZaiLimitEntry?) -> String? {\n guard let limit else { return nil }\n\n if let currentValue = limit.currentValue,\n let usage = limit.usage,\n let remaining = limit.remaining\n {\n let currentStr = UsageFormatter.tokenCountString(currentValue)\n let usageStr = UsageFormatter.tokenCountString(usage)\n let remainingStr = UsageFormatter.tokenCountString(remaining)\n return \"\\(currentStr) / \\(usageStr) (\\(remainingStr) remaining)\"\n }\n\n return nil\n }\n\n private static func openRouterQuotaDetail(provider: UsageProvider, snapshot: UsageSnapshot) -> String? {\n guard provider == .openrouter,\n let usage = snapshot.openRouterUsage,\n usage.hasValidKeyQuota,\n let keyRemaining = usage.keyRemaining,\n let keyLimit = usage.keyLimit\n else {\n return nil\n }\n\n let remaining = UsageFormatter.usdString(keyRemaining)\n let limit = UsageFormatter.usdString(keyLimit)\n return \"\\(remaining)/\\(limit) left\"\n }\n\n private struct PaceDetail {\n let leftLabel: String\n let rightLabel: String?\n let pacePercent: Double?\n let paceOnTop: Bool\n }\n\n private static func weeklyPaceDetail(\n window: RateWindow,\n now: Date,\n pace: UsagePace?,\n showUsed: Bool) -> PaceDetail?\n {\n guard let pace else { return nil }\n let detail = UsagePaceText.weeklyDetail(pace: pace, now: now)\n let expectedUsed = detail.expectedUsedPercent\n let actualUsed = window.usedPercent\n let expectedPercent = showUsed ? expectedUsed : (100 - expectedUsed)\n let actualPercent = showUsed ? actualUsed : (100 - actualUsed)\n if expectedPercent.isFinite == false || actualPercent.isFinite == false { return nil }\n let paceOnTop = actualUsed <= expectedUsed\n let pacePercent: Double? = if detail.stage == .onTrack { nil } else { expectedPercent }\n return PaceDetail(\n leftLabel: detail.leftLabel,\n rightLabel: detail.rightLabel,\n pacePercent: pacePercent,\n paceOnTop: paceOnTop)\n }\n\n private static func creditsLine(\n metadata: ProviderMetadata,\n credits: CreditsSnapshot?,\n error: String?) -> String?\n {\n guard metadata.supportsCredits else { return nil }\n if let credits {\n return UsageFormatter.creditsString(from: credits.remaining)\n }\n if let error, !error.isEmpty {\n return error.trimmingCharacters(in: .whitespacesAndNewlines)\n }\n return metadata.creditsHint\n }\n\n private static func dashboardHint(provider: UsageProvider, error: String?) -> String? {\n guard provider == .codex else { return nil }\n guard let error, !error.isEmpty else { return nil }\n return error\n }\n\n private static func tokenUsageSection(\n provider: UsageProvider,\n enabled: Bool,\n snapshot: CostUsageTokenSnapshot?,\n error: String?) -> TokenUsageSection?\n {\n guard provider == .codex || provider == .claude || provider == .vertexai else { return nil }\n guard enabled else { return nil }\n guard let snapshot else { return nil }\n\n let sessionCost = snapshot.sessionCostUSD.map { UsageFormatter.usdString($0) } ?? \"—\"\n let sessionTokens = snapshot.sessionTokens.map { UsageFormatter.tokenCountString($0) }\n let sessionLine: String = {\n if let sessionTokens {\n return \"Today: \\(sessionCost) · \\(sessionTokens) tokens\"\n }\n return \"Today: \\(sessionCost)\"\n }()\n\n let monthCost = snapshot.last30DaysCostUSD.map { UsageFormatter.usdString($0) } ?? \"—\"\n let fallbackTokens = snapshot.daily.compactMap(\\.totalTokens).reduce(0, +)\n let monthTokensValue = snapshot.last30DaysTokens ?? (fallbackTokens > 0 ? fallbackTokens : nil)\n let monthTokens = monthTokensValue.map { UsageFormatter.tokenCountString($0) }\n let monthLine: String = {\n if let monthTokens {\n return \"Last 30 days: \\(monthCost) · \\(monthTokens) tokens\"\n }\n return \"Last 30 days: \\(monthCost)\"\n }()\n let err = (error?.isEmpty ?? true) ? nil : error\n return TokenUsageSection(\n sessionLine: sessionLine,\n monthLine: monthLine,\n hintLine: nil,\n errorLine: err,\n errorCopyText: (error?.isEmpty ?? true) ? nil : error)\n }\n\n private static func providerCostSection(\n provider: UsageProvider,\n cost: ProviderCostSnapshot?) -> ProviderCostSection?\n {\n guard let cost else { return nil }\n guard cost.limit > 0 else { return nil }\n\n let used: String\n let limit: String\n let title: String\n\n if cost.currencyCode == \"Quota\" {\n title = \"Quota usage\"\n used = String(format: \"%.0f\", cost.used)\n limit = String(format: \"%.0f\", cost.limit)\n } else {\n title = \"Extra usage\"\n used = UsageFormatter.currencyString(cost.used, currencyCode: cost.currencyCode)\n limit = UsageFormatter.currencyString(cost.limit, currencyCode: cost.currencyCode)\n }\n\n let percentUsed = Self.clamped((cost.used / cost.limit) * 100)\n let periodLabel = cost.period ?? \"This month\"\n\n return ProviderCostSection(\n title: title,\n percentUsed: percentUsed,\n spendLine: \"\\(periodLabel): \\(used) / \\(limit)\")\n }\n\n private static func clamped(_ value: Double) -> Double {\n min(100, max(0, value))\n }\n\n private static func progressColor(for provider: UsageProvider) -> Color {\n let color = ProviderDescriptorRegistry.descriptor(for: provider).branding.color\n return Color(red: color.red, green: color.green, blue: color.blue)\n }\n\n private static func resetText(\n for window: RateWindow,\n style: ResetTimeDisplayStyle,\n now: Date) -> String?\n {\n UsageFormatter.resetLine(for: window, style: style, now: now)\n }\n}\n\n// MARK: - Copy-on-click overlay\n\nprivate struct ClickToCopyOverlay: NSViewRepresentable {\n let copyText: String\n\n func makeNSView(context: Context) -> ClickToCopyView {\n ClickToCopyView(copyText: self.copyText)\n }\n\n func updateNSView(_ nsView: ClickToCopyView, context: Context) {\n nsView.copyText = self.copyText\n }\n}\n\nprivate final class ClickToCopyView: NSView {\n var copyText: String\n\n init(copyText: String) {\n self.copyText = copyText\n super.init(frame: .zero)\n self.wantsLayer = false\n }\n\n @available(*, unavailable)\n required init?(coder: NSCoder) {\n fatalError(\"init(coder:) has not been implemented\")\n }\n\n override func acceptsFirstMouse(for event: NSEvent?) -> Bool {\n true\n }\n\n override func mouseDown(with event: NSEvent) {\n _ = event\n let pb = NSPasteboard.general\n pb.clearContents()\n pb.setString(self.copyText, forType: .string)\n }\n}\n" }, { "path": "Sources/CodexBar/MenuContent.swift", "content": "import AppKit\nimport CodexBarCore\nimport SwiftUI\n\n@MainActor\nstruct MenuContent: View {\n @Bindable var store: UsageStore\n @Bindable var settings: SettingsStore\n let account: AccountInfo\n let updater: UpdaterProviding\n let provider: UsageProvider?\n let actions: MenuActions\n\n var body: some View {\n let descriptor = MenuDescriptor.build(\n provider: self.provider,\n store: self.store,\n settings: self.settings,\n account: self.account,\n updateReady: self.updater.updateStatus.isUpdateReady)\n\n VStack(alignment: .leading, spacing: 8) {\n ForEach(Array(descriptor.sections.enumerated()), id: \\.offset) { index, section in\n VStack(alignment: .leading, spacing: 4) {\n ForEach(Array(section.entries.enumerated()), id: \\.offset) { _, entry in\n self.row(for: entry)\n }\n }\n if index < descriptor.sections.count - 1 {\n Divider()\n }\n }\n }\n .padding(.horizontal, 10)\n .padding(.vertical, 6)\n .frame(minWidth: 260, alignment: .leading)\n }\n\n @ViewBuilder\n private func row(for entry: MenuDescriptor.Entry) -> some View {\n switch entry {\n case let .text(text, style):\n switch style {\n case .headline:\n Text(text).font(.headline)\n case .primary:\n Text(text)\n case .secondary:\n Text(text).foregroundStyle(.secondary).font(.footnote)\n }\n case let .action(title, action):\n Button {\n self.perform(action)\n } label: {\n if let icon = self.iconName(for: action) {\n HStack(spacing: 8) {\n Image(systemName: icon)\n .imageScale(.medium)\n .frame(width: 18, alignment: .center)\n Text(title)\n }\n .foregroundStyle(.primary)\n } else {\n Text(title)\n }\n }\n .buttonStyle(.plain)\n case .divider:\n Divider()\n }\n }\n\n private func iconName(for action: MenuDescriptor.MenuAction) -> String? {\n action.systemImageName\n }\n\n private func perform(_ action: MenuDescriptor.MenuAction) {\n switch action {\n case .refresh:\n self.actions.refresh()\n case .refreshAugmentSession:\n self.actions.refreshAugmentSession()\n case .installUpdate:\n self.actions.installUpdate()\n case .dashboard:\n self.actions.openDashboard()\n case .statusPage:\n self.actions.openStatusPage()\n case let .switchAccount(provider):\n self.actions.switchAccount(provider)\n case let .openTerminal(command):\n self.actions.openTerminal(command)\n case let .loginToProvider(url):\n if let urlObj = URL(string: url) {\n NSWorkspace.shared.open(urlObj)\n }\n case .settings:\n self.actions.openSettings()\n case .about:\n self.actions.openAbout()\n case .quit:\n self.actions.quit()\n case let .copyError(message):\n self.actions.copyError(message)\n }\n }\n}\n\nstruct MenuActions {\n let installUpdate: () -> Void\n let refresh: () -> Void\n let refreshAugmentSession: () -> Void\n let openDashboard: () -> Void\n let openStatusPage: () -> Void\n let switchAccount: (UsageProvider) -> Void\n let openTerminal: (String) -> Void\n let openSettings: () -> Void\n let openAbout: () -> Void\n let quit: () -> Void\n let copyError: (String) -> Void\n}\n\n@MainActor\nstruct StatusIconView: View {\n @Bindable var store: UsageStore\n let provider: UsageProvider\n\n var body: some View {\n Image(nsImage: self.icon)\n .renderingMode(.template)\n .interpolation(.none)\n }\n\n private var icon: NSImage {\n IconRenderer.makeIcon(\n primaryRemaining: self.store.snapshot(for: self.provider)?.primary?.remainingPercent,\n weeklyRemaining: self.store.snapshot(for: self.provider)?.secondary?.remainingPercent,\n creditsRemaining: self.provider == .codex ? self.store.credits?.remaining : nil,\n stale: self.store.isStale(provider: self.provider),\n style: self.store.style(for: self.provider),\n statusIndicator: self.store.statusIndicator(for: self.provider))\n }\n}\n" }, { "path": "Sources/CodexBar/MenuDescriptor.swift", "content": "import CodexBarCore\nimport Foundation\n\n@MainActor\nstruct MenuDescriptor {\n struct Section {\n var entries: [Entry]\n }\n\n enum Entry {\n case text(String, TextStyle)\n case action(String, MenuAction)\n case divider\n }\n\n enum MenuActionSystemImage: String {\n case refresh = \"arrow.clockwise\"\n case dashboard = \"chart.bar\"\n case statusPage = \"waveform.path.ecg\"\n case switchAccount = \"key\"\n case openTerminal = \"terminal\"\n case loginToProvider = \"arrow.right.square\"\n case settings = \"gearshape\"\n case about = \"info.circle\"\n case quit = \"xmark.rectangle\"\n case copyError = \"doc.on.doc\"\n }\n\n enum TextStyle {\n case headline\n case primary\n case secondary\n }\n\n enum MenuAction {\n case installUpdate\n case refresh\n case refreshAugmentSession\n case dashboard\n case statusPage\n case switchAccount(UsageProvider)\n case openTerminal(command: String)\n case loginToProvider(url: String)\n case settings\n case about\n case quit\n case copyError(String)\n }\n\n var sections: [Section]\n\n static func build(\n provider: UsageProvider?,\n store: UsageStore,\n settings: SettingsStore,\n account: AccountInfo,\n updateReady: Bool,\n includeContextualActions: Bool = true) -> MenuDescriptor\n {\n var sections: [Section] = []\n\n if let provider {\n sections.append(Self.usageSection(for: provider, store: store, settings: settings))\n if let accountSection = Self.accountSection(\n for: provider,\n store: store,\n settings: settings,\n account: account)\n {\n sections.append(accountSection)\n }\n } else {\n var addedUsage = false\n\n for enabledProvider in store.enabledProviders() {\n sections.append(Self.usageSection(for: enabledProvider, store: store, settings: settings))\n addedUsage = true\n }\n if addedUsage {\n if let accountProvider = Self.accountProviderForCombined(store: store),\n let accountSection = Self.accountSection(\n for: accountProvider,\n store: store,\n settings: settings,\n account: account)\n {\n sections.append(accountSection)\n }\n } else {\n sections.append(Section(entries: [.text(\"No usage configured.\", .secondary)]))\n }\n }\n\n if includeContextualActions {\n let actions = Self.actionsSection(for: provider, store: store, account: account)\n if !actions.entries.isEmpty {\n sections.append(actions)\n }\n }\n sections.append(Self.metaSection(updateReady: updateReady))\n\n return MenuDescriptor(sections: sections)\n }\n\n private static func usageSection(\n for provider: UsageProvider,\n store: UsageStore,\n settings: SettingsStore) -> Section\n {\n let meta = store.metadata(for: provider)\n var entries: [Entry] = []\n let headlineText: String = {\n if let ver = Self.versionNumber(for: provider, store: store) { return \"\\(meta.displayName) \\(ver)\" }\n return meta.displayName\n }()\n entries.append(.text(headlineText, .headline))\n\n if let snap = store.snapshot(for: provider) {\n let resetStyle = settings.resetTimeDisplayStyle\n if let primary = snap.primary {\n let primaryWindow = if provider == .warp || provider == .kilo {\n // Warp/Kilo primary uses resetDescription for non-reset detail (e.g., \"Unlimited\", \"X/Y credits\").\n // Avoid rendering it as a \"Resets ...\" line.\n RateWindow(\n usedPercent: primary.usedPercent,\n windowMinutes: primary.windowMinutes,\n resetsAt: primary.resetsAt,\n resetDescription: nil)\n } else {\n primary\n }\n Self.appendRateWindow(\n entries: &entries,\n title: meta.sessionLabel,\n window: primaryWindow,\n resetStyle: resetStyle,\n showUsed: settings.usageBarsShowUsed)\n if provider == .warp || provider == .kilo,\n let detail = primary.resetDescription?.trimmingCharacters(in: .whitespacesAndNewlines),\n !detail.isEmpty\n {\n entries.append(.text(detail, .secondary))\n }\n }\n if let weekly = snap.secondary {\n let weeklyResetOverride: String? = {\n guard provider == .warp || provider == .kilo else { return nil }\n let detail = weekly.resetDescription?.trimmingCharacters(in: .whitespacesAndNewlines)\n guard let detail, !detail.isEmpty else { return nil }\n if provider == .kilo, weekly.resetsAt != nil {\n return nil\n }\n return detail\n }()\n Self.appendRateWindow(\n entries: &entries,\n title: meta.weeklyLabel,\n window: weekly,\n resetStyle: resetStyle,\n showUsed: settings.usageBarsShowUsed,\n resetOverride: weeklyResetOverride)\n if provider == .kilo,\n weekly.resetsAt != nil,\n let detail = weekly.resetDescription?.trimmingCharacters(in: .whitespacesAndNewlines),\n !detail.isEmpty\n {\n entries.append(.text(detail, .secondary))\n }\n if let pace = store.weeklyPace(provider: provider, window: weekly) {\n let paceSummary = UsagePaceText.weeklySummary(pace: pace)\n entries.append(.text(paceSummary, .secondary))\n }\n }\n if meta.supportsOpus, let opus = snap.tertiary {\n Self.appendRateWindow(\n entries: &entries,\n title: meta.opusLabel ?? \"Sonnet\",\n window: opus,\n resetStyle: resetStyle,\n showUsed: settings.usageBarsShowUsed)\n }\n\n if let cost = snap.providerCost {\n if cost.currencyCode == \"Quota\" {\n let used = String(format: \"%.0f\", cost.used)\n let limit = String(format: \"%.0f\", cost.limit)\n entries.append(.text(\"Quota: \\(used) / \\(limit)\", .primary))\n }\n }\n } else {\n entries.append(.text(\"No usage yet\", .secondary))\n }\n\n let usageContext = ProviderMenuUsageContext(\n provider: provider,\n store: store,\n settings: settings,\n metadata: meta,\n snapshot: store.snapshot(for: provider))\n ProviderCatalog.implementation(for: provider)?\n .appendUsageMenuEntries(context: usageContext, entries: &entries)\n\n return Section(entries: entries)\n }\n\n private static func accountSection(\n for provider: UsageProvider,\n store: UsageStore,\n settings: SettingsStore,\n account: AccountInfo) -> Section?\n {\n let snapshot = store.snapshot(for: provider)\n let metadata = store.metadata(for: provider)\n let entries = Self.accountEntries(\n provider: provider,\n snapshot: snapshot,\n metadata: metadata,\n fallback: account,\n hidePersonalInfo: settings.hidePersonalInfo)\n guard !entries.isEmpty else { return nil }\n return Section(entries: entries)\n }\n\n private static func accountEntries(\n provider: UsageProvider,\n snapshot: UsageSnapshot?,\n metadata: ProviderMetadata,\n fallback: AccountInfo,\n hidePersonalInfo: Bool) -> [Entry]\n {\n var entries: [Entry] = []\n let emailText = snapshot?.accountEmail(for: provider)?\n .trimmingCharacters(in: .whitespacesAndNewlines)\n let loginMethodText = snapshot?.loginMethod(for: provider)?\n .trimmingCharacters(in: .whitespacesAndNewlines)\n let redactedEmail = PersonalInfoRedactor.redactEmail(emailText, isEnabled: hidePersonalInfo)\n\n if let emailText, !emailText.isEmpty {\n entries.append(.text(\"Account: \\(redactedEmail)\", .secondary))\n }\n if provider == .kilo {\n let kiloLogin = self.kiloLoginParts(loginMethod: loginMethodText)\n if let pass = kiloLogin.pass {\n entries.append(.text(\"Plan: \\(AccountFormatter.plan(pass))\", .secondary))\n }\n for detail in kiloLogin.details {\n entries.append(.text(\"Activity: \\(detail)\", .secondary))\n }\n } else if let loginMethodText, !loginMethodText.isEmpty {\n entries.append(.text(\"Plan: \\(AccountFormatter.plan(loginMethodText))\", .secondary))\n }\n\n if metadata.usesAccountFallback {\n if emailText?.isEmpty ?? true, let fallbackEmail = fallback.email, !fallbackEmail.isEmpty {\n let redacted = PersonalInfoRedactor.redactEmail(fallbackEmail, isEnabled: hidePersonalInfo)\n entries.append(.text(\"Account: \\(redacted)\", .secondary))\n }\n if loginMethodText?.isEmpty ?? true, let fallbackPlan = fallback.plan, !fallbackPlan.isEmpty {\n entries.append(.text(\"Plan: \\(AccountFormatter.plan(fallbackPlan))\", .secondary))\n }\n }\n\n return entries\n }\n\n private static func kiloLoginParts(loginMethod: String?) -> (pass: String?, details: [String]) {\n guard let loginMethod else {\n return (nil, [])\n }\n let parts = loginMethod\n .components(separatedBy: \"·\")\n .map { $0.trimmingCharacters(in: .whitespacesAndNewlines) }\n .filter { !$0.isEmpty }\n guard !parts.isEmpty else {\n return (nil, [])\n }\n let first = parts[0]\n if self.isKiloActivitySegment(first) {\n return (nil, parts)\n }\n return (first, Array(parts.dropFirst()))\n }\n\n private static func isKiloActivitySegment(_ text: String) -> Bool {\n let normalized = text.trimmingCharacters(in: .whitespacesAndNewlines).lowercased()\n return normalized.hasPrefix(\"auto top-up:\")\n }\n\n private static func accountProviderForCombined(store: UsageStore) -> UsageProvider? {\n for provider in store.enabledProviders() {\n let metadata = store.metadata(for: provider)\n if store.snapshot(for: provider)?.identity(for: provider) != nil {\n return provider\n }\n if metadata.usesAccountFallback {\n return provider\n }\n }\n return nil\n }\n\n private static func actionsSection(\n for provider: UsageProvider?,\n store: UsageStore,\n account: AccountInfo) -> Section\n {\n var entries: [Entry] = []\n let targetProvider = provider ?? store.enabledProviders().first\n let metadata = targetProvider.map { store.metadata(for: $0) }\n let loginContext = targetProvider.map {\n ProviderMenuLoginContext(\n provider: $0,\n store: store,\n settings: store.settings,\n account: account)\n }\n\n // Show \"Add Account\" if no account, \"Switch Account\" if logged in\n if let targetProvider,\n let implementation = ProviderCatalog.implementation(for: targetProvider),\n implementation.supportsLoginFlow\n {\n if let loginContext,\n let override = implementation.loginMenuAction(context: loginContext)\n {\n entries.append(.action(override.label, override.action))\n } else {\n let loginAction = self.switchAccountTarget(for: provider, store: store)\n let hasAccount = self.hasAccount(for: provider, store: store, account: account)\n let accountLabel = hasAccount ? \"Switch Account...\" : \"Add Account...\"\n entries.append(.action(accountLabel, loginAction))\n }\n }\n\n if let targetProvider {\n let actionContext = ProviderMenuActionContext(\n provider: targetProvider,\n store: store,\n settings: store.settings,\n account: account)\n ProviderCatalog.implementation(for: targetProvider)?\n .appendActionMenuEntries(context: actionContext, entries: &entries)\n }\n\n if metadata?.dashboardURL != nil {\n entries.append(.action(\"Usage Dashboard\", .dashboard))\n }\n if metadata?.statusPageURL != nil || metadata?.statusLinkURL != nil {\n entries.append(.action(\"Status Page\", .statusPage))\n }\n\n if let statusLine = self.statusLine(for: provider, store: store) {\n entries.append(.text(statusLine, .secondary))\n }\n\n return Section(entries: entries)\n }\n\n private static func metaSection(updateReady: Bool) -> Section {\n var entries: [Entry] = []\n if updateReady {\n entries.append(.action(\"Update ready, restart now?\", .installUpdate))\n }\n entries.append(contentsOf: [\n .action(\"Settings...\", .settings),\n .action(\"About CodexBar\", .about),\n .action(\"Quit\", .quit),\n ])\n return Section(entries: entries)\n }\n\n private static func statusLine(for provider: UsageProvider?, store: UsageStore) -> String? {\n let target = provider ?? store.enabledProviders().first\n guard let target,\n let status = store.status(for: target),\n status.indicator != .none else { return nil }\n\n let description = status.description?.trimmingCharacters(in: .whitespacesAndNewlines)\n let label = description?.isEmpty == false ? description! : status.indicator.label\n if let updated = status.updatedAt {\n let freshness = UsageFormatter.updatedString(from: updated)\n return \"\\(label) — \\(freshness)\"\n }\n return label\n }\n\n private static func switchAccountTarget(for provider: UsageProvider?, store: UsageStore) -> MenuAction {\n if let provider { return .switchAccount(provider) }\n if let enabled = store.enabledProviders().first { return .switchAccount(enabled) }\n return .switchAccount(.codex)\n }\n\n private static func hasAccount(for provider: UsageProvider?, store: UsageStore, account: AccountInfo) -> Bool {\n let target = provider ?? store.enabledProviders().first ?? .codex\n if let email = store.snapshot(for: target)?.accountEmail(for: target),\n !email.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty\n {\n return true\n }\n let metadata = store.metadata(for: target)\n if metadata.usesAccountFallback,\n let fallback = account.email?.trimmingCharacters(in: .whitespacesAndNewlines),\n !fallback.isEmpty\n {\n return true\n }\n return false\n }\n\n private static func appendRateWindow(\n entries: inout [Entry],\n title: String,\n window: RateWindow,\n resetStyle: ResetTimeDisplayStyle,\n showUsed: Bool,\n resetOverride: String? = nil)\n {\n let line = UsageFormatter\n .usageLine(remaining: window.remainingPercent, used: window.usedPercent, showUsed: showUsed)\n entries.append(.text(\"\\(title): \\(line)\", .primary))\n if let resetOverride {\n entries.append(.text(resetOverride, .secondary))\n } else if let reset = UsageFormatter.resetLine(for: window, style: resetStyle) {\n entries.append(.text(reset, .secondary))\n }\n }\n\n private static func versionNumber(for provider: UsageProvider, store: UsageStore) -> String? {\n guard let raw = store.version(for: provider) else { return nil }\n let pattern = #\"[0-9]+(?:\\.[0-9]+)*\"#\n guard let regex = try? NSRegularExpression(pattern: pattern) else { return nil }\n let range = NSRange(raw.startIndex.. String {\n let cleaned = UsageFormatter.cleanPlanName(text)\n return cleaned.isEmpty ? text : cleaned\n }\n\n static func email(_ text: String) -> String {\n text\n }\n}\n\nextension MenuDescriptor.MenuAction {\n var systemImageName: String? {\n switch self {\n case .installUpdate, .settings, .about, .quit:\n nil\n case .refresh: MenuDescriptor.MenuActionSystemImage.refresh.rawValue\n case .refreshAugmentSession: MenuDescriptor.MenuActionSystemImage.refresh.rawValue\n case .dashboard: MenuDescriptor.MenuActionSystemImage.dashboard.rawValue\n case .statusPage: MenuDescriptor.MenuActionSystemImage.statusPage.rawValue\n case .switchAccount: MenuDescriptor.MenuActionSystemImage.switchAccount.rawValue\n case .openTerminal: MenuDescriptor.MenuActionSystemImage.openTerminal.rawValue\n case .loginToProvider: MenuDescriptor.MenuActionSystemImage.loginToProvider.rawValue\n case .copyError: MenuDescriptor.MenuActionSystemImage.copyError.rawValue\n }\n }\n}\n" }, { "path": "Sources/CodexBar/MenuHighlightStyle.swift", "content": "import SwiftUI\n\nextension EnvironmentValues {\n @Entry var menuItemHighlighted: Bool = false\n}\n\nenum MenuHighlightStyle {\n static let selectionText = Color(nsColor: .selectedMenuItemTextColor)\n static let normalPrimaryText = Color(nsColor: .controlTextColor)\n static let normalSecondaryText = Color(nsColor: .secondaryLabelColor)\n\n static func primary(_ highlighted: Bool) -> Color {\n highlighted ? self.selectionText : self.normalPrimaryText\n }\n\n static func secondary(_ highlighted: Bool) -> Color {\n highlighted ? self.selectionText : self.normalSecondaryText\n }\n\n static func error(_ highlighted: Bool) -> Color {\n highlighted ? self.selectionText : Color(nsColor: .systemRed)\n }\n\n static func progressTrack(_ highlighted: Bool) -> Color {\n highlighted ? self.selectionText.opacity(0.22) : Color(nsColor: .tertiaryLabelColor).opacity(0.22)\n }\n\n static func progressTint(_ highlighted: Bool, fallback: Color) -> Color {\n highlighted ? self.selectionText : fallback\n }\n\n static func selectionBackground(_ highlighted: Bool) -> Color {\n highlighted ? Color(nsColor: .selectedContentBackgroundColor) : .clear\n }\n}\n" }, { "path": "Sources/CodexBar/MiniMaxAPITokenStore.swift", "content": "import CodexBarCore\nimport Foundation\nimport Security\n\nprotocol MiniMaxAPITokenStoring: Sendable {\n func loadToken() throws -> String?\n func storeToken(_ token: String?) throws\n}\n\nenum MiniMaxAPITokenStoreError: LocalizedError {\n case keychainStatus(OSStatus)\n case invalidData\n\n var errorDescription: String? {\n switch self {\n case let .keychainStatus(status):\n \"Keychain error: \\(status)\"\n case .invalidData:\n \"Keychain returned invalid data.\"\n }\n }\n}\n\nstruct KeychainMiniMaxAPITokenStore: MiniMaxAPITokenStoring {\n private static let log = CodexBarLog.logger(LogCategories.minimaxAPITokenStore)\n\n private let service = \"com.steipete.CodexBar\"\n private let account = \"minimax-api-token\"\n\n func loadToken() throws -> String? {\n guard !KeychainAccessGate.isDisabled else {\n Self.log.debug(\"Keychain access disabled; skipping token load\")\n return nil\n }\n var result: CFTypeRef?\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n kSecMatchLimit as String: kSecMatchLimitOne,\n kSecReturnData as String: true,\n ]\n\n if case .interactionRequired = KeychainAccessPreflight\n .checkGenericPassword(service: self.service, account: self.account)\n {\n KeychainPromptHandler.handler?(KeychainPromptContext(\n kind: .minimaxToken,\n service: self.service,\n account: self.account))\n }\n\n let status = SecItemCopyMatching(query as CFDictionary, &result)\n if status == errSecItemNotFound {\n return nil\n }\n guard status == errSecSuccess else {\n Self.log.error(\"Keychain read failed: \\(status)\")\n throw MiniMaxAPITokenStoreError.keychainStatus(status)\n }\n\n guard let data = result as? Data else {\n throw MiniMaxAPITokenStoreError.invalidData\n }\n let token = String(data: data, encoding: .utf8)?.trimmingCharacters(in: .whitespacesAndNewlines)\n if let token, !token.isEmpty {\n return token\n }\n return nil\n }\n\n func storeToken(_ token: String?) throws {\n guard !KeychainAccessGate.isDisabled else {\n Self.log.debug(\"Keychain access disabled; skipping token store\")\n return\n }\n let cleaned = token?.trimmingCharacters(in: .whitespacesAndNewlines)\n if cleaned == nil || cleaned?.isEmpty == true {\n try self.deleteIfPresent()\n return\n }\n\n let data = cleaned!.data(using: .utf8)!\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n ]\n let attributes: [String: Any] = [\n kSecValueData as String: data,\n kSecAttrAccessible as String: kSecAttrAccessibleAfterFirstUnlockThisDeviceOnly,\n ]\n\n let updateStatus = SecItemUpdate(query as CFDictionary, attributes as CFDictionary)\n if updateStatus == errSecSuccess {\n return\n }\n if updateStatus != errSecItemNotFound {\n Self.log.error(\"Keychain update failed: \\(updateStatus)\")\n throw MiniMaxAPITokenStoreError.keychainStatus(updateStatus)\n }\n\n var addQuery = query\n for (key, value) in attributes {\n addQuery[key] = value\n }\n let addStatus = SecItemAdd(addQuery as CFDictionary, nil)\n guard addStatus == errSecSuccess else {\n Self.log.error(\"Keychain add failed: \\(addStatus)\")\n throw MiniMaxAPITokenStoreError.keychainStatus(addStatus)\n }\n }\n\n private func deleteIfPresent() throws {\n guard !KeychainAccessGate.isDisabled else { return }\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n ]\n let status = SecItemDelete(query as CFDictionary)\n if status == errSecSuccess || status == errSecItemNotFound {\n return\n }\n Self.log.error(\"Keychain delete failed: \\(status)\")\n throw MiniMaxAPITokenStoreError.keychainStatus(status)\n }\n}\n" }, { "path": "Sources/CodexBar/MiniMaxCookieStore.swift", "content": "import CodexBarCore\nimport Foundation\nimport Security\n\nprotocol MiniMaxCookieStoring: Sendable {\n func loadCookieHeader() throws -> String?\n func storeCookieHeader(_ header: String?) throws\n}\n\nenum MiniMaxCookieStoreError: LocalizedError {\n case keychainStatus(OSStatus)\n case invalidData\n\n var errorDescription: String? {\n switch self {\n case let .keychainStatus(status):\n \"Keychain error: \\(status)\"\n case .invalidData:\n \"Keychain returned invalid data.\"\n }\n }\n}\n\nstruct KeychainMiniMaxCookieStore: MiniMaxCookieStoring {\n private static let log = CodexBarLog.logger(LogCategories.minimaxCookieStore)\n\n private let service = \"com.steipete.CodexBar\"\n private let account = \"minimax-cookie\"\n\n func loadCookieHeader() throws -> String? {\n guard !KeychainAccessGate.isDisabled else {\n Self.log.debug(\"Keychain access disabled; skipping cookie load\")\n return nil\n }\n var result: CFTypeRef?\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n kSecMatchLimit as String: kSecMatchLimitOne,\n kSecReturnData as String: true,\n ]\n\n if case .interactionRequired = KeychainAccessPreflight\n .checkGenericPassword(service: self.service, account: self.account)\n {\n KeychainPromptHandler.handler?(KeychainPromptContext(\n kind: .minimaxCookie,\n service: self.service,\n account: self.account))\n }\n\n let status = SecItemCopyMatching(query as CFDictionary, &result)\n if status == errSecItemNotFound {\n return nil\n }\n guard status == errSecSuccess else {\n Self.log.error(\"Keychain read failed: \\(status)\")\n throw MiniMaxCookieStoreError.keychainStatus(status)\n }\n\n guard let data = result as? Data else {\n throw MiniMaxCookieStoreError.invalidData\n }\n let header = String(data: data, encoding: .utf8)?.trimmingCharacters(in: .whitespacesAndNewlines)\n if let header, !header.isEmpty {\n return header\n }\n return nil\n }\n\n func storeCookieHeader(_ header: String?) throws {\n guard !KeychainAccessGate.isDisabled else {\n Self.log.debug(\"Keychain access disabled; skipping cookie store\")\n return\n }\n guard let raw = header?.trimmingCharacters(in: .whitespacesAndNewlines),\n !raw.isEmpty\n else {\n try self.deleteIfPresent()\n return\n }\n guard MiniMaxCookieHeader.normalized(from: raw) != nil else {\n try self.deleteIfPresent()\n return\n }\n\n let data = raw.data(using: .utf8)!\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n ]\n let attributes: [String: Any] = [\n kSecValueData as String: data,\n kSecAttrAccessible as String: kSecAttrAccessibleAfterFirstUnlockThisDeviceOnly,\n ]\n\n let updateStatus = SecItemUpdate(query as CFDictionary, attributes as CFDictionary)\n if updateStatus == errSecSuccess {\n return\n }\n if updateStatus != errSecItemNotFound {\n Self.log.error(\"Keychain update failed: \\(updateStatus)\")\n throw MiniMaxCookieStoreError.keychainStatus(updateStatus)\n }\n\n var addQuery = query\n for (key, value) in attributes {\n addQuery[key] = value\n }\n let addStatus = SecItemAdd(addQuery as CFDictionary, nil)\n guard addStatus == errSecSuccess else {\n Self.log.error(\"Keychain add failed: \\(addStatus)\")\n throw MiniMaxCookieStoreError.keychainStatus(addStatus)\n }\n }\n\n private func deleteIfPresent() throws {\n guard !KeychainAccessGate.isDisabled else { return }\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n ]\n let status = SecItemDelete(query as CFDictionary)\n if status == errSecSuccess || status == errSecItemNotFound {\n return\n }\n Self.log.error(\"Keychain delete failed: \\(status)\")\n throw MiniMaxCookieStoreError.keychainStatus(status)\n }\n}\n" }, { "path": "Sources/CodexBar/MouseLocationReader.swift", "content": "import AppKit\nimport SwiftUI\n\n/// Lightweight NSView-based mouse tracking with local coordinates.\n///\n/// Why: SwiftUI's `onHover` doesn't provide location, but we want \"hover a bar to see values\" on macOS.\n@MainActor\nstruct MouseLocationReader: NSViewRepresentable {\n let onMoved: (CGPoint?) -> Void\n\n func makeNSView(context: Context) -> TrackingView {\n let view = TrackingView()\n view.onMoved = self.onMoved\n return view\n }\n\n func updateNSView(_ nsView: TrackingView, context: Context) {\n nsView.onMoved = self.onMoved\n }\n\n final class TrackingView: NSView {\n var onMoved: ((CGPoint?) -> Void)?\n private var trackingArea: NSTrackingArea?\n\n override func viewDidMoveToWindow() {\n super.viewDidMoveToWindow()\n self.window?.acceptsMouseMovedEvents = true\n self.updateTrackingAreas()\n }\n\n override func updateTrackingAreas() {\n super.updateTrackingAreas()\n if let trackingArea {\n self.removeTrackingArea(trackingArea)\n }\n\n let options: NSTrackingArea.Options = [\n // NSMenu popups aren't \"key windows\", so `.activeInKeyWindow` would drop events and cause hover\n // state to flicker. `.activeAlways` keeps tracking stable while the menu is open.\n .activeAlways,\n .inVisibleRect,\n .mouseEnteredAndExited,\n .mouseMoved,\n ]\n let area = NSTrackingArea(rect: .zero, options: options, owner: self, userInfo: nil)\n self.addTrackingArea(area)\n self.trackingArea = area\n }\n\n override func mouseEntered(with event: NSEvent) {\n super.mouseEntered(with: event)\n self.onMoved?(self.convert(event.locationInWindow, from: nil))\n }\n\n override func mouseMoved(with event: NSEvent) {\n super.mouseMoved(with: event)\n self.onMoved?(self.convert(event.locationInWindow, from: nil))\n }\n\n override func mouseExited(with event: NSEvent) {\n super.mouseExited(with: event)\n self.onMoved?(nil)\n }\n }\n}\n" }, { "path": "Sources/CodexBar/Notifications+CodexBar.swift", "content": "import Foundation\n\nextension Notification.Name {\n static let codexbarOpenSettings = Notification.Name(\"codexbarOpenSettings\")\n static let codexbarDebugBlinkNow = Notification.Name(\"codexbarDebugBlinkNow\")\n static let codexbarProviderConfigDidChange = Notification.Name(\"codexbarProviderConfigDidChange\")\n}\n" }, { "path": "Sources/CodexBar/OpenAICreditsPurchaseWindowController.swift", "content": "import AppKit\nimport CodexBarCore\nimport WebKit\n\n@MainActor\nfinal class OpenAICreditsPurchaseWindowController: NSWindowController, WKNavigationDelegate, WKScriptMessageHandler {\n private static let defaultSize = NSSize(width: 980, height: 760)\n private static let logHandlerName = \"codexbarLog\"\n private static let debugLogURL = URL(fileURLWithPath: NSTemporaryDirectory())\n .appendingPathComponent(\"codexbar-buy-credits.log\")\n private static let autoStartScript = \"\"\"\n (() => {\n if (window.__codexbarAutoBuyCreditsStarted) return 'already';\n const log = (...args) => {\n try {\n window.webkit?.messageHandlers?.codexbarLog?.postMessage(args);\n } catch {}\n };\n const buttonSelector = 'button, a, [role=\"button\"], input[type=\"button\"], input[type=\"submit\"]';\n const isVisible = (el) => {\n if (!el || !el.getBoundingClientRect) return false;\n const rect = el.getBoundingClientRect();\n if (rect.width < 2 || rect.height < 2) return false;\n const style = window.getComputedStyle ? window.getComputedStyle(el) : null;\n if (style) {\n if (style.display === 'none' || style.visibility === 'hidden') return false;\n if (parseFloat(style.opacity || '1') === 0) return false;\n }\n return true;\n };\n const textOf = el => {\n const raw = el && (el.innerText || el.textContent) ? String(el.innerText || el.textContent) : '';\n return raw.trim();\n };\n const matches = text => {\n const lower = String(text || '').toLowerCase();\n if (!lower.includes('credit')) return false;\n return (\n lower.includes('buy') ||\n lower.includes('add') ||\n lower.includes('purchase') ||\n lower.includes('top up') ||\n lower.includes('top-up')\n );\n };\n const matchesAddMore = text => {\n const lower = String(text || '').toLowerCase();\n return lower.includes('add more');\n };\n const labelFor = el => {\n if (!el) return '';\n return textOf(el) || el.getAttribute('aria-label') || el.getAttribute('title') || el.value || '';\n };\n const summarize = el => {\n if (!el) return null;\n return {\n tag: el.tagName,\n type: el.getAttribute('type'),\n role: el.getAttribute('role'),\n label: labelFor(el),\n aria: el.getAttribute('aria-label'),\n disabled: isDisabled(el),\n href: el.getAttribute('href'),\n testId: el.getAttribute('data-testid'),\n className: (el.className && String(el.className).slice(0, 120)) || ''\n };\n };\n const collectButtons = () => {\n const results = new Set();\n const addAll = (root) => {\n if (!root || !root.querySelectorAll) return;\n root.querySelectorAll(buttonSelector).forEach(el => results.add(el));\n };\n addAll(document);\n document.querySelectorAll('*').forEach(el => {\n if (el.shadowRoot) addAll(el.shadowRoot);\n });\n document.querySelectorAll('iframe').forEach(frame => {\n try {\n const doc = frame.contentDocument;\n if (!doc) return;\n addAll(doc);\n doc.querySelectorAll('*').forEach(el => {\n if (el.shadowRoot) addAll(el.shadowRoot);\n });\n } catch {}\n });\n return Array.from(results);\n };\n const findDialogNextButton = () => {\n const dialog = document.querySelector('[role=\\\"dialog\\\"], dialog, [aria-modal=\\\"true\\\"]');\n if (!dialog) return null;\n const buttons = Array.from(dialog.querySelectorAll(buttonSelector));\n const labeled = buttons.filter(btn => labelFor(btn).toLowerCase().startsWith('next'));\n const visible = labeled.find(isVisible);\n return visible || labeled[0] || null;\n };\n const clickButton = (el) => {\n if (!el) return false;\n try {\n el.dispatchEvent(new MouseEvent('click', { bubbles: true, cancelable: true, view: window }));\n } catch {\n try {\n el.click();\n } catch {\n return false;\n }\n }\n return true;\n };\n const triggerPointerClick = (el) => {\n if (!el) return false;\n const rect = el.getBoundingClientRect ? el.getBoundingClientRect() : null;\n if (!rect) return false;\n const x = rect.left + rect.width / 2;\n const y = rect.top + rect.height / 2;\n const events = ['pointerdown', 'mousedown', 'pointerup', 'mouseup', 'click'];\n for (const type of events) {\n try {\n el.dispatchEvent(new MouseEvent(type, {\n bubbles: true,\n cancelable: true,\n view: window,\n clientX: x,\n clientY: y\n }));\n } catch {\n return false;\n }\n }\n return true;\n };\n const pickLikelyButton = (buttons) => {\n if (!buttons || buttons.length === 0) return null;\n const labeled = buttons.find(btn => {\n const label = labelFor(btn);\n if (matches(label) || matchesAddMore(label)) return true;\n const aria = String(btn.getAttribute('aria-label') || '').toLowerCase();\n return aria.includes('credit') || aria.includes('buy') || aria.includes('add');\n });\n return labeled || buttons[0];\n };\n const findAddMoreButton = () => {\n const buttons = collectButtons();\n return buttons.find(btn => matchesAddMore(labelFor(btn))) || null;\n };\n const findNextButton = () => {\n const dialogNext = findDialogNextButton();\n if (dialogNext) return dialogNext;\n const buttons = collectButtons();\n const labeled = buttons.filter(btn => {\n const label = labelFor(btn).toLowerCase();\n return label === 'next' || label.startsWith('next ');\n });\n const visible = labeled.find(isVisible);\n if (visible) return visible;\n const submit = buttons.find(btn => btn.type && String(btn.type).toLowerCase() === 'submit' && isVisible(btn));\n return submit || labeled[0] || null;\n };\n const isDisabled = (el) => {\n if (!el) return true;\n if (el.disabled) return true;\n const ariaDisabled = String(el.getAttribute('aria-disabled') || '').toLowerCase();\n if (ariaDisabled === 'true') return true;\n if (el.classList && (el.classList.contains('disabled') || el.classList.contains('is-disabled'))) {\n return true;\n }\n return false;\n };\n const forceClickElement = (el) => {\n if (!el) return false;\n const rect = el.getBoundingClientRect ? el.getBoundingClientRect() : null;\n if (rect) {\n const x = rect.left + rect.width / 2;\n const y = rect.top + rect.height / 2;\n const target = document.elementFromPoint(x, y);\n if (target) {\n try {\n target.dispatchEvent(new MouseEvent('click', { bubbles: true, cancelable: true, view: window }));\n return true;\n } catch {\n return false;\n }\n }\n }\n return false;\n };\n const requestSubmit = (el) => {\n if (!el || !el.closest) return false;\n const form = el.closest('form');\n if (!form) return false;\n if (typeof form.requestSubmit === 'function') {\n form.requestSubmit(el);\n return true;\n }\n if (typeof form.submit === 'function') {\n form.submit();\n return true;\n }\n return false;\n };\n const clickNextIfReady = (attempts) => {\n const nextButton = findNextButton();\n if (!nextButton) {\n if (attempts && attempts % 5 === 0) log('next_missing', { attempts });\n return false;\n }\n if (isDisabled(nextButton)) {\n if (attempts && attempts % 5 === 0) log('next_disabled', summarize(nextButton));\n return false;\n }\n if (!isVisible(nextButton)) {\n if (attempts && attempts % 5 === 0) log('next_hidden', summarize(nextButton));\n return false;\n }\n nextButton.focus?.();\n if (requestSubmit(nextButton)) {\n log('next_submit', summarize(nextButton));\n return true;\n }\n if (triggerPointerClick(nextButton)) {\n log('next_pointer', summarize(nextButton));\n return true;\n }\n if (clickButton(nextButton)) {\n log('next_click', summarize(nextButton));\n return true;\n }\n return forceClickElement(nextButton);\n };\n const startNextPolling = (initialDelay = 500, interval = 500, maxAttempts = 90) => {\n if (window.__codexbarNextPolling) return;\n window.__codexbarNextPolling = true;\n log('start_next_poll', { initialDelay, interval, maxAttempts });\n setTimeout(() => {\n let attempts = 0;\n const nextTimer = setInterval(() => {\n attempts += 1;\n if (attempts % 5 === 0) {\n const nextButton = findNextButton();\n log('next_poll', {\n attempts,\n found: Boolean(nextButton),\n summary: summarize(nextButton)\n });\n }\n if (clickNextIfReady(attempts) || attempts >= maxAttempts) {\n clearInterval(nextTimer);\n }\n }, interval);\n }, initialDelay);\n };\n const observeNextButton = () => {\n if (window.__codexbarNextObserver || !window.MutationObserver) return;\n const observer = new MutationObserver(() => {\n if (clickNextIfReady(1)) {\n observer.disconnect();\n window.__codexbarNextObserver = null;\n }\n });\n observer.observe(document.body, { subtree: true, childList: true, attributes: true });\n window.__codexbarNextObserver = observer;\n };\n const findCreditsCardButton = () => {\n const nodes = Array.from(document.querySelectorAll('h1,h2,h3,div,span,p'));\n const labelMatch = nodes.find(node => {\n const lower = textOf(node).toLowerCase();\n return lower === 'credits remaining' || (lower.includes('credits') && lower.includes('remaining'));\n });\n if (!labelMatch) return null;\n let cur = labelMatch;\n for (let i = 0; i < 6 && cur; i++) {\n const buttons = Array.from(cur.querySelectorAll(buttonSelector));\n const picked = pickLikelyButton(buttons);\n if (picked) return picked;\n cur = cur.parentElement;\n }\n return null;\n };\n const findAndClick = () => {\n const addMoreButton = findAddMoreButton();\n if (addMoreButton) {\n log('add_more_click', summarize(addMoreButton));\n clickButton(addMoreButton);\n return true;\n }\n const cardButton = findCreditsCardButton();\n if (!cardButton) return false;\n log('credits_card_click', summarize(cardButton));\n return clickButton(cardButton);\n };\n const logDialogButtons = () => {\n const dialog = document.querySelector('[role=\\\"dialog\\\"], dialog, [aria-modal=\\\"true\\\"]');\n if (dialog) {\n const buttons = Array.from(dialog.querySelectorAll(buttonSelector)).map(summarize).filter(Boolean);\n if (buttons.length) {\n log('dialog_buttons', { count: buttons.length, buttons: buttons.slice(0, 6) });\n }\n const nextButton = findDialogNextButton();\n if (nextButton) {\n log('dialog_next', summarize(nextButton));\n setTimeout(() => clickNextIfReady(1), 100);\n }\n return;\n }\n const candidates = collectButtons()\n .map(summarize)\n .filter(Boolean)\n .filter(entry => {\n const label = (entry.label || '').toLowerCase();\n return label.includes('next')\n || label.includes('continue')\n || label.includes('confirm')\n || label.includes('buy');\n });\n if (candidates.length) {\n log('button_candidates', { count: candidates.length, buttons: candidates.slice(0, 8) });\n }\n };\n log('auto_start', { href: location.href, ready: document.readyState });\n const iframeSources = Array.from(document.querySelectorAll('iframe'))\n .map(frame => frame.getAttribute('src') || '')\n .filter(Boolean)\n .slice(0, 6);\n if (iframeSources.length) {\n log('iframes', iframeSources);\n }\n const shadowHostCount = Array.from(document.querySelectorAll('*')).filter(el => el.shadowRoot).length;\n if (shadowHostCount > 0) {\n log('shadow_roots', { count: shadowHostCount });\n }\n if (findAndClick()) {\n window.__codexbarAutoBuyCreditsStarted = true;\n startNextPolling();\n observeNextButton();\n logDialogButtons();\n return 'clicked';\n }\n startNextPolling(500);\n observeNextButton();\n logDialogButtons();\n let attempts = 0;\n const maxAttempts = 14;\n const timer = setInterval(() => {\n attempts += 1;\n if (findAndClick()) {\n logDialogButtons();\n startNextPolling();\n clearInterval(timer);\n return;\n }\n if (attempts >= maxAttempts) {\n clearInterval(timer);\n }\n }, 500);\n window.__codexbarAutoBuyCreditsStarted = true;\n return 'scheduled';\n })();\n \"\"\"\n\n private let logger = CodexBarLog.logger(LogCategories.creditsPurchase)\n private var webView: WKWebView?\n private var accountEmail: String?\n private var pendingAutoStart = false\n private let logHandler = WeakScriptMessageHandler()\n\n init() {\n super.init(window: nil)\n self.logHandler.delegate = self\n }\n\n @available(*, unavailable)\n required init?(coder: NSCoder) {\n fatalError(\"init(coder:) has not been implemented\")\n }\n\n func show(purchaseURL: URL, accountEmail: String?, autoStartPurchase: Bool) {\n let normalizedEmail = Self.normalizeEmail(accountEmail)\n if self.window == nil || normalizedEmail != self.accountEmail {\n self.accountEmail = normalizedEmail\n self.buildWindow()\n }\n Self.resetDebugLog()\n let accountValue = normalizedEmail == nil ? \"none\" : \"set\"\n let sanitizedURL = Self.sanitizedURLString(purchaseURL)\n Self.appendDebugLog(\n \"show autoStart=\\(autoStartPurchase) url=\\(sanitizedURL) account=\\(accountValue)\")\n self.logger.info(\"Buy credits window opened\")\n self.logger.debug(\"Auto-start purchase\", metadata: [\"enabled\": autoStartPurchase ? \"1\" : \"0\"])\n self.logger.debug(\"Purchase URL\", metadata: [\"url\": sanitizedURL])\n self.logger.debug(\"Account email\", metadata: [\"state\": accountValue])\n self.pendingAutoStart = autoStartPurchase\n self.load(url: purchaseURL)\n self.window?.center()\n self.showWindow(nil)\n NSApp.activate(ignoringOtherApps: true)\n }\n\n private func buildWindow() {\n let config = WKWebViewConfiguration()\n config.userContentController.add(self.logHandler, name: Self.logHandlerName)\n config.websiteDataStore = OpenAIDashboardWebsiteDataStore.store(forAccountEmail: self.accountEmail)\n\n let webView = WKWebView(frame: .zero, configuration: config)\n webView.navigationDelegate = self\n webView.allowsBackForwardNavigationGestures = true\n webView.translatesAutoresizingMaskIntoConstraints = false\n\n let container = NSView(frame: .zero)\n container.addSubview(webView)\n NSLayoutConstraint.activate([\n webView.leadingAnchor.constraint(equalTo: container.leadingAnchor),\n webView.trailingAnchor.constraint(equalTo: container.trailingAnchor),\n webView.topAnchor.constraint(equalTo: container.topAnchor),\n webView.bottomAnchor.constraint(equalTo: container.bottomAnchor),\n ])\n\n let window = NSWindow(\n contentRect: Self.defaultFrame(),\n styleMask: [.titled, .closable, .resizable],\n backing: .buffered,\n defer: false)\n window.title = \"Buy Credits\"\n window.isReleasedWhenClosed = false\n window.collectionBehavior = [.moveToActiveSpace, .fullScreenAuxiliary]\n window.contentView = container\n window.center()\n window.delegate = self\n\n self.window = window\n self.webView = webView\n }\n\n private func load(url: URL) {\n guard let webView else { return }\n let request = URLRequest(url: url)\n webView.load(request)\n }\n\n func webView(_ webView: WKWebView, didFinish navigation: WKNavigation!) {\n guard self.pendingAutoStart else { return }\n self.pendingAutoStart = false\n let currentURL = webView.url?.absoluteString ?? \"unknown\"\n Self.appendDebugLog(\"didFinish url=\\(currentURL)\")\n self.logger.debug(\"Buy credits navigation finished\", metadata: [\"url\": currentURL])\n webView.evaluateJavaScript(Self.autoStartScript) { [logger] result, error in\n if let error {\n Self.appendDebugLog(\"autoStart error=\\(error.localizedDescription)\")\n logger.error(\"Auto-start purchase failed\", metadata: [\"error\": error.localizedDescription])\n return\n }\n if let result {\n Self.appendDebugLog(\"autoStart result=\\(String(describing: result))\")\n logger.debug(\"Auto-start purchase result\", metadata: [\"result\": String(describing: result)])\n }\n }\n }\n\n func userContentController(_ userContentController: WKUserContentController, didReceive message: WKScriptMessage) {\n guard message.name == Self.logHandlerName else { return }\n let payload = String(describing: message.body)\n Self.appendDebugLog(\"js \\(payload)\")\n self.logger.debug(\"Auto-buy log\", metadata: [\"payload\": payload])\n }\n\n private static func normalizeEmail(_ email: String?) -> String? {\n guard let raw = email?.trimmingCharacters(in: .whitespacesAndNewlines), !raw.isEmpty else { return nil }\n return raw.lowercased()\n }\n\n private static func defaultFrame() -> NSRect {\n let visible = NSScreen.main?.visibleFrame ?? NSRect(x: 0, y: 0, width: 1200, height: 900)\n let width = min(Self.defaultSize.width, visible.width * 0.92)\n let height = min(Self.defaultSize.height, visible.height * 0.88)\n let origin = NSPoint(x: visible.midX - width / 2, y: visible.midY - height / 2)\n return NSRect(origin: origin, size: NSSize(width: width, height: height))\n }\n\n private static func appendDebugLog(_ message: String) {\n let timestamp = ISO8601DateFormatter().string(from: Date())\n let line = \"[\\(timestamp)] \\(LogRedactor.redact(message))\\n\"\n guard let data = line.data(using: .utf8) else { return }\n if FileManager.default.fileExists(atPath: Self.debugLogURL.path) {\n if let handle = try? FileHandle(forWritingTo: Self.debugLogURL) {\n handle.seekToEndOfFile()\n handle.write(data)\n try? handle.close()\n }\n } else {\n try? data.write(to: Self.debugLogURL, options: .atomic)\n }\n }\n\n private static func resetDebugLog() {\n try? FileManager.default.removeItem(at: self.debugLogURL)\n }\n\n private static func sanitizedURLString(_ url: URL) -> String {\n guard var components = URLComponents(url: url, resolvingAgainstBaseURL: false) else {\n return url.absoluteString\n }\n components.query = nil\n components.fragment = nil\n return components.string ?? url.absoluteString\n }\n}\n\nprivate final class WeakScriptMessageHandler: NSObject, WKScriptMessageHandler {\n weak var delegate: WKScriptMessageHandler?\n\n func userContentController(_ userContentController: WKUserContentController, didReceive message: WKScriptMessage) {\n self.delegate?.userContentController(userContentController, didReceive: message)\n }\n}\n\n// MARK: - NSWindowDelegate\n\nextension OpenAICreditsPurchaseWindowController: NSWindowDelegate {\n func windowWillClose(_ notification: Notification) {\n guard let window = self.window else { return }\n let webView = self.webView\n self.pendingAutoStart = false\n self.webView = nil\n self.window = nil\n self.logger.info(\"Buy credits window closing\")\n WebKitTeardown.scheduleCleanup(owner: window, window: window, webView: webView)\n }\n}\n" }, { "path": "Sources/CodexBar/PersonalInfoRedactor.swift", "content": "import Foundation\n\nenum PersonalInfoRedactor {\n static let emailPlaceholder = \"Hidden\"\n\n private static let emailRegex: NSRegularExpression? = {\n let pattern = #\"[A-Z0-9._%+-]+@[A-Z0-9.-]+\\.[A-Z]{2,}\"#\n return try? NSRegularExpression(pattern: pattern, options: [.caseInsensitive])\n }()\n\n static func redactEmail(_ email: String?, isEnabled: Bool) -> String {\n guard let email, !email.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty else { return \"\" }\n guard isEnabled else { return email }\n return Self.emailPlaceholder\n }\n\n static func redactEmails(in text: String?, isEnabled: Bool) -> String? {\n guard let text else { return nil }\n guard isEnabled else { return text }\n guard let regex = Self.emailRegex else { return text }\n let range = NSRange(text.startIndex.. {\n Binding(\n get: { self.updateChannel },\n set: { newValue in\n self.updateChannelRaw = newValue.rawValue\n self.updater.checkForUpdates(nil)\n })\n }\n\n private func openProjectHome() {\n guard let url = URL(string: \"https://github.com/steipete/CodexBar\") else { return }\n NSWorkspace.shared.open(url)\n }\n}\n" }, { "path": "Sources/CodexBar/PreferencesAdvancedPane.swift", "content": "import KeyboardShortcuts\nimport SwiftUI\n\n@MainActor\nstruct AdvancedPane: View {\n @Bindable var settings: SettingsStore\n @State private var isInstallingCLI = false\n @State private var cliStatus: String?\n\n var body: some View {\n ScrollView(.vertical, showsIndicators: true) {\n VStack(alignment: .leading, spacing: 16) {\n SettingsSection(contentSpacing: 8) {\n Text(\"Keyboard shortcut\")\n .font(.caption)\n .foregroundStyle(.secondary)\n .textCase(.uppercase)\n HStack(alignment: .center, spacing: 12) {\n Text(\"Open menu\")\n .font(.body)\n Spacer()\n KeyboardShortcuts.Recorder(for: .openMenu)\n }\n Text(\"Trigger the menu bar menu from anywhere.\")\n .font(.footnote)\n .foregroundStyle(.tertiary)\n }\n\n Divider()\n\n SettingsSection(contentSpacing: 10) {\n HStack(spacing: 12) {\n Button {\n Task { await self.installCLI() }\n } label: {\n if self.isInstallingCLI {\n ProgressView().controlSize(.small)\n } else {\n Text(\"Install CLI\")\n }\n }\n .disabled(self.isInstallingCLI)\n\n if let status = self.cliStatus {\n Text(status)\n .font(.footnote)\n .foregroundStyle(.tertiary)\n .lineLimit(2)\n }\n }\n Text(\"Symlink CodexBarCLI to /usr/local/bin and /opt/homebrew/bin as codexbar.\")\n .font(.footnote)\n .foregroundStyle(.tertiary)\n }\n\n Divider()\n\n SettingsSection(contentSpacing: 10) {\n PreferenceToggleRow(\n title: \"Show Debug Settings\",\n subtitle: \"Expose troubleshooting tools in the Debug tab.\",\n binding: self.$settings.debugMenuEnabled)\n PreferenceToggleRow(\n title: \"Surprise me\",\n subtitle: \"Check if you like your agents having some fun up there.\",\n binding: self.$settings.randomBlinkEnabled)\n }\n\n Divider()\n\n SettingsSection(contentSpacing: 10) {\n PreferenceToggleRow(\n title: \"Hide personal information\",\n subtitle: \"Obscure email addresses in the menu bar and menu UI.\",\n binding: self.$settings.hidePersonalInfo)\n }\n\n Divider()\n\n SettingsSection(\n title: \"Keychain access\",\n caption: \"\"\"\n Disable all Keychain reads and writes. Browser cookie import is unavailable; paste Cookie \\\n headers manually in Providers.\n \"\"\") {\n PreferenceToggleRow(\n title: \"Disable Keychain access\",\n subtitle: \"Prevents any Keychain access while enabled.\",\n binding: self.$settings.debugDisableKeychainAccess)\n }\n }\n .frame(maxWidth: .infinity, alignment: .leading)\n .padding(.horizontal, 20)\n .padding(.vertical, 12)\n }\n }\n}\n\nextension AdvancedPane {\n private func installCLI() async {\n if self.isInstallingCLI { return }\n self.isInstallingCLI = true\n defer { self.isInstallingCLI = false }\n\n let helperURL = Bundle.main.bundleURL.appendingPathComponent(\"Contents/Helpers/CodexBarCLI\")\n let fm = FileManager.default\n guard fm.fileExists(atPath: helperURL.path) else {\n self.cliStatus = \"CodexBarCLI not found in app bundle.\"\n return\n }\n\n let destinations = [\n \"/usr/local/bin/codexbar\",\n \"/opt/homebrew/bin/codexbar\",\n ]\n\n var results: [String] = []\n for dest in destinations {\n let dir = (dest as NSString).deletingLastPathComponent\n guard fm.fileExists(atPath: dir) else { continue }\n guard fm.isWritableFile(atPath: dir) else {\n results.append(\"No write access: \\(dir)\")\n continue\n }\n\n if fm.fileExists(atPath: dest) {\n if Self.isLink(atPath: dest, pointingTo: helperURL.path) {\n results.append(\"Installed: \\(dir)\")\n } else {\n results.append(\"Exists: \\(dir)\")\n }\n continue\n }\n\n do {\n try fm.createSymbolicLink(atPath: dest, withDestinationPath: helperURL.path)\n results.append(\"Installed: \\(dir)\")\n } catch {\n results.append(\"Failed: \\(dir)\")\n }\n }\n\n self.cliStatus = results.isEmpty\n ? \"No writable bin dirs found.\"\n : results.joined(separator: \" · \")\n }\n\n private static func isLink(atPath path: String, pointingTo destination: String) -> Bool {\n guard let link = try? FileManager.default.destinationOfSymbolicLink(atPath: path) else { return false }\n let dir = (path as NSString).deletingLastPathComponent\n let resolved = URL(fileURLWithPath: link, relativeTo: URL(fileURLWithPath: dir))\n .standardizedFileURL\n .path\n return resolved == destination\n }\n}\n" }, { "path": "Sources/CodexBar/PreferencesComponents.swift", "content": "import AppKit\nimport SwiftUI\n\n@MainActor\nstruct PreferenceToggleRow: View {\n let title: String\n let subtitle: String?\n @Binding var binding: Bool\n\n var body: some View {\n VStack(alignment: .leading, spacing: 5.4) {\n Toggle(isOn: self.$binding) {\n Text(self.title)\n .font(.body)\n }\n .toggleStyle(.checkbox)\n\n if let subtitle, !subtitle.isEmpty {\n Text(subtitle)\n .font(.footnote)\n .foregroundStyle(.tertiary)\n .fixedSize(horizontal: false, vertical: true)\n }\n }\n }\n}\n\n@MainActor\nstruct SettingsSection: View {\n let title: String?\n let caption: String?\n let contentSpacing: CGFloat\n private let content: () -> Content\n\n init(\n title: String? = nil,\n caption: String? = nil,\n contentSpacing: CGFloat = 14,\n @ViewBuilder content: @escaping () -> Content)\n {\n self.title = title\n self.caption = caption\n self.contentSpacing = contentSpacing\n self.content = content\n }\n\n var body: some View {\n VStack(alignment: .leading, spacing: 10) {\n if let title, !title.isEmpty {\n Text(title)\n .font(.subheadline.weight(.semibold))\n }\n if let caption {\n Text(caption)\n .font(.footnote)\n .foregroundStyle(.tertiary)\n .fixedSize(horizontal: false, vertical: true)\n }\n VStack(alignment: .leading, spacing: self.contentSpacing) {\n self.content()\n }\n .frame(maxWidth: .infinity, alignment: .leading)\n }\n }\n}\n\n@MainActor\nstruct AboutLinkRow: View {\n let icon: String\n let title: String\n let url: String\n @State private var hovering = false\n\n var body: some View {\n Button {\n if let url = URL(string: self.url) { NSWorkspace.shared.open(url) }\n } label: {\n HStack(spacing: 8) {\n Image(systemName: self.icon)\n Text(self.title)\n .underline(self.hovering, color: .accentColor)\n }\n .frame(maxWidth: .infinity)\n .padding(.vertical, 4)\n .foregroundColor(.accentColor)\n }\n .buttonStyle(.plain)\n .contentShape(Rectangle())\n .onHover { self.hovering = $0 }\n }\n}\n" }, { "path": "Sources/CodexBar/PreferencesDebugPane.swift", "content": "import AppKit\nimport CodexBarCore\nimport SwiftUI\n\n@MainActor\nstruct DebugPane: View {\n @Bindable var settings: SettingsStore\n @Bindable var store: UsageStore\n @AppStorage(\"debugFileLoggingEnabled\") private var debugFileLoggingEnabled = false\n @State private var currentLogProvider: UsageProvider = .codex\n @State private var currentFetchProvider: UsageProvider = .codex\n @State private var isLoadingLog = false\n @State private var logText: String = \"\"\n @State private var isClearingCostCache = false\n @State private var costCacheStatus: String?\n #if DEBUG\n @State private var currentErrorProvider: UsageProvider = .codex\n @State private var simulatedErrorText: String = \"\"\"\n Simulated error for testing layout.\n Second line.\n Third line.\n Fourth line.\n \"\"\"\n #endif\n\n var body: some View {\n ScrollView(.vertical, showsIndicators: true) {\n VStack(alignment: .leading, spacing: 20) {\n SettingsSection(title: \"Logging\") {\n PreferenceToggleRow(\n title: \"Enable file logging\",\n subtitle: \"Write logs to \\(self.fileLogPath) for debugging.\",\n binding: self.$debugFileLoggingEnabled)\n .onChange(of: self.debugFileLoggingEnabled) { _, newValue in\n if self.settings.debugFileLoggingEnabled != newValue {\n self.settings.debugFileLoggingEnabled = newValue\n }\n }\n\n HStack(alignment: .center, spacing: 12) {\n VStack(alignment: .leading, spacing: 4) {\n Text(\"Verbosity\")\n .font(.body)\n Text(\"Controls how much detail is logged.\")\n .font(.footnote)\n .foregroundStyle(.tertiary)\n }\n Spacer()\n Picker(\"Verbosity\", selection: self.$settings.debugLogLevel) {\n ForEach(CodexBarLog.Level.allCases) { level in\n Text(level.displayName).tag(level)\n }\n }\n .labelsHidden()\n .pickerStyle(.menu)\n .frame(maxWidth: 160)\n }\n\n Button {\n NSWorkspace.shared.open(CodexBarLog.fileLogURL)\n } label: {\n Label(\"Open log file\", systemImage: \"doc.text.magnifyingglass\")\n }\n .controlSize(.small)\n }\n\n SettingsSection {\n PreferenceToggleRow(\n title: \"Force animation on next refresh\",\n subtitle: \"Temporarily shows the loading animation after the next refresh.\",\n binding: self.$store.debugForceAnimation)\n }\n\n SettingsSection(\n title: \"Loading animations\",\n caption: \"Pick a pattern and replay it in the menu bar. \\\"Random\\\" keeps the existing behavior.\")\n {\n Picker(\"Animation pattern\", selection: self.animationPatternBinding) {\n Text(\"Random (default)\").tag(nil as LoadingPattern?)\n ForEach(LoadingPattern.allCases) { pattern in\n Text(pattern.displayName).tag(Optional(pattern))\n }\n }\n .pickerStyle(.radioGroup)\n\n Button(\"Replay selected animation\") {\n self.replaySelectedAnimation()\n }\n .keyboardShortcut(.defaultAction)\n\n Button {\n NotificationCenter.default.post(name: .codexbarDebugBlinkNow, object: nil)\n } label: {\n Label(\"Blink now\", systemImage: \"eyes\")\n }\n .controlSize(.small)\n }\n\n SettingsSection(\n title: \"Probe logs\",\n caption: \"Fetch the latest probe output for debugging; Copy keeps the full text.\")\n {\n Picker(\"Provider\", selection: self.$currentLogProvider) {\n Text(\"Codex\").tag(UsageProvider.codex)\n Text(\"Claude\").tag(UsageProvider.claude)\n Text(\"Cursor\").tag(UsageProvider.cursor)\n Text(\"Augment\").tag(UsageProvider.augment)\n Text(\"Amp\").tag(UsageProvider.amp)\n Text(\"Ollama\").tag(UsageProvider.ollama)\n }\n .pickerStyle(.segmented)\n .frame(width: 460)\n\n HStack(spacing: 12) {\n Button { self.loadLog(self.currentLogProvider) } label: {\n Label(\"Fetch log\", systemImage: \"arrow.clockwise\")\n }\n .disabled(self.isLoadingLog)\n\n Button { self.copyToPasteboard(self.logText) } label: {\n Label(\"Copy\", systemImage: \"doc.on.doc\")\n }\n .disabled(self.logText.isEmpty)\n\n Button { self.saveLog(self.currentLogProvider) } label: {\n Label(\"Save to file\", systemImage: \"externaldrive.badge.plus\")\n }\n .disabled(self.isLoadingLog && self.logText.isEmpty)\n\n if self.currentLogProvider == .claude {\n Button { self.loadClaudeDump() } label: {\n Label(\"Load parse dump\", systemImage: \"doc.text.magnifyingglass\")\n }\n .disabled(self.isLoadingLog)\n }\n }\n\n Button {\n self.settings.rerunProviderDetection()\n self.loadLog(self.currentLogProvider)\n } label: {\n Label(\"Re-run provider autodetect\", systemImage: \"dot.radiowaves.left.and.right\")\n }\n .controlSize(.small)\n\n ZStack(alignment: .topLeading) {\n ScrollView {\n Text(self.displayedLog)\n .font(.system(.footnote, design: .monospaced))\n .textSelection(.enabled)\n .frame(maxWidth: .infinity, alignment: .leading)\n .padding(8)\n }\n .frame(minHeight: 160, maxHeight: 220)\n .background(Color(NSColor.textBackgroundColor))\n .cornerRadius(6)\n\n if self.isLoadingLog {\n ProgressView()\n .progressViewStyle(.circular)\n .frame(maxWidth: .infinity, maxHeight: .infinity, alignment: .center)\n .padding()\n }\n }\n }\n\n SettingsSection(\n title: \"Fetch strategy attempts\",\n caption: \"Last fetch pipeline decisions and errors for a provider.\")\n {\n Picker(\"Provider\", selection: self.$currentFetchProvider) {\n ForEach(UsageProvider.allCases, id: \\.self) { provider in\n Text(provider.rawValue.capitalized).tag(provider)\n }\n }\n .pickerStyle(.menu)\n .frame(width: 240)\n\n ScrollView {\n Text(self.fetchAttemptsText(for: self.currentFetchProvider))\n .font(.system(.footnote, design: .monospaced))\n .textSelection(.enabled)\n .frame(maxWidth: .infinity, alignment: .leading)\n .padding(8)\n }\n .frame(minHeight: 120, maxHeight: 220)\n .background(Color(NSColor.textBackgroundColor))\n .cornerRadius(6)\n }\n\n if !self.settings.debugDisableKeychainAccess {\n SettingsSection(\n title: \"OpenAI cookies\",\n caption: \"Cookie import + WebKit scrape logs from the last OpenAI cookies attempt.\")\n {\n HStack(spacing: 12) {\n Button {\n self.copyToPasteboard(self.store.openAIDashboardCookieImportDebugLog ?? \"\")\n } label: {\n Label(\"Copy\", systemImage: \"doc.on.doc\")\n }\n .disabled((self.store.openAIDashboardCookieImportDebugLog ?? \"\").isEmpty)\n }\n\n ScrollView {\n Text(\n self.store.openAIDashboardCookieImportDebugLog?.isEmpty == false\n ? (self.store.openAIDashboardCookieImportDebugLog ?? \"\")\n : \"No log yet. Update OpenAI cookies in Providers → Codex to run an import.\")\n .font(.system(.footnote, design: .monospaced))\n .textSelection(.enabled)\n .frame(maxWidth: .infinity, alignment: .leading)\n .padding(8)\n }\n .frame(minHeight: 120, maxHeight: 180)\n .background(Color(NSColor.textBackgroundColor))\n .cornerRadius(6)\n }\n }\n\n SettingsSection(\n title: \"Caches\",\n caption: \"Clear cached cost scan results.\")\n {\n let isTokenRefreshActive = self.store.isTokenRefreshInFlight(for: .codex)\n || self.store.isTokenRefreshInFlight(for: .claude)\n\n HStack(spacing: 12) {\n Button {\n Task { await self.clearCostCache() }\n } label: {\n Label(\"Clear cost cache\", systemImage: \"trash\")\n }\n .disabled(self.isClearingCostCache || isTokenRefreshActive)\n\n if let status = self.costCacheStatus {\n Text(status)\n .font(.footnote)\n .foregroundStyle(.tertiary)\n }\n }\n }\n\n SettingsSection(\n title: \"Notifications\",\n caption: \"Trigger test notifications for the 5-hour session window (depleted/restored).\")\n {\n Picker(\"Provider\", selection: self.$currentLogProvider) {\n Text(\"Codex\").tag(UsageProvider.codex)\n Text(\"Claude\").tag(UsageProvider.claude)\n }\n .pickerStyle(.segmented)\n .frame(width: 240)\n\n HStack(spacing: 12) {\n Button {\n self.postSessionNotification(.depleted, provider: self.currentLogProvider)\n } label: {\n Label(\"Post depleted\", systemImage: \"bell.badge\")\n }\n .controlSize(.small)\n\n Button {\n self.postSessionNotification(.restored, provider: self.currentLogProvider)\n } label: {\n Label(\"Post restored\", systemImage: \"bell\")\n }\n .controlSize(.small)\n }\n }\n\n SettingsSection(\n title: \"CLI sessions\",\n caption: \"Keep Codex/Claude CLI sessions alive after a probe. Default exits once data is captured.\")\n {\n PreferenceToggleRow(\n title: \"Keep CLI sessions alive\",\n subtitle: \"Skip teardown between probes (debug-only).\",\n binding: self.$settings.debugKeepCLISessionsAlive)\n\n Button {\n Task {\n await CLIProbeSessionResetter.resetAll()\n }\n } label: {\n Label(\"Reset CLI sessions\", systemImage: \"arrow.counterclockwise\")\n }\n .controlSize(.small)\n }\n\n #if DEBUG\n SettingsSection(\n title: \"Error simulation\",\n caption: \"Inject a fake error message into the menu card for layout testing.\")\n {\n Picker(\"Provider\", selection: self.$currentErrorProvider) {\n Text(\"Codex\").tag(UsageProvider.codex)\n Text(\"Claude\").tag(UsageProvider.claude)\n Text(\"Gemini\").tag(UsageProvider.gemini)\n Text(\"Antigravity\").tag(UsageProvider.antigravity)\n Text(\"Augment\").tag(UsageProvider.augment)\n Text(\"Amp\").tag(UsageProvider.amp)\n Text(\"Ollama\").tag(UsageProvider.ollama)\n }\n .pickerStyle(.segmented)\n .frame(width: 360)\n\n TextField(\"Simulated error text\", text: self.$simulatedErrorText, axis: .vertical)\n .lineLimit(4)\n\n HStack(spacing: 12) {\n Button {\n self.store._setErrorForTesting(\n self.simulatedErrorText,\n provider: self.currentErrorProvider)\n } label: {\n Label(\"Set menu error\", systemImage: \"exclamationmark.triangle\")\n }\n .controlSize(.small)\n\n Button {\n self.store._setErrorForTesting(nil, provider: self.currentErrorProvider)\n } label: {\n Label(\"Clear menu error\", systemImage: \"xmark.circle\")\n }\n .controlSize(.small)\n }\n\n let supportsTokenError = self.currentErrorProvider == .codex || self.currentErrorProvider == .claude\n HStack(spacing: 12) {\n Button {\n self.store._setTokenErrorForTesting(\n self.simulatedErrorText,\n provider: self.currentErrorProvider)\n } label: {\n Label(\"Set cost error\", systemImage: \"banknote\")\n }\n .controlSize(.small)\n .disabled(!supportsTokenError)\n\n Button {\n self.store._setTokenErrorForTesting(nil, provider: self.currentErrorProvider)\n } label: {\n Label(\"Clear cost error\", systemImage: \"xmark.circle\")\n }\n .controlSize(.small)\n .disabled(!supportsTokenError)\n }\n }\n #endif\n\n SettingsSection(\n title: \"CLI paths\",\n caption: \"Resolved Codex binary and PATH layers; startup login PATH capture (short timeout).\")\n {\n self.binaryRow(title: \"Codex binary\", value: self.store.pathDebugInfo.codexBinary)\n self.binaryRow(title: \"Claude binary\", value: self.store.pathDebugInfo.claudeBinary)\n\n VStack(alignment: .leading, spacing: 6) {\n Text(\"Effective PATH\")\n .font(.callout.weight(.semibold))\n ScrollView {\n Text(\n self.store.pathDebugInfo.effectivePATH.isEmpty\n ? \"Unavailable\"\n : self.store.pathDebugInfo.effectivePATH)\n .font(.system(.footnote, design: .monospaced))\n .textSelection(.enabled)\n .frame(maxWidth: .infinity, alignment: .leading)\n .padding(6)\n }\n .frame(minHeight: 60, maxHeight: 110)\n .background(Color(NSColor.textBackgroundColor))\n .cornerRadius(6)\n }\n\n if let loginPATH = self.store.pathDebugInfo.loginShellPATH {\n VStack(alignment: .leading, spacing: 6) {\n Text(\"Login shell PATH (startup capture)\")\n .font(.callout.weight(.semibold))\n ScrollView {\n Text(loginPATH)\n .font(.system(.footnote, design: .monospaced))\n .textSelection(.enabled)\n .frame(maxWidth: .infinity, alignment: .leading)\n .padding(6)\n }\n .frame(minHeight: 60, maxHeight: 110)\n .background(Color(NSColor.textBackgroundColor))\n .cornerRadius(6)\n }\n }\n }\n }\n .frame(maxWidth: .infinity, alignment: .leading)\n .padding(.horizontal, 20)\n .padding(.vertical, 12)\n }\n }\n\n private var fileLogPath: String {\n CodexBarLog.fileLogURL.path\n }\n\n private var animationPatternBinding: Binding {\n Binding(\n get: { self.settings.debugLoadingPattern },\n set: { self.settings.debugLoadingPattern = $0 })\n }\n\n private func replaySelectedAnimation() {\n var userInfo: [AnyHashable: Any] = [:]\n if let pattern = self.settings.debugLoadingPattern {\n userInfo[\"pattern\"] = pattern.rawValue\n }\n NotificationCenter.default.post(\n name: .codexbarDebugReplayAllAnimations,\n object: nil,\n userInfo: userInfo.isEmpty ? nil : userInfo)\n self.store.replayLoadingAnimation(duration: 4)\n }\n\n private var displayedLog: String {\n if self.logText.isEmpty {\n return self.isLoadingLog ? \"Loading…\" : \"No log yet. Fetch to load.\"\n }\n return self.logText\n }\n\n private func loadLog(_ provider: UsageProvider) {\n self.isLoadingLog = true\n Task {\n let text = await ProviderInteractionContext.$current.withValue(.userInitiated) {\n await ProviderRefreshContext.$current.withValue(.regular) {\n await self.store.debugLog(for: provider)\n }\n }\n await MainActor.run {\n self.logText = text\n self.isLoadingLog = false\n }\n }\n }\n\n private func saveLog(_ provider: UsageProvider) {\n Task {\n if self.logText.isEmpty {\n self.isLoadingLog = true\n let text = await ProviderInteractionContext.$current.withValue(.userInitiated) {\n await ProviderRefreshContext.$current.withValue(.regular) {\n await self.store.debugLog(for: provider)\n }\n }\n await MainActor.run { self.logText = text }\n self.isLoadingLog = false\n }\n _ = await ProviderInteractionContext.$current.withValue(.userInitiated) {\n await ProviderRefreshContext.$current.withValue(.regular) {\n await self.store.dumpLog(toFileFor: provider)\n }\n }\n }\n }\n\n private func copyToPasteboard(_ text: String) {\n let pb = NSPasteboard.general\n pb.clearContents()\n pb.setString(text, forType: .string)\n }\n\n private func binaryRow(title: String, value: String?) -> some View {\n VStack(alignment: .leading, spacing: 6) {\n Text(title)\n .font(.callout.weight(.semibold))\n Text(value ?? \"Not found\")\n .font(.system(.footnote, design: .monospaced))\n .foregroundStyle(value == nil ? .secondary : .primary)\n }\n }\n\n private func loadClaudeDump() {\n self.isLoadingLog = true\n Task {\n let text = await self.store.debugClaudeDump()\n await MainActor.run {\n self.logText = text\n self.isLoadingLog = false\n }\n }\n }\n\n private func postSessionNotification(_ transition: SessionQuotaTransition, provider: UsageProvider) {\n SessionQuotaNotifier().post(transition: transition, provider: provider, badge: 1)\n }\n\n private func clearCostCache() async {\n guard !self.isClearingCostCache else { return }\n self.isClearingCostCache = true\n self.costCacheStatus = nil\n defer { self.isClearingCostCache = false }\n\n if let error = await self.store.clearCostUsageCache() {\n self.costCacheStatus = \"Failed: \\(error)\"\n return\n }\n\n self.costCacheStatus = \"Cleared.\"\n }\n\n private func fetchAttemptsText(for provider: UsageProvider) -> String {\n let attempts = self.store.fetchAttempts(for: provider)\n guard !attempts.isEmpty else { return \"No fetch attempts yet.\" }\n return attempts.map { attempt in\n let kind = Self.fetchKindLabel(attempt.kind)\n var line = \"\\(attempt.strategyID) (\\(kind))\"\n line += attempt.wasAvailable ? \" available\" : \" unavailable\"\n if let error = attempt.errorDescription, !error.isEmpty {\n line += \" error=\\(error)\"\n }\n return line\n }.joined(separator: \"\\n\")\n }\n\n private static func fetchKindLabel(_ kind: ProviderFetchKind) -> String {\n switch kind {\n case .cli: \"cli\"\n case .web: \"web\"\n case .oauth: \"oauth\"\n case .apiToken: \"api\"\n case .localProbe: \"local\"\n case .webDashboard: \"web\"\n }\n }\n}\n" }, { "path": "Sources/CodexBar/PreferencesDisplayPane.swift", "content": "import CodexBarCore\nimport SwiftUI\n\n@MainActor\nstruct DisplayPane: View {\n private static let maxOverviewProviders = SettingsStore.mergedOverviewProviderLimit\n\n @State private var isOverviewProviderPopoverPresented = false\n @Bindable var settings: SettingsStore\n @Bindable var store: UsageStore\n\n var body: some View {\n ScrollView(.vertical, showsIndicators: true) {\n VStack(alignment: .leading, spacing: 16) {\n SettingsSection(contentSpacing: 12) {\n Text(\"Menu bar\")\n .font(.caption)\n .foregroundStyle(.secondary)\n .textCase(.uppercase)\n PreferenceToggleRow(\n title: \"Merge Icons\",\n subtitle: \"Use a single menu bar icon with a provider switcher.\",\n binding: self.$settings.mergeIcons)\n PreferenceToggleRow(\n title: \"Switcher shows icons\",\n subtitle: \"Show provider icons in the switcher (otherwise show a weekly progress line).\",\n binding: self.$settings.switcherShowsIcons)\n .disabled(!self.settings.mergeIcons)\n .opacity(self.settings.mergeIcons ? 1 : 0.5)\n PreferenceToggleRow(\n title: \"Show most-used provider\",\n subtitle: \"Menu bar auto-shows the provider closest to its rate limit.\",\n binding: self.$settings.menuBarShowsHighestUsage)\n .disabled(!self.settings.mergeIcons)\n .opacity(self.settings.mergeIcons ? 1 : 0.5)\n PreferenceToggleRow(\n title: \"Menu bar shows percent\",\n subtitle: \"Replace critter bars with provider branding icons and a percentage.\",\n binding: self.$settings.menuBarShowsBrandIconWithPercent)\n HStack(alignment: .top, spacing: 12) {\n VStack(alignment: .leading, spacing: 4) {\n Text(\"Display mode\")\n .font(.body)\n Text(\"Choose what to show in the menu bar (Pace shows usage vs. expected).\")\n .font(.footnote)\n .foregroundStyle(.tertiary)\n }\n Spacer()\n Picker(\"Display mode\", selection: self.$settings.menuBarDisplayMode) {\n ForEach(MenuBarDisplayMode.allCases) { mode in\n Text(mode.label).tag(mode)\n }\n }\n .labelsHidden()\n .pickerStyle(.menu)\n .frame(maxWidth: 200)\n }\n .disabled(!self.settings.menuBarShowsBrandIconWithPercent)\n .opacity(self.settings.menuBarShowsBrandIconWithPercent ? 1 : 0.5)\n }\n\n Divider()\n\n SettingsSection(contentSpacing: 12) {\n Text(\"Menu content\")\n .font(.caption)\n .foregroundStyle(.secondary)\n .textCase(.uppercase)\n PreferenceToggleRow(\n title: \"Show usage as used\",\n subtitle: \"Progress bars fill as you consume quota (instead of showing remaining).\",\n binding: self.$settings.usageBarsShowUsed)\n PreferenceToggleRow(\n title: \"Show reset time as clock\",\n subtitle: \"Display reset times as absolute clock values instead of countdowns.\",\n binding: self.$settings.resetTimesShowAbsolute)\n PreferenceToggleRow(\n title: \"Show credits + extra usage\",\n subtitle: \"Show Codex Credits and Claude Extra usage sections in the menu.\",\n binding: self.$settings.showOptionalCreditsAndExtraUsage)\n PreferenceToggleRow(\n title: \"Show all token accounts\",\n subtitle: \"Stack token accounts in the menu (otherwise show an account switcher bar).\",\n binding: self.$settings.showAllTokenAccountsInMenu)\n self.overviewProviderSelector\n }\n }\n .frame(maxWidth: .infinity, alignment: .leading)\n .padding(.horizontal, 20)\n .padding(.vertical, 12)\n .onAppear {\n self.reconcileOverviewSelection()\n }\n .onChange(of: self.settings.mergeIcons) { _, isEnabled in\n guard isEnabled else {\n self.isOverviewProviderPopoverPresented = false\n return\n }\n self.reconcileOverviewSelection()\n }\n .onChange(of: self.activeProvidersInOrder) { _, _ in\n if self.activeProvidersInOrder.isEmpty {\n self.isOverviewProviderPopoverPresented = false\n }\n self.reconcileOverviewSelection()\n }\n }\n }\n\n private var overviewProviderSelector: some View {\n VStack(alignment: .leading, spacing: 6) {\n HStack(alignment: .center, spacing: 12) {\n Text(\"Overview tab providers\")\n .font(.body)\n Spacer(minLength: 0)\n if self.showsOverviewConfigureButton {\n Button(\"Configure…\") {\n self.isOverviewProviderPopoverPresented = true\n }\n .offset(y: 1)\n .popover(isPresented: self.$isOverviewProviderPopoverPresented, arrowEdge: .bottom) {\n self.overviewProviderPopover\n }\n }\n }\n\n if !self.settings.mergeIcons {\n Text(\"Enable Merge Icons to configure Overview tab providers.\")\n .font(.footnote)\n .foregroundStyle(.tertiary)\n } else if self.activeProvidersInOrder.isEmpty {\n Text(\"No enabled providers available for Overview.\")\n .font(.footnote)\n .foregroundStyle(.tertiary)\n } else {\n Text(self.overviewProviderSelectionSummary)\n .font(.footnote)\n .foregroundStyle(.tertiary)\n .lineLimit(2)\n .truncationMode(.tail)\n }\n }\n }\n\n private var overviewProviderPopover: some View {\n VStack(alignment: .leading, spacing: 10) {\n Text(\"Choose up to \\(Self.maxOverviewProviders) providers\")\n .font(.headline)\n Text(\"Overview rows always follow provider order.\")\n .font(.footnote)\n .foregroundStyle(.tertiary)\n\n ScrollView(.vertical, showsIndicators: true) {\n VStack(alignment: .leading, spacing: 6) {\n ForEach(self.activeProvidersInOrder, id: \\.self) { provider in\n Toggle(\n isOn: Binding(\n get: { self.overviewSelectedProviders.contains(provider) },\n set: { shouldSelect in\n self.setOverviewProviderSelection(provider: provider, isSelected: shouldSelect)\n })) {\n Text(self.providerDisplayName(provider))\n .font(.body)\n }\n .toggleStyle(.checkbox)\n .disabled(\n !self.overviewSelectedProviders.contains(provider) &&\n self.overviewSelectedProviders.count >= Self.maxOverviewProviders)\n }\n }\n }\n .frame(maxHeight: 220)\n }\n .padding(12)\n .frame(width: 280)\n }\n\n private var activeProvidersInOrder: [UsageProvider] {\n self.store.enabledProviders()\n }\n\n private var overviewSelectedProviders: [UsageProvider] {\n self.settings.resolvedMergedOverviewProviders(\n activeProviders: self.activeProvidersInOrder,\n maxVisibleProviders: Self.maxOverviewProviders)\n }\n\n private var showsOverviewConfigureButton: Bool {\n self.settings.mergeIcons && !self.activeProvidersInOrder.isEmpty\n }\n\n private var overviewProviderSelectionSummary: String {\n let selectedNames = self.overviewSelectedProviders.map(self.providerDisplayName)\n guard !selectedNames.isEmpty else { return \"No providers selected\" }\n return selectedNames.joined(separator: \", \")\n }\n\n private func providerDisplayName(_ provider: UsageProvider) -> String {\n ProviderDescriptorRegistry.descriptor(for: provider).metadata.displayName\n }\n\n private func setOverviewProviderSelection(provider: UsageProvider, isSelected: Bool) {\n _ = self.settings.setMergedOverviewProviderSelection(\n provider: provider,\n isSelected: isSelected,\n activeProviders: self.activeProvidersInOrder,\n maxVisibleProviders: Self.maxOverviewProviders)\n }\n\n private func reconcileOverviewSelection() {\n _ = self.settings.reconcileMergedOverviewSelectedProviders(\n activeProviders: self.activeProvidersInOrder,\n maxVisibleProviders: Self.maxOverviewProviders)\n }\n}\n" }, { "path": "Sources/CodexBar/PreferencesGeneralPane.swift", "content": "import AppKit\nimport CodexBarCore\nimport SwiftUI\n\n@MainActor\nstruct GeneralPane: View {\n @Bindable var settings: SettingsStore\n @Bindable var store: UsageStore\n\n var body: some View {\n ScrollView(.vertical, showsIndicators: true) {\n VStack(alignment: .leading, spacing: 16) {\n SettingsSection(contentSpacing: 12) {\n Text(\"System\")\n .font(.caption)\n .foregroundStyle(.secondary)\n .textCase(.uppercase)\n PreferenceToggleRow(\n title: \"Start at Login\",\n subtitle: \"Automatically opens CodexBar when you start your Mac.\",\n binding: self.$settings.launchAtLogin)\n }\n\n Divider()\n\n SettingsSection(contentSpacing: 12) {\n Text(\"Usage\")\n .font(.caption)\n .foregroundStyle(.secondary)\n .textCase(.uppercase)\n\n VStack(alignment: .leading, spacing: 10) {\n VStack(alignment: .leading, spacing: 4) {\n Toggle(isOn: self.$settings.costUsageEnabled) {\n Text(\"Show cost summary\")\n .font(.body)\n }\n .toggleStyle(.checkbox)\n\n Text(\"Reads local usage logs. Shows today + last 30 days cost in the menu.\")\n .font(.footnote)\n .foregroundStyle(.tertiary)\n .fixedSize(horizontal: false, vertical: true)\n\n if self.settings.costUsageEnabled {\n Text(\"Auto-refresh: hourly · Timeout: 10m\")\n .font(.footnote)\n .foregroundStyle(.tertiary)\n\n self.costStatusLine(provider: .claude)\n self.costStatusLine(provider: .codex)\n }\n }\n }\n }\n\n Divider()\n\n SettingsSection(contentSpacing: 12) {\n Text(\"Automation\")\n .font(.caption)\n .foregroundStyle(.secondary)\n .textCase(.uppercase)\n VStack(alignment: .leading, spacing: 6) {\n HStack(alignment: .top, spacing: 12) {\n VStack(alignment: .leading, spacing: 4) {\n Text(\"Refresh cadence\")\n .font(.body)\n Text(\"How often CodexBar polls providers in the background.\")\n .font(.footnote)\n .foregroundStyle(.tertiary)\n }\n Spacer()\n Picker(\"Refresh cadence\", selection: self.$settings.refreshFrequency) {\n ForEach(RefreshFrequency.allCases) { option in\n Text(option.label).tag(option)\n }\n }\n .labelsHidden()\n .pickerStyle(.menu)\n .frame(maxWidth: 200)\n }\n if self.settings.refreshFrequency == .manual {\n Text(\"Auto-refresh is off; use the menu's Refresh command.\")\n .font(.footnote)\n .foregroundStyle(.secondary)\n }\n }\n PreferenceToggleRow(\n title: \"Check provider status\",\n subtitle: \"Polls OpenAI/Claude status pages and Google Workspace for \" +\n \"Gemini/Antigravity, surfacing incidents in the icon and menu.\",\n binding: self.$settings.statusChecksEnabled)\n PreferenceToggleRow(\n title: \"Session quota notifications\",\n subtitle: \"Notifies when the 5-hour session quota hits 0% and when it becomes \" +\n \"available again.\",\n binding: self.$settings.sessionQuotaNotificationsEnabled)\n }\n\n Divider()\n\n SettingsSection(contentSpacing: 12) {\n HStack {\n Spacer()\n Button(\"Quit CodexBar\") { NSApp.terminate(nil) }\n .buttonStyle(.borderedProminent)\n .controlSize(.large)\n }\n }\n }\n .frame(maxWidth: .infinity, alignment: .leading)\n .padding(.horizontal, 20)\n .padding(.vertical, 12)\n }\n }\n\n private func costStatusLine(provider: UsageProvider) -> some View {\n let name = ProviderDescriptorRegistry.descriptor(for: provider).metadata.displayName\n\n guard provider == .claude || provider == .codex else {\n return Text(\"\\(name): unsupported\")\n .font(.footnote)\n .foregroundStyle(.tertiary)\n }\n\n if self.store.isTokenRefreshInFlight(for: provider) {\n let elapsed: String = {\n guard let startedAt = self.store.tokenLastAttemptAt(for: provider) else { return \"\" }\n let seconds = max(0, Date().timeIntervalSince(startedAt))\n let formatter = DateComponentsFormatter()\n formatter.allowedUnits = seconds < 60 ? [.second] : [.minute, .second]\n formatter.unitsStyle = .abbreviated\n return formatter.string(from: seconds).map { \" (\\($0))\" } ?? \"\"\n }()\n return Text(\"\\(name): fetching…\\(elapsed)\")\n .font(.footnote)\n .foregroundStyle(.tertiary)\n }\n if let snapshot = self.store.tokenSnapshot(for: provider) {\n let updated = UsageFormatter.updatedString(from: snapshot.updatedAt)\n let cost = snapshot.last30DaysCostUSD.map { UsageFormatter.usdString($0) } ?? \"—\"\n return Text(\"\\(name): \\(updated) · 30d \\(cost)\")\n .font(.footnote)\n .foregroundStyle(.tertiary)\n }\n if let error = self.store.tokenError(for: provider), !error.isEmpty {\n let truncated = UsageFormatter.truncatedSingleLine(error, max: 120)\n return Text(\"\\(name): \\(truncated)\")\n .font(.footnote)\n .foregroundStyle(.tertiary)\n }\n if let lastAttempt = self.store.tokenLastAttemptAt(for: provider) {\n let rel = RelativeDateTimeFormatter()\n rel.unitsStyle = .abbreviated\n let when = rel.localizedString(for: lastAttempt, relativeTo: Date())\n return Text(\"\\(name): last attempt \\(when)\")\n .font(.footnote)\n .foregroundStyle(.tertiary)\n }\n return Text(\"\\(name): no data yet\")\n .font(.footnote)\n .foregroundStyle(.tertiary)\n }\n}\n" }, { "path": "Sources/CodexBar/PreferencesProviderDetailView.swift", "content": "import CodexBarCore\nimport SwiftUI\n\n@MainActor\nstruct ProviderDetailView: View {\n let provider: UsageProvider\n @Bindable var store: UsageStore\n @Binding var isEnabled: Bool\n let subtitle: String\n let model: UsageMenuCardView.Model\n let settingsPickers: [ProviderSettingsPickerDescriptor]\n let settingsToggles: [ProviderSettingsToggleDescriptor]\n let settingsFields: [ProviderSettingsFieldDescriptor]\n let settingsTokenAccounts: ProviderSettingsTokenAccountsDescriptor?\n let errorDisplay: ProviderErrorDisplay?\n @Binding var isErrorExpanded: Bool\n let onCopyError: (String) -> Void\n let onRefresh: () -> Void\n\n static func metricTitle(provider: UsageProvider, metric: UsageMenuCardView.Model.Metric) -> String {\n UsageMenuCardView.popupMetricTitle(provider: provider, metric: metric)\n }\n\n static func planRow(provider: UsageProvider, planText: String?) -> (label: String, value: String)? {\n guard let rawPlan = planText?.trimmingCharacters(in: .whitespacesAndNewlines),\n !rawPlan.isEmpty\n else {\n return nil\n }\n guard provider == .openrouter else {\n return (label: \"Plan\", value: rawPlan)\n }\n\n let prefix = \"Balance:\"\n if rawPlan.hasPrefix(prefix) {\n let valueStart = rawPlan.index(rawPlan.startIndex, offsetBy: prefix.count)\n let trimmedValue = rawPlan[valueStart...].trimmingCharacters(in: .whitespacesAndNewlines)\n if !trimmedValue.isEmpty {\n return (label: \"Balance\", value: trimmedValue)\n }\n }\n return (label: \"Balance\", value: rawPlan)\n }\n\n var body: some View {\n ScrollView {\n VStack(alignment: .leading, spacing: 16) {\n let labelWidth = self.detailLabelWidth\n ProviderDetailHeaderView(\n provider: self.provider,\n store: self.store,\n isEnabled: self.$isEnabled,\n subtitle: self.subtitle,\n model: self.model,\n labelWidth: labelWidth,\n onRefresh: self.onRefresh)\n\n ProviderMetricsInlineView(\n provider: self.provider,\n model: self.model,\n isEnabled: self.isEnabled,\n labelWidth: labelWidth)\n\n if let errorDisplay {\n ProviderErrorView(\n title: \"Last \\(self.store.metadata(for: self.provider).displayName) fetch failed:\",\n display: errorDisplay,\n isExpanded: self.$isErrorExpanded,\n onCopy: { self.onCopyError(errorDisplay.full) })\n }\n\n if self.hasSettings {\n ProviderSettingsSection(title: \"Settings\") {\n ForEach(self.settingsPickers) { picker in\n ProviderSettingsPickerRowView(picker: picker)\n }\n if let tokenAccounts = self.settingsTokenAccounts,\n tokenAccounts.isVisible?() ?? true\n {\n ProviderSettingsTokenAccountsRowView(descriptor: tokenAccounts)\n }\n ForEach(self.settingsFields) { field in\n ProviderSettingsFieldRowView(field: field)\n }\n }\n }\n\n if !self.settingsToggles.isEmpty {\n ProviderSettingsSection(title: \"Options\") {\n ForEach(self.settingsToggles) { toggle in\n ProviderSettingsToggleRowView(toggle: toggle)\n }\n }\n }\n }\n .frame(maxWidth: ProviderSettingsMetrics.detailMaxWidth, alignment: .leading)\n .padding(.vertical, 12)\n .padding(.horizontal, 8)\n }\n .frame(maxWidth: .infinity, alignment: .leading)\n }\n\n private var hasSettings: Bool {\n !self.settingsPickers.isEmpty ||\n !self.settingsFields.isEmpty ||\n self.settingsTokenAccounts != nil\n }\n\n private var detailLabelWidth: CGFloat {\n var infoLabels = [\"State\", \"Source\", \"Version\", \"Updated\"]\n if self.store.status(for: self.provider) != nil {\n infoLabels.append(\"Status\")\n }\n if !self.model.email.isEmpty {\n infoLabels.append(\"Account\")\n }\n if let planRow = Self.planRow(provider: self.provider, planText: self.model.planText) {\n infoLabels.append(planRow.label)\n }\n\n var metricLabels = self.model.metrics.map { metric in\n Self.metricTitle(provider: self.provider, metric: metric)\n }\n if self.model.creditsText != nil {\n metricLabels.append(\"Credits\")\n }\n if let providerCost = self.model.providerCost {\n metricLabels.append(providerCost.title)\n }\n if self.model.tokenUsage != nil {\n metricLabels.append(\"Cost\")\n }\n\n let infoWidth = ProviderSettingsMetrics.labelWidth(\n for: infoLabels,\n font: ProviderSettingsMetrics.infoLabelFont())\n let metricWidth = ProviderSettingsMetrics.labelWidth(\n for: metricLabels,\n font: ProviderSettingsMetrics.metricLabelFont())\n return max(infoWidth, metricWidth)\n }\n}\n\n@MainActor\nprivate struct ProviderDetailHeaderView: View {\n let provider: UsageProvider\n @Bindable var store: UsageStore\n @Binding var isEnabled: Bool\n let subtitle: String\n let model: UsageMenuCardView.Model\n let labelWidth: CGFloat\n let onRefresh: () -> Void\n\n var body: some View {\n VStack(alignment: .leading, spacing: 12) {\n HStack(alignment: .center, spacing: 12) {\n ProviderDetailBrandIcon(provider: self.provider)\n\n VStack(alignment: .leading, spacing: 4) {\n Text(self.store.metadata(for: self.provider).displayName)\n .font(.title3.weight(.semibold))\n\n Text(self.detailSubtitle)\n .font(.footnote)\n .foregroundStyle(.secondary)\n }\n\n Spacer(minLength: 12)\n\n Button {\n self.onRefresh()\n } label: {\n Image(systemName: \"arrow.clockwise\")\n }\n .buttonStyle(.bordered)\n .controlSize(.small)\n .help(\"Refresh\")\n\n Toggle(\"\", isOn: self.$isEnabled)\n .labelsHidden()\n .toggleStyle(.switch)\n .controlSize(.small)\n }\n\n ProviderDetailInfoGrid(\n provider: self.provider,\n store: self.store,\n isEnabled: self.isEnabled,\n model: self.model,\n labelWidth: self.labelWidth)\n }\n }\n\n private var detailSubtitle: String {\n let lines = self.subtitle.split(separator: \"\\n\", omittingEmptySubsequences: false)\n guard lines.count >= 2 else { return self.subtitle }\n let first = lines[0]\n let rest = lines.dropFirst().joined(separator: \"\\n\")\n let tail = rest.trimmingCharacters(in: .whitespacesAndNewlines)\n if tail.isEmpty { return String(first) }\n return \"\\(first) • \\(tail)\"\n }\n}\n\n@MainActor\nprivate struct ProviderDetailBrandIcon: View {\n let provider: UsageProvider\n\n var body: some View {\n if let brand = ProviderBrandIcon.image(for: self.provider) {\n Image(nsImage: brand)\n .resizable()\n .scaledToFit()\n .frame(width: 28, height: 28)\n .foregroundStyle(.secondary)\n .accessibilityHidden(true)\n } else {\n Image(systemName: \"circle.dotted\")\n .font(.system(size: 24, weight: .regular))\n .foregroundStyle(.secondary)\n .accessibilityHidden(true)\n }\n }\n}\n\n@MainActor\nprivate struct ProviderDetailInfoGrid: View {\n let provider: UsageProvider\n @Bindable var store: UsageStore\n let isEnabled: Bool\n let model: UsageMenuCardView.Model\n let labelWidth: CGFloat\n\n var body: some View {\n let status = self.store.status(for: self.provider)\n let source = self.store.sourceLabel(for: self.provider)\n let version = self.store.version(for: self.provider) ?? \"not detected\"\n let updated = self.updatedText\n let email = self.model.email\n let enabledText = self.isEnabled ? \"Enabled\" : \"Disabled\"\n\n Grid(alignment: .leading, horizontalSpacing: 12, verticalSpacing: 6) {\n ProviderDetailInfoRow(label: \"State\", value: enabledText, labelWidth: self.labelWidth)\n ProviderDetailInfoRow(label: \"Source\", value: source, labelWidth: self.labelWidth)\n ProviderDetailInfoRow(label: \"Version\", value: version, labelWidth: self.labelWidth)\n ProviderDetailInfoRow(label: \"Updated\", value: updated, labelWidth: self.labelWidth)\n\n if let status {\n ProviderDetailInfoRow(\n label: \"Status\",\n value: status.description ?? status.indicator.label,\n labelWidth: self.labelWidth)\n }\n\n if !email.isEmpty {\n ProviderDetailInfoRow(label: \"Account\", value: email, labelWidth: self.labelWidth)\n }\n\n if let planRow = ProviderDetailView.planRow(provider: self.provider, planText: self.model.planText) {\n ProviderDetailInfoRow(label: planRow.label, value: planRow.value, labelWidth: self.labelWidth)\n }\n }\n .font(.footnote)\n .foregroundStyle(.secondary)\n }\n\n private var updatedText: String {\n if let updated = self.store.snapshot(for: self.provider)?.updatedAt {\n return UsageFormatter.updatedString(from: updated)\n }\n if self.store.refreshingProviders.contains(self.provider) {\n return \"Refreshing\"\n }\n return \"Not fetched yet\"\n }\n}\n\nprivate struct ProviderDetailInfoRow: View {\n let label: String\n let value: String\n let labelWidth: CGFloat\n\n var body: some View {\n GridRow {\n Text(self.label)\n .frame(width: self.labelWidth, alignment: .leading)\n Text(self.value)\n .lineLimit(2)\n }\n }\n}\n\n@MainActor\nstruct ProviderMetricsInlineView: View {\n let provider: UsageProvider\n let model: UsageMenuCardView.Model\n let isEnabled: Bool\n let labelWidth: CGFloat\n\n var body: some View {\n let hasMetrics = !self.model.metrics.isEmpty\n let hasUsageNotes = !self.model.usageNotes.isEmpty\n let hasCredits = self.model.creditsText != nil\n let hasProviderCost = self.model.providerCost != nil\n let hasTokenUsage = self.model.tokenUsage != nil\n ProviderSettingsSection(\n title: \"Usage\",\n spacing: 8,\n verticalPadding: 6,\n horizontalPadding: 0)\n {\n if !hasMetrics, !hasUsageNotes, !hasProviderCost, !hasCredits, !hasTokenUsage {\n Text(self.placeholderText)\n .font(.footnote)\n .foregroundStyle(.secondary)\n } else {\n ForEach(self.model.metrics, id: \\.id) { metric in\n ProviderMetricInlineRow(\n metric: metric,\n title: ProviderDetailView.metricTitle(provider: self.provider, metric: metric),\n progressColor: self.model.progressColor,\n labelWidth: self.labelWidth)\n }\n\n if hasUsageNotes {\n ProviderUsageNotesInlineView(\n notes: self.model.usageNotes,\n labelWidth: self.labelWidth,\n alignsWithMetricContent: hasMetrics)\n }\n\n if let credits = self.model.creditsText {\n ProviderMetricInlineTextRow(\n title: \"Credits\",\n value: credits,\n labelWidth: self.labelWidth)\n }\n\n if let providerCost = self.model.providerCost {\n ProviderMetricInlineCostRow(\n section: providerCost,\n progressColor: self.model.progressColor,\n labelWidth: self.labelWidth)\n }\n\n if let tokenUsage = self.model.tokenUsage {\n ProviderMetricInlineTextRow(\n title: \"Cost\",\n value: tokenUsage.sessionLine,\n labelWidth: self.labelWidth)\n ProviderMetricInlineTextRow(\n title: \"\",\n value: tokenUsage.monthLine,\n labelWidth: self.labelWidth)\n }\n }\n }\n }\n\n private var placeholderText: String {\n if !self.isEnabled {\n return \"Disabled — no recent data\"\n }\n return self.model.placeholder ?? \"No usage yet\"\n }\n}\n\nprivate struct ProviderMetricInlineRow: View {\n let metric: UsageMenuCardView.Model.Metric\n let title: String\n let progressColor: Color\n let labelWidth: CGFloat\n\n var body: some View {\n HStack(alignment: .top, spacing: 10) {\n Text(self.title)\n .font(.subheadline.weight(.semibold))\n .lineLimit(1)\n .frame(width: self.labelWidth, alignment: .leading)\n\n VStack(alignment: .leading, spacing: 4) {\n UsageProgressBar(\n percent: self.metric.percent,\n tint: self.progressColor,\n accessibilityLabel: self.metric.percentStyle.accessibilityLabel,\n pacePercent: self.metric.pacePercent,\n paceOnTop: self.metric.paceOnTop)\n .frame(minWidth: ProviderSettingsMetrics.metricBarWidth, maxWidth: .infinity)\n\n HStack(alignment: .firstTextBaseline, spacing: 8) {\n Text(self.metric.percentLabel)\n .font(.footnote)\n .foregroundStyle(.secondary)\n .monospacedDigit()\n Spacer(minLength: 8)\n if let resetText = self.metric.resetText, !resetText.isEmpty {\n Text(resetText)\n .font(.footnote)\n .foregroundStyle(.secondary)\n }\n }\n\n let hasLeftDetail = self.metric.detailLeftText?.isEmpty == false\n let hasRightDetail = self.metric.detailRightText?.isEmpty == false\n if hasLeftDetail || hasRightDetail {\n HStack(alignment: .firstTextBaseline, spacing: 8) {\n if let leftDetail = self.metric.detailLeftText, !leftDetail.isEmpty {\n Text(leftDetail)\n .font(.footnote)\n .foregroundStyle(.secondary)\n }\n Spacer(minLength: 8)\n if let rightDetail = self.metric.detailRightText, !rightDetail.isEmpty {\n Text(rightDetail)\n .font(.footnote)\n .foregroundStyle(.secondary)\n }\n }\n }\n\n if let detail = self.detailText, !detail.isEmpty {\n Text(detail)\n .font(.footnote)\n .foregroundStyle(.tertiary)\n }\n }\n .frame(maxWidth: .infinity, alignment: .leading)\n }\n .padding(.vertical, 2)\n }\n\n private var detailText: String? {\n guard let detailText = self.metric.detailText, !detailText.isEmpty else { return nil }\n return detailText\n }\n}\n\nprivate struct ProviderUsageNotesInlineView: View {\n let notes: [String]\n let labelWidth: CGFloat\n let alignsWithMetricContent: Bool\n\n var body: some View {\n HStack(alignment: .top, spacing: 10) {\n if self.alignsWithMetricContent {\n Spacer()\n .frame(width: self.labelWidth)\n }\n VStack(alignment: .leading, spacing: 4) {\n ForEach(Array(self.notes.enumerated()), id: \\.offset) { _, note in\n Text(note)\n .font(.footnote)\n .foregroundStyle(.secondary)\n .lineLimit(2)\n .fixedSize(horizontal: false, vertical: true)\n }\n }\n .frame(maxWidth: .infinity, alignment: .leading)\n }\n .padding(.vertical, 2)\n }\n}\n\nprivate struct ProviderMetricInlineTextRow: View {\n let title: String\n let value: String\n let labelWidth: CGFloat\n\n var body: some View {\n HStack(alignment: .firstTextBaseline, spacing: 12) {\n Text(self.title)\n .font(.subheadline.weight(.semibold))\n .frame(width: self.labelWidth, alignment: .leading)\n\n Text(self.value)\n .font(.footnote)\n .foregroundStyle(.secondary)\n\n Spacer(minLength: 0)\n }\n .padding(.vertical, 1)\n }\n}\n\nprivate struct ProviderMetricInlineCostRow: View {\n let section: UsageMenuCardView.Model.ProviderCostSection\n let progressColor: Color\n let labelWidth: CGFloat\n\n var body: some View {\n HStack(alignment: .top, spacing: 10) {\n Text(self.section.title)\n .font(.subheadline.weight(.semibold))\n .frame(width: self.labelWidth, alignment: .leading)\n\n VStack(alignment: .leading, spacing: 4) {\n UsageProgressBar(\n percent: self.section.percentUsed,\n tint: self.progressColor,\n accessibilityLabel: \"Usage used\")\n .frame(minWidth: ProviderSettingsMetrics.metricBarWidth, maxWidth: .infinity)\n\n HStack(alignment: .firstTextBaseline, spacing: 8) {\n Text(String(format: \"%.0f%% used\", self.section.percentUsed))\n .font(.footnote)\n .foregroundStyle(.secondary)\n .monospacedDigit()\n Spacer(minLength: 8)\n Text(self.section.spendLine)\n .font(.footnote)\n .foregroundStyle(.secondary)\n }\n }\n\n Spacer(minLength: 0)\n }\n .padding(.vertical, 2)\n }\n}\n" }, { "path": "Sources/CodexBar/PreferencesProviderErrorView.swift", "content": "import SwiftUI\n\nstruct ProviderErrorDisplay {\n let preview: String\n let full: String\n}\n\n@MainActor\nstruct ProviderErrorView: View {\n let title: String\n let display: ProviderErrorDisplay\n @Binding var isExpanded: Bool\n let onCopy: () -> Void\n\n var body: some View {\n VStack(alignment: .leading, spacing: 6) {\n HStack(alignment: .firstTextBaseline, spacing: 8) {\n Text(self.title)\n .font(.footnote.weight(.semibold))\n .foregroundStyle(.secondary)\n Spacer()\n Button {\n self.onCopy()\n } label: {\n Image(systemName: \"doc.on.doc\")\n }\n .buttonStyle(.plain)\n .foregroundStyle(.secondary)\n .help(\"Copy error\")\n }\n\n Text(self.display.preview)\n .font(.footnote)\n .foregroundStyle(.secondary)\n .lineLimit(3)\n .fixedSize(horizontal: false, vertical: true)\n\n if self.display.preview != self.display.full {\n Button(self.isExpanded ? \"Hide details\" : \"Show details\") { self.isExpanded.toggle() }\n .buttonStyle(.link)\n .font(.footnote)\n }\n\n if self.isExpanded {\n Text(self.display.full)\n .font(.footnote)\n .textSelection(.enabled)\n .fixedSize(horizontal: false, vertical: true)\n }\n }\n .padding(.leading, 2)\n }\n}\n" }, { "path": "Sources/CodexBar/PreferencesProviderSettingsMetrics.swift", "content": "import AppKit\nimport SwiftUI\n\nenum ProviderSettingsMetrics {\n static let rowSpacing: CGFloat = 12\n static let rowInsets = EdgeInsets(top: 6, leading: 0, bottom: 6, trailing: 0)\n static let dividerBottomInset: CGFloat = 8\n static let listTopPadding: CGFloat = 12\n static let checkboxSize: CGFloat = 18\n static let iconSize: CGFloat = 18\n static let reorderHandleSize: CGFloat = 12\n static let reorderDotSize: CGFloat = 2\n static let reorderDotSpacing: CGFloat = 3\n static let pickerLabelWidth: CGFloat = 92\n static let sidebarWidth: CGFloat = 240\n static let sidebarCornerRadius: CGFloat = 12\n static let sidebarSubtitleHeight: CGFloat = {\n let font = NSFont.systemFont(ofSize: NSFont.smallSystemFontSize)\n let layout = NSLayoutManager()\n return ceil(layout.defaultLineHeight(for: font) * 2)\n }()\n\n static let detailMaxWidth: CGFloat = 640\n static let metricLabelWidth: CGFloat = 120\n static let metricBarWidth: CGFloat = 220\n\n static func labelWidth(for labels: [String], font: NSFont, minimum: CGFloat = 0) -> CGFloat {\n let maxWidth = labels\n .filter { !$0.isEmpty }\n .map { ($0 as NSString).size(withAttributes: [.font: font]).width }\n .max() ?? 0\n return max(minimum, ceil(maxWidth))\n }\n\n static func metricLabelFont() -> NSFont {\n let baseSize = NSFont.preferredFont(forTextStyle: .subheadline).pointSize\n return NSFont.systemFont(ofSize: baseSize, weight: .semibold)\n }\n\n static func infoLabelFont() -> NSFont {\n NSFont.preferredFont(forTextStyle: .footnote)\n }\n}\n" }, { "path": "Sources/CodexBar/PreferencesProviderSettingsRows.swift", "content": "import SwiftUI\n\nstruct ProviderSettingsSection: View {\n let title: String\n let spacing: CGFloat\n let verticalPadding: CGFloat\n let horizontalPadding: CGFloat\n @ViewBuilder let content: () -> Content\n\n init(\n title: String,\n spacing: CGFloat = 12,\n verticalPadding: CGFloat = 10,\n horizontalPadding: CGFloat = 4,\n @ViewBuilder content: @escaping () -> Content)\n {\n self.title = title\n self.spacing = spacing\n self.verticalPadding = verticalPadding\n self.horizontalPadding = horizontalPadding\n self.content = content\n }\n\n var body: some View {\n VStack(alignment: .leading, spacing: self.spacing) {\n Text(self.title)\n .font(.headline)\n self.content()\n }\n .frame(maxWidth: .infinity, alignment: .leading)\n .padding(.vertical, self.verticalPadding)\n .padding(.horizontal, self.horizontalPadding)\n }\n}\n\n@MainActor\nstruct ProviderSettingsToggleRowView: View {\n let toggle: ProviderSettingsToggleDescriptor\n\n var body: some View {\n VStack(alignment: .leading, spacing: 8) {\n HStack(alignment: .firstTextBaseline, spacing: 12) {\n VStack(alignment: .leading, spacing: 4) {\n Text(self.toggle.title)\n .font(.subheadline.weight(.semibold))\n Text(self.toggle.subtitle)\n .font(.footnote)\n .foregroundStyle(.secondary)\n .fixedSize(horizontal: false, vertical: true)\n }\n Spacer(minLength: 8)\n Toggle(\"\", isOn: self.toggle.binding)\n .labelsHidden()\n .toggleStyle(.switch)\n }\n\n if self.toggle.binding.wrappedValue {\n if let status = self.toggle.statusText?(), !status.isEmpty {\n Text(status)\n .font(.footnote)\n .foregroundStyle(.secondary)\n .lineLimit(4)\n .fixedSize(horizontal: false, vertical: true)\n }\n\n let actions = self.toggle.actions.filter { $0.isVisible?() ?? true }\n if !actions.isEmpty {\n HStack(spacing: 10) {\n ForEach(actions) { action in\n Button(action.title) {\n Task { @MainActor in\n await action.perform()\n }\n }\n .applyProviderSettingsButtonStyle(action.style)\n .controlSize(.small)\n }\n }\n }\n }\n }\n .onChange(of: self.toggle.binding.wrappedValue) { _, enabled in\n guard let onChange = self.toggle.onChange else { return }\n Task { @MainActor in\n await onChange(enabled)\n }\n }\n .task(id: self.toggle.binding.wrappedValue) {\n guard self.toggle.binding.wrappedValue else { return }\n guard let onAppear = self.toggle.onAppearWhenEnabled else { return }\n await onAppear()\n }\n }\n}\n\n@MainActor\nstruct ProviderSettingsPickerRowView: View {\n let picker: ProviderSettingsPickerDescriptor\n\n var body: some View {\n let isEnabled = self.picker.isEnabled?() ?? true\n VStack(alignment: .leading, spacing: 6) {\n HStack(alignment: .firstTextBaseline, spacing: 10) {\n Text(self.picker.title)\n .font(.subheadline.weight(.semibold))\n .frame(width: ProviderSettingsMetrics.pickerLabelWidth, alignment: .leading)\n\n Picker(\"\", selection: self.picker.binding) {\n ForEach(self.picker.options) { option in\n Text(option.title).tag(option.id)\n }\n }\n .labelsHidden()\n .pickerStyle(.menu)\n .controlSize(.small)\n\n if let trailingText = self.picker.trailingText?(), !trailingText.isEmpty {\n Text(trailingText)\n .font(.footnote)\n .foregroundStyle(.secondary)\n .lineLimit(1)\n .truncationMode(.tail)\n .padding(.leading, 4)\n }\n\n Spacer(minLength: 0)\n }\n\n let subtitle = self.picker.dynamicSubtitle?() ?? self.picker.subtitle\n if !subtitle.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty {\n Text(subtitle)\n .font(.footnote)\n .foregroundStyle(.secondary)\n .fixedSize(horizontal: false, vertical: true)\n }\n }\n .disabled(!isEnabled)\n .onChange(of: self.picker.binding.wrappedValue) { _, selection in\n guard let onChange = self.picker.onChange else { return }\n Task { @MainActor in\n await onChange(selection)\n }\n }\n }\n}\n\n@MainActor\nstruct ProviderSettingsFieldRowView: View {\n let field: ProviderSettingsFieldDescriptor\n\n var body: some View {\n VStack(alignment: .leading, spacing: 8) {\n let trimmedTitle = self.field.title.trimmingCharacters(in: .whitespacesAndNewlines)\n let trimmedSubtitle = self.field.subtitle.trimmingCharacters(in: .whitespacesAndNewlines)\n let hasHeader = !trimmedTitle.isEmpty || !trimmedSubtitle.isEmpty\n\n if hasHeader {\n VStack(alignment: .leading, spacing: 4) {\n if !trimmedTitle.isEmpty {\n Text(trimmedTitle)\n .font(.subheadline.weight(.semibold))\n }\n if !trimmedSubtitle.isEmpty {\n Text(trimmedSubtitle)\n .font(.footnote)\n .foregroundStyle(.secondary)\n .fixedSize(horizontal: false, vertical: true)\n }\n }\n }\n\n switch self.field.kind {\n case .plain:\n TextField(self.field.placeholder ?? \"\", text: self.field.binding)\n .textFieldStyle(.roundedBorder)\n .font(.footnote)\n .onTapGesture { self.field.onActivate?() }\n case .secure:\n SecureField(self.field.placeholder ?? \"\", text: self.field.binding)\n .textFieldStyle(.roundedBorder)\n .font(.footnote)\n .onTapGesture { self.field.onActivate?() }\n }\n\n let actions = self.field.actions.filter { $0.isVisible?() ?? true }\n if !actions.isEmpty {\n HStack(spacing: 10) {\n ForEach(actions) { action in\n Button(action.title) {\n Task { @MainActor in\n await action.perform()\n }\n }\n .applyProviderSettingsButtonStyle(action.style)\n .controlSize(.small)\n }\n }\n }\n }\n }\n}\n\n@MainActor\nstruct ProviderSettingsTokenAccountsRowView: View {\n let descriptor: ProviderSettingsTokenAccountsDescriptor\n @State private var newLabel: String = \"\"\n @State private var newToken: String = \"\"\n\n var body: some View {\n VStack(alignment: .leading, spacing: 8) {\n Text(self.descriptor.title)\n .font(.subheadline.weight(.semibold))\n\n if !self.descriptor.subtitle.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty {\n Text(self.descriptor.subtitle)\n .font(.footnote)\n .foregroundStyle(.secondary)\n .fixedSize(horizontal: false, vertical: true)\n }\n\n let accounts = self.descriptor.accounts()\n if accounts.isEmpty {\n Text(\"No token accounts yet.\")\n .font(.footnote)\n .foregroundStyle(.secondary)\n } else {\n let selectedIndex = min(self.descriptor.activeIndex(), max(0, accounts.count - 1))\n Picker(\"\", selection: Binding(\n get: { selectedIndex },\n set: { index in self.descriptor.setActiveIndex(index) }))\n {\n ForEach(Array(accounts.enumerated()), id: \\.offset) { index, account in\n Text(account.displayName).tag(index)\n }\n }\n .labelsHidden()\n .pickerStyle(.menu)\n .controlSize(.small)\n\n Button(\"Remove selected account\") {\n let account = accounts[selectedIndex]\n self.descriptor.removeAccount(account.id)\n }\n .buttonStyle(.bordered)\n .controlSize(.small)\n }\n\n HStack(spacing: 8) {\n TextField(\"Label\", text: self.$newLabel)\n .textFieldStyle(.roundedBorder)\n .font(.footnote)\n SecureField(self.descriptor.placeholder, text: self.$newToken)\n .textFieldStyle(.roundedBorder)\n .font(.footnote)\n Button(\"Add\") {\n let label = self.newLabel.trimmingCharacters(in: .whitespacesAndNewlines)\n let token = self.newToken.trimmingCharacters(in: .whitespacesAndNewlines)\n guard !label.isEmpty, !token.isEmpty else { return }\n self.descriptor.addAccount(label, token)\n self.newLabel = \"\"\n self.newToken = \"\"\n }\n .buttonStyle(.bordered)\n .controlSize(.small)\n .disabled(self.newLabel.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty ||\n self.newToken.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty)\n }\n\n HStack(spacing: 10) {\n Button(\"Open token file\") {\n self.descriptor.openConfigFile()\n }\n .buttonStyle(.link)\n .controlSize(.small)\n Button(\"Reload\") {\n self.descriptor.reloadFromDisk()\n }\n .buttonStyle(.link)\n .controlSize(.small)\n }\n }\n }\n}\n\nextension View {\n @ViewBuilder\n fileprivate func applyProviderSettingsButtonStyle(_ style: ProviderSettingsActionDescriptor.Style) -> some View {\n switch style {\n case .bordered:\n self.buttonStyle(.bordered)\n case .link:\n self.buttonStyle(.link)\n }\n }\n}\n" }, { "path": "Sources/CodexBar/PreferencesProviderSidebarView.swift", "content": "import CodexBarCore\nimport SwiftUI\nimport UniformTypeIdentifiers\n\n@MainActor\nstruct ProviderSidebarListView: View {\n let providers: [UsageProvider]\n @Bindable var store: UsageStore\n let isEnabled: (UsageProvider) -> Binding\n let subtitle: (UsageProvider) -> String\n @Binding var selection: UsageProvider?\n let moveProviders: (IndexSet, Int) -> Void\n @State private var draggingProvider: UsageProvider?\n\n var body: some View {\n List(selection: self.$selection) {\n ForEach(self.providers, id: \\.self) { provider in\n ProviderSidebarRowView(\n provider: provider,\n store: self.store,\n isEnabled: self.isEnabled(provider),\n subtitle: self.subtitle(provider),\n draggingProvider: self.$draggingProvider)\n .tag(provider)\n .onDrop(\n of: [UTType.plainText],\n delegate: ProviderSidebarDropDelegate(\n item: provider,\n providers: self.providers,\n dragging: self.$draggingProvider,\n moveProviders: self.moveProviders))\n }\n }\n .listStyle(.sidebar)\n .scrollContentBackground(.hidden)\n .background(\n RoundedRectangle(cornerRadius: ProviderSettingsMetrics.sidebarCornerRadius, style: .continuous)\n .fill(.regularMaterial))\n .overlay(\n RoundedRectangle(cornerRadius: ProviderSettingsMetrics.sidebarCornerRadius, style: .continuous)\n .stroke(Color(nsColor: .separatorColor).opacity(0.7), lineWidth: 1))\n .clipShape(RoundedRectangle(cornerRadius: ProviderSettingsMetrics.sidebarCornerRadius, style: .continuous))\n .frame(minWidth: ProviderSettingsMetrics.sidebarWidth, maxWidth: ProviderSettingsMetrics.sidebarWidth)\n }\n}\n\n@MainActor\nprivate struct ProviderSidebarRowView: View {\n let provider: UsageProvider\n @Bindable var store: UsageStore\n @Binding var isEnabled: Bool\n let subtitle: String\n @Binding var draggingProvider: UsageProvider?\n\n var body: some View {\n let isRefreshing = self.store.refreshingProviders.contains(self.provider)\n let showStatus = self.store.statusChecksEnabled\n let statusText = self.statusText\n\n HStack(alignment: .center, spacing: 10) {\n ProviderSidebarReorderHandle()\n .contentShape(Rectangle())\n .padding(.vertical, 4)\n .padding(.horizontal, 2)\n .help(\"Drag to reorder\")\n .onDrag {\n self.draggingProvider = self.provider\n return NSItemProvider(object: self.provider.rawValue as NSString)\n }\n\n ProviderSidebarBrandIcon(provider: self.provider)\n\n VStack(alignment: .leading, spacing: 2) {\n HStack(spacing: 6) {\n Text(self.store.metadata(for: self.provider).displayName)\n .font(.subheadline.weight(.semibold))\n .foregroundStyle(.primary)\n\n if showStatus {\n ProviderStatusDot(indicator: self.store.statusIndicator(for: self.provider))\n }\n\n if isRefreshing {\n ProgressView()\n .controlSize(.mini)\n }\n }\n Text(statusText)\n .font(.caption)\n .foregroundStyle(.secondary)\n .lineLimit(2)\n .frame(height: ProviderSettingsMetrics.sidebarSubtitleHeight, alignment: .topLeading)\n }\n\n Spacer(minLength: 8)\n\n Toggle(\"\", isOn: self.$isEnabled)\n .labelsHidden()\n .toggleStyle(.checkbox)\n .controlSize(.small)\n }\n .contentShape(Rectangle())\n .padding(.vertical, 2)\n }\n\n private var statusText: String {\n guard !self.isEnabled else { return self.subtitle }\n let lines = self.subtitle.split(separator: \"\\n\", omittingEmptySubsequences: false)\n if lines.count >= 2 {\n let first = lines[0]\n let rest = lines.dropFirst().joined(separator: \"\\n\")\n return \"Disabled — \\(first)\\n\\(rest)\"\n }\n return \"Disabled — \\(self.subtitle)\"\n }\n}\n\nprivate struct ProviderSidebarReorderHandle: View {\n var body: some View {\n VStack(spacing: ProviderSettingsMetrics.reorderDotSpacing) {\n ForEach(0..<3, id: \\.self) { _ in\n HStack(spacing: ProviderSettingsMetrics.reorderDotSpacing) {\n Circle()\n .frame(\n width: ProviderSettingsMetrics.reorderDotSize,\n height: ProviderSettingsMetrics.reorderDotSize)\n Circle()\n .frame(\n width: ProviderSettingsMetrics.reorderDotSize,\n height: ProviderSettingsMetrics.reorderDotSize)\n }\n }\n }\n .frame(\n width: ProviderSettingsMetrics.reorderHandleSize,\n height: ProviderSettingsMetrics.reorderHandleSize)\n .foregroundStyle(.tertiary)\n .accessibilityLabel(\"Reorder\")\n }\n}\n\n@MainActor\nprivate struct ProviderSidebarBrandIcon: View {\n let provider: UsageProvider\n\n var body: some View {\n if let brand = ProviderBrandIcon.image(for: self.provider) {\n Image(nsImage: brand)\n .resizable()\n .scaledToFit()\n .frame(width: ProviderSettingsMetrics.iconSize, height: ProviderSettingsMetrics.iconSize)\n .foregroundStyle(.secondary)\n .accessibilityHidden(true)\n } else {\n Image(systemName: \"circle.dotted\")\n .font(.system(size: ProviderSettingsMetrics.iconSize, weight: .regular))\n .foregroundStyle(.secondary)\n .accessibilityHidden(true)\n }\n }\n}\n\nprivate struct ProviderSidebarDropDelegate: DropDelegate {\n let item: UsageProvider\n let providers: [UsageProvider]\n @Binding var dragging: UsageProvider?\n let moveProviders: (IndexSet, Int) -> Void\n\n func dropEntered(info _: DropInfo) {\n guard let dragging, dragging != self.item else { return }\n guard let fromIndex = self.providers.firstIndex(of: dragging),\n let toIndex = self.providers.firstIndex(of: self.item)\n else { return }\n\n if fromIndex == toIndex { return }\n let adjustedIndex = toIndex > fromIndex ? toIndex + 1 : toIndex\n self.moveProviders(IndexSet(integer: fromIndex), adjustedIndex)\n }\n\n func dropUpdated(info _: DropInfo) -> DropProposal? {\n DropProposal(operation: .move)\n }\n\n func performDrop(info _: DropInfo) -> Bool {\n self.dragging = nil\n return true\n }\n}\n\nprivate struct ProviderStatusDot: View {\n let indicator: ProviderStatusIndicator\n\n var body: some View {\n Circle()\n .fill(self.statusColor)\n .frame(width: 6, height: 6)\n .accessibilityHidden(true)\n }\n\n private var statusColor: Color {\n switch self.indicator {\n case .none: .green\n case .minor: .yellow\n case .major: .orange\n case .critical: .red\n case .maintenance: .gray\n case .unknown: .gray\n }\n }\n}\n" }, { "path": "Sources/CodexBar/PreferencesProvidersPane+Testing.swift", "content": "#if DEBUG\nimport CodexBarCore\nimport SwiftUI\n\nextension ProvidersPane {\n func _test_binding(for provider: UsageProvider) -> Binding {\n self.binding(for: provider)\n }\n\n func _test_providerSubtitle(_ provider: UsageProvider) -> String {\n self.providerSubtitle(provider)\n }\n\n func _test_menuBarMetricPicker(for provider: UsageProvider) -> ProviderSettingsPickerDescriptor? {\n self.menuBarMetricPicker(for: provider)\n }\n\n func _test_tokenAccountDescriptor(for provider: UsageProvider) -> ProviderSettingsTokenAccountsDescriptor? {\n self.tokenAccountDescriptor(for: provider)\n }\n\n func _test_menuCardModel(for provider: UsageProvider) -> UsageMenuCardView.Model {\n self.menuCardModel(for: provider)\n }\n}\n\n@MainActor\nenum ProvidersPaneTestHarness {\n static func exercise(settings: SettingsStore, store: UsageStore) {\n self.prepareTestState(settings: settings, store: store)\n let pane = ProvidersPane(settings: settings, store: store)\n self.exercisePaneBasics(pane: pane)\n\n let descriptors = self.makeDescriptors()\n self.exerciseDetailViews(store: store, descriptors: descriptors)\n }\n\n private static func prepareTestState(settings: SettingsStore, store: UsageStore) {\n store.versions[.codex] = \"1.0.0\"\n store.versions[.claude] = \"2.0.0 (build 123)\"\n store.versions[.cursor] = nil\n store._setSnapshotForTesting(\n UsageSnapshot(primary: nil, secondary: nil, updatedAt: Date()),\n provider: .codex)\n store._setSnapshotForTesting(\n UsageSnapshot(primary: nil, secondary: nil, updatedAt: Date()),\n provider: .minimax)\n store._setErrorForTesting(String(repeating: \"x\", count: 200), provider: .cursor)\n store.lastSourceLabels[.minimax] = \"cookies\"\n store.refreshingProviders.insert(.codex)\n\n settings.claudeCookieSource = .manual\n settings.cursorCookieSource = .manual\n settings.opencodeCookieSource = .manual\n settings.factoryCookieSource = .manual\n settings.minimaxCookieSource = .manual\n settings.augmentCookieSource = .manual\n }\n\n private static func exercisePaneBasics(pane: ProvidersPane) {\n _ = pane._test_binding(for: .codex).wrappedValue\n _ = pane._test_providerSubtitle(.codex)\n _ = pane._test_providerSubtitle(.claude)\n _ = pane._test_providerSubtitle(.cursor)\n _ = pane._test_providerSubtitle(.opencode)\n _ = pane._test_providerSubtitle(.zai)\n _ = pane._test_providerSubtitle(.synthetic)\n _ = pane._test_providerSubtitle(.minimax)\n _ = pane._test_providerSubtitle(.kimi)\n _ = pane._test_providerSubtitle(.gemini)\n\n _ = pane._test_menuBarMetricPicker(for: .codex)\n _ = pane._test_menuBarMetricPicker(for: .gemini)\n _ = pane._test_menuBarMetricPicker(for: .zai)\n\n if let descriptor = pane._test_tokenAccountDescriptor(for: .claude) {\n _ = descriptor.isVisible?()\n _ = descriptor.accounts()\n }\n }\n\n private static func exerciseDetailViews(store: UsageStore, descriptors: ProviderListTestDescriptors) {\n var isEnabled = true\n let enabledBinding = Binding(get: { isEnabled }, set: { isEnabled = $0 })\n let pane = ProvidersPane(settings: store.settings, store: store)\n let model = pane._test_menuCardModel(for: .codex)\n var expanded = false\n let expandedBinding = Binding(get: { expanded }, set: { expanded = $0 })\n\n _ = ProviderDetailView(\n provider: .codex,\n store: store,\n isEnabled: enabledBinding,\n subtitle: \"Subtitle\",\n model: model,\n settingsPickers: [descriptors.picker],\n settingsToggles: [descriptors.toggle],\n settingsFields: [descriptors.fieldPlain, descriptors.fieldSecure],\n settingsTokenAccounts: descriptors.tokenAccountsEmpty,\n errorDisplay: ProviderErrorDisplay(preview: \"Preview\", full: \"Full\"),\n isErrorExpanded: expandedBinding,\n onCopyError: { _ in },\n onRefresh: {}).body\n }\n\n private static func makeDescriptors() -> ProviderListTestDescriptors {\n let toggleBinding = Binding(get: { true }, set: { _ in })\n let actionBordered = ProviderSettingsActionDescriptor(\n id: \"action-bordered\",\n title: \"Bordered\",\n style: .bordered,\n isVisible: { true },\n perform: { await Task.yield() })\n let actionLink = ProviderSettingsActionDescriptor(\n id: \"action-link\",\n title: \"Link\",\n style: .link,\n isVisible: { true },\n perform: { await Task.yield() })\n let toggle = ProviderSettingsToggleDescriptor(\n id: \"toggle\",\n title: \"Toggle\",\n subtitle: \"Toggle subtitle\",\n binding: toggleBinding,\n statusText: { \"Status\" },\n actions: [actionBordered, actionLink],\n isVisible: { true },\n onChange: nil,\n onAppDidBecomeActive: nil,\n onAppearWhenEnabled: nil)\n let picker = ProviderSettingsPickerDescriptor(\n id: \"picker\",\n title: \"Picker\",\n subtitle: \"Picker subtitle\",\n dynamicSubtitle: nil,\n binding: Binding(get: { \"a\" }, set: { _ in }),\n options: [\n ProviderSettingsPickerOption(id: \"a\", title: \"Option A\"),\n ProviderSettingsPickerOption(id: \"b\", title: \"Option B\"),\n ],\n isVisible: { true },\n onChange: nil,\n trailingText: { \"Trailing\" })\n let fieldPlain = ProviderSettingsFieldDescriptor(\n id: \"plain\",\n title: \"Field\",\n subtitle: \"Field subtitle\",\n kind: .plain,\n placeholder: \"Placeholder\",\n binding: Binding(get: { \"\" }, set: { _ in }),\n actions: [actionBordered],\n isVisible: { true },\n onActivate: nil)\n let fieldSecure = ProviderSettingsFieldDescriptor(\n id: \"secure\",\n title: \"Secure\",\n subtitle: \"Secure subtitle\",\n kind: .secure,\n placeholder: \"Secure\",\n binding: Binding(get: { \"\" }, set: { _ in }),\n actions: [actionLink],\n isVisible: { true },\n onActivate: nil)\n let tokenAccountsEmpty = ProviderSettingsTokenAccountsDescriptor(\n id: \"accounts-empty\",\n title: \"Accounts\",\n subtitle: \"Accounts subtitle\",\n placeholder: \"Token\",\n provider: .codex,\n isVisible: { true },\n accounts: { [] },\n activeIndex: { 0 },\n setActiveIndex: { _ in },\n addAccount: { _, _ in },\n removeAccount: { _ in },\n openConfigFile: {},\n reloadFromDisk: {})\n\n return ProviderListTestDescriptors(\n toggle: toggle,\n picker: picker,\n fieldPlain: fieldPlain,\n fieldSecure: fieldSecure,\n tokenAccountsEmpty: tokenAccountsEmpty)\n }\n}\n\nprivate struct ProviderListTestDescriptors {\n let toggle: ProviderSettingsToggleDescriptor\n let picker: ProviderSettingsPickerDescriptor\n let fieldPlain: ProviderSettingsFieldDescriptor\n let fieldSecure: ProviderSettingsFieldDescriptor\n let tokenAccountsEmpty: ProviderSettingsTokenAccountsDescriptor\n}\n#endif\n" }, { "path": "Sources/CodexBar/PreferencesProvidersPane.swift", "content": "import AppKit\nimport CodexBarCore\nimport SwiftUI\n\n@MainActor\nstruct ProvidersPane: View {\n @Bindable var settings: SettingsStore\n @Bindable var store: UsageStore\n @State private var expandedErrors: Set = []\n @State private var settingsStatusTextByID: [String: String] = [:]\n @State private var settingsLastAppActiveRunAtByID: [String: Date] = [:]\n @State private var activeConfirmation: ProviderSettingsConfirmationState?\n @State private var selectedProvider: UsageProvider?\n\n private var providers: [UsageProvider] {\n self.settings.orderedProviders()\n }\n\n var body: some View {\n HStack(alignment: .top, spacing: 16) {\n ProviderSidebarListView(\n providers: self.providers,\n store: self.store,\n isEnabled: { provider in self.binding(for: provider) },\n subtitle: { provider in self.providerSubtitle(provider) },\n selection: self.$selectedProvider,\n moveProviders: { fromOffsets, toOffset in\n self.settings.moveProvider(fromOffsets: fromOffsets, toOffset: toOffset)\n })\n\n if let provider = self.selectedProvider ?? self.providers.first {\n ProviderDetailView(\n provider: provider,\n store: self.store,\n isEnabled: self.binding(for: provider),\n subtitle: self.providerSubtitle(provider),\n model: self.menuCardModel(for: provider),\n settingsPickers: self.extraSettingsPickers(for: provider),\n settingsToggles: self.extraSettingsToggles(for: provider),\n settingsFields: self.extraSettingsFields(for: provider),\n settingsTokenAccounts: self.tokenAccountDescriptor(for: provider),\n errorDisplay: self.providerErrorDisplay(provider),\n isErrorExpanded: self.expandedBinding(for: provider),\n onCopyError: { text in self.copyToPasteboard(text) },\n onRefresh: {\n Task { @MainActor in\n await ProviderInteractionContext.$current.withValue(.userInitiated) {\n await self.store.refreshProvider(provider, allowDisabled: true)\n }\n }\n })\n } else {\n Text(\"Select a provider\")\n .foregroundStyle(.secondary)\n .frame(maxWidth: .infinity, maxHeight: .infinity, alignment: .center)\n }\n }\n .frame(maxWidth: .infinity, maxHeight: .infinity, alignment: .leading)\n .onAppear {\n self.ensureSelection()\n }\n .onChange(of: self.providers) { _, _ in\n self.ensureSelection()\n }\n .onReceive(NotificationCenter.default.publisher(for: NSApplication.didBecomeActiveNotification)) { _ in\n self.runSettingsDidBecomeActiveHooks()\n }\n .alert(\n self.activeConfirmation?.title ?? \"\",\n isPresented: Binding(\n get: { self.activeConfirmation != nil },\n set: { isPresented in\n if !isPresented { self.activeConfirmation = nil }\n }),\n actions: {\n if let active = self.activeConfirmation {\n Button(active.confirmTitle) {\n active.onConfirm()\n self.activeConfirmation = nil\n }\n Button(\"Cancel\", role: .cancel) { self.activeConfirmation = nil }\n }\n },\n message: {\n if let active = self.activeConfirmation {\n Text(active.message)\n }\n })\n }\n\n private func ensureSelection() {\n guard !self.providers.isEmpty else {\n self.selectedProvider = nil\n return\n }\n if let selected = self.selectedProvider, self.providers.contains(selected) {\n return\n }\n self.selectedProvider = self.providers.first\n }\n\n func binding(for provider: UsageProvider) -> Binding {\n let meta = self.store.metadata(for: provider)\n return Binding(\n get: { self.settings.isProviderEnabled(provider: provider, metadata: meta) },\n set: { newValue in\n self.settings.setProviderEnabled(provider: provider, metadata: meta, enabled: newValue)\n })\n }\n\n func providerSubtitle(_ provider: UsageProvider) -> String {\n let meta = self.store.metadata(for: provider)\n let usageText: String\n if let snapshot = self.store.snapshot(for: provider) {\n let relative = snapshot.updatedAt.relativeDescription()\n usageText = relative\n } else if self.store.isStale(provider: provider) {\n usageText = \"last fetch failed\"\n } else {\n usageText = \"usage not fetched yet\"\n }\n\n let presentationContext = ProviderPresentationContext(\n provider: provider,\n settings: self.settings,\n store: self.store,\n metadata: meta)\n let presentation = ProviderCatalog.implementation(for: provider)?\n .presentation(context: presentationContext)\n ?? ProviderPresentation(detailLine: ProviderPresentation.standardDetailLine)\n let detailLine = presentation.detailLine(presentationContext)\n\n return \"\\(detailLine)\\n\\(usageText)\"\n }\n\n private func providerErrorDisplay(_ provider: UsageProvider) -> ProviderErrorDisplay? {\n guard let raw = self.store.error(for: provider), !raw.isEmpty else { return nil }\n return ProviderErrorDisplay(\n preview: self.truncated(raw, prefix: \"\"),\n full: raw)\n }\n\n private func extraSettingsToggles(for provider: UsageProvider) -> [ProviderSettingsToggleDescriptor] {\n guard let impl = ProviderCatalog.implementation(for: provider) else { return [] }\n let context = self.makeSettingsContext(provider: provider)\n return impl.settingsToggles(context: context)\n .filter { $0.isVisible?() ?? true }\n }\n\n private func extraSettingsPickers(for provider: UsageProvider) -> [ProviderSettingsPickerDescriptor] {\n guard let impl = ProviderCatalog.implementation(for: provider) else { return [] }\n let context = self.makeSettingsContext(provider: provider)\n let providerPickers = impl.settingsPickers(context: context)\n .filter { $0.isVisible?() ?? true }\n if let menuBarPicker = self.menuBarMetricPicker(for: provider) {\n return [menuBarPicker] + providerPickers\n }\n return providerPickers\n }\n\n private func extraSettingsFields(for provider: UsageProvider) -> [ProviderSettingsFieldDescriptor] {\n guard let impl = ProviderCatalog.implementation(for: provider) else { return [] }\n let context = self.makeSettingsContext(provider: provider)\n return impl.settingsFields(context: context)\n .filter { $0.isVisible?() ?? true }\n }\n\n func tokenAccountDescriptor(for provider: UsageProvider) -> ProviderSettingsTokenAccountsDescriptor? {\n guard let support = TokenAccountSupportCatalog.support(for: provider) else { return nil }\n let context = self.makeSettingsContext(provider: provider)\n return ProviderSettingsTokenAccountsDescriptor(\n id: \"token-accounts-\\(provider.rawValue)\",\n title: support.title,\n subtitle: support.subtitle,\n placeholder: support.placeholder,\n provider: provider,\n isVisible: {\n ProviderCatalog.implementation(for: provider)?\n .tokenAccountsVisibility(context: context, support: support)\n ?? (!support.requiresManualCookieSource ||\n !context.settings.tokenAccounts(for: provider).isEmpty)\n },\n accounts: { self.settings.tokenAccounts(for: provider) },\n activeIndex: {\n let data = self.settings.tokenAccountsData(for: provider)\n return data?.clampedActiveIndex() ?? 0\n },\n setActiveIndex: { index in\n self.settings.setActiveTokenAccountIndex(index, for: provider)\n Task { @MainActor in\n await ProviderInteractionContext.$current.withValue(.userInitiated) {\n await self.store.refreshProvider(provider, allowDisabled: true)\n }\n }\n },\n addAccount: { label, token in\n self.settings.addTokenAccount(provider: provider, label: label, token: token)\n Task { @MainActor in\n await ProviderInteractionContext.$current.withValue(.userInitiated) {\n await self.store.refreshProvider(provider, allowDisabled: true)\n }\n }\n },\n removeAccount: { accountID in\n self.settings.removeTokenAccount(provider: provider, accountID: accountID)\n Task { @MainActor in\n await ProviderInteractionContext.$current.withValue(.userInitiated) {\n await self.store.refreshProvider(provider, allowDisabled: true)\n }\n }\n },\n openConfigFile: {\n self.settings.openTokenAccountsFile()\n },\n reloadFromDisk: {\n self.settings.reloadTokenAccounts()\n Task { @MainActor in\n await ProviderInteractionContext.$current.withValue(.userInitiated) {\n await self.store.refreshProvider(provider, allowDisabled: true)\n }\n }\n })\n }\n\n private func makeSettingsContext(provider: UsageProvider) -> ProviderSettingsContext {\n ProviderSettingsContext(\n provider: provider,\n settings: self.settings,\n store: self.store,\n boolBinding: { keyPath in\n Binding(\n get: { self.settings[keyPath: keyPath] },\n set: { self.settings[keyPath: keyPath] = $0 })\n },\n stringBinding: { keyPath in\n Binding(\n get: { self.settings[keyPath: keyPath] },\n set: { self.settings[keyPath: keyPath] = $0 })\n },\n statusText: { id in\n self.settingsStatusTextByID[id]\n },\n setStatusText: { id, text in\n if let text {\n self.settingsStatusTextByID[id] = text\n } else {\n self.settingsStatusTextByID.removeValue(forKey: id)\n }\n },\n lastAppActiveRunAt: { id in\n self.settingsLastAppActiveRunAtByID[id]\n },\n setLastAppActiveRunAt: { id, date in\n if let date {\n self.settingsLastAppActiveRunAtByID[id] = date\n } else {\n self.settingsLastAppActiveRunAtByID.removeValue(forKey: id)\n }\n },\n requestConfirmation: { confirmation in\n self.activeConfirmation = ProviderSettingsConfirmationState(confirmation: confirmation)\n })\n }\n\n func menuBarMetricPicker(for provider: UsageProvider) -> ProviderSettingsPickerDescriptor? {\n if provider == .zai { return nil }\n let options: [ProviderSettingsPickerOption]\n if provider == .openrouter {\n options = [\n ProviderSettingsPickerOption(id: MenuBarMetricPreference.automatic.rawValue, title: \"Automatic\"),\n ProviderSettingsPickerOption(\n id: MenuBarMetricPreference.primary.rawValue,\n title: \"Primary (API key limit)\"),\n ]\n } else {\n let metadata = self.store.metadata(for: provider)\n let supportsAverage = self.settings.menuBarMetricSupportsAverage(for: provider)\n var metricOptions: [ProviderSettingsPickerOption] = [\n ProviderSettingsPickerOption(id: MenuBarMetricPreference.automatic.rawValue, title: \"Automatic\"),\n ProviderSettingsPickerOption(\n id: MenuBarMetricPreference.primary.rawValue,\n title: \"Primary (\\(metadata.sessionLabel))\"),\n ProviderSettingsPickerOption(\n id: MenuBarMetricPreference.secondary.rawValue,\n title: \"Secondary (\\(metadata.weeklyLabel))\"),\n ]\n if supportsAverage {\n metricOptions.append(ProviderSettingsPickerOption(\n id: MenuBarMetricPreference.average.rawValue,\n title: \"Average (\\(metadata.sessionLabel) + \\(metadata.weeklyLabel))\"))\n }\n options = metricOptions\n }\n return ProviderSettingsPickerDescriptor(\n id: \"menuBarMetric\",\n title: \"Menu bar metric\",\n subtitle: \"Choose which window drives the menu bar percent.\",\n binding: Binding(\n get: { self.settings.menuBarMetricPreference(for: provider).rawValue },\n set: { rawValue in\n guard let preference = MenuBarMetricPreference(rawValue: rawValue) else { return }\n self.settings.setMenuBarMetricPreference(preference, for: provider)\n }),\n options: options,\n isVisible: { true },\n onChange: nil)\n }\n\n func menuCardModel(for provider: UsageProvider) -> UsageMenuCardView.Model {\n let metadata = self.store.metadata(for: provider)\n let snapshot = self.store.snapshot(for: provider)\n let credits: CreditsSnapshot?\n let creditsError: String?\n let dashboard: OpenAIDashboardSnapshot?\n let dashboardError: String?\n let tokenSnapshot: CostUsageTokenSnapshot?\n let tokenError: String?\n if provider == .codex {\n credits = self.store.credits\n creditsError = self.store.lastCreditsError\n dashboard = self.store.openAIDashboardRequiresLogin ? nil : self.store.openAIDashboard\n dashboardError = self.store.lastOpenAIDashboardError\n tokenSnapshot = self.store.tokenSnapshot(for: provider)\n tokenError = self.store.tokenError(for: provider)\n } else if provider == .claude || provider == .vertexai {\n credits = nil\n creditsError = nil\n dashboard = nil\n dashboardError = nil\n tokenSnapshot = self.store.tokenSnapshot(for: provider)\n tokenError = self.store.tokenError(for: provider)\n } else {\n credits = nil\n creditsError = nil\n dashboard = nil\n dashboardError = nil\n tokenSnapshot = nil\n tokenError = nil\n }\n\n let now = Date()\n let weeklyPace = snapshot?.secondary.flatMap { window in\n self.store.weeklyPace(provider: provider, window: window, now: now)\n }\n let input = UsageMenuCardView.Model.Input(\n provider: provider,\n metadata: metadata,\n snapshot: snapshot,\n credits: credits,\n creditsError: creditsError,\n dashboard: dashboard,\n dashboardError: dashboardError,\n tokenSnapshot: tokenSnapshot,\n tokenError: tokenError,\n account: self.store.accountInfo(),\n isRefreshing: self.store.refreshingProviders.contains(provider),\n lastError: self.store.error(for: provider),\n usageBarsShowUsed: self.settings.usageBarsShowUsed,\n resetTimeDisplayStyle: self.settings.resetTimeDisplayStyle,\n tokenCostUsageEnabled: self.settings.isCostUsageEffectivelyEnabled(for: provider),\n showOptionalCreditsAndExtraUsage: self.settings.showOptionalCreditsAndExtraUsage,\n hidePersonalInfo: self.settings.hidePersonalInfo,\n weeklyPace: weeklyPace,\n now: now)\n return UsageMenuCardView.Model.make(input)\n }\n\n private func runSettingsDidBecomeActiveHooks() {\n for provider in UsageProvider.allCases {\n for toggle in self.extraSettingsToggles(for: provider) {\n guard let hook = toggle.onAppDidBecomeActive else { continue }\n Task { @MainActor in\n await hook()\n }\n }\n }\n }\n\n private func truncated(_ text: String, prefix: String, maxLength: Int = 160) -> String {\n var message = text.trimmingCharacters(in: .whitespacesAndNewlines)\n if message.count > maxLength {\n let idx = message.index(message.startIndex, offsetBy: maxLength)\n message = \"\\(message[.. Binding {\n Binding(\n get: { self.expandedErrors.contains(provider) },\n set: { expanded in\n if expanded {\n self.expandedErrors.insert(provider)\n } else {\n self.expandedErrors.remove(provider)\n }\n })\n }\n\n private func copyToPasteboard(_ text: String) {\n let pb = NSPasteboard.general\n pb.clearContents()\n pb.setString(text, forType: .string)\n }\n}\n\n@MainActor\nstruct ProviderSettingsConfirmationState: Identifiable {\n let id = UUID()\n let title: String\n let message: String\n let confirmTitle: String\n let onConfirm: () -> Void\n\n init(confirmation: ProviderSettingsConfirmation) {\n self.title = confirmation.title\n self.message = confirmation.message\n self.confirmTitle = confirmation.confirmTitle\n self.onConfirm = confirmation.onConfirm\n }\n}\n" }, { "path": "Sources/CodexBar/PreferencesSelection.swift", "content": "import Foundation\nimport Observation\n\n@MainActor\n@Observable\nfinal class PreferencesSelection {\n var tab: PreferencesTab = .general\n}\n" }, { "path": "Sources/CodexBar/PreferencesView.swift", "content": "import AppKit\nimport SwiftUI\n\nenum PreferencesTab: String, Hashable {\n case general\n case providers\n case display\n case advanced\n case about\n case debug\n\n static let defaultWidth: CGFloat = 496\n static let providersWidth: CGFloat = 720\n static let windowHeight: CGFloat = 580\n\n var preferredWidth: CGFloat {\n self == .providers ? PreferencesTab.providersWidth : PreferencesTab.defaultWidth\n }\n\n var preferredHeight: CGFloat {\n PreferencesTab.windowHeight\n }\n}\n\n@MainActor\nstruct PreferencesView: View {\n @Bindable var settings: SettingsStore\n @Bindable var store: UsageStore\n let updater: UpdaterProviding\n @Bindable var selection: PreferencesSelection\n @State private var contentWidth: CGFloat = PreferencesTab.general.preferredWidth\n @State private var contentHeight: CGFloat = PreferencesTab.general.preferredHeight\n\n var body: some View {\n TabView(selection: self.$selection.tab) {\n GeneralPane(settings: self.settings, store: self.store)\n .tabItem { Label(\"General\", systemImage: \"gearshape\") }\n .tag(PreferencesTab.general)\n\n ProvidersPane(settings: self.settings, store: self.store)\n .tabItem { Label(\"Providers\", systemImage: \"square.grid.2x2\") }\n .tag(PreferencesTab.providers)\n\n DisplayPane(settings: self.settings, store: self.store)\n .tabItem { Label(\"Display\", systemImage: \"eye\") }\n .tag(PreferencesTab.display)\n\n AdvancedPane(settings: self.settings)\n .tabItem { Label(\"Advanced\", systemImage: \"slider.horizontal.3\") }\n .tag(PreferencesTab.advanced)\n\n AboutPane(updater: self.updater)\n .tabItem { Label(\"About\", systemImage: \"info.circle\") }\n .tag(PreferencesTab.about)\n\n if self.settings.debugMenuEnabled {\n DebugPane(settings: self.settings, store: self.store)\n .tabItem { Label(\"Debug\", systemImage: \"ladybug\") }\n .tag(PreferencesTab.debug)\n }\n }\n .padding(.horizontal, 24)\n .padding(.vertical, 16)\n .frame(width: self.contentWidth, height: self.contentHeight)\n .onAppear {\n self.updateLayout(for: self.selection.tab, animate: false)\n self.ensureValidTabSelection()\n }\n .onChange(of: self.selection.tab) { _, newValue in\n self.updateLayout(for: newValue, animate: true)\n }\n .onChange(of: self.settings.debugMenuEnabled) { _, _ in\n self.ensureValidTabSelection()\n }\n }\n\n private func updateLayout(for tab: PreferencesTab, animate: Bool) {\n let change = {\n self.contentWidth = tab.preferredWidth\n self.contentHeight = tab.preferredHeight\n }\n if animate {\n withAnimation(.spring(response: 0.32, dampingFraction: 0.85)) { change() }\n } else {\n change()\n }\n }\n\n private func ensureValidTabSelection() {\n if !self.settings.debugMenuEnabled, self.selection.tab == .debug {\n self.selection.tab = .general\n self.updateLayout(for: .general, animate: true)\n }\n }\n}\n" }, { "path": "Sources/CodexBar/ProviderBrandIcon.swift", "content": "import AppKit\nimport CodexBarCore\n\nenum ProviderBrandIcon {\n private static let size = NSSize(width: 16, height: 16)\n\n /// Lazy-loaded resource bundle for provider icons.\n private static let resourceBundle: Bundle? = {\n // SwiftPM creates a CodexBar_CodexBar.bundle for resources in the CodexBar target.\n if let bundleURL = Bundle.main.url(forResource: \"CodexBar_CodexBar\", withExtension: \"bundle\"),\n let bundle = Bundle(url: bundleURL)\n {\n return bundle\n }\n // Fallback to main bundle for development/testing.\n return Bundle.main\n }()\n\n static func image(for provider: UsageProvider) -> NSImage? {\n let baseName = ProviderDescriptorRegistry.descriptor(for: provider).branding.iconResourceName\n guard let bundle = self.resourceBundle,\n let url = bundle.url(forResource: baseName, withExtension: \"svg\"),\n let image = NSImage(contentsOf: url)\n else {\n return nil\n }\n\n image.size = self.size\n image.isTemplate = true\n return image\n }\n}\n" }, { "path": "Sources/CodexBar/ProviderRegistry.swift", "content": "import CodexBarCore\nimport Foundation\n\nstruct ProviderSpec {\n let style: IconStyle\n let isEnabled: @MainActor () -> Bool\n let descriptor: ProviderDescriptor\n let makeFetchContext: @MainActor () -> ProviderFetchContext\n}\n\nstruct ProviderRegistry {\n let metadata: [UsageProvider: ProviderMetadata]\n\n static let shared: ProviderRegistry = .init()\n\n init(metadata: [UsageProvider: ProviderMetadata] = ProviderDescriptorRegistry.metadata) {\n self.metadata = metadata\n }\n\n @MainActor\n func specs(\n settings: SettingsStore,\n metadata: [UsageProvider: ProviderMetadata],\n codexFetcher: UsageFetcher,\n claudeFetcher: any ClaudeUsageFetching,\n browserDetection: BrowserDetection) -> [UsageProvider: ProviderSpec]\n {\n var specs: [UsageProvider: ProviderSpec] = [:]\n specs.reserveCapacity(UsageProvider.allCases.count)\n\n for provider in UsageProvider.allCases {\n let descriptor = ProviderDescriptorRegistry.descriptor(for: provider)\n let meta = metadata[provider]!\n let spec = ProviderSpec(\n style: descriptor.branding.iconStyle,\n isEnabled: { settings.isProviderEnabled(provider: provider, metadata: meta) },\n descriptor: descriptor,\n makeFetchContext: {\n let sourceMode = ProviderCatalog.implementation(for: provider)?\n .sourceMode(context: ProviderSourceModeContext(provider: provider, settings: settings))\n ?? .auto\n let snapshot = Self.makeSettingsSnapshot(settings: settings, tokenOverride: nil)\n let env = Self.makeEnvironment(\n base: ProcessInfo.processInfo.environment,\n provider: provider,\n settings: settings,\n tokenOverride: nil)\n let verbose = settings.isVerboseLoggingEnabled\n return ProviderFetchContext(\n runtime: .app,\n sourceMode: sourceMode,\n includeCredits: false,\n webTimeout: 60,\n webDebugDumpHTML: false,\n verbose: verbose,\n env: env,\n settings: snapshot,\n fetcher: codexFetcher,\n claudeFetcher: claudeFetcher,\n browserDetection: browserDetection)\n })\n specs[provider] = spec\n }\n\n return specs\n }\n\n @MainActor\n static func makeSettingsSnapshot(\n settings: SettingsStore,\n tokenOverride: TokenAccountOverride?) -> ProviderSettingsSnapshot\n {\n settings.ensureTokenAccountsLoaded()\n var builder = ProviderSettingsSnapshotBuilder(\n debugMenuEnabled: settings.debugMenuEnabled,\n debugKeepCLISessionsAlive: settings.debugKeepCLISessionsAlive)\n let context = ProviderSettingsSnapshotContext(settings: settings, tokenOverride: tokenOverride)\n for implementation in ProviderCatalog.all {\n if let contribution = implementation.settingsSnapshot(context: context) {\n builder.apply(contribution)\n }\n }\n return builder.build()\n }\n\n @MainActor\n static func makeEnvironment(\n base: [String: String],\n provider: UsageProvider,\n settings: SettingsStore,\n tokenOverride: TokenAccountOverride?) -> [String: String]\n {\n let account = ProviderTokenAccountSelection.selectedAccount(\n provider: provider,\n settings: settings,\n override: tokenOverride)\n var env = ProviderConfigEnvironment.applyAPIKeyOverride(\n base: base,\n provider: provider,\n config: settings.providerConfig(for: provider))\n // If token account is selected, use its token instead of config's apiKey\n if let account, let override = TokenAccountSupportCatalog.envOverride(\n for: provider,\n token: account.token)\n {\n for (key, value) in override {\n env[key] = value\n }\n }\n return env\n }\n}\n" }, { "path": "Sources/CodexBar/ProviderSwitcherButtons.swift", "content": "import AppKit\n\nfinal class PaddedToggleButton: NSButton {\n var contentPadding = NSEdgeInsets(top: 4, left: 7, bottom: 4, right: 7) {\n didSet {\n if oldValue.top != self.contentPadding.top ||\n oldValue.left != self.contentPadding.left ||\n oldValue.bottom != self.contentPadding.bottom ||\n oldValue.right != self.contentPadding.right\n {\n self.invalidateIntrinsicContentSize()\n }\n }\n }\n\n override var intrinsicContentSize: NSSize {\n let size = super.intrinsicContentSize\n return NSSize(\n width: size.width + self.contentPadding.left + self.contentPadding.right,\n height: size.height + self.contentPadding.top + self.contentPadding.bottom)\n }\n}\n\nfinal class InlineIconToggleButton: NSButton {\n private let iconView = NSImageView()\n private let titleField = NSTextField(labelWithString: \"\")\n private let stack = NSStackView()\n private var paddingConstraints: [NSLayoutConstraint] = []\n private var iconSizeConstraints: [NSLayoutConstraint] = []\n private var isConfiguring = false // Batch invalidation during setup\n\n var contentPadding = NSEdgeInsets(top: 4, left: 7, bottom: 4, right: 7) {\n didSet {\n self.paddingConstraints.first { $0.firstAttribute == .top }?.constant = self.contentPadding.top\n self.paddingConstraints.first { $0.firstAttribute == .leading }?.constant = self.contentPadding.left\n self.paddingConstraints.first { $0.firstAttribute == .trailing }?.constant = -self.contentPadding.right\n self.paddingConstraints.first { $0.firstAttribute == .bottom }?.constant = -(self.contentPadding.bottom + 4)\n if !self.isConfiguring { self.invalidateIntrinsicContentSize() }\n }\n }\n\n override var title: String {\n get { \"\" }\n set {\n super.title = \"\"\n super.alternateTitle = \"\"\n super.attributedTitle = NSAttributedString(string: \"\")\n super.attributedAlternateTitle = NSAttributedString(string: \"\")\n self.titleField.stringValue = newValue\n if !self.isConfiguring { self.invalidateIntrinsicContentSize() }\n }\n }\n\n override var image: NSImage? {\n get { nil }\n set {\n super.image = nil\n super.alternateImage = nil\n self.iconView.image = newValue\n if !self.isConfiguring { self.invalidateIntrinsicContentSize() }\n }\n }\n\n func setContentTintColor(_ color: NSColor?) {\n self.iconView.contentTintColor = color\n self.titleField.textColor = color\n }\n\n func setTitleFontSize(_ size: CGFloat) {\n self.titleField.font = NSFont.systemFont(ofSize: size)\n }\n\n func setAllowsTwoLineTitle(_ allow: Bool) {\n let hasWhitespace = self.titleField.stringValue.rangeOfCharacter(from: .whitespacesAndNewlines) != nil\n let shouldWrap = allow && hasWhitespace\n self.titleField.maximumNumberOfLines = shouldWrap ? 2 : 1\n self.titleField.usesSingleLineMode = !shouldWrap\n self.titleField.lineBreakMode = shouldWrap ? .byWordWrapping : .byTruncatingTail\n }\n\n override var intrinsicContentSize: NSSize {\n let size = self.stack.fittingSize\n return NSSize(\n width: size.width + self.contentPadding.left + self.contentPadding.right,\n height: size.height + self.contentPadding.top + self.contentPadding.bottom)\n }\n\n init(title: String, image: NSImage, target: AnyObject?, action: Selector?) {\n super.init(frame: .zero)\n self.target = target\n self.action = action\n self.isConfiguring = true // Batch invalidations during setup\n self.configure()\n self.title = title\n self.image = image\n self.isConfiguring = false\n self.invalidateIntrinsicContentSize() // Single invalidation after setup\n }\n\n @available(*, unavailable)\n required init?(coder: NSCoder) {\n nil\n }\n\n private func configure() {\n self.bezelStyle = .regularSquare\n self.isBordered = false\n self.setButtonType(.toggle)\n self.controlSize = .small\n self.wantsLayer = true\n\n self.iconView.imageScaling = .scaleNone\n self.iconView.translatesAutoresizingMaskIntoConstraints = false\n self.titleField.font = NSFont.systemFont(ofSize: NSFont.smallSystemFontSize)\n self.titleField.alignment = .left\n self.titleField.lineBreakMode = .byTruncatingTail\n self.titleField.setContentCompressionResistancePriority(.defaultLow, for: .horizontal)\n self.titleField.setContentHuggingPriority(.defaultLow, for: .horizontal)\n self.setContentTintColor(NSColor.secondaryLabelColor)\n\n self.stack.orientation = .horizontal\n self.stack.alignment = .centerY\n self.stack.spacing = 1\n self.stack.translatesAutoresizingMaskIntoConstraints = false\n self.stack.addArrangedSubview(self.iconView)\n self.stack.addArrangedSubview(self.titleField)\n self.addSubview(self.stack)\n\n let iconWidth = self.iconView.widthAnchor.constraint(equalToConstant: 16)\n let iconHeight = self.iconView.heightAnchor.constraint(equalToConstant: 16)\n self.iconSizeConstraints = [iconWidth, iconHeight]\n\n let top = self.stack.topAnchor.constraint(\n equalTo: self.topAnchor,\n constant: self.contentPadding.top)\n let leading = self.stack.leadingAnchor.constraint(\n greaterThanOrEqualTo: self.leadingAnchor,\n constant: self.contentPadding.left)\n let trailing = self.stack.trailingAnchor.constraint(\n lessThanOrEqualTo: self.trailingAnchor,\n constant: -self.contentPadding.right)\n let centerX = self.stack.centerXAnchor.constraint(equalTo: self.centerXAnchor)\n centerX.priority = .defaultHigh\n let bottom = self.stack.bottomAnchor.constraint(\n lessThanOrEqualTo: self.bottomAnchor,\n constant: -(self.contentPadding.bottom + 4))\n self.paddingConstraints = [top, leading, trailing, bottom, centerX]\n\n NSLayoutConstraint.activate(self.paddingConstraints + self.iconSizeConstraints)\n }\n}\n\nfinal class StackedToggleButton: NSButton {\n private let iconView = NSImageView()\n private let titleField = NSTextField(labelWithString: \"\")\n private let stack = NSStackView()\n private var paddingConstraints: [NSLayoutConstraint] = []\n private var iconSizeConstraints: [NSLayoutConstraint] = []\n private var isConfiguring = false // Batch invalidation during setup\n\n var contentPadding = NSEdgeInsets(top: 2, left: 4, bottom: 2, right: 4) {\n didSet {\n self.paddingConstraints.first { $0.firstAttribute == .top }?.constant = self.contentPadding.top\n self.paddingConstraints.first { $0.firstAttribute == .leading }?.constant = self.contentPadding.left\n self.paddingConstraints.first { $0.firstAttribute == .trailing }?.constant = -self.contentPadding.right\n self.paddingConstraints.first { $0.firstAttribute == .bottom }?.constant = -self.contentPadding.bottom\n if !self.isConfiguring { self.invalidateIntrinsicContentSize() }\n }\n }\n\n override var title: String {\n get { \"\" }\n set {\n super.title = \"\"\n super.alternateTitle = \"\"\n super.attributedTitle = NSAttributedString(string: \"\")\n super.attributedAlternateTitle = NSAttributedString(string: \"\")\n self.titleField.stringValue = newValue\n if !self.isConfiguring { self.invalidateIntrinsicContentSize() }\n }\n }\n\n override var image: NSImage? {\n get { nil }\n set {\n super.image = nil\n super.alternateImage = nil\n self.iconView.image = newValue\n if !self.isConfiguring { self.invalidateIntrinsicContentSize() }\n }\n }\n\n func setContentTintColor(_ color: NSColor?) {\n self.iconView.contentTintColor = color\n self.titleField.textColor = color\n }\n\n func setTitleFontSize(_ size: CGFloat) {\n self.titleField.font = NSFont.systemFont(ofSize: size)\n }\n\n func setAllowsTwoLineTitle(_ allow: Bool) {\n let hasWhitespace = self.titleField.stringValue.rangeOfCharacter(from: .whitespacesAndNewlines) != nil\n let shouldWrap = allow && hasWhitespace\n self.titleField.maximumNumberOfLines = shouldWrap ? 2 : 1\n self.titleField.usesSingleLineMode = !shouldWrap\n self.titleField.lineBreakMode = shouldWrap ? .byWordWrapping : .byTruncatingTail\n }\n\n override var intrinsicContentSize: NSSize {\n let size = self.stack.fittingSize\n return NSSize(\n width: size.width + self.contentPadding.left + self.contentPadding.right,\n height: size.height + self.contentPadding.top + self.contentPadding.bottom)\n }\n\n init(title: String, image: NSImage, target: AnyObject?, action: Selector?) {\n super.init(frame: .zero)\n self.target = target\n self.action = action\n self.isConfiguring = true // Batch invalidations during setup\n self.configure()\n self.title = title\n self.image = image\n self.isConfiguring = false\n self.invalidateIntrinsicContentSize() // Single invalidation after setup\n }\n\n @available(*, unavailable)\n required init?(coder: NSCoder) {\n nil\n }\n\n private func configure() {\n self.bezelStyle = .regularSquare\n self.isBordered = false\n self.setButtonType(.toggle)\n self.controlSize = .small\n self.wantsLayer = true\n\n self.iconView.imageScaling = .scaleNone\n self.iconView.translatesAutoresizingMaskIntoConstraints = false\n self.titleField.font = NSFont.systemFont(ofSize: NSFont.smallSystemFontSize - 2)\n self.titleField.alignment = .center\n self.titleField.lineBreakMode = .byTruncatingTail\n self.titleField.maximumNumberOfLines = 1\n self.titleField.usesSingleLineMode = true\n self.setContentTintColor(NSColor.secondaryLabelColor)\n\n self.stack.orientation = .vertical\n self.stack.alignment = .centerX\n self.stack.spacing = 0\n self.stack.translatesAutoresizingMaskIntoConstraints = false\n self.stack.addArrangedSubview(self.iconView)\n self.stack.addArrangedSubview(self.titleField)\n self.addSubview(self.stack)\n\n let iconWidth = self.iconView.widthAnchor.constraint(equalToConstant: 16)\n let iconHeight = self.iconView.heightAnchor.constraint(equalToConstant: 16)\n self.iconSizeConstraints = [iconWidth, iconHeight]\n\n // Avoid subpixel centering: pin from the top so the icon sits on whole-point coordinates.\n // Force an even layout width (button width minus padding) so the icon doesn't land on 0.5pt centers.\n // Reserve some bottom space for the \"weekly remaining\" indicator line.\n let top = self.stack.topAnchor.constraint(\n equalTo: self.topAnchor,\n constant: self.contentPadding.top)\n let leading = self.stack.leadingAnchor.constraint(\n equalTo: self.leadingAnchor,\n constant: self.contentPadding.left)\n let trailing = self.stack.trailingAnchor.constraint(\n equalTo: self.trailingAnchor,\n constant: -self.contentPadding.right)\n let bottom = self.stack.bottomAnchor.constraint(\n lessThanOrEqualTo: self.bottomAnchor,\n constant: -(self.contentPadding.bottom + 4))\n self.paddingConstraints = [top, leading, trailing, bottom]\n\n NSLayoutConstraint.activate(self.paddingConstraints + self.iconSizeConstraints)\n }\n}\n" }, { "path": "Sources/CodexBar/ProviderToggleStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nstruct ProviderToggleStore {\n private let userDefaults: UserDefaults\n private let key = \"providerToggles\"\n\n init(userDefaults: UserDefaults = .standard) {\n self.userDefaults = userDefaults\n }\n\n func isEnabled(metadata: ProviderMetadata) -> Bool {\n self.load()[metadata.cliName] ?? metadata.defaultEnabled\n }\n\n func setEnabled(_ enabled: Bool, metadata: ProviderMetadata) {\n var toggles = self.load()\n toggles[metadata.cliName] = enabled\n self.userDefaults.set(toggles, forKey: self.key)\n }\n\n private func load() -> [String: Bool] {\n (self.userDefaults.dictionary(forKey: self.key) as? [String: Bool]) ?? [:]\n }\n\n func purgeLegacyKeys() {\n self.userDefaults.removeObject(forKey: \"showCodexUsage\")\n self.userDefaults.removeObject(forKey: \"showClaudeUsage\")\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Alibaba/AlibabaCodingPlanProviderImplementation.swift", "content": "import AppKit\nimport CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\nimport SwiftUI\n\n@ProviderImplementationRegistration\nstruct AlibabaCodingPlanProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .alibaba\n\n @MainActor\n func presentation(context _: ProviderPresentationContext) -> ProviderPresentation {\n ProviderPresentation { context in\n context.store.sourceLabel(for: context.provider)\n }\n }\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.alibabaCodingPlanAPIToken\n _ = settings.alibabaCodingPlanCookieSource\n _ = settings.alibabaCodingPlanCookieHeader\n _ = settings.alibabaCodingPlanAPIRegion\n }\n\n @MainActor\n func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {\n _ = context\n return .alibaba(context.settings.alibabaCodingPlanSettingsSnapshot())\n }\n\n @MainActor\n func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {\n let binding = Binding(\n get: { context.settings.alibabaCodingPlanAPIRegion.rawValue },\n set: { raw in\n context.settings\n .alibabaCodingPlanAPIRegion = AlibabaCodingPlanAPIRegion(rawValue: raw) ?? .international\n })\n let options = AlibabaCodingPlanAPIRegion.allCases.map {\n ProviderSettingsPickerOption(id: $0.rawValue, title: $0.displayName)\n }\n\n let cookieBinding = Binding(\n get: { context.settings.alibabaCodingPlanCookieSource.rawValue },\n set: { raw in\n context.settings.alibabaCodingPlanCookieSource = ProviderCookieSource(rawValue: raw) ?? .auto\n })\n let cookieOptions = ProviderCookieSourceUI.options(\n allowsOff: false,\n keychainDisabled: context.settings.debugDisableKeychainAccess)\n let cookieSubtitle: () -> String? = {\n ProviderCookieSourceUI.subtitle(\n source: context.settings.alibabaCodingPlanCookieSource,\n keychainDisabled: context.settings.debugDisableKeychainAccess,\n auto: \"Automatic imports browser cookies from Model Studio/Bailian.\",\n manual: \"Paste a Cookie header from modelstudio.console.alibabacloud.com.\",\n off: \"Alibaba cookies are disabled.\")\n }\n\n return [\n ProviderSettingsPickerDescriptor(\n id: \"alibaba-coding-plan-cookie-source\",\n title: \"Cookie source\",\n subtitle: \"Automatic imports browser cookies from Model Studio/Bailian.\",\n dynamicSubtitle: cookieSubtitle,\n binding: cookieBinding,\n options: cookieOptions,\n isVisible: nil,\n onChange: nil,\n trailingText: {\n guard let entry = CookieHeaderCache.load(provider: .alibaba) else { return nil }\n let when = entry.storedAt.relativeDescription()\n return \"Cached: \\(entry.sourceLabel) • \\(when)\"\n }),\n ProviderSettingsPickerDescriptor(\n id: \"alibaba-coding-plan-region\",\n title: \"Gateway region\",\n subtitle: \"Use international or China mainland console gateways for quota fetches.\",\n binding: binding,\n options: options,\n isVisible: nil,\n onChange: nil),\n ]\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n [\n ProviderSettingsFieldDescriptor(\n id: \"alibaba-coding-plan-api-key\",\n title: \"API key\",\n subtitle: \"Stored in ~/.codexbar/config.json. Paste your Coding Plan API key from Model Studio.\",\n kind: .secure,\n placeholder: \"cpk-...\",\n binding: context.stringBinding(\\.alibabaCodingPlanAPIToken),\n actions: [\n ProviderSettingsActionDescriptor(\n id: \"alibaba-coding-plan-open-dashboard\",\n title: \"Open Coding Plan\",\n style: .link,\n isVisible: nil,\n perform: {\n NSWorkspace.shared.open(context.settings.alibabaCodingPlanAPIRegion.dashboardURL)\n }),\n ],\n isVisible: nil,\n onActivate: { context.settings.ensureAlibabaCodingPlanAPITokenLoaded() }),\n ProviderSettingsFieldDescriptor(\n id: \"alibaba-coding-plan-cookie\",\n title: \"Cookie header\",\n subtitle: \"\",\n kind: .secure,\n placeholder: \"Cookie: ...\",\n binding: context.stringBinding(\\.alibabaCodingPlanCookieHeader),\n actions: [\n ProviderSettingsActionDescriptor(\n id: \"alibaba-coding-plan-open-dashboard-cookie\",\n title: \"Open Coding Plan\",\n style: .link,\n isVisible: nil,\n perform: {\n NSWorkspace.shared.open(context.settings.alibabaCodingPlanAPIRegion.dashboardURL)\n }),\n ],\n isVisible: {\n context.settings.alibabaCodingPlanCookieSource == .manual\n },\n onActivate: nil),\n ]\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Alibaba/AlibabaCodingPlanSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n private static let alibabaAutoEnableAppliedKey = \"alibabaCodingPlanAutoEnableApplied\"\n\n var alibabaCodingPlanAPIRegion: AlibabaCodingPlanAPIRegion {\n get {\n let raw = self.configSnapshot.providerConfig(for: .alibaba)?.region\n return AlibabaCodingPlanAPIRegion(rawValue: raw ?? \"\") ?? .international\n }\n set {\n self.updateProviderConfig(provider: .alibaba) { entry in\n entry.region = newValue.rawValue\n }\n }\n }\n\n var alibabaCodingPlanCookieHeader: String {\n get { self.configSnapshot.providerConfig(for: .alibaba)?.sanitizedCookieHeader ?? \"\" }\n set {\n self.updateProviderConfig(provider: .alibaba) { entry in\n entry.cookieHeader = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .alibaba, field: \"cookieHeader\", value: newValue)\n }\n }\n\n var alibabaCodingPlanCookieSource: ProviderCookieSource {\n get { self.resolvedCookieSource(provider: .alibaba, fallback: .auto) }\n set {\n self.updateProviderConfig(provider: .alibaba) { entry in\n entry.cookieSource = newValue\n }\n self.logProviderModeChange(provider: .alibaba, field: \"cookieSource\", value: newValue.rawValue)\n }\n }\n\n var alibabaCodingPlanAPIToken: String {\n get { self.configSnapshot.providerConfig(for: .alibaba)?.sanitizedAPIKey ?? \"\" }\n set {\n self.updateProviderConfig(provider: .alibaba) { entry in\n entry.apiKey = self.normalizedConfigValue(newValue)\n }\n let hasToken = !newValue.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty\n if hasToken,\n let metadata = ProviderDescriptorRegistry.metadata[.alibaba],\n !self.isProviderEnabled(provider: .alibaba, metadata: metadata)\n {\n self.setProviderEnabled(provider: .alibaba, metadata: metadata, enabled: true)\n }\n self.logSecretUpdate(provider: .alibaba, field: \"apiKey\", value: newValue)\n }\n }\n\n func ensureAlibabaCodingPlanAPITokenLoaded() {}\n\n func ensureAlibabaProviderAutoEnabledIfNeeded(\n environment: [String: String] = ProcessInfo.processInfo.environment)\n {\n guard self.userDefaults.bool(forKey: Self.alibabaAutoEnableAppliedKey) == false else { return }\n\n let hasConfigToken = self.configSnapshot.providerConfig(for: .alibaba)?.sanitizedAPIKey != nil\n let hasEnvironmentToken = AlibabaCodingPlanSettingsReader.apiToken(environment: environment) != nil\n guard hasConfigToken || hasEnvironmentToken else { return }\n\n if let metadata = ProviderDescriptorRegistry.metadata[.alibaba],\n !self.isProviderEnabled(provider: .alibaba, metadata: metadata)\n {\n self.setProviderEnabled(provider: .alibaba, metadata: metadata, enabled: true)\n }\n\n self.userDefaults.set(true, forKey: Self.alibabaAutoEnableAppliedKey)\n }\n}\n\nextension SettingsStore {\n func alibabaCodingPlanSettingsSnapshot() -> ProviderSettingsSnapshot.AlibabaCodingPlanProviderSettings {\n ProviderSettingsSnapshot.AlibabaCodingPlanProviderSettings(\n cookieSource: self.alibabaCodingPlanCookieSource,\n manualCookieHeader: self.alibabaCodingPlanCookieHeader,\n apiRegion: self.alibabaCodingPlanAPIRegion)\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Amp/AmpProviderImplementation.swift", "content": "import AppKit\nimport CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\nimport SwiftUI\n\n@ProviderImplementationRegistration\nstruct AmpProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .amp\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.ampCookieSource\n _ = settings.ampCookieHeader\n }\n\n @MainActor\n func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {\n .amp(context.settings.ampSettingsSnapshot(tokenOverride: context.tokenOverride))\n }\n\n @MainActor\n func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {\n let cookieBinding = Binding(\n get: { context.settings.ampCookieSource.rawValue },\n set: { raw in\n context.settings.ampCookieSource = ProviderCookieSource(rawValue: raw) ?? .auto\n })\n let cookieOptions = ProviderCookieSourceUI.options(\n allowsOff: false,\n keychainDisabled: context.settings.debugDisableKeychainAccess)\n\n let cookieSubtitle: () -> String? = {\n ProviderCookieSourceUI.subtitle(\n source: context.settings.ampCookieSource,\n keychainDisabled: context.settings.debugDisableKeychainAccess,\n auto: \"Automatic imports browser cookies.\",\n manual: \"Paste a Cookie header or cURL capture from Amp settings.\",\n off: \"Amp cookies are disabled.\")\n }\n\n return [\n ProviderSettingsPickerDescriptor(\n id: \"amp-cookie-source\",\n title: \"Cookie source\",\n subtitle: \"Automatic imports browser cookies.\",\n dynamicSubtitle: cookieSubtitle,\n binding: cookieBinding,\n options: cookieOptions,\n isVisible: nil,\n onChange: nil),\n ]\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n [\n ProviderSettingsFieldDescriptor(\n id: \"amp-cookie\",\n title: \"\",\n subtitle: \"\",\n kind: .secure,\n placeholder: \"Cookie: …\",\n binding: context.stringBinding(\\.ampCookieHeader),\n actions: [\n ProviderSettingsActionDescriptor(\n id: \"amp-open-settings\",\n title: \"Open Amp Settings\",\n style: .link,\n isVisible: nil,\n perform: {\n if let url = URL(string: \"https://ampcode.com/settings\") {\n NSWorkspace.shared.open(url)\n }\n }),\n ],\n isVisible: { context.settings.ampCookieSource == .manual },\n onActivate: { context.settings.ensureAmpCookieLoaded() }),\n ]\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Amp/AmpSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var ampCookieHeader: String {\n get { self.configSnapshot.providerConfig(for: .amp)?.sanitizedCookieHeader ?? \"\" }\n set {\n self.updateProviderConfig(provider: .amp) { entry in\n entry.cookieHeader = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .amp, field: \"cookieHeader\", value: newValue)\n }\n }\n\n var ampCookieSource: ProviderCookieSource {\n get { self.resolvedCookieSource(provider: .amp, fallback: .auto) }\n set {\n self.updateProviderConfig(provider: .amp) { entry in\n entry.cookieSource = newValue\n }\n self.logProviderModeChange(provider: .amp, field: \"cookieSource\", value: newValue.rawValue)\n }\n }\n\n func ensureAmpCookieLoaded() {}\n}\n\nextension SettingsStore {\n func ampSettingsSnapshot(tokenOverride: TokenAccountOverride?) -> ProviderSettingsSnapshot.AmpProviderSettings {\n ProviderSettingsSnapshot.AmpProviderSettings(\n cookieSource: self.ampSnapshotCookieSource(tokenOverride: tokenOverride),\n manualCookieHeader: self.ampSnapshotCookieHeader(tokenOverride: tokenOverride))\n }\n\n private func ampSnapshotCookieHeader(tokenOverride: TokenAccountOverride?) -> String {\n let fallback = self.ampCookieHeader\n guard let support = TokenAccountSupportCatalog.support(for: .amp),\n case .cookieHeader = support.injection\n else {\n return fallback\n }\n guard let account = ProviderTokenAccountSelection.selectedAccount(\n provider: .amp,\n settings: self,\n override: tokenOverride)\n else {\n return fallback\n }\n return TokenAccountSupportCatalog.normalizedCookieHeader(account.token, support: support)\n }\n\n private func ampSnapshotCookieSource(tokenOverride: TokenAccountOverride?) -> ProviderCookieSource {\n let fallback = self.ampCookieSource\n guard let support = TokenAccountSupportCatalog.support(for: .amp),\n support.requiresManualCookieSource\n else {\n return fallback\n }\n if self.tokenAccounts(for: .amp).isEmpty { return fallback }\n return .manual\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Antigravity/AntigravityLoginFlow.swift", "content": "import CodexBarCore\n\n@MainActor\nextension StatusItemController {\n func runAntigravityLoginFlow() async {\n self.loginPhase = .idle\n self.presentLoginAlert(\n title: \"Antigravity login is managed in the app\",\n message: \"Open Antigravity to sign in, then refresh CodexBar.\")\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Antigravity/AntigravityProviderImplementation.swift", "content": "import CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\n\n@ProviderImplementationRegistration\nstruct AntigravityProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .antigravity\n\n func detectVersion(context _: ProviderVersionContext) async -> String? {\n await AntigravityStatusProbe.detectVersion()\n }\n\n @MainActor\n func runLoginFlow(context: ProviderLoginContext) async -> Bool {\n await context.controller.runAntigravityLoginFlow()\n return false\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Augment/AugmentProviderImplementation.swift", "content": "import AppKit\nimport CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\nimport SwiftUI\n\n@ProviderImplementationRegistration\nstruct AugmentProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .augment\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.augmentCookieSource\n _ = settings.augmentCookieHeader\n }\n\n @MainActor\n func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {\n .augment(context.settings.augmentSettingsSnapshot(tokenOverride: context.tokenOverride))\n }\n\n @MainActor\n func tokenAccountsVisibility(context: ProviderSettingsContext, support: TokenAccountSupport) -> Bool {\n guard support.requiresManualCookieSource else { return true }\n if !context.settings.tokenAccounts(for: context.provider).isEmpty { return true }\n return context.settings.augmentCookieSource == .manual\n }\n\n @MainActor\n func applyTokenAccountCookieSource(settings: SettingsStore) {\n if settings.augmentCookieSource != .manual {\n settings.augmentCookieSource = .manual\n }\n }\n\n func makeRuntime() -> (any ProviderRuntime)? {\n AugmentProviderRuntime()\n }\n\n @MainActor\n func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {\n let cookieBinding = Binding(\n get: { context.settings.augmentCookieSource.rawValue },\n set: { raw in\n context.settings.augmentCookieSource = ProviderCookieSource(rawValue: raw) ?? .auto\n })\n let cookieOptions = ProviderCookieSourceUI.options(\n allowsOff: false,\n keychainDisabled: context.settings.debugDisableKeychainAccess)\n\n let cookieSubtitle: () -> String? = {\n ProviderCookieSourceUI.subtitle(\n source: context.settings.augmentCookieSource,\n keychainDisabled: context.settings.debugDisableKeychainAccess,\n auto: \"Automatic imports browser cookies.\",\n manual: \"Paste a Cookie header or cURL capture from the Augment dashboard.\",\n off: \"Augment cookies are disabled.\")\n }\n\n return [\n ProviderSettingsPickerDescriptor(\n id: \"augment-cookie-source\",\n title: \"Cookie source\",\n subtitle: \"Automatic imports browser cookies.\",\n dynamicSubtitle: cookieSubtitle,\n binding: cookieBinding,\n options: cookieOptions,\n isVisible: nil,\n onChange: nil,\n trailingText: {\n guard let entry = CookieHeaderCache.load(provider: .augment) else { return nil }\n let when = entry.storedAt.relativeDescription()\n return \"Cached: \\(entry.sourceLabel) • \\(when)\"\n }),\n ]\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n _ = context\n return []\n }\n\n @MainActor\n func appendActionMenuEntries(context: ProviderMenuActionContext, entries: inout [ProviderMenuEntry]) {\n entries.append(.action(\"Refresh Session\", .refreshAugmentSession))\n\n if let error = context.store.error(for: .augment) {\n if error.contains(\"session has expired\") ||\n error.contains(\"No Augment session cookie found\")\n {\n entries.append(.action(\n \"Open Augment (Log Out & Back In)\",\n .loginToProvider(url: \"https://app.augmentcode.com\")))\n }\n }\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Augment/AugmentProviderRuntime.swift", "content": "import CodexBarCore\nimport Foundation\n\n@MainActor\nfinal class AugmentProviderRuntime: ProviderRuntime {\n let id: UsageProvider = .augment\n private var keepalive: AugmentSessionKeepalive?\n\n func start(context: ProviderRuntimeContext) {\n self.updateKeepalive(context: context)\n }\n\n func stop(context: ProviderRuntimeContext) {\n self.stopKeepalive(context: context, reason: \"provider disabled\")\n }\n\n func settingsDidChange(context: ProviderRuntimeContext) {\n self.updateKeepalive(context: context)\n }\n\n func providerDidFail(context: ProviderRuntimeContext, provider: UsageProvider, error: Error) {\n guard provider == .augment else { return }\n let message = error.localizedDescription\n guard message.contains(\"session expired\") else { return }\n context.store.augmentLogger.warning(\"Augment session expired; triggering recovery\")\n Task { [weak self] in\n guard let self else { return }\n await self.forceRefresh(context: context)\n }\n }\n\n func perform(action: ProviderRuntimeAction, context: ProviderRuntimeContext) async {\n switch action {\n case .forceSessionRefresh:\n await self.forceRefresh(context: context)\n case .openAIWebAccessToggled:\n break\n }\n }\n\n private func updateKeepalive(context: ProviderRuntimeContext) {\n #if os(macOS)\n let shouldRun = context.store.isEnabled(.augment)\n let isRunning = self.keepalive != nil\n\n if shouldRun, !isRunning {\n self.startKeepalive(context: context)\n } else if !shouldRun, isRunning {\n self.stopKeepalive(context: context, reason: \"provider disabled\")\n }\n #endif\n }\n\n private func startKeepalive(context: ProviderRuntimeContext) {\n #if os(macOS)\n context.store.augmentLogger.info(\n \"Augment keepalive check\",\n metadata: [\n \"enabled\": context.store.isEnabled(.augment) ? \"1\" : \"0\",\n \"available\": context.store.isProviderAvailable(.augment) ? \"1\" : \"0\",\n ])\n\n guard context.store.isEnabled(.augment) else {\n context.store.augmentLogger.warning(\"Augment keepalive not started (provider disabled)\")\n return\n }\n\n let logger: (String) -> Void = { [augmentLogger = context.store.augmentLogger] message in\n augmentLogger.verbose(message)\n }\n\n let onSessionRecovered: () async -> Void = { [weak store = context.store] in\n guard let store else { return }\n store.augmentLogger.info(\"Augment session recovered; refreshing usage\")\n await store.refreshProvider(.augment)\n }\n\n self.keepalive = AugmentSessionKeepalive(logger: logger, onSessionRecovered: onSessionRecovered)\n self.keepalive?.start()\n context.store.augmentLogger.info(\"Augment keepalive started\")\n #endif\n }\n\n private func stopKeepalive(context: ProviderRuntimeContext, reason: String) {\n #if os(macOS)\n self.keepalive?.stop()\n self.keepalive = nil\n context.store.augmentLogger.info(\"Augment keepalive stopped (\\(reason))\")\n #endif\n }\n\n private func forceRefresh(context: ProviderRuntimeContext) async {\n #if os(macOS)\n context.store.augmentLogger.info(\"Augment force refresh requested\")\n guard let keepalive = self.keepalive else {\n context.store.augmentLogger.warning(\"Augment keepalive not running; starting\")\n self.startKeepalive(context: context)\n try? await Task.sleep(for: .seconds(1))\n guard let keepalive = self.keepalive else {\n context.store.augmentLogger.error(\"Augment keepalive failed to start\")\n return\n }\n await keepalive.forceRefresh()\n return\n }\n\n await keepalive.forceRefresh()\n context.store.augmentLogger.info(\"Refreshing Augment usage after session refresh\")\n await context.store.refreshProvider(.augment)\n #endif\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Augment/AugmentSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var augmentCookieHeader: String {\n get { self.configSnapshot.providerConfig(for: .augment)?.sanitizedCookieHeader ?? \"\" }\n set {\n self.updateProviderConfig(provider: .augment) { entry in\n entry.cookieHeader = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .augment, field: \"cookieHeader\", value: newValue)\n }\n }\n\n var augmentCookieSource: ProviderCookieSource {\n get { self.resolvedCookieSource(provider: .augment, fallback: .auto) }\n set {\n self.updateProviderConfig(provider: .augment) { entry in\n entry.cookieSource = newValue\n }\n self.logProviderModeChange(provider: .augment, field: \"cookieSource\", value: newValue.rawValue)\n }\n }\n\n func ensureAugmentCookieLoaded() {}\n}\n\nextension SettingsStore {\n func augmentSettingsSnapshot(tokenOverride: TokenAccountOverride?) -> ProviderSettingsSnapshot\n .AugmentProviderSettings {\n ProviderSettingsSnapshot.AugmentProviderSettings(\n cookieSource: self.augmentSnapshotCookieSource(tokenOverride: tokenOverride),\n manualCookieHeader: self.augmentSnapshotCookieHeader(tokenOverride: tokenOverride))\n }\n\n private func augmentSnapshotCookieHeader(tokenOverride: TokenAccountOverride?) -> String {\n let fallback = self.augmentCookieHeader\n guard let support = TokenAccountSupportCatalog.support(for: .augment),\n case .cookieHeader = support.injection\n else {\n return fallback\n }\n guard let account = ProviderTokenAccountSelection.selectedAccount(\n provider: .augment,\n settings: self,\n override: tokenOverride)\n else {\n return fallback\n }\n return TokenAccountSupportCatalog.normalizedCookieHeader(account.token, support: support)\n }\n\n private func augmentSnapshotCookieSource(tokenOverride: TokenAccountOverride?) -> ProviderCookieSource {\n let fallback = self.augmentCookieSource\n guard let support = TokenAccountSupportCatalog.support(for: .augment),\n support.requiresManualCookieSource\n else {\n return fallback\n }\n if self.tokenAccounts(for: .augment).isEmpty { return fallback }\n return .manual\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Claude/ClaudeLoginFlow.swift", "content": "import CodexBarCore\n\n@MainActor\nextension StatusItemController {\n func runClaudeLoginFlow() async {\n let phaseHandler: @Sendable (ClaudeLoginRunner.Phase) -> Void = { [weak self] phase in\n Task { @MainActor in\n switch phase {\n case .requesting: self?.loginPhase = .requesting\n case .waitingBrowser: self?.loginPhase = .waitingBrowser\n }\n }\n }\n let result = await ClaudeLoginRunner.run(timeout: 120, onPhaseChange: phaseHandler)\n guard !Task.isCancelled else { return }\n self.loginPhase = .idle\n self.presentClaudeLoginResult(result)\n let outcome = self.describe(result.outcome)\n let length = result.output.count\n self.loginLogger.info(\"Claude login\", metadata: [\"outcome\": outcome, \"length\": \"\\(length)\"])\n if case .success = result.outcome {\n self.postLoginNotification(for: .claude)\n }\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Claude/ClaudeProviderImplementation.swift", "content": "import CodexBarCore\nimport CodexBarMacroSupport\nimport SwiftUI\n\n@ProviderImplementationRegistration\nstruct ClaudeProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .claude\n let supportsLoginFlow: Bool = true\n\n @MainActor\n func presentation(context _: ProviderPresentationContext) -> ProviderPresentation {\n ProviderPresentation { context in\n var versionText = context.store.version(for: context.provider) ?? \"not detected\"\n if let parenRange = versionText.range(of: \"(\") {\n versionText = versionText[.. ProviderSettingsSnapshotContribution? {\n .claude(context.settings.claudeSettingsSnapshot(tokenOverride: context.tokenOverride))\n }\n\n @MainActor\n func tokenAccountsVisibility(context: ProviderSettingsContext, support: TokenAccountSupport) -> Bool {\n guard support.requiresManualCookieSource else { return true }\n if !context.settings.tokenAccounts(for: context.provider).isEmpty { return true }\n return context.settings.claudeCookieSource == .manual\n }\n\n @MainActor\n func applyTokenAccountCookieSource(settings: SettingsStore) {\n if settings.claudeCookieSource != .manual {\n settings.claudeCookieSource = .manual\n }\n }\n\n @MainActor\n func defaultSourceLabel(context: ProviderSourceLabelContext) -> String? {\n context.settings.claudeUsageDataSource.rawValue\n }\n\n @MainActor\n func sourceMode(context: ProviderSourceModeContext) -> ProviderSourceMode {\n switch context.settings.claudeUsageDataSource {\n case .auto: .auto\n case .oauth: .oauth\n case .web: .web\n case .cli: .cli\n }\n }\n\n @MainActor\n func settingsToggles(context: ProviderSettingsContext) -> [ProviderSettingsToggleDescriptor] {\n let subtitle = if context.settings.debugDisableKeychainAccess {\n \"Inactive while \\\"Disable Keychain access\\\" is enabled in Advanced.\"\n } else {\n \"Use /usr/bin/security to read Claude credentials and avoid CodexBar keychain prompts.\"\n }\n\n let promptFreeBinding = Binding(\n get: { context.settings.claudeOAuthPromptFreeCredentialsEnabled },\n set: { enabled in\n guard !context.settings.debugDisableKeychainAccess else { return }\n context.settings.claudeOAuthPromptFreeCredentialsEnabled = enabled\n })\n\n return [\n ProviderSettingsToggleDescriptor(\n id: \"claude-oauth-prompt-free-credentials\",\n title: \"Avoid Keychain prompts (experimental)\",\n subtitle: subtitle,\n binding: promptFreeBinding,\n statusText: nil,\n actions: [],\n isVisible: nil,\n onChange: nil,\n onAppDidBecomeActive: nil,\n onAppearWhenEnabled: nil),\n ]\n }\n\n @MainActor\n func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {\n let usageBinding = Binding(\n get: { context.settings.claudeUsageDataSource.rawValue },\n set: { raw in\n context.settings.claudeUsageDataSource = ClaudeUsageDataSource(rawValue: raw) ?? .auto\n })\n let cookieBinding = Binding(\n get: { context.settings.claudeCookieSource.rawValue },\n set: { raw in\n context.settings.claudeCookieSource = ProviderCookieSource(rawValue: raw) ?? .auto\n })\n let keychainPromptPolicyBinding = Binding(\n get: { context.settings.claudeOAuthKeychainPromptMode.rawValue },\n set: { raw in\n context.settings.claudeOAuthKeychainPromptMode = ClaudeOAuthKeychainPromptMode(rawValue: raw)\n ?? .onlyOnUserAction\n })\n\n let usageOptions = ClaudeUsageDataSource.allCases.map {\n ProviderSettingsPickerOption(id: $0.rawValue, title: $0.displayName)\n }\n let cookieOptions = ProviderCookieSourceUI.options(\n allowsOff: false,\n keychainDisabled: context.settings.debugDisableKeychainAccess)\n let keychainPromptPolicyOptions: [ProviderSettingsPickerOption] = [\n ProviderSettingsPickerOption(\n id: ClaudeOAuthKeychainPromptMode.never.rawValue,\n title: \"Never prompt\"),\n ProviderSettingsPickerOption(\n id: ClaudeOAuthKeychainPromptMode.onlyOnUserAction.rawValue,\n title: \"Only on user action\"),\n ProviderSettingsPickerOption(\n id: ClaudeOAuthKeychainPromptMode.always.rawValue,\n title: \"Always allow prompts\"),\n ]\n let cookieSubtitle: () -> String? = {\n ProviderCookieSourceUI.subtitle(\n source: context.settings.claudeCookieSource,\n keychainDisabled: context.settings.debugDisableKeychainAccess,\n auto: \"Automatic imports browser cookies for the web API.\",\n manual: \"Paste a Cookie header from a claude.ai request.\",\n off: \"Claude cookies are disabled.\")\n }\n let keychainPromptPolicySubtitle: () -> String? = {\n if context.settings.debugDisableKeychainAccess {\n return \"Global Keychain access is disabled in Advanced, so this setting is currently inactive.\"\n }\n return \"Controls Claude OAuth Keychain prompts when experimental reader mode is off. Choosing \" +\n \"\\\"Never prompt\\\" can make OAuth unavailable; use Web/CLI when needed.\"\n }\n\n return [\n ProviderSettingsPickerDescriptor(\n id: \"claude-usage-source\",\n title: \"Usage source\",\n subtitle: \"Auto falls back to the next source if the preferred one fails.\",\n binding: usageBinding,\n options: usageOptions,\n isVisible: nil,\n onChange: nil,\n trailingText: {\n guard context.settings.claudeUsageDataSource == .auto else { return nil }\n let label = context.store.sourceLabel(for: .claude)\n return label == \"auto\" ? nil : label\n }),\n ProviderSettingsPickerDescriptor(\n id: \"claude-keychain-prompt-policy\",\n title: \"Keychain prompt policy\",\n subtitle: \"Applies only to the Security.framework OAuth keychain reader.\",\n dynamicSubtitle: keychainPromptPolicySubtitle,\n binding: keychainPromptPolicyBinding,\n options: keychainPromptPolicyOptions,\n isVisible: { context.settings.claudeOAuthKeychainReadStrategy == .securityFramework },\n isEnabled: { !context.settings.debugDisableKeychainAccess },\n onChange: nil),\n ProviderSettingsPickerDescriptor(\n id: \"claude-cookie-source\",\n title: \"Claude cookies\",\n subtitle: \"Automatic imports browser cookies for the web API.\",\n dynamicSubtitle: cookieSubtitle,\n binding: cookieBinding,\n options: cookieOptions,\n isVisible: nil,\n onChange: nil,\n trailingText: {\n guard let entry = CookieHeaderCache.load(provider: .claude) else { return nil }\n let when = entry.storedAt.relativeDescription()\n return \"Cached: \\(entry.sourceLabel) • \\(when)\"\n }),\n ]\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n _ = context\n return []\n }\n\n @MainActor\n func runLoginFlow(context: ProviderLoginContext) async -> Bool {\n await context.controller.runClaudeLoginFlow()\n return true\n }\n\n @MainActor\n func appendUsageMenuEntries(context: ProviderMenuUsageContext, entries: inout [ProviderMenuEntry]) {\n if context.snapshot?.secondary == nil {\n entries.append(.text(\"Weekly usage unavailable for this account.\", .secondary))\n }\n\n if let cost = context.snapshot?.providerCost,\n context.settings.showOptionalCreditsAndExtraUsage,\n cost.currencyCode != \"Quota\"\n {\n let used = UsageFormatter.currencyString(cost.used, currencyCode: cost.currencyCode)\n let limit = UsageFormatter.currencyString(cost.limit, currencyCode: cost.currencyCode)\n entries.append(.text(\"Extra usage: \\(used) / \\(limit)\", .primary))\n }\n }\n\n @MainActor\n func loginMenuAction(context: ProviderMenuLoginContext)\n -> (label: String, action: MenuDescriptor.MenuAction)?\n {\n guard self.shouldOpenTerminalForOAuthError(store: context.store) else { return nil }\n return (\"Open Terminal\", .openTerminal(command: \"claude\"))\n }\n\n @MainActor\n private func shouldOpenTerminalForOAuthError(store: UsageStore) -> Bool {\n guard store.error(for: .claude) != nil else { return false }\n let attempts = store.fetchAttempts(for: .claude)\n if attempts.contains(where: { $0.kind == .oauth && ($0.errorDescription?.isEmpty == false) }) {\n return true\n }\n if let error = store.error(for: .claude)?.lowercased(), error.contains(\"oauth\") {\n return true\n }\n return false\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Claude/ClaudeSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var claudeUsageDataSource: ClaudeUsageDataSource {\n get {\n let source = self.configSnapshot.providerConfig(for: .claude)?.source\n return Self.claudeUsageDataSource(from: source)\n }\n set {\n let source: ProviderSourceMode? = switch newValue {\n case .auto: .auto\n case .oauth: .oauth\n case .web: .web\n case .cli: .cli\n }\n self.updateProviderConfig(provider: .claude) { entry in\n entry.source = source\n }\n self.logProviderModeChange(provider: .claude, field: \"usageSource\", value: newValue.rawValue)\n if newValue != .cli {\n self.claudeWebExtrasEnabled = false\n }\n }\n }\n\n var claudeCookieHeader: String {\n get { self.configSnapshot.providerConfig(for: .claude)?.sanitizedCookieHeader ?? \"\" }\n set {\n self.updateProviderConfig(provider: .claude) { entry in\n entry.cookieHeader = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .claude, field: \"cookieHeader\", value: newValue)\n }\n }\n\n var claudeCookieSource: ProviderCookieSource {\n get { self.resolvedCookieSource(provider: .claude, fallback: .auto) }\n set {\n self.updateProviderConfig(provider: .claude) { entry in\n entry.cookieSource = newValue\n }\n self.logProviderModeChange(provider: .claude, field: \"cookieSource\", value: newValue.rawValue)\n }\n }\n\n func ensureClaudeCookieLoaded() {}\n}\n\nextension SettingsStore {\n func claudeSettingsSnapshot(tokenOverride: TokenAccountOverride?) -> ProviderSettingsSnapshot\n .ClaudeProviderSettings {\n let account = self.selectedClaudeTokenAccount(tokenOverride: tokenOverride)\n let routing = self.claudeCredentialRouting(account: account)\n return ProviderSettingsSnapshot.ClaudeProviderSettings(\n usageDataSource: self.claudeUsageDataSource,\n webExtrasEnabled: self.claudeWebExtrasEnabled,\n cookieSource: self.claudeSnapshotCookieSource(tokenOverride: tokenOverride, routing: routing),\n manualCookieHeader: self.claudeSnapshotCookieHeader(\n routing: routing,\n hasSelectedAccount: account != nil))\n }\n\n private static func claudeUsageDataSource(from source: ProviderSourceMode?) -> ClaudeUsageDataSource {\n guard let source else { return .auto }\n switch source {\n case .auto, .api:\n return .auto\n case .web:\n return .web\n case .cli:\n return .cli\n case .oauth:\n return .oauth\n }\n }\n\n private func claudeSnapshotCookieHeader(\n routing: ClaudeCredentialRouting,\n hasSelectedAccount: Bool) -> String\n {\n switch routing {\n case .none:\n hasSelectedAccount ? \"\" : self.claudeCookieHeader\n case .oauth:\n \"\"\n case let .webCookie(header):\n header\n }\n }\n\n private func claudeSnapshotCookieSource(\n tokenOverride: TokenAccountOverride?,\n routing: ClaudeCredentialRouting) -> ProviderCookieSource\n {\n let fallback = self.claudeCookieSource\n guard let support = TokenAccountSupportCatalog.support(for: .claude),\n support.requiresManualCookieSource\n else {\n return fallback\n }\n if routing.isOAuth {\n return .off\n }\n if self.tokenAccounts(for: .claude).isEmpty { return fallback }\n return .manual\n }\n\n private func claudeCredentialRouting(account: ProviderTokenAccount?) -> ClaudeCredentialRouting {\n let manualCookieHeader = account == nil ? self.claudeCookieHeader : nil\n return ClaudeCredentialRouting.resolve(\n tokenAccountToken: account?.token,\n manualCookieHeader: manualCookieHeader)\n }\n\n private func selectedClaudeTokenAccount(tokenOverride: TokenAccountOverride?) -> ProviderTokenAccount? {\n ProviderTokenAccountSelection.selectedAccount(\n provider: .claude,\n settings: self,\n override: tokenOverride)\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Codex/CodexLoginFlow.swift", "content": "import CodexBarCore\n\n@MainActor\nextension StatusItemController {\n func runCodexLoginFlow() async {\n let result = await CodexLoginRunner.run(timeout: 120)\n guard !Task.isCancelled else { return }\n self.loginPhase = .idle\n self.presentCodexLoginResult(result)\n let outcome = self.describe(result.outcome)\n let length = result.output.count\n self.loginLogger.info(\"Codex login\", metadata: [\"outcome\": outcome, \"length\": \"\\(length)\"])\n if case .success = result.outcome {\n self.postLoginNotification(for: .codex)\n }\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Codex/CodexProviderImplementation.swift", "content": "import CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\nimport SwiftUI\n\n@ProviderImplementationRegistration\nstruct CodexProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .codex\n let supportsLoginFlow: Bool = true\n\n @MainActor\n func presentation(context _: ProviderPresentationContext) -> ProviderPresentation {\n ProviderPresentation { context in\n context.store.version(for: context.provider) ?? \"not detected\"\n }\n }\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.codexUsageDataSource\n _ = settings.codexCookieSource\n _ = settings.codexCookieHeader\n }\n\n @MainActor\n func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {\n .codex(context.settings.codexSettingsSnapshot(tokenOverride: context.tokenOverride))\n }\n\n @MainActor\n func defaultSourceLabel(context: ProviderSourceLabelContext) -> String? {\n context.settings.codexUsageDataSource.rawValue\n }\n\n @MainActor\n func decorateSourceLabel(context: ProviderSourceLabelContext, baseLabel: String) -> String {\n if context.settings.codexCookieSource.isEnabled,\n context.store.openAIDashboard != nil,\n !context.store.openAIDashboardRequiresLogin,\n !baseLabel.contains(\"openai-web\")\n {\n return \"\\(baseLabel) + openai-web\"\n }\n return baseLabel\n }\n\n @MainActor\n func sourceMode(context: ProviderSourceModeContext) -> ProviderSourceMode {\n switch context.settings.codexUsageDataSource {\n case .auto: .auto\n case .oauth: .oauth\n case .cli: .cli\n }\n }\n\n func makeRuntime() -> (any ProviderRuntime)? {\n CodexProviderRuntime()\n }\n\n @MainActor\n func settingsToggles(context: ProviderSettingsContext) -> [ProviderSettingsToggleDescriptor] {\n let extrasBinding = Binding(\n get: { context.settings.openAIWebAccessEnabled },\n set: { enabled in\n context.settings.openAIWebAccessEnabled = enabled\n Task { @MainActor in\n await context.store.performRuntimeAction(\n .openAIWebAccessToggled(enabled),\n for: .codex)\n }\n })\n\n return [\n ProviderSettingsToggleDescriptor(\n id: \"codex-historical-tracking\",\n title: \"Historical tracking\",\n subtitle: \"Stores local Codex usage history (8 weeks) to personalize Pace predictions.\",\n binding: context.boolBinding(\\.historicalTrackingEnabled),\n statusText: nil,\n actions: [],\n isVisible: nil,\n onChange: nil,\n onAppDidBecomeActive: nil,\n onAppearWhenEnabled: nil),\n ProviderSettingsToggleDescriptor(\n id: \"codex-openai-web-extras\",\n title: \"OpenAI web extras\",\n subtitle: \"Show usage breakdown, credits history, and code review via chatgpt.com.\",\n binding: extrasBinding,\n statusText: nil,\n actions: [],\n isVisible: nil,\n onChange: nil,\n onAppDidBecomeActive: nil,\n onAppearWhenEnabled: nil),\n ]\n }\n\n @MainActor\n func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {\n let usageBinding = Binding(\n get: { context.settings.codexUsageDataSource.rawValue },\n set: { raw in\n context.settings.codexUsageDataSource = CodexUsageDataSource(rawValue: raw) ?? .auto\n })\n let cookieBinding = Binding(\n get: { context.settings.codexCookieSource.rawValue },\n set: { raw in\n context.settings.codexCookieSource = ProviderCookieSource(rawValue: raw) ?? .auto\n })\n\n let usageOptions = CodexUsageDataSource.allCases.map {\n ProviderSettingsPickerOption(id: $0.rawValue, title: $0.displayName)\n }\n let cookieOptions = ProviderCookieSourceUI.options(\n allowsOff: true,\n keychainDisabled: context.settings.debugDisableKeychainAccess)\n\n let cookieSubtitle: () -> String? = {\n ProviderCookieSourceUI.subtitle(\n source: context.settings.codexCookieSource,\n keychainDisabled: context.settings.debugDisableKeychainAccess,\n auto: \"Automatic imports browser cookies for dashboard extras.\",\n manual: \"Paste a Cookie header from a chatgpt.com request.\",\n off: \"Disable OpenAI dashboard cookie usage.\")\n }\n\n return [\n ProviderSettingsPickerDescriptor(\n id: \"codex-usage-source\",\n title: \"Usage source\",\n subtitle: \"Auto falls back to the next source if the preferred one fails.\",\n binding: usageBinding,\n options: usageOptions,\n isVisible: nil,\n onChange: nil,\n trailingText: {\n guard context.settings.codexUsageDataSource == .auto else { return nil }\n let label = context.store.sourceLabel(for: .codex)\n return label == \"auto\" ? nil : label\n }),\n ProviderSettingsPickerDescriptor(\n id: \"codex-cookie-source\",\n title: \"OpenAI cookies\",\n subtitle: \"Automatic imports browser cookies for dashboard extras.\",\n dynamicSubtitle: cookieSubtitle,\n binding: cookieBinding,\n options: cookieOptions,\n isVisible: { context.settings.openAIWebAccessEnabled },\n onChange: nil,\n trailingText: {\n guard let entry = CookieHeaderCache.load(provider: .codex) else { return nil }\n let when = entry.storedAt.relativeDescription()\n return \"Cached: \\(entry.sourceLabel) • \\(when)\"\n }),\n ]\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n [\n ProviderSettingsFieldDescriptor(\n id: \"codex-cookie-header\",\n title: \"\",\n subtitle: \"\",\n kind: .secure,\n placeholder: \"Cookie: …\",\n binding: context.stringBinding(\\.codexCookieHeader),\n actions: [],\n isVisible: {\n context.settings.codexCookieSource == .manual\n },\n onActivate: { context.settings.ensureCodexCookieLoaded() }),\n ]\n }\n\n @MainActor\n func appendUsageMenuEntries(context: ProviderMenuUsageContext, entries: inout [ProviderMenuEntry]) {\n guard context.settings.showOptionalCreditsAndExtraUsage,\n context.metadata.supportsCredits\n else { return }\n\n if let credits = context.store.credits {\n entries.append(.text(\"Credits: \\(UsageFormatter.creditsString(from: credits.remaining))\", .primary))\n if let latest = credits.events.first {\n entries.append(.text(\"Last spend: \\(UsageFormatter.creditEventSummary(latest))\", .secondary))\n }\n } else {\n let hint = context.store.lastCreditsError ?? context.metadata.creditsHint\n entries.append(.text(hint, .secondary))\n }\n }\n\n @MainActor\n func runLoginFlow(context: ProviderLoginContext) async -> Bool {\n await context.controller.runCodexLoginFlow()\n return true\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Codex/CodexProviderRuntime.swift", "content": "import CodexBarCore\nimport Foundation\n\n@MainActor\nfinal class CodexProviderRuntime: ProviderRuntime {\n let id: UsageProvider = .codex\n\n func perform(action: ProviderRuntimeAction, context: ProviderRuntimeContext) async {\n switch action {\n case let .openAIWebAccessToggled(enabled):\n guard enabled == false else { return }\n context.store.resetOpenAIWebState()\n case .forceSessionRefresh:\n break\n }\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Codex/CodexSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var codexUsageDataSource: CodexUsageDataSource {\n get {\n let source = self.configSnapshot.providerConfig(for: .codex)?.source\n return Self.codexUsageDataSource(from: source)\n }\n set {\n let source: ProviderSourceMode? = switch newValue {\n case .auto: .auto\n case .oauth: .oauth\n case .cli: .cli\n }\n self.updateProviderConfig(provider: .codex) { entry in\n entry.source = source\n }\n self.logProviderModeChange(provider: .codex, field: \"usageSource\", value: newValue.rawValue)\n }\n }\n\n var codexCookieHeader: String {\n get { self.configSnapshot.providerConfig(for: .codex)?.sanitizedCookieHeader ?? \"\" }\n set {\n self.updateProviderConfig(provider: .codex) { entry in\n entry.cookieHeader = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .codex, field: \"cookieHeader\", value: newValue)\n }\n }\n\n var codexCookieSource: ProviderCookieSource {\n get {\n let resolved = self.resolvedCookieSource(provider: .codex, fallback: .auto)\n return self.openAIWebAccessEnabled ? resolved : .off\n }\n set {\n self.updateProviderConfig(provider: .codex) { entry in\n entry.cookieSource = newValue\n }\n self.logProviderModeChange(provider: .codex, field: \"cookieSource\", value: newValue.rawValue)\n self.openAIWebAccessEnabled = newValue.isEnabled\n }\n }\n\n func ensureCodexCookieLoaded() {}\n}\n\nextension SettingsStore {\n func codexSettingsSnapshot(tokenOverride: TokenAccountOverride?) -> ProviderSettingsSnapshot.CodexProviderSettings {\n ProviderSettingsSnapshot.CodexProviderSettings(\n usageDataSource: self.codexUsageDataSource,\n cookieSource: self.codexSnapshotCookieSource(tokenOverride: tokenOverride),\n manualCookieHeader: self.codexSnapshotCookieHeader(tokenOverride: tokenOverride))\n }\n\n private static func codexUsageDataSource(from source: ProviderSourceMode?) -> CodexUsageDataSource {\n guard let source else { return .auto }\n switch source {\n case .auto, .web, .api:\n return .auto\n case .cli:\n return .cli\n case .oauth:\n return .oauth\n }\n }\n\n private func codexSnapshotCookieHeader(tokenOverride: TokenAccountOverride?) -> String {\n let fallback = self.codexCookieHeader\n guard let support = TokenAccountSupportCatalog.support(for: .codex),\n case .cookieHeader = support.injection\n else {\n return fallback\n }\n guard let account = ProviderTokenAccountSelection.selectedAccount(\n provider: .codex,\n settings: self,\n override: tokenOverride)\n else {\n return fallback\n }\n return TokenAccountSupportCatalog.normalizedCookieHeader(account.token, support: support)\n }\n\n private func codexSnapshotCookieSource(tokenOverride: TokenAccountOverride?) -> ProviderCookieSource {\n let fallback = self.codexCookieSource\n guard let support = TokenAccountSupportCatalog.support(for: .codex),\n support.requiresManualCookieSource\n else {\n return fallback\n }\n if self.tokenAccounts(for: .codex).isEmpty { return fallback }\n return .manual\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Copilot/CopilotLoginFlow.swift", "content": "import AppKit\nimport CodexBarCore\nimport SwiftUI\n\n@MainActor\nstruct CopilotLoginFlow {\n static func run(settings: SettingsStore) async {\n let flow = CopilotDeviceFlow()\n\n do {\n let code = try await flow.requestDeviceCode()\n\n // Copy code to clipboard\n let pb = NSPasteboard.general\n pb.clearContents()\n pb.setString(code.userCode, forType: .string)\n\n let alert = NSAlert()\n alert.messageText = \"GitHub Copilot Login\"\n alert.informativeText = \"\"\"\n A device code has been copied to your clipboard: \\(code.userCode)\n\n Please verify it at: \\(code.verificationUri)\n \"\"\"\n alert.addButton(withTitle: \"Open Browser\")\n alert.addButton(withTitle: \"Cancel\")\n\n let response = alert.runModal()\n if response == .alertSecondButtonReturn {\n return // Cancelled\n }\n\n if let url = URL(string: code.verificationUri) {\n NSWorkspace.shared.open(url)\n }\n\n // Poll in background (modal blocks, but we need to wait for token effectively)\n // Ideally we'd show a \"Waiting...\" modal or spinner.\n // For simplicity, we can use a non-modal window or just block a Task?\n // `runModal` blocks the thread. We need to poll while the user is doing auth in browser.\n // But we already returned from runModal to open the browser.\n // We need a secondary \"Waiting for confirmation...\" alert or state.\n\n // Let's show a \"Waiting\" alert that can be cancelled.\n let waitingAlert = NSAlert()\n waitingAlert.messageText = \"Waiting for Authentication...\"\n waitingAlert.informativeText = \"\"\"\n Please complete the login in your browser.\n This window will close automatically when finished.\n \"\"\"\n waitingAlert.addButton(withTitle: \"Cancel\")\n let parentWindow = Self.resolveWaitingParentWindow()\n let hostWindow = parentWindow ?? Self.makeWaitingHostWindow()\n let shouldCloseHostWindow = parentWindow == nil\n let tokenTask = Task.detached(priority: .userInitiated) {\n try await flow.pollForToken(deviceCode: code.deviceCode, interval: code.interval)\n }\n\n let waitTask = Task { @MainActor in\n let response = await Self.presentWaitingAlert(waitingAlert, parentWindow: hostWindow)\n if response == .alertFirstButtonReturn {\n tokenTask.cancel()\n }\n return response\n }\n\n let tokenResult: Result\n do {\n let token = try await tokenTask.value\n tokenResult = .success(token)\n } catch {\n tokenResult = .failure(error)\n }\n\n Self.dismissWaitingAlert(waitingAlert, parentWindow: hostWindow, closeHost: shouldCloseHostWindow)\n let waitResponse = await waitTask.value\n if waitResponse == .alertFirstButtonReturn {\n return\n }\n\n switch tokenResult {\n case let .success(token):\n settings.copilotAPIToken = token\n settings.setProviderEnabled(\n provider: .copilot,\n metadata: ProviderRegistry.shared.metadata[.copilot]!,\n enabled: true)\n\n let success = NSAlert()\n success.messageText = \"Login Successful\"\n success.runModal()\n case let .failure(error):\n guard !(error is CancellationError) else { return }\n let err = NSAlert()\n err.messageText = \"Login Failed\"\n err.informativeText = error.localizedDescription\n err.runModal()\n }\n\n } catch {\n let err = NSAlert()\n err.messageText = \"Login Failed\"\n err.informativeText = error.localizedDescription\n err.runModal()\n }\n }\n\n @MainActor\n private static func presentWaitingAlert(\n _ alert: NSAlert,\n parentWindow: NSWindow) async -> NSApplication.ModalResponse\n {\n await withCheckedContinuation { continuation in\n alert.beginSheetModal(for: parentWindow) { response in\n continuation.resume(returning: response)\n }\n }\n }\n\n @MainActor\n private static func dismissWaitingAlert(\n _ alert: NSAlert,\n parentWindow: NSWindow,\n closeHost: Bool)\n {\n let alertWindow = alert.window\n if alertWindow.sheetParent != nil {\n parentWindow.endSheet(alertWindow)\n } else {\n alertWindow.orderOut(nil)\n }\n\n guard closeHost else { return }\n parentWindow.orderOut(nil)\n parentWindow.close()\n }\n\n @MainActor\n private static func resolveWaitingParentWindow() -> NSWindow? {\n if let window = NSApp.keyWindow ?? NSApp.mainWindow {\n return window\n }\n if let window = NSApp.windows.first(where: { $0.isVisible && !$0.ignoresMouseEvents }) {\n return window\n }\n return NSApp.windows.first\n }\n\n @MainActor\n private static func makeWaitingHostWindow() -> NSWindow {\n let window = NSWindow(\n contentRect: NSRect(x: 0, y: 0, width: 420, height: 1),\n styleMask: [.titled, .fullSizeContentView],\n backing: .buffered,\n defer: false)\n window.isReleasedWhenClosed = false\n window.titleVisibility = .hidden\n window.titlebarAppearsTransparent = true\n window.standardWindowButton(.closeButton)?.isHidden = true\n window.standardWindowButton(.miniaturizeButton)?.isHidden = true\n window.standardWindowButton(.zoomButton)?.isHidden = true\n window.backgroundColor = .clear\n window.isOpaque = false\n window.hasShadow = false\n window.level = .floating\n window.collectionBehavior = [.canJoinAllSpaces, .fullScreenAuxiliary]\n window.center()\n window.makeKeyAndOrderFront(nil)\n return window\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Copilot/CopilotProviderImplementation.swift", "content": "import AppKit\nimport CodexBarCore\nimport CodexBarMacroSupport\nimport SwiftUI\n\n@ProviderImplementationRegistration\nstruct CopilotProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .copilot\n let supportsLoginFlow: Bool = true\n\n @MainActor\n func presentation(context _: ProviderPresentationContext) -> ProviderPresentation {\n ProviderPresentation { _ in \"github api\" }\n }\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.copilotAPIToken\n }\n\n @MainActor\n func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {\n _ = context\n return .copilot(context.settings.copilotSettingsSnapshot())\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n [\n ProviderSettingsFieldDescriptor(\n id: \"copilot-api-token\",\n title: \"GitHub Login\",\n subtitle: \"Requires authentication via GitHub Device Flow.\",\n kind: .secure,\n placeholder: \"Sign in via button below\",\n binding: context.stringBinding(\\.copilotAPIToken),\n actions: [\n ProviderSettingsActionDescriptor(\n id: \"copilot-login\",\n title: \"Sign in with GitHub\",\n style: .bordered,\n isVisible: { context.settings.copilotAPIToken.isEmpty },\n perform: {\n await CopilotLoginFlow.run(settings: context.settings)\n }),\n ProviderSettingsActionDescriptor(\n id: \"copilot-relogin\",\n title: \"Sign in again\",\n style: .link,\n isVisible: { !context.settings.copilotAPIToken.isEmpty },\n perform: {\n await CopilotLoginFlow.run(settings: context.settings)\n }),\n ],\n isVisible: nil,\n onActivate: { context.settings.ensureCopilotAPITokenLoaded() }),\n ]\n }\n\n @MainActor\n func runLoginFlow(context: ProviderLoginContext) async -> Bool {\n await CopilotLoginFlow.run(settings: context.controller.settings)\n return true\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Copilot/CopilotSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var copilotAPIToken: String {\n get { self.configSnapshot.providerConfig(for: .copilot)?.sanitizedAPIKey ?? \"\" }\n set {\n self.updateProviderConfig(provider: .copilot) { entry in\n entry.apiKey = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .copilot, field: \"apiKey\", value: newValue)\n }\n }\n\n func ensureCopilotAPITokenLoaded() {}\n}\n\nextension SettingsStore {\n func copilotSettingsSnapshot() -> ProviderSettingsSnapshot.CopilotProviderSettings {\n ProviderSettingsSnapshot.CopilotProviderSettings()\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Cursor/CursorLoginFlow.swift", "content": "import CodexBarCore\n\n@MainActor\nextension StatusItemController {\n func runCursorLoginFlow() async {\n let cursorRunner = CursorLoginRunner(browserDetection: self.store.browserDetection)\n let phaseHandler: @Sendable (CursorLoginRunner.Phase) -> Void = { [weak self] phase in\n Task { @MainActor in\n switch phase {\n case .loading, .waitingLogin:\n self?.loginPhase = .waitingBrowser\n case .success, .failed:\n self?.loginPhase = .idle\n }\n }\n }\n let result = await cursorRunner.run(onPhaseChange: phaseHandler)\n guard !Task.isCancelled else { return }\n self.loginPhase = .idle\n self.presentCursorLoginResult(result)\n let outcome = self.describe(result.outcome)\n self.loginLogger.info(\"Cursor login\", metadata: [\"outcome\": outcome])\n if case .success = result.outcome {\n self.postLoginNotification(for: .cursor)\n }\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Cursor/CursorProviderImplementation.swift", "content": "import CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\nimport SwiftUI\n\n@ProviderImplementationRegistration\nstruct CursorProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .cursor\n let supportsLoginFlow: Bool = true\n\n @MainActor\n func presentation(context _: ProviderPresentationContext) -> ProviderPresentation {\n ProviderPresentation { _ in \"web\" }\n }\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.cursorCookieSource\n _ = settings.cursorCookieHeader\n }\n\n @MainActor\n func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {\n .cursor(context.settings.cursorSettingsSnapshot(tokenOverride: context.tokenOverride))\n }\n\n @MainActor\n func tokenAccountsVisibility(context: ProviderSettingsContext, support: TokenAccountSupport) -> Bool {\n guard support.requiresManualCookieSource else { return true }\n if !context.settings.tokenAccounts(for: context.provider).isEmpty { return true }\n return context.settings.cursorCookieSource == .manual\n }\n\n @MainActor\n func applyTokenAccountCookieSource(settings: SettingsStore) {\n if settings.cursorCookieSource != .manual {\n settings.cursorCookieSource = .manual\n }\n }\n\n @MainActor\n func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {\n let cookieBinding = Binding(\n get: { context.settings.cursorCookieSource.rawValue },\n set: { raw in\n context.settings.cursorCookieSource = ProviderCookieSource(rawValue: raw) ?? .auto\n })\n let cookieOptions = ProviderCookieSourceUI.options(\n allowsOff: false,\n keychainDisabled: context.settings.debugDisableKeychainAccess)\n\n let cookieSubtitle: () -> String? = {\n ProviderCookieSourceUI.subtitle(\n source: context.settings.cursorCookieSource,\n keychainDisabled: context.settings.debugDisableKeychainAccess,\n auto: \"Automatic imports browser cookies or stored sessions.\",\n manual: \"Paste a Cookie header from a cursor.com request.\",\n off: \"Cursor cookies are disabled.\")\n }\n\n return [\n ProviderSettingsPickerDescriptor(\n id: \"cursor-cookie-source\",\n title: \"Cookie source\",\n subtitle: \"Automatic imports browser cookies or stored sessions.\",\n dynamicSubtitle: cookieSubtitle,\n binding: cookieBinding,\n options: cookieOptions,\n isVisible: nil,\n onChange: nil,\n trailingText: {\n guard let entry = CookieHeaderCache.load(provider: .cursor) else { return nil }\n let when = entry.storedAt.relativeDescription()\n return \"Cached: \\(entry.sourceLabel) • \\(when)\"\n }),\n ]\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n _ = context\n return []\n }\n\n @MainActor\n func runLoginFlow(context: ProviderLoginContext) async -> Bool {\n await context.controller.runCursorLoginFlow()\n return true\n }\n\n @MainActor\n func appendUsageMenuEntries(context: ProviderMenuUsageContext, entries: inout [ProviderMenuEntry]) {\n guard let cost = context.snapshot?.providerCost, cost.currencyCode != \"Quota\" else { return }\n let used = UsageFormatter.currencyString(cost.used, currencyCode: cost.currencyCode)\n if cost.limit > 0 {\n let limitStr = UsageFormatter.currencyString(cost.limit, currencyCode: cost.currencyCode)\n entries.append(.text(\"On-Demand: \\(used) / \\(limitStr)\", .primary))\n } else {\n entries.append(.text(\"On-Demand: \\(used)\", .primary))\n }\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Cursor/CursorSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var cursorCookieHeader: String {\n get { self.configSnapshot.providerConfig(for: .cursor)?.sanitizedCookieHeader ?? \"\" }\n set {\n self.updateProviderConfig(provider: .cursor) { entry in\n entry.cookieHeader = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .cursor, field: \"cookieHeader\", value: newValue)\n }\n }\n\n var cursorCookieSource: ProviderCookieSource {\n get { self.resolvedCookieSource(provider: .cursor, fallback: .auto) }\n set {\n self.updateProviderConfig(provider: .cursor) { entry in\n entry.cookieSource = newValue\n }\n self.logProviderModeChange(provider: .cursor, field: \"cookieSource\", value: newValue.rawValue)\n }\n }\n\n func ensureCursorCookieLoaded() {}\n}\n\nextension SettingsStore {\n func cursorSettingsSnapshot(tokenOverride: TokenAccountOverride?) -> ProviderSettingsSnapshot\n .CursorProviderSettings {\n ProviderSettingsSnapshot.CursorProviderSettings(\n cookieSource: self.cursorSnapshotCookieSource(tokenOverride: tokenOverride),\n manualCookieHeader: self.cursorSnapshotCookieHeader(tokenOverride: tokenOverride))\n }\n\n private func cursorSnapshotCookieHeader(tokenOverride: TokenAccountOverride?) -> String {\n let fallback = self.cursorCookieHeader\n guard let support = TokenAccountSupportCatalog.support(for: .cursor),\n case .cookieHeader = support.injection\n else {\n return fallback\n }\n guard let account = ProviderTokenAccountSelection.selectedAccount(\n provider: .cursor,\n settings: self,\n override: tokenOverride)\n else {\n return fallback\n }\n return TokenAccountSupportCatalog.normalizedCookieHeader(account.token, support: support)\n }\n\n private func cursorSnapshotCookieSource(tokenOverride: TokenAccountOverride?) -> ProviderCookieSource {\n let fallback = self.cursorCookieSource\n guard let support = TokenAccountSupportCatalog.support(for: .cursor),\n support.requiresManualCookieSource\n else {\n return fallback\n }\n if self.tokenAccounts(for: .cursor).isEmpty { return fallback }\n return .manual\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Factory/FactoryLoginFlow.swift", "content": "import AppKit\nimport CodexBarCore\n\n@MainActor\nextension StatusItemController {\n func runFactoryLoginFlow() async {\n // Open Factory login page in default browser\n if let url = URL(string: \"https://app.factory.ai\") {\n NSWorkspace.shared.open(url)\n }\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Factory/FactoryProviderImplementation.swift", "content": "import CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\nimport SwiftUI\n\n@ProviderImplementationRegistration\nstruct FactoryProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .factory\n let supportsLoginFlow: Bool = true\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.factoryCookieSource\n _ = settings.factoryCookieHeader\n }\n\n @MainActor\n func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {\n .factory(context.settings.factorySettingsSnapshot(tokenOverride: context.tokenOverride))\n }\n\n @MainActor\n func tokenAccountsVisibility(context: ProviderSettingsContext, support: TokenAccountSupport) -> Bool {\n guard support.requiresManualCookieSource else { return true }\n if !context.settings.tokenAccounts(for: context.provider).isEmpty { return true }\n return context.settings.factoryCookieSource == .manual\n }\n\n @MainActor\n func applyTokenAccountCookieSource(settings: SettingsStore) {\n if settings.factoryCookieSource != .manual {\n settings.factoryCookieSource = .manual\n }\n }\n\n @MainActor\n func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {\n let cookieBinding = Binding(\n get: { context.settings.factoryCookieSource.rawValue },\n set: { raw in\n context.settings.factoryCookieSource = ProviderCookieSource(rawValue: raw) ?? .auto\n })\n let cookieOptions = ProviderCookieSourceUI.options(\n allowsOff: false,\n keychainDisabled: context.settings.debugDisableKeychainAccess)\n\n let cookieSubtitle: () -> String? = {\n ProviderCookieSourceUI.subtitle(\n source: context.settings.factoryCookieSource,\n keychainDisabled: context.settings.debugDisableKeychainAccess,\n auto: \"Automatic imports browser cookies and WorkOS tokens.\",\n manual: \"Paste a Cookie header from app.factory.ai.\",\n off: \"Factory cookies are disabled.\")\n }\n\n return [\n ProviderSettingsPickerDescriptor(\n id: \"factory-cookie-source\",\n title: \"Cookie source\",\n subtitle: \"Automatic imports browser cookies and WorkOS tokens.\",\n dynamicSubtitle: cookieSubtitle,\n binding: cookieBinding,\n options: cookieOptions,\n isVisible: nil,\n onChange: nil,\n trailingText: {\n guard let entry = CookieHeaderCache.load(provider: .factory) else { return nil }\n let when = entry.storedAt.relativeDescription()\n return \"Cached: \\(entry.sourceLabel) • \\(when)\"\n }),\n ]\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n _ = context\n return []\n }\n\n @MainActor\n func runLoginFlow(context: ProviderLoginContext) async -> Bool {\n await context.controller.runFactoryLoginFlow()\n return true\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Factory/FactorySettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var factoryCookieHeader: String {\n get { self.configSnapshot.providerConfig(for: .factory)?.sanitizedCookieHeader ?? \"\" }\n set {\n self.updateProviderConfig(provider: .factory) { entry in\n entry.cookieHeader = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .factory, field: \"cookieHeader\", value: newValue)\n }\n }\n\n var factoryCookieSource: ProviderCookieSource {\n get { self.resolvedCookieSource(provider: .factory, fallback: .auto) }\n set {\n self.updateProviderConfig(provider: .factory) { entry in\n entry.cookieSource = newValue\n }\n self.logProviderModeChange(provider: .factory, field: \"cookieSource\", value: newValue.rawValue)\n }\n }\n\n func ensureFactoryCookieLoaded() {}\n}\n\nextension SettingsStore {\n func factorySettingsSnapshot(tokenOverride: TokenAccountOverride?) -> ProviderSettingsSnapshot\n .FactoryProviderSettings {\n ProviderSettingsSnapshot.FactoryProviderSettings(\n cookieSource: self.factorySnapshotCookieSource(tokenOverride: tokenOverride),\n manualCookieHeader: self.factorySnapshotCookieHeader(tokenOverride: tokenOverride))\n }\n\n private func factorySnapshotCookieHeader(tokenOverride: TokenAccountOverride?) -> String {\n let fallback = self.factoryCookieHeader\n guard let support = TokenAccountSupportCatalog.support(for: .factory),\n case .cookieHeader = support.injection\n else {\n return fallback\n }\n guard let account = ProviderTokenAccountSelection.selectedAccount(\n provider: .factory,\n settings: self,\n override: tokenOverride)\n else {\n return fallback\n }\n return TokenAccountSupportCatalog.normalizedCookieHeader(account.token, support: support)\n }\n\n private func factorySnapshotCookieSource(tokenOverride: TokenAccountOverride?) -> ProviderCookieSource {\n let fallback = self.factoryCookieSource\n guard let support = TokenAccountSupportCatalog.support(for: .factory),\n support.requiresManualCookieSource\n else {\n return fallback\n }\n if self.tokenAccounts(for: .factory).isEmpty { return fallback }\n return .manual\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Gemini/GeminiLoginFlow.swift", "content": "import CodexBarCore\n\n@MainActor\nextension StatusItemController {\n func runGeminiLoginFlow() async {\n let store = self.store\n let result = await GeminiLoginRunner.run {\n Task { @MainActor in\n await store.refresh()\n CodexBarLog.logger(LogCategories.login).info(\"Auto-refreshed after Gemini auth\")\n }\n }\n guard !Task.isCancelled else { return }\n self.loginPhase = .idle\n self.presentGeminiLoginResult(result)\n let outcome = self.describe(result.outcome)\n self.loginLogger.info(\"Gemini login\", metadata: [\"outcome\": outcome])\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Gemini/GeminiProviderImplementation.swift", "content": "import CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\n\n@ProviderImplementationRegistration\nstruct GeminiProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .gemini\n let supportsLoginFlow: Bool = true\n\n @MainActor\n func runLoginFlow(context: ProviderLoginContext) async -> Bool {\n await context.controller.runGeminiLoginFlow()\n return false\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/JetBrains/JetBrainsLoginFlow.swift", "content": "import CodexBarCore\n\n@MainActor\nextension StatusItemController {\n func runJetBrainsLoginFlow() async {\n self.loginPhase = .idle\n let detectedIDEs = JetBrainsIDEDetector.detectInstalledIDEs(includeMissingQuota: true)\n if detectedIDEs.isEmpty {\n let message = [\n \"Install a JetBrains IDE with AI Assistant enabled, then refresh CodexBar.\",\n \"Alternatively, set a custom path in Settings.\",\n ].joined(separator: \" \")\n self.presentLoginAlert(\n title: \"No JetBrains IDE detected\",\n message: message)\n } else {\n let ideNames = detectedIDEs.prefix(3).map(\\.displayName).joined(separator: \", \")\n let hasQuotaFile = !JetBrainsIDEDetector.detectInstalledIDEs().isEmpty\n let message = hasQuotaFile\n ? \"Detected: \\(ideNames). Select your preferred IDE in Settings, then refresh CodexBar.\"\n : \"Detected: \\(ideNames). Use AI Assistant once to generate quota data, then refresh CodexBar.\"\n self.presentLoginAlert(\n title: \"JetBrains AI is ready\",\n message: message)\n }\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/JetBrains/JetBrainsProviderImplementation.swift", "content": "import CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\nimport SwiftUI\n\n@ProviderImplementationRegistration\nstruct JetBrainsProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .jetbrains\n\n @MainActor\n func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {\n _ = context\n return .jetbrains(context.settings.jetbrainsSettingsSnapshot())\n }\n\n @MainActor\n func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {\n let detectedIDEs = JetBrainsIDEDetector.detectInstalledIDEs(includeMissingQuota: true)\n guard !detectedIDEs.isEmpty else { return [] }\n\n var options: [ProviderSettingsPickerOption] = [\n ProviderSettingsPickerOption(id: \"\", title: \"Auto-detect\"),\n ]\n for ide in detectedIDEs {\n options.append(ProviderSettingsPickerOption(id: ide.basePath, title: ide.displayName))\n }\n\n return [\n ProviderSettingsPickerDescriptor(\n id: \"jetbrains.ide\",\n title: \"JetBrains IDE\",\n subtitle: \"Select the IDE to monitor\",\n binding: context.stringBinding(\\.jetbrainsIDEBasePath),\n options: options,\n isVisible: nil,\n onChange: nil,\n trailingText: {\n if context.settings.jetbrainsIDEBasePath.isEmpty {\n if let latest = JetBrainsIDEDetector.detectLatestIDE() {\n return latest.displayName\n }\n }\n return nil\n }),\n ]\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n [\n ProviderSettingsFieldDescriptor(\n id: \"jetbrains.customPath\",\n title: \"Custom Path\",\n subtitle: \"Override auto-detection with a custom IDE base path\",\n kind: .plain,\n placeholder: \"~/Library/Application Support/JetBrains/IntelliJIdea2024.3\",\n binding: context.stringBinding(\\.jetbrainsIDEBasePath),\n actions: [],\n isVisible: {\n let detectedIDEs = JetBrainsIDEDetector.detectInstalledIDEs()\n return detectedIDEs.isEmpty || !context.settings.jetbrainsIDEBasePath.isEmpty\n },\n onActivate: nil),\n ]\n }\n\n @MainActor\n func runLoginFlow(context: ProviderLoginContext) async -> Bool {\n await context.controller.runJetBrainsLoginFlow()\n return false\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/JetBrains/JetBrainsSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n func jetbrainsSettingsSnapshot() -> ProviderSettingsSnapshot.JetBrainsProviderSettings {\n ProviderSettingsSnapshot.JetBrainsProviderSettings(\n ideBasePath: self.jetbrainsIDEBasePath.isEmpty ? nil : self.jetbrainsIDEBasePath)\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Kilo/KiloProviderImplementation.swift", "content": "import AppKit\nimport CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\nimport SwiftUI\n\n@ProviderImplementationRegistration\nstruct KiloProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .kilo\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.kiloUsageDataSource\n _ = settings.kiloExtrasEnabled\n _ = settings.kiloAPIToken\n }\n\n @MainActor\n func isAvailable(context _: ProviderAvailabilityContext) -> Bool {\n // Keep availability permissive to avoid main-thread auth-file I/O while still showing Kilo for auth.json-only\n // setups. Fetch-time auth resolution remains authoritative (env first, then auth file fallback).\n true\n }\n\n @MainActor\n func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {\n .kilo(context.settings.kiloSettingsSnapshot(tokenOverride: context.tokenOverride))\n }\n\n @MainActor\n func defaultSourceLabel(context: ProviderSourceLabelContext) -> String? {\n context.settings.kiloUsageDataSource.rawValue\n }\n\n @MainActor\n func sourceMode(context: ProviderSourceModeContext) -> ProviderSourceMode {\n switch context.settings.kiloUsageDataSource {\n case .auto: .auto\n case .api: .api\n case .cli: .cli\n }\n }\n\n @MainActor\n func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {\n let usageBinding = Binding(\n get: { context.settings.kiloUsageDataSource.rawValue },\n set: { raw in\n context.settings.kiloUsageDataSource = KiloUsageDataSource(rawValue: raw) ?? .auto\n })\n let usageOptions = KiloUsageDataSource.allCases.map {\n ProviderSettingsPickerOption(id: $0.rawValue, title: $0.displayName)\n }\n return [\n ProviderSettingsPickerDescriptor(\n id: \"kilo-usage-source\",\n title: \"Usage source\",\n subtitle: \"Auto uses API first, then falls back to CLI on auth failures.\",\n binding: usageBinding,\n options: usageOptions,\n isVisible: nil,\n onChange: nil,\n trailingText: {\n guard context.settings.kiloUsageDataSource == .auto else { return nil }\n let label = context.store.sourceLabel(for: .kilo)\n return label == \"auto\" ? nil : label\n }),\n ]\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n [\n ProviderSettingsFieldDescriptor(\n id: \"kilo-api-key\",\n title: \"API key\",\n subtitle: \"Stored in ~/.codexbar/config.json. You can also provide KILO_API_KEY or \"\n + \"~/.local/share/kilo/auth.json (kilo.access).\",\n kind: .secure,\n placeholder: \"kilo_...\",\n binding: context.stringBinding(\\.kiloAPIToken),\n actions: [],\n isVisible: nil,\n onActivate: nil),\n ]\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Kilo/KiloSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var kiloUsageDataSource: KiloUsageDataSource {\n get {\n let source = self.configSnapshot.providerConfig(for: .kilo)?.source\n return Self.kiloUsageDataSource(from: source)\n }\n set {\n let source: ProviderSourceMode? = switch newValue {\n case .auto: .auto\n case .api: .api\n case .cli: .cli\n }\n self.updateProviderConfig(provider: .kilo) { entry in\n entry.source = source\n }\n self.logProviderModeChange(provider: .kilo, field: \"usageSource\", value: newValue.rawValue)\n }\n }\n\n var kiloExtrasEnabled: Bool {\n get {\n guard self.kiloUsageDataSource == .auto else { return false }\n return self.kiloExtrasEnabledRaw\n }\n set {\n self.kiloExtrasEnabledRaw = newValue\n }\n }\n\n var kiloAPIToken: String {\n get { self.configSnapshot.providerConfig(for: .kilo)?.sanitizedAPIKey ?? \"\" }\n set {\n self.updateProviderConfig(provider: .kilo) { entry in\n entry.apiKey = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .kilo, field: \"apiKey\", value: newValue)\n }\n }\n\n private var kiloExtrasEnabledRaw: Bool {\n get { self.configSnapshot.providerConfig(for: .kilo)?.extrasEnabled ?? false }\n set {\n self.updateProviderConfig(provider: .kilo) { entry in\n entry.extrasEnabled = newValue\n }\n self.logProviderModeChange(\n provider: .kilo,\n field: \"extrasEnabled\",\n value: newValue ? \"1\" : \"0\")\n }\n }\n}\n\nextension SettingsStore {\n func kiloSettingsSnapshot(tokenOverride _: TokenAccountOverride?) -> ProviderSettingsSnapshot.KiloProviderSettings {\n ProviderSettingsSnapshot.KiloProviderSettings(\n usageDataSource: self.kiloUsageDataSource,\n extrasEnabled: self.kiloExtrasEnabled)\n }\n\n private static func kiloUsageDataSource(from source: ProviderSourceMode?) -> KiloUsageDataSource {\n guard let source else { return .auto }\n switch source {\n case .auto, .web, .oauth:\n return .auto\n case .api:\n return .api\n case .cli:\n return .cli\n }\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Kimi/KimiProviderImplementation.swift", "content": "import AppKit\nimport CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\nimport SwiftUI\n\n@ProviderImplementationRegistration\nstruct KimiProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .kimi\n\n @MainActor\n func presentation(context _: ProviderPresentationContext) -> ProviderPresentation {\n ProviderPresentation { _ in \"web\" }\n }\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.kimiCookieSource\n _ = settings.kimiManualCookieHeader\n }\n\n @MainActor\n func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {\n .kimi(context.settings.kimiSettingsSnapshot(tokenOverride: context.tokenOverride))\n }\n\n @MainActor\n func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {\n let cookieBinding = Binding(\n get: { context.settings.kimiCookieSource.rawValue },\n set: { raw in\n context.settings.kimiCookieSource = ProviderCookieSource(rawValue: raw) ?? .auto\n })\n let options = ProviderCookieSourceUI.options(\n allowsOff: true,\n keychainDisabled: context.settings.debugDisableKeychainAccess)\n\n let subtitle: () -> String? = {\n ProviderCookieSourceUI.subtitle(\n source: context.settings.kimiCookieSource,\n keychainDisabled: context.settings.debugDisableKeychainAccess,\n auto: \"Automatic imports browser cookies.\",\n manual: \"Paste a cookie header or the kimi-auth token value.\",\n off: \"Kimi cookies are disabled.\")\n }\n\n return [\n ProviderSettingsPickerDescriptor(\n id: \"kimi-cookie-source\",\n title: \"Cookie source\",\n subtitle: \"Automatic imports browser cookies.\",\n dynamicSubtitle: subtitle,\n binding: cookieBinding,\n options: options,\n isVisible: nil,\n onChange: nil),\n ]\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n [\n ProviderSettingsFieldDescriptor(\n id: \"kimi-cookie\",\n title: \"\",\n subtitle: \"\",\n kind: .secure,\n placeholder: \"Cookie: \\u{2026}\\n\\nor paste the kimi-auth token value\",\n binding: context.stringBinding(\\.kimiManualCookieHeader),\n actions: [\n ProviderSettingsActionDescriptor(\n id: \"kimi-open-console\",\n title: \"Open Console\",\n style: .link,\n isVisible: nil,\n perform: {\n if let url = URL(string: \"https://www.kimi.com/code/console\") {\n NSWorkspace.shared.open(url)\n }\n }),\n ],\n isVisible: { context.settings.kimiCookieSource == .manual },\n onActivate: { context.settings.ensureKimiAuthTokenLoaded() }),\n ]\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Kimi/KimiSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var kimiManualCookieHeader: String {\n get { self.configSnapshot.providerConfig(for: .kimi)?.sanitizedCookieHeader ?? \"\" }\n set {\n self.updateProviderConfig(provider: .kimi) { entry in\n entry.cookieHeader = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .kimi, field: \"cookieHeader\", value: newValue)\n }\n }\n\n var kimiCookieSource: ProviderCookieSource {\n get { self.resolvedCookieSource(provider: .kimi, fallback: .auto) }\n set {\n self.updateProviderConfig(provider: .kimi) { entry in\n entry.cookieSource = newValue\n }\n self.logProviderModeChange(provider: .kimi, field: \"cookieSource\", value: newValue.rawValue)\n }\n }\n\n func ensureKimiAuthTokenLoaded() {}\n}\n\nextension SettingsStore {\n func kimiSettingsSnapshot(tokenOverride: TokenAccountOverride?) -> ProviderSettingsSnapshot.KimiProviderSettings {\n _ = tokenOverride\n self.ensureKimiAuthTokenLoaded()\n return ProviderSettingsSnapshot.KimiProviderSettings(\n cookieSource: self.kimiCookieSource,\n manualCookieHeader: self.kimiManualCookieHeader)\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/KimiK2/KimiK2ProviderImplementation.swift", "content": "import AppKit\nimport CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\n\n@ProviderImplementationRegistration\nstruct KimiK2ProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .kimik2\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.kimiK2APIToken\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n [\n ProviderSettingsFieldDescriptor(\n id: \"kimi-k2-api-token\",\n title: \"API key\",\n subtitle: \"Stored in ~/.codexbar/config.json. Generate one at kimi-k2.ai.\",\n kind: .secure,\n placeholder: \"Paste API key…\",\n binding: context.stringBinding(\\.kimiK2APIToken),\n actions: [\n ProviderSettingsActionDescriptor(\n id: \"kimi-k2-open-api-keys\",\n title: \"Open API Keys\",\n style: .link,\n isVisible: nil,\n perform: {\n if let url = URL(string: \"https://kimi-k2.ai/user-center/api-keys\") {\n NSWorkspace.shared.open(url)\n }\n }),\n ],\n isVisible: nil,\n onActivate: { context.settings.ensureKimiK2APITokenLoaded() }),\n ]\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/KimiK2/KimiK2SettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var kimiK2APIToken: String {\n get { self.configSnapshot.providerConfig(for: .kimik2)?.sanitizedAPIKey ?? \"\" }\n set {\n self.updateProviderConfig(provider: .kimik2) { entry in\n entry.apiKey = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .kimik2, field: \"apiKey\", value: newValue)\n }\n }\n\n func ensureKimiK2APITokenLoaded() {}\n}\n" }, { "path": "Sources/CodexBar/Providers/Kiro/KiroProviderImplementation.swift", "content": "import CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\n\n@ProviderImplementationRegistration\nstruct KiroProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .kiro\n}\n" }, { "path": "Sources/CodexBar/Providers/MiniMax/MiniMaxProviderImplementation.swift", "content": "import AppKit\nimport CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\nimport SwiftUI\n\n@ProviderImplementationRegistration\nstruct MiniMaxProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .minimax\n\n @MainActor\n func presentation(context _: ProviderPresentationContext) -> ProviderPresentation {\n ProviderPresentation { context in\n context.store.sourceLabel(for: context.provider)\n }\n }\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.minimaxCookieSource\n _ = settings.minimaxCookieHeader\n _ = settings.minimaxAPIToken\n _ = settings.minimaxAPIRegion\n }\n\n @MainActor\n func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {\n .minimax(context.settings.minimaxSettingsSnapshot(tokenOverride: context.tokenOverride))\n }\n\n @MainActor\n func tokenAccountsVisibility(context: ProviderSettingsContext, support: TokenAccountSupport) -> Bool {\n guard support.requiresManualCookieSource else { return true }\n if !context.settings.tokenAccounts(for: context.provider).isEmpty { return true }\n context.settings.ensureMiniMaxAPITokenLoaded()\n if context.settings.minimaxAuthMode().usesAPIToken { return false }\n return context.settings.minimaxCookieSource == .manual\n }\n\n @MainActor\n func applyTokenAccountCookieSource(settings: SettingsStore) {\n if settings.minimaxCookieSource != .manual {\n settings.minimaxCookieSource = .manual\n }\n }\n\n @MainActor\n func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {\n context.settings.ensureMiniMaxAPITokenLoaded()\n let authMode: () -> MiniMaxAuthMode = {\n context.settings.minimaxAuthMode()\n }\n\n let cookieBinding = Binding(\n get: { context.settings.minimaxCookieSource.rawValue },\n set: { raw in\n context.settings.minimaxCookieSource = ProviderCookieSource(rawValue: raw) ?? .auto\n })\n let cookieOptions = ProviderCookieSourceUI.options(\n allowsOff: false,\n keychainDisabled: context.settings.debugDisableKeychainAccess)\n\n let cookieSubtitle: () -> String? = {\n ProviderCookieSourceUI.subtitle(\n source: context.settings.minimaxCookieSource,\n keychainDisabled: context.settings.debugDisableKeychainAccess,\n auto: \"Automatic imports browser cookies and local storage tokens.\",\n manual: \"Paste a Cookie header or cURL capture from the Coding Plan page.\",\n off: \"MiniMax cookies are disabled.\")\n }\n\n let regionBinding = Binding(\n get: { context.settings.minimaxAPIRegion.rawValue },\n set: { raw in\n context.settings.minimaxAPIRegion = MiniMaxAPIRegion(rawValue: raw) ?? .global\n })\n let regionOptions = MiniMaxAPIRegion.allCases.map {\n ProviderSettingsPickerOption(id: $0.rawValue, title: $0.displayName)\n }\n\n return [\n ProviderSettingsPickerDescriptor(\n id: \"minimax-cookie-source\",\n title: \"Cookie source\",\n subtitle: \"Automatic imports browser cookies and local storage tokens.\",\n dynamicSubtitle: cookieSubtitle,\n binding: cookieBinding,\n options: cookieOptions,\n isVisible: { authMode().allowsCookies },\n onChange: nil,\n trailingText: {\n guard let entry = CookieHeaderCache.load(provider: .minimax) else { return nil }\n let when = entry.storedAt.relativeDescription()\n return \"Cached: \\(entry.sourceLabel) • \\(when)\"\n }),\n ProviderSettingsPickerDescriptor(\n id: \"minimax-region\",\n title: \"API region\",\n subtitle: \"Choose the MiniMax host (global .io or China mainland .com).\",\n binding: regionBinding,\n options: regionOptions,\n isVisible: nil,\n onChange: nil),\n ]\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n context.settings.ensureMiniMaxAPITokenLoaded()\n let authMode: () -> MiniMaxAuthMode = {\n context.settings.minimaxAuthMode()\n }\n\n return [\n ProviderSettingsFieldDescriptor(\n id: \"minimax-api-token\",\n title: \"API token\",\n subtitle: \"Stored in ~/.codexbar/config.json. Paste your MiniMax API key.\",\n kind: .secure,\n placeholder: \"Paste API token…\",\n binding: context.stringBinding(\\.minimaxAPIToken),\n actions: [\n ProviderSettingsActionDescriptor(\n id: \"minimax-open-dashboard\",\n title: \"Open Coding Plan\",\n style: .link,\n isVisible: nil,\n perform: {\n NSWorkspace.shared.open(context.settings.minimaxAPIRegion.codingPlanURL)\n }),\n ],\n isVisible: nil,\n onActivate: { context.settings.ensureMiniMaxAPITokenLoaded() }),\n ProviderSettingsFieldDescriptor(\n id: \"minimax-cookie\",\n title: \"Cookie header\",\n subtitle: \"\",\n kind: .secure,\n placeholder: \"Cookie: …\",\n binding: context.stringBinding(\\.minimaxCookieHeader),\n actions: [\n ProviderSettingsActionDescriptor(\n id: \"minimax-open-dashboard-cookie\",\n title: \"Open Coding Plan\",\n style: .link,\n isVisible: nil,\n perform: {\n NSWorkspace.shared.open(context.settings.minimaxAPIRegion.codingPlanURL)\n }),\n ],\n isVisible: {\n authMode().allowsCookies && context.settings.minimaxCookieSource == .manual\n },\n onActivate: { context.settings.ensureMiniMaxCookieLoaded() }),\n ]\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/MiniMax/MiniMaxSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var minimaxAPIRegion: MiniMaxAPIRegion {\n get {\n let raw = self.configSnapshot.providerConfig(for: .minimax)?.region\n return MiniMaxAPIRegion(rawValue: raw ?? \"\") ?? .global\n }\n set {\n self.updateProviderConfig(provider: .minimax) { entry in\n entry.region = newValue.rawValue\n }\n }\n }\n\n var minimaxCookieHeader: String {\n get { self.configSnapshot.providerConfig(for: .minimax)?.sanitizedCookieHeader ?? \"\" }\n set {\n self.updateProviderConfig(provider: .minimax) { entry in\n entry.cookieHeader = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .minimax, field: \"cookieHeader\", value: newValue)\n }\n }\n\n var minimaxAPIToken: String {\n get { self.configSnapshot.providerConfig(for: .minimax)?.sanitizedAPIKey ?? \"\" }\n set {\n self.updateProviderConfig(provider: .minimax) { entry in\n entry.apiKey = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .minimax, field: \"apiKey\", value: newValue)\n }\n }\n\n var minimaxCookieSource: ProviderCookieSource {\n get { self.resolvedCookieSource(provider: .minimax, fallback: .auto) }\n set {\n self.updateProviderConfig(provider: .minimax) { entry in\n entry.cookieSource = newValue\n }\n self.logProviderModeChange(provider: .minimax, field: \"cookieSource\", value: newValue.rawValue)\n }\n }\n\n func ensureMiniMaxCookieLoaded() {}\n\n func ensureMiniMaxAPITokenLoaded() {}\n\n func minimaxAuthMode(\n environment: [String: String] = ProcessInfo.processInfo.environment) -> MiniMaxAuthMode\n {\n let apiToken = MiniMaxAPISettingsReader.apiToken(environment: environment) ?? self.minimaxAPIToken\n let cookieHeader = MiniMaxSettingsReader.cookieHeader(environment: environment) ?? self.minimaxCookieHeader\n return MiniMaxAuthMode.resolve(apiToken: apiToken, cookieHeader: cookieHeader)\n }\n}\n\nextension SettingsStore {\n func minimaxSettingsSnapshot(tokenOverride: TokenAccountOverride?) -> ProviderSettingsSnapshot\n .MiniMaxProviderSettings {\n ProviderSettingsSnapshot.MiniMaxProviderSettings(\n cookieSource: self.minimaxSnapshotCookieSource(tokenOverride: tokenOverride),\n manualCookieHeader: self.minimaxSnapshotCookieHeader(tokenOverride: tokenOverride),\n apiRegion: self.minimaxAPIRegion)\n }\n\n private func minimaxSnapshotCookieHeader(tokenOverride: TokenAccountOverride?) -> String {\n let fallback = self.minimaxCookieHeader\n guard let support = TokenAccountSupportCatalog.support(for: .minimax),\n case .cookieHeader = support.injection\n else {\n return fallback\n }\n guard let account = ProviderTokenAccountSelection.selectedAccount(\n provider: .minimax,\n settings: self,\n override: tokenOverride)\n else {\n return fallback\n }\n return TokenAccountSupportCatalog.normalizedCookieHeader(account.token, support: support)\n }\n\n private func minimaxSnapshotCookieSource(tokenOverride: TokenAccountOverride?) -> ProviderCookieSource {\n let fallback = self.minimaxCookieSource\n guard let support = TokenAccountSupportCatalog.support(for: .minimax),\n support.requiresManualCookieSource\n else {\n return fallback\n }\n if self.tokenAccounts(for: .minimax).isEmpty { return fallback }\n return .manual\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Ollama/OllamaProviderImplementation.swift", "content": "import AppKit\nimport CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\nimport SwiftUI\n\n@ProviderImplementationRegistration\nstruct OllamaProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .ollama\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.ollamaCookieSource\n _ = settings.ollamaCookieHeader\n }\n\n @MainActor\n func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {\n .ollama(context.settings.ollamaSettingsSnapshot(tokenOverride: context.tokenOverride))\n }\n\n @MainActor\n func tokenAccountsVisibility(context: ProviderSettingsContext, support: TokenAccountSupport) -> Bool {\n guard support.requiresManualCookieSource else { return true }\n if !context.settings.tokenAccounts(for: context.provider).isEmpty { return true }\n return context.settings.ollamaCookieSource == .manual\n }\n\n @MainActor\n func applyTokenAccountCookieSource(settings: SettingsStore) {\n if settings.ollamaCookieSource != .manual {\n settings.ollamaCookieSource = .manual\n }\n }\n\n @MainActor\n func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {\n let cookieBinding = Binding(\n get: { context.settings.ollamaCookieSource.rawValue },\n set: { raw in\n context.settings.ollamaCookieSource = ProviderCookieSource(rawValue: raw) ?? .auto\n })\n let cookieOptions = ProviderCookieSourceUI.options(\n allowsOff: false,\n keychainDisabled: context.settings.debugDisableKeychainAccess)\n\n let cookieSubtitle: () -> String? = {\n ProviderCookieSourceUI.subtitle(\n source: context.settings.ollamaCookieSource,\n keychainDisabled: context.settings.debugDisableKeychainAccess,\n auto: \"Automatic imports browser cookies.\",\n manual: \"Paste a Cookie header or cURL capture from Ollama settings.\",\n off: \"Ollama cookies are disabled.\")\n }\n\n return [\n ProviderSettingsPickerDescriptor(\n id: \"ollama-cookie-source\",\n title: \"Cookie source\",\n subtitle: \"Automatic imports browser cookies.\",\n dynamicSubtitle: cookieSubtitle,\n binding: cookieBinding,\n options: cookieOptions,\n isVisible: nil,\n onChange: nil),\n ]\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n [\n ProviderSettingsFieldDescriptor(\n id: \"ollama-cookie\",\n title: \"\",\n subtitle: \"\",\n kind: .secure,\n placeholder: \"Cookie: …\",\n binding: context.stringBinding(\\.ollamaCookieHeader),\n actions: [\n ProviderSettingsActionDescriptor(\n id: \"ollama-open-settings\",\n title: \"Open Ollama Settings\",\n style: .link,\n isVisible: nil,\n perform: {\n if let url = URL(string: \"https://ollama.com/settings\") {\n NSWorkspace.shared.open(url)\n }\n }),\n ],\n isVisible: { context.settings.ollamaCookieSource == .manual },\n onActivate: { context.settings.ensureOllamaCookieLoaded() }),\n ]\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Ollama/OllamaSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var ollamaCookieHeader: String {\n get { self.configSnapshot.providerConfig(for: .ollama)?.sanitizedCookieHeader ?? \"\" }\n set {\n self.updateProviderConfig(provider: .ollama) { entry in\n entry.cookieHeader = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .ollama, field: \"cookieHeader\", value: newValue)\n }\n }\n\n var ollamaCookieSource: ProviderCookieSource {\n get { self.resolvedCookieSource(provider: .ollama, fallback: .auto) }\n set {\n self.updateProviderConfig(provider: .ollama) { entry in\n entry.cookieSource = newValue\n }\n self.logProviderModeChange(provider: .ollama, field: \"cookieSource\", value: newValue.rawValue)\n }\n }\n\n func ensureOllamaCookieLoaded() {}\n}\n\nextension SettingsStore {\n func ollamaSettingsSnapshot(tokenOverride: TokenAccountOverride?) -> ProviderSettingsSnapshot\n .OllamaProviderSettings {\n ProviderSettingsSnapshot.OllamaProviderSettings(\n cookieSource: self.ollamaSnapshotCookieSource(tokenOverride: tokenOverride),\n manualCookieHeader: self.ollamaSnapshotCookieHeader(tokenOverride: tokenOverride))\n }\n\n private func ollamaSnapshotCookieHeader(tokenOverride: TokenAccountOverride?) -> String {\n let fallback = self.ollamaCookieHeader\n guard let support = TokenAccountSupportCatalog.support(for: .ollama),\n case .cookieHeader = support.injection\n else {\n return fallback\n }\n guard let account = ProviderTokenAccountSelection.selectedAccount(\n provider: .ollama,\n settings: self,\n override: tokenOverride)\n else {\n return fallback\n }\n return TokenAccountSupportCatalog.normalizedCookieHeader(account.token, support: support)\n }\n\n private func ollamaSnapshotCookieSource(tokenOverride: TokenAccountOverride?) -> ProviderCookieSource {\n let fallback = self.ollamaCookieSource\n guard let support = TokenAccountSupportCatalog.support(for: .ollama),\n support.requiresManualCookieSource\n else {\n return fallback\n }\n if self.tokenAccounts(for: .ollama).isEmpty { return fallback }\n return .manual\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/OpenCode/OpenCodeProviderImplementation.swift", "content": "import AppKit\nimport CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\nimport SwiftUI\n\n@ProviderImplementationRegistration\nstruct OpenCodeProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .opencode\n\n @MainActor\n func presentation(context _: ProviderPresentationContext) -> ProviderPresentation {\n ProviderPresentation { _ in \"web\" }\n }\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.opencodeCookieSource\n _ = settings.opencodeCookieHeader\n _ = settings.opencodeWorkspaceID\n }\n\n @MainActor\n func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {\n .opencode(context.settings.opencodeSettingsSnapshot(tokenOverride: context.tokenOverride))\n }\n\n @MainActor\n func tokenAccountsVisibility(context: ProviderSettingsContext, support: TokenAccountSupport) -> Bool {\n guard support.requiresManualCookieSource else { return true }\n if !context.settings.tokenAccounts(for: context.provider).isEmpty { return true }\n return context.settings.opencodeCookieSource == .manual\n }\n\n @MainActor\n func applyTokenAccountCookieSource(settings: SettingsStore) {\n if settings.opencodeCookieSource != .manual {\n settings.opencodeCookieSource = .manual\n }\n }\n\n @MainActor\n func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {\n let cookieBinding = Binding(\n get: { context.settings.opencodeCookieSource.rawValue },\n set: { raw in\n context.settings.opencodeCookieSource = ProviderCookieSource(rawValue: raw) ?? .auto\n })\n let cookieOptions = ProviderCookieSourceUI.options(\n allowsOff: false,\n keychainDisabled: context.settings.debugDisableKeychainAccess)\n\n let cookieSubtitle: () -> String? = {\n ProviderCookieSourceUI.subtitle(\n source: context.settings.opencodeCookieSource,\n keychainDisabled: context.settings.debugDisableKeychainAccess,\n auto: \"Automatic imports browser cookies from opencode.ai.\",\n manual: \"Paste a Cookie header captured from the billing page.\",\n off: \"OpenCode cookies are disabled.\")\n }\n\n return [\n ProviderSettingsPickerDescriptor(\n id: \"opencode-cookie-source\",\n title: \"Cookie source\",\n subtitle: \"Automatic imports browser cookies from opencode.ai.\",\n dynamicSubtitle: cookieSubtitle,\n binding: cookieBinding,\n options: cookieOptions,\n isVisible: nil,\n onChange: nil,\n trailingText: {\n guard let entry = CookieHeaderCache.load(provider: .opencode) else { return nil }\n let when = entry.storedAt.relativeDescription()\n return \"Cached: \\(entry.sourceLabel) • \\(when)\"\n }),\n ]\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n [\n ProviderSettingsFieldDescriptor(\n id: \"opencode-workspace-id\",\n title: \"Workspace ID\",\n subtitle: \"Optional override if workspace lookup fails.\",\n kind: .plain,\n placeholder: \"wrk_…\",\n binding: context.stringBinding(\\.opencodeWorkspaceID),\n actions: [],\n isVisible: nil,\n onActivate: nil),\n ]\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/OpenCode/OpenCodeSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var opencodeWorkspaceID: String {\n get { self.configSnapshot.providerConfig(for: .opencode)?.workspaceID ?? \"\" }\n set {\n let trimmed = newValue.trimmingCharacters(in: .whitespacesAndNewlines)\n let value = trimmed.isEmpty ? nil : trimmed\n self.updateProviderConfig(provider: .opencode) { entry in\n entry.workspaceID = value\n }\n }\n }\n\n var opencodeCookieHeader: String {\n get { self.configSnapshot.providerConfig(for: .opencode)?.sanitizedCookieHeader ?? \"\" }\n set {\n self.updateProviderConfig(provider: .opencode) { entry in\n entry.cookieHeader = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .opencode, field: \"cookieHeader\", value: newValue)\n }\n }\n\n var opencodeCookieSource: ProviderCookieSource {\n get { self.resolvedCookieSource(provider: .opencode, fallback: .auto) }\n set {\n self.updateProviderConfig(provider: .opencode) { entry in\n entry.cookieSource = newValue\n }\n self.logProviderModeChange(provider: .opencode, field: \"cookieSource\", value: newValue.rawValue)\n }\n }\n\n func ensureOpenCodeCookieLoaded() {}\n}\n\nextension SettingsStore {\n func opencodeSettingsSnapshot(tokenOverride: TokenAccountOverride?) -> ProviderSettingsSnapshot\n .OpenCodeProviderSettings {\n ProviderSettingsSnapshot.OpenCodeProviderSettings(\n cookieSource: self.opencodeSnapshotCookieSource(tokenOverride: tokenOverride),\n manualCookieHeader: self.opencodeSnapshotCookieHeader(tokenOverride: tokenOverride),\n workspaceID: self.opencodeWorkspaceID)\n }\n\n private func opencodeSnapshotCookieHeader(tokenOverride: TokenAccountOverride?) -> String {\n let fallback = self.opencodeCookieHeader\n guard let support = TokenAccountSupportCatalog.support(for: .opencode),\n case .cookieHeader = support.injection\n else {\n return fallback\n }\n guard let account = ProviderTokenAccountSelection.selectedAccount(\n provider: .opencode,\n settings: self,\n override: tokenOverride)\n else {\n return fallback\n }\n return TokenAccountSupportCatalog.normalizedCookieHeader(account.token, support: support)\n }\n\n private func opencodeSnapshotCookieSource(tokenOverride: TokenAccountOverride?) -> ProviderCookieSource {\n let fallback = self.opencodeCookieSource\n guard let support = TokenAccountSupportCatalog.support(for: .opencode),\n support.requiresManualCookieSource\n else {\n return fallback\n }\n if self.tokenAccounts(for: .opencode).isEmpty { return fallback }\n return .manual\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/OpenRouter/OpenRouterProviderImplementation.swift", "content": "import AppKit\nimport CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\nimport SwiftUI\n\n@ProviderImplementationRegistration\nstruct OpenRouterProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .openrouter\n\n @MainActor\n func presentation(context _: ProviderPresentationContext) -> ProviderPresentation {\n ProviderPresentation { _ in \"api\" }\n }\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.openRouterAPIToken\n }\n\n @MainActor\n func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {\n _ = context\n return nil\n }\n\n @MainActor\n func isAvailable(context: ProviderAvailabilityContext) -> Bool {\n if OpenRouterSettingsReader.apiToken(environment: context.environment) != nil {\n return true\n }\n return !context.settings.openRouterAPIToken.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty\n }\n\n @MainActor\n func settingsPickers(context _: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {\n []\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n [\n ProviderSettingsFieldDescriptor(\n id: \"openrouter-api-key\",\n title: \"API key\",\n subtitle: \"Stored in ~/.codexbar/config.json. \"\n + \"Get your key from openrouter.ai/settings/keys and set a key spending limit \"\n + \"there to enable API key quota tracking.\",\n kind: .secure,\n placeholder: \"sk-or-v1-...\",\n binding: context.stringBinding(\\.openRouterAPIToken),\n actions: [],\n isVisible: nil,\n onActivate: nil),\n ]\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/OpenRouter/OpenRouterSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var openRouterAPIToken: String {\n get { self.configSnapshot.providerConfig(for: .openrouter)?.sanitizedAPIKey ?? \"\" }\n set {\n self.updateProviderConfig(provider: .openrouter) { entry in\n entry.apiKey = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .openrouter, field: \"apiKey\", value: newValue)\n }\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Shared/ProviderCatalog.swift", "content": "import CodexBarCore\n\n/// Source of truth for app-side provider implementations.\n///\n/// Keep provider registration centralized here. The rest of the app should *not* have to be updated when a new\n/// provider is added, aside from enum/metadata work in `CodexBarCore`.\nenum ProviderCatalog {\n /// All provider implementations shipped in the app.\n static let all: [any ProviderImplementation] = ProviderImplementationRegistry.all\n\n /// Lookup for a single provider implementation.\n static func implementation(for id: UsageProvider) -> (any ProviderImplementation)? {\n ProviderImplementationRegistry.implementation(for: id)\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Shared/ProviderContext.swift", "content": "import CodexBarCore\nimport Foundation\n\nstruct ProviderPresentationContext {\n let provider: UsageProvider\n let settings: SettingsStore\n let store: UsageStore\n let metadata: ProviderMetadata\n}\n\nstruct ProviderAvailabilityContext {\n let provider: UsageProvider\n let settings: SettingsStore\n let environment: [String: String]\n}\n\nstruct ProviderSourceLabelContext {\n let provider: UsageProvider\n let settings: SettingsStore\n let store: UsageStore\n let descriptor: ProviderDescriptor\n}\n\nstruct ProviderSourceModeContext {\n let provider: UsageProvider\n let settings: SettingsStore\n}\n\nstruct ProviderVersionContext {\n let provider: UsageProvider\n let browserDetection: BrowserDetection\n}\n\nstruct ProviderSettingsSnapshotContext {\n let settings: SettingsStore\n let tokenOverride: TokenAccountOverride?\n}\n" }, { "path": "Sources/CodexBar/Providers/Shared/ProviderCookieSourceUI.swift", "content": "import CodexBarCore\n\nenum ProviderCookieSourceUI {\n static let keychainDisabledPrefix =\n \"Keychain access is disabled in Advanced, so browser cookie import is unavailable.\"\n\n static func options(allowsOff: Bool, keychainDisabled: Bool) -> [ProviderSettingsPickerOption] {\n var options: [ProviderSettingsPickerOption] = []\n if !keychainDisabled {\n options.append(ProviderSettingsPickerOption(\n id: ProviderCookieSource.auto.rawValue,\n title: ProviderCookieSource.auto.displayName))\n }\n options.append(ProviderSettingsPickerOption(\n id: ProviderCookieSource.manual.rawValue,\n title: ProviderCookieSource.manual.displayName))\n if allowsOff {\n options.append(ProviderSettingsPickerOption(\n id: ProviderCookieSource.off.rawValue,\n title: ProviderCookieSource.off.displayName))\n }\n return options\n }\n\n static func subtitle(\n source: ProviderCookieSource,\n keychainDisabled: Bool,\n auto: String,\n manual: String,\n off: String) -> String\n {\n if keychainDisabled {\n return source == .off ? off : \"\\(self.keychainDisabledPrefix) \\(manual)\"\n }\n switch source {\n case .auto:\n return auto\n case .manual:\n return manual\n case .off:\n return off\n }\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Shared/ProviderImplementation.swift", "content": "import CodexBarCore\nimport Foundation\n\n/// App-side provider implementation.\n///\n/// Rules:\n/// - Provider implementations return *data/behavior descriptors*; the app owns UI.\n/// - Do not mix identity fields across providers (email/org/plan/loginMethod stays siloed).\nprotocol ProviderImplementation: Sendable {\n var id: UsageProvider { get }\n var supportsLoginFlow: Bool { get }\n\n @MainActor\n func presentation(context: ProviderPresentationContext) -> ProviderPresentation\n\n @MainActor\n func observeSettings(_ settings: SettingsStore)\n\n @MainActor\n func isAvailable(context: ProviderAvailabilityContext) -> Bool\n\n @MainActor\n func defaultSourceLabel(context: ProviderSourceLabelContext) -> String?\n\n @MainActor\n func decorateSourceLabel(context: ProviderSourceLabelContext, baseLabel: String) -> String\n\n @MainActor\n func sourceMode(context: ProviderSourceModeContext) -> ProviderSourceMode\n\n func detectVersion(context: ProviderVersionContext) async -> String?\n\n func makeRuntime() -> (any ProviderRuntime)?\n\n /// Optional provider-specific settings toggles to render in the Providers pane.\n ///\n /// Important: Providers must not return custom SwiftUI views here. Only shared toggle/action descriptors.\n @MainActor\n func settingsToggles(context: ProviderSettingsContext) -> [ProviderSettingsToggleDescriptor]\n\n /// Optional provider-specific settings fields to render in the Providers pane.\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor]\n\n /// Optional provider-specific settings pickers to render in the Providers pane.\n @MainActor\n func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor]\n\n /// Optional visibility gate for token account settings.\n @MainActor\n func tokenAccountsVisibility(context: ProviderSettingsContext, support: TokenAccountSupport) -> Bool\n\n /// Optional provider-specific settings snapshot contribution.\n @MainActor\n func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution?\n\n /// Optional hook to update provider settings when token accounts change.\n @MainActor\n func applyTokenAccountCookieSource(settings: SettingsStore)\n\n /// Optional provider-specific menu entries for the usage section.\n @MainActor\n func appendUsageMenuEntries(context: ProviderMenuUsageContext, entries: inout [ProviderMenuEntry])\n\n /// Optional provider-specific menu entries for the actions section.\n @MainActor\n func appendActionMenuEntries(context: ProviderMenuActionContext, entries: inout [ProviderMenuEntry])\n\n /// Optional override for the login/switch account menu action.\n @MainActor\n func loginMenuAction(context: ProviderMenuLoginContext) -> (label: String, action: MenuDescriptor.MenuAction)?\n\n /// Optional provider-specific login flow. Returns whether to refresh after completion.\n @MainActor\n func runLoginFlow(context: ProviderLoginContext) async -> Bool\n}\n\nextension ProviderImplementation {\n var supportsLoginFlow: Bool {\n false\n }\n\n @MainActor\n func presentation(context _: ProviderPresentationContext) -> ProviderPresentation {\n ProviderPresentation(detailLine: ProviderPresentation.standardDetailLine)\n }\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings\n }\n\n @MainActor\n func isAvailable(context _: ProviderAvailabilityContext) -> Bool {\n true\n }\n\n @MainActor\n func defaultSourceLabel(context _: ProviderSourceLabelContext) -> String? {\n nil\n }\n\n @MainActor\n func decorateSourceLabel(context _: ProviderSourceLabelContext, baseLabel: String) -> String {\n baseLabel\n }\n\n @MainActor\n func sourceMode(context _: ProviderSourceModeContext) -> ProviderSourceMode {\n .auto\n }\n\n func detectVersion(context: ProviderVersionContext) async -> String? {\n let detector = ProviderDescriptorRegistry.descriptor(for: self.id).cli.versionDetector\n return detector?(context.browserDetection)\n }\n\n func makeRuntime() -> (any ProviderRuntime)? {\n nil\n }\n\n @MainActor\n func settingsToggles(context _: ProviderSettingsContext) -> [ProviderSettingsToggleDescriptor] {\n []\n }\n\n @MainActor\n func settingsFields(context _: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n []\n }\n\n @MainActor\n func settingsPickers(context _: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {\n []\n }\n\n @MainActor\n func tokenAccountsVisibility(context: ProviderSettingsContext, support: TokenAccountSupport) -> Bool {\n guard support.requiresManualCookieSource else { return true }\n return !context.settings.tokenAccounts(for: context.provider).isEmpty\n }\n\n @MainActor\n func settingsSnapshot(context _: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {\n nil\n }\n\n @MainActor\n func applyTokenAccountCookieSource(settings _: SettingsStore) {}\n\n @MainActor\n func appendUsageMenuEntries(context _: ProviderMenuUsageContext, entries _: inout [ProviderMenuEntry]) {}\n\n @MainActor\n func appendActionMenuEntries(context _: ProviderMenuActionContext, entries _: inout [ProviderMenuEntry]) {}\n\n @MainActor\n func loginMenuAction(context _: ProviderMenuLoginContext)\n -> (label: String, action: MenuDescriptor.MenuAction)?\n {\n nil\n }\n\n @MainActor\n func runLoginFlow(context _: ProviderLoginContext) async -> Bool {\n false\n }\n}\n\nstruct ProviderLoginContext {\n unowned let controller: StatusItemController\n}\n" }, { "path": "Sources/CodexBar/Providers/Shared/ProviderImplementationRegistry.swift", "content": "import CodexBarCore\nimport Foundation\n\nenum ProviderImplementationRegistry {\n private final class Store: @unchecked Sendable {\n var ordered: [any ProviderImplementation] = []\n var byID: [UsageProvider: any ProviderImplementation] = [:]\n }\n\n private static let lock = NSLock()\n private static let store = Store()\n\n // swiftlint:disable:next cyclomatic_complexity\n private static func makeImplementation(for provider: UsageProvider) -> (any ProviderImplementation) {\n switch provider {\n case .codex: CodexProviderImplementation()\n case .claude: ClaudeProviderImplementation()\n case .cursor: CursorProviderImplementation()\n case .opencode: OpenCodeProviderImplementation()\n case .alibaba: AlibabaCodingPlanProviderImplementation()\n case .factory: FactoryProviderImplementation()\n case .gemini: GeminiProviderImplementation()\n case .antigravity: AntigravityProviderImplementation()\n case .copilot: CopilotProviderImplementation()\n case .zai: ZaiProviderImplementation()\n case .minimax: MiniMaxProviderImplementation()\n case .kimi: KimiProviderImplementation()\n case .kilo: KiloProviderImplementation()\n case .kiro: KiroProviderImplementation()\n case .vertexai: VertexAIProviderImplementation()\n case .augment: AugmentProviderImplementation()\n case .jetbrains: JetBrainsProviderImplementation()\n case .kimik2: KimiK2ProviderImplementation()\n case .amp: AmpProviderImplementation()\n case .ollama: OllamaProviderImplementation()\n case .synthetic: SyntheticProviderImplementation()\n case .openrouter: OpenRouterProviderImplementation()\n case .warp: WarpProviderImplementation()\n }\n }\n\n private static let bootstrap: Void = {\n for provider in UsageProvider.allCases {\n _ = ProviderImplementationRegistry.register(makeImplementation(for: provider))\n }\n }()\n\n private static func ensureBootstrapped() {\n _ = self.bootstrap\n }\n\n @discardableResult\n static func register(_ implementation: any ProviderImplementation) -> any ProviderImplementation {\n self.lock.lock()\n defer { self.lock.unlock() }\n if self.store.byID[implementation.id] == nil {\n self.store.ordered.append(implementation)\n }\n self.store.byID[implementation.id] = implementation\n return implementation\n }\n\n static var all: [any ProviderImplementation] {\n self.ensureBootstrapped()\n self.lock.lock()\n defer { self.lock.unlock() }\n return self.store.ordered\n }\n\n static func implementation(for id: UsageProvider) -> (any ProviderImplementation)? {\n self.ensureBootstrapped()\n if let found = self.store.byID[id] { return found }\n return self.all.first(where: { $0.id == id })\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Shared/ProviderLoginFlow.swift", "content": "import AppKit\nimport CodexBarCore\n\n@MainActor\nextension StatusItemController {\n /// Runs the provider-specific login flow.\n /// - Returns: Whether CodexBar should refresh after the flow completes.\n func runLoginFlow(provider: UsageProvider) async -> Bool {\n guard let impl = ProviderCatalog.implementation(for: provider) else { return false }\n return await impl.runLoginFlow(context: ProviderLoginContext(controller: self))\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Shared/ProviderMenuContext.swift", "content": "import CodexBarCore\nimport Foundation\n\ntypealias ProviderMenuEntry = MenuDescriptor.Entry\n\nstruct ProviderMenuUsageContext {\n let provider: UsageProvider\n let store: UsageStore\n let settings: SettingsStore\n let metadata: ProviderMetadata\n let snapshot: UsageSnapshot?\n}\n\nstruct ProviderMenuActionContext {\n let provider: UsageProvider\n let store: UsageStore\n let settings: SettingsStore\n let account: AccountInfo\n}\n\nstruct ProviderMenuLoginContext {\n let provider: UsageProvider\n let store: UsageStore\n let settings: SettingsStore\n let account: AccountInfo\n}\n" }, { "path": "Sources/CodexBar/Providers/Shared/ProviderPresentation.swift", "content": "import CodexBarCore\nimport Foundation\n\nstruct ProviderPresentation {\n let detailLine: @MainActor (ProviderPresentationContext) -> String\n\n @MainActor\n static func standardDetailLine(context: ProviderPresentationContext) -> String {\n let versionText = context.store.version(for: context.provider) ?? \"not detected\"\n return \"\\(context.metadata.cliName) \\(versionText)\"\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Shared/ProviderRuntime.swift", "content": "import CodexBarCore\nimport Foundation\n\nstruct ProviderRuntimeContext {\n let provider: UsageProvider\n let settings: SettingsStore\n let store: UsageStore\n}\n\nenum ProviderRuntimeAction {\n case forceSessionRefresh\n case openAIWebAccessToggled(Bool)\n}\n\n@MainActor\nprotocol ProviderRuntime: AnyObject {\n var id: UsageProvider { get }\n\n func start(context: ProviderRuntimeContext)\n func stop(context: ProviderRuntimeContext)\n func settingsDidChange(context: ProviderRuntimeContext)\n func providerDidRefresh(context: ProviderRuntimeContext, provider: UsageProvider)\n func providerDidFail(context: ProviderRuntimeContext, provider: UsageProvider, error: Error)\n func perform(action: ProviderRuntimeAction, context: ProviderRuntimeContext) async\n}\n\nextension ProviderRuntime {\n func start(context _: ProviderRuntimeContext) {}\n func stop(context _: ProviderRuntimeContext) {}\n func settingsDidChange(context _: ProviderRuntimeContext) {}\n func providerDidRefresh(context _: ProviderRuntimeContext, provider _: UsageProvider) {}\n func providerDidFail(context _: ProviderRuntimeContext, provider _: UsageProvider, error _: Error) {}\n func perform(action _: ProviderRuntimeAction, context _: ProviderRuntimeContext) async {}\n}\n" }, { "path": "Sources/CodexBar/Providers/Shared/ProviderSettingsDescriptors.swift", "content": "import CodexBarCore\nimport Foundation\nimport SwiftUI\n\n/// Settings UI context passed to provider implementations.\n///\n/// Providers use this to:\n/// - bind to `SettingsStore` values\n/// - read current provider state from `UsageStore`\n/// - surface transient status text (e.g. \"Importing cookies…\")\n/// - request a shared confirmation alert (no provider-specific UI)\n@MainActor\nstruct ProviderSettingsContext {\n let provider: UsageProvider\n let settings: SettingsStore\n let store: UsageStore\n\n let boolBinding: (ReferenceWritableKeyPath) -> Binding\n let stringBinding: (ReferenceWritableKeyPath) -> Binding\n\n let statusText: (String) -> String?\n let setStatusText: (String, String?) -> Void\n\n let lastAppActiveRunAt: (String) -> Date?\n let setLastAppActiveRunAt: (String, Date?) -> Void\n\n let requestConfirmation: (ProviderSettingsConfirmation) -> Void\n}\n\n/// Shared confirmation alert descriptor.\n///\n/// Providers can request confirmations (e.g. permission prompts) without supplying custom UI.\n@MainActor\nstruct ProviderSettingsConfirmation {\n let title: String\n let message: String\n let confirmTitle: String\n let onConfirm: () -> Void\n}\n\n/// Shared toggle descriptor rendered in the Providers settings pane.\n@MainActor\nstruct ProviderSettingsToggleDescriptor: Identifiable {\n let id: String\n let title: String\n let subtitle: String\n let binding: Binding\n\n /// Optional short status text shown under the toggle when enabled.\n let statusText: (() -> String?)?\n\n /// Optional actions shown under the toggle when enabled.\n let actions: [ProviderSettingsActionDescriptor]\n\n /// Optional runtime visibility gate.\n let isVisible: (() -> Bool)?\n\n /// Called whenever the toggle changes.\n let onChange: ((_ enabled: Bool) async -> Void)?\n\n /// Called when the app becomes active (used for \"retry after permission grant\" flows).\n let onAppDidBecomeActive: (() async -> Void)?\n\n /// Called when the view appears while the toggle is enabled.\n let onAppearWhenEnabled: (() async -> Void)?\n}\n\n/// Shared text field descriptor rendered in the Providers settings pane.\n@MainActor\nstruct ProviderSettingsFieldDescriptor: Identifiable {\n enum Kind {\n case plain\n case secure\n }\n\n let id: String\n let title: String\n let subtitle: String\n let kind: Kind\n let placeholder: String?\n let binding: Binding\n let actions: [ProviderSettingsActionDescriptor]\n let isVisible: (() -> Bool)?\n let onActivate: (() -> Void)?\n}\n\n/// Shared token account descriptor rendered in the Providers settings pane.\n@MainActor\nstruct ProviderSettingsTokenAccountsDescriptor: Identifiable {\n let id: String\n let title: String\n let subtitle: String\n let placeholder: String\n let provider: UsageProvider\n let isVisible: (() -> Bool)?\n let accounts: () -> [ProviderTokenAccount]\n let activeIndex: () -> Int\n let setActiveIndex: (Int) -> Void\n let addAccount: (_ label: String, _ token: String) -> Void\n let removeAccount: (_ accountID: UUID) -> Void\n let openConfigFile: () -> Void\n let reloadFromDisk: () -> Void\n}\n\n/// Shared picker descriptor rendered in the Providers settings pane.\n@MainActor\nstruct ProviderSettingsPickerDescriptor: Identifiable {\n let id: String\n let title: String\n let subtitle: String\n let dynamicSubtitle: (() -> String?)?\n let binding: Binding\n let options: [ProviderSettingsPickerOption]\n let isVisible: (() -> Bool)?\n let isEnabled: (() -> Bool)?\n let onChange: ((_ selection: String) async -> Void)?\n let trailingText: (() -> String?)?\n\n init(\n id: String,\n title: String,\n subtitle: String,\n dynamicSubtitle: (() -> String?)? = nil,\n binding: Binding,\n options: [ProviderSettingsPickerOption],\n isVisible: (() -> Bool)?,\n isEnabled: (() -> Bool)? = nil,\n onChange: ((_ selection: String) async -> Void)?,\n trailingText: (() -> String?)? = nil)\n {\n self.id = id\n self.title = title\n self.subtitle = subtitle\n self.dynamicSubtitle = dynamicSubtitle\n self.binding = binding\n self.options = options\n self.isVisible = isVisible\n self.isEnabled = isEnabled\n self.onChange = onChange\n self.trailingText = trailingText\n }\n}\n\nstruct ProviderSettingsPickerOption: Identifiable {\n let id: String\n let title: String\n}\n\n/// Shared action descriptor rendered under a settings toggle.\n@MainActor\nstruct ProviderSettingsActionDescriptor: Identifiable {\n enum Style {\n case bordered\n case link\n }\n\n let id: String\n let title: String\n let style: Style\n let isVisible: (() -> Bool)?\n let perform: () async -> Void\n}\n" }, { "path": "Sources/CodexBar/Providers/Shared/ProviderTokenAccountSelection.swift", "content": "import CodexBarCore\nimport Foundation\n\nstruct TokenAccountOverride {\n let provider: UsageProvider\n let account: ProviderTokenAccount\n}\n\nenum ProviderTokenAccountSelection {\n @MainActor\n static func selectedAccount(\n provider: UsageProvider,\n settings: SettingsStore,\n override: TokenAccountOverride?) -> ProviderTokenAccount?\n {\n if let override, override.provider == provider { return override.account }\n return settings.selectedTokenAccount(for: provider)\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Shared/SystemSettingsLinks.swift", "content": "import AppKit\nimport Foundation\n\nenum SystemSettingsLinks {\n /// Opens System Settings → Privacy & Security → Full Disk Access (best effort).\n static func openFullDiskAccess() {\n // Best-effort deep link. On older betas it sometimes opened the wrong pane; on modern macOS this is stable.\n let urls: [URL] = [\n URL(string: \"x-apple.systempreferences:com.apple.preference.security?Privacy_AllFiles\"),\n URL(string: \"x-apple.systempreferences:com.apple.preference.security?Privacy\"),\n URL(string: \"x-apple.systempreferences:com.apple.preference.security\"),\n ].compactMap(\\.self)\n\n for url in urls where NSWorkspace.shared.open(url) {\n return\n }\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Synthetic/SyntheticProviderImplementation.swift", "content": "import AppKit\nimport CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\n\n@ProviderImplementationRegistration\nstruct SyntheticProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .synthetic\n\n @MainActor\n func presentation(context _: ProviderPresentationContext) -> ProviderPresentation {\n ProviderPresentation { _ in \"api\" }\n }\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.syntheticAPIToken\n }\n\n @MainActor\n func isAvailable(context: ProviderAvailabilityContext) -> Bool {\n if SyntheticSettingsReader.apiKey(environment: context.environment) != nil {\n return true\n }\n context.settings.ensureSyntheticAPITokenLoaded()\n return !context.settings.syntheticAPIToken.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n [\n ProviderSettingsFieldDescriptor(\n id: \"synthetic-api-key\",\n title: \"API key\",\n subtitle: \"Stored in ~/.codexbar/config.json. Paste the key from the Synthetic dashboard.\",\n kind: .secure,\n placeholder: \"Paste key…\",\n binding: context.stringBinding(\\.syntheticAPIToken),\n actions: [],\n isVisible: nil,\n onActivate: { context.settings.ensureSyntheticAPITokenLoaded() }),\n ]\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Synthetic/SyntheticSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var syntheticAPIToken: String {\n get { self.configSnapshot.providerConfig(for: .synthetic)?.sanitizedAPIKey ?? \"\" }\n set {\n self.updateProviderConfig(provider: .synthetic) { entry in\n entry.apiKey = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .synthetic, field: \"apiKey\", value: newValue)\n }\n }\n\n func ensureSyntheticAPITokenLoaded() {}\n}\n" }, { "path": "Sources/CodexBar/Providers/VertexAI/VertexAILoginFlow.swift", "content": "import AppKit\nimport CodexBarCore\nimport Foundation\n\n@MainActor\nextension StatusItemController {\n func runVertexAILoginFlow() async {\n // Show alert with instructions\n let alert = NSAlert()\n alert.messageText = \"Vertex AI Login\"\n alert.informativeText = \"\"\"\n To use Vertex AI tracking, you need to authenticate with Google Cloud.\n\n 1. Open Terminal\n 2. Run: gcloud auth application-default login\n 3. Follow the browser prompts to sign in\n 4. Set your project: gcloud config set project PROJECT_ID\n\n Would you like to open Terminal now?\n \"\"\"\n alert.alertStyle = .informational\n alert.addButton(withTitle: \"Open Terminal\")\n alert.addButton(withTitle: \"Cancel\")\n\n let response = alert.runModal()\n\n if response == .alertFirstButtonReturn {\n Self.openTerminalWithGcloudCommand()\n }\n\n // Refresh after user may have logged in\n self.loginPhase = .idle\n Task { @MainActor in\n try? await Task.sleep(for: .seconds(2))\n await self.store.refresh()\n }\n }\n\n private static func openTerminalWithGcloudCommand() {\n let script = \"\"\"\n tell application \"Terminal\"\n activate\n do script \"gcloud auth application-default login --scopes=openid,https://www.googleapis.com/auth/userinfo.email,https://www.googleapis.com/auth/cloud-platform\"\n end tell\n \"\"\"\n\n if let appleScript = NSAppleScript(source: script) {\n var error: NSDictionary?\n appleScript.executeAndReturnError(&error)\n if let error {\n CodexBarLog.logger(LogCategories.terminal).error(\n \"Failed to open Terminal\",\n metadata: [\"error\": String(describing: error)])\n }\n }\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/VertexAI/VertexAIProviderImplementation.swift", "content": "import CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\n\n@ProviderImplementationRegistration\nstruct VertexAIProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .vertexai\n let supportsLoginFlow: Bool = true\n\n @MainActor\n func runLoginFlow(context: ProviderLoginContext) async -> Bool {\n await context.controller.runVertexAILoginFlow()\n return false\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Warp/WarpProviderImplementation.swift", "content": "import AppKit\nimport CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\n\n@ProviderImplementationRegistration\nstruct WarpProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .warp\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.warpAPIToken\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n [\n ProviderSettingsFieldDescriptor(\n id: \"warp-api-token\",\n title: \"API key\",\n subtitle: \"Stored in ~/.codexbar/config.json. In Warp, open Settings > Platform > API Keys, \"\n + \"then create one.\",\n kind: .secure,\n placeholder: \"wk-...\",\n binding: context.stringBinding(\\.warpAPIToken),\n actions: [\n ProviderSettingsActionDescriptor(\n id: \"warp-open-api-keys\",\n title: \"Open Warp API Key Guide\",\n style: .link,\n isVisible: nil,\n perform: {\n if let url = URL(string: \"https://docs.warp.dev/reference/cli/api-keys\") {\n NSWorkspace.shared.open(url)\n }\n }),\n ],\n isVisible: nil,\n onActivate: nil),\n ]\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Warp/WarpSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var warpAPIToken: String {\n get { self.configSnapshot.providerConfig(for: .warp)?.sanitizedAPIKey ?? \"\" }\n set {\n self.updateProviderConfig(provider: .warp) { entry in\n entry.apiKey = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .warp, field: \"apiKey\", value: newValue)\n }\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Zai/ZaiProviderImplementation.swift", "content": "import AppKit\nimport CodexBarCore\nimport CodexBarMacroSupport\nimport Foundation\nimport SwiftUI\n\n@ProviderImplementationRegistration\nstruct ZaiProviderImplementation: ProviderImplementation {\n let id: UsageProvider = .zai\n\n @MainActor\n func presentation(context _: ProviderPresentationContext) -> ProviderPresentation {\n ProviderPresentation { _ in \"api\" }\n }\n\n @MainActor\n func observeSettings(_ settings: SettingsStore) {\n _ = settings.zaiAPIToken\n _ = settings.zaiAPIRegion\n }\n\n @MainActor\n func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {\n _ = context\n return .zai(context.settings.zaiSettingsSnapshot())\n }\n\n @MainActor\n func isAvailable(context: ProviderAvailabilityContext) -> Bool {\n if ZaiSettingsReader.apiToken(environment: context.environment) != nil {\n return true\n }\n context.settings.ensureZaiAPITokenLoaded()\n return !context.settings.zaiAPIToken.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty\n }\n\n @MainActor\n func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {\n let binding = Binding(\n get: { context.settings.zaiAPIRegion.rawValue },\n set: { raw in\n context.settings.zaiAPIRegion = ZaiAPIRegion(rawValue: raw) ?? .global\n })\n let options = ZaiAPIRegion.allCases.map {\n ProviderSettingsPickerOption(id: $0.rawValue, title: $0.displayName)\n }\n\n return [\n ProviderSettingsPickerDescriptor(\n id: \"zai-api-region\",\n title: \"API region\",\n subtitle: \"Use BigModel for the China mainland endpoints (open.bigmodel.cn).\",\n binding: binding,\n options: options,\n isVisible: nil,\n onChange: nil),\n ]\n }\n\n @MainActor\n func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {\n _ = context\n return []\n }\n}\n" }, { "path": "Sources/CodexBar/Providers/Zai/ZaiSettingsStore.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n var zaiAPIRegion: ZaiAPIRegion {\n get {\n let raw = self.configSnapshot.providerConfig(for: .zai)?.region\n return ZaiAPIRegion(rawValue: raw ?? \"\") ?? .global\n }\n set {\n self.updateProviderConfig(provider: .zai) { entry in\n entry.region = newValue.rawValue\n }\n }\n }\n\n var zaiAPIToken: String {\n get { self.configSnapshot.providerConfig(for: .zai)?.sanitizedAPIKey ?? \"\" }\n set {\n self.updateProviderConfig(provider: .zai) { entry in\n entry.apiKey = self.normalizedConfigValue(newValue)\n }\n self.logSecretUpdate(provider: .zai, field: \"apiKey\", value: newValue)\n }\n }\n\n func ensureZaiAPITokenLoaded() {}\n}\n\nextension SettingsStore {\n func zaiSettingsSnapshot() -> ProviderSettingsSnapshot.ZaiProviderSettings {\n ProviderSettingsSnapshot.ZaiProviderSettings(apiRegion: self.zaiAPIRegion)\n }\n}\n" }, { "path": "Sources/CodexBar/SessionQuotaNotifications.swift", "content": "import CodexBarCore\nimport Foundation\n@preconcurrency import UserNotifications\n\nenum SessionQuotaTransition: Equatable {\n case none\n case depleted\n case restored\n}\n\nenum SessionQuotaNotificationLogic {\n static let depletedThreshold: Double = 0.0001\n\n static func isDepleted(_ remaining: Double?) -> Bool {\n guard let remaining else { return false }\n return remaining <= Self.depletedThreshold\n }\n\n static func transition(previousRemaining: Double?, currentRemaining: Double?) -> SessionQuotaTransition {\n guard let currentRemaining else { return .none }\n guard let previousRemaining else { return .none }\n\n let wasDepleted = previousRemaining <= Self.depletedThreshold\n let isDepleted = currentRemaining <= Self.depletedThreshold\n\n if !wasDepleted, isDepleted { return .depleted }\n if wasDepleted, !isDepleted { return .restored }\n return .none\n }\n}\n\n@MainActor\nprotocol SessionQuotaNotifying: AnyObject {\n func post(transition: SessionQuotaTransition, provider: UsageProvider, badge: NSNumber?)\n}\n\n@MainActor\nfinal class SessionQuotaNotifier: SessionQuotaNotifying {\n private let logger = CodexBarLog.logger(LogCategories.sessionQuotaNotifications)\n\n init() {}\n\n func post(transition: SessionQuotaTransition, provider: UsageProvider, badge: NSNumber? = nil) {\n guard transition != .none else { return }\n\n let providerName = ProviderDescriptorRegistry.descriptor(for: provider).metadata.displayName\n\n let (title, body) = switch transition {\n case .none:\n (\"\", \"\")\n case .depleted:\n (\"\\(providerName) session depleted\", \"0% left. Will notify when it's available again.\")\n case .restored:\n (\"\\(providerName) session restored\", \"Session quota is available again.\")\n }\n\n let providerText = provider.rawValue\n let transitionText = String(describing: transition)\n let idPrefix = \"session-\\(providerText)-\\(transitionText)\"\n self.logger.info(\"enqueuing\", metadata: [\"prefix\": idPrefix])\n AppNotifications.shared.post(idPrefix: idPrefix, title: title, body: body, badge: badge)\n }\n}\n" }, { "path": "Sources/CodexBar/SettingsStore+Config.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n func providerConfig(for provider: UsageProvider) -> ProviderConfig? {\n self.configSnapshot.providerConfig(for: provider)\n }\n\n var tokenAccountsByProvider: [UsageProvider: ProviderTokenAccountData] {\n get {\n Dictionary(uniqueKeysWithValues: self.configSnapshot.providers.compactMap { entry in\n guard let accounts = entry.tokenAccounts else { return nil }\n return (entry.id, accounts)\n })\n }\n set {\n self.updateProviderTokenAccounts(newValue)\n }\n }\n}\n\nextension SettingsStore {\n func resolvedCookieSource(\n provider: UsageProvider,\n fallback: ProviderCookieSource) -> ProviderCookieSource\n {\n let source = self.configSnapshot.providerConfig(for: provider)?.cookieSource ?? fallback\n guard self.debugDisableKeychainAccess == false else { return source == .off ? .off : .manual }\n return source\n }\n\n func logProviderModeChange(provider: UsageProvider, field: String, value: String) {\n CodexBarLog.logger(LogCategories.settings).info(\n \"Provider mode updated\",\n metadata: [\"provider\": provider.rawValue, \"field\": field, \"value\": value])\n }\n\n func logSecretUpdate(provider: UsageProvider, field: String, value: String) {\n var metadata = LogMetadata.secretSummary(value)\n metadata[\"provider\"] = provider.rawValue\n metadata[\"field\"] = field\n CodexBarLog.logger(LogCategories.settings).info(\n \"Provider secret updated\",\n metadata: metadata)\n }\n}\n" }, { "path": "Sources/CodexBar/SettingsStore+ConfigPersistence.swift", "content": "import CodexBarCore\nimport Foundation\n\nprivate enum ConfigChangeOrigin {\n case localUser\n case externalSync\n case reload\n}\n\nprivate struct ConfigChangeContext {\n let origin: ConfigChangeOrigin\n let reason: String\n\n static func local(reason: String) -> Self {\n Self(origin: .localUser, reason: reason)\n }\n\n static func external(reason: String) -> Self {\n Self(origin: .externalSync, reason: reason)\n }\n\n static func reload(reason: String) -> Self {\n Self(origin: .reload, reason: reason)\n }\n\n var shouldBroadcast: Bool {\n switch self.origin {\n case .localUser:\n true\n case .externalSync, .reload:\n false\n }\n }\n}\n\nextension SettingsStore {\n private func updateConfig(reason: String, mutate: (inout CodexBarConfig) -> Void) {\n guard !self.configLoading else { return }\n var config = self.config\n mutate(&config)\n self.config = config.normalized()\n self.updateProviderState(config: self.config)\n self.schedulePersistConfig()\n self.bumpConfigRevision(.local(reason: reason))\n }\n\n func updateProviderConfig(provider: UsageProvider, mutate: (inout ProviderConfig) -> Void) {\n self.updateConfig(reason: \"provider-\\(provider.rawValue)\") { config in\n if let index = config.providers.firstIndex(where: { $0.id == provider }) {\n var entry = config.providers[index]\n mutate(&entry)\n config.providers[index] = entry\n } else {\n var entry = ProviderConfig(id: provider)\n mutate(&entry)\n config.providers.append(entry)\n }\n }\n }\n\n func updateProviderTokenAccounts(_ accounts: [UsageProvider: ProviderTokenAccountData]) {\n let summary = accounts\n .sorted { $0.key.rawValue < $1.key.rawValue }\n .map { \"\\($0.key.rawValue)=\\($0.value.accounts.count)\" }\n .joined(separator: \",\")\n CodexBarLog.logger(LogCategories.tokenAccounts).info(\n \"Token accounts updated\",\n metadata: [\n \"providers\": \"\\(accounts.count)\",\n \"summary\": summary,\n ])\n self.updateConfig(reason: \"token-accounts\") { config in\n var seen: Set = []\n for index in config.providers.indices {\n let provider = config.providers[index].id\n config.providers[index].tokenAccounts = accounts[provider]\n seen.insert(provider)\n }\n for (provider, data) in accounts where !seen.contains(provider) {\n config.providers.append(ProviderConfig(id: provider, tokenAccounts: data))\n }\n }\n }\n\n func setProviderOrder(_ order: [UsageProvider]) {\n self.updateConfig(reason: \"order\") { config in\n let configsByID = Dictionary(uniqueKeysWithValues: config.providers.map { ($0.id, $0) })\n var seen: Set = []\n var ordered: [ProviderConfig] = []\n ordered.reserveCapacity(max(order.count, config.providers.count))\n\n for provider in order {\n guard !seen.contains(provider) else { continue }\n seen.insert(provider)\n ordered.append(configsByID[provider] ?? ProviderConfig(id: provider))\n }\n\n for provider in UsageProvider.allCases where !seen.contains(provider) {\n ordered.append(configsByID[provider] ?? ProviderConfig(id: provider))\n }\n\n config.providers = ordered\n }\n }\n\n func reloadConfig(reason: String) {\n guard !self.configLoading else { return }\n do {\n guard let loaded = try self.configStore.load() else { return }\n self.applyExternalConfig(loaded, reason: \"reload-\\(reason)\")\n } catch {\n CodexBarLog.logger(LogCategories.configStore).error(\"Failed to reload config: \\(error)\")\n }\n }\n\n func applyExternalConfig(_ config: CodexBarConfig, reason: String) {\n guard !self.configLoading else { return }\n self.configLoading = true\n self.config = config\n self.updateProviderState(config: config)\n self.configLoading = false\n self.bumpConfigRevision(.external(reason: \"sync-\\(reason)\"))\n }\n\n private func bumpConfigRevision(_ context: ConfigChangeContext) {\n self.configRevision &+= 1\n CodexBarLog.logger(LogCategories.settings)\n .debug(\"Config revision bumped (\\(context.reason)) -> \\(self.configRevision)\")\n guard context.shouldBroadcast else { return }\n NotificationCenter.default.post(\n name: .codexbarProviderConfigDidChange,\n object: self,\n userInfo: [\n \"config\": self.config,\n \"reason\": context.reason,\n \"revision\": self.configRevision,\n ])\n }\n\n func normalizedConfigValue(_ raw: String) -> String? {\n let trimmed = raw.trimmingCharacters(in: .whitespacesAndNewlines)\n return trimmed.isEmpty ? nil : trimmed\n }\n\n func schedulePersistConfig() {\n guard !self.configLoading else { return }\n self.configPersistTask?.cancel()\n if Self.isRunningTests {\n do {\n try self.configStore.save(self.config)\n } catch {\n CodexBarLog.logger(LogCategories.configStore).error(\"Failed to persist config: \\(error)\")\n }\n return\n }\n let store = self.configStore\n self.configPersistTask = Task { @MainActor in\n do {\n try await Task.sleep(nanoseconds: 350_000_000)\n } catch {\n return\n }\n guard !Task.isCancelled else { return }\n let snapshot = self.config\n let error: (any Error)? = await Task.detached(priority: .utility) {\n do {\n try store.save(snapshot)\n return nil\n } catch {\n return error\n }\n }.value\n if let error {\n CodexBarLog.logger(LogCategories.configStore).error(\"Failed to persist config: \\(error)\")\n }\n }\n }\n}\n" }, { "path": "Sources/CodexBar/SettingsStore+Defaults.swift", "content": "import CodexBarCore\nimport Foundation\nimport ServiceManagement\n\nextension SettingsStore {\n private static let mergedOverviewSelectionEditedActiveProvidersKey = \"mergedOverviewSelectionEditedActiveProviders\"\n\n var refreshFrequency: RefreshFrequency {\n get { self.defaultsState.refreshFrequency }\n set {\n self.defaultsState.refreshFrequency = newValue\n self.userDefaults.set(newValue.rawValue, forKey: \"refreshFrequency\")\n }\n }\n\n var launchAtLogin: Bool {\n get { self.defaultsState.launchAtLogin }\n set {\n self.defaultsState.launchAtLogin = newValue\n self.userDefaults.set(newValue, forKey: \"launchAtLogin\")\n LaunchAtLoginManager.setEnabled(newValue)\n }\n }\n\n var debugMenuEnabled: Bool {\n get { self.defaultsState.debugMenuEnabled }\n set {\n self.defaultsState.debugMenuEnabled = newValue\n self.userDefaults.set(newValue, forKey: \"debugMenuEnabled\")\n }\n }\n\n var debugDisableKeychainAccess: Bool {\n get { self.defaultsState.debugDisableKeychainAccess }\n set {\n self.defaultsState.debugDisableKeychainAccess = newValue\n self.userDefaults.set(newValue, forKey: \"debugDisableKeychainAccess\")\n Self.sharedDefaults?.set(newValue, forKey: \"debugDisableKeychainAccess\")\n KeychainAccessGate.isDisabled = newValue\n }\n }\n\n var debugFileLoggingEnabled: Bool {\n get { self.defaultsState.debugFileLoggingEnabled }\n set {\n self.defaultsState.debugFileLoggingEnabled = newValue\n self.userDefaults.set(newValue, forKey: \"debugFileLoggingEnabled\")\n CodexBarLog.setFileLoggingEnabled(newValue)\n }\n }\n\n var debugLogLevel: CodexBarLog.Level {\n get {\n let raw = self.defaultsState.debugLogLevelRaw\n return CodexBarLog.parseLevel(raw) ?? .verbose\n }\n set {\n self.defaultsState.debugLogLevelRaw = newValue.rawValue\n self.userDefaults.set(newValue.rawValue, forKey: \"debugLogLevel\")\n CodexBarLog.setLogLevel(newValue)\n }\n }\n\n var debugKeepCLISessionsAlive: Bool {\n get { self.defaultsState.debugKeepCLISessionsAlive }\n set {\n self.defaultsState.debugKeepCLISessionsAlive = newValue\n self.userDefaults.set(newValue, forKey: \"debugKeepCLISessionsAlive\")\n }\n }\n\n var isVerboseLoggingEnabled: Bool {\n self.debugLogLevel.rank <= CodexBarLog.Level.verbose.rank\n }\n\n private var debugLoadingPatternRaw: String? {\n get { self.defaultsState.debugLoadingPatternRaw }\n set {\n self.defaultsState.debugLoadingPatternRaw = newValue\n if let raw = newValue {\n self.userDefaults.set(raw, forKey: \"debugLoadingPattern\")\n } else {\n self.userDefaults.removeObject(forKey: \"debugLoadingPattern\")\n }\n }\n }\n\n var statusChecksEnabled: Bool {\n get { self.defaultsState.statusChecksEnabled }\n set {\n self.defaultsState.statusChecksEnabled = newValue\n self.userDefaults.set(newValue, forKey: \"statusChecksEnabled\")\n }\n }\n\n var sessionQuotaNotificationsEnabled: Bool {\n get { self.defaultsState.sessionQuotaNotificationsEnabled }\n set {\n self.defaultsState.sessionQuotaNotificationsEnabled = newValue\n self.userDefaults.set(newValue, forKey: \"sessionQuotaNotificationsEnabled\")\n }\n }\n\n var usageBarsShowUsed: Bool {\n get { self.defaultsState.usageBarsShowUsed }\n set {\n self.defaultsState.usageBarsShowUsed = newValue\n self.userDefaults.set(newValue, forKey: \"usageBarsShowUsed\")\n }\n }\n\n var resetTimesShowAbsolute: Bool {\n get { self.defaultsState.resetTimesShowAbsolute }\n set {\n self.defaultsState.resetTimesShowAbsolute = newValue\n self.userDefaults.set(newValue, forKey: \"resetTimesShowAbsolute\")\n }\n }\n\n var menuBarShowsBrandIconWithPercent: Bool {\n get { self.defaultsState.menuBarShowsBrandIconWithPercent }\n set {\n self.defaultsState.menuBarShowsBrandIconWithPercent = newValue\n self.userDefaults.set(newValue, forKey: \"menuBarShowsBrandIconWithPercent\")\n }\n }\n\n private var menuBarDisplayModeRaw: String? {\n get { self.defaultsState.menuBarDisplayModeRaw }\n set {\n self.defaultsState.menuBarDisplayModeRaw = newValue\n if let raw = newValue {\n self.userDefaults.set(raw, forKey: \"menuBarDisplayMode\")\n } else {\n self.userDefaults.removeObject(forKey: \"menuBarDisplayMode\")\n }\n }\n }\n\n var menuBarDisplayMode: MenuBarDisplayMode {\n get { MenuBarDisplayMode(rawValue: self.menuBarDisplayModeRaw ?? \"\") ?? .percent }\n set { self.menuBarDisplayModeRaw = newValue.rawValue }\n }\n\n var showAllTokenAccountsInMenu: Bool {\n get { self.defaultsState.showAllTokenAccountsInMenu }\n set {\n self.defaultsState.showAllTokenAccountsInMenu = newValue\n self.userDefaults.set(newValue, forKey: \"showAllTokenAccountsInMenu\")\n }\n }\n\n var historicalTrackingEnabled: Bool {\n get { self.defaultsState.historicalTrackingEnabled }\n set {\n self.defaultsState.historicalTrackingEnabled = newValue\n self.userDefaults.set(newValue, forKey: \"historicalTrackingEnabled\")\n }\n }\n\n var menuBarMetricPreferencesRaw: [String: String] {\n get { self.defaultsState.menuBarMetricPreferencesRaw }\n set {\n self.defaultsState.menuBarMetricPreferencesRaw = newValue\n self.userDefaults.set(newValue, forKey: \"menuBarMetricPreferences\")\n }\n }\n\n var costUsageEnabled: Bool {\n get { self.defaultsState.costUsageEnabled }\n set {\n self.defaultsState.costUsageEnabled = newValue\n self.userDefaults.set(newValue, forKey: \"tokenCostUsageEnabled\")\n }\n }\n\n var hidePersonalInfo: Bool {\n get { self.defaultsState.hidePersonalInfo }\n set {\n self.defaultsState.hidePersonalInfo = newValue\n self.userDefaults.set(newValue, forKey: \"hidePersonalInfo\")\n }\n }\n\n var randomBlinkEnabled: Bool {\n get { self.defaultsState.randomBlinkEnabled }\n set {\n self.defaultsState.randomBlinkEnabled = newValue\n self.userDefaults.set(newValue, forKey: \"randomBlinkEnabled\")\n }\n }\n\n var menuBarShowsHighestUsage: Bool {\n get { self.defaultsState.menuBarShowsHighestUsage }\n set {\n self.defaultsState.menuBarShowsHighestUsage = newValue\n self.userDefaults.set(newValue, forKey: \"menuBarShowsHighestUsage\")\n }\n }\n\n var claudeOAuthKeychainPromptMode: ClaudeOAuthKeychainPromptMode {\n get {\n let raw = self.defaultsState.claudeOAuthKeychainPromptModeRaw\n return ClaudeOAuthKeychainPromptMode(rawValue: raw ?? \"\") ?? .onlyOnUserAction\n }\n set {\n self.defaultsState.claudeOAuthKeychainPromptModeRaw = newValue.rawValue\n self.userDefaults.set(newValue.rawValue, forKey: \"claudeOAuthKeychainPromptMode\")\n }\n }\n\n var claudeOAuthKeychainReadStrategy: ClaudeOAuthKeychainReadStrategy {\n get {\n let raw = self.defaultsState.claudeOAuthKeychainReadStrategyRaw\n return ClaudeOAuthKeychainReadStrategy(rawValue: raw ?? \"\") ?? .securityFramework\n }\n set {\n self.defaultsState.claudeOAuthKeychainReadStrategyRaw = newValue.rawValue\n self.userDefaults.set(newValue.rawValue, forKey: \"claudeOAuthKeychainReadStrategy\")\n }\n }\n\n var claudeOAuthPromptFreeCredentialsEnabled: Bool {\n get { self.claudeOAuthKeychainReadStrategy == .securityCLIExperimental }\n set {\n self.claudeOAuthKeychainReadStrategy = newValue\n ? .securityCLIExperimental\n : .securityFramework\n }\n }\n\n var claudeWebExtrasEnabled: Bool {\n get { self.claudeWebExtrasEnabledRaw }\n set { self.claudeWebExtrasEnabledRaw = newValue }\n }\n\n private var claudeWebExtrasEnabledRaw: Bool {\n get { self.defaultsState.claudeWebExtrasEnabledRaw }\n set {\n self.defaultsState.claudeWebExtrasEnabledRaw = newValue\n self.userDefaults.set(newValue, forKey: \"claudeWebExtrasEnabled\")\n CodexBarLog.logger(LogCategories.settings).info(\n \"Claude web extras updated\",\n metadata: [\"enabled\": newValue ? \"1\" : \"0\"])\n }\n }\n\n var showOptionalCreditsAndExtraUsage: Bool {\n get { self.defaultsState.showOptionalCreditsAndExtraUsage }\n set {\n self.defaultsState.showOptionalCreditsAndExtraUsage = newValue\n self.userDefaults.set(newValue, forKey: \"showOptionalCreditsAndExtraUsage\")\n }\n }\n\n var openAIWebAccessEnabled: Bool {\n get { self.defaultsState.openAIWebAccessEnabled }\n set {\n self.defaultsState.openAIWebAccessEnabled = newValue\n self.userDefaults.set(newValue, forKey: \"openAIWebAccessEnabled\")\n CodexBarLog.logger(LogCategories.settings).info(\n \"OpenAI web access updated\",\n metadata: [\"enabled\": newValue ? \"1\" : \"0\"])\n }\n }\n\n var jetbrainsIDEBasePath: String {\n get { self.defaultsState.jetbrainsIDEBasePath }\n set {\n self.defaultsState.jetbrainsIDEBasePath = newValue\n self.userDefaults.set(newValue, forKey: \"jetbrainsIDEBasePath\")\n }\n }\n\n var mergeIcons: Bool {\n get { self.defaultsState.mergeIcons }\n set {\n self.defaultsState.mergeIcons = newValue\n self.userDefaults.set(newValue, forKey: \"mergeIcons\")\n }\n }\n\n var switcherShowsIcons: Bool {\n get { self.defaultsState.switcherShowsIcons }\n set {\n self.defaultsState.switcherShowsIcons = newValue\n self.userDefaults.set(newValue, forKey: \"switcherShowsIcons\")\n }\n }\n\n var mergedMenuLastSelectedWasOverview: Bool {\n get { self.defaultsState.mergedMenuLastSelectedWasOverview }\n set {\n self.defaultsState.mergedMenuLastSelectedWasOverview = newValue\n self.userDefaults.set(newValue, forKey: \"mergedMenuLastSelectedWasOverview\")\n }\n }\n\n private var mergedOverviewSelectedProvidersRaw: [String] {\n get { self.defaultsState.mergedOverviewSelectedProvidersRaw }\n set {\n self.defaultsState.mergedOverviewSelectedProvidersRaw = newValue\n self.userDefaults.set(newValue, forKey: \"mergedOverviewSelectedProviders\")\n }\n }\n\n private var selectedMenuProviderRaw: String? {\n get { self.defaultsState.selectedMenuProviderRaw }\n set {\n self.defaultsState.selectedMenuProviderRaw = newValue\n if let raw = newValue {\n self.userDefaults.set(raw, forKey: \"selectedMenuProvider\")\n } else {\n self.userDefaults.removeObject(forKey: \"selectedMenuProvider\")\n }\n }\n }\n\n var selectedMenuProvider: UsageProvider? {\n get { self.selectedMenuProviderRaw.flatMap(UsageProvider.init(rawValue:)) }\n set {\n self.selectedMenuProviderRaw = newValue?.rawValue\n }\n }\n\n var mergedOverviewSelectedProviders: [UsageProvider] {\n get {\n Self.decodeProviders(\n self.mergedOverviewSelectedProvidersRaw,\n maxCount: Self.mergedOverviewProviderLimit)\n }\n set {\n let normalized = Self.normalizeProviders(newValue, maxCount: Self.mergedOverviewProviderLimit)\n self.mergedOverviewSelectedProvidersRaw = normalized.map(\\.rawValue)\n }\n }\n\n private var hasMergedOverviewSelectionPreference: Bool {\n self.userDefaults.object(forKey: \"mergedOverviewSelectedProviders\") != nil\n }\n\n private var mergedOverviewSelectionEditedActiveProvidersRaw: [String]? {\n get {\n self.userDefaults.array(forKey: Self.mergedOverviewSelectionEditedActiveProvidersKey) as? [String]\n }\n set {\n if let newValue {\n self.userDefaults.set(newValue, forKey: Self.mergedOverviewSelectionEditedActiveProvidersKey)\n } else {\n self.userDefaults.removeObject(forKey: Self.mergedOverviewSelectionEditedActiveProvidersKey)\n }\n }\n }\n\n private func mergedOverviewSelectionApplies(to activeProviders: [UsageProvider]) -> Bool {\n guard let editedRaw = self.mergedOverviewSelectionEditedActiveProvidersRaw else { return false }\n let editedSet = Set(editedRaw)\n let activeSet = Set(Self.normalizeProviders(activeProviders).map(\\.rawValue))\n return editedSet == activeSet\n }\n\n private func markMergedOverviewSelectionEdited(for activeProviders: [UsageProvider]) {\n let signature = Set(Self.normalizeProviders(activeProviders).map(\\.rawValue))\n self.mergedOverviewSelectionEditedActiveProvidersRaw = Array(signature).sorted()\n }\n\n private func clearMergedOverviewSelectionPreference() {\n self.defaultsState.mergedOverviewSelectedProvidersRaw = []\n self.userDefaults.removeObject(forKey: \"mergedOverviewSelectedProviders\")\n self.mergedOverviewSelectionEditedActiveProvidersRaw = nil\n }\n\n func resolvedMergedOverviewProviders(\n activeProviders: [UsageProvider],\n maxVisibleProviders: Int = SettingsStore.mergedOverviewProviderLimit) -> [UsageProvider]\n {\n guard maxVisibleProviders > 0 else { return [] }\n let normalizedActive = Self.normalizeProviders(activeProviders)\n guard self.hasMergedOverviewSelectionPreference else {\n return Array(normalizedActive.prefix(maxVisibleProviders))\n }\n if normalizedActive.count <= maxVisibleProviders,\n !self.mergedOverviewSelectionApplies(to: normalizedActive)\n {\n return normalizedActive\n }\n\n let selectedSet = Set(self.mergedOverviewSelectedProviders)\n return Array(normalizedActive.filter { selectedSet.contains($0) }.prefix(maxVisibleProviders))\n }\n\n @discardableResult\n func reconcileMergedOverviewSelectedProviders(\n activeProviders: [UsageProvider],\n maxVisibleProviders: Int = SettingsStore.mergedOverviewProviderLimit) -> [UsageProvider]\n {\n guard maxVisibleProviders > 0 else {\n self.clearMergedOverviewSelectionPreference()\n return []\n }\n\n let normalizedActive = Self.normalizeProviders(activeProviders)\n if normalizedActive.isEmpty {\n self.clearMergedOverviewSelectionPreference()\n return []\n }\n\n let shouldPersistResolvedSelection = normalizedActive.count > maxVisibleProviders ||\n self.mergedOverviewSelectionApplies(to: normalizedActive)\n\n if self.hasMergedOverviewSelectionPreference, shouldPersistResolvedSelection {\n let selectedSet = Set(self.mergedOverviewSelectedProviders)\n let sanitizedSelection = Array(\n normalizedActive\n .filter { selectedSet.contains($0) }\n .prefix(maxVisibleProviders))\n if sanitizedSelection != self.mergedOverviewSelectedProviders {\n self.mergedOverviewSelectedProviders = sanitizedSelection\n }\n }\n\n return self.resolvedMergedOverviewProviders(\n activeProviders: normalizedActive,\n maxVisibleProviders: maxVisibleProviders)\n }\n\n @discardableResult\n func setMergedOverviewProviderSelection(\n provider: UsageProvider,\n isSelected: Bool,\n activeProviders: [UsageProvider],\n maxVisibleProviders: Int = SettingsStore.mergedOverviewProviderLimit) -> [UsageProvider]\n {\n guard maxVisibleProviders > 0 else {\n self.clearMergedOverviewSelectionPreference()\n return []\n }\n\n let normalizedActive = Self.normalizeProviders(activeProviders)\n guard normalizedActive.contains(provider) else {\n return self.resolvedMergedOverviewProviders(\n activeProviders: normalizedActive,\n maxVisibleProviders: maxVisibleProviders)\n }\n\n let currentSelection = self.resolvedMergedOverviewProviders(\n activeProviders: normalizedActive,\n maxVisibleProviders: maxVisibleProviders)\n var updatedSet = Set(currentSelection)\n\n if isSelected {\n guard updatedSet.contains(provider) || currentSelection.count < maxVisibleProviders else {\n return currentSelection\n }\n updatedSet.insert(provider)\n } else {\n updatedSet.remove(provider)\n }\n\n let updatedSelection = Array(\n normalizedActive\n .filter { updatedSet.contains($0) }\n .prefix(maxVisibleProviders))\n self.mergedOverviewSelectedProviders = updatedSelection\n self.markMergedOverviewSelectionEdited(for: normalizedActive)\n return updatedSelection\n }\n\n var providerDetectionCompleted: Bool {\n get { self.defaultsState.providerDetectionCompleted }\n set {\n self.defaultsState.providerDetectionCompleted = newValue\n self.userDefaults.set(newValue, forKey: \"providerDetectionCompleted\")\n }\n }\n\n var debugLoadingPattern: LoadingPattern? {\n get { self.debugLoadingPatternRaw.flatMap(LoadingPattern.init(rawValue:)) }\n set { self.debugLoadingPatternRaw = newValue?.rawValue }\n }\n}\n\nextension SettingsStore {\n private static func normalizeProviders(_ providers: [UsageProvider], maxCount: Int? = nil) -> [UsageProvider] {\n var seen: Set = []\n var normalized: [UsageProvider] = []\n for provider in providers where !seen.contains(provider) {\n seen.insert(provider)\n normalized.append(provider)\n if let maxCount, normalized.count >= maxCount { break }\n }\n return normalized\n }\n\n private static func decodeProviders(_ rawProviders: [String], maxCount: Int? = nil) -> [UsageProvider] {\n var providers: [UsageProvider] = []\n providers.reserveCapacity(rawProviders.count)\n for raw in rawProviders {\n guard let provider = UsageProvider(rawValue: raw) else { continue }\n providers.append(provider)\n }\n return self.normalizeProviders(providers, maxCount: maxCount)\n }\n}\n" }, { "path": "Sources/CodexBar/SettingsStore+MenuObservation.swift", "content": "import Foundation\n\nextension SettingsStore {\n var menuObservationToken: Int {\n _ = self.providerOrder\n _ = self.providerEnablement\n _ = self.refreshFrequency\n _ = self.launchAtLogin\n _ = self.debugMenuEnabled\n _ = self.debugDisableKeychainAccess\n _ = self.debugKeepCLISessionsAlive\n _ = self.statusChecksEnabled\n _ = self.sessionQuotaNotificationsEnabled\n _ = self.usageBarsShowUsed\n _ = self.resetTimesShowAbsolute\n _ = self.menuBarShowsBrandIconWithPercent\n _ = self.menuBarShowsHighestUsage\n _ = self.menuBarDisplayMode\n _ = self.historicalTrackingEnabled\n _ = self.showAllTokenAccountsInMenu\n _ = self.menuBarMetricPreferencesRaw\n _ = self.costUsageEnabled\n _ = self.hidePersonalInfo\n _ = self.randomBlinkEnabled\n _ = self.claudeOAuthKeychainPromptMode\n _ = self.claudeOAuthKeychainReadStrategy\n _ = self.claudeWebExtrasEnabled\n _ = self.showOptionalCreditsAndExtraUsage\n _ = self.openAIWebAccessEnabled\n _ = self.codexUsageDataSource\n _ = self.claudeUsageDataSource\n _ = self.kiloUsageDataSource\n _ = self.kiloExtrasEnabled\n _ = self.codexCookieSource\n _ = self.claudeCookieSource\n _ = self.cursorCookieSource\n _ = self.opencodeCookieSource\n _ = self.factoryCookieSource\n _ = self.minimaxCookieSource\n _ = self.minimaxAPIRegion\n _ = self.kimiCookieSource\n _ = self.augmentCookieSource\n _ = self.ampCookieSource\n _ = self.ollamaCookieSource\n _ = self.mergeIcons\n _ = self.switcherShowsIcons\n _ = self.mergedMenuLastSelectedWasOverview\n _ = self.mergedOverviewSelectedProviders\n _ = self.zaiAPIToken\n _ = self.syntheticAPIToken\n _ = self.codexCookieHeader\n _ = self.claudeCookieHeader\n _ = self.cursorCookieHeader\n _ = self.opencodeCookieHeader\n _ = self.opencodeWorkspaceID\n _ = self.factoryCookieHeader\n _ = self.minimaxCookieHeader\n _ = self.minimaxAPIToken\n _ = self.kimiManualCookieHeader\n _ = self.kimiK2APIToken\n _ = self.kiloAPIToken\n _ = self.augmentCookieHeader\n _ = self.ampCookieHeader\n _ = self.ollamaCookieHeader\n _ = self.copilotAPIToken\n _ = self.warpAPIToken\n _ = self.tokenAccountsByProvider\n _ = self.debugLoadingPattern\n _ = self.selectedMenuProvider\n _ = self.configRevision\n return 0\n }\n}\n" }, { "path": "Sources/CodexBar/SettingsStore+MenuPreferences.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n func menuBarMetricPreference(for provider: UsageProvider) -> MenuBarMetricPreference {\n if provider == .zai { return .primary }\n if provider == .openrouter {\n let raw = self.menuBarMetricPreferencesRaw[provider.rawValue] ?? \"\"\n let preference = MenuBarMetricPreference(rawValue: raw) ?? .automatic\n switch preference {\n case .automatic, .primary:\n return preference\n case .secondary, .average:\n return .automatic\n }\n }\n let raw = self.menuBarMetricPreferencesRaw[provider.rawValue] ?? \"\"\n let preference = MenuBarMetricPreference(rawValue: raw) ?? .automatic\n if preference == .average, !self.menuBarMetricSupportsAverage(for: provider) {\n return .automatic\n }\n return preference\n }\n\n func setMenuBarMetricPreference(_ preference: MenuBarMetricPreference, for provider: UsageProvider) {\n if provider == .zai {\n self.menuBarMetricPreferencesRaw[provider.rawValue] = MenuBarMetricPreference.primary.rawValue\n return\n }\n if provider == .openrouter {\n switch preference {\n case .automatic, .primary:\n self.menuBarMetricPreferencesRaw[provider.rawValue] = preference.rawValue\n case .secondary, .average:\n self.menuBarMetricPreferencesRaw[provider.rawValue] = MenuBarMetricPreference.automatic.rawValue\n }\n return\n }\n self.menuBarMetricPreferencesRaw[provider.rawValue] = preference.rawValue\n }\n\n func menuBarMetricSupportsAverage(for provider: UsageProvider) -> Bool {\n provider == .gemini\n }\n\n func isCostUsageEffectivelyEnabled(for provider: UsageProvider) -> Bool {\n self.costUsageEnabled\n && ProviderDescriptorRegistry.descriptor(for: provider).tokenCost.supportsTokenCost\n }\n\n var resetTimeDisplayStyle: ResetTimeDisplayStyle {\n self.resetTimesShowAbsolute ? .absolute : .countdown\n }\n}\n" }, { "path": "Sources/CodexBar/SettingsStore+ProviderDetection.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n func runInitialProviderDetectionIfNeeded(force: Bool = false) {\n guard force || !self.providerDetectionCompleted else { return }\n LoginShellPathCache.shared.captureOnce { [weak self] _ in\n Task { @MainActor in\n await self?.applyProviderDetection()\n }\n }\n }\n\n func applyProviderDetection() async {\n guard !self.providerDetectionCompleted else { return }\n let codexInstalled = BinaryLocator.resolveCodexBinary() != nil\n let claudeInstalled = BinaryLocator.resolveClaudeBinary() != nil\n let geminiInstalled = BinaryLocator.resolveGeminiBinary() != nil\n let antigravityRunning = await AntigravityStatusProbe.isRunning()\n let logger = CodexBarLog.logger(LogCategories.providerDetection)\n\n // If none installed, keep Codex enabled to match previous behavior.\n let noneInstalled = !codexInstalled && !claudeInstalled && !geminiInstalled && !antigravityRunning\n let enableCodex = codexInstalled || noneInstalled\n let enableClaude = claudeInstalled\n let enableGemini = geminiInstalled\n let enableAntigravity = antigravityRunning\n\n logger.info(\n \"Provider detection results\",\n metadata: [\n \"codexInstalled\": codexInstalled ? \"1\" : \"0\",\n \"claudeInstalled\": claudeInstalled ? \"1\" : \"0\",\n \"geminiInstalled\": geminiInstalled ? \"1\" : \"0\",\n \"antigravityRunning\": antigravityRunning ? \"1\" : \"0\",\n ])\n logger.info(\n \"Provider detection enablement\",\n metadata: [\n \"codex\": enableCodex ? \"1\" : \"0\",\n \"claude\": enableClaude ? \"1\" : \"0\",\n \"gemini\": enableGemini ? \"1\" : \"0\",\n \"antigravity\": enableAntigravity ? \"1\" : \"0\",\n ])\n\n self.updateProviderConfig(provider: .codex) { entry in\n entry.enabled = enableCodex\n }\n self.updateProviderConfig(provider: .claude) { entry in\n entry.enabled = enableClaude\n }\n self.updateProviderConfig(provider: .gemini) { entry in\n entry.enabled = enableGemini\n }\n self.updateProviderConfig(provider: .antigravity) { entry in\n entry.enabled = enableAntigravity\n }\n self.providerDetectionCompleted = true\n logger.info(\"Provider detection completed\")\n }\n}\n" }, { "path": "Sources/CodexBar/SettingsStore+TokenAccounts.swift", "content": "import AppKit\nimport CodexBarCore\nimport Foundation\n\nextension SettingsStore {\n func tokenAccountsData(for provider: UsageProvider) -> ProviderTokenAccountData? {\n guard TokenAccountSupportCatalog.support(for: provider) != nil else { return nil }\n return self.configSnapshot.providerConfig(for: provider)?.tokenAccounts\n }\n\n func tokenAccounts(for provider: UsageProvider) -> [ProviderTokenAccount] {\n self.tokenAccountsData(for: provider)?.accounts ?? []\n }\n\n func selectedTokenAccount(for provider: UsageProvider) -> ProviderTokenAccount? {\n guard let data = self.tokenAccountsData(for: provider), !data.accounts.isEmpty else { return nil }\n let index = data.clampedActiveIndex()\n return data.accounts[index]\n }\n\n func setActiveTokenAccountIndex(_ index: Int, for provider: UsageProvider) {\n guard let data = self.tokenAccountsData(for: provider), !data.accounts.isEmpty else { return }\n let clamped = min(max(index, 0), data.accounts.count - 1)\n let updated = ProviderTokenAccountData(\n version: data.version,\n accounts: data.accounts,\n activeIndex: clamped)\n self.updateProviderConfig(provider: provider) { entry in\n entry.tokenAccounts = updated\n }\n CodexBarLog.logger(LogCategories.tokenAccounts).info(\n \"Active token account updated\",\n metadata: [\n \"provider\": provider.rawValue,\n \"index\": \"\\(clamped)\",\n ])\n }\n\n func addTokenAccount(provider: UsageProvider, label: String, token: String) {\n guard TokenAccountSupportCatalog.support(for: provider) != nil else { return }\n let trimmedToken = token.trimmingCharacters(in: .whitespacesAndNewlines)\n guard !trimmedToken.isEmpty else { return }\n let trimmedLabel = label.trimmingCharacters(in: .whitespacesAndNewlines)\n let existing = self.tokenAccountsData(for: provider)\n let accounts = existing?.accounts ?? []\n let fallbackLabel = trimmedLabel.isEmpty ? \"Account \\(accounts.count + 1)\" : trimmedLabel\n let account = ProviderTokenAccount(\n id: UUID(),\n label: fallbackLabel,\n token: trimmedToken,\n addedAt: Date().timeIntervalSince1970,\n lastUsed: nil)\n let updated = ProviderTokenAccountData(\n version: existing?.version ?? 1,\n accounts: accounts + [account],\n activeIndex: accounts.count)\n self.updateProviderConfig(provider: provider) { entry in\n entry.tokenAccounts = updated\n }\n self.applyTokenAccountCookieSourceIfNeeded(provider: provider)\n CodexBarLog.logger(LogCategories.tokenAccounts).info(\n \"Token account added\",\n metadata: [\n \"provider\": provider.rawValue,\n \"count\": \"\\(updated.accounts.count)\",\n ])\n }\n\n func removeTokenAccount(provider: UsageProvider, accountID: UUID) {\n guard let data = self.tokenAccountsData(for: provider), !data.accounts.isEmpty else { return }\n let filtered = data.accounts.filter { $0.id != accountID }\n self.updateProviderConfig(provider: provider) { entry in\n if filtered.isEmpty {\n entry.tokenAccounts = nil\n } else {\n let clamped = min(max(data.activeIndex, 0), filtered.count - 1)\n entry.tokenAccounts = ProviderTokenAccountData(\n version: data.version,\n accounts: filtered,\n activeIndex: clamped)\n }\n }\n CodexBarLog.logger(LogCategories.tokenAccounts).info(\n \"Token account removed\",\n metadata: [\n \"provider\": provider.rawValue,\n \"count\": \"\\(filtered.count)\",\n ])\n }\n\n func ensureTokenAccountsLoaded() {\n if self.tokenAccountsLoaded { return }\n self.tokenAccountsLoaded = true\n }\n\n func reloadTokenAccounts() {\n let log = CodexBarLog.logger(LogCategories.tokenAccounts)\n let accounts: [UsageProvider: ProviderTokenAccountData]\n do {\n guard let loaded = try self.configStore.load() else { return }\n accounts = Dictionary(uniqueKeysWithValues: loaded.providers.compactMap { entry in\n guard let data = entry.tokenAccounts else { return nil }\n return (entry.id, data)\n })\n } catch {\n log.error(\"Failed to reload token accounts: \\(error)\")\n return\n }\n self.tokenAccountsLoaded = true\n self.updateProviderTokenAccounts(accounts)\n }\n\n func openTokenAccountsFile() {\n do {\n try self.configStore.save(self.config)\n } catch {\n CodexBarLog.logger(LogCategories.tokenAccounts).error(\"Failed to persist config: \\(error)\")\n return\n }\n NSWorkspace.shared.open(self.configStore.fileURL)\n }\n\n private func applyTokenAccountCookieSourceIfNeeded(provider: UsageProvider) {\n guard let support = TokenAccountSupportCatalog.support(for: provider),\n support.requiresManualCookieSource\n else { return }\n ProviderCatalog.implementation(for: provider)?.applyTokenAccountCookieSource(settings: self)\n }\n}\n" }, { "path": "Sources/CodexBar/SettingsStore+TokenCost.swift", "content": "import Foundation\n\nextension SettingsStore {\n func applyTokenCostDefaultIfNeeded() {\n // Settings are persisted in UserDefaults.standard.\n guard UserDefaults.standard.object(forKey: \"tokenCostUsageEnabled\") == nil else { return }\n\n Task { @MainActor [weak self] in\n guard let self else { return }\n let hasSources = await Task.detached(priority: .utility) {\n Self.hasAnyTokenCostUsageSources()\n }.value\n guard hasSources else { return }\n guard UserDefaults.standard.object(forKey: \"tokenCostUsageEnabled\") == nil else { return }\n self.costUsageEnabled = true\n }\n }\n\n nonisolated static func hasAnyTokenCostUsageSources(\n env: [String: String] = ProcessInfo.processInfo.environment,\n fileManager: FileManager = .default) -> Bool\n {\n func hasAnyJsonl(in root: URL) -> Bool {\n guard fileManager.fileExists(atPath: root.path) else { return false }\n guard let enumerator = fileManager.enumerator(\n at: root,\n includingPropertiesForKeys: [.isRegularFileKey],\n options: [.skipsHiddenFiles, .skipsPackageDescendants])\n else { return false }\n\n for case let url as URL in enumerator where url.pathExtension.lowercased() == \"jsonl\" {\n return true\n }\n return false\n }\n\n let codexRoot: URL = {\n let raw = env[\"CODEX_HOME\"]?.trimmingCharacters(in: .whitespacesAndNewlines)\n if let raw, !raw.isEmpty {\n return URL(fileURLWithPath: raw).appendingPathComponent(\"sessions\", isDirectory: true)\n }\n return fileManager.homeDirectoryForCurrentUser\n .appendingPathComponent(\".codex\", isDirectory: true)\n .appendingPathComponent(\"sessions\", isDirectory: true)\n }()\n\n let archivedCodexRoot: URL? = {\n guard codexRoot.lastPathComponent == \"sessions\" else { return nil }\n return codexRoot\n .deletingLastPathComponent()\n .appendingPathComponent(\"archived_sessions\", isDirectory: true)\n }()\n\n if hasAnyJsonl(in: codexRoot) { return true }\n if let archivedCodexRoot, hasAnyJsonl(in: archivedCodexRoot) { return true }\n\n let claudeRoots: [URL] = {\n if let env = env[\"CLAUDE_CONFIG_DIR\"]?.trimmingCharacters(in: .whitespacesAndNewlines),\n !env.isEmpty\n {\n return env.split(separator: \",\").map { part in\n let raw = String(part).trimmingCharacters(in: .whitespacesAndNewlines)\n let url = URL(fileURLWithPath: raw)\n if url.lastPathComponent == \"projects\" {\n return url\n }\n return url.appendingPathComponent(\"projects\", isDirectory: true)\n }\n }\n\n let home = fileManager.homeDirectoryForCurrentUser\n return [\n home.appendingPathComponent(\".config/claude/projects\", isDirectory: true),\n home.appendingPathComponent(\".claude/projects\", isDirectory: true),\n ]\n }()\n\n return claudeRoots.contains(where: hasAnyJsonl(in:))\n }\n}\n" }, { "path": "Sources/CodexBar/SettingsStore.swift", "content": "import AppKit\nimport CodexBarCore\nimport Observation\nimport ServiceManagement\n\nenum RefreshFrequency: String, CaseIterable, Identifiable {\n case manual\n case oneMinute\n case twoMinutes\n case fiveMinutes\n case fifteenMinutes\n case thirtyMinutes\n\n var id: String {\n self.rawValue\n }\n\n var seconds: TimeInterval? {\n switch self {\n case .manual: nil\n case .oneMinute: 60\n case .twoMinutes: 120\n case .fiveMinutes: 300\n case .fifteenMinutes: 900\n case .thirtyMinutes: 1800\n }\n }\n\n var label: String {\n switch self {\n case .manual: \"Manual\"\n case .oneMinute: \"1 min\"\n case .twoMinutes: \"2 min\"\n case .fiveMinutes: \"5 min\"\n case .fifteenMinutes: \"15 min\"\n case .thirtyMinutes: \"30 min\"\n }\n }\n}\n\nenum MenuBarMetricPreference: String, CaseIterable, Identifiable {\n case automatic\n case primary\n case secondary\n case average\n\n var id: String {\n self.rawValue\n }\n\n var label: String {\n switch self {\n case .automatic: \"Automatic\"\n case .primary: \"Primary\"\n case .secondary: \"Secondary\"\n case .average: \"Average\"\n }\n }\n}\n\n@MainActor\n@Observable\nfinal class SettingsStore {\n static let sharedDefaults = UserDefaults(suiteName: \"group.com.steipete.codexbar\")\n static let mergedOverviewProviderLimit = 3\n static let isRunningTests: Bool = {\n let env = ProcessInfo.processInfo.environment\n if env[\"XCTestConfigurationFilePath\"] != nil { return true }\n if env[\"TESTING_LIBRARY_VERSION\"] != nil { return true }\n if env[\"SWIFT_TESTING\"] != nil { return true }\n return NSClassFromString(\"XCTestCase\") != nil\n }()\n\n @ObservationIgnored let userDefaults: UserDefaults\n @ObservationIgnored let configStore: CodexBarConfigStore\n @ObservationIgnored var config: CodexBarConfig\n @ObservationIgnored var configPersistTask: Task?\n @ObservationIgnored var configLoading = false\n @ObservationIgnored var tokenAccountsLoaded = false\n var defaultsState: SettingsDefaultsState\n var configRevision: Int = 0\n var providerOrder: [UsageProvider] = []\n var providerEnablement: [UsageProvider: Bool] = [:]\n\n init(\n userDefaults: UserDefaults = .standard,\n configStore: CodexBarConfigStore = CodexBarConfigStore(),\n zaiTokenStore: any ZaiTokenStoring = KeychainZaiTokenStore(),\n syntheticTokenStore: any SyntheticTokenStoring = KeychainSyntheticTokenStore(),\n codexCookieStore: any CookieHeaderStoring = KeychainCookieHeaderStore(\n account: \"codex-cookie\",\n promptKind: .codexCookie),\n claudeCookieStore: any CookieHeaderStoring = KeychainCookieHeaderStore(\n account: \"claude-cookie\",\n promptKind: .claudeCookie),\n cursorCookieStore: any CookieHeaderStoring = KeychainCookieHeaderStore(\n account: \"cursor-cookie\",\n promptKind: .cursorCookie),\n opencodeCookieStore: any CookieHeaderStoring = KeychainCookieHeaderStore(\n account: \"opencode-cookie\",\n promptKind: .opencodeCookie),\n factoryCookieStore: any CookieHeaderStoring = KeychainCookieHeaderStore(\n account: \"factory-cookie\",\n promptKind: .factoryCookie),\n minimaxCookieStore: any MiniMaxCookieStoring = KeychainMiniMaxCookieStore(),\n minimaxAPITokenStore: any MiniMaxAPITokenStoring = KeychainMiniMaxAPITokenStore(),\n kimiTokenStore: any KimiTokenStoring = KeychainKimiTokenStore(),\n kimiK2TokenStore: any KimiK2TokenStoring = KeychainKimiK2TokenStore(),\n augmentCookieStore: any CookieHeaderStoring = KeychainCookieHeaderStore(\n account: \"augment-cookie\",\n promptKind: .augmentCookie),\n ampCookieStore: any CookieHeaderStoring = KeychainCookieHeaderStore(\n account: \"amp-cookie\",\n promptKind: .ampCookie),\n copilotTokenStore: any CopilotTokenStoring = KeychainCopilotTokenStore(),\n tokenAccountStore: any ProviderTokenAccountStoring = FileTokenAccountStore())\n {\n let legacyStores = CodexBarConfigMigrator.LegacyStores(\n zaiTokenStore: zaiTokenStore,\n syntheticTokenStore: syntheticTokenStore,\n codexCookieStore: codexCookieStore,\n claudeCookieStore: claudeCookieStore,\n cursorCookieStore: cursorCookieStore,\n opencodeCookieStore: opencodeCookieStore,\n factoryCookieStore: factoryCookieStore,\n minimaxCookieStore: minimaxCookieStore,\n minimaxAPITokenStore: minimaxAPITokenStore,\n kimiTokenStore: kimiTokenStore,\n kimiK2TokenStore: kimiK2TokenStore,\n augmentCookieStore: augmentCookieStore,\n ampCookieStore: ampCookieStore,\n copilotTokenStore: copilotTokenStore,\n tokenAccountStore: tokenAccountStore)\n let config = CodexBarConfigMigrator.loadOrMigrate(\n configStore: configStore,\n userDefaults: userDefaults,\n stores: legacyStores)\n self.userDefaults = userDefaults\n self.configStore = configStore\n self.config = config\n self.configLoading = true\n self.defaultsState = Self.loadDefaultsState(userDefaults: userDefaults)\n self.updateProviderState(config: config)\n self.configLoading = false\n CodexBarLog.setFileLoggingEnabled(self.debugFileLoggingEnabled)\n userDefaults.removeObject(forKey: \"showCodexUsage\")\n userDefaults.removeObject(forKey: \"showClaudeUsage\")\n LaunchAtLoginManager.setEnabled(self.launchAtLogin)\n self.runInitialProviderDetectionIfNeeded()\n self.ensureAlibabaProviderAutoEnabledIfNeeded()\n self.applyTokenCostDefaultIfNeeded()\n if self.claudeUsageDataSource != .cli { self.claudeWebExtrasEnabled = false }\n self.openAIWebAccessEnabled = self.codexCookieSource.isEnabled\n Self.sharedDefaults?.set(self.debugDisableKeychainAccess, forKey: \"debugDisableKeychainAccess\")\n KeychainAccessGate.isDisabled = self.debugDisableKeychainAccess\n }\n}\n\nextension SettingsStore {\n private static func loadDefaultsState(userDefaults: UserDefaults) -> SettingsDefaultsState {\n let refreshRaw = userDefaults.string(forKey: \"refreshFrequency\") ?? RefreshFrequency.fiveMinutes.rawValue\n let refreshFrequency = RefreshFrequency(rawValue: refreshRaw) ?? .fiveMinutes\n let launchAtLogin = userDefaults.object(forKey: \"launchAtLogin\") as? Bool ?? false\n let debugMenuEnabled = userDefaults.object(forKey: \"debugMenuEnabled\") as? Bool ?? false\n let debugDisableKeychainAccess: Bool = {\n if let stored = userDefaults.object(forKey: \"debugDisableKeychainAccess\") as? Bool {\n return stored\n }\n if let shared = Self.sharedDefaults?.object(forKey: \"debugDisableKeychainAccess\") as? Bool {\n userDefaults.set(shared, forKey: \"debugDisableKeychainAccess\")\n return shared\n }\n return false\n }()\n let debugFileLoggingEnabled = userDefaults.object(forKey: \"debugFileLoggingEnabled\") as? Bool ?? false\n let debugLogLevelRaw = userDefaults.string(forKey: \"debugLogLevel\") ?? CodexBarLog.Level.verbose.rawValue\n if userDefaults.string(forKey: \"debugLogLevel\") == nil {\n userDefaults.set(debugLogLevelRaw, forKey: \"debugLogLevel\")\n }\n let debugLoadingPatternRaw = userDefaults.string(forKey: \"debugLoadingPattern\")\n let debugKeepCLISessionsAlive = userDefaults.object(forKey: \"debugKeepCLISessionsAlive\") as? Bool ?? false\n let statusChecksEnabled = userDefaults.object(forKey: \"statusChecksEnabled\") as? Bool ?? true\n let sessionQuotaDefault = userDefaults.object(forKey: \"sessionQuotaNotificationsEnabled\") as? Bool\n let sessionQuotaNotificationsEnabled = sessionQuotaDefault ?? true\n if sessionQuotaDefault == nil {\n userDefaults.set(true, forKey: \"sessionQuotaNotificationsEnabled\")\n }\n let usageBarsShowUsed = userDefaults.object(forKey: \"usageBarsShowUsed\") as? Bool ?? false\n let resetTimesShowAbsolute = userDefaults.object(forKey: \"resetTimesShowAbsolute\") as? Bool ?? false\n let menuBarShowsBrandIconWithPercent = userDefaults.object(\n forKey: \"menuBarShowsBrandIconWithPercent\") as? Bool ?? false\n let menuBarDisplayModeRaw = userDefaults.string(forKey: \"menuBarDisplayMode\")\n ?? MenuBarDisplayMode.percent.rawValue\n let historicalTrackingEnabled = userDefaults.object(forKey: \"historicalTrackingEnabled\") as? Bool ?? false\n let showAllTokenAccountsInMenu = userDefaults.object(forKey: \"showAllTokenAccountsInMenu\") as? Bool ?? false\n let storedPreferences = userDefaults.dictionary(forKey: \"menuBarMetricPreferences\") as? [String: String] ?? [:]\n var resolvedPreferences = storedPreferences\n if resolvedPreferences.isEmpty,\n let menuBarMetricRaw = userDefaults.string(forKey: \"menuBarMetricPreference\"),\n let legacyPreference = MenuBarMetricPreference(rawValue: menuBarMetricRaw)\n {\n resolvedPreferences = Dictionary(\n uniqueKeysWithValues: UsageProvider.allCases.map { ($0.rawValue, legacyPreference.rawValue) })\n }\n let costUsageEnabled = userDefaults.object(forKey: \"tokenCostUsageEnabled\") as? Bool ?? false\n let hidePersonalInfo = userDefaults.object(forKey: \"hidePersonalInfo\") as? Bool ?? false\n let randomBlinkEnabled = userDefaults.object(forKey: \"randomBlinkEnabled\") as? Bool ?? false\n let menuBarShowsHighestUsage = userDefaults.object(forKey: \"menuBarShowsHighestUsage\") as? Bool ?? false\n let claudeOAuthKeychainPromptModeRaw = userDefaults.string(forKey: \"claudeOAuthKeychainPromptMode\")\n let claudeOAuthKeychainReadStrategyRaw = userDefaults.string(forKey: \"claudeOAuthKeychainReadStrategy\")\n let claudeWebExtrasEnabledRaw = userDefaults.object(forKey: \"claudeWebExtrasEnabled\") as? Bool ?? false\n let creditsExtrasDefault = userDefaults.object(forKey: \"showOptionalCreditsAndExtraUsage\") as? Bool\n let showOptionalCreditsAndExtraUsage = creditsExtrasDefault ?? true\n if creditsExtrasDefault == nil { userDefaults.set(true, forKey: \"showOptionalCreditsAndExtraUsage\") }\n let openAIWebAccessDefault = userDefaults.object(forKey: \"openAIWebAccessEnabled\") as? Bool\n let openAIWebAccessEnabled = openAIWebAccessDefault ?? true\n if openAIWebAccessDefault == nil { userDefaults.set(true, forKey: \"openAIWebAccessEnabled\") }\n let jetbrainsIDEBasePath = userDefaults.string(forKey: \"jetbrainsIDEBasePath\") ?? \"\"\n let mergeIcons = userDefaults.object(forKey: \"mergeIcons\") as? Bool ?? true\n let switcherShowsIcons = userDefaults.object(forKey: \"switcherShowsIcons\") as? Bool ?? true\n let mergedMenuLastSelectedWasOverview = userDefaults.object(\n forKey: \"mergedMenuLastSelectedWasOverview\") as? Bool ?? false\n let mergedOverviewSelectedProvidersRaw = userDefaults.array(\n forKey: \"mergedOverviewSelectedProviders\") as? [String] ?? []\n let selectedMenuProviderRaw = userDefaults.string(forKey: \"selectedMenuProvider\")\n let providerDetectionCompleted = userDefaults.object(forKey: \"providerDetectionCompleted\") as? Bool ?? false\n\n return SettingsDefaultsState(\n refreshFrequency: refreshFrequency,\n launchAtLogin: launchAtLogin,\n debugMenuEnabled: debugMenuEnabled,\n debugDisableKeychainAccess: debugDisableKeychainAccess,\n debugFileLoggingEnabled: debugFileLoggingEnabled,\n debugLogLevelRaw: debugLogLevelRaw,\n debugLoadingPatternRaw: debugLoadingPatternRaw,\n debugKeepCLISessionsAlive: debugKeepCLISessionsAlive,\n statusChecksEnabled: statusChecksEnabled,\n sessionQuotaNotificationsEnabled: sessionQuotaNotificationsEnabled,\n usageBarsShowUsed: usageBarsShowUsed,\n resetTimesShowAbsolute: resetTimesShowAbsolute,\n menuBarShowsBrandIconWithPercent: menuBarShowsBrandIconWithPercent,\n menuBarDisplayModeRaw: menuBarDisplayModeRaw,\n historicalTrackingEnabled: historicalTrackingEnabled,\n showAllTokenAccountsInMenu: showAllTokenAccountsInMenu,\n menuBarMetricPreferencesRaw: resolvedPreferences,\n costUsageEnabled: costUsageEnabled,\n hidePersonalInfo: hidePersonalInfo,\n randomBlinkEnabled: randomBlinkEnabled,\n menuBarShowsHighestUsage: menuBarShowsHighestUsage,\n claudeOAuthKeychainPromptModeRaw: claudeOAuthKeychainPromptModeRaw,\n claudeOAuthKeychainReadStrategyRaw: claudeOAuthKeychainReadStrategyRaw,\n claudeWebExtrasEnabledRaw: claudeWebExtrasEnabledRaw,\n showOptionalCreditsAndExtraUsage: showOptionalCreditsAndExtraUsage,\n openAIWebAccessEnabled: openAIWebAccessEnabled,\n jetbrainsIDEBasePath: jetbrainsIDEBasePath,\n mergeIcons: mergeIcons,\n switcherShowsIcons: switcherShowsIcons,\n mergedMenuLastSelectedWasOverview: mergedMenuLastSelectedWasOverview,\n mergedOverviewSelectedProvidersRaw: mergedOverviewSelectedProvidersRaw,\n selectedMenuProviderRaw: selectedMenuProviderRaw,\n providerDetectionCompleted: providerDetectionCompleted)\n }\n}\n\nextension SettingsStore {\n var configSnapshot: CodexBarConfig {\n _ = self.configRevision\n return self.config\n }\n\n func updateProviderState(config: CodexBarConfig) {\n let rawOrder = config.providers.map(\\.id.rawValue)\n self.providerOrder = Self.effectiveProviderOrder(raw: rawOrder)\n let metadata = ProviderDescriptorRegistry.metadata\n var enablement: [UsageProvider: Bool] = [:]\n enablement.reserveCapacity(metadata.count)\n for provider in UsageProvider.allCases {\n let defaultEnabled = metadata[provider]?.defaultEnabled ?? false\n enablement[provider] = config.providerConfig(for: provider)?.enabled ?? defaultEnabled\n }\n self.providerEnablement = enablement\n }\n\n func orderedProviders() -> [UsageProvider] {\n if self.providerOrder.isEmpty {\n self.updateProviderState(config: self.configSnapshot)\n }\n return self.providerOrder\n }\n\n func moveProvider(fromOffsets: IndexSet, toOffset: Int) {\n var order = self.orderedProviders()\n order.move(fromOffsets: fromOffsets, toOffset: toOffset)\n self.setProviderOrder(order)\n }\n\n func isProviderEnabled(provider: UsageProvider, metadata: ProviderMetadata) -> Bool {\n self.providerEnablement[provider] ?? metadata.defaultEnabled\n }\n\n func isProviderEnabledCached(\n provider: UsageProvider,\n metadataByProvider: [UsageProvider: ProviderMetadata]) -> Bool\n {\n let defaultEnabled = metadataByProvider[provider]?.defaultEnabled ?? false\n return self.providerEnablement[provider] ?? defaultEnabled\n }\n\n func enabledProvidersOrdered(metadataByProvider: [UsageProvider: ProviderMetadata]) -> [UsageProvider] {\n _ = metadataByProvider\n return self.orderedProviders().filter { self.providerEnablement[$0] ?? false }\n }\n\n func setProviderEnabled(provider: UsageProvider, metadata _: ProviderMetadata, enabled: Bool) {\n CodexBarLog.logger(LogCategories.settings).debug(\n \"Provider toggle updated\",\n metadata: [\"provider\": provider.rawValue, \"enabled\": \"\\(enabled)\"])\n self.updateProviderConfig(provider: provider) { entry in\n entry.enabled = enabled\n }\n }\n\n func rerunProviderDetection() {\n self.runInitialProviderDetectionIfNeeded(force: true)\n }\n}\n\nextension SettingsStore {\n private static func effectiveProviderOrder(raw: [String]) -> [UsageProvider] {\n var seen: Set = []\n var ordered: [UsageProvider] = []\n\n for rawValue in raw {\n guard let provider = UsageProvider(rawValue: rawValue) else { continue }\n guard !seen.contains(provider) else { continue }\n seen.insert(provider)\n ordered.append(provider)\n }\n\n if ordered.isEmpty {\n ordered = UsageProvider.allCases\n seen = Set(ordered)\n }\n\n if !seen.contains(.factory), let zaiIndex = ordered.firstIndex(of: .zai) {\n ordered.insert(.factory, at: zaiIndex)\n seen.insert(.factory)\n }\n\n if !seen.contains(.minimax), let zaiIndex = ordered.firstIndex(of: .zai) {\n let insertIndex = ordered.index(after: zaiIndex)\n ordered.insert(.minimax, at: insertIndex)\n seen.insert(.minimax)\n }\n\n for provider in UsageProvider.allCases where !seen.contains(provider) {\n ordered.append(provider)\n }\n\n return ordered\n }\n}\n" }, { "path": "Sources/CodexBar/SettingsStoreState.swift", "content": "import Foundation\n\nstruct SettingsDefaultsState {\n var refreshFrequency: RefreshFrequency\n var launchAtLogin: Bool\n var debugMenuEnabled: Bool\n var debugDisableKeychainAccess: Bool\n var debugFileLoggingEnabled: Bool\n var debugLogLevelRaw: String?\n var debugLoadingPatternRaw: String?\n var debugKeepCLISessionsAlive: Bool\n var statusChecksEnabled: Bool\n var sessionQuotaNotificationsEnabled: Bool\n var usageBarsShowUsed: Bool\n var resetTimesShowAbsolute: Bool\n var menuBarShowsBrandIconWithPercent: Bool\n var menuBarDisplayModeRaw: String?\n var historicalTrackingEnabled: Bool\n var showAllTokenAccountsInMenu: Bool\n var menuBarMetricPreferencesRaw: [String: String]\n var costUsageEnabled: Bool\n var hidePersonalInfo: Bool\n var randomBlinkEnabled: Bool\n var menuBarShowsHighestUsage: Bool\n var claudeOAuthKeychainPromptModeRaw: String?\n var claudeOAuthKeychainReadStrategyRaw: String?\n var claudeWebExtrasEnabledRaw: Bool\n var showOptionalCreditsAndExtraUsage: Bool\n var openAIWebAccessEnabled: Bool\n var jetbrainsIDEBasePath: String\n var mergeIcons: Bool\n var switcherShowsIcons: Bool\n var mergedMenuLastSelectedWasOverview: Bool\n var mergedOverviewSelectedProvidersRaw: [String]\n var selectedMenuProviderRaw: String?\n var providerDetectionCompleted: Bool\n}\n" }, { "path": "Sources/CodexBar/StatusItemController+Actions.swift", "content": "import AppKit\nimport CodexBarCore\n\nextension StatusItemController {\n // MARK: - Actions reachable from menus\n\n func refreshStore(forceTokenUsage: Bool) {\n Task {\n await ProviderInteractionContext.$current.withValue(.userInitiated) {\n await self.store.refresh(forceTokenUsage: forceTokenUsage)\n }\n }\n }\n\n @objc func refreshNow() {\n self.refreshStore(forceTokenUsage: true)\n }\n\n @objc func refreshAugmentSession() {\n Task {\n await self.store.forceRefreshAugmentSession()\n // Also trigger a full refresh to update the menu and clear any stale errors\n await ProviderInteractionContext.$current.withValue(.userInitiated) {\n await self.store.refresh(forceTokenUsage: false)\n }\n }\n }\n\n @objc func installUpdate() {\n self.updater.checkForUpdates(nil)\n }\n\n @objc func openDashboard() {\n let preferred = self.lastMenuProvider\n ?? (self.store.isEnabled(.codex) ? .codex : self.store.enabledProviders().first)\n\n let provider = preferred ?? .codex\n guard let url = self.dashboardURL(for: provider) else { return }\n NSWorkspace.shared.open(url)\n }\n\n func dashboardURL(for provider: UsageProvider) -> URL? {\n if provider == .alibaba {\n return self.settings.alibabaCodingPlanAPIRegion.dashboardURL\n }\n\n let meta = self.store.metadata(for: provider)\n let urlString: String? = if provider == .claude, self.store.isClaudeSubscription() {\n meta.subscriptionDashboardURL ?? meta.dashboardURL\n } else {\n meta.dashboardURL\n }\n\n guard let urlString else { return nil }\n return URL(string: urlString)\n }\n\n @objc func openCreditsPurchase() {\n let preferred = self.lastMenuProvider\n ?? (self.store.isEnabled(.codex) ? .codex : self.store.enabledProviders().first)\n let provider = preferred ?? .codex\n guard provider == .codex else { return }\n\n let dashboardURL = self.store.metadata(for: .codex).dashboardURL\n let purchaseURL = Self.sanitizedCreditsPurchaseURL(self.store.openAIDashboard?.creditsPurchaseURL)\n let urlString = purchaseURL ?? dashboardURL\n guard let urlString,\n let url = URL(string: urlString) else { return }\n\n let autoStart = true\n let accountEmail = self.store.codexAccountEmailForOpenAIDashboard()\n let controller = self.creditsPurchaseWindow ?? OpenAICreditsPurchaseWindowController()\n controller.show(purchaseURL: url, accountEmail: accountEmail, autoStartPurchase: autoStart)\n self.creditsPurchaseWindow = controller\n }\n\n private static func sanitizedCreditsPurchaseURL(_ raw: String?) -> String? {\n guard let raw, let url = URL(string: raw) else { return nil }\n guard let host = url.host?.lowercased(), host.contains(\"chatgpt.com\") else { return nil }\n let path = url.path.lowercased()\n let allowed = [\"settings\", \"usage\", \"billing\", \"credits\"]\n guard allowed.contains(where: { path.contains($0) }) else { return nil }\n return url.absoluteString\n }\n\n @objc func openStatusPage() {\n let preferred = self.lastMenuProvider\n ?? (self.store.isEnabled(.codex) ? .codex : self.store.enabledProviders().first)\n\n let provider = preferred ?? .codex\n let meta = self.store.metadata(for: provider)\n let urlString = meta.statusPageURL ?? meta.statusLinkURL\n guard let urlString, let url = URL(string: urlString) else { return }\n NSWorkspace.shared.open(url)\n }\n\n @objc func openTerminalCommand(_ sender: NSMenuItem) {\n let command = sender.representedObject as? String ?? \"claude\"\n Self.openTerminal(command: command)\n }\n\n @objc func openLoginToProvider(_ sender: NSMenuItem) {\n guard let urlString = sender.representedObject as? String,\n let url = URL(string: urlString) else { return }\n NSWorkspace.shared.open(url)\n }\n\n @objc func runSwitchAccount(_ sender: NSMenuItem) {\n if self.loginTask != nil {\n self.loginLogger.info(\"Switch Account tap ignored: login already in-flight\")\n return\n }\n\n let rawProvider = sender.representedObject as? String\n let provider = rawProvider.flatMap(UsageProvider.init(rawValue:)) ?? self.lastMenuProvider ?? .codex\n self.loginLogger.info(\"Switch Account tapped\", metadata: [\"provider\": provider.rawValue])\n\n self.loginTask = Task { @MainActor [weak self] in\n guard let self else { return }\n defer {\n self.activeLoginProvider = nil\n self.loginTask = nil\n }\n self.activeLoginProvider = provider\n self.loginPhase = .requesting\n self.loginLogger.info(\"Starting login task\", metadata: [\"provider\": provider.rawValue])\n\n let shouldRefresh = await self.runLoginFlow(provider: provider)\n if shouldRefresh {\n await ProviderInteractionContext.$current.withValue(.userInitiated) {\n await self.store.refresh()\n }\n self.loginLogger.info(\"Triggered refresh after login\", metadata: [\"provider\": provider.rawValue])\n }\n }\n }\n\n @objc func showSettingsGeneral() {\n self.openSettings(tab: .general)\n }\n\n @objc func showSettingsAbout() {\n self.openSettings(tab: .about)\n }\n\n func openMenuFromShortcut() {\n if self.shouldMergeIcons {\n self.statusItem.button?.performClick(nil)\n return\n }\n\n let provider = self.resolvedShortcutProvider()\n // Use the lazy accessor to ensure the item exists\n let item = self.lazyStatusItem(for: provider)\n item.button?.performClick(nil)\n }\n\n private func openSettings(tab: PreferencesTab) {\n DispatchQueue.main.async {\n self.preferencesSelection.tab = tab\n NSApp.activate(ignoringOtherApps: true)\n NotificationCenter.default.post(\n name: .codexbarOpenSettings,\n object: nil,\n userInfo: [\"tab\": tab.rawValue])\n }\n }\n\n @objc func quit() {\n NSApp.terminate(nil)\n }\n\n @objc func copyError(_ sender: NSMenuItem) {\n if let err = sender.representedObject as? String {\n let pb = NSPasteboard.general\n pb.clearContents()\n pb.setString(err, forType: .string)\n }\n }\n\n private static func openTerminal(command: String) {\n let escaped = command\n .replacingOccurrences(of: \"\\\\\\\\\", with: \"\\\\\\\\\\\\\\\\\")\n .replacingOccurrences(of: \"\\\"\", with: \"\\\\\\\"\")\n let script = \"\"\"\n tell application \"Terminal\"\n activate\n do script \"\\(escaped)\"\n end tell\n \"\"\"\n if let appleScript = NSAppleScript(source: script) {\n var error: NSDictionary?\n appleScript.executeAndReturnError(&error)\n if let error {\n CodexBarLog.logger(LogCategories.terminal).error(\n \"Failed to open Terminal\",\n metadata: [\"error\": String(describing: error)])\n }\n }\n }\n\n private func resolvedShortcutProvider() -> UsageProvider {\n if let last = self.lastMenuProvider, self.isEnabled(last) {\n return last\n }\n if let first = self.store.enabledProviders().first {\n return first\n }\n return .codex\n }\n\n func presentCodexLoginResult(_ result: CodexLoginRunner.Result) {\n switch result.outcome {\n case .success:\n return\n case .missingBinary:\n self.presentLoginAlert(\n title: \"Codex CLI not found\",\n message: \"Install the Codex CLI (npm i -g @openai/codex) and try again.\")\n case let .launchFailed(message):\n self.presentLoginAlert(title: \"Could not start codex login\", message: message)\n case .timedOut:\n self.presentLoginAlert(\n title: \"Codex login timed out\",\n message: self.trimmedLoginOutput(result.output))\n case let .failed(status):\n let statusLine = \"codex login exited with status \\(status).\"\n let message = self.trimmedLoginOutput(result.output.isEmpty ? statusLine : result.output)\n self.presentLoginAlert(title: \"Codex login failed\", message: message)\n }\n }\n\n func presentClaudeLoginResult(_ result: ClaudeLoginRunner.Result) {\n switch result.outcome {\n case .success:\n return\n case .missingBinary:\n self.presentLoginAlert(\n title: \"Claude CLI not found\",\n message: \"Install the Claude CLI (npm i -g @anthropic-ai/claude-code) and try again.\")\n case let .launchFailed(message):\n self.presentLoginAlert(title: \"Could not start claude /login\", message: message)\n case .timedOut:\n self.presentLoginAlert(\n title: \"Claude login timed out\",\n message: self.trimmedLoginOutput(result.output))\n case let .failed(status):\n let statusLine = \"claude /login exited with status \\(status).\"\n let message = self.trimmedLoginOutput(result.output.isEmpty ? statusLine : result.output)\n self.presentLoginAlert(title: \"Claude login failed\", message: message)\n }\n }\n\n func describe(_ outcome: CodexLoginRunner.Result.Outcome) -> String {\n switch outcome {\n case .success: \"success\"\n case .timedOut: \"timedOut\"\n case let .failed(status): \"failed(status: \\(status))\"\n case .missingBinary: \"missingBinary\"\n case let .launchFailed(message): \"launchFailed(\\(message))\"\n }\n }\n\n func describe(_ outcome: ClaudeLoginRunner.Result.Outcome) -> String {\n switch outcome {\n case .success: \"success\"\n case .timedOut: \"timedOut\"\n case let .failed(status): \"failed(status: \\(status))\"\n case .missingBinary: \"missingBinary\"\n case let .launchFailed(message): \"launchFailed(\\(message))\"\n }\n }\n\n func describe(_ outcome: GeminiLoginRunner.Result.Outcome) -> String {\n switch outcome {\n case .success: \"success\"\n case .missingBinary: \"missingBinary\"\n case let .launchFailed(message): \"launchFailed(\\(message))\"\n }\n }\n\n func presentGeminiLoginResult(_ result: GeminiLoginRunner.Result) {\n guard let info = Self.geminiLoginAlertInfo(for: result) else { return }\n self.presentLoginAlert(title: info.title, message: info.message)\n }\n\n struct LoginAlertInfo: Equatable {\n let title: String\n let message: String\n }\n\n nonisolated static func geminiLoginAlertInfo(for result: GeminiLoginRunner.Result) -> LoginAlertInfo? {\n switch result.outcome {\n case .success:\n nil\n case .missingBinary:\n LoginAlertInfo(\n title: \"Gemini CLI not found\",\n message: \"Install the Gemini CLI (npm i -g @google/gemini-cli) and try again.\")\n case let .launchFailed(message):\n LoginAlertInfo(title: \"Could not open Terminal for Gemini\", message: message)\n }\n }\n\n func presentLoginAlert(title: String, message: String) {\n let alert = NSAlert()\n alert.messageText = title\n alert.informativeText = message\n alert.alertStyle = .warning\n alert.runModal()\n }\n\n private func trimmedLoginOutput(_ text: String) -> String {\n let trimmed = text.trimmingCharacters(in: .whitespacesAndNewlines)\n let limit = 600\n if trimmed.isEmpty { return \"No output captured.\" }\n if trimmed.count <= limit { return trimmed }\n let idx = trimmed.index(trimmed.startIndex, offsetBy: limit)\n return \"\\(trimmed[.. String {\n switch outcome {\n case .success: \"success\"\n case .cancelled: \"cancelled\"\n case let .failed(message): \"failed(\\(message))\"\n }\n }\n}\n" }, { "path": "Sources/CodexBar/StatusItemController+Animation.swift", "content": "import AppKit\nimport CodexBarCore\nimport QuartzCore\n\nextension StatusItemController {\n private static let loadingPercentEpsilon = 0.0001\n private static let blinkActiveTickInterval: Duration = .milliseconds(75)\n private static let blinkIdleFallbackInterval: Duration = .seconds(1)\n\n func needsMenuBarIconAnimation() -> Bool {\n if self.shouldMergeIcons {\n let primaryProvider = self.primaryProviderForUnifiedIcon()\n return self.shouldAnimate(provider: primaryProvider)\n }\n return UsageProvider.allCases.contains { self.shouldAnimate(provider: $0) }\n }\n\n func updateBlinkingState() {\n // During the loading animation, blink ticks can overwrite the animated menu bar icon and cause flicker.\n if self.needsMenuBarIconAnimation() {\n self.stopBlinking()\n return\n }\n\n let blinkingEnabled = self.isBlinkingAllowed()\n // Use display list so merged-mode visibility stays consistent with shouldMergeIcons.\n let displayProviders = self.store.enabledProvidersForDisplay()\n let anyEnabled = !displayProviders.isEmpty || self.store.debugForceAnimation\n let anyVisible = UsageProvider.allCases.contains { self.isVisible($0) }\n let mergeIcons = self.shouldMergeIcons\n let shouldBlink = mergeIcons ? anyEnabled : anyVisible\n if blinkingEnabled, shouldBlink {\n if self.blinkTask == nil {\n self.seedBlinkStatesIfNeeded()\n self.blinkTask = Task { [weak self] in\n while !Task.isCancelled {\n let delay = await MainActor.run {\n self?.blinkTickSleepDuration(now: Date()) ?? Self.blinkIdleFallbackInterval\n }\n try? await Task.sleep(for: delay)\n await MainActor.run { self?.tickBlink() }\n }\n }\n }\n } else {\n self.stopBlinking()\n }\n }\n\n private func seedBlinkStatesIfNeeded() {\n let now = Date()\n for provider in UsageProvider.allCases where self.blinkStates[provider] == nil {\n self.blinkStates[provider] = BlinkState(nextBlink: now.addingTimeInterval(BlinkState.randomDelay()))\n }\n }\n\n private func stopBlinking() {\n self.blinkTask?.cancel()\n self.blinkTask = nil\n self.blinkAmounts.removeAll()\n let phase: Double? = self.needsMenuBarIconAnimation() ? self.animationPhase : nil\n if self.shouldMergeIcons {\n self.applyIcon(phase: phase)\n } else {\n for provider in UsageProvider.allCases {\n self.applyIcon(for: provider, phase: phase)\n }\n }\n }\n\n private func blinkTickSleepDuration(now: Date) -> Duration {\n let mergeIcons = self.shouldMergeIcons\n var nextWakeAt: Date?\n\n for provider in UsageProvider.allCases {\n let shouldRender = mergeIcons ? self.isEnabled(provider) : self.isVisible(provider)\n guard shouldRender, !self.shouldAnimate(provider: provider, mergeIcons: mergeIcons) else { continue }\n\n let state = self\n .blinkStates[provider] ?? BlinkState(nextBlink: now.addingTimeInterval(BlinkState.randomDelay()))\n if state.blinkStart != nil {\n return Self.blinkActiveTickInterval\n }\n\n let candidate: Date = state.pendingSecondStart ?? state.nextBlink\n if let current = nextWakeAt {\n if candidate < current {\n nextWakeAt = candidate\n }\n } else {\n nextWakeAt = candidate\n }\n }\n\n guard let nextWakeAt else { return Self.blinkIdleFallbackInterval }\n let delay = nextWakeAt.timeIntervalSince(now)\n if delay <= 0 { return Self.blinkActiveTickInterval }\n return .seconds(delay)\n }\n\n private func tickBlink(now: Date = .init()) {\n guard self.isBlinkingAllowed(at: now) else {\n self.stopBlinking()\n return\n }\n\n let blinkDuration: TimeInterval = 0.36\n let doubleBlinkChance = 0.18\n let doubleDelayRange: ClosedRange = 0.22...0.34\n // Cache merge state once per tick to avoid repeated enabled-provider lookups.\n let mergeIcons = self.shouldMergeIcons\n\n for provider in UsageProvider.allCases {\n let shouldRender = mergeIcons ? self.isEnabled(provider) : self.isVisible(provider)\n guard shouldRender, !self.shouldAnimate(provider: provider, mergeIcons: mergeIcons) else {\n self.clearMotion(for: provider)\n continue\n }\n\n var state = self\n .blinkStates[provider] ?? BlinkState(nextBlink: now.addingTimeInterval(BlinkState.randomDelay()))\n\n if let pendingSecond = state.pendingSecondStart, now >= pendingSecond {\n state.blinkStart = now\n state.pendingSecondStart = nil\n }\n\n if let start = state.blinkStart {\n let elapsed = now.timeIntervalSince(start)\n if elapsed >= blinkDuration {\n state.blinkStart = nil\n if let pending = state.pendingSecondStart, now < pending {\n // Wait for the planned double-blink.\n } else {\n state.pendingSecondStart = nil\n state.nextBlink = now.addingTimeInterval(BlinkState.randomDelay())\n }\n self.clearMotion(for: provider)\n } else {\n let progress = max(0, min(elapsed / blinkDuration, 1))\n let symmetric = progress < 0.5 ? progress * 2 : (1 - progress) * 2\n let eased = pow(symmetric, 2.2) // slightly punchier than smoothstep\n self.assignMotion(amount: CGFloat(eased), for: provider, effect: state.effect)\n }\n } else if now >= state.nextBlink {\n state.blinkStart = now\n state.effect = self.randomEffect(for: provider)\n if state.effect == .blink, Double.random(in: 0...1) < doubleBlinkChance {\n state.pendingSecondStart = now.addingTimeInterval(Double.random(in: doubleDelayRange))\n }\n self.clearMotion(for: provider)\n } else {\n self.clearMotion(for: provider)\n }\n\n self.blinkStates[provider] = state\n if !mergeIcons {\n self.applyIcon(for: provider, phase: nil)\n }\n }\n if mergeIcons {\n let phase: Double? = self.needsMenuBarIconAnimation() ? self.animationPhase : nil\n self.applyIcon(phase: phase)\n }\n }\n\n private func blinkAmount(for provider: UsageProvider) -> CGFloat {\n guard self.isBlinkingAllowed() else { return 0 }\n return self.blinkAmounts[provider] ?? 0\n }\n\n private func wiggleAmount(for provider: UsageProvider) -> CGFloat {\n guard self.isBlinkingAllowed() else { return 0 }\n return self.wiggleAmounts[provider] ?? 0\n }\n\n private func tiltAmount(for provider: UsageProvider) -> CGFloat {\n guard self.isBlinkingAllowed() else { return 0 }\n return self.tiltAmounts[provider] ?? 0\n }\n\n private func assignMotion(amount: CGFloat, for provider: UsageProvider, effect: MotionEffect) {\n switch effect {\n case .blink:\n self.blinkAmounts[provider] = amount\n self.wiggleAmounts[provider] = 0\n self.tiltAmounts[provider] = 0\n case .wiggle:\n self.wiggleAmounts[provider] = amount\n self.blinkAmounts[provider] = 0\n self.tiltAmounts[provider] = 0\n case .tilt:\n self.tiltAmounts[provider] = amount\n self.blinkAmounts[provider] = 0\n self.wiggleAmounts[provider] = 0\n }\n }\n\n private func clearMotion(for provider: UsageProvider) {\n self.blinkAmounts[provider] = 0\n self.wiggleAmounts[provider] = 0\n self.tiltAmounts[provider] = 0\n }\n\n private func randomEffect(for provider: UsageProvider) -> MotionEffect {\n if provider == .claude {\n Bool.random() ? .blink : .wiggle\n } else {\n Bool.random() ? .blink : .tilt\n }\n }\n\n private func isBlinkingAllowed(at date: Date = .init()) -> Bool {\n if self.settings.randomBlinkEnabled { return true }\n if let until = self.blinkForceUntil, until > date { return true }\n self.blinkForceUntil = nil\n return false\n }\n\n func applyIcon(phase: Double?) {\n guard let button = self.statusItem.button else { return }\n\n let style = self.store.iconStyle\n let showUsed = self.settings.usageBarsShowUsed\n let showBrandPercent = self.settings.menuBarShowsBrandIconWithPercent\n let primaryProvider = self.primaryProviderForUnifiedIcon()\n let snapshot = self.store.snapshot(for: primaryProvider)\n\n // IconRenderer treats these values as a left-to-right \"progress fill\" percentage; depending on the\n // user setting we pass either \"percent left\" or \"percent used\".\n var primary = showUsed ? snapshot?.primary?.usedPercent : snapshot?.primary?.remainingPercent\n var weekly = showUsed ? snapshot?.secondary?.usedPercent : snapshot?.secondary?.remainingPercent\n if showUsed,\n primaryProvider == .warp,\n let remaining = snapshot?.secondary?.remainingPercent,\n remaining <= 0\n {\n // Preserve Warp \"no bonus/exhausted bonus\" layout even in show-used mode.\n weekly = 0\n }\n if showUsed,\n primaryProvider == .warp,\n let remaining = snapshot?.secondary?.remainingPercent,\n remaining > 0,\n weekly == 0\n {\n // In show-used mode, `0` means \"unused\", not \"missing\". Keep the weekly lane present.\n weekly = Self.loadingPercentEpsilon\n }\n var credits: Double? = primaryProvider == .codex ? self.store.credits?.remaining : nil\n var stale = self.store.isStale(provider: primaryProvider)\n var morphProgress: Double?\n\n let needsAnimation = self.needsMenuBarIconAnimation()\n if let phase, needsAnimation {\n var pattern = self.animationPattern\n if style == .combined, pattern == .unbraid {\n pattern = .cylon\n }\n if pattern == .unbraid {\n morphProgress = pattern.value(phase: phase) / 100\n primary = nil\n weekly = nil\n credits = nil\n stale = false\n } else {\n // Keep loading animation layout stable: IconRenderer uses `weeklyRemaining > 0` to switch layouts,\n // so hitting an exact 0 would flip between \"normal\" and \"weekly exhausted\" rendering.\n primary = max(pattern.value(phase: phase), Self.loadingPercentEpsilon)\n weekly = max(pattern.value(phase: phase + pattern.secondaryOffset), Self.loadingPercentEpsilon)\n credits = nil\n stale = false\n }\n }\n\n let blink: CGFloat = style == .combined ? 0 : self.blinkAmount(for: primaryProvider)\n let wiggle: CGFloat = style == .combined ? 0 : self.wiggleAmount(for: primaryProvider)\n let tilt: CGFloat = style == .combined ? 0 : self.tiltAmount(for: primaryProvider) * .pi / 28\n\n let statusIndicator: ProviderStatusIndicator = {\n for provider in self.store.enabledProvidersForDisplay() {\n let indicator = self.store.statusIndicator(for: provider)\n if indicator.hasIssue { return indicator }\n }\n return .none\n }()\n\n if showBrandPercent,\n let brand = ProviderBrandIcon.image(for: primaryProvider)\n {\n let displayText = self.menuBarDisplayText(for: primaryProvider, snapshot: snapshot)\n self.setButtonImage(brand, for: button)\n self.setButtonTitle(displayText, for: button)\n return\n }\n\n if Self.shouldUseOpenRouterBrandFallback(provider: primaryProvider, snapshot: snapshot),\n let brand = ProviderBrandIcon.image(for: primaryProvider)\n {\n self.setButtonTitle(nil, for: button)\n self.setButtonImage(\n Self.brandImageWithStatusOverlay(brand: brand, statusIndicator: statusIndicator),\n for: button)\n return\n }\n\n self.setButtonTitle(nil, for: button)\n if let morphProgress {\n let image = IconRenderer.makeMorphIcon(progress: morphProgress, style: style)\n self.setButtonImage(image, for: button)\n } else {\n let image = IconRenderer.makeIcon(\n primaryRemaining: primary,\n weeklyRemaining: weekly,\n creditsRemaining: credits,\n stale: stale,\n style: style,\n blink: blink,\n wiggle: wiggle,\n tilt: tilt,\n statusIndicator: statusIndicator)\n self.setButtonImage(image, for: button)\n }\n }\n\n func applyIcon(for provider: UsageProvider, phase: Double?) {\n guard let button = self.statusItems[provider]?.button else { return }\n let snapshot = self.store.snapshot(for: provider)\n // IconRenderer treats these values as a left-to-right \"progress fill\" percentage; depending on the\n // user setting we pass either \"percent left\" or \"percent used\".\n let showUsed = self.settings.usageBarsShowUsed\n let showBrandPercent = self.settings.menuBarShowsBrandIconWithPercent\n\n if showBrandPercent,\n let brand = ProviderBrandIcon.image(for: provider)\n {\n let displayText = self.menuBarDisplayText(for: provider, snapshot: snapshot)\n self.setButtonImage(brand, for: button)\n self.setButtonTitle(displayText, for: button)\n return\n }\n\n if Self.shouldUseOpenRouterBrandFallback(provider: provider, snapshot: snapshot),\n let brand = ProviderBrandIcon.image(for: provider)\n {\n self.setButtonTitle(nil, for: button)\n self.setButtonImage(\n Self.brandImageWithStatusOverlay(\n brand: brand,\n statusIndicator: self.store.statusIndicator(for: provider)),\n for: button)\n return\n }\n var primary = showUsed ? snapshot?.primary?.usedPercent : snapshot?.primary?.remainingPercent\n var weekly = showUsed ? snapshot?.secondary?.usedPercent : snapshot?.secondary?.remainingPercent\n if showUsed,\n provider == .warp,\n let remaining = snapshot?.secondary?.remainingPercent,\n remaining <= 0\n {\n // Preserve Warp \"no bonus/exhausted bonus\" layout even in show-used mode.\n weekly = 0\n }\n if showUsed,\n provider == .warp,\n let remaining = snapshot?.secondary?.remainingPercent,\n remaining > 0,\n weekly == 0\n {\n // In show-used mode, `0` means \"unused\", not \"missing\". Keep the weekly lane present.\n weekly = Self.loadingPercentEpsilon\n }\n var credits: Double? = provider == .codex ? self.store.credits?.remaining : nil\n var stale = self.store.isStale(provider: provider)\n var morphProgress: Double?\n\n if let phase, self.shouldAnimate(provider: provider) {\n var pattern = self.animationPattern\n if provider == .claude, pattern == .unbraid {\n pattern = .cylon\n }\n if pattern == .unbraid {\n morphProgress = pattern.value(phase: phase) / 100\n primary = nil\n weekly = nil\n credits = nil\n stale = false\n } else {\n // Keep loading animation layout stable: IconRenderer switches layouts at `weeklyRemaining == 0`.\n primary = max(pattern.value(phase: phase), Self.loadingPercentEpsilon)\n weekly = max(pattern.value(phase: phase + pattern.secondaryOffset), Self.loadingPercentEpsilon)\n credits = nil\n stale = false\n }\n }\n\n let style: IconStyle = self.store.style(for: provider)\n let isLoading = phase != nil && self.shouldAnimate(provider: provider)\n let blink: CGFloat = {\n guard isLoading, style == .warp, let phase else {\n return self.blinkAmount(for: provider)\n }\n let normalized = (sin(phase * 3) + 1) / 2\n return CGFloat(max(0, min(normalized, 1)))\n }()\n let wiggle = self.wiggleAmount(for: provider)\n let tilt = self.tiltAmount(for: provider) * .pi / 28 // limit to ~6.4°\n if let morphProgress {\n let image = IconRenderer.makeMorphIcon(progress: morphProgress, style: style)\n self.setButtonImage(image, for: button)\n } else {\n self.setButtonTitle(nil, for: button)\n let image = IconRenderer.makeIcon(\n primaryRemaining: primary,\n weeklyRemaining: weekly,\n creditsRemaining: credits,\n stale: stale,\n style: style,\n blink: blink,\n wiggle: wiggle,\n tilt: tilt,\n statusIndicator: self.store.statusIndicator(for: provider))\n self.setButtonImage(image, for: button)\n }\n }\n\n private func setButtonImage(_ image: NSImage, for button: NSStatusBarButton) {\n if button.image === image { return }\n button.image = image\n }\n\n private func setButtonTitle(_ title: String?, for button: NSStatusBarButton) {\n let value = title ?? \"\"\n if button.title != value {\n button.title = value\n }\n let position: NSControl.ImagePosition = value.isEmpty ? .imageOnly : .imageLeft\n if button.imagePosition != position {\n button.imagePosition = position\n }\n }\n\n func menuBarDisplayText(for provider: UsageProvider, snapshot: UsageSnapshot?) -> String? {\n let percentWindow = self.menuBarPercentWindow(for: provider, snapshot: snapshot)\n let mode = self.settings.menuBarDisplayMode\n let now = Date()\n let pace: UsagePace? = switch mode {\n case .percent:\n nil\n case .pace, .both:\n snapshot?.secondary.flatMap { window in\n self.store.weeklyPace(provider: provider, window: window, now: now)\n }\n }\n let displayText = MenuBarDisplayText.displayText(\n mode: mode,\n percentWindow: percentWindow,\n pace: pace,\n showUsed: self.settings.usageBarsShowUsed)\n\n let sessionExhausted = (snapshot?.primary?.remainingPercent ?? 100) <= 0\n let weeklyExhausted = (snapshot?.secondary?.remainingPercent ?? 100) <= 0\n\n if provider == .codex,\n mode == .percent,\n !self.settings.usageBarsShowUsed,\n sessionExhausted || weeklyExhausted,\n let creditsRemaining = self.store.credits?.remaining,\n creditsRemaining > 0\n {\n return UsageFormatter\n .creditsString(from: creditsRemaining)\n .replacingOccurrences(of: \" left\", with: \"\")\n }\n\n return displayText\n }\n\n private func menuBarPercentWindow(for provider: UsageProvider, snapshot: UsageSnapshot?) -> RateWindow? {\n self.menuBarMetricWindow(for: provider, snapshot: snapshot)\n }\n\n private func primaryProviderForUnifiedIcon() -> UsageProvider {\n // When \"show highest usage\" is enabled, auto-select the provider closest to rate limit.\n if self.settings.menuBarShowsHighestUsage,\n self.shouldMergeIcons,\n let highest = self.store.providerWithHighestUsage()\n {\n return highest.provider\n }\n if self.shouldMergeIcons,\n let selected = self.selectedMenuProvider,\n self.store.isEnabled(selected)\n {\n return selected\n }\n for provider in UsageProvider.allCases {\n if self.store.isEnabled(provider), self.store.snapshot(for: provider) != nil {\n return provider\n }\n }\n // Use availability-filtered list: fallback must pick a provider that can\n // actually animate, otherwise shouldAnimate() fails on credential-less providers.\n if let enabled = self.store.enabledProviders().first {\n return enabled\n }\n return .codex\n }\n\n @objc func handleDebugBlinkNotification() {\n self.forceBlinkNow()\n }\n\n private func forceBlinkNow() {\n let now = Date()\n self.blinkForceUntil = now.addingTimeInterval(0.6)\n self.seedBlinkStatesIfNeeded()\n\n for provider in UsageProvider.allCases {\n let shouldBlink = self.shouldMergeIcons ? self.isEnabled(provider) : self.isVisible(provider)\n guard shouldBlink, !self.shouldAnimate(provider: provider) else { continue }\n var state = self\n .blinkStates[provider] ?? BlinkState(nextBlink: now.addingTimeInterval(BlinkState.randomDelay()))\n state.blinkStart = now\n state.pendingSecondStart = nil\n state.effect = self.randomEffect(for: provider)\n state.nextBlink = now.addingTimeInterval(BlinkState.randomDelay())\n self.blinkStates[provider] = state\n self.assignMotion(amount: 0, for: provider, effect: state.effect)\n }\n\n // If the blink task is currently in a long idle sleep, restart it so this forced blink\n // keeps animating on the active frame cadence immediately.\n self.blinkTask?.cancel()\n self.blinkTask = nil\n self.updateBlinkingState()\n self.tickBlink(now: now)\n }\n\n private func shouldAnimate(provider: UsageProvider, mergeIcons: Bool? = nil) -> Bool {\n if self.store.debugForceAnimation { return true }\n\n let isMerged = mergeIcons ?? self.shouldMergeIcons\n let isVisible = isMerged ? self.isEnabled(provider) : self.isVisible(provider)\n guard isVisible else { return false }\n\n // Don't animate for fallback provider - it's only shown as a placeholder when nothing is enabled.\n // Animating the fallback causes unnecessary CPU usage (battery drain). See #269, #139.\n let isEnabled = self.isEnabled(provider)\n let isFallbackOnly = !isEnabled && self.fallbackProvider == provider\n if isFallbackOnly { return false }\n\n let isStale = self.store.isStale(provider: provider)\n let hasData = self.store.snapshot(for: provider) != nil\n if provider == .warp, !hasData, self.store.refreshingProviders.contains(provider) {\n return true\n }\n return !hasData && !isStale\n }\n\n func updateAnimationState() {\n let needsAnimation = self.needsMenuBarIconAnimation()\n if needsAnimation {\n if self.animationDriver == nil {\n if let forced = self.settings.debugLoadingPattern {\n self.animationPattern = forced\n } else if !LoadingPattern.allCases.contains(self.animationPattern) {\n self.animationPattern = .knightRider\n }\n self.animationPhase = 0\n let driver = DisplayLinkDriver(onTick: { [weak self] in\n self?.updateAnimationFrame()\n })\n self.animationDriver = driver\n driver.start(fps: 60)\n } else if let forced = self.settings.debugLoadingPattern, forced != self.animationPattern {\n self.animationPattern = forced\n self.animationPhase = 0\n }\n } else {\n self.animationDriver?.stop()\n self.animationDriver = nil\n self.animationPhase = 0\n if self.shouldMergeIcons {\n self.applyIcon(phase: nil)\n } else {\n UsageProvider.allCases.forEach { self.applyIcon(for: $0, phase: nil) }\n }\n }\n }\n\n private func updateAnimationFrame() {\n self.animationPhase += 0.045 // half-speed animation\n if self.shouldMergeIcons {\n self.applyIcon(phase: self.animationPhase)\n } else {\n UsageProvider.allCases.forEach { self.applyIcon(for: $0, phase: self.animationPhase) }\n }\n }\n\n nonisolated static func shouldUseOpenRouterBrandFallback(\n provider: UsageProvider,\n snapshot: UsageSnapshot?) -> Bool\n {\n guard provider == .openrouter,\n let openRouterUsage = snapshot?.openRouterUsage\n else {\n return false\n }\n return openRouterUsage.keyQuotaStatus == .noLimitConfigured\n }\n\n nonisolated static func brandImageWithStatusOverlay(\n brand: NSImage,\n statusIndicator: ProviderStatusIndicator) -> NSImage\n {\n guard statusIndicator.hasIssue else { return brand }\n\n let image = NSImage(size: brand.size)\n image.lockFocus()\n brand.draw(\n at: .zero,\n from: NSRect(origin: .zero, size: brand.size),\n operation: .sourceOver,\n fraction: 1.0)\n Self.drawBrandStatusOverlay(indicator: statusIndicator, size: brand.size)\n image.unlockFocus()\n image.isTemplate = brand.isTemplate\n return image\n }\n\n private nonisolated static func drawBrandStatusOverlay(indicator: ProviderStatusIndicator, size: NSSize) {\n guard indicator.hasIssue else { return }\n\n let color = NSColor.labelColor\n switch indicator {\n case .minor, .maintenance:\n let dotSize = CGSize(width: 4, height: 4)\n let dotOrigin = CGPoint(x: size.width - dotSize.width - 2, y: 2)\n color.setFill()\n NSBezierPath(ovalIn: CGRect(origin: dotOrigin, size: dotSize)).fill()\n case .major, .critical, .unknown:\n color.setFill()\n let lineRect = CGRect(x: size.width - 6, y: 4, width: 2, height: 6)\n NSBezierPath(roundedRect: lineRect, xRadius: 1, yRadius: 1).fill()\n let dotRect = CGRect(x: size.width - 6, y: 2, width: 2, height: 2)\n NSBezierPath(ovalIn: dotRect).fill()\n case .none:\n break\n }\n }\n\n private func advanceAnimationPattern() {\n let patterns = LoadingPattern.allCases\n if let idx = patterns.firstIndex(of: self.animationPattern) {\n let next = patterns.indices.contains(idx + 1) ? patterns[idx + 1] : patterns.first\n self.animationPattern = next ?? .knightRider\n } else {\n self.animationPattern = .knightRider\n }\n }\n\n @objc func handleDebugReplayNotification(_ notification: Notification) {\n if let raw = notification.userInfo?[\"pattern\"] as? String,\n let selected = LoadingPattern(rawValue: raw)\n {\n self.animationPattern = selected\n } else if let forced = self.settings.debugLoadingPattern {\n self.animationPattern = forced\n } else {\n self.advanceAnimationPattern()\n }\n self.animationPhase = 0\n self.updateAnimationState()\n }\n}\n" }, { "path": "Sources/CodexBar/StatusItemController+Menu.swift", "content": "import AppKit\nimport CodexBarCore\nimport Observation\nimport QuartzCore\nimport SwiftUI\n\nextension ProviderSwitcherSelection {\n fileprivate var provider: UsageProvider? {\n switch self {\n case .overview:\n nil\n case let .provider(provider):\n provider\n }\n }\n}\n\nprivate struct OverviewMenuCardRowView: View {\n let model: UsageMenuCardView.Model\n let width: CGFloat\n\n var body: some View {\n VStack(alignment: .leading, spacing: 0) {\n UsageMenuCardHeaderSectionView(\n model: self.model,\n showDivider: self.hasUsageBlock,\n width: self.width)\n if self.hasUsageBlock {\n UsageMenuCardUsageSectionView(\n model: self.model,\n showBottomDivider: false,\n bottomPadding: 6,\n width: self.width)\n }\n }\n .frame(width: self.width, alignment: .leading)\n }\n\n private var hasUsageBlock: Bool {\n !self.model.metrics.isEmpty || !self.model.usageNotes.isEmpty || self.model.placeholder != nil\n }\n}\n\n// MARK: - NSMenu construction\n\nextension StatusItemController {\n private static let menuCardBaseWidth: CGFloat = 310\n private static let maxOverviewProviders = SettingsStore.mergedOverviewProviderLimit\n private static let overviewRowIdentifierPrefix = \"overviewRow-\"\n private static let menuOpenRefreshDelay: Duration = .seconds(1.2)\n private struct OpenAIWebMenuItems {\n let hasUsageBreakdown: Bool\n let hasCreditsHistory: Bool\n let hasCostHistory: Bool\n }\n\n private struct TokenAccountMenuDisplay {\n let provider: UsageProvider\n let accounts: [ProviderTokenAccount]\n let snapshots: [TokenAccountUsageSnapshot]\n let activeIndex: Int\n let showAll: Bool\n let showSwitcher: Bool\n }\n\n private func menuCardWidth(for providers: [UsageProvider], menu: NSMenu? = nil) -> CGFloat {\n _ = menu\n return Self.menuCardBaseWidth\n }\n\n func makeMenu() -> NSMenu {\n guard self.shouldMergeIcons else {\n return self.makeMenu(for: nil)\n }\n let menu = NSMenu()\n menu.autoenablesItems = false\n menu.delegate = self\n return menu\n }\n\n func menuWillOpen(_ menu: NSMenu) {\n if self.isHostedSubviewMenu(menu) {\n self.refreshHostedSubviewHeights(in: menu)\n if Self.menuRefreshEnabled, self.isOpenAIWebSubviewMenu(menu) {\n self.store.requestOpenAIDashboardRefreshIfStale(reason: \"submenu open\")\n }\n self.openMenus[ObjectIdentifier(menu)] = menu\n // Removed redundant async refresh - single pass is sufficient after initial layout\n return\n }\n\n var provider: UsageProvider?\n if self.shouldMergeIcons {\n let resolvedProvider = self.resolvedMenuProvider()\n self.lastMenuProvider = resolvedProvider ?? .codex\n provider = resolvedProvider\n } else {\n if let menuProvider = self.menuProviders[ObjectIdentifier(menu)] {\n self.lastMenuProvider = menuProvider\n provider = menuProvider\n } else if menu === self.fallbackMenu {\n self.lastMenuProvider = self.store.enabledProvidersForDisplay().first ?? .codex\n provider = nil\n } else {\n let resolved = self.store.enabledProvidersForDisplay().first ?? .codex\n self.lastMenuProvider = resolved\n provider = resolved\n }\n }\n\n let didRefresh = self.menuNeedsRefresh(menu)\n if didRefresh {\n self.populateMenu(menu, provider: provider)\n self.markMenuFresh(menu)\n // Heights are already set during populateMenu, no need to remeasure\n }\n self.openMenus[ObjectIdentifier(menu)] = menu\n // Only schedule refresh after menu is registered as open - refreshNow is called async\n if Self.menuRefreshEnabled {\n self.scheduleOpenMenuRefresh(for: menu)\n }\n }\n\n func menuDidClose(_ menu: NSMenu) {\n let key = ObjectIdentifier(menu)\n\n self.openMenus.removeValue(forKey: key)\n self.menuRefreshTasks.removeValue(forKey: key)?.cancel()\n\n let isPersistentMenu = menu === self.mergedMenu ||\n menu === self.fallbackMenu ||\n self.providerMenus.values.contains { $0 === menu }\n if !isPersistentMenu {\n self.menuProviders.removeValue(forKey: key)\n self.menuVersions.removeValue(forKey: key)\n }\n for menuItem in menu.items {\n (menuItem.view as? MenuCardHighlighting)?.setHighlighted(false)\n }\n }\n\n func menu(_ menu: NSMenu, willHighlight item: NSMenuItem?) {\n for menuItem in menu.items {\n let highlighted = menuItem == item && menuItem.isEnabled\n (menuItem.view as? MenuCardHighlighting)?.setHighlighted(highlighted)\n }\n }\n\n private func populateMenu(_ menu: NSMenu, provider: UsageProvider?) {\n let enabledProviders = self.store.enabledProvidersForDisplay()\n let includesOverview = self.includesOverviewTab(enabledProviders: enabledProviders)\n let switcherSelection = self.shouldMergeIcons && enabledProviders.count > 1\n ? self.resolvedSwitcherSelection(\n enabledProviders: enabledProviders,\n includesOverview: includesOverview)\n : nil\n let isOverviewSelected = switcherSelection == .overview\n let selectedProvider = if isOverviewSelected {\n self.resolvedMenuProvider(enabledProviders: enabledProviders)\n } else {\n switcherSelection?.provider ?? provider\n }\n let menuWidth = self.menuCardWidth(for: enabledProviders, menu: menu)\n let currentProvider = selectedProvider ?? enabledProviders.first ?? .codex\n let tokenAccountDisplay = isOverviewSelected ? nil : self.tokenAccountMenuDisplay(for: currentProvider)\n let showAllTokenAccounts = tokenAccountDisplay?.showAll ?? false\n let openAIContext = self.openAIWebContext(\n currentProvider: currentProvider,\n showAllTokenAccounts: showAllTokenAccounts)\n\n let hasTokenAccountSwitcher = menu.items.contains { $0.view is TokenAccountSwitcherView }\n let switcherProvidersMatch = enabledProviders == self.lastSwitcherProviders\n let switcherUsageBarsShowUsedMatch = self.settings.usageBarsShowUsed == self.lastSwitcherUsageBarsShowUsed\n let switcherSelectionMatches = switcherSelection == self.lastMergedSwitcherSelection\n let switcherOverviewAvailabilityMatches = includesOverview == self.lastSwitcherIncludesOverview\n let canSmartUpdate = self.shouldMergeIcons &&\n enabledProviders.count > 1 &&\n !isOverviewSelected &&\n switcherProvidersMatch &&\n switcherUsageBarsShowUsedMatch &&\n switcherSelectionMatches &&\n switcherOverviewAvailabilityMatches &&\n tokenAccountDisplay == nil &&\n !hasTokenAccountSwitcher &&\n !menu.items.isEmpty &&\n menu.items.first?.view is ProviderSwitcherView\n\n if canSmartUpdate {\n self.updateMenuContent(\n menu,\n provider: selectedProvider,\n currentProvider: currentProvider,\n menuWidth: menuWidth,\n openAIContext: openAIContext)\n return\n }\n\n menu.removeAllItems()\n\n let descriptor = MenuDescriptor.build(\n provider: selectedProvider,\n store: self.store,\n settings: self.settings,\n account: self.account,\n updateReady: self.updater.updateStatus.isUpdateReady,\n includeContextualActions: !isOverviewSelected)\n\n self.addProviderSwitcherIfNeeded(\n to: menu,\n enabledProviders: enabledProviders,\n includesOverview: includesOverview,\n selection: switcherSelection ?? .provider(currentProvider))\n // Track which providers the switcher was built with for smart update detection\n if self.shouldMergeIcons, enabledProviders.count > 1 {\n self.lastSwitcherProviders = enabledProviders\n self.lastSwitcherUsageBarsShowUsed = self.settings.usageBarsShowUsed\n self.lastMergedSwitcherSelection = switcherSelection\n self.lastSwitcherIncludesOverview = includesOverview\n }\n self.addTokenAccountSwitcherIfNeeded(to: menu, display: tokenAccountDisplay)\n let menuContext = MenuCardContext(\n currentProvider: currentProvider,\n selectedProvider: selectedProvider,\n menuWidth: menuWidth,\n tokenAccountDisplay: tokenAccountDisplay,\n openAIContext: openAIContext)\n if isOverviewSelected {\n if self.addOverviewRows(\n to: menu,\n enabledProviders: enabledProviders,\n menuWidth: menuWidth)\n {\n menu.addItem(.separator())\n } else {\n self.addOverviewEmptyState(to: menu, enabledProviders: enabledProviders)\n menu.addItem(.separator())\n }\n } else {\n let addedOpenAIWebItems = self.addMenuCards(to: menu, context: menuContext)\n self.addOpenAIWebItemsIfNeeded(\n to: menu,\n currentProvider: currentProvider,\n context: openAIContext,\n addedOpenAIWebItems: addedOpenAIWebItems)\n }\n self.addActionableSections(descriptor.sections, to: menu, width: menuWidth)\n }\n\n /// Smart update: only rebuild content sections when switching providers (keep the switcher intact).\n private func updateMenuContent(\n _ menu: NSMenu,\n provider: UsageProvider?,\n currentProvider: UsageProvider,\n menuWidth: CGFloat,\n openAIContext: OpenAIWebContext)\n {\n // Batch menu updates to prevent visual flickering during provider switch.\n CATransaction.begin()\n CATransaction.setDisableActions(true)\n defer { CATransaction.commit() }\n\n var contentStartIndex = 0\n if menu.items.first?.view is ProviderSwitcherView {\n contentStartIndex = 2\n }\n if menu.items.count > contentStartIndex,\n menu.items[contentStartIndex].view is TokenAccountSwitcherView\n {\n contentStartIndex += 2\n }\n while menu.items.count > contentStartIndex {\n menu.removeItem(at: contentStartIndex)\n }\n\n let descriptor = MenuDescriptor.build(\n provider: provider,\n store: self.store,\n settings: self.settings,\n account: self.account,\n updateReady: self.updater.updateStatus.isUpdateReady)\n\n let menuContext = MenuCardContext(\n currentProvider: currentProvider,\n selectedProvider: provider,\n menuWidth: menuWidth,\n tokenAccountDisplay: nil,\n openAIContext: openAIContext)\n let addedOpenAIWebItems = self.addMenuCards(to: menu, context: menuContext)\n self.addOpenAIWebItemsIfNeeded(\n to: menu,\n currentProvider: currentProvider,\n context: openAIContext,\n addedOpenAIWebItems: addedOpenAIWebItems)\n self.addActionableSections(descriptor.sections, to: menu, width: menuWidth)\n }\n\n private struct OpenAIWebContext {\n let hasUsageBreakdown: Bool\n let hasCreditsHistory: Bool\n let hasCostHistory: Bool\n let hasOpenAIWebMenuItems: Bool\n }\n\n private struct MenuCardContext {\n let currentProvider: UsageProvider\n let selectedProvider: UsageProvider?\n let menuWidth: CGFloat\n let tokenAccountDisplay: TokenAccountMenuDisplay?\n let openAIContext: OpenAIWebContext\n }\n\n private func openAIWebContext(\n currentProvider: UsageProvider,\n showAllTokenAccounts: Bool) -> OpenAIWebContext\n {\n let dashboard = self.store.openAIDashboard\n let openAIWebEligible = currentProvider == .codex &&\n self.store.openAIDashboardRequiresLogin == false &&\n dashboard != nil\n let hasCreditsHistory = openAIWebEligible && !(dashboard?.dailyBreakdown ?? []).isEmpty\n let hasUsageBreakdown = openAIWebEligible && !(dashboard?.usageBreakdown ?? []).isEmpty\n let hasCostHistory = self.settings.isCostUsageEffectivelyEnabled(for: currentProvider) &&\n (self.store.tokenSnapshot(for: currentProvider)?.daily.isEmpty == false)\n let hasOpenAIWebMenuItems = !showAllTokenAccounts &&\n (hasCreditsHistory || hasUsageBreakdown || hasCostHistory)\n return OpenAIWebContext(\n hasUsageBreakdown: hasUsageBreakdown,\n hasCreditsHistory: hasCreditsHistory,\n hasCostHistory: hasCostHistory,\n hasOpenAIWebMenuItems: hasOpenAIWebMenuItems)\n }\n\n private func addProviderSwitcherIfNeeded(\n to menu: NSMenu,\n enabledProviders: [UsageProvider],\n includesOverview: Bool,\n selection: ProviderSwitcherSelection)\n {\n guard self.shouldMergeIcons, enabledProviders.count > 1 else { return }\n let switcherItem = self.makeProviderSwitcherItem(\n providers: enabledProviders,\n includesOverview: includesOverview,\n selected: selection,\n menu: menu)\n menu.addItem(switcherItem)\n menu.addItem(.separator())\n }\n\n private func addTokenAccountSwitcherIfNeeded(to menu: NSMenu, display: TokenAccountMenuDisplay?) {\n guard let display, display.showSwitcher else { return }\n let switcherItem = self.makeTokenAccountSwitcherItem(display: display, menu: menu)\n menu.addItem(switcherItem)\n menu.addItem(.separator())\n }\n\n @discardableResult\n private func addOverviewRows(\n to menu: NSMenu,\n enabledProviders: [UsageProvider],\n menuWidth: CGFloat) -> Bool\n {\n let overviewProviders = self.settings.reconcileMergedOverviewSelectedProviders(\n activeProviders: enabledProviders)\n let rows: [(provider: UsageProvider, model: UsageMenuCardView.Model)] = overviewProviders\n .compactMap { provider in\n guard let model = self.menuCardModel(for: provider) else { return nil }\n return (provider: provider, model: model)\n }\n guard !rows.isEmpty else { return false }\n\n for (index, row) in rows.enumerated() {\n let identifier = \"\\(Self.overviewRowIdentifierPrefix)\\(row.provider.rawValue)\"\n let item = self.makeMenuCardItem(\n OverviewMenuCardRowView(model: row.model, width: menuWidth),\n id: identifier,\n width: menuWidth,\n onClick: { [weak self, weak menu] in\n guard let self, let menu else { return }\n self.selectOverviewProvider(row.provider, menu: menu)\n })\n // Keep menu item action wired for keyboard activation and accessibility action paths.\n item.target = self\n item.action = #selector(self.selectOverviewProvider(_:))\n menu.addItem(item)\n if index < rows.count - 1 {\n menu.addItem(.separator())\n }\n }\n return true\n }\n\n private func addOverviewEmptyState(to menu: NSMenu, enabledProviders: [UsageProvider]) {\n let resolvedProviders = self.settings.resolvedMergedOverviewProviders(\n activeProviders: enabledProviders,\n maxVisibleProviders: Self.maxOverviewProviders)\n let message = if resolvedProviders.isEmpty {\n \"No providers selected for Overview.\"\n } else {\n \"No overview data available.\"\n }\n let item = NSMenuItem(title: message, action: nil, keyEquivalent: \"\")\n item.isEnabled = false\n item.representedObject = \"overviewEmptyState\"\n menu.addItem(item)\n }\n\n private func addMenuCards(to menu: NSMenu, context: MenuCardContext) -> Bool {\n if let tokenAccountDisplay = context.tokenAccountDisplay, tokenAccountDisplay.showAll {\n let accountSnapshots = tokenAccountDisplay.snapshots\n let cards = accountSnapshots.isEmpty\n ? []\n : accountSnapshots.compactMap { accountSnapshot in\n self.menuCardModel(\n for: context.currentProvider,\n snapshotOverride: accountSnapshot.snapshot,\n errorOverride: accountSnapshot.error)\n }\n if cards.isEmpty, let model = self.menuCardModel(for: context.selectedProvider) {\n menu.addItem(self.makeMenuCardItem(\n UsageMenuCardView(model: model, width: context.menuWidth),\n id: \"menuCard\",\n width: context.menuWidth))\n menu.addItem(.separator())\n } else {\n for (index, model) in cards.enumerated() {\n menu.addItem(self.makeMenuCardItem(\n UsageMenuCardView(model: model, width: context.menuWidth),\n id: \"menuCard-\\(index)\",\n width: context.menuWidth))\n if index < cards.count - 1 {\n menu.addItem(.separator())\n }\n }\n if !cards.isEmpty {\n menu.addItem(.separator())\n }\n }\n return false\n }\n\n guard let model = self.menuCardModel(for: context.selectedProvider) else { return false }\n if context.openAIContext.hasOpenAIWebMenuItems {\n let webItems = OpenAIWebMenuItems(\n hasUsageBreakdown: context.openAIContext.hasUsageBreakdown,\n hasCreditsHistory: context.openAIContext.hasCreditsHistory,\n hasCostHistory: context.openAIContext.hasCostHistory)\n self.addMenuCardSections(\n to: menu,\n model: model,\n provider: context.currentProvider,\n width: context.menuWidth,\n webItems: webItems)\n return true\n }\n\n menu.addItem(self.makeMenuCardItem(\n UsageMenuCardView(model: model, width: context.menuWidth),\n id: \"menuCard\",\n width: context.menuWidth))\n if context.currentProvider == .codex, model.creditsText != nil {\n menu.addItem(self.makeBuyCreditsItem())\n }\n menu.addItem(.separator())\n return false\n }\n\n private func addOpenAIWebItemsIfNeeded(\n to menu: NSMenu,\n currentProvider: UsageProvider,\n context: OpenAIWebContext,\n addedOpenAIWebItems: Bool)\n {\n guard context.hasOpenAIWebMenuItems else { return }\n if !addedOpenAIWebItems {\n // Only show these when we actually have additional data.\n if context.hasUsageBreakdown {\n _ = self.addUsageBreakdownSubmenu(to: menu)\n }\n if context.hasCreditsHistory {\n _ = self.addCreditsHistorySubmenu(to: menu)\n }\n if context.hasCostHistory {\n _ = self.addCostHistorySubmenu(to: menu, provider: currentProvider)\n }\n }\n menu.addItem(.separator())\n }\n\n private func addActionableSections(_ sections: [MenuDescriptor.Section], to menu: NSMenu, width: CGFloat) {\n let actionableSections = sections.filter { section in\n section.entries.contains { entry in\n if case .action = entry { return true }\n return false\n }\n }\n for (index, section) in actionableSections.enumerated() {\n for entry in section.entries {\n switch entry {\n case let .text(text, style):\n if style == .secondary {\n menu.addItem(self.makeWrappedSecondaryTextItem(text: text, width: width))\n continue\n }\n let item = NSMenuItem(title: text, action: nil, keyEquivalent: \"\")\n item.isEnabled = false\n if style == .headline {\n let font = NSFont.systemFont(ofSize: NSFont.systemFontSize, weight: .semibold)\n item.attributedTitle = NSAttributedString(string: text, attributes: [.font: font])\n } else if style == .secondary {\n let font = NSFont.systemFont(ofSize: NSFont.smallSystemFontSize)\n item.attributedTitle = NSAttributedString(\n string: text,\n attributes: [.font: font, .foregroundColor: NSColor.secondaryLabelColor])\n }\n menu.addItem(item)\n case let .action(title, action):\n let (selector, represented) = self.selector(for: action)\n let item = NSMenuItem(title: title, action: selector, keyEquivalent: \"\")\n item.target = self\n item.representedObject = represented\n if let iconName = action.systemImageName,\n let image = NSImage(systemSymbolName: iconName, accessibilityDescription: nil)\n {\n image.isTemplate = true\n image.size = NSSize(width: 16, height: 16)\n item.image = image\n }\n if case let .switchAccount(targetProvider) = action,\n let subtitle = self.switchAccountSubtitle(for: targetProvider)\n {\n item.isEnabled = false\n self.applySubtitle(subtitle, to: item, title: title)\n }\n menu.addItem(item)\n case .divider:\n menu.addItem(.separator())\n }\n }\n if index < actionableSections.count - 1 {\n menu.addItem(.separator())\n }\n }\n }\n\n private func makeWrappedSecondaryTextItem(text: String, width: CGFloat) -> NSMenuItem {\n let item = NSMenuItem(title: text, action: nil, keyEquivalent: \"\")\n let view = self.makeWrappedSecondaryTextView(text: text)\n let height = self.menuTextItemHeight(for: view, width: width)\n view.frame = NSRect(origin: .zero, size: NSSize(width: width, height: height))\n item.view = view\n item.isEnabled = false\n item.toolTip = text\n return item\n }\n\n private func makeWrappedSecondaryTextView(text: String) -> NSView {\n let container = NSView()\n container.translatesAutoresizingMaskIntoConstraints = false\n\n let textField = NSTextField(wrappingLabelWithString: text)\n textField.font = NSFont.menuFont(ofSize: NSFont.smallSystemFontSize)\n textField.textColor = NSColor.secondaryLabelColor\n textField.lineBreakMode = .byWordWrapping\n textField.maximumNumberOfLines = 0\n textField.setContentCompressionResistancePriority(.defaultLow, for: .horizontal)\n textField.translatesAutoresizingMaskIntoConstraints = false\n\n container.addSubview(textField)\n NSLayoutConstraint.activate([\n textField.leadingAnchor.constraint(equalTo: container.leadingAnchor, constant: 18),\n textField.trailingAnchor.constraint(equalTo: container.trailingAnchor, constant: -10),\n textField.topAnchor.constraint(equalTo: container.topAnchor, constant: 2),\n textField.bottomAnchor.constraint(equalTo: container.bottomAnchor, constant: -2),\n ])\n\n return container\n }\n\n private func menuTextItemHeight(for view: NSView, width: CGFloat) -> CGFloat {\n view.frame = NSRect(origin: .zero, size: NSSize(width: width, height: 1))\n view.layoutSubtreeIfNeeded()\n return max(1, ceil(view.fittingSize.height))\n }\n\n func makeMenu(for provider: UsageProvider?) -> NSMenu {\n let menu = NSMenu()\n menu.autoenablesItems = false\n menu.delegate = self\n if let provider {\n self.menuProviders[ObjectIdentifier(menu)] = provider\n }\n return menu\n }\n\n private func makeProviderSwitcherItem(\n providers: [UsageProvider],\n includesOverview: Bool,\n selected: ProviderSwitcherSelection,\n menu: NSMenu) -> NSMenuItem\n {\n let view = ProviderSwitcherView(\n providers: providers,\n selected: selected,\n includesOverview: includesOverview,\n width: self.menuCardWidth(for: providers, menu: menu),\n showsIcons: self.settings.switcherShowsIcons,\n iconProvider: { [weak self] provider in\n self?.switcherIcon(for: provider) ?? NSImage()\n },\n weeklyRemainingProvider: { [weak self] provider in\n self?.switcherWeeklyRemaining(for: provider)\n },\n onSelect: { [weak self, weak menu] selection in\n guard let self, let menu else { return }\n switch selection {\n case .overview:\n self.settings.mergedMenuLastSelectedWasOverview = true\n self.lastMergedSwitcherSelection = .overview\n let provider = self.resolvedMenuProvider()\n self.lastMenuProvider = provider ?? .codex\n self.populateMenu(menu, provider: provider)\n case let .provider(provider):\n self.settings.mergedMenuLastSelectedWasOverview = false\n self.lastMergedSwitcherSelection = .provider(provider)\n self.selectedMenuProvider = provider\n self.lastMenuProvider = provider\n self.populateMenu(menu, provider: provider)\n }\n self.markMenuFresh(menu)\n self.applyIcon(phase: nil)\n })\n let item = NSMenuItem()\n item.view = view\n item.isEnabled = false\n return item\n }\n\n private func makeTokenAccountSwitcherItem(\n display: TokenAccountMenuDisplay,\n menu: NSMenu) -> NSMenuItem\n {\n let view = TokenAccountSwitcherView(\n accounts: display.accounts,\n selectedIndex: display.activeIndex,\n width: self.menuCardWidth(for: self.store.enabledProvidersForDisplay(), menu: menu),\n onSelect: { [weak self, weak menu] index in\n guard let self, let menu else { return }\n self.settings.setActiveTokenAccountIndex(index, for: display.provider)\n Task { @MainActor in\n await ProviderInteractionContext.$current.withValue(.userInitiated) {\n await self.store.refresh()\n }\n }\n self.populateMenu(menu, provider: display.provider)\n self.markMenuFresh(menu)\n self.applyIcon(phase: nil)\n })\n let item = NSMenuItem()\n item.view = view\n item.isEnabled = false\n return item\n }\n\n private func resolvedMenuProvider(enabledProviders: [UsageProvider]? = nil) -> UsageProvider? {\n let enabled = enabledProviders ?? self.store.enabledProvidersForDisplay()\n if enabled.isEmpty { return .codex }\n if let selected = self.selectedMenuProvider, enabled.contains(selected) {\n return selected\n }\n // Prefer an available provider so the default menu content matches the status icon.\n // Falls back to first display provider when all lack credentials.\n return enabled.first(where: { self.store.isProviderAvailable($0) }) ?? enabled.first\n }\n\n private func includesOverviewTab(enabledProviders: [UsageProvider]) -> Bool {\n !self.settings.resolvedMergedOverviewProviders(\n activeProviders: enabledProviders,\n maxVisibleProviders: Self.maxOverviewProviders).isEmpty\n }\n\n private func resolvedSwitcherSelection(\n enabledProviders: [UsageProvider],\n includesOverview: Bool) -> ProviderSwitcherSelection\n {\n if includesOverview, self.settings.mergedMenuLastSelectedWasOverview {\n return .overview\n }\n return .provider(self.resolvedMenuProvider(enabledProviders: enabledProviders) ?? .codex)\n }\n\n private func tokenAccountMenuDisplay(for provider: UsageProvider) -> TokenAccountMenuDisplay? {\n guard TokenAccountSupportCatalog.support(for: provider) != nil else { return nil }\n let accounts = self.settings.tokenAccounts(for: provider)\n guard accounts.count > 1 else { return nil }\n let activeIndex = self.settings.tokenAccountsData(for: provider)?.clampedActiveIndex() ?? 0\n let showAll = self.settings.showAllTokenAccountsInMenu\n let snapshots = showAll ? (self.store.accountSnapshots[provider] ?? []) : []\n return TokenAccountMenuDisplay(\n provider: provider,\n accounts: accounts,\n snapshots: snapshots,\n activeIndex: activeIndex,\n showAll: showAll,\n showSwitcher: !showAll)\n }\n\n private func menuNeedsRefresh(_ menu: NSMenu) -> Bool {\n let key = ObjectIdentifier(menu)\n return self.menuVersions[key] != self.menuContentVersion\n }\n\n private func markMenuFresh(_ menu: NSMenu) {\n let key = ObjectIdentifier(menu)\n self.menuVersions[key] = self.menuContentVersion\n }\n\n func refreshOpenMenusIfNeeded() {\n guard !self.openMenus.isEmpty else { return }\n for (key, menu) in self.openMenus {\n guard key == ObjectIdentifier(menu) else {\n // Clean up orphaned menu entries from all tracking dictionaries\n self.openMenus.removeValue(forKey: key)\n self.menuRefreshTasks.removeValue(forKey: key)?.cancel()\n self.menuProviders.removeValue(forKey: key)\n self.menuVersions.removeValue(forKey: key)\n continue\n }\n\n if self.isHostedSubviewMenu(menu) {\n self.refreshHostedSubviewHeights(in: menu)\n continue\n }\n\n if self.menuNeedsRefresh(menu) {\n let provider = self.menuProvider(for: menu)\n self.populateMenu(menu, provider: provider)\n self.markMenuFresh(menu)\n // Heights are already set during populateMenu, no need to remeasure\n }\n }\n }\n\n private func menuProvider(for menu: NSMenu) -> UsageProvider? {\n if self.shouldMergeIcons {\n return self.resolvedMenuProvider()\n }\n if let provider = self.menuProviders[ObjectIdentifier(menu)] {\n return provider\n }\n if menu === self.fallbackMenu {\n return nil\n }\n return self.store.enabledProvidersForDisplay().first ?? .codex\n }\n\n private func scheduleOpenMenuRefresh(for menu: NSMenu) {\n // Kick off a user-initiated refresh on open (non-forced) and re-check after a delay.\n // NEVER block menu opening with network requests.\n if !self.store.isRefreshing {\n self.refreshStore(forceTokenUsage: false)\n }\n let key = ObjectIdentifier(menu)\n self.menuRefreshTasks[key]?.cancel()\n self.menuRefreshTasks[key] = Task { @MainActor [weak self, weak menu] in\n guard let self, let menu else { return }\n try? await Task.sleep(for: Self.menuOpenRefreshDelay)\n guard !Task.isCancelled else { return }\n guard self.openMenus[ObjectIdentifier(menu)] != nil else { return }\n guard !self.store.isRefreshing else { return }\n guard self.menuNeedsDelayedRefreshRetry(for: menu) else { return }\n self.refreshStore(forceTokenUsage: false)\n }\n }\n\n private func menuNeedsDelayedRefreshRetry(for menu: NSMenu) -> Bool {\n let providersToCheck = self.delayedRefreshRetryProviders(for: menu)\n guard !providersToCheck.isEmpty else { return false }\n return providersToCheck.contains { provider in\n self.store.isStale(provider: provider) || self.store.snapshot(for: provider) == nil\n }\n }\n\n private func delayedRefreshRetryProviders(for menu: NSMenu) -> [UsageProvider] {\n let enabledProviders = self.store.enabledProvidersForDisplay()\n guard !enabledProviders.isEmpty else { return [] }\n let includesOverview = self.includesOverviewTab(enabledProviders: enabledProviders)\n\n if self.shouldMergeIcons,\n enabledProviders.count > 1,\n self.resolvedSwitcherSelection(\n enabledProviders: enabledProviders,\n includesOverview: includesOverview) == .overview\n {\n return self.settings.resolvedMergedOverviewProviders(\n activeProviders: enabledProviders,\n maxVisibleProviders: Self.maxOverviewProviders)\n }\n\n if let provider = self.menuProvider(for: menu)\n ?? self.resolvedMenuProvider(enabledProviders: enabledProviders)\n {\n return [provider]\n }\n return enabledProviders\n }\n\n private func refreshMenuCardHeights(in menu: NSMenu) {\n // Re-measure the menu card height right before display to avoid stale/incorrect sizing when content\n // changes (e.g. dashboard error lines causing wrapping).\n let cardItems = menu.items.filter { item in\n (item.representedObject as? String)?.hasPrefix(\"menuCard\") == true\n }\n for item in cardItems {\n guard let view = item.view else { continue }\n let width = self.menuCardWidth(for: self.store.enabledProvidersForDisplay(), menu: menu)\n let height = self.menuCardHeight(for: view, width: width)\n view.frame = NSRect(\n origin: .zero,\n size: NSSize(width: width, height: height))\n }\n }\n\n private func makeMenuCardItem(\n _ view: some View,\n id: String,\n width: CGFloat,\n submenu: NSMenu? = nil,\n onClick: (() -> Void)? = nil) -> NSMenuItem\n {\n if !Self.menuCardRenderingEnabled {\n let item = NSMenuItem()\n item.isEnabled = true\n item.representedObject = id\n item.submenu = submenu\n if submenu != nil {\n item.target = self\n item.action = #selector(self.menuCardNoOp(_:))\n }\n return item\n }\n\n let highlightState = MenuCardHighlightState()\n let wrapped = MenuCardSectionContainerView(\n highlightState: highlightState,\n showsSubmenuIndicator: submenu != nil)\n {\n view\n }\n let hosting = MenuCardItemHostingView(rootView: wrapped, highlightState: highlightState, onClick: onClick)\n // Set frame with target width immediately\n let height = self.menuCardHeight(for: hosting, width: width)\n hosting.frame = NSRect(origin: .zero, size: NSSize(width: width, height: height))\n let item = NSMenuItem()\n item.view = hosting\n item.isEnabled = true\n item.representedObject = id\n item.submenu = submenu\n if submenu != nil {\n item.target = self\n item.action = #selector(self.menuCardNoOp(_:))\n }\n return item\n }\n\n private func menuCardHeight(for view: NSView, width: CGFloat) -> CGFloat {\n let basePadding: CGFloat = 6\n let descenderSafety: CGFloat = 1\n\n // Fast path: use protocol-based measurement when available (avoids layout passes)\n if let measured = view as? MenuCardMeasuring {\n return max(1, ceil(measured.measuredHeight(width: width) + basePadding + descenderSafety))\n }\n\n // Set frame with target width before measuring.\n view.frame = NSRect(origin: .zero, size: NSSize(width: width, height: 1))\n\n // Use fittingSize directly - SwiftUI hosting views respect the frame width for wrapping\n let fitted = view.fittingSize\n\n return max(1, ceil(fitted.height + basePadding + descenderSafety))\n }\n\n private func addMenuCardSections(\n to menu: NSMenu,\n model: UsageMenuCardView.Model,\n provider: UsageProvider,\n width: CGFloat,\n webItems: OpenAIWebMenuItems)\n {\n let hasUsageBlock = !model.metrics.isEmpty || model.placeholder != nil\n let hasCredits = model.creditsText != nil\n let hasExtraUsage = model.providerCost != nil\n let hasCost = model.tokenUsage != nil\n let bottomPadding = CGFloat(hasCredits ? 4 : 6)\n let sectionSpacing = CGFloat(6)\n let usageBottomPadding = bottomPadding\n let creditsBottomPadding = bottomPadding\n\n let headerView = UsageMenuCardHeaderSectionView(\n model: model,\n showDivider: hasUsageBlock,\n width: width)\n menu.addItem(self.makeMenuCardItem(headerView, id: \"menuCardHeader\", width: width))\n\n if hasUsageBlock {\n let usageView = UsageMenuCardUsageSectionView(\n model: model,\n showBottomDivider: false,\n bottomPadding: usageBottomPadding,\n width: width)\n let usageSubmenu = self.makeUsageSubmenu(\n provider: provider,\n snapshot: self.store.snapshot(for: provider),\n webItems: webItems)\n menu.addItem(self.makeMenuCardItem(\n usageView,\n id: \"menuCardUsage\",\n width: width,\n submenu: usageSubmenu))\n }\n\n if hasCredits || hasExtraUsage || hasCost {\n menu.addItem(.separator())\n }\n\n if hasCredits {\n if hasExtraUsage || hasCost {\n menu.addItem(.separator())\n }\n let creditsView = UsageMenuCardCreditsSectionView(\n model: model,\n showBottomDivider: false,\n topPadding: sectionSpacing,\n bottomPadding: creditsBottomPadding,\n width: width)\n let creditsSubmenu = webItems.hasCreditsHistory ? self.makeCreditsHistorySubmenu() : nil\n menu.addItem(self.makeMenuCardItem(\n creditsView,\n id: \"menuCardCredits\",\n width: width,\n submenu: creditsSubmenu))\n if provider == .codex {\n menu.addItem(self.makeBuyCreditsItem())\n }\n }\n if hasExtraUsage {\n if hasCredits {\n menu.addItem(.separator())\n }\n let extraUsageView = UsageMenuCardExtraUsageSectionView(\n model: model,\n topPadding: sectionSpacing,\n bottomPadding: bottomPadding,\n width: width)\n menu.addItem(self.makeMenuCardItem(\n extraUsageView,\n id: \"menuCardExtraUsage\",\n width: width))\n }\n if hasCost {\n if hasCredits || hasExtraUsage {\n menu.addItem(.separator())\n }\n let costView = UsageMenuCardCostSectionView(\n model: model,\n topPadding: sectionSpacing,\n bottomPadding: bottomPadding,\n width: width)\n let costSubmenu = webItems.hasCostHistory ? self.makeCostHistorySubmenu(provider: provider) : nil\n menu.addItem(self.makeMenuCardItem(\n costView,\n id: \"menuCardCost\",\n width: width,\n submenu: costSubmenu))\n }\n }\n\n private func switcherIcon(for provider: UsageProvider) -> NSImage {\n if let brand = ProviderBrandIcon.image(for: provider) {\n return brand\n }\n\n // Fallback to the dynamic icon renderer if resources are missing (e.g. dev bundle mismatch).\n let snapshot = self.store.snapshot(for: provider)\n let showUsed = self.settings.usageBarsShowUsed\n let primary = showUsed ? snapshot?.primary?.usedPercent : snapshot?.primary?.remainingPercent\n var weekly = showUsed ? snapshot?.secondary?.usedPercent : snapshot?.secondary?.remainingPercent\n if showUsed,\n provider == .warp,\n let remaining = snapshot?.secondary?.remainingPercent,\n remaining <= 0\n {\n // Preserve Warp \"no bonus/exhausted bonus\" layout even in show-used mode.\n weekly = 0\n }\n if showUsed,\n provider == .warp,\n let remaining = snapshot?.secondary?.remainingPercent,\n remaining > 0,\n weekly == 0\n {\n // In show-used mode, `0` means \"unused\", not \"missing\". Keep the weekly lane present.\n weekly = 0.0001\n }\n let credits = provider == .codex ? self.store.credits?.remaining : nil\n let stale = self.store.isStale(provider: provider)\n let style = self.store.style(for: provider)\n let indicator = self.store.statusIndicator(for: provider)\n let image = IconRenderer.makeIcon(\n primaryRemaining: primary,\n weeklyRemaining: weekly,\n creditsRemaining: credits,\n stale: stale,\n style: style,\n blink: 0,\n wiggle: 0,\n tilt: 0,\n statusIndicator: indicator)\n image.isTemplate = true\n return image\n }\n\n nonisolated static func switcherWeeklyMetricPercent(\n for provider: UsageProvider,\n snapshot: UsageSnapshot?,\n showUsed: Bool) -> Double?\n {\n let window = snapshot?.switcherWeeklyWindow(for: provider, showUsed: showUsed)\n guard let window else { return nil }\n return showUsed ? window.usedPercent : window.remainingPercent\n }\n\n private func switcherWeeklyRemaining(for provider: UsageProvider) -> Double? {\n Self.switcherWeeklyMetricPercent(\n for: provider,\n snapshot: self.store.snapshot(for: provider),\n showUsed: self.settings.usageBarsShowUsed)\n }\n\n private func selector(for action: MenuDescriptor.MenuAction) -> (Selector, Any?) {\n switch action {\n case .installUpdate: (#selector(self.installUpdate), nil)\n case .refresh: (#selector(self.refreshNow), nil)\n case .refreshAugmentSession: (#selector(self.refreshAugmentSession), nil)\n case .dashboard: (#selector(self.openDashboard), nil)\n case .statusPage: (#selector(self.openStatusPage), nil)\n case let .switchAccount(provider): (#selector(self.runSwitchAccount(_:)), provider.rawValue)\n case let .openTerminal(command): (#selector(self.openTerminalCommand(_:)), command)\n case let .loginToProvider(url): (#selector(self.openLoginToProvider(_:)), url)\n case .settings: (#selector(self.showSettingsGeneral), nil)\n case .about: (#selector(self.showSettingsAbout), nil)\n case .quit: (#selector(self.quit), nil)\n case let .copyError(message): (#selector(self.copyError(_:)), message)\n }\n }\n\n @MainActor\n private protocol MenuCardHighlighting: AnyObject {\n func setHighlighted(_ highlighted: Bool)\n }\n\n @MainActor\n private protocol MenuCardMeasuring: AnyObject {\n func measuredHeight(width: CGFloat) -> CGFloat\n }\n\n @MainActor\n @Observable\n fileprivate final class MenuCardHighlightState {\n var isHighlighted = false\n }\n\n private final class MenuHostingView: NSHostingView {\n override var allowsVibrancy: Bool {\n true\n }\n }\n\n @MainActor\n private final class MenuCardItemHostingView: NSHostingView, MenuCardHighlighting,\n MenuCardMeasuring {\n private let highlightState: MenuCardHighlightState\n private let onClick: (() -> Void)?\n override var allowsVibrancy: Bool {\n true\n }\n\n override var intrinsicContentSize: NSSize {\n let size = super.intrinsicContentSize\n guard self.frame.width > 0 else { return size }\n return NSSize(width: self.frame.width, height: size.height)\n }\n\n init(rootView: Content, highlightState: MenuCardHighlightState, onClick: (() -> Void)? = nil) {\n self.highlightState = highlightState\n self.onClick = onClick\n super.init(rootView: rootView)\n if onClick != nil {\n let recognizer = NSClickGestureRecognizer(target: self, action: #selector(self.handlePrimaryClick(_:)))\n recognizer.buttonMask = 0x1\n self.addGestureRecognizer(recognizer)\n }\n }\n\n required init(rootView: Content) {\n self.highlightState = MenuCardHighlightState()\n self.onClick = nil\n super.init(rootView: rootView)\n }\n\n @available(*, unavailable)\n required init?(coder: NSCoder) {\n fatalError(\"init(coder:) has not been implemented\")\n }\n\n override func acceptsFirstMouse(for event: NSEvent?) -> Bool {\n true\n }\n\n @objc private func handlePrimaryClick(_ recognizer: NSClickGestureRecognizer) {\n guard recognizer.state == .ended else { return }\n self.onClick?()\n }\n\n func measuredHeight(width: CGFloat) -> CGFloat {\n let controller = NSHostingController(rootView: self.rootView)\n let measured = controller.sizeThatFits(in: CGSize(width: width, height: .greatestFiniteMagnitude))\n return measured.height\n }\n\n func setHighlighted(_ highlighted: Bool) {\n guard self.highlightState.isHighlighted != highlighted else { return }\n self.highlightState.isHighlighted = highlighted\n }\n }\n\n private struct MenuCardSectionContainerView: View {\n @Bindable var highlightState: MenuCardHighlightState\n let showsSubmenuIndicator: Bool\n let content: Content\n\n init(\n highlightState: MenuCardHighlightState,\n showsSubmenuIndicator: Bool,\n @ViewBuilder content: () -> Content)\n {\n self.highlightState = highlightState\n self.showsSubmenuIndicator = showsSubmenuIndicator\n self.content = content()\n }\n\n var body: some View {\n self.content\n .environment(\\.menuItemHighlighted, self.highlightState.isHighlighted)\n .foregroundStyle(MenuHighlightStyle.primary(self.highlightState.isHighlighted))\n .background(alignment: .topLeading) {\n if self.highlightState.isHighlighted {\n RoundedRectangle(cornerRadius: 6, style: .continuous)\n .fill(MenuHighlightStyle.selectionBackground(true))\n .padding(.horizontal, 6)\n .padding(.vertical, 2)\n }\n }\n .overlay(alignment: .topTrailing) {\n if self.showsSubmenuIndicator {\n Image(systemName: \"chevron.right\")\n .font(.caption2.weight(.semibold))\n .foregroundStyle(MenuHighlightStyle.secondary(self.highlightState.isHighlighted))\n .padding(.top, 8)\n .padding(.trailing, 10)\n }\n }\n }\n }\n\n private func makeBuyCreditsItem() -> NSMenuItem {\n let item = NSMenuItem(title: \"Buy Credits...\", action: #selector(self.openCreditsPurchase), keyEquivalent: \"\")\n item.target = self\n if let image = NSImage(systemSymbolName: \"plus.circle\", accessibilityDescription: nil) {\n image.isTemplate = true\n image.size = NSSize(width: 16, height: 16)\n item.image = image\n }\n return item\n }\n\n @discardableResult\n private func addCreditsHistorySubmenu(to menu: NSMenu) -> Bool {\n guard let submenu = self.makeCreditsHistorySubmenu() else { return false }\n let item = NSMenuItem(title: \"Credits history\", action: nil, keyEquivalent: \"\")\n item.isEnabled = true\n item.submenu = submenu\n menu.addItem(item)\n return true\n }\n\n @discardableResult\n private func addUsageBreakdownSubmenu(to menu: NSMenu) -> Bool {\n guard let submenu = self.makeUsageBreakdownSubmenu() else { return false }\n let item = NSMenuItem(title: \"Usage breakdown\", action: nil, keyEquivalent: \"\")\n item.isEnabled = true\n item.submenu = submenu\n menu.addItem(item)\n return true\n }\n\n @discardableResult\n private func addCostHistorySubmenu(to menu: NSMenu, provider: UsageProvider) -> Bool {\n guard let submenu = self.makeCostHistorySubmenu(provider: provider) else { return false }\n let item = NSMenuItem(title: \"Usage history (30 days)\", action: nil, keyEquivalent: \"\")\n item.isEnabled = true\n item.submenu = submenu\n menu.addItem(item)\n return true\n }\n\n private func makeUsageSubmenu(\n provider: UsageProvider,\n snapshot: UsageSnapshot?,\n webItems: OpenAIWebMenuItems) -> NSMenu?\n {\n if provider == .codex, webItems.hasUsageBreakdown {\n return self.makeUsageBreakdownSubmenu()\n }\n if provider == .zai {\n return self.makeZaiUsageDetailsSubmenu(snapshot: snapshot)\n }\n return nil\n }\n\n private func makeZaiUsageDetailsSubmenu(snapshot: UsageSnapshot?) -> NSMenu? {\n guard let timeLimit = snapshot?.zaiUsage?.timeLimit else { return nil }\n guard !timeLimit.usageDetails.isEmpty else { return nil }\n\n let submenu = NSMenu()\n submenu.delegate = self\n let titleItem = NSMenuItem(title: \"MCP details\", action: nil, keyEquivalent: \"\")\n titleItem.isEnabled = false\n submenu.addItem(titleItem)\n\n if let window = timeLimit.windowLabel {\n let item = NSMenuItem(title: \"Window: \\(window)\", action: nil, keyEquivalent: \"\")\n item.isEnabled = false\n submenu.addItem(item)\n }\n if let resetTime = timeLimit.nextResetTime {\n let reset = self.settings.resetTimeDisplayStyle == .absolute\n ? UsageFormatter.resetDescription(from: resetTime)\n : UsageFormatter.resetCountdownDescription(from: resetTime)\n let item = NSMenuItem(title: \"Resets: \\(reset)\", action: nil, keyEquivalent: \"\")\n item.isEnabled = false\n submenu.addItem(item)\n }\n submenu.addItem(.separator())\n\n let sortedDetails = timeLimit.usageDetails.sorted {\n $0.modelCode.localizedCaseInsensitiveCompare($1.modelCode) == .orderedAscending\n }\n for detail in sortedDetails {\n let usage = UsageFormatter.tokenCountString(detail.usage)\n let item = NSMenuItem(title: \"\\(detail.modelCode): \\(usage)\", action: nil, keyEquivalent: \"\")\n submenu.addItem(item)\n }\n return submenu\n }\n\n private func makeUsageBreakdownSubmenu() -> NSMenu? {\n let breakdown = self.store.openAIDashboard?.usageBreakdown ?? []\n let width = Self.menuCardBaseWidth\n guard !breakdown.isEmpty else { return nil }\n\n if !Self.menuCardRenderingEnabled {\n let submenu = NSMenu()\n submenu.delegate = self\n let chartItem = NSMenuItem()\n chartItem.isEnabled = false\n chartItem.representedObject = \"usageBreakdownChart\"\n submenu.addItem(chartItem)\n return submenu\n }\n\n let submenu = NSMenu()\n submenu.delegate = self\n let chartView = UsageBreakdownChartMenuView(breakdown: breakdown, width: width)\n let hosting = MenuHostingView(rootView: chartView)\n // Use NSHostingController for efficient size calculation without multiple layout passes\n let controller = NSHostingController(rootView: chartView)\n let size = controller.sizeThatFits(in: CGSize(width: width, height: .greatestFiniteMagnitude))\n hosting.frame = NSRect(origin: .zero, size: NSSize(width: width, height: size.height))\n\n let chartItem = NSMenuItem()\n chartItem.view = hosting\n chartItem.isEnabled = false\n chartItem.representedObject = \"usageBreakdownChart\"\n submenu.addItem(chartItem)\n return submenu\n }\n\n private func makeCreditsHistorySubmenu() -> NSMenu? {\n let breakdown = self.store.openAIDashboard?.dailyBreakdown ?? []\n let width = Self.menuCardBaseWidth\n guard !breakdown.isEmpty else { return nil }\n\n if !Self.menuCardRenderingEnabled {\n let submenu = NSMenu()\n submenu.delegate = self\n let chartItem = NSMenuItem()\n chartItem.isEnabled = false\n chartItem.representedObject = \"creditsHistoryChart\"\n submenu.addItem(chartItem)\n return submenu\n }\n\n let submenu = NSMenu()\n submenu.delegate = self\n let chartView = CreditsHistoryChartMenuView(breakdown: breakdown, width: width)\n let hosting = MenuHostingView(rootView: chartView)\n // Use NSHostingController for efficient size calculation without multiple layout passes\n let controller = NSHostingController(rootView: chartView)\n let size = controller.sizeThatFits(in: CGSize(width: width, height: .greatestFiniteMagnitude))\n hosting.frame = NSRect(origin: .zero, size: NSSize(width: width, height: size.height))\n\n let chartItem = NSMenuItem()\n chartItem.view = hosting\n chartItem.isEnabled = false\n chartItem.representedObject = \"creditsHistoryChart\"\n submenu.addItem(chartItem)\n return submenu\n }\n\n private func makeCostHistorySubmenu(provider: UsageProvider) -> NSMenu? {\n guard provider == .codex || provider == .claude || provider == .vertexai else { return nil }\n let width = Self.menuCardBaseWidth\n guard let tokenSnapshot = self.store.tokenSnapshot(for: provider) else { return nil }\n guard !tokenSnapshot.daily.isEmpty else { return nil }\n\n if !Self.menuCardRenderingEnabled {\n let submenu = NSMenu()\n submenu.delegate = self\n let chartItem = NSMenuItem()\n chartItem.isEnabled = false\n chartItem.representedObject = \"costHistoryChart\"\n submenu.addItem(chartItem)\n return submenu\n }\n\n let submenu = NSMenu()\n submenu.delegate = self\n let chartView = CostHistoryChartMenuView(\n provider: provider,\n daily: tokenSnapshot.daily,\n totalCostUSD: tokenSnapshot.last30DaysCostUSD,\n width: width)\n let hosting = MenuHostingView(rootView: chartView)\n // Use NSHostingController for efficient size calculation without multiple layout passes\n let controller = NSHostingController(rootView: chartView)\n let size = controller.sizeThatFits(in: CGSize(width: width, height: .greatestFiniteMagnitude))\n hosting.frame = NSRect(origin: .zero, size: NSSize(width: width, height: size.height))\n\n let chartItem = NSMenuItem()\n chartItem.view = hosting\n chartItem.isEnabled = false\n chartItem.representedObject = \"costHistoryChart\"\n submenu.addItem(chartItem)\n return submenu\n }\n\n private func isHostedSubviewMenu(_ menu: NSMenu) -> Bool {\n let ids: Set = [\n \"usageBreakdownChart\",\n \"creditsHistoryChart\",\n \"costHistoryChart\",\n ]\n return menu.items.contains { item in\n guard let id = item.representedObject as? String else { return false }\n return ids.contains(id)\n }\n }\n\n private func isOpenAIWebSubviewMenu(_ menu: NSMenu) -> Bool {\n let ids: Set = [\n \"usageBreakdownChart\",\n \"creditsHistoryChart\",\n ]\n return menu.items.contains { item in\n guard let id = item.representedObject as? String else { return false }\n return ids.contains(id)\n }\n }\n\n private func refreshHostedSubviewHeights(in menu: NSMenu) {\n let enabledProviders = self.store.enabledProvidersForDisplay()\n let width = self.menuCardWidth(for: enabledProviders, menu: menu)\n\n for item in menu.items {\n guard let view = item.view else { continue }\n view.frame = NSRect(origin: .zero, size: NSSize(width: width, height: 1))\n view.layoutSubtreeIfNeeded()\n let height = view.fittingSize.height\n view.frame = NSRect(origin: .zero, size: NSSize(width: width, height: height))\n }\n }\n\n private func menuCardModel(\n for provider: UsageProvider?,\n snapshotOverride: UsageSnapshot? = nil,\n errorOverride: String? = nil) -> UsageMenuCardView.Model?\n {\n let target = provider ?? self.store.enabledProvidersForDisplay().first ?? .codex\n let metadata = self.store.metadata(for: target)\n\n let snapshot = snapshotOverride ?? self.store.snapshot(for: target)\n let credits: CreditsSnapshot?\n let creditsError: String?\n let dashboard: OpenAIDashboardSnapshot?\n let dashboardError: String?\n let tokenSnapshot: CostUsageTokenSnapshot?\n let tokenError: String?\n if target == .codex, snapshotOverride == nil {\n credits = self.store.credits\n creditsError = self.store.lastCreditsError\n dashboard = self.store.openAIDashboardRequiresLogin ? nil : self.store.openAIDashboard\n dashboardError = self.store.lastOpenAIDashboardError\n tokenSnapshot = self.store.tokenSnapshot(for: target)\n tokenError = self.store.tokenError(for: target)\n } else if target == .claude || target == .vertexai, snapshotOverride == nil {\n credits = nil\n creditsError = nil\n dashboard = nil\n dashboardError = nil\n tokenSnapshot = self.store.tokenSnapshot(for: target)\n tokenError = self.store.tokenError(for: target)\n } else {\n credits = nil\n creditsError = nil\n dashboard = nil\n dashboardError = nil\n tokenSnapshot = nil\n tokenError = nil\n }\n\n let sourceLabel = snapshotOverride == nil ? self.store.sourceLabel(for: target) : nil\n let kiloAutoMode = target == .kilo && self.settings.kiloUsageDataSource == .auto\n let now = Date()\n let weeklyPace = snapshot?.secondary.flatMap { window in\n self.store.weeklyPace(provider: target, window: window, now: now)\n }\n let input = UsageMenuCardView.Model.Input(\n provider: target,\n metadata: metadata,\n snapshot: snapshot,\n credits: credits,\n creditsError: creditsError,\n dashboard: dashboard,\n dashboardError: dashboardError,\n tokenSnapshot: tokenSnapshot,\n tokenError: tokenError,\n account: self.account,\n isRefreshing: self.store.isRefreshing,\n lastError: errorOverride ?? self.store.error(for: target),\n usageBarsShowUsed: self.settings.usageBarsShowUsed,\n resetTimeDisplayStyle: self.settings.resetTimeDisplayStyle,\n tokenCostUsageEnabled: self.settings.isCostUsageEffectivelyEnabled(for: target),\n showOptionalCreditsAndExtraUsage: self.settings.showOptionalCreditsAndExtraUsage,\n sourceLabel: sourceLabel,\n kiloAutoMode: kiloAutoMode,\n hidePersonalInfo: self.settings.hidePersonalInfo,\n weeklyPace: weeklyPace,\n now: now)\n return UsageMenuCardView.Model.make(input)\n }\n\n @objc private func menuCardNoOp(_ sender: NSMenuItem) {\n _ = sender\n }\n\n @objc private func selectOverviewProvider(_ sender: NSMenuItem) {\n guard let represented = sender.representedObject as? String,\n represented.hasPrefix(Self.overviewRowIdentifierPrefix)\n else {\n return\n }\n let rawProvider = String(represented.dropFirst(Self.overviewRowIdentifierPrefix.count))\n guard let provider = UsageProvider(rawValue: rawProvider),\n let menu = sender.menu\n else {\n return\n }\n\n self.selectOverviewProvider(provider, menu: menu)\n }\n\n private func selectOverviewProvider(_ provider: UsageProvider, menu: NSMenu) {\n if !self.settings.mergedMenuLastSelectedWasOverview, self.selectedMenuProvider == provider { return }\n self.settings.mergedMenuLastSelectedWasOverview = false\n self.lastMergedSwitcherSelection = nil\n self.selectedMenuProvider = provider\n self.lastMenuProvider = provider\n self.populateMenu(menu, provider: provider)\n self.markMenuFresh(menu)\n self.applyIcon(phase: nil)\n }\n\n private func applySubtitle(_ subtitle: String, to item: NSMenuItem, title: String) {\n if #available(macOS 14.4, *) {\n // NSMenuItem.subtitle is only available on macOS 14.4+.\n item.subtitle = subtitle\n } else {\n item.view = self.makeMenuSubtitleView(title: title, subtitle: subtitle, isEnabled: item.isEnabled)\n item.toolTip = \"\\(title) — \\(subtitle)\"\n }\n }\n\n private func makeMenuSubtitleView(title: String, subtitle: String, isEnabled: Bool) -> NSView {\n let container = NSView()\n container.translatesAutoresizingMaskIntoConstraints = false\n container.alphaValue = isEnabled ? 1.0 : 0.7\n\n let titleField = NSTextField(labelWithString: title)\n titleField.font = NSFont.menuFont(ofSize: NSFont.systemFontSize)\n titleField.textColor = NSColor.labelColor\n titleField.lineBreakMode = .byTruncatingTail\n titleField.maximumNumberOfLines = 1\n titleField.setContentCompressionResistancePriority(.defaultLow, for: .horizontal)\n\n let subtitleField = NSTextField(labelWithString: subtitle)\n subtitleField.font = NSFont.menuFont(ofSize: NSFont.smallSystemFontSize)\n subtitleField.textColor = NSColor.secondaryLabelColor\n subtitleField.lineBreakMode = .byTruncatingTail\n subtitleField.maximumNumberOfLines = 1\n subtitleField.setContentCompressionResistancePriority(.defaultLow, for: .horizontal)\n\n let stack = NSStackView(views: [titleField, subtitleField])\n stack.orientation = .vertical\n stack.alignment = .leading\n stack.spacing = 1\n stack.translatesAutoresizingMaskIntoConstraints = false\n container.addSubview(stack)\n\n NSLayoutConstraint.activate([\n stack.leadingAnchor.constraint(equalTo: container.leadingAnchor, constant: 18),\n stack.trailingAnchor.constraint(equalTo: container.trailingAnchor, constant: -10),\n stack.topAnchor.constraint(equalTo: container.topAnchor, constant: 2),\n stack.bottomAnchor.constraint(equalTo: container.bottomAnchor, constant: -2),\n ])\n\n return container\n }\n}\n" }, { "path": "Sources/CodexBar/StatusItemController+SwitcherViews.swift", "content": "import AppKit\nimport CodexBarCore\n\nenum ProviderSwitcherSelection: Equatable {\n case overview\n case provider(UsageProvider)\n}\n\nfinal class ProviderSwitcherView: NSView {\n private struct Segment {\n let selection: ProviderSwitcherSelection\n let image: NSImage\n let title: String\n }\n\n private struct WeeklyIndicator {\n let track: NSView\n let fill: NSView\n }\n\n private let segments: [Segment]\n private let onSelect: (ProviderSwitcherSelection) -> Void\n private let showsIcons: Bool\n private let weeklyRemainingProvider: (UsageProvider) -> Double?\n private var buttons: [NSButton] = []\n private var weeklyIndicators: [ObjectIdentifier: WeeklyIndicator] = [:]\n private var hoverTrackingArea: NSTrackingArea?\n private var segmentWidths: [CGFloat] = []\n private let selectedBackground = NSColor.controlAccentColor.cgColor\n private let unselectedBackground = NSColor.clear.cgColor\n private let selectedTextColor = NSColor.white\n private let unselectedTextColor = NSColor.secondaryLabelColor\n private let stackedIcons: Bool\n private let rowCount: Int\n private let rowSpacing: CGFloat\n private let rowHeight: CGFloat\n private var preferredWidth: CGFloat = 0\n private var hoveredButtonTag: Int?\n private let lightModeOverlayLayer = CALayer()\n\n init(\n providers: [UsageProvider],\n selected: ProviderSwitcherSelection?,\n includesOverview: Bool,\n width: CGFloat,\n showsIcons: Bool,\n iconProvider: (UsageProvider) -> NSImage,\n weeklyRemainingProvider: @escaping (UsageProvider) -> Double?,\n onSelect: @escaping (ProviderSwitcherSelection) -> Void)\n {\n let minimumGap: CGFloat = 1\n var segments = providers.map { provider in\n let fullTitle = Self.switcherTitle(for: provider)\n let icon = iconProvider(provider)\n icon.isTemplate = true\n // Avoid any resampling: we ship exact 16pt/32px assets for crisp rendering.\n icon.size = NSSize(width: 16, height: 16)\n return Segment(\n selection: .provider(provider),\n image: icon,\n title: fullTitle)\n }\n if includesOverview {\n let overviewIcon = Self.overviewIcon()\n overviewIcon.isTemplate = true\n overviewIcon.size = NSSize(width: 16, height: 16)\n segments.insert(\n Segment(\n selection: .overview,\n image: overviewIcon,\n title: \"Overview\"),\n at: 0)\n }\n self.segments = segments\n self.onSelect = onSelect\n self.showsIcons = showsIcons\n self.weeklyRemainingProvider = weeklyRemainingProvider\n self.stackedIcons = showsIcons && self.segments.count > 3\n let initialOuterPadding = Self.switcherOuterPadding(\n for: width,\n count: self.segments.count,\n minimumGap: minimumGap)\n let initialMaxAllowedSegmentWidth = Self.maxAllowedUniformSegmentWidth(\n for: width,\n count: self.segments.count,\n outerPadding: initialOuterPadding,\n minimumGap: minimumGap)\n self.rowCount = Self.switcherRowCount(\n width: width,\n count: self.segments.count,\n maxAllowedSegmentWidth: initialMaxAllowedSegmentWidth,\n stackedIcons: self.stackedIcons)\n self.rowSpacing = self.stackedIcons ? 4 : 2\n if self.stackedIcons && self.rowCount >= 3 {\n self.rowHeight = 40\n } else {\n self.rowHeight = self.stackedIcons ? 36 : 30\n }\n let height: CGFloat = self.rowHeight * CGFloat(self.rowCount)\n + self.rowSpacing * CGFloat(max(0, self.rowCount - 1))\n self.preferredWidth = width\n super.init(frame: NSRect(x: 0, y: 0, width: width, height: height))\n Self.clearButtonWidthCache()\n self.wantsLayer = true\n self.layer?.masksToBounds = false\n self.lightModeOverlayLayer.masksToBounds = false\n self.layer?.insertSublayer(self.lightModeOverlayLayer, at: 0)\n self.updateLightModeStyling()\n\n let layoutCount = Self.layoutCount(for: self.segments.count, rows: self.rowCount)\n let outerPadding: CGFloat = Self.switcherOuterPadding(\n for: width,\n count: layoutCount,\n minimumGap: minimumGap)\n let maxAllowedSegmentWidth = Self.maxAllowedUniformSegmentWidth(\n for: width,\n count: layoutCount,\n outerPadding: outerPadding,\n minimumGap: minimumGap)\n\n func makeButton(index: Int, segment: Segment) -> NSButton {\n let button: NSButton\n if self.stackedIcons {\n let stacked = StackedToggleButton(\n title: segment.title,\n image: segment.image,\n target: self,\n action: #selector(self.handleSelection(_:)))\n stacked.setAllowsTwoLineTitle(self.rowCount >= 3)\n if self.rowCount >= 4 {\n stacked.setTitleFontSize(NSFont.smallSystemFontSize - 3)\n }\n button = stacked\n } else if self.showsIcons {\n let inline = InlineIconToggleButton(\n title: segment.title,\n image: segment.image,\n target: self,\n action: #selector(self.handleSelection(_:)))\n button = inline\n } else {\n button = PaddedToggleButton(\n title: segment.title,\n target: self,\n action: #selector(self.handleSelection(_:)))\n }\n button.tag = index\n if self.showsIcons {\n if self.stackedIcons {\n // StackedToggleButton manages its own image view.\n } else {\n // InlineIconToggleButton manages its own image view.\n }\n } else {\n button.image = nil\n button.imagePosition = .noImage\n }\n\n let remaining: Double? = switch segment.selection {\n case let .provider(provider):\n self.weeklyRemainingProvider(provider)\n case .overview:\n nil\n }\n self.addWeeklyIndicator(to: button, selection: segment.selection, remainingPercent: remaining)\n button.bezelStyle = .regularSquare\n button.isBordered = false\n button.controlSize = .small\n button.font = NSFont.systemFont(ofSize: NSFont.smallSystemFontSize)\n button.setButtonType(.toggle)\n button.contentTintColor = self.unselectedTextColor\n button.alignment = .center\n button.wantsLayer = true\n button.layer?.cornerRadius = 6\n button.state = (selected == segment.selection) ? .on : .off\n button.toolTip = nil\n button.translatesAutoresizingMaskIntoConstraints = false\n self.buttons.append(button)\n return button\n }\n\n for (index, segment) in self.segments.enumerated() {\n let button = makeButton(index: index, segment: segment)\n self.addSubview(button)\n }\n\n let uniformWidth: CGFloat\n if self.rowCount > 1 || !self.stackedIcons {\n uniformWidth = self.applyUniformSegmentWidth(maxAllowedWidth: maxAllowedSegmentWidth)\n if uniformWidth > 0 {\n self.segmentWidths = Array(repeating: uniformWidth, count: self.buttons.count)\n }\n } else {\n self.segmentWidths = self.applyNonUniformSegmentWidths(\n totalWidth: width,\n outerPadding: outerPadding,\n minimumGap: minimumGap)\n uniformWidth = 0\n }\n\n self.applyLayout(\n outerPadding: outerPadding,\n minimumGap: minimumGap,\n uniformWidth: uniformWidth)\n if width > 0 {\n self.preferredWidth = width\n self.frame.size.width = width\n }\n\n self.updateButtonStyles()\n }\n\n override func layout() {\n super.layout()\n self.lightModeOverlayLayer.frame = self.bounds\n }\n\n override func viewDidChangeEffectiveAppearance() {\n super.viewDidChangeEffectiveAppearance()\n self.updateLightModeStyling()\n self.updateButtonStyles()\n }\n\n override func viewDidMoveToWindow() {\n super.viewDidMoveToWindow()\n self.window?.acceptsMouseMovedEvents = true\n }\n\n override func updateTrackingAreas() {\n super.updateTrackingAreas()\n\n if let hoverTrackingArea {\n self.removeTrackingArea(hoverTrackingArea)\n }\n\n let trackingArea = NSTrackingArea(\n rect: .zero,\n options: [\n .activeAlways,\n .inVisibleRect,\n .mouseEnteredAndExited,\n .mouseMoved,\n ],\n owner: self,\n userInfo: nil)\n self.addTrackingArea(trackingArea)\n self.hoverTrackingArea = trackingArea\n }\n\n override func mouseMoved(with event: NSEvent) {\n let location = self.convert(event.locationInWindow, from: nil)\n let hoveredTag = self.buttons.first(where: { $0.frame.contains(location) })?.tag\n guard hoveredTag != self.hoveredButtonTag else { return }\n self.hoveredButtonTag = hoveredTag\n self.updateButtonStyles()\n }\n\n override func mouseExited(with event: NSEvent) {\n guard self.hoveredButtonTag != nil else { return }\n self.hoveredButtonTag = nil\n self.updateButtonStyles()\n }\n\n private func applyLayout(\n outerPadding: CGFloat,\n minimumGap: CGFloat,\n uniformWidth: CGFloat)\n {\n if self.rowCount > 1 {\n self.applyMultiRowLayout(\n rowCount: self.rowCount,\n outerPadding: outerPadding,\n minimumGap: minimumGap,\n uniformWidth: uniformWidth)\n return\n }\n\n if self.buttons.count == 2 {\n let left = self.buttons[0]\n let right = self.buttons[1]\n let gap = right.leadingAnchor.constraint(greaterThanOrEqualTo: left.trailingAnchor, constant: minimumGap)\n gap.priority = .defaultHigh\n NSLayoutConstraint.activate([\n left.leadingAnchor.constraint(equalTo: self.leadingAnchor, constant: outerPadding),\n left.centerYAnchor.constraint(equalTo: self.centerYAnchor),\n right.trailingAnchor.constraint(equalTo: self.trailingAnchor, constant: -outerPadding),\n right.centerYAnchor.constraint(equalTo: self.centerYAnchor),\n gap,\n ])\n return\n }\n\n if self.buttons.count == 3 {\n let left = self.buttons[0]\n let mid = self.buttons[1]\n let right = self.buttons[2]\n\n let leftGap = mid.leadingAnchor.constraint(greaterThanOrEqualTo: left.trailingAnchor, constant: minimumGap)\n leftGap.priority = .defaultHigh\n let rightGap = right.leadingAnchor.constraint(\n greaterThanOrEqualTo: mid.trailingAnchor,\n constant: minimumGap)\n rightGap.priority = .defaultHigh\n\n NSLayoutConstraint.activate([\n left.leadingAnchor.constraint(equalTo: self.leadingAnchor, constant: outerPadding),\n left.centerYAnchor.constraint(equalTo: self.centerYAnchor),\n mid.centerXAnchor.constraint(equalTo: self.centerXAnchor),\n mid.centerYAnchor.constraint(equalTo: self.centerYAnchor),\n right.trailingAnchor.constraint(equalTo: self.trailingAnchor, constant: -outerPadding),\n right.centerYAnchor.constraint(equalTo: self.centerYAnchor),\n leftGap,\n rightGap,\n ])\n return\n }\n\n if self.buttons.count >= 4 {\n let widths = self.segmentWidths.isEmpty\n ? self.buttons.map { ceil($0.fittingSize.width) }\n : self.segmentWidths\n let layoutWidth = self.preferredWidth > 0 ? self.preferredWidth : self.bounds.width\n let availableWidth = max(0, layoutWidth - outerPadding * 2)\n let gaps = max(1, widths.count - 1)\n let computedGap = gaps > 0\n ? max(minimumGap, (availableWidth - widths.reduce(0, +)) / CGFloat(gaps))\n : 0\n let rowContainer = NSView()\n rowContainer.translatesAutoresizingMaskIntoConstraints = false\n self.addSubview(rowContainer)\n\n NSLayoutConstraint.activate([\n rowContainer.topAnchor.constraint(equalTo: self.topAnchor),\n rowContainer.bottomAnchor.constraint(equalTo: self.bottomAnchor),\n rowContainer.leadingAnchor.constraint(equalTo: self.leadingAnchor, constant: outerPadding),\n rowContainer.trailingAnchor.constraint(equalTo: self.trailingAnchor, constant: -outerPadding),\n ])\n\n var xOffset: CGFloat = 0\n for (index, button) in self.buttons.enumerated() {\n let width = index < widths.count ? widths[index] : 0\n if self.stackedIcons {\n NSLayoutConstraint.activate([\n button.leadingAnchor.constraint(equalTo: rowContainer.leadingAnchor, constant: xOffset),\n button.topAnchor.constraint(equalTo: rowContainer.topAnchor),\n ])\n } else {\n NSLayoutConstraint.activate([\n button.leadingAnchor.constraint(equalTo: rowContainer.leadingAnchor, constant: xOffset),\n button.centerYAnchor.constraint(equalTo: rowContainer.centerYAnchor),\n ])\n }\n xOffset += width + computedGap\n }\n return\n }\n\n if let first = self.buttons.first {\n NSLayoutConstraint.activate([\n first.centerXAnchor.constraint(equalTo: self.centerXAnchor),\n first.centerYAnchor.constraint(equalTo: self.centerYAnchor),\n ])\n }\n }\n\n private func applyMultiRowLayout(\n rowCount: Int,\n outerPadding: CGFloat,\n minimumGap: CGFloat,\n uniformWidth: CGFloat)\n {\n let rows = Self.splitRows(for: self.buttons, rowCount: rowCount)\n let columns = rows.map(\\.count).max() ?? 0\n let layoutWidth = self.preferredWidth > 0 ? self.preferredWidth : self.bounds.width\n let availableWidth = max(0, layoutWidth - outerPadding * 2)\n let gaps = max(1, columns - 1)\n let totalWidth = uniformWidth * CGFloat(columns)\n let computedGap = gaps > 0\n ? max(minimumGap, (availableWidth - totalWidth) / CGFloat(gaps))\n : 0\n let gridContainer = NSView()\n gridContainer.translatesAutoresizingMaskIntoConstraints = false\n self.addSubview(gridContainer)\n\n NSLayoutConstraint.activate([\n gridContainer.topAnchor.constraint(equalTo: self.topAnchor),\n gridContainer.bottomAnchor.constraint(equalTo: self.bottomAnchor),\n gridContainer.leadingAnchor.constraint(equalTo: self.leadingAnchor, constant: outerPadding),\n gridContainer.trailingAnchor.constraint(equalTo: self.trailingAnchor, constant: -outerPadding),\n ])\n\n var rowViews: [NSView] = []\n for _ in 0.. Int\n {\n guard count > 1 else { return 1 }\n let maxRows = min(4, count)\n let fourRowThreshold = 15\n let minimumComfortableAverage: CGFloat = stackedIcons ? 50 : 54\n if count >= fourRowThreshold { return maxRows }\n if maxAllowedSegmentWidth >= minimumComfortableAverage { return 1 }\n\n for rows in 2...maxRows {\n let perRow = self.layoutCount(for: count, rows: rows)\n let outerPadding = self.switcherOuterPadding(for: width, count: perRow, minimumGap: 1)\n let allowedWidth = self.maxAllowedUniformSegmentWidth(\n for: width,\n count: perRow,\n outerPadding: outerPadding,\n minimumGap: 1)\n if allowedWidth >= minimumComfortableAverage { return rows }\n }\n\n return maxRows\n }\n\n private static func layoutCount(for count: Int, rows: Int) -> Int {\n guard rows > 0 else { return count }\n return Int(ceil(Double(count) / Double(rows)))\n }\n\n private static func splitRows(for buttons: [NSButton], rowCount: Int) -> [[NSButton]] {\n guard rowCount > 1 else { return [buttons] }\n let base = buttons.count / rowCount\n let extra = buttons.count % rowCount\n var rows: [[NSButton]] = []\n var start = 0\n for index in 0.. CGFloat {\n // Align with the card's left/right content grid when possible.\n let preferred: CGFloat = 16\n let reduced: CGFloat = 10\n let minimal: CGFloat = 6\n\n func averageButtonWidth(outerPadding: CGFloat) -> CGFloat {\n let available = width - outerPadding * 2 - minimumGap * CGFloat(max(0, count - 1))\n guard count > 0 else { return 0 }\n return available / CGFloat(count)\n }\n\n // Only sacrifice padding when we'd otherwise squeeze buttons into unreadable widths.\n let minimumComfortableAverage: CGFloat = count >= 5 ? 50 : 54\n\n if averageButtonWidth(outerPadding: preferred) >= minimumComfortableAverage { return preferred }\n if averageButtonWidth(outerPadding: reduced) >= minimumComfortableAverage { return reduced }\n return minimal\n }\n\n @available(*, unavailable)\n required init?(coder: NSCoder) {\n nil\n }\n\n override var intrinsicContentSize: NSSize {\n NSSize(width: self.preferredWidth, height: self.frame.size.height)\n }\n\n @objc private func handleSelection(_ sender: NSButton) {\n let index = sender.tag\n guard self.segments.indices.contains(index) else { return }\n for (idx, button) in self.buttons.enumerated() {\n button.state = (idx == index) ? .on : .off\n }\n self.updateButtonStyles()\n self.onSelect(self.segments[index].selection)\n }\n\n private func updateButtonStyles() {\n for button in self.buttons {\n let isSelected = button.state == .on\n let isHovered = self.hoveredButtonTag == button.tag\n button.contentTintColor = isSelected ? self.selectedTextColor : self.unselectedTextColor\n button.layer?.backgroundColor = if isSelected {\n self.selectedBackground\n } else if isHovered {\n self.hoverPlateColor()\n } else {\n self.unselectedBackground\n }\n self.updateWeeklyIndicatorVisibility(for: button)\n (button as? StackedToggleButton)?.setContentTintColor(button.contentTintColor)\n (button as? InlineIconToggleButton)?.setContentTintColor(button.contentTintColor)\n }\n }\n\n private func isLightMode() -> Bool {\n self.effectiveAppearance.bestMatch(from: [.aqua, .darkAqua]) == .aqua\n }\n\n private func updateLightModeStyling() {\n guard self.isLightMode() else {\n self.lightModeOverlayLayer.backgroundColor = nil\n return\n }\n // The menu card background is very bright in light mode; add a subtle neutral wash to ground the switcher.\n self.lightModeOverlayLayer.backgroundColor = NSColor.black.withAlphaComponent(0.035).cgColor\n }\n\n private func hoverPlateColor() -> CGColor {\n if self.isLightMode() {\n return NSColor.black.withAlphaComponent(0.095).cgColor\n }\n return NSColor.labelColor.withAlphaComponent(0.06).cgColor\n }\n\n /// Cache for button width measurements to avoid repeated layout passes.\n private static var buttonWidthCache: [ObjectIdentifier: CGFloat] = [:]\n\n private static func maxToggleWidth(for button: NSButton) -> CGFloat {\n let buttonId = ObjectIdentifier(button)\n\n // Return cached value if available.\n if let cached = buttonWidthCache[buttonId] {\n return cached\n }\n\n let originalState = button.state\n defer { button.state = originalState }\n\n button.state = .off\n button.layoutSubtreeIfNeeded()\n let offWidth = button.fittingSize.width\n\n button.state = .on\n button.layoutSubtreeIfNeeded()\n let onWidth = button.fittingSize.width\n\n let maxWidth = max(offWidth, onWidth)\n self.buttonWidthCache[buttonId] = maxWidth\n return maxWidth\n }\n\n private static func clearButtonWidthCache() {\n self.buttonWidthCache.removeAll()\n }\n\n private func applyUniformSegmentWidth(maxAllowedWidth: CGFloat) -> CGFloat {\n guard !self.buttons.isEmpty else { return 0 }\n\n var desiredWidths: [CGFloat] = []\n desiredWidths.reserveCapacity(self.buttons.count)\n\n for (index, button) in self.buttons.enumerated() {\n if self.stackedIcons,\n self.segments.indices.contains(index)\n {\n let font = NSFont.systemFont(ofSize: NSFont.smallSystemFontSize)\n let titleWidth = ceil(\n (self.segments[index].title as NSString).size(withAttributes: [.font: font])\n .width)\n let contentPadding: CGFloat = 4 + 4\n let extraSlack: CGFloat = 1\n desiredWidths.append(ceil(titleWidth + contentPadding + extraSlack))\n } else {\n desiredWidths.append(ceil(Self.maxToggleWidth(for: button)))\n }\n }\n\n let maxDesired = desiredWidths.max() ?? 0\n let evenMaxDesired = maxDesired.truncatingRemainder(dividingBy: 2) == 0 ? maxDesired : maxDesired + 1\n let evenMaxAllowed = maxAllowedWidth > 0\n ? (maxAllowedWidth.truncatingRemainder(dividingBy: 2) == 0 ? maxAllowedWidth : maxAllowedWidth - 1)\n : 0\n let finalWidth: CGFloat = if evenMaxAllowed > 0 {\n min(evenMaxDesired, evenMaxAllowed)\n } else {\n evenMaxDesired\n }\n\n if finalWidth > 0 {\n for button in self.buttons {\n button.widthAnchor.constraint(equalToConstant: finalWidth).isActive = true\n }\n }\n\n return finalWidth\n }\n\n @discardableResult\n private func applyNonUniformSegmentWidths(\n totalWidth: CGFloat,\n outerPadding: CGFloat,\n minimumGap: CGFloat) -> [CGFloat]\n {\n guard !self.buttons.isEmpty else { return [] }\n\n let count = self.buttons.count\n let available = totalWidth -\n outerPadding * 2 -\n minimumGap * CGFloat(max(0, count - 1))\n guard available > 0 else { return [] }\n\n func evenFloor(_ value: CGFloat) -> CGFloat {\n var v = floor(value)\n if Int(v) % 2 != 0 { v -= 1 }\n return v\n }\n\n let desired = self.buttons.map { ceil(Self.maxToggleWidth(for: $0)) }\n let desiredSum = desired.reduce(0, +)\n let avg = floor(available / CGFloat(count))\n let minWidth = max(24, min(40, avg))\n\n var widths: [CGFloat]\n if desiredSum <= available {\n widths = desired\n } else {\n let totalCapacity = max(0, desiredSum - minWidth * CGFloat(count))\n if totalCapacity <= 0 {\n widths = Array(repeating: available / CGFloat(count), count: count)\n } else {\n let overflow = desiredSum - available\n widths = desired.map { desiredWidth in\n let capacity = max(0, desiredWidth - minWidth)\n let shrink = overflow * (capacity / totalCapacity)\n return desiredWidth - shrink\n }\n }\n }\n\n widths = widths.map { max(minWidth, evenFloor($0)) }\n var used = widths.reduce(0, +)\n\n while available - used >= 2 {\n if let best = widths.indices\n .filter({ desired[$0] - widths[$0] >= 2 })\n .max(by: { lhs, rhs in\n (desired[lhs] - widths[lhs]) < (desired[rhs] - widths[rhs])\n })\n {\n widths[best] += 2\n used += 2\n continue\n }\n\n guard let best = widths.indices.min(by: { lhs, rhs in widths[lhs] < widths[rhs] }) else { break }\n widths[best] += 2\n used += 2\n }\n\n for (index, button) in self.buttons.enumerated() where index < widths.count {\n button.widthAnchor.constraint(equalToConstant: widths[index]).isActive = true\n }\n\n return widths\n }\n\n private static func maxAllowedUniformSegmentWidth(\n for totalWidth: CGFloat,\n count: Int,\n outerPadding: CGFloat,\n minimumGap: CGFloat) -> CGFloat\n {\n guard count > 0 else { return 0 }\n let available = totalWidth -\n outerPadding * 2 -\n minimumGap * CGFloat(max(0, count - 1))\n guard available > 0 else { return 0 }\n return floor(available / CGFloat(count))\n }\n\n private static func paddedImage(_ image: NSImage, leading: CGFloat) -> NSImage {\n let size = NSSize(width: image.size.width + leading, height: image.size.height)\n let newImage = NSImage(size: size)\n newImage.lockFocus()\n let y = (size.height - image.size.height) / 2\n image.draw(\n at: NSPoint(x: leading, y: y),\n from: NSRect(origin: .zero, size: image.size),\n operation: .sourceOver,\n fraction: 1.0)\n newImage.unlockFocus()\n newImage.isTemplate = image.isTemplate\n return newImage\n }\n\n private func addWeeklyIndicator(to view: NSView, selection: ProviderSwitcherSelection, remainingPercent: Double?) {\n guard let remainingPercent else { return }\n\n let track = NSView()\n track.wantsLayer = true\n track.layer?.backgroundColor = NSColor.tertiaryLabelColor.withAlphaComponent(0.22).cgColor\n track.layer?.cornerRadius = 2\n track.layer?.masksToBounds = true\n track.translatesAutoresizingMaskIntoConstraints = false\n view.addSubview(track)\n\n let fill = NSView()\n fill.wantsLayer = true\n fill.layer?.backgroundColor = Self.weeklyIndicatorColor(for: selection).cgColor\n fill.layer?.cornerRadius = 2\n fill.translatesAutoresizingMaskIntoConstraints = false\n track.addSubview(fill)\n\n let ratio = CGFloat(max(0, min(1, remainingPercent / 100)))\n\n NSLayoutConstraint.activate([\n track.leadingAnchor.constraint(equalTo: view.leadingAnchor, constant: 6),\n track.trailingAnchor.constraint(equalTo: view.trailingAnchor, constant: -6),\n track.bottomAnchor.constraint(equalTo: view.bottomAnchor, constant: -1),\n track.heightAnchor.constraint(equalToConstant: 4),\n fill.leadingAnchor.constraint(equalTo: track.leadingAnchor),\n fill.topAnchor.constraint(equalTo: track.topAnchor),\n fill.bottomAnchor.constraint(equalTo: track.bottomAnchor),\n ])\n\n fill.widthAnchor.constraint(equalTo: track.widthAnchor, multiplier: ratio).isActive = true\n\n self.weeklyIndicators[ObjectIdentifier(view)] = WeeklyIndicator(track: track, fill: fill)\n self.updateWeeklyIndicatorVisibility(for: view)\n }\n\n private func updateWeeklyIndicatorVisibility(for view: NSView) {\n guard let indicator = self.weeklyIndicators[ObjectIdentifier(view)] else { return }\n let isSelected = (view as? NSButton)?.state == .on\n indicator.track.isHidden = isSelected\n indicator.fill.isHidden = isSelected\n }\n\n private static func weeklyIndicatorColor(for selection: ProviderSwitcherSelection) -> NSColor {\n switch selection {\n case let .provider(provider):\n let color = ProviderDescriptorRegistry.descriptor(for: provider).branding.color\n return NSColor(deviceRed: color.red, green: color.green, blue: color.blue, alpha: 1)\n case .overview:\n return NSColor.secondaryLabelColor\n }\n }\n\n private static func overviewIcon() -> NSImage {\n if let symbol = NSImage(systemSymbolName: \"square.grid.2x2\", accessibilityDescription: nil) {\n return symbol\n }\n return NSImage(size: NSSize(width: 16, height: 16))\n }\n\n private static func switcherTitle(for provider: UsageProvider) -> String {\n ProviderDescriptorRegistry.descriptor(for: provider).metadata.displayName\n }\n}\n\nfinal class TokenAccountSwitcherView: NSView {\n private let accounts: [ProviderTokenAccount]\n private let onSelect: (Int) -> Void\n private var selectedIndex: Int\n private var buttons: [NSButton] = []\n private let rowSpacing: CGFloat = 4\n private let rowHeight: CGFloat = 26\n private let selectedBackground = NSColor.controlAccentColor.cgColor\n private let unselectedBackground = NSColor.clear.cgColor\n private let selectedTextColor = NSColor.white\n private let unselectedTextColor = NSColor.secondaryLabelColor\n\n init(accounts: [ProviderTokenAccount], selectedIndex: Int, width: CGFloat, onSelect: @escaping (Int) -> Void) {\n self.accounts = accounts\n self.onSelect = onSelect\n self.selectedIndex = min(max(selectedIndex, 0), max(0, accounts.count - 1))\n let useTwoRows = accounts.count > 3\n let rows = useTwoRows ? 2 : 1\n let height = self.rowHeight * CGFloat(rows) + (useTwoRows ? self.rowSpacing : 0)\n super.init(frame: NSRect(x: 0, y: 0, width: width, height: height))\n self.wantsLayer = true\n self.buildButtons(useTwoRows: useTwoRows)\n self.updateButtonStyles()\n }\n\n @available(*, unavailable)\n required init?(coder: NSCoder) {\n nil\n }\n\n private func buildButtons(useTwoRows: Bool) {\n let perRow = useTwoRows ? Int(ceil(Double(self.accounts.count) / 2.0)) : self.accounts.count\n let rows: [[ProviderTokenAccount]] = {\n if !useTwoRows { return [self.accounts] }\n let first = Array(self.accounts.prefix(perRow))\n let second = Array(self.accounts.dropFirst(perRow))\n return [first, second]\n }()\n\n let stack = NSStackView()\n stack.orientation = .vertical\n stack.alignment = .centerX\n stack.spacing = self.rowSpacing\n stack.translatesAutoresizingMaskIntoConstraints = false\n\n var globalIndex = 0\n for rowAccounts in rows {\n let row = NSStackView()\n row.orientation = .horizontal\n row.alignment = .centerY\n row.distribution = .fillEqually\n row.spacing = self.rowSpacing\n row.translatesAutoresizingMaskIntoConstraints = false\n\n for account in rowAccounts {\n let button = PaddedToggleButton(\n title: account.displayName,\n target: self,\n action: #selector(self.handleSelect))\n button.tag = globalIndex\n button.toolTip = account.displayName\n button.isBordered = false\n button.setButtonType(.toggle)\n button.controlSize = .small\n button.font = NSFont.systemFont(ofSize: NSFont.smallSystemFontSize)\n button.wantsLayer = true\n button.layer?.cornerRadius = 6\n row.addArrangedSubview(button)\n self.buttons.append(button)\n globalIndex += 1\n }\n\n stack.addArrangedSubview(row)\n }\n\n self.addSubview(stack)\n NSLayoutConstraint.activate([\n stack.leadingAnchor.constraint(equalTo: self.leadingAnchor, constant: 6),\n stack.trailingAnchor.constraint(equalTo: self.trailingAnchor, constant: -6),\n stack.topAnchor.constraint(equalTo: self.topAnchor),\n stack.bottomAnchor.constraint(equalTo: self.bottomAnchor),\n stack.heightAnchor.constraint(equalToConstant: self.rowHeight * CGFloat(rows.count) +\n (useTwoRows ? self.rowSpacing : 0)),\n ])\n }\n\n private func updateButtonStyles() {\n for (index, button) in self.buttons.enumerated() {\n let selected = index == self.selectedIndex\n button.state = selected ? .on : .off\n button.layer?.backgroundColor = selected ? self.selectedBackground : self.unselectedBackground\n button.contentTintColor = selected ? self.selectedTextColor : self.unselectedTextColor\n }\n }\n\n @objc private func handleSelect(_ sender: NSButton) {\n let index = sender.tag\n guard index >= 0, index < self.accounts.count else { return }\n self.selectedIndex = index\n self.updateButtonStyles()\n self.onSelect(index)\n }\n}\n" }, { "path": "Sources/CodexBar/StatusItemController.swift", "content": "import AppKit\nimport CodexBarCore\nimport Observation\nimport QuartzCore\nimport SwiftUI\n\n// MARK: - Status item controller (AppKit-hosted icons, SwiftUI popovers)\n\n@MainActor\nprotocol StatusItemControlling: AnyObject {\n func openMenuFromShortcut()\n}\n\n@MainActor\nfinal class StatusItemController: NSObject, NSMenuDelegate, StatusItemControlling {\n // Disable SwiftUI menu cards + menu refresh work in tests to avoid swiftpm-testing-helper crashes.\n static var menuCardRenderingEnabled = !SettingsStore.isRunningTests\n static var menuRefreshEnabled = !SettingsStore.isRunningTests\n typealias Factory = (UsageStore, SettingsStore, AccountInfo, UpdaterProviding, PreferencesSelection)\n -> StatusItemControlling\n static let defaultFactory: Factory = { store, settings, account, updater, selection in\n StatusItemController(\n store: store,\n settings: settings,\n account: account,\n updater: updater,\n preferencesSelection: selection)\n }\n\n static var factory: Factory = StatusItemController.defaultFactory\n\n let store: UsageStore\n let settings: SettingsStore\n let account: AccountInfo\n let updater: UpdaterProviding\n private let statusBar: NSStatusBar\n var statusItem: NSStatusItem\n var statusItems: [UsageProvider: NSStatusItem] = [:]\n var lastMenuProvider: UsageProvider?\n var menuProviders: [ObjectIdentifier: UsageProvider] = [:]\n var menuContentVersion: Int = 0\n var menuVersions: [ObjectIdentifier: Int] = [:]\n var mergedMenu: NSMenu?\n var providerMenus: [UsageProvider: NSMenu] = [:]\n var fallbackMenu: NSMenu?\n var openMenus: [ObjectIdentifier: NSMenu] = [:]\n var menuRefreshTasks: [ObjectIdentifier: Task] = [:]\n var blinkTask: Task?\n var loginTask: Task? {\n didSet { self.refreshMenusForLoginStateChange() }\n }\n\n var creditsPurchaseWindow: OpenAICreditsPurchaseWindowController?\n\n var activeLoginProvider: UsageProvider? {\n didSet {\n if oldValue != self.activeLoginProvider {\n self.refreshMenusForLoginStateChange()\n }\n }\n }\n\n var blinkStates: [UsageProvider: BlinkState] = [:]\n var blinkAmounts: [UsageProvider: CGFloat] = [:]\n var wiggleAmounts: [UsageProvider: CGFloat] = [:]\n var tiltAmounts: [UsageProvider: CGFloat] = [:]\n var blinkForceUntil: Date?\n var loginPhase: LoginPhase = .idle {\n didSet {\n if oldValue != self.loginPhase {\n self.refreshMenusForLoginStateChange()\n }\n }\n }\n\n let preferencesSelection: PreferencesSelection\n var animationDriver: DisplayLinkDriver?\n var animationPhase: Double = 0\n var animationPattern: LoadingPattern = .knightRider\n private var lastConfigRevision: Int\n private var lastProviderOrder: [UsageProvider]\n private var lastMergeIcons: Bool\n private var lastSwitcherShowsIcons: Bool\n private var lastObservedUsageBarsShowUsed: Bool\n /// Tracks which `usageBarsShowUsed` mode the provider switcher was built with.\n /// Used to decide whether we can \"smart update\" menu content without rebuilding the switcher.\n var lastSwitcherUsageBarsShowUsed: Bool\n /// Tracks whether the merged-menu switcher was built with the Overview tab visible.\n /// Used to force switcher rebuilds when Overview availability toggles.\n var lastSwitcherIncludesOverview: Bool = false\n /// Tracks which providers the merged menu's switcher was built with, to detect when it needs full rebuild.\n var lastSwitcherProviders: [UsageProvider] = []\n /// Tracks which switcher tab state was used for the current merged-menu switcher instance.\n var lastMergedSwitcherSelection: ProviderSwitcherSelection?\n let loginLogger = CodexBarLog.logger(LogCategories.login)\n var selectedMenuProvider: UsageProvider? {\n get { self.settings.selectedMenuProvider }\n set { self.settings.selectedMenuProvider = newValue }\n }\n\n struct BlinkState {\n var nextBlink: Date\n var blinkStart: Date?\n var pendingSecondStart: Date?\n var effect: MotionEffect = .blink\n\n static func randomDelay() -> TimeInterval {\n Double.random(in: 3...12)\n }\n }\n\n enum MotionEffect {\n case blink\n case wiggle\n case tilt\n }\n\n enum LoginPhase {\n case idle\n case requesting\n case waitingBrowser\n }\n\n func menuBarMetricWindow(for provider: UsageProvider, snapshot: UsageSnapshot?) -> RateWindow? {\n switch self.settings.menuBarMetricPreference(for: provider) {\n case .primary:\n return snapshot?.primary ?? snapshot?.secondary\n case .secondary:\n return snapshot?.secondary ?? snapshot?.primary\n case .average:\n guard let primary = snapshot?.primary, let secondary = snapshot?.secondary else {\n return snapshot?.primary ?? snapshot?.secondary\n }\n let usedPercent = (primary.usedPercent + secondary.usedPercent) / 2\n return RateWindow(usedPercent: usedPercent, windowMinutes: nil, resetsAt: nil, resetDescription: nil)\n case .automatic:\n if provider == .factory || provider == .kimi {\n return snapshot?.secondary ?? snapshot?.primary\n }\n if provider == .copilot,\n let primary = snapshot?.primary,\n let secondary = snapshot?.secondary\n {\n // Copilot can expose chat + completions quotas; show the more constrained one by default.\n return primary.usedPercent >= secondary.usedPercent ? primary : secondary\n }\n return snapshot?.primary ?? snapshot?.secondary\n }\n }\n\n init(\n store: UsageStore,\n settings: SettingsStore,\n account: AccountInfo,\n updater: UpdaterProviding,\n preferencesSelection: PreferencesSelection,\n statusBar: NSStatusBar = .system)\n {\n if SettingsStore.isRunningTests {\n _ = NSApplication.shared\n }\n self.store = store\n self.settings = settings\n self.account = account\n self.updater = updater\n self.preferencesSelection = preferencesSelection\n self.lastConfigRevision = settings.configRevision\n self.lastProviderOrder = settings.providerOrder\n self.lastMergeIcons = settings.mergeIcons\n self.lastSwitcherShowsIcons = settings.switcherShowsIcons\n self.lastObservedUsageBarsShowUsed = settings.usageBarsShowUsed\n self.lastSwitcherUsageBarsShowUsed = settings.usageBarsShowUsed\n self.statusBar = statusBar\n let item = statusBar.statusItem(withLength: NSStatusItem.variableLength)\n // Ensure the icon is rendered at 1:1 without resampling (crisper edges for template images).\n item.button?.imageScaling = .scaleNone\n self.statusItem = item\n // Status items for individual providers are now created lazily in updateVisibility()\n super.init()\n self.wireBindings()\n self.updateIcons()\n self.updateVisibility()\n NotificationCenter.default.addObserver(\n self,\n selector: #selector(self.handleDebugReplayNotification(_:)),\n name: .codexbarDebugReplayAllAnimations,\n object: nil)\n NotificationCenter.default.addObserver(\n self,\n selector: #selector(self.handleDebugBlinkNotification),\n name: .codexbarDebugBlinkNow,\n object: nil)\n NotificationCenter.default.addObserver(\n self,\n selector: #selector(self.handleProviderConfigDidChange),\n name: .codexbarProviderConfigDidChange,\n object: nil)\n }\n\n private func wireBindings() {\n self.observeStoreChanges()\n self.observeDebugForceAnimation()\n self.observeSettingsChanges()\n self.observeUpdaterChanges()\n }\n\n private func observeStoreChanges() {\n withObservationTracking {\n _ = self.store.menuObservationToken\n } onChange: { [weak self] in\n Task { @MainActor [weak self] in\n guard let self else { return }\n self.observeStoreChanges()\n self.invalidateMenus()\n self.updateIcons()\n self.updateBlinkingState()\n }\n }\n }\n\n private func observeDebugForceAnimation() {\n withObservationTracking {\n _ = self.store.debugForceAnimation\n } onChange: { [weak self] in\n Task { @MainActor [weak self] in\n guard let self else { return }\n self.observeDebugForceAnimation()\n self.updateVisibility()\n self.updateBlinkingState()\n }\n }\n }\n\n private func observeSettingsChanges() {\n withObservationTracking {\n _ = self.settings.menuObservationToken\n } onChange: { [weak self] in\n Task { @MainActor [weak self] in\n guard let self else { return }\n self.observeSettingsChanges()\n self.handleSettingsChange(reason: \"observation\")\n }\n }\n }\n\n func handleProviderConfigChange(reason: String) {\n self.handleSettingsChange(reason: \"config:\\(reason)\")\n }\n\n @objc private func handleProviderConfigDidChange(_ notification: Notification) {\n let reason = notification.userInfo?[\"reason\"] as? String ?? \"unknown\"\n if let source = notification.object as? SettingsStore,\n source !== self.settings\n {\n if let config = notification.userInfo?[\"config\"] as? CodexBarConfig {\n self.settings.applyExternalConfig(config, reason: \"external-\\(reason)\")\n } else {\n self.settings.reloadConfig(reason: \"external-\\(reason)\")\n }\n }\n self.handleProviderConfigChange(reason: \"notification:\\(reason)\")\n }\n\n private func observeUpdaterChanges() {\n withObservationTracking {\n _ = self.updater.updateStatus.isUpdateReady\n } onChange: { [weak self] in\n Task { @MainActor [weak self] in\n guard let self else { return }\n self.observeUpdaterChanges()\n self.invalidateMenus()\n }\n }\n }\n\n private func invalidateMenus() {\n self.menuContentVersion &+= 1\n // Don't refresh menus while they're open - wait until they close and reopen\n // This prevents expensive rebuilds while user is navigating the menu\n guard self.openMenus.isEmpty else { return }\n self.refreshOpenMenusIfNeeded()\n Task { @MainActor in\n // AppKit can ignore menu mutations while tracking; retry on the next run loop.\n await Task.yield()\n guard self.openMenus.isEmpty else { return }\n self.refreshOpenMenusIfNeeded()\n }\n }\n\n private func shouldRefreshOpenMenusForProviderSwitcher() -> Bool {\n var shouldRefresh = false\n let revision = self.settings.configRevision\n if revision != self.lastConfigRevision {\n self.lastConfigRevision = revision\n shouldRefresh = true\n }\n let order = self.settings.providerOrder\n if order != self.lastProviderOrder {\n self.lastProviderOrder = order\n shouldRefresh = true\n }\n let mergeIcons = self.settings.mergeIcons\n if mergeIcons != self.lastMergeIcons {\n self.lastMergeIcons = mergeIcons\n shouldRefresh = true\n }\n let showsIcons = self.settings.switcherShowsIcons\n if showsIcons != self.lastSwitcherShowsIcons {\n self.lastSwitcherShowsIcons = showsIcons\n shouldRefresh = true\n }\n let usageBarsShowUsed = self.settings.usageBarsShowUsed\n if usageBarsShowUsed != self.lastObservedUsageBarsShowUsed {\n self.lastObservedUsageBarsShowUsed = usageBarsShowUsed\n shouldRefresh = true\n }\n return shouldRefresh\n }\n\n private func handleSettingsChange(reason: String) {\n let configChanged = self.settings.configRevision != self.lastConfigRevision\n let orderChanged = self.settings.providerOrder != self.lastProviderOrder\n let shouldRefreshOpenMenus = self.shouldRefreshOpenMenusForProviderSwitcher()\n self.invalidateMenus()\n if orderChanged || configChanged {\n self.rebuildProviderStatusItems()\n }\n self.updateVisibility()\n self.updateIcons()\n if shouldRefreshOpenMenus {\n self.refreshOpenMenusIfNeeded()\n }\n }\n\n private func updateIcons() {\n // Avoid flicker: when an animation driver is active, store updates can call `updateIcons()` and\n // briefly overwrite the animated frame with the static (phase=nil) icon.\n let phase: Double? = self.needsMenuBarIconAnimation() ? self.animationPhase : nil\n if self.shouldMergeIcons {\n self.applyIcon(phase: phase)\n self.attachMenus()\n } else {\n UsageProvider.allCases.forEach { self.applyIcon(for: $0, phase: phase) }\n self.attachMenus(fallback: self.fallbackProvider)\n }\n self.updateAnimationState()\n self.updateBlinkingState()\n }\n\n /// Lazily retrieves or creates a status item for the given provider\n func lazyStatusItem(for provider: UsageProvider) -> NSStatusItem {\n if let existing = self.statusItems[provider] {\n return existing\n }\n let item = self.statusBar.statusItem(withLength: NSStatusItem.variableLength)\n item.button?.imageScaling = .scaleNone\n self.statusItems[provider] = item\n return item\n }\n\n private func updateVisibility() {\n let anyEnabled = !self.store.enabledProvidersForDisplay().isEmpty\n let force = self.store.debugForceAnimation\n let mergeIcons = self.shouldMergeIcons\n if mergeIcons {\n self.statusItem.isVisible = anyEnabled || force\n for item in self.statusItems.values {\n item.isVisible = false\n }\n self.attachMenus()\n } else {\n self.statusItem.isVisible = false\n let fallback = self.fallbackProvider\n for provider in UsageProvider.allCases {\n let isEnabled = self.isEnabled(provider)\n let shouldBeVisible = isEnabled || fallback == provider || force\n if shouldBeVisible {\n let item = self.lazyStatusItem(for: provider)\n item.isVisible = true\n } else if let item = self.statusItems[provider] {\n item.isVisible = false\n }\n }\n self.attachMenus(fallback: fallback)\n }\n self.updateAnimationState()\n self.updateBlinkingState()\n }\n\n var fallbackProvider: UsageProvider? {\n // Intentionally uses availability-filtered list: fallback activates when no provider\n // can actually work, ensuring at least a codex icon is always visible.\n self.store.enabledProviders().isEmpty ? .codex : nil\n }\n\n func isEnabled(_ provider: UsageProvider) -> Bool {\n self.store.isEnabled(provider)\n }\n\n private func refreshMenusForLoginStateChange() {\n self.invalidateMenus()\n if self.shouldMergeIcons {\n self.attachMenus()\n } else {\n self.attachMenus(fallback: self.fallbackProvider)\n }\n }\n\n private func attachMenus() {\n if self.mergedMenu == nil {\n self.mergedMenu = self.makeMenu()\n }\n if self.statusItem.menu !== self.mergedMenu {\n self.statusItem.menu = self.mergedMenu\n }\n }\n\n private func attachMenus(fallback: UsageProvider? = nil) {\n for provider in UsageProvider.allCases {\n // Only access/create the status item if it's actually needed\n let shouldHaveItem = self.isEnabled(provider) || fallback == provider\n\n if shouldHaveItem {\n let item = self.lazyStatusItem(for: provider)\n\n if self.isEnabled(provider) {\n if self.providerMenus[provider] == nil {\n self.providerMenus[provider] = self.makeMenu(for: provider)\n }\n let menu = self.providerMenus[provider]\n if item.menu !== menu {\n item.menu = menu\n }\n } else if fallback == provider {\n if self.fallbackMenu == nil {\n self.fallbackMenu = self.makeMenu(for: nil)\n }\n if item.menu !== self.fallbackMenu {\n item.menu = self.fallbackMenu\n }\n }\n } else if let item = self.statusItems[provider] {\n // Item exists but is no longer needed - clear its menu\n if item.menu != nil {\n item.menu = nil\n }\n }\n }\n }\n\n private func rebuildProviderStatusItems() {\n for item in self.statusItems.values {\n self.statusBar.removeStatusItem(item)\n }\n self.statusItems.removeAll(keepingCapacity: true)\n\n for provider in self.settings.orderedProviders() {\n let item = self.statusBar.statusItem(withLength: NSStatusItem.variableLength)\n item.button?.imageScaling = .scaleNone\n self.statusItems[provider] = item\n }\n }\n\n func isVisible(_ provider: UsageProvider) -> Bool {\n self.store.debugForceAnimation || self.isEnabled(provider) || self.fallbackProvider == provider\n }\n\n var shouldMergeIcons: Bool {\n self.settings.mergeIcons && self.store.enabledProvidersForDisplay().count > 1\n }\n\n func switchAccountSubtitle(for target: UsageProvider) -> String? {\n guard self.loginTask != nil, let provider = self.activeLoginProvider, provider == target else { return nil }\n let base: String\n switch self.loginPhase {\n case .idle: return nil\n case .requesting: base = \"Requesting login…\"\n case .waitingBrowser: base = \"Waiting in browser…\"\n }\n let prefix = ProviderDescriptorRegistry.descriptor(for: provider).metadata.displayName\n return \"\\(prefix): \\(base)\"\n }\n\n deinit {\n self.blinkTask?.cancel()\n self.loginTask?.cancel()\n NotificationCenter.default.removeObserver(self)\n }\n}\n" }, { "path": "Sources/CodexBar/SyntheticTokenStore.swift", "content": "import CodexBarCore\nimport Foundation\nimport Security\n\nprotocol SyntheticTokenStoring: Sendable {\n func loadToken() throws -> String?\n func storeToken(_ token: String?) throws\n}\n\nenum SyntheticTokenStoreError: LocalizedError {\n case keychainStatus(OSStatus)\n case invalidData\n\n var errorDescription: String? {\n switch self {\n case let .keychainStatus(status):\n \"Keychain error: \\(status)\"\n case .invalidData:\n \"Keychain returned invalid data.\"\n }\n }\n}\n\nstruct KeychainSyntheticTokenStore: SyntheticTokenStoring {\n private static let log = CodexBarLog.logger(LogCategories.syntheticTokenStore)\n\n private let service = \"com.steipete.CodexBar\"\n private let account = \"synthetic-api-key\"\n\n func loadToken() throws -> String? {\n guard !KeychainAccessGate.isDisabled else {\n Self.log.debug(\"Keychain access disabled; skipping token load\")\n return nil\n }\n var result: CFTypeRef?\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n kSecMatchLimit as String: kSecMatchLimitOne,\n kSecReturnData as String: true,\n ]\n\n if case .interactionRequired = KeychainAccessPreflight\n .checkGenericPassword(service: self.service, account: self.account)\n {\n KeychainPromptHandler.handler?(KeychainPromptContext(\n kind: .syntheticToken,\n service: self.service,\n account: self.account))\n }\n\n let status = SecItemCopyMatching(query as CFDictionary, &result)\n if status == errSecItemNotFound {\n return nil\n }\n guard status == errSecSuccess else {\n Self.log.error(\"Keychain read failed: \\(status)\")\n throw SyntheticTokenStoreError.keychainStatus(status)\n }\n\n guard let data = result as? Data else {\n throw SyntheticTokenStoreError.invalidData\n }\n let token = String(data: data, encoding: .utf8)?.trimmingCharacters(in: .whitespacesAndNewlines)\n if let token, !token.isEmpty {\n return token\n }\n return nil\n }\n\n func storeToken(_ token: String?) throws {\n guard !KeychainAccessGate.isDisabled else {\n Self.log.debug(\"Keychain access disabled; skipping token store\")\n return\n }\n let cleaned = token?.trimmingCharacters(in: .whitespacesAndNewlines)\n if cleaned == nil || cleaned?.isEmpty == true {\n try self.deleteTokenIfPresent()\n return\n }\n\n let data = cleaned!.data(using: .utf8)!\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n ]\n let attributes: [String: Any] = [\n kSecValueData as String: data,\n kSecAttrAccessible as String: kSecAttrAccessibleAfterFirstUnlockThisDeviceOnly,\n ]\n\n let updateStatus = SecItemUpdate(query as CFDictionary, attributes as CFDictionary)\n if updateStatus == errSecSuccess {\n return\n }\n if updateStatus != errSecItemNotFound {\n Self.log.error(\"Keychain update failed: \\(updateStatus)\")\n throw SyntheticTokenStoreError.keychainStatus(updateStatus)\n }\n\n var addQuery = query\n for (key, value) in attributes {\n addQuery[key] = value\n }\n let addStatus = SecItemAdd(addQuery as CFDictionary, nil)\n guard addStatus == errSecSuccess else {\n Self.log.error(\"Keychain add failed: \\(addStatus)\")\n throw SyntheticTokenStoreError.keychainStatus(addStatus)\n }\n }\n\n private func deleteTokenIfPresent() throws {\n guard !KeychainAccessGate.isDisabled else { return }\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n ]\n let status = SecItemDelete(query as CFDictionary)\n if status == errSecSuccess || status == errSecItemNotFound {\n return\n }\n Self.log.error(\"Keychain delete failed: \\(status)\")\n throw SyntheticTokenStoreError.keychainStatus(status)\n }\n}\n" }, { "path": "Sources/CodexBar/UpdateChannel.swift", "content": "import Foundation\n\nenum UpdateChannel: String, CaseIterable, Codable {\n case stable\n case beta\n\n static let userDefaultsKey = \"updateChannel\"\n static let sparkleBetaChannel = \"beta\"\n\n var displayName: String {\n switch self {\n case .stable:\n \"Stable\"\n case .beta:\n \"Beta\"\n }\n }\n\n var description: String {\n switch self {\n case .stable:\n \"Receive only stable, production-ready releases.\"\n case .beta:\n \"Receive stable releases plus beta previews.\"\n }\n }\n\n var allowedSparkleChannels: Set {\n switch self {\n case .stable:\n [\"\"]\n case .beta:\n [\"\", UpdateChannel.sparkleBetaChannel]\n }\n }\n\n static var current: Self {\n if let rawValue = UserDefaults.standard.string(forKey: userDefaultsKey),\n let channel = Self(rawValue: rawValue)\n {\n return channel\n }\n return defaultChannel\n }\n\n static var defaultChannel: Self {\n defaultChannel(for: Bundle.main.infoDictionary?[\"CFBundleShortVersionString\"] as? String ?? \"1.0\")\n }\n\n static func defaultChannel(for appVersion: String) -> Self {\n if let isPrereleaseValue = Bundle.main.object(forInfoDictionaryKey: \"IS_PRERELEASE_BUILD\"),\n let isPrerelease = isPrereleaseValue as? Bool,\n isPrerelease\n {\n return .beta\n }\n\n let prereleaseKeywords = [\"beta\", \"alpha\", \"rc\", \"pre\", \"dev\"]\n let lowercaseVersion = appVersion.lowercased()\n\n for keyword in prereleaseKeywords where lowercaseVersion.contains(keyword) {\n return .beta\n }\n\n return .stable\n }\n}\n\nextension UpdateChannel: Identifiable {\n var id: String {\n rawValue\n }\n}\n" }, { "path": "Sources/CodexBar/UsageBreakdownChartMenuView.swift", "content": "import Charts\nimport CodexBarCore\nimport SwiftUI\n\n@MainActor\nstruct UsageBreakdownChartMenuView: View {\n private struct Point: Identifiable {\n let id: String\n let date: Date\n let service: String\n let creditsUsed: Double\n\n init(date: Date, service: String, creditsUsed: Double) {\n self.date = date\n self.service = service\n self.creditsUsed = creditsUsed\n self.id = \"\\(service)-\\(Int(date.timeIntervalSince1970))- \\(creditsUsed)\"\n }\n }\n\n private let breakdown: [OpenAIDashboardDailyBreakdown]\n private let width: CGFloat\n @State private var selectedDayKey: String?\n\n init(breakdown: [OpenAIDashboardDailyBreakdown], width: CGFloat) {\n self.breakdown = breakdown\n self.width = width\n }\n\n var body: some View {\n let model = Self.makeModel(from: self.breakdown)\n VStack(alignment: .leading, spacing: 10) {\n if model.points.isEmpty {\n Text(\"No usage breakdown data.\")\n .font(.footnote)\n .foregroundStyle(.secondary)\n } else {\n Chart {\n ForEach(model.points) { point in\n BarMark(\n x: .value(\"Day\", point.date, unit: .day),\n y: .value(\"Credits used\", point.creditsUsed))\n .foregroundStyle(by: .value(\"Service\", point.service))\n }\n if let peak = model.peakPoint {\n let capStart = max(peak.creditsUsed - Self.capHeight(maxValue: model.maxCreditsUsed), 0)\n BarMark(\n x: .value(\"Day\", peak.date, unit: .day),\n yStart: .value(\"Cap start\", capStart),\n yEnd: .value(\"Cap end\", peak.creditsUsed))\n .foregroundStyle(Color(nsColor: .systemYellow))\n }\n }\n .chartForegroundStyleScale(domain: model.services, range: model.serviceColors)\n .chartYAxis(.hidden)\n .chartXAxis {\n AxisMarks(values: model.axisDates) { _ in\n AxisGridLine().foregroundStyle(Color.clear)\n AxisTick().foregroundStyle(Color.clear)\n AxisValueLabel(format: .dateTime.month(.abbreviated).day())\n .font(.caption2)\n .foregroundStyle(Color(nsColor: .tertiaryLabelColor))\n }\n }\n .chartLegend(.hidden)\n .frame(height: 130)\n .chartOverlay { proxy in\n GeometryReader { geo in\n ZStack(alignment: .topLeading) {\n if let rect = self.selectionBandRect(model: model, proxy: proxy, geo: geo) {\n Rectangle()\n .fill(Self.selectionBandColor)\n .frame(width: rect.width, height: rect.height)\n .position(x: rect.midX, y: rect.midY)\n .allowsHitTesting(false)\n }\n MouseLocationReader { location in\n self.updateSelection(location: location, model: model, proxy: proxy, geo: geo)\n }\n .frame(maxWidth: .infinity, maxHeight: .infinity)\n .contentShape(Rectangle())\n }\n }\n }\n\n let detail = self.detailLines(model: model)\n VStack(alignment: .leading, spacing: 0) {\n Text(detail.primary)\n .font(.caption)\n .foregroundStyle(.secondary)\n .lineLimit(1)\n .truncationMode(.tail)\n .frame(height: 16, alignment: .leading)\n Text(detail.secondary ?? \" \")\n .font(.caption)\n .foregroundStyle(.secondary)\n .lineLimit(1)\n .truncationMode(.tail)\n .frame(height: 16, alignment: .leading)\n .opacity(detail.secondary == nil ? 0 : 1)\n }\n\n LazyVGrid(\n columns: [GridItem(.adaptive(minimum: 110), alignment: .leading)],\n alignment: .leading,\n spacing: 6)\n {\n ForEach(model.services, id: \\.self) { service in\n HStack(spacing: 6) {\n Circle()\n .fill(model.color(for: service))\n .frame(width: 7, height: 7)\n Text(service)\n .font(.caption2)\n .foregroundStyle(.secondary)\n .lineLimit(1)\n }\n }\n }\n }\n }\n .padding(.horizontal, 16)\n .padding(.vertical, 10)\n .frame(minWidth: self.width, maxWidth: .infinity, alignment: .leading)\n }\n\n private struct Model {\n let points: [Point]\n let breakdownByDayKey: [String: OpenAIDashboardDailyBreakdown]\n let dayDates: [(dayKey: String, date: Date)]\n let selectableDayDates: [(dayKey: String, date: Date)]\n let peakPoint: (date: Date, creditsUsed: Double)?\n let services: [String]\n let serviceColors: [Color]\n let axisDates: [Date]\n let maxCreditsUsed: Double\n\n func color(for service: String) -> Color {\n guard let idx = self.services.firstIndex(of: service), idx < self.serviceColors.count else {\n return .secondary\n }\n return self.serviceColors[idx]\n }\n }\n\n private static let selectionBandColor = Color(nsColor: .labelColor).opacity(0.1)\n\n private static func makeModel(from breakdown: [OpenAIDashboardDailyBreakdown]) -> Model {\n let sorted = breakdown\n .sorted { lhs, rhs in lhs.day < rhs.day }\n\n var points: [Point] = []\n points.reserveCapacity(sorted.count * 2)\n\n var breakdownByDayKey: [String: OpenAIDashboardDailyBreakdown] = [:]\n breakdownByDayKey.reserveCapacity(sorted.count)\n\n var dayDates: [(dayKey: String, date: Date)] = []\n dayDates.reserveCapacity(sorted.count)\n\n var selectableDayDates: [(dayKey: String, date: Date)] = []\n selectableDayDates.reserveCapacity(sorted.count)\n\n var peak: (date: Date, creditsUsed: Double)?\n var maxCreditsUsed: Double = 0\n\n for day in sorted {\n guard let date = self.dateFromDayKey(day.day) else { continue }\n breakdownByDayKey[day.day] = day\n dayDates.append((dayKey: day.day, date: date))\n if day.totalCreditsUsed > 0 {\n if let cur = peak {\n if day.totalCreditsUsed > cur.creditsUsed { peak = (date, day.totalCreditsUsed) }\n } else {\n peak = (date, day.totalCreditsUsed)\n }\n maxCreditsUsed = max(maxCreditsUsed, day.totalCreditsUsed)\n }\n var addedSelectable = false\n for service in day.services where service.creditsUsed > 0 {\n points.append(Point(date: date, service: service.service, creditsUsed: service.creditsUsed))\n if !addedSelectable {\n selectableDayDates.append((dayKey: day.day, date: date))\n addedSelectable = true\n }\n }\n }\n\n let services = Self.serviceOrder(from: sorted)\n let colors = services.map { Self.colorForService($0) }\n let axisDates = Self.axisDates(fromSortedDays: sorted)\n\n return Model(\n points: points,\n breakdownByDayKey: breakdownByDayKey,\n dayDates: dayDates,\n selectableDayDates: selectableDayDates,\n peakPoint: peak,\n services: services,\n serviceColors: colors,\n axisDates: axisDates,\n maxCreditsUsed: maxCreditsUsed)\n }\n\n private static func capHeight(maxValue: Double) -> Double {\n maxValue * 0.05\n }\n\n private static func serviceOrder(from breakdown: [OpenAIDashboardDailyBreakdown]) -> [String] {\n var totals: [String: Double] = [:]\n for day in breakdown {\n for service in day.services {\n totals[service.service, default: 0] += service.creditsUsed\n }\n }\n\n return totals\n .sorted { lhs, rhs in\n if lhs.value == rhs.value { return lhs.key < rhs.key }\n return lhs.value > rhs.value\n }\n .map(\\.key)\n }\n\n private static func colorForService(_ service: String) -> Color {\n let lower = service.lowercased()\n if lower == \"cli\" {\n return Color(red: 0.26, green: 0.55, blue: 0.96)\n }\n if lower.contains(\"github\"), lower.contains(\"review\") {\n return Color(red: 0.94, green: 0.53, blue: 0.18)\n }\n let palette: [Color] = [\n Color(red: 0.46, green: 0.75, blue: 0.36),\n Color(red: 0.80, green: 0.45, blue: 0.92),\n Color(red: 0.26, green: 0.78, blue: 0.86),\n Color(red: 0.94, green: 0.74, blue: 0.26),\n ]\n let idx = abs(service.hashValue) % palette.count\n return palette[idx]\n }\n\n private static func axisDates(fromSortedDays sortedDays: [OpenAIDashboardDailyBreakdown]) -> [Date] {\n guard let first = sortedDays.first, let last = sortedDays.last else { return [] }\n guard let firstDate = self.dateFromDayKey(first.day),\n let lastDate = self.dateFromDayKey(last.day)\n else {\n return []\n }\n if Calendar.current.isDate(firstDate, inSameDayAs: lastDate) {\n return [firstDate]\n }\n return [firstDate, lastDate]\n }\n\n private static func dateFromDayKey(_ key: String) -> Date? {\n let parts = key.split(separator: \"-\")\n guard parts.count == 3,\n let year = Int(parts[0]),\n let month = Int(parts[1]),\n let day = Int(parts[2])\n else {\n return nil\n }\n\n var comps = DateComponents()\n comps.calendar = Calendar.current\n comps.timeZone = TimeZone.current\n comps.year = year\n comps.month = month\n comps.day = day\n // Noon avoids off-by-one-day shifts if anything ends up interpreted in UTC.\n comps.hour = 12\n return comps.date\n }\n\n private func selectionBandRect(model: Model, proxy: ChartProxy, geo: GeometryProxy) -> CGRect? {\n guard let key = self.selectedDayKey else { return nil }\n guard let plotAnchor = proxy.plotFrame else { return nil }\n let plotFrame = geo[plotAnchor]\n guard let index = model.dayDates.firstIndex(where: { $0.dayKey == key }) else { return nil }\n let date = model.dayDates[index].date\n guard let x = proxy.position(forX: date) else { return nil }\n\n func xForIndex(_ idx: Int) -> CGFloat? {\n guard idx >= 0, idx < model.dayDates.count else { return nil }\n return proxy.position(forX: model.dayDates[idx].date)\n }\n\n let xPrev = xForIndex(index - 1)\n let xNext = xForIndex(index + 1)\n\n if model.dayDates.count <= 1 {\n return CGRect(\n x: plotFrame.origin.x,\n y: plotFrame.origin.y,\n width: plotFrame.width,\n height: plotFrame.height)\n }\n\n let leftInPlot: CGFloat = if let xPrev {\n (xPrev + x) / 2\n } else if let xNext {\n x - (xNext - x) / 2\n } else {\n x - 8\n }\n\n let rightInPlot: CGFloat = if let xNext {\n (xNext + x) / 2\n } else if let xPrev {\n x + (x - xPrev) / 2\n } else {\n x + 8\n }\n\n let left = plotFrame.origin.x + min(leftInPlot, rightInPlot)\n let right = plotFrame.origin.x + max(leftInPlot, rightInPlot)\n return CGRect(x: left, y: plotFrame.origin.y, width: right - left, height: plotFrame.height)\n }\n\n private func updateSelection(\n location: CGPoint?,\n model: Model,\n proxy: ChartProxy,\n geo: GeometryProxy)\n {\n guard let location else {\n if self.selectedDayKey != nil { self.selectedDayKey = nil }\n return\n }\n\n guard let plotAnchor = proxy.plotFrame else { return }\n let plotFrame = geo[plotAnchor]\n guard plotFrame.contains(location) else { return }\n\n let xInPlot = location.x - plotFrame.origin.x\n guard let date: Date = proxy.value(atX: xInPlot) else { return }\n guard let nearest = self.nearestDayKey(to: date, model: model) else { return }\n\n if self.selectedDayKey != nearest {\n self.selectedDayKey = nearest\n }\n }\n\n private func nearestDayKey(to date: Date, model: Model) -> String? {\n guard !model.selectableDayDates.isEmpty else { return nil }\n var best: (key: String, distance: TimeInterval)?\n for entry in model.selectableDayDates {\n let dist = abs(entry.date.timeIntervalSince(date))\n if let cur = best {\n if dist < cur.distance { best = (entry.dayKey, dist) }\n } else {\n best = (entry.dayKey, dist)\n }\n }\n return best?.key\n }\n\n private func detailLines(model: Model) -> (primary: String, secondary: String?) {\n guard let key = self.selectedDayKey,\n let day = model.breakdownByDayKey[key],\n let date = Self.dateFromDayKey(key)\n else {\n return (\"Hover a bar for details\", nil)\n }\n\n let dayLabel = date.formatted(.dateTime.month(.abbreviated).day())\n let total = day.totalCreditsUsed.formatted(.number.precision(.fractionLength(0...2)))\n if day.services.isEmpty {\n return (\"\\(dayLabel): \\(total)\", nil)\n }\n if day.services.count <= 1, let first = day.services.first {\n let used = first.creditsUsed.formatted(.number.precision(.fractionLength(0...2)))\n return (\"\\(dayLabel): \\(used)\", first.service)\n }\n\n let services = day.services\n .sorted { lhs, rhs in\n if lhs.creditsUsed == rhs.creditsUsed { return lhs.service < rhs.service }\n return lhs.creditsUsed > rhs.creditsUsed\n }\n .prefix(3)\n .map { \"\\($0.service) \\($0.creditsUsed.formatted(.number.precision(.fractionLength(0...2))))\" }\n .joined(separator: \" · \")\n\n return (\"\\(dayLabel): \\(total)\", services)\n }\n}\n" }, { "path": "Sources/CodexBar/UsagePaceText.swift", "content": "import CodexBarCore\nimport Foundation\n\nenum UsagePaceText {\n struct WeeklyDetail {\n let leftLabel: String\n let rightLabel: String?\n let expectedUsedPercent: Double\n let stage: UsagePace.Stage\n }\n\n static func weeklySummary(pace: UsagePace, now: Date = .init()) -> String {\n let detail = self.weeklyDetail(pace: pace, now: now)\n if let rightLabel = detail.rightLabel {\n return \"Pace: \\(detail.leftLabel) · \\(rightLabel)\"\n }\n return \"Pace: \\(detail.leftLabel)\"\n }\n\n static func weeklyDetail(pace: UsagePace, now: Date = .init()) -> WeeklyDetail {\n WeeklyDetail(\n leftLabel: self.detailLeftLabel(for: pace),\n rightLabel: self.detailRightLabel(for: pace, now: now),\n expectedUsedPercent: pace.expectedUsedPercent,\n stage: pace.stage)\n }\n\n private static func detailLeftLabel(for pace: UsagePace) -> String {\n let deltaValue = Int(abs(pace.deltaPercent).rounded())\n switch pace.stage {\n case .onTrack:\n return \"On pace\"\n case .slightlyAhead, .ahead, .farAhead:\n return \"\\(deltaValue)% in deficit\"\n case .slightlyBehind, .behind, .farBehind:\n return \"\\(deltaValue)% in reserve\"\n }\n }\n\n private static func detailRightLabel(for pace: UsagePace, now: Date) -> String? {\n let etaLabel: String?\n if pace.willLastToReset {\n etaLabel = \"Lasts until reset\"\n } else if let etaSeconds = pace.etaSeconds {\n let etaText = Self.durationText(seconds: etaSeconds, now: now)\n etaLabel = etaText == \"now\" ? \"Runs out now\" : \"Runs out in \\(etaText)\"\n } else {\n etaLabel = nil\n }\n\n guard let runOutProbability = pace.runOutProbability else { return etaLabel }\n let roundedRisk = self.roundedRiskPercent(runOutProbability)\n let riskLabel = \"≈ \\(roundedRisk)% run-out risk\"\n if let etaLabel {\n return \"\\(etaLabel) · \\(riskLabel)\"\n }\n return riskLabel\n }\n\n private static func durationText(seconds: TimeInterval, now: Date) -> String {\n let date = now.addingTimeInterval(seconds)\n let countdown = UsageFormatter.resetCountdownDescription(from: date, now: now)\n if countdown == \"now\" { return \"now\" }\n if countdown.hasPrefix(\"in \") { return String(countdown.dropFirst(3)) }\n return countdown\n }\n\n private static func roundedRiskPercent(_ probability: Double) -> Int {\n let percent = probability.clamped(to: 0...1) * 100\n let rounded = (percent / 5).rounded() * 5\n return Int(rounded)\n }\n}\n" }, { "path": "Sources/CodexBar/UsageProgressBar.swift", "content": "import SwiftUI\n\n/// Static progress fill with no implicit animations, used inside the menu card.\nstruct UsageProgressBar: View {\n private static let paceStripeCount = 3\n private static func paceStripeWidth(for scale: CGFloat) -> CGFloat {\n 2\n }\n\n private static func paceStripeSpan(for scale: CGFloat) -> CGFloat {\n let stripeCount = max(1, Self.paceStripeCount)\n return Self.paceStripeWidth(for: scale) * CGFloat(stripeCount)\n }\n\n let percent: Double\n let tint: Color\n let accessibilityLabel: String\n let pacePercent: Double?\n let paceOnTop: Bool\n @Environment(\\.menuItemHighlighted) private var isHighlighted\n @Environment(\\.displayScale) private var displayScale\n\n init(\n percent: Double,\n tint: Color,\n accessibilityLabel: String,\n pacePercent: Double? = nil,\n paceOnTop: Bool = true)\n {\n self.percent = percent\n self.tint = tint\n self.accessibilityLabel = accessibilityLabel\n self.pacePercent = pacePercent\n self.paceOnTop = paceOnTop\n }\n\n private var clamped: Double {\n min(100, max(0, self.percent))\n }\n\n var body: some View {\n GeometryReader { proxy in\n let scale = max(self.displayScale, 1)\n let fillWidth = proxy.size.width * self.clamped / 100\n let paceWidth = proxy.size.width * Self.clampedPercent(self.pacePercent) / 100\n let tipWidth = max(25, proxy.size.height * 6.5)\n let stripeInset = 1 / scale\n let tipOffset = paceWidth - tipWidth + (Self.paceStripeSpan(for: scale) / 2) + stripeInset\n let showTip = self.pacePercent != nil && tipWidth > 0.5\n let needsPunchCompositing = showTip\n let bar = ZStack(alignment: .leading) {\n Capsule()\n .fill(MenuHighlightStyle.progressTrack(self.isHighlighted))\n self.actualBar(width: fillWidth)\n if showTip {\n self.paceTip(width: tipWidth)\n .offset(x: tipOffset)\n }\n }\n .clipped()\n if self.isHighlighted {\n bar\n .compositingGroup()\n .drawingGroup()\n } else if needsPunchCompositing {\n bar\n .compositingGroup()\n } else {\n bar\n }\n }\n .frame(height: 6)\n .accessibilityLabel(self.accessibilityLabel)\n .accessibilityValue(\"\\(Int(self.clamped)) percent\")\n }\n\n private func actualBar(width: CGFloat) -> some View {\n Capsule()\n .fill(MenuHighlightStyle.progressTint(self.isHighlighted, fallback: self.tint))\n .frame(width: width)\n .contentShape(Rectangle())\n .allowsHitTesting(false)\n }\n\n private func paceTip(width: CGFloat) -> some View {\n let isDeficit = self.paceOnTop == false\n let useDeficitRed = isDeficit && self.isHighlighted == false\n return GeometryReader { proxy in\n let size = proxy.size\n let rect = CGRect(origin: .zero, size: size)\n let scale = max(self.displayScale, 1)\n let stripes = Self.paceStripePaths(size: size, scale: scale)\n let stripeColor: Color = if self.isHighlighted {\n .white\n } else if useDeficitRed {\n .red\n } else {\n .green\n }\n\n ZStack {\n Canvas { context, _ in\n context.clip(to: Path(rect))\n context.fill(stripes.punched, with: .color(.white.opacity(0.9)))\n }\n .blendMode(.destinationOut)\n\n Canvas { context, _ in\n context.clip(to: Path(rect))\n context.fill(stripes.center, with: .color(stripeColor))\n }\n }\n }\n .frame(width: width)\n .contentShape(Rectangle())\n .allowsHitTesting(false)\n }\n\n private static func paceStripePaths(size: CGSize, scale: CGFloat) -> (punched: Path, center: Path) {\n let rect = CGRect(origin: .zero, size: size)\n let extend = size.height * 2\n let stripeTopY: CGFloat = -extend\n let stripeBottomY: CGFloat = size.height + extend\n let align: (CGFloat) -> CGFloat = { value in\n (value * scale).rounded() / scale\n }\n\n let stripeWidth = Self.paceStripeWidth(for: scale)\n let punchWidth = stripeWidth * 3\n let stripeInset = 1 / scale\n let stripeAnchorX = align(rect.maxX - stripeInset)\n let stripeMinY = align(stripeTopY)\n let stripeMaxY = align(stripeBottomY)\n let anchorTopX = stripeAnchorX\n var punchedStripe = Path()\n var centerStripe = Path()\n let availableWidth = (anchorTopX - punchWidth) - rect.minX\n guard availableWidth >= 0 else { return (punchedStripe, centerStripe) }\n\n let punchRightTopX = align(anchorTopX)\n let punchLeftTopX = punchRightTopX - punchWidth\n let punchRightBottomX = punchRightTopX\n let punchLeftBottomX = punchLeftTopX\n punchedStripe.addPath(Path { path in\n path.move(to: CGPoint(x: punchLeftTopX, y: stripeMinY))\n path.addLine(to: CGPoint(x: punchRightTopX, y: stripeMinY))\n path.addLine(to: CGPoint(x: punchRightBottomX, y: stripeMaxY))\n path.addLine(to: CGPoint(x: punchLeftBottomX, y: stripeMaxY))\n path.closeSubpath()\n })\n\n let centerLeftTopX = align(punchLeftTopX + (punchWidth - stripeWidth) / 2)\n let centerRightTopX = centerLeftTopX + stripeWidth\n let centerRightBottomX = centerRightTopX\n let centerLeftBottomX = centerLeftTopX\n centerStripe.addPath(Path { path in\n path.move(to: CGPoint(x: centerLeftTopX, y: stripeMinY))\n path.addLine(to: CGPoint(x: centerRightTopX, y: stripeMinY))\n path.addLine(to: CGPoint(x: centerRightBottomX, y: stripeMaxY))\n path.addLine(to: CGPoint(x: centerLeftBottomX, y: stripeMaxY))\n path.closeSubpath()\n })\n\n return (punchedStripe, centerStripe)\n }\n\n private static func clampedPercent(_ value: Double?) -> Double {\n guard let value else { return 0 }\n return min(100, max(0, value))\n }\n}\n" }, { "path": "Sources/CodexBar/UsageStore+Accessors.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension UsageStore {\n var codexSnapshot: UsageSnapshot? {\n self.snapshots[.codex]\n }\n\n var claudeSnapshot: UsageSnapshot? {\n self.snapshots[.claude]\n }\n\n var lastCodexError: String? {\n self.errors[.codex]\n }\n\n var lastClaudeError: String? {\n self.errors[.claude]\n }\n\n func error(for provider: UsageProvider) -> String? {\n self.errors[provider]\n }\n\n func status(for provider: UsageProvider) -> ProviderStatus? {\n guard self.statusChecksEnabled else { return nil }\n return self.statuses[provider]\n }\n\n func statusIndicator(for provider: UsageProvider) -> ProviderStatusIndicator {\n self.status(for: provider)?.indicator ?? .none\n }\n\n func accountInfo() -> AccountInfo {\n self.codexFetcher.loadAccountInfo()\n }\n}\n" }, { "path": "Sources/CodexBar/UsageStore+ClaudeDebug.swift", "content": "import CodexBarCore\nimport Foundation\nimport SweetCookieKit\n\n@MainActor\nextension UsageStore {\n func debugClaudeDump() async -> String {\n await ClaudeStatusProbe.latestDumps()\n }\n}\n\nextension UsageStore {\n struct ClaudeDebugLogConfiguration: Sendable {\n let runtime: CodexBarCore.ProviderRuntime\n let sourceMode: ProviderSourceMode\n let environment: [String: String]\n let webExtrasEnabled: Bool\n let usageDataSource: ClaudeUsageDataSource\n let cookieSource: ProviderCookieSource\n let cookieHeader: String\n let keepCLISessionsAlive: Bool\n }\n\n static func debugClaudeLog(\n browserDetection: BrowserDetection,\n configuration: ClaudeDebugLogConfiguration) async -> String\n {\n struct OAuthDebugProbe: Sendable {\n let hasCredentials: Bool\n let ownerRawValue: String\n let sourceRawValue: String\n let isExpired: Bool\n }\n\n return await runWithTimeout(seconds: 15) {\n var lines: [String] = []\n let manualHeader = configuration.cookieSource == .manual\n ? CookieHeaderNormalizer.normalize(configuration.cookieHeader)\n : nil\n let hasKey = if configuration.cookieSource == .off {\n false\n } else if let manualHeader {\n ClaudeWebAPIFetcher.hasSessionKey(cookieHeader: manualHeader)\n } else {\n ClaudeWebAPIFetcher.hasSessionKey(browserDetection: browserDetection) { msg in lines.append(msg) }\n }\n let oauthProbe = await withTaskGroup(of: OAuthDebugProbe.self) { group in\n // Preserve task-local test overrides while keeping the keychain read off the calling task.\n group.addTask(priority: .utility) {\n let oauthRecord = try? ClaudeOAuthCredentialsStore.loadRecord(\n environment: configuration.environment,\n allowKeychainPrompt: false,\n respectKeychainPromptCooldown: true,\n allowClaudeKeychainRepairWithoutPrompt: false)\n return OAuthDebugProbe(\n hasCredentials: oauthRecord?.credentials.scopes.contains(\"user:profile\") == true,\n ownerRawValue: oauthRecord?.owner.rawValue ?? \"none\",\n sourceRawValue: oauthRecord?.source.rawValue ?? \"none\",\n isExpired: oauthRecord?.credentials.isExpired ?? false)\n }\n return await group.next() ?? OAuthDebugProbe(\n hasCredentials: false,\n ownerRawValue: \"none\",\n sourceRawValue: \"none\",\n isExpired: false)\n }\n let hasOAuthCredentials = ClaudeOAuthPlanningAvailability.isAvailable(\n runtime: configuration.runtime,\n sourceMode: configuration.sourceMode,\n environment: configuration.environment)\n let hasClaudeBinary = ClaudeCLIResolver.isAvailable(environment: configuration.environment)\n let delegatedCooldownSeconds = ClaudeOAuthDelegatedRefreshCoordinator.cooldownRemainingSeconds()\n let planningInput = ClaudeSourcePlanningInput(\n runtime: configuration.runtime,\n selectedDataSource: configuration.usageDataSource,\n webExtrasEnabled: configuration.webExtrasEnabled,\n hasWebSession: hasKey,\n hasCLI: hasClaudeBinary,\n hasOAuthCredentials: hasOAuthCredentials)\n let plan = ClaudeSourcePlanner.resolve(input: planningInput)\n let strategy = plan.compatibilityStrategy\n\n lines.append(contentsOf: plan.debugLines())\n lines.append(\"hasSessionKey=\\(hasKey)\")\n lines.append(\"hasOAuthCredentials=\\(hasOAuthCredentials)\")\n lines.append(\"oauthCredentialOwner=\\(oauthProbe.ownerRawValue)\")\n lines.append(\"oauthCredentialSource=\\(oauthProbe.sourceRawValue)\")\n lines.append(\"oauthCredentialExpired=\\(oauthProbe.isExpired)\")\n lines.append(\"delegatedRefreshCLIAvailable=\\(hasClaudeBinary)\")\n lines.append(\"delegatedRefreshCooldownActive=\\(delegatedCooldownSeconds != nil)\")\n if let delegatedCooldownSeconds {\n lines.append(\"delegatedRefreshCooldownSeconds=\\(delegatedCooldownSeconds)\")\n }\n lines.append(\"hasClaudeBinary=\\(hasClaudeBinary)\")\n if strategy?.useWebExtras == true {\n lines.append(\"web_extras=enabled\")\n }\n lines.append(\"\")\n\n guard let strategy else {\n lines.append(\"No planner-selected Claude source.\")\n return lines.joined(separator: \"\\n\")\n }\n\n switch strategy.dataSource {\n case .auto:\n lines.append(\"Auto source selected.\")\n return lines.joined(separator: \"\\n\")\n case .web:\n do {\n let web: ClaudeWebAPIFetcher.WebUsageData =\n if let manualHeader {\n try await ClaudeWebAPIFetcher.fetchUsage(cookieHeader: manualHeader) { msg in\n lines.append(msg)\n }\n } else {\n try await ClaudeWebAPIFetcher.fetchUsage(browserDetection: browserDetection) { msg in\n lines.append(msg)\n }\n }\n lines.append(\"\")\n lines.append(\"Web API summary:\")\n\n let sessionReset = web.sessionResetsAt?.description ?? \"nil\"\n lines.append(\"session_used=\\(web.sessionPercentUsed)% resetsAt=\\(sessionReset)\")\n\n if let weekly = web.weeklyPercentUsed {\n let weeklyReset = web.weeklyResetsAt?.description ?? \"nil\"\n lines.append(\"weekly_used=\\(weekly)% resetsAt=\\(weeklyReset)\")\n } else {\n lines.append(\"weekly_used=nil\")\n }\n\n lines.append(\"opus_used=\\(web.opusPercentUsed?.description ?? \"nil\")\")\n\n if let extra = web.extraUsageCost {\n let resetsAt = extra.resetsAt?.description ?? \"nil\"\n let period = extra.period ?? \"nil\"\n let line =\n \"extra_usage used=\\(extra.used) limit=\\(extra.limit) \" +\n \"currency=\\(extra.currencyCode) period=\\(period) resetsAt=\\(resetsAt)\"\n lines.append(line)\n } else {\n lines.append(\"extra_usage=nil\")\n }\n\n return lines.joined(separator: \"\\n\")\n } catch {\n lines.append(\"Web API failed: \\(error.localizedDescription)\")\n return lines.joined(separator: \"\\n\")\n }\n case .cli:\n let fetcher = ClaudeUsageFetcher(\n browserDetection: browserDetection,\n environment: configuration.environment,\n runtime: configuration.runtime,\n dataSource: configuration.usageDataSource,\n keepCLISessionsAlive: configuration.keepCLISessionsAlive)\n let cli = await fetcher.debugRawProbe(model: \"sonnet\")\n lines.append(cli)\n return lines.joined(separator: \"\\n\")\n case .oauth:\n lines.append(\"OAuth source selected.\")\n return lines.joined(separator: \"\\n\")\n }\n }\n }\n}\n" }, { "path": "Sources/CodexBar/UsageStore+HighestUsage.swift", "content": "import CodexBarCore\nimport Foundation\n\n@MainActor\nextension UsageStore {\n /// Returns the enabled provider with the highest usage percentage (closest to rate limit).\n /// Excludes providers that are fully rate-limited.\n func providerWithHighestUsage() -> (provider: UsageProvider, usedPercent: Double)? {\n var highest: (provider: UsageProvider, usedPercent: Double)?\n for provider in self.enabledProviders() {\n guard let snapshot = self.snapshots[provider] else { continue }\n let window = self.menuBarMetricWindowForHighestUsage(provider: provider, snapshot: snapshot)\n let percent = window?.usedPercent ?? 0\n guard !self.shouldExcludeFromHighestUsage(\n provider: provider,\n snapshot: snapshot,\n metricPercent: percent)\n else {\n continue\n }\n if highest == nil || percent > highest!.usedPercent {\n highest = (provider, percent)\n }\n }\n return highest\n }\n\n private func menuBarMetricWindowForHighestUsage(provider: UsageProvider, snapshot: UsageSnapshot) -> RateWindow? {\n switch self.settings.menuBarMetricPreference(for: provider) {\n case .primary:\n return snapshot.primary ?? snapshot.secondary\n case .secondary:\n return snapshot.secondary ?? snapshot.primary\n case .average:\n guard let primary = snapshot.primary, let secondary = snapshot.secondary else {\n return snapshot.primary ?? snapshot.secondary\n }\n let usedPercent = (primary.usedPercent + secondary.usedPercent) / 2\n return RateWindow(usedPercent: usedPercent, windowMinutes: nil, resetsAt: nil, resetDescription: nil)\n case .automatic:\n if provider == .factory || provider == .kimi {\n return snapshot.secondary ?? snapshot.primary\n }\n if provider == .copilot,\n let primary = snapshot.primary,\n let secondary = snapshot.secondary\n {\n // Copilot can expose chat + completions quotas; rank by the more constrained one.\n return primary.usedPercent >= secondary.usedPercent ? primary : secondary\n }\n return snapshot.primary ?? snapshot.secondary\n }\n }\n\n private func shouldExcludeFromHighestUsage(\n provider: UsageProvider,\n snapshot: UsageSnapshot,\n metricPercent: Double)\n -> Bool\n {\n guard metricPercent >= 100 else { return false }\n if provider == .copilot,\n self.settings.menuBarMetricPreference(for: provider) == .automatic,\n let primary = snapshot.primary,\n let secondary = snapshot.secondary\n {\n // In automatic mode Copilot can have one depleted lane while another still has quota.\n return primary.usedPercent >= 100 && secondary.usedPercent >= 100\n }\n return true\n }\n}\n" }, { "path": "Sources/CodexBar/UsageStore+HistoricalPace.swift", "content": "import CodexBarCore\nimport CryptoKit\nimport Foundation\n\n@MainActor\nextension UsageStore {\n private static let minimumPaceExpectedPercent: Double = 3\n private static let backfillMaxTimestampMismatch: TimeInterval = 5 * 60\n\n func weeklyPace(provider: UsageProvider, window: RateWindow, now: Date = .init()) -> UsagePace? {\n guard provider == .codex || provider == .claude else { return nil }\n guard window.remainingPercent > 0 else { return nil }\n let resolved: UsagePace?\n if provider == .codex, self.settings.historicalTrackingEnabled {\n let codexAccountKey = self.codexHistoricalAccountKey()\n if self.codexHistoricalDatasetAccountKey == codexAccountKey,\n let historical = CodexHistoricalPaceEvaluator.evaluate(\n window: window,\n now: now,\n dataset: self.codexHistoricalDataset)\n {\n resolved = historical\n } else {\n resolved = UsagePace.weekly(window: window, now: now, defaultWindowMinutes: 10080)\n }\n } else {\n resolved = UsagePace.weekly(window: window, now: now, defaultWindowMinutes: 10080)\n }\n\n guard let resolved else { return nil }\n guard resolved.expectedUsedPercent >= Self.minimumPaceExpectedPercent else { return nil }\n return resolved\n }\n\n func recordCodexHistoricalSampleIfNeeded(snapshot: UsageSnapshot) {\n guard self.settings.historicalTrackingEnabled else { return }\n guard let weekly = snapshot.secondary else { return }\n\n let sampledAt = snapshot.updatedAt\n let accountKey = self.codexHistoricalAccountKey(preferredEmail: snapshot.accountEmail(for: .codex))\n let historyStore = self.historicalUsageHistoryStore\n Task.detached(priority: .utility) { [weak self] in\n let dataset = await historyStore.recordCodexWeekly(\n window: weekly,\n sampledAt: sampledAt,\n accountKey: accountKey)\n await MainActor.run { [weak self] in\n self?.setCodexHistoricalDataset(dataset, accountKey: accountKey)\n }\n }\n }\n\n func refreshHistoricalDatasetIfNeeded() async {\n if !self.settings.historicalTrackingEnabled {\n self.setCodexHistoricalDataset(nil, accountKey: nil)\n return\n }\n let accountKey = self.codexHistoricalAccountKey(dashboard: self.openAIDashboard)\n let dataset = await self.historicalUsageHistoryStore.loadCodexDataset(accountKey: accountKey)\n self.setCodexHistoricalDataset(dataset, accountKey: accountKey)\n if let dashboard = self.openAIDashboard {\n self.backfillCodexHistoricalFromDashboardIfNeeded(dashboard)\n }\n }\n\n func backfillCodexHistoricalFromDashboardIfNeeded(_ dashboard: OpenAIDashboardSnapshot) {\n guard self.settings.historicalTrackingEnabled else { return }\n guard !dashboard.usageBreakdown.isEmpty else { return }\n\n let codexSnapshot = self.snapshots[.codex]\n let accountKey = self.codexHistoricalAccountKey(\n preferredEmail: codexSnapshot?.accountEmail(for: .codex),\n dashboard: dashboard)\n let referenceWindow: RateWindow\n let calibrationAt: Date\n if let dashboardWeekly = dashboard.secondaryLimit {\n referenceWindow = dashboardWeekly\n calibrationAt = dashboard.updatedAt\n } else if let codexSnapshot, let snapshotWeekly = codexSnapshot.secondary {\n let mismatch = abs(codexSnapshot.updatedAt.timeIntervalSince(dashboard.updatedAt))\n guard mismatch <= Self.backfillMaxTimestampMismatch else { return }\n referenceWindow = snapshotWeekly\n calibrationAt = min(codexSnapshot.updatedAt, dashboard.updatedAt)\n } else {\n return\n }\n\n let historyStore = self.historicalUsageHistoryStore\n let usageBreakdown = dashboard.usageBreakdown\n Task.detached(priority: .utility) { [weak self] in\n let dataset = await historyStore.backfillCodexWeeklyFromUsageBreakdown(\n usageBreakdown,\n referenceWindow: referenceWindow,\n now: calibrationAt,\n accountKey: accountKey)\n await MainActor.run { [weak self] in\n self?.setCodexHistoricalDataset(dataset, accountKey: accountKey)\n }\n }\n }\n\n private func setCodexHistoricalDataset(_ dataset: CodexHistoricalDataset?, accountKey: String?) {\n self.codexHistoricalDataset = dataset\n self.codexHistoricalDatasetAccountKey = accountKey\n self.historicalPaceRevision += 1\n }\n\n private func codexHistoricalAccountKey(\n preferredEmail: String? = nil,\n dashboard: OpenAIDashboardSnapshot? = nil) -> String?\n {\n let sourceEmail = preferredEmail ??\n self.snapshots[.codex]?.accountEmail(for: .codex) ??\n dashboard?.signedInEmail ??\n self.codexAccountEmailForOpenAIDashboard()\n guard let sourceEmail else { return nil }\n let normalized = sourceEmail\n .trimmingCharacters(in: .whitespacesAndNewlines)\n .lowercased()\n guard !normalized.isEmpty else { return nil }\n return Self.sha256Hex(normalized)\n }\n\n private static func sha256Hex(_ input: String) -> String {\n let digest = SHA256.hash(data: Data(input.utf8))\n return digest.map { String(format: \"%02x\", $0) }.joined()\n }\n}\n" }, { "path": "Sources/CodexBar/UsageStore+Logging.swift", "content": "import CodexBarCore\n\nextension UsageStore {\n func logStartupState() {\n let modeSnapshot: [String: String] = [\n \"codexUsageSource\": self.settings.codexUsageDataSource.rawValue,\n \"claudeUsageSource\": self.settings.claudeUsageDataSource.rawValue,\n \"kiloUsageSource\": self.settings.kiloUsageDataSource.rawValue,\n \"codexCookieSource\": self.settings.codexCookieSource.rawValue,\n \"claudeCookieSource\": self.settings.claudeCookieSource.rawValue,\n \"cursorCookieSource\": self.settings.cursorCookieSource.rawValue,\n \"opencodeCookieSource\": self.settings.opencodeCookieSource.rawValue,\n \"factoryCookieSource\": self.settings.factoryCookieSource.rawValue,\n \"minimaxCookieSource\": self.settings.minimaxCookieSource.rawValue,\n \"kimiCookieSource\": self.settings.kimiCookieSource.rawValue,\n \"augmentCookieSource\": self.settings.augmentCookieSource.rawValue,\n \"ampCookieSource\": self.settings.ampCookieSource.rawValue,\n \"ollamaCookieSource\": self.settings.ollamaCookieSource.rawValue,\n \"openAIWebAccess\": self.settings.openAIWebAccessEnabled ? \"1\" : \"0\",\n \"claudeWebExtras\": self.settings.claudeWebExtrasEnabled ? \"1\" : \"0\",\n \"kiloExtras\": self.settings.kiloExtrasEnabled ? \"1\" : \"0\",\n ]\n ProviderLogging.logStartupState(\n logger: self.providerLogger,\n providers: Array(self.providerMetadata.keys),\n isEnabled: { provider in\n self.settings.isProviderEnabled(\n provider: provider,\n metadata: self.providerMetadata[provider]!)\n },\n modeSnapshot: modeSnapshot)\n }\n}\n" }, { "path": "Sources/CodexBar/UsageStore+OpenAIWeb.swift", "content": "import Foundation\n\n// MARK: - OpenAI web error messaging\n\nextension UsageStore {\n func openAIDashboardFriendlyError(\n body: String,\n targetEmail: String?,\n cookieImportStatus: String?) -> String?\n {\n let trimmed = body.trimmingCharacters(in: .whitespacesAndNewlines)\n let status = cookieImportStatus?.trimmingCharacters(in: .whitespacesAndNewlines)\n if trimmed.isEmpty {\n return [\n \"OpenAI web dashboard returned an empty page.\",\n \"Sign in to chatgpt.com and update OpenAI cookies in Providers → Codex.\",\n ].joined(separator: \" \")\n }\n\n let lower = trimmed.lowercased()\n let looksLikePublicLanding = lower.contains(\"skip to content\")\n && (lower.contains(\"about\") || lower.contains(\"openai\") || lower.contains(\"chatgpt\"))\n let looksLoggedOut = lower.contains(\"sign in\")\n || lower.contains(\"log in\")\n || lower.contains(\"create account\")\n || lower.contains(\"continue with google\")\n || lower.contains(\"continue with apple\")\n || lower.contains(\"continue with microsoft\")\n\n guard looksLikePublicLanding || looksLoggedOut else { return nil }\n let emailLabel = targetEmail?.trimmingCharacters(in: .whitespacesAndNewlines)\n let targetLabel = (emailLabel?.isEmpty == false) ? emailLabel! : \"your OpenAI account\"\n if let status, !status.isEmpty {\n if status.contains(\"cookies do not match Codex account\")\n || status.localizedCaseInsensitiveContains(\"cookie import failed\")\n {\n return [\n status,\n \"Sign in to chatgpt.com as \\(targetLabel), then update OpenAI cookies in Providers → Codex.\",\n ].joined(separator: \" \")\n }\n }\n return [\n \"OpenAI web dashboard returned a public page (not signed in).\",\n \"Sign in to chatgpt.com as \\(targetLabel), then update OpenAI cookies in Providers → Codex.\",\n ].joined(separator: \" \")\n }\n}\n" }, { "path": "Sources/CodexBar/UsageStore+Refresh.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension UsageStore {\n /// Force refresh Augment session (called from UI button)\n func forceRefreshAugmentSession() async {\n await self.performRuntimeAction(.forceSessionRefresh, for: .augment)\n }\n\n func refreshProvider(_ provider: UsageProvider, allowDisabled: Bool = false) async {\n guard let spec = self.providerSpecs[provider] else { return }\n\n if !spec.isEnabled(), !allowDisabled {\n self.refreshingProviders.remove(provider)\n await MainActor.run {\n self.snapshots.removeValue(forKey: provider)\n self.errors[provider] = nil\n self.lastSourceLabels.removeValue(forKey: provider)\n self.lastFetchAttempts.removeValue(forKey: provider)\n self.accountSnapshots.removeValue(forKey: provider)\n self.tokenSnapshots.removeValue(forKey: provider)\n self.tokenErrors[provider] = nil\n self.failureGates[provider]?.reset()\n self.tokenFailureGates[provider]?.reset()\n self.statuses.removeValue(forKey: provider)\n self.lastKnownSessionRemaining.removeValue(forKey: provider)\n self.lastKnownSessionWindowSource.removeValue(forKey: provider)\n self.lastTokenFetchAt.removeValue(forKey: provider)\n }\n return\n }\n\n self.refreshingProviders.insert(provider)\n defer { self.refreshingProviders.remove(provider) }\n\n let tokenAccounts = self.tokenAccounts(for: provider)\n if self.shouldFetchAllTokenAccounts(provider: provider, accounts: tokenAccounts) {\n await self.refreshTokenAccounts(provider: provider, accounts: tokenAccounts)\n return\n } else {\n _ = await MainActor.run {\n self.accountSnapshots.removeValue(forKey: provider)\n }\n }\n\n let fetchContext = spec.makeFetchContext()\n let descriptor = spec.descriptor\n // Keep provider fetch work off MainActor so slow keychain/process reads don't stall menu/UI responsiveness.\n let outcome = await withTaskGroup(\n of: ProviderFetchOutcome.self,\n returning: ProviderFetchOutcome.self)\n { group in\n group.addTask {\n await descriptor.fetchOutcome(context: fetchContext)\n }\n return await group.next()!\n }\n if provider == .claude,\n ClaudeOAuthCredentialsStore.invalidateCacheIfCredentialsFileChanged()\n {\n await MainActor.run {\n self.snapshots.removeValue(forKey: .claude)\n self.errors[.claude] = nil\n self.lastSourceLabels.removeValue(forKey: .claude)\n self.lastFetchAttempts.removeValue(forKey: .claude)\n self.accountSnapshots.removeValue(forKey: .claude)\n self.tokenSnapshots.removeValue(forKey: .claude)\n self.tokenErrors[.claude] = nil\n self.failureGates[.claude]?.reset()\n self.tokenFailureGates[.claude]?.reset()\n self.lastTokenFetchAt.removeValue(forKey: .claude)\n }\n }\n await MainActor.run {\n self.lastFetchAttempts[provider] = outcome.attempts\n }\n\n switch outcome.result {\n case let .success(result):\n let scoped = result.usage.scoped(to: provider)\n await MainActor.run {\n self.handleSessionQuotaTransition(provider: provider, snapshot: scoped)\n self.snapshots[provider] = scoped\n self.lastSourceLabels[provider] = result.sourceLabel\n self.errors[provider] = nil\n self.failureGates[provider]?.recordSuccess()\n }\n if let runtime = self.providerRuntimes[provider] {\n let context = ProviderRuntimeContext(\n provider: provider, settings: self.settings, store: self)\n runtime.providerDidRefresh(context: context, provider: provider)\n }\n if provider == .codex {\n self.recordCodexHistoricalSampleIfNeeded(snapshot: scoped)\n }\n case let .failure(error):\n await MainActor.run {\n let hadPriorData = self.snapshots[provider] != nil\n let shouldSurface =\n self.failureGates[provider]?\n .shouldSurfaceError(onFailureWithPriorData: hadPriorData) ?? true\n if shouldSurface {\n self.errors[provider] = error.localizedDescription\n self.snapshots.removeValue(forKey: provider)\n } else {\n self.errors[provider] = nil\n }\n }\n if let runtime = self.providerRuntimes[provider] {\n let context = ProviderRuntimeContext(\n provider: provider, settings: self.settings, store: self)\n runtime.providerDidFail(context: context, provider: provider, error: error)\n }\n }\n }\n}\n" }, { "path": "Sources/CodexBar/UsageStore+Status.swift", "content": "import Foundation\n\nextension UsageStore {\n static func fetchStatus(from baseURL: URL) async throws -> ProviderStatus {\n let apiURL = baseURL.appendingPathComponent(\"api/v2/status.json\")\n var request = URLRequest(url: apiURL)\n request.timeoutInterval = 10\n\n let (data, _) = try await URLSession.shared.data(for: request, delegate: nil)\n\n struct Response: Decodable {\n struct Status: Decodable {\n let indicator: String\n let description: String?\n }\n\n struct Page: Decodable {\n let updatedAt: Date?\n\n private enum CodingKeys: String, CodingKey {\n case updatedAt = \"updated_at\"\n }\n }\n\n let page: Page?\n let status: Status\n }\n\n let decoder = JSONDecoder()\n decoder.dateDecodingStrategy = .custom { decoder in\n let container = try decoder.singleValueContainer()\n let raw = try container.decode(String.self)\n let formatter = ISO8601DateFormatter()\n formatter.formatOptions = [.withInternetDateTime, .withFractionalSeconds]\n if let date = formatter.date(from: raw) { return date }\n formatter.formatOptions = [.withInternetDateTime]\n if let date = formatter.date(from: raw) { return date }\n throw DecodingError.dataCorruptedError(in: container, debugDescription: \"Invalid ISO8601 date\")\n }\n\n let response = try decoder.decode(Response.self, from: data)\n let indicator = ProviderStatusIndicator(rawValue: response.status.indicator) ?? .unknown\n return ProviderStatus(\n indicator: indicator,\n description: response.status.description,\n updatedAt: response.page?.updatedAt)\n }\n\n static func fetchWorkspaceStatus(productID: String) async throws -> ProviderStatus {\n guard let url = URL(string: \"https://www.google.com/appsstatus/dashboard/incidents.json\") else {\n throw URLError(.badURL)\n }\n var request = URLRequest(url: url)\n request.timeoutInterval = 10\n let (data, _) = try await URLSession.shared.data(for: request, delegate: nil)\n return try Self.parseGoogleWorkspaceStatus(data: data, productID: productID)\n }\n\n static func parseGoogleWorkspaceStatus(data: Data, productID: String) throws -> ProviderStatus {\n let decoder = JSONDecoder()\n decoder.keyDecodingStrategy = .convertFromSnakeCase\n decoder.dateDecodingStrategy = .custom { decoder in\n let container = try decoder.singleValueContainer()\n let raw = try container.decode(String.self)\n let formatter = ISO8601DateFormatter()\n formatter.formatOptions = [.withInternetDateTime, .withFractionalSeconds]\n if let date = formatter.date(from: raw) { return date }\n formatter.formatOptions = [.withInternetDateTime]\n if let date = formatter.date(from: raw) { return date }\n throw DecodingError.dataCorruptedError(in: container, debugDescription: \"Invalid ISO8601 date\")\n }\n\n let incidents = try decoder.decode([GoogleWorkspaceIncident].self, from: data)\n let active = incidents.filter { $0.isRelevant(productID: productID) && $0.isActive }\n guard !active.isEmpty else {\n return ProviderStatus(indicator: .none, description: nil, updatedAt: nil)\n }\n\n var best: (\n indicator: ProviderStatusIndicator,\n incident: GoogleWorkspaceIncident,\n update: GoogleWorkspaceUpdate?)\n best = (indicator: .none, incident: active[0], update: active[0].mostRecentUpdate ?? active[0].updates?.last)\n\n for incident in active {\n let update = incident.mostRecentUpdate ?? incident.updates?.last\n let indicator = Self.workspaceIndicator(\n status: update?.status ?? incident.statusImpact,\n severity: incident.severity)\n if Self.indicatorRank(indicator) <= Self.indicatorRank(best.indicator) { continue }\n best = (indicator: indicator, incident: incident, update: update)\n }\n\n let description = Self.workspaceSummary(from: best.update?.text ?? best.incident.externalDesc)\n let updatedAt = best.update?.when ?? best.incident.modified ?? best.incident.begin\n return ProviderStatus(indicator: best.indicator, description: description, updatedAt: updatedAt)\n }\n\n private static func indicatorRank(_ indicator: ProviderStatusIndicator) -> Int {\n switch indicator {\n case .none: 0\n case .maintenance: 1\n case .minor: 2\n case .major: 3\n case .critical: 4\n case .unknown: 1\n }\n }\n\n private static func workspaceIndicator(status: String?, severity: String?) -> ProviderStatusIndicator {\n switch status?.uppercased() {\n case \"AVAILABLE\": return .none\n case \"SERVICE_INFORMATION\": return .minor\n case \"SERVICE_DISRUPTION\": return .major\n case \"SERVICE_OUTAGE\": return .critical\n case \"SERVICE_MAINTENANCE\", \"SCHEDULED_MAINTENANCE\": return .maintenance\n default: break\n }\n\n switch severity?.lowercased() {\n case \"low\": return .minor\n case \"medium\": return .major\n case \"high\": return .critical\n default: return .minor\n }\n }\n\n private static func workspaceSummary(from text: String?) -> String? {\n guard let text else { return nil }\n let normalized = text\n .replacingOccurrences(of: \"\\r\\n\", with: \"\\n\")\n .replacingOccurrences(of: \"\\r\", with: \"\\n\")\n let lines = normalized.split(separator: \"\\n\", omittingEmptySubsequences: true)\n for rawLine in lines {\n let trimmed = rawLine.trimmingCharacters(in: .whitespacesAndNewlines)\n if trimmed.isEmpty { continue }\n let lower = trimmed.lowercased()\n if lower.hasPrefix(\"**summary\") || lower.hasPrefix(\"**description\") || lower == \"summary\" {\n continue\n }\n var cleaned = trimmed.replacingOccurrences(of: \"**\", with: \"\")\n cleaned = cleaned.replacingOccurrences(\n of: #\"\\[([^\\]]+)\\]\\([^)]+\\)\"#,\n with: \"$1\",\n options: .regularExpression)\n if cleaned.hasPrefix(\"- \") {\n cleaned.removeFirst(2)\n }\n cleaned = cleaned.trimmingCharacters(in: .whitespacesAndNewlines)\n if !cleaned.isEmpty { return cleaned }\n }\n return nil\n }\n\n private struct GoogleWorkspaceIncident: Decodable {\n let begin: Date?\n let end: Date?\n let modified: Date?\n let externalDesc: String?\n let statusImpact: String?\n let severity: String?\n let affectedProducts: [GoogleWorkspaceProduct]?\n let currentlyAffectedProducts: [GoogleWorkspaceProduct]?\n let mostRecentUpdate: GoogleWorkspaceUpdate?\n let updates: [GoogleWorkspaceUpdate]?\n\n var isActive: Bool {\n self.end == nil\n }\n\n func isRelevant(productID: String) -> Bool {\n if let current = currentlyAffectedProducts {\n return current.contains(where: { $0.id == productID })\n }\n return self.affectedProducts?.contains(where: { $0.id == productID }) ?? false\n }\n }\n\n private struct GoogleWorkspaceProduct: Decodable {\n let title: String?\n let id: String\n }\n\n private struct GoogleWorkspaceUpdate: Decodable {\n let when: Date?\n let status: String?\n let text: String?\n }\n}\n" }, { "path": "Sources/CodexBar/UsageStore+Timeout.swift", "content": "import Foundation\n\nextension UsageStore {\n nonisolated static func runWithTimeout(\n seconds: Double,\n operation: @escaping @Sendable () async -> String) async -> String\n {\n await withTaskGroup(of: String?.self) { group -> String in\n group.addTask { await operation() }\n group.addTask {\n try? await Task.sleep(nanoseconds: UInt64(seconds * 1_000_000_000))\n return nil\n }\n let result = await group.next()?.flatMap(\\.self)\n group.cancelAll()\n return result ?? \"Probe timed out after \\(Int(seconds))s\"\n }\n }\n}\n" }, { "path": "Sources/CodexBar/UsageStore+TokenAccounts.swift", "content": "import CodexBarCore\nimport Foundation\n\nstruct TokenAccountUsageSnapshot: Identifiable {\n let id: UUID\n let account: ProviderTokenAccount\n let snapshot: UsageSnapshot?\n let error: String?\n let sourceLabel: String?\n\n init(account: ProviderTokenAccount, snapshot: UsageSnapshot?, error: String?, sourceLabel: String?) {\n self.id = account.id\n self.account = account\n self.snapshot = snapshot\n self.error = error\n self.sourceLabel = sourceLabel\n }\n}\n\nextension UsageStore {\n func tokenAccounts(for provider: UsageProvider) -> [ProviderTokenAccount] {\n guard TokenAccountSupportCatalog.support(for: provider) != nil else { return [] }\n return self.settings.tokenAccounts(for: provider)\n }\n\n func shouldFetchAllTokenAccounts(provider: UsageProvider, accounts: [ProviderTokenAccount]) -> Bool {\n guard TokenAccountSupportCatalog.support(for: provider) != nil else { return false }\n return self.settings.showAllTokenAccountsInMenu && accounts.count > 1\n }\n\n func refreshTokenAccounts(provider: UsageProvider, accounts: [ProviderTokenAccount]) async {\n let selectedAccount = self.settings.selectedTokenAccount(for: provider)\n let limitedAccounts = self.limitedTokenAccounts(accounts, selected: selectedAccount)\n let effectiveSelected = selectedAccount ?? limitedAccounts.first\n var snapshots: [TokenAccountUsageSnapshot] = []\n var selectedOutcome: ProviderFetchOutcome?\n var selectedSnapshot: UsageSnapshot?\n\n for account in limitedAccounts {\n let override = TokenAccountOverride(provider: provider, account: account)\n let outcome = await self.fetchOutcome(provider: provider, override: override)\n let resolved = self.resolveAccountOutcome(outcome, provider: provider, account: account)\n snapshots.append(resolved.snapshot)\n if account.id == effectiveSelected?.id {\n selectedOutcome = outcome\n selectedSnapshot = resolved.usage\n }\n }\n\n await MainActor.run {\n self.accountSnapshots[provider] = snapshots\n }\n\n if let selectedOutcome {\n await self.applySelectedOutcome(\n selectedOutcome,\n provider: provider,\n account: effectiveSelected,\n fallbackSnapshot: selectedSnapshot)\n }\n }\n\n func limitedTokenAccounts(\n _ accounts: [ProviderTokenAccount],\n selected: ProviderTokenAccount?) -> [ProviderTokenAccount]\n {\n let limit = 6\n if accounts.count <= limit { return accounts }\n var limited = Array(accounts.prefix(limit))\n if let selected, !limited.contains(where: { $0.id == selected.id }) {\n limited.removeLast()\n limited.append(selected)\n }\n return limited\n }\n\n func fetchOutcome(\n provider: UsageProvider,\n override: TokenAccountOverride?) async -> ProviderFetchOutcome\n {\n let descriptor = ProviderDescriptorRegistry.descriptor(for: provider)\n let sourceMode = self.sourceMode(for: provider)\n let snapshot = ProviderRegistry.makeSettingsSnapshot(settings: self.settings, tokenOverride: override)\n let env = ProviderRegistry.makeEnvironment(\n base: ProcessInfo.processInfo.environment,\n provider: provider,\n settings: self.settings,\n tokenOverride: override)\n let verbose = self.settings.isVerboseLoggingEnabled\n let context = ProviderFetchContext(\n runtime: .app,\n sourceMode: sourceMode,\n includeCredits: false,\n webTimeout: 60,\n webDebugDumpHTML: false,\n verbose: verbose,\n env: env,\n settings: snapshot,\n fetcher: self.codexFetcher,\n claudeFetcher: self.claudeFetcher,\n browserDetection: self.browserDetection)\n return await descriptor.fetchOutcome(context: context)\n }\n\n func sourceMode(for provider: UsageProvider) -> ProviderSourceMode {\n ProviderCatalog.implementation(for: provider)?\n .sourceMode(context: ProviderSourceModeContext(provider: provider, settings: self.settings))\n ?? .auto\n }\n\n private struct ResolvedAccountOutcome {\n let snapshot: TokenAccountUsageSnapshot\n let usage: UsageSnapshot?\n }\n\n private func resolveAccountOutcome(\n _ outcome: ProviderFetchOutcome,\n provider: UsageProvider,\n account: ProviderTokenAccount) -> ResolvedAccountOutcome\n {\n switch outcome.result {\n case let .success(result):\n let scoped = result.usage.scoped(to: provider)\n let labeled = self.applyAccountLabel(scoped, provider: provider, account: account)\n let snapshot = TokenAccountUsageSnapshot(\n account: account,\n snapshot: labeled,\n error: nil,\n sourceLabel: result.sourceLabel)\n return ResolvedAccountOutcome(snapshot: snapshot, usage: labeled)\n case let .failure(error):\n let snapshot = TokenAccountUsageSnapshot(\n account: account,\n snapshot: nil,\n error: error.localizedDescription,\n sourceLabel: nil)\n return ResolvedAccountOutcome(snapshot: snapshot, usage: nil)\n }\n }\n\n func applySelectedOutcome(\n _ outcome: ProviderFetchOutcome,\n provider: UsageProvider,\n account: ProviderTokenAccount?,\n fallbackSnapshot: UsageSnapshot?) async\n {\n await MainActor.run {\n self.lastFetchAttempts[provider] = outcome.attempts\n }\n switch outcome.result {\n case let .success(result):\n let scoped = result.usage.scoped(to: provider)\n let labeled: UsageSnapshot = if let account {\n self.applyAccountLabel(scoped, provider: provider, account: account)\n } else {\n scoped\n }\n await MainActor.run {\n self.handleSessionQuotaTransition(provider: provider, snapshot: labeled)\n self.snapshots[provider] = labeled\n self.lastSourceLabels[provider] = result.sourceLabel\n self.errors[provider] = nil\n self.failureGates[provider]?.recordSuccess()\n }\n case let .failure(error):\n await MainActor.run {\n let hadPriorData = self.snapshots[provider] != nil || fallbackSnapshot != nil\n let shouldSurface = self.failureGates[provider]?\n .shouldSurfaceError(onFailureWithPriorData: hadPriorData) ?? true\n if shouldSurface {\n self.errors[provider] = error.localizedDescription\n self.snapshots.removeValue(forKey: provider)\n } else {\n self.errors[provider] = nil\n }\n }\n }\n }\n\n func applyAccountLabel(\n _ snapshot: UsageSnapshot,\n provider: UsageProvider,\n account: ProviderTokenAccount) -> UsageSnapshot\n {\n let label = account.label.trimmingCharacters(in: .whitespacesAndNewlines)\n guard !label.isEmpty else { return snapshot }\n let existing = snapshot.identity(for: provider)\n let email = existing?.accountEmail?.trimmingCharacters(in: .whitespacesAndNewlines)\n let resolvedEmail = (email?.isEmpty ?? true) ? label : email\n let identity = ProviderIdentitySnapshot(\n providerID: provider,\n accountEmail: resolvedEmail,\n accountOrganization: existing?.accountOrganization,\n loginMethod: existing?.loginMethod)\n return snapshot.withIdentity(identity)\n }\n}\n" }, { "path": "Sources/CodexBar/UsageStore+TokenCost.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension UsageStore {\n func tokenSnapshot(for provider: UsageProvider) -> CostUsageTokenSnapshot? {\n self.tokenSnapshots[provider]\n }\n\n func tokenError(for provider: UsageProvider) -> String? {\n self.tokenErrors[provider]\n }\n\n func tokenLastAttemptAt(for provider: UsageProvider) -> Date? {\n self.lastTokenFetchAt[provider]\n }\n\n func isTokenRefreshInFlight(for provider: UsageProvider) -> Bool {\n self.tokenRefreshInFlight.contains(provider)\n }\n\n nonisolated static func costUsageCacheDirectory(\n fileManager: FileManager = .default) -> URL\n {\n let root = fileManager.urls(for: .cachesDirectory, in: .userDomainMask).first!\n return root\n .appendingPathComponent(\"CodexBar\", isDirectory: true)\n .appendingPathComponent(\"cost-usage\", isDirectory: true)\n }\n\n nonisolated static func tokenCostNoDataMessage(for provider: UsageProvider) -> String {\n ProviderDescriptorRegistry.descriptor(for: provider).tokenCost.noDataMessage()\n }\n}\n" }, { "path": "Sources/CodexBar/UsageStore+WidgetSnapshot.swift", "content": "import CodexBarCore\nimport Foundation\n#if canImport(WidgetKit)\nimport WidgetKit\n#endif\n\nextension UsageStore {\n func persistWidgetSnapshot(reason: String) {\n let snapshot = self.makeWidgetSnapshot()\n Task.detached(priority: .utility) {\n WidgetSnapshotStore.save(snapshot)\n #if canImport(WidgetKit)\n await MainActor.run {\n WidgetCenter.shared.reloadAllTimelines()\n }\n #endif\n }\n }\n\n private func makeWidgetSnapshot() -> WidgetSnapshot {\n let enabledProviders = self.enabledProviders()\n let entries = UsageProvider.allCases.compactMap { provider in\n self.makeWidgetEntry(for: provider)\n }\n return WidgetSnapshot(entries: entries, enabledProviders: enabledProviders, generatedAt: Date())\n }\n\n private func makeWidgetEntry(for provider: UsageProvider) -> WidgetSnapshot.ProviderEntry? {\n guard let snapshot = self.snapshots[provider] else { return nil }\n\n let tokenSnapshot = self.tokenSnapshots[provider]\n let dailyUsage = tokenSnapshot?.daily.map { entry in\n WidgetSnapshot.DailyUsagePoint(\n dayKey: entry.date,\n totalTokens: entry.totalTokens,\n costUSD: entry.costUSD)\n } ?? []\n\n let tokenUsage = Self.widgetTokenUsageSummary(from: tokenSnapshot)\n let creditsRemaining = provider == .codex ? self.credits?.remaining : nil\n let codeReviewRemaining = provider == .codex ? self.openAIDashboard?.codeReviewRemainingPercent : nil\n\n return WidgetSnapshot.ProviderEntry(\n provider: provider,\n updatedAt: snapshot.updatedAt,\n primary: snapshot.primary,\n secondary: snapshot.secondary,\n tertiary: snapshot.tertiary,\n creditsRemaining: creditsRemaining,\n codeReviewRemainingPercent: codeReviewRemaining,\n tokenUsage: tokenUsage,\n dailyUsage: dailyUsage)\n }\n\n private nonisolated static func widgetTokenUsageSummary(\n from snapshot: CostUsageTokenSnapshot?) -> WidgetSnapshot.TokenUsageSummary?\n {\n guard let snapshot else { return nil }\n let fallbackTokens = snapshot.daily.compactMap(\\.totalTokens).reduce(0, +)\n let monthTokensValue = snapshot.last30DaysTokens ?? (fallbackTokens > 0 ? fallbackTokens : nil)\n return WidgetSnapshot.TokenUsageSummary(\n sessionCostUSD: snapshot.sessionCostUSD,\n sessionTokens: snapshot.sessionTokens,\n last30DaysCostUSD: snapshot.last30DaysCostUSD,\n last30DaysTokens: monthTokensValue)\n }\n}\n" }, { "path": "Sources/CodexBar/UsageStore.swift", "content": "import AppKit\nimport CodexBarCore\nimport Foundation\nimport Observation\nimport SweetCookieKit\n\n// MARK: - Observation helpers\n\n@MainActor\nextension UsageStore {\n var menuObservationToken: Int {\n _ = self.snapshots\n _ = self.errors\n _ = self.lastSourceLabels\n _ = self.lastFetchAttempts\n _ = self.accountSnapshots\n _ = self.tokenSnapshots\n _ = self.tokenErrors\n _ = self.tokenRefreshInFlight\n _ = self.credits\n _ = self.lastCreditsError\n _ = self.openAIDashboard\n _ = self.lastOpenAIDashboardError\n _ = self.openAIDashboardRequiresLogin\n _ = self.openAIDashboardCookieImportStatus\n _ = self.openAIDashboardCookieImportDebugLog\n _ = self.versions\n _ = self.isRefreshing\n _ = self.refreshingProviders\n _ = self.pathDebugInfo\n _ = self.statuses\n _ = self.probeLogs\n _ = self.historicalPaceRevision\n return 0\n }\n\n func observeSettingsChanges() {\n withObservationTracking {\n _ = self.settings.refreshFrequency\n _ = self.settings.statusChecksEnabled\n _ = self.settings.sessionQuotaNotificationsEnabled\n _ = self.settings.usageBarsShowUsed\n _ = self.settings.costUsageEnabled\n _ = self.settings.randomBlinkEnabled\n _ = self.settings.configRevision\n for implementation in ProviderCatalog.all {\n implementation.observeSettings(self.settings)\n }\n _ = self.settings.showAllTokenAccountsInMenu\n _ = self.settings.tokenAccountsByProvider\n _ = self.settings.mergeIcons\n _ = self.settings.selectedMenuProvider\n _ = self.settings.debugLoadingPattern\n _ = self.settings.debugKeepCLISessionsAlive\n _ = self.settings.historicalTrackingEnabled\n } onChange: { [weak self] in\n Task { @MainActor [weak self] in\n guard let self else { return }\n self.observeSettingsChanges()\n self.probeLogs = [:]\n guard self.startupBehavior.automaticallyStartsBackgroundWork else { return }\n self.startTimer()\n self.updateProviderRuntimes()\n await self.refreshHistoricalDatasetIfNeeded()\n await self.refresh()\n }\n }\n }\n}\n\n@MainActor\n@Observable\nfinal class UsageStore {\n enum StartupBehavior {\n case automatic\n case full\n case testing\n\n var automaticallyStartsBackgroundWork: Bool {\n switch self {\n case .automatic, .full:\n true\n case .testing:\n false\n }\n }\n\n func resolved(isRunningTests: Bool) -> StartupBehavior {\n switch self {\n case .automatic:\n isRunningTests ? .testing : .full\n case .full, .testing:\n self\n }\n }\n }\n\n var snapshots: [UsageProvider: UsageSnapshot] = [:]\n var errors: [UsageProvider: String] = [:]\n var lastSourceLabels: [UsageProvider: String] = [:]\n var lastFetchAttempts: [UsageProvider: [ProviderFetchAttempt]] = [:]\n var accountSnapshots: [UsageProvider: [TokenAccountUsageSnapshot]] = [:]\n var tokenSnapshots: [UsageProvider: CostUsageTokenSnapshot] = [:]\n var tokenErrors: [UsageProvider: String] = [:]\n var tokenRefreshInFlight: Set = []\n var credits: CreditsSnapshot?\n var lastCreditsError: String?\n var openAIDashboard: OpenAIDashboardSnapshot?\n var lastOpenAIDashboardError: String?\n var openAIDashboardRequiresLogin: Bool = false\n var openAIDashboardCookieImportStatus: String?\n var openAIDashboardCookieImportDebugLog: String?\n var versions: [UsageProvider: String] = [:]\n var isRefreshing = false\n var refreshingProviders: Set = []\n var debugForceAnimation = false\n var pathDebugInfo: PathDebugSnapshot = .empty\n var statuses: [UsageProvider: ProviderStatus] = [:]\n var probeLogs: [UsageProvider: String] = [:]\n var historicalPaceRevision: Int = 0\n @ObservationIgnored private var lastCreditsSnapshot: CreditsSnapshot?\n @ObservationIgnored private var creditsFailureStreak: Int = 0\n @ObservationIgnored private var lastOpenAIDashboardSnapshot: OpenAIDashboardSnapshot?\n @ObservationIgnored private var lastOpenAIDashboardTargetEmail: String?\n @ObservationIgnored private var lastOpenAIDashboardCookieImportAttemptAt: Date?\n @ObservationIgnored private var lastOpenAIDashboardCookieImportEmail: String?\n @ObservationIgnored private var openAIWebAccountDidChange: Bool = false\n\n @ObservationIgnored let codexFetcher: UsageFetcher\n @ObservationIgnored let claudeFetcher: any ClaudeUsageFetching\n @ObservationIgnored private let costUsageFetcher: CostUsageFetcher\n @ObservationIgnored let browserDetection: BrowserDetection\n @ObservationIgnored private let registry: ProviderRegistry\n @ObservationIgnored let settings: SettingsStore\n @ObservationIgnored private let sessionQuotaNotifier: any SessionQuotaNotifying\n @ObservationIgnored private let sessionQuotaLogger = CodexBarLog.logger(LogCategories.sessionQuota)\n @ObservationIgnored private let openAIWebLogger = CodexBarLog.logger(LogCategories.openAIWeb)\n @ObservationIgnored private let tokenCostLogger = CodexBarLog.logger(LogCategories.tokenCost)\n @ObservationIgnored let augmentLogger = CodexBarLog.logger(LogCategories.augment)\n @ObservationIgnored let providerLogger = CodexBarLog.logger(LogCategories.providers)\n @ObservationIgnored private var openAIWebDebugLines: [String] = []\n @ObservationIgnored var failureGates: [UsageProvider: ConsecutiveFailureGate] = [:]\n @ObservationIgnored var tokenFailureGates: [UsageProvider: ConsecutiveFailureGate] = [:]\n @ObservationIgnored var providerSpecs: [UsageProvider: ProviderSpec] = [:]\n @ObservationIgnored let providerMetadata: [UsageProvider: ProviderMetadata]\n @ObservationIgnored var providerRuntimes: [UsageProvider: any ProviderRuntime] = [:]\n @ObservationIgnored private var timerTask: Task?\n @ObservationIgnored private var tokenTimerTask: Task?\n @ObservationIgnored private var tokenRefreshSequenceTask: Task?\n @ObservationIgnored private var pathDebugRefreshTask: Task?\n @ObservationIgnored let historicalUsageHistoryStore: HistoricalUsageHistoryStore\n @ObservationIgnored var codexHistoricalDataset: CodexHistoricalDataset?\n @ObservationIgnored var codexHistoricalDatasetAccountKey: String?\n @ObservationIgnored var lastKnownSessionRemaining: [UsageProvider: Double] = [:]\n @ObservationIgnored var lastKnownSessionWindowSource: [UsageProvider: SessionQuotaWindowSource] = [:]\n @ObservationIgnored var lastTokenFetchAt: [UsageProvider: Date] = [:]\n @ObservationIgnored private var hasCompletedInitialRefresh: Bool = false\n @ObservationIgnored private let tokenFetchTTL: TimeInterval = 60 * 60\n @ObservationIgnored private let tokenFetchTimeout: TimeInterval = 10 * 60\n @ObservationIgnored private let startupBehavior: StartupBehavior\n\n init(\n fetcher: UsageFetcher,\n browserDetection: BrowserDetection,\n claudeFetcher: (any ClaudeUsageFetching)? = nil,\n costUsageFetcher: CostUsageFetcher = CostUsageFetcher(),\n settings: SettingsStore,\n registry: ProviderRegistry = .shared,\n historicalUsageHistoryStore: HistoricalUsageHistoryStore = HistoricalUsageHistoryStore(),\n sessionQuotaNotifier: any SessionQuotaNotifying = SessionQuotaNotifier(),\n startupBehavior: StartupBehavior = .automatic)\n {\n self.codexFetcher = fetcher\n self.browserDetection = browserDetection\n self.claudeFetcher = claudeFetcher ?? ClaudeUsageFetcher(browserDetection: browserDetection)\n self.costUsageFetcher = costUsageFetcher\n self.settings = settings\n self.registry = registry\n self.historicalUsageHistoryStore = historicalUsageHistoryStore\n self.sessionQuotaNotifier = sessionQuotaNotifier\n self.startupBehavior = startupBehavior.resolved(isRunningTests: Self.isRunningTestsProcess())\n self.providerMetadata = registry.metadata\n self\n .failureGates = Dictionary(\n uniqueKeysWithValues: UsageProvider.allCases\n .map { ($0, ConsecutiveFailureGate()) })\n self.tokenFailureGates = Dictionary(\n uniqueKeysWithValues: UsageProvider.allCases\n .map { ($0, ConsecutiveFailureGate()) })\n self.providerSpecs = registry.specs(\n settings: settings,\n metadata: self.providerMetadata,\n codexFetcher: fetcher,\n claudeFetcher: self.claudeFetcher,\n browserDetection: browserDetection)\n self.providerRuntimes = Dictionary(uniqueKeysWithValues: ProviderCatalog.all.compactMap { implementation in\n implementation.makeRuntime().map { (implementation.id, $0) }\n })\n self.logStartupState()\n self.bindSettings()\n self.pathDebugInfo = PathDebugSnapshot(\n codexBinary: nil,\n claudeBinary: nil,\n geminiBinary: nil,\n effectivePATH: PathBuilder.effectivePATH(purposes: [.rpc, .tty, .nodeTooling]),\n loginShellPATH: LoginShellPathCache.shared.current?.joined(separator: \":\"))\n guard self.startupBehavior.automaticallyStartsBackgroundWork else { return }\n self.detectVersions()\n self.updateProviderRuntimes()\n Task { @MainActor [weak self] in\n self?.schedulePathDebugInfoRefresh()\n }\n LoginShellPathCache.shared.captureOnce { [weak self] _ in\n Task { @MainActor [weak self] in\n self?.schedulePathDebugInfoRefresh()\n }\n }\n Task { @MainActor [weak self] in\n await self?.refreshHistoricalDatasetIfNeeded()\n }\n Task { await self.refresh() }\n self.startTimer()\n self.startTokenTimer()\n }\n\n private static func isRunningTestsProcess() -> Bool {\n let environment = ProcessInfo.processInfo.environment\n if environment[\"XCTestConfigurationFilePath\"] != nil { return true }\n if environment[\"XCTestSessionIdentifier\"] != nil { return true }\n if environment[\"SWIFT_TESTING_ENABLED\"] != nil { return true }\n return CommandLine.arguments.contains { argument in\n argument.contains(\"xctest\") || argument.contains(\"swift-testing\")\n }\n }\n\n /// Returns the login method (plan type) for the specified provider, if available.\n private func loginMethod(for provider: UsageProvider) -> String? {\n self.snapshots[provider]?.loginMethod(for: provider)\n }\n\n /// Returns true if the Claude account appears to be a subscription (Max, Pro, Ultra, Team).\n /// Returns false for API users or when plan cannot be determined.\n func isClaudeSubscription() -> Bool {\n Self.isSubscriptionPlan(self.loginMethod(for: .claude))\n }\n\n /// Determines if a login method string indicates a Claude subscription plan.\n /// Known subscription indicators: Max, Pro, Ultra, Team (case-insensitive).\n nonisolated static func isSubscriptionPlan(_ loginMethod: String?) -> Bool {\n ClaudePlan.isSubscriptionLoginMethod(loginMethod)\n }\n\n func version(for provider: UsageProvider) -> String? {\n self.versions[provider]\n }\n\n var preferredSnapshot: UsageSnapshot? {\n for provider in self.enabledProviders() {\n if let snap = self.snapshots[provider] { return snap }\n }\n return nil\n }\n\n var iconStyle: IconStyle {\n let enabled = self.enabledProviders()\n if enabled.count > 1 { return .combined }\n if let provider = enabled.first {\n return self.style(for: provider)\n }\n return .codex\n }\n\n var isStale: Bool {\n for provider in self.enabledProviders() where self.errors[provider] != nil {\n return true\n }\n return false\n }\n\n func enabledProviders() -> [UsageProvider] {\n // Use cached enablement to avoid repeated UserDefaults lookups in animation ticks.\n let enabled = self.settings.enabledProvidersOrdered(metadataByProvider: self.providerMetadata)\n return enabled.filter { self.isProviderAvailable($0) }\n }\n\n /// Enabled providers without availability filtering. Used for display (switcher, merge-icons).\n func enabledProvidersForDisplay() -> [UsageProvider] {\n self.settings.enabledProvidersOrdered(metadataByProvider: self.providerMetadata)\n }\n\n var statusChecksEnabled: Bool {\n self.settings.statusChecksEnabled\n }\n\n func metadata(for provider: UsageProvider) -> ProviderMetadata {\n self.providerMetadata[provider]!\n }\n\n private var codexBrowserCookieOrder: BrowserCookieImportOrder {\n self.metadata(for: .codex).browserCookieOrder ?? Browser.defaultImportOrder\n }\n\n func snapshot(for provider: UsageProvider) -> UsageSnapshot? {\n self.snapshots[provider]\n }\n\n func sourceLabel(for provider: UsageProvider) -> String {\n var label = self.lastSourceLabels[provider] ?? \"\"\n if label.isEmpty {\n let descriptor = ProviderDescriptorRegistry.descriptor(for: provider)\n let modes = descriptor.fetchPlan.sourceModes\n if modes.count == 1, let mode = modes.first {\n label = mode.rawValue\n } else {\n let context = ProviderSourceLabelContext(\n provider: provider,\n settings: self.settings,\n store: self,\n descriptor: descriptor)\n label = ProviderCatalog.implementation(for: provider)?\n .defaultSourceLabel(context: context)\n ?? \"auto\"\n }\n }\n\n let context = ProviderSourceLabelContext(\n provider: provider,\n settings: self.settings,\n store: self,\n descriptor: ProviderDescriptorRegistry.descriptor(for: provider))\n return ProviderCatalog.implementation(for: provider)?\n .decorateSourceLabel(context: context, baseLabel: label)\n ?? label\n }\n\n func fetchAttempts(for provider: UsageProvider) -> [ProviderFetchAttempt] {\n self.lastFetchAttempts[provider] ?? []\n }\n\n func style(for provider: UsageProvider) -> IconStyle {\n self.providerSpecs[provider]?.style ?? .codex\n }\n\n func isStale(provider: UsageProvider) -> Bool {\n self.errors[provider] != nil\n }\n\n func isEnabled(_ provider: UsageProvider) -> Bool {\n let enabled = self.settings.isProviderEnabledCached(\n provider: provider,\n metadataByProvider: self.providerMetadata)\n guard enabled else { return false }\n return self.isProviderAvailable(provider)\n }\n\n func isProviderAvailable(_ provider: UsageProvider) -> Bool {\n // Availability should mirror the effective fetch environment, including token-account overrides.\n // Otherwise providers (notably token-account-backed API providers) can fetch successfully but be\n // hidden from the menu because their credentials are not in ProcessInfo's environment.\n let environment = ProviderRegistry.makeEnvironment(\n base: ProcessInfo.processInfo.environment,\n provider: provider,\n settings: self.settings,\n tokenOverride: nil)\n let context = ProviderAvailabilityContext(\n provider: provider,\n settings: self.settings,\n environment: environment)\n return ProviderCatalog.implementation(for: provider)?\n .isAvailable(context: context)\n ?? true\n }\n\n func performRuntimeAction(_ action: ProviderRuntimeAction, for provider: UsageProvider) async {\n guard let runtime = self.providerRuntimes[provider] else { return }\n let context = ProviderRuntimeContext(provider: provider, settings: self.settings, store: self)\n await runtime.perform(action: action, context: context)\n }\n\n private func updateProviderRuntimes() {\n for (provider, runtime) in self.providerRuntimes {\n let context = ProviderRuntimeContext(provider: provider, settings: self.settings, store: self)\n if self.isEnabled(provider) {\n runtime.start(context: context)\n } else {\n runtime.stop(context: context)\n }\n runtime.settingsDidChange(context: context)\n }\n }\n\n func refresh(forceTokenUsage: Bool = false) async {\n guard !self.isRefreshing else { return }\n let refreshPhase: ProviderRefreshPhase = self.hasCompletedInitialRefresh ? .regular : .startup\n\n await ProviderRefreshContext.$current.withValue(refreshPhase) {\n self.isRefreshing = true\n defer {\n self.isRefreshing = false\n self.hasCompletedInitialRefresh = true\n }\n\n await withTaskGroup(of: Void.self) { group in\n for provider in UsageProvider.allCases {\n group.addTask { await self.refreshProvider(provider) }\n group.addTask { await self.refreshStatus(provider) }\n }\n group.addTask { await self.refreshCreditsIfNeeded() }\n }\n\n // Token-cost usage can be slow; run it outside the refresh group so we don't block menu updates.\n self.scheduleTokenRefresh(force: forceTokenUsage)\n\n // OpenAI web scrape depends on the current Codex account email (which can change after login/account\n // switch). Run this after Codex usage refresh so we don't accidentally scrape with stale credentials.\n await self.refreshOpenAIDashboardIfNeeded(force: forceTokenUsage)\n\n if self.openAIDashboardRequiresLogin {\n await self.refreshProvider(.codex)\n await self.refreshCreditsIfNeeded()\n }\n\n self.persistWidgetSnapshot(reason: \"refresh\")\n }\n }\n\n /// For demo/testing: drop the snapshot so the loading animation plays, then restore the last snapshot.\n func replayLoadingAnimation(duration: TimeInterval = 3) {\n let current = self.preferredSnapshot\n self.snapshots.removeAll()\n self.debugForceAnimation = true\n Task { @MainActor in\n try? await Task.sleep(for: .seconds(duration))\n if let current, let provider = self.enabledProviders().first {\n self.snapshots[provider] = current\n }\n self.debugForceAnimation = false\n }\n }\n\n // MARK: - Private\n\n private func bindSettings() {\n self.observeSettingsChanges()\n }\n\n private func startTimer() {\n self.timerTask?.cancel()\n guard let wait = self.settings.refreshFrequency.seconds else { return }\n\n // Background poller so the menu stays responsive; canceled when settings change or store deallocates.\n self.timerTask = Task.detached(priority: .utility) { [weak self] in\n while !Task.isCancelled {\n try? await Task.sleep(for: .seconds(wait))\n await self?.refresh()\n }\n }\n }\n\n private func startTokenTimer() {\n self.tokenTimerTask?.cancel()\n let wait = self.tokenFetchTTL\n self.tokenTimerTask = Task.detached(priority: .utility) { [weak self] in\n while !Task.isCancelled {\n try? await Task.sleep(for: .seconds(wait))\n await self?.scheduleTokenRefresh(force: false)\n }\n }\n }\n\n private func scheduleTokenRefresh(force: Bool) {\n if force {\n self.tokenRefreshSequenceTask?.cancel()\n self.tokenRefreshSequenceTask = nil\n } else if self.tokenRefreshSequenceTask != nil {\n return\n }\n\n self.tokenRefreshSequenceTask = Task(priority: .utility) { [weak self] in\n guard let self else { return }\n defer {\n Task { @MainActor [weak self] in\n self?.tokenRefreshSequenceTask = nil\n }\n }\n for provider in UsageProvider.allCases {\n if Task.isCancelled { break }\n await self.refreshTokenUsage(provider, force: force)\n }\n }\n }\n\n deinit {\n self.timerTask?.cancel()\n self.tokenTimerTask?.cancel()\n self.tokenRefreshSequenceTask?.cancel()\n }\n\n enum SessionQuotaWindowSource: String {\n case primary\n case copilotSecondaryFallback\n }\n\n private func sessionQuotaWindow(\n provider: UsageProvider,\n snapshot: UsageSnapshot) -> (window: RateWindow, source: SessionQuotaWindowSource)?\n {\n if let primary = snapshot.primary {\n return (primary, .primary)\n }\n if provider == .copilot, let secondary = snapshot.secondary {\n return (secondary, .copilotSecondaryFallback)\n }\n return nil\n }\n\n func handleSessionQuotaTransition(provider: UsageProvider, snapshot: UsageSnapshot) {\n // Session quota notifications are tied to the primary session window. Copilot free plans can\n // expose only chat quota, so allow Copilot to fall back to secondary for transition tracking.\n guard let sessionWindow = self.sessionQuotaWindow(provider: provider, snapshot: snapshot) else {\n self.lastKnownSessionRemaining.removeValue(forKey: provider)\n self.lastKnownSessionWindowSource.removeValue(forKey: provider)\n return\n }\n let currentRemaining = sessionWindow.window.remainingPercent\n let currentSource = sessionWindow.source\n let previousRemaining = self.lastKnownSessionRemaining[provider]\n let previousSource = self.lastKnownSessionWindowSource[provider]\n\n if let previousSource, previousSource != currentSource {\n let providerText = provider.rawValue\n self.sessionQuotaLogger.debug(\n \"session window source changed: provider=\\(providerText) prevSource=\\(previousSource.rawValue) \" +\n \"currSource=\\(currentSource.rawValue) curr=\\(currentRemaining)\")\n self.lastKnownSessionRemaining[provider] = currentRemaining\n self.lastKnownSessionWindowSource[provider] = currentSource\n return\n }\n\n defer {\n self.lastKnownSessionRemaining[provider] = currentRemaining\n self.lastKnownSessionWindowSource[provider] = currentSource\n }\n\n guard self.settings.sessionQuotaNotificationsEnabled else {\n if SessionQuotaNotificationLogic.isDepleted(currentRemaining) ||\n SessionQuotaNotificationLogic.isDepleted(previousRemaining)\n {\n let providerText = provider.rawValue\n let message =\n \"notifications disabled: provider=\\(providerText) \" +\n \"prev=\\(previousRemaining ?? -1) curr=\\(currentRemaining)\"\n self.sessionQuotaLogger.debug(message)\n }\n return\n }\n\n guard previousRemaining != nil else {\n if SessionQuotaNotificationLogic.isDepleted(currentRemaining) {\n let providerText = provider.rawValue\n let message = \"startup depleted: provider=\\(providerText) curr=\\(currentRemaining)\"\n self.sessionQuotaLogger.info(message)\n self.sessionQuotaNotifier.post(transition: .depleted, provider: provider, badge: nil)\n }\n return\n }\n\n let transition = SessionQuotaNotificationLogic.transition(\n previousRemaining: previousRemaining,\n currentRemaining: currentRemaining)\n guard transition != .none else {\n if SessionQuotaNotificationLogic.isDepleted(currentRemaining) ||\n SessionQuotaNotificationLogic.isDepleted(previousRemaining)\n {\n let providerText = provider.rawValue\n let message =\n \"no transition: provider=\\(providerText) \" +\n \"prev=\\(previousRemaining ?? -1) curr=\\(currentRemaining)\"\n self.sessionQuotaLogger.debug(message)\n }\n return\n }\n\n let providerText = provider.rawValue\n let transitionText = String(describing: transition)\n let message =\n \"transition \\(transitionText): provider=\\(providerText) \" +\n \"prev=\\(previousRemaining ?? -1) curr=\\(currentRemaining)\"\n self.sessionQuotaLogger.info(message)\n\n self.sessionQuotaNotifier.post(transition: transition, provider: provider, badge: nil)\n }\n\n private func refreshStatus(_ provider: UsageProvider) async {\n guard self.settings.statusChecksEnabled else { return }\n guard let meta = self.providerMetadata[provider] else { return }\n\n do {\n let status: ProviderStatus\n if let urlString = meta.statusPageURL, let baseURL = URL(string: urlString) {\n status = try await Self.fetchStatus(from: baseURL)\n } else if let productID = meta.statusWorkspaceProductID {\n status = try await Self.fetchWorkspaceStatus(productID: productID)\n } else {\n return\n }\n await MainActor.run { self.statuses[provider] = status }\n } catch {\n // Keep the previous status to avoid flapping when the API hiccups.\n await MainActor.run {\n if self.statuses[provider] == nil {\n self.statuses[provider] = ProviderStatus(\n indicator: .unknown,\n description: error.localizedDescription,\n updatedAt: nil)\n }\n }\n }\n }\n\n private func refreshCreditsIfNeeded() async {\n guard self.isEnabled(.codex) else { return }\n do {\n let credits = try await self.codexFetcher.loadLatestCredits(\n keepCLISessionsAlive: self.settings.debugKeepCLISessionsAlive)\n await MainActor.run {\n self.credits = credits\n self.lastCreditsError = nil\n self.lastCreditsSnapshot = credits\n self.creditsFailureStreak = 0\n }\n } catch {\n let message = error.localizedDescription\n if message.localizedCaseInsensitiveContains(\"data not available yet\") {\n await MainActor.run {\n if let cached = self.lastCreditsSnapshot {\n self.credits = cached\n self.lastCreditsError = nil\n } else {\n self.credits = nil\n self.lastCreditsError = \"Codex credits are still loading; will retry shortly.\"\n }\n }\n return\n }\n\n await MainActor.run {\n self.creditsFailureStreak += 1\n if let cached = self.lastCreditsSnapshot {\n self.credits = cached\n let stamp = cached.updatedAt.formatted(date: .abbreviated, time: .shortened)\n self.lastCreditsError =\n \"Last Codex credits refresh failed: \\(message). Cached values from \\(stamp).\"\n } else {\n self.lastCreditsError = message\n self.credits = nil\n }\n }\n }\n }\n}\n\nextension UsageStore {\n private static let openAIWebRefreshMultiplier: TimeInterval = 5\n private static let openAIWebPrimaryFetchTimeout: TimeInterval = 15\n private static let openAIWebRetryFetchTimeout: TimeInterval = 8\n\n private func openAIWebRefreshIntervalSeconds() -> TimeInterval {\n let base = max(self.settings.refreshFrequency.seconds ?? 0, 120)\n return base * Self.openAIWebRefreshMultiplier\n }\n\n func requestOpenAIDashboardRefreshIfStale(reason: String) {\n guard self.isEnabled(.codex), self.settings.codexCookieSource.isEnabled else { return }\n let now = Date()\n let refreshInterval = self.openAIWebRefreshIntervalSeconds()\n let lastUpdatedAt = self.openAIDashboard?.updatedAt ?? self.lastOpenAIDashboardSnapshot?.updatedAt\n if let lastUpdatedAt, now.timeIntervalSince(lastUpdatedAt) < refreshInterval { return }\n let stamp = now.formatted(date: .abbreviated, time: .shortened)\n self.logOpenAIWeb(\"[\\(stamp)] OpenAI web refresh request: \\(reason)\")\n Task { await self.refreshOpenAIDashboardIfNeeded(force: true) }\n }\n\n private func applyOpenAIDashboard(_ dash: OpenAIDashboardSnapshot, targetEmail: String?) async {\n await MainActor.run {\n self.openAIDashboard = dash\n self.lastOpenAIDashboardError = nil\n self.lastOpenAIDashboardSnapshot = dash\n self.openAIDashboardRequiresLogin = false\n // Only fill gaps; OAuth/CLI remain the primary sources for usage + credits.\n if self.snapshots[.codex] == nil,\n let usage = dash.toUsageSnapshot(provider: .codex, accountEmail: targetEmail)\n {\n self.snapshots[.codex] = usage\n self.errors[.codex] = nil\n self.failureGates[.codex]?.recordSuccess()\n self.lastSourceLabels[.codex] = \"openai-web\"\n }\n if self.credits == nil, let credits = dash.toCreditsSnapshot() {\n self.credits = credits\n self.lastCreditsSnapshot = credits\n self.lastCreditsError = nil\n self.creditsFailureStreak = 0\n }\n }\n\n if let email = targetEmail, !email.isEmpty {\n OpenAIDashboardCacheStore.save(OpenAIDashboardCache(accountEmail: email, snapshot: dash))\n }\n self.backfillCodexHistoricalFromDashboardIfNeeded(dash)\n }\n\n private func applyOpenAIDashboardFailure(message: String) async {\n await MainActor.run {\n if let cached = self.lastOpenAIDashboardSnapshot {\n self.openAIDashboard = cached\n let stamp = cached.updatedAt.formatted(date: .abbreviated, time: .shortened)\n self.lastOpenAIDashboardError =\n \"Last OpenAI dashboard refresh failed: \\(message). Cached values from \\(stamp).\"\n } else {\n self.lastOpenAIDashboardError = message\n self.openAIDashboard = nil\n }\n }\n }\n\n private func refreshOpenAIDashboardIfNeeded(force: Bool = false) async {\n guard self.isEnabled(.codex), self.settings.codexCookieSource.isEnabled else {\n self.resetOpenAIWebState()\n return\n }\n\n let targetEmail = self.codexAccountEmailForOpenAIDashboard()\n self.handleOpenAIWebTargetEmailChangeIfNeeded(targetEmail: targetEmail)\n\n let now = Date()\n let minInterval = self.openAIWebRefreshIntervalSeconds()\n if !force,\n !self.openAIWebAccountDidChange,\n self.lastOpenAIDashboardError == nil,\n let snapshot = self.lastOpenAIDashboardSnapshot,\n now.timeIntervalSince(snapshot.updatedAt) < minInterval\n {\n return\n }\n\n if self.openAIWebDebugLines.isEmpty {\n self.resetOpenAIWebDebugLog(context: \"refresh\")\n } else {\n let stamp = Date().formatted(date: .abbreviated, time: .shortened)\n self.logOpenAIWeb(\"[\\(stamp)] OpenAI web refresh start\")\n }\n let log: (String) -> Void = { [weak self] line in\n guard let self else { return }\n self.logOpenAIWeb(line)\n }\n\n do {\n let normalized = targetEmail?\n .trimmingCharacters(in: .whitespacesAndNewlines)\n .lowercased()\n var effectiveEmail = targetEmail\n\n // Use a per-email persistent `WKWebsiteDataStore` so multiple dashboard sessions can coexist.\n // Strategy:\n // - Try the existing per-email WebKit cookie store first (fast; avoids Keychain prompts).\n // - On login-required or account mismatch, import cookies from the configured browser order and retry once.\n if self.openAIWebAccountDidChange, let targetEmail, !targetEmail.isEmpty {\n // On account switches, proactively re-import cookies so we don't show stale data from the previous\n // user.\n if let imported = await self.importOpenAIDashboardCookiesIfNeeded(\n targetEmail: targetEmail,\n force: true)\n {\n effectiveEmail = imported\n }\n self.openAIWebAccountDidChange = false\n }\n\n var dash = try await OpenAIDashboardFetcher().loadLatestDashboard(\n accountEmail: effectiveEmail,\n logger: log,\n debugDumpHTML: false,\n timeout: Self.openAIWebPrimaryFetchTimeout)\n\n if self.dashboardEmailMismatch(expected: normalized, actual: dash.signedInEmail) {\n if let imported = await self.importOpenAIDashboardCookiesIfNeeded(\n targetEmail: targetEmail,\n force: true)\n {\n effectiveEmail = imported\n }\n dash = try await OpenAIDashboardFetcher().loadLatestDashboard(\n accountEmail: effectiveEmail,\n logger: log,\n debugDumpHTML: false,\n timeout: Self.openAIWebRetryFetchTimeout)\n }\n\n if self.dashboardEmailMismatch(expected: normalized, actual: dash.signedInEmail) {\n let signedIn = dash.signedInEmail?.trimmingCharacters(in: .whitespacesAndNewlines) ?? \"unknown\"\n await MainActor.run {\n self.openAIDashboard = nil\n self.lastOpenAIDashboardError = [\n \"OpenAI dashboard signed in as \\(signedIn), but Codex uses \\(normalized ?? \"unknown\").\",\n \"Switch accounts in your browser and update OpenAI cookies in Providers → Codex.\",\n ].joined(separator: \" \")\n self.openAIDashboardRequiresLogin = true\n }\n return\n }\n\n await self.applyOpenAIDashboard(dash, targetEmail: effectiveEmail)\n } catch let OpenAIDashboardFetcher.FetchError.noDashboardData(body) {\n // Often indicates a missing/stale session without an obvious login prompt. Retry once after\n // importing cookies from the user's browser.\n let targetEmail = self.codexAccountEmailForOpenAIDashboard()\n var effectiveEmail = targetEmail\n if let imported = await self.importOpenAIDashboardCookiesIfNeeded(targetEmail: targetEmail, force: true) {\n effectiveEmail = imported\n }\n do {\n let dash = try await OpenAIDashboardFetcher().loadLatestDashboard(\n accountEmail: effectiveEmail,\n logger: log,\n debugDumpHTML: true,\n timeout: Self.openAIWebRetryFetchTimeout)\n await self.applyOpenAIDashboard(dash, targetEmail: effectiveEmail)\n } catch let OpenAIDashboardFetcher.FetchError.noDashboardData(retryBody) {\n let finalBody = retryBody.isEmpty ? body : retryBody\n let message = self.openAIDashboardFriendlyError(\n body: finalBody,\n targetEmail: targetEmail,\n cookieImportStatus: self.openAIDashboardCookieImportStatus)\n ?? OpenAIDashboardFetcher.FetchError.noDashboardData(body: finalBody).localizedDescription\n await self.applyOpenAIDashboardFailure(message: message)\n } catch {\n await self.applyOpenAIDashboardFailure(message: error.localizedDescription)\n }\n } catch OpenAIDashboardFetcher.FetchError.loginRequired {\n let targetEmail = self.codexAccountEmailForOpenAIDashboard()\n var effectiveEmail = targetEmail\n if let imported = await self.importOpenAIDashboardCookiesIfNeeded(targetEmail: targetEmail, force: true) {\n effectiveEmail = imported\n }\n do {\n let dash = try await OpenAIDashboardFetcher().loadLatestDashboard(\n accountEmail: effectiveEmail,\n logger: log,\n debugDumpHTML: true,\n timeout: Self.openAIWebRetryFetchTimeout)\n await self.applyOpenAIDashboard(dash, targetEmail: effectiveEmail)\n } catch OpenAIDashboardFetcher.FetchError.loginRequired {\n await MainActor.run {\n self.lastOpenAIDashboardError = [\n \"OpenAI web access requires a signed-in chatgpt.com session.\",\n \"Sign in using \\(self.codexBrowserCookieOrder.loginHint), \" +\n \"then update OpenAI cookies in Providers → Codex.\",\n ].joined(separator: \" \")\n self.openAIDashboard = self.lastOpenAIDashboardSnapshot\n self.openAIDashboardRequiresLogin = true\n }\n } catch {\n await self.applyOpenAIDashboardFailure(message: error.localizedDescription)\n }\n } catch {\n await self.applyOpenAIDashboardFailure(message: error.localizedDescription)\n }\n }\n\n // MARK: - OpenAI web account switching\n\n /// Detect Codex account email changes and clear stale OpenAI web state so the UI can't show the wrong user.\n /// This does not delete other per-email WebKit cookie stores (we keep multiple accounts around).\n func handleOpenAIWebTargetEmailChangeIfNeeded(targetEmail: String?) {\n let normalized = targetEmail?\n .trimmingCharacters(in: .whitespacesAndNewlines)\n .lowercased()\n\n guard let normalized, !normalized.isEmpty else { return }\n\n let previous = self.lastOpenAIDashboardTargetEmail\n self.lastOpenAIDashboardTargetEmail = normalized\n\n if let previous,\n !previous.isEmpty,\n previous != normalized\n {\n let stamp = Date().formatted(date: .abbreviated, time: .shortened)\n self.logOpenAIWeb(\n \"[\\(stamp)] Codex account changed: \\(previous) → \\(normalized); \" +\n \"clearing OpenAI web snapshot\")\n self.openAIWebAccountDidChange = true\n self.openAIDashboard = nil\n self.lastOpenAIDashboardSnapshot = nil\n self.lastOpenAIDashboardError = nil\n self.openAIDashboardRequiresLogin = true\n self.openAIDashboardCookieImportStatus = \"Codex account changed; importing browser cookies…\"\n self.lastOpenAIDashboardCookieImportAttemptAt = nil\n self.lastOpenAIDashboardCookieImportEmail = nil\n }\n }\n\n func importOpenAIDashboardBrowserCookiesNow() async {\n self.resetOpenAIWebDebugLog(context: \"manual import\")\n let targetEmail = self.codexAccountEmailForOpenAIDashboard()\n _ = await self.importOpenAIDashboardCookiesIfNeeded(targetEmail: targetEmail, force: true)\n await self.refreshOpenAIDashboardIfNeeded(force: true)\n }\n\n private func importOpenAIDashboardCookiesIfNeeded(targetEmail: String?, force: Bool) async -> String? {\n let normalizedTarget = targetEmail?.trimmingCharacters(in: .whitespacesAndNewlines)\n let allowAnyAccount = normalizedTarget == nil || normalizedTarget?.isEmpty == true\n let cookieSource = self.settings.codexCookieSource\n\n let now = Date()\n let lastEmail = self.lastOpenAIDashboardCookieImportEmail\n let lastAttempt = self.lastOpenAIDashboardCookieImportAttemptAt ?? .distantPast\n\n let shouldAttempt: Bool = if force {\n true\n } else {\n if allowAnyAccount {\n now.timeIntervalSince(lastAttempt) > 300\n } else {\n self.openAIDashboardRequiresLogin &&\n (\n lastEmail?.lowercased() != normalizedTarget?.lowercased() || now\n .timeIntervalSince(lastAttempt) > 300)\n }\n }\n\n guard shouldAttempt else { return normalizedTarget }\n self.lastOpenAIDashboardCookieImportEmail = normalizedTarget\n self.lastOpenAIDashboardCookieImportAttemptAt = now\n\n let stamp = now.formatted(date: .abbreviated, time: .shortened)\n let targetLabel = normalizedTarget ?? \"unknown\"\n self.logOpenAIWeb(\"[\\(stamp)] import start (target=\\(targetLabel))\")\n\n do {\n let log: (String) -> Void = { [weak self] message in\n guard let self else { return }\n self.logOpenAIWeb(message)\n }\n\n let importer = OpenAIDashboardBrowserCookieImporter(browserDetection: self.browserDetection)\n let result: OpenAIDashboardBrowserCookieImporter.ImportResult\n switch cookieSource {\n case .manual:\n self.settings.ensureCodexCookieLoaded()\n let manualHeader = self.settings.codexCookieHeader\n guard CookieHeaderNormalizer.normalize(manualHeader) != nil else {\n throw OpenAIDashboardBrowserCookieImporter.ImportError.manualCookieHeaderInvalid\n }\n result = try await importer.importManualCookies(\n cookieHeader: manualHeader,\n intoAccountEmail: normalizedTarget,\n allowAnyAccount: allowAnyAccount,\n logger: log)\n case .auto:\n result = try await importer.importBestCookies(\n intoAccountEmail: normalizedTarget,\n allowAnyAccount: allowAnyAccount,\n logger: log)\n case .off:\n result = OpenAIDashboardBrowserCookieImporter.ImportResult(\n sourceLabel: \"Off\",\n cookieCount: 0,\n signedInEmail: normalizedTarget,\n matchesCodexEmail: true)\n }\n let effectiveEmail = result.signedInEmail?\n .trimmingCharacters(in: .whitespacesAndNewlines)\n .isEmpty == false\n ? result.signedInEmail?.trimmingCharacters(in: .whitespacesAndNewlines)\n : normalizedTarget\n self.lastOpenAIDashboardCookieImportEmail = effectiveEmail ?? normalizedTarget\n await MainActor.run {\n let signed = result.signedInEmail?.trimmingCharacters(in: .whitespacesAndNewlines)\n let matchText = result.matchesCodexEmail ? \"matches Codex\" : \"does not match Codex\"\n let sourceLabel = switch cookieSource {\n case .manual:\n \"Manual cookie header\"\n case .auto:\n \"\\(result.sourceLabel) cookies\"\n case .off:\n \"OpenAI cookies disabled\"\n }\n if let signed, !signed.isEmpty {\n self.openAIDashboardCookieImportStatus =\n allowAnyAccount\n ? [\n \"Using \\(sourceLabel) (\\(result.cookieCount)).\",\n \"Signed in as \\(signed).\",\n ].joined(separator: \" \")\n : [\n \"Using \\(sourceLabel) (\\(result.cookieCount)).\",\n \"Signed in as \\(signed) (\\(matchText)).\",\n ].joined(separator: \" \")\n } else {\n self.openAIDashboardCookieImportStatus =\n \"Using \\(sourceLabel) (\\(result.cookieCount)).\"\n }\n }\n return effectiveEmail\n } catch let err as OpenAIDashboardBrowserCookieImporter.ImportError {\n switch err {\n case let .noMatchingAccount(found):\n let foundText: String = if found.isEmpty {\n \"no signed-in session detected in \\(self.codexBrowserCookieOrder.loginHint)\"\n } else {\n found\n .sorted { lhs, rhs in\n if lhs.sourceLabel == rhs.sourceLabel { return lhs.email < rhs.email }\n return lhs.sourceLabel < rhs.sourceLabel\n }\n .map { \"\\($0.sourceLabel): \\($0.email)\" }\n .joined(separator: \" • \")\n }\n self.logOpenAIWeb(\"[\\(stamp)] import mismatch: \\(foundText)\")\n await MainActor.run {\n self.openAIDashboardCookieImportStatus = allowAnyAccount\n ? [\n \"No signed-in OpenAI web session found.\",\n \"Found \\(foundText).\",\n ].joined(separator: \" \")\n : [\n \"Browser cookies do not match Codex account (\\(normalizedTarget ?? \"unknown\")).\",\n \"Found \\(foundText).\",\n ].joined(separator: \" \")\n // Treat mismatch like \"not logged in\" for the current Codex account.\n self.openAIDashboardRequiresLogin = true\n self.openAIDashboard = nil\n }\n case .noCookiesFound,\n .browserAccessDenied,\n .dashboardStillRequiresLogin,\n .manualCookieHeaderInvalid:\n self.logOpenAIWeb(\"[\\(stamp)] import failed: \\(err.localizedDescription)\")\n await MainActor.run {\n self.openAIDashboardCookieImportStatus =\n \"OpenAI cookie import failed: \\(err.localizedDescription)\"\n self.openAIDashboardRequiresLogin = true\n }\n }\n } catch {\n self.logOpenAIWeb(\"[\\(stamp)] import failed: \\(error.localizedDescription)\")\n await MainActor.run {\n self.openAIDashboardCookieImportStatus =\n \"Browser cookie import failed: \\(error.localizedDescription)\"\n }\n }\n return nil\n }\n\n private func resetOpenAIWebDebugLog(context: String) {\n let stamp = Date().formatted(date: .abbreviated, time: .shortened)\n self.openAIWebDebugLines.removeAll(keepingCapacity: true)\n self.openAIDashboardCookieImportDebugLog = nil\n self.logOpenAIWeb(\"[\\(stamp)] OpenAI web \\(context) start\")\n }\n\n private func logOpenAIWeb(_ message: String) {\n let safeMessage = LogRedactor.redact(message)\n self.openAIWebLogger.debug(safeMessage)\n self.openAIWebDebugLines.append(safeMessage)\n if self.openAIWebDebugLines.count > 240 {\n self.openAIWebDebugLines.removeFirst(self.openAIWebDebugLines.count - 240)\n }\n self.openAIDashboardCookieImportDebugLog = self.openAIWebDebugLines.joined(separator: \"\\n\")\n }\n\n func resetOpenAIWebState() {\n self.openAIDashboard = nil\n self.lastOpenAIDashboardError = nil\n self.lastOpenAIDashboardSnapshot = nil\n self.lastOpenAIDashboardTargetEmail = nil\n self.openAIDashboardRequiresLogin = false\n self.openAIDashboardCookieImportStatus = nil\n self.openAIDashboardCookieImportDebugLog = nil\n self.lastOpenAIDashboardCookieImportAttemptAt = nil\n self.lastOpenAIDashboardCookieImportEmail = nil\n }\n\n private func dashboardEmailMismatch(expected: String?, actual: String?) -> Bool {\n guard let expected, !expected.isEmpty else { return false }\n guard let raw = actual?.trimmingCharacters(in: .whitespacesAndNewlines), !raw.isEmpty else { return false }\n return raw.lowercased() != expected.lowercased()\n }\n\n func codexAccountEmailForOpenAIDashboard() -> String? {\n let direct = self.snapshots[.codex]?.accountEmail(for: .codex)?\n .trimmingCharacters(in: .whitespacesAndNewlines)\n if let direct, !direct.isEmpty { return direct }\n let fallback = self.codexFetcher.loadAccountInfo().email?.trimmingCharacters(in: .whitespacesAndNewlines)\n if let fallback, !fallback.isEmpty { return fallback }\n let cached = self.openAIDashboard?.signedInEmail?.trimmingCharacters(in: .whitespacesAndNewlines)\n if let cached, !cached.isEmpty { return cached }\n let imported = self.lastOpenAIDashboardCookieImportEmail?.trimmingCharacters(in: .whitespacesAndNewlines)\n if let imported, !imported.isEmpty { return imported }\n return nil\n }\n}\n\nextension UsageStore {\n func debugDumpClaude() async {\n let fetcher = ClaudeUsageFetcher(\n browserDetection: self.browserDetection,\n keepCLISessionsAlive: self.settings.debugKeepCLISessionsAlive)\n let output = await fetcher.debugRawProbe(model: \"sonnet\")\n let url = URL(fileURLWithPath: NSTemporaryDirectory()).appendingPathComponent(\"codexbar-claude-probe.txt\")\n try? output.write(to: url, atomically: true, encoding: .utf8)\n await MainActor.run {\n let snippet = String(output.prefix(180)).replacingOccurrences(of: \"\\n\", with: \" \")\n self.errors[.claude] = \"[Claude] \\(snippet) (saved: \\(url.path))\"\n NSWorkspace.shared.open(url)\n }\n }\n\n func dumpLog(toFileFor provider: UsageProvider) async -> URL? {\n let text = await self.debugLog(for: provider)\n let filename = \"codexbar-\\(provider.rawValue)-probe.txt\"\n let url = URL(fileURLWithPath: NSTemporaryDirectory()).appendingPathComponent(filename)\n do {\n try text.write(to: url, atomically: true, encoding: .utf8)\n _ = await MainActor.run { NSWorkspace.shared.open(url) }\n return url\n } catch {\n await MainActor.run {\n self.errors[provider] = \"Failed to save log: \\(error.localizedDescription)\"\n }\n return nil\n }\n }\n\n func debugAugmentDump() async -> String {\n await AugmentStatusProbe.latestDumps()\n }\n\n func debugLog(for provider: UsageProvider) async -> String {\n if let cached = self.probeLogs[provider], !cached.isEmpty {\n return cached\n }\n\n let claudeWebExtrasEnabled = self.settings.claudeWebExtrasEnabled\n let claudeUsageDataSource = self.settings.claudeUsageDataSource\n let claudeCookieSource = self.settings.claudeCookieSource\n let claudeCookieHeader = self.settings.claudeCookieHeader\n let claudeDebugConfiguration: ClaudeDebugLogConfiguration? = if provider == .claude {\n await self.makeClaudeDebugConfiguration(\n fallbackUsageDataSource: claudeUsageDataSource,\n fallbackWebExtrasEnabled: claudeWebExtrasEnabled,\n fallbackCookieSource: claudeCookieSource,\n fallbackCookieHeader: claudeCookieHeader)\n } else {\n nil\n }\n let cursorCookieSource = self.settings.cursorCookieSource\n let cursorCookieHeader = self.settings.cursorCookieHeader\n let ampCookieSource = self.settings.ampCookieSource\n let ampCookieHeader = self.settings.ampCookieHeader\n let ollamaCookieSource = self.settings.ollamaCookieSource\n let ollamaCookieHeader = self.settings.ollamaCookieHeader\n let processEnvironment = ProcessInfo.processInfo.environment\n let openRouterConfigToken = self.settings.providerConfig(for: .openrouter)?.sanitizedAPIKey\n let openRouterHasConfigToken = !(openRouterConfigToken?.trimmingCharacters(in: .whitespacesAndNewlines)\n .isEmpty ?? true)\n let openRouterHasEnvToken = OpenRouterSettingsReader.apiToken(environment: processEnvironment) != nil\n let openRouterEnvironment = ProviderConfigEnvironment.applyAPIKeyOverride(\n base: processEnvironment,\n provider: .openrouter,\n config: self.settings.providerConfig(for: .openrouter))\n let codexFetcher = self.codexFetcher\n let browserDetection = self.browserDetection\n let claudeDebugExecutionContext = self.currentClaudeDebugExecutionContext()\n let text = await Task.detached(priority: .utility) { () -> String in\n let unimplementedDebugLogMessages: [UsageProvider: String] = [\n .gemini: \"Gemini debug log not yet implemented\",\n .antigravity: \"Antigravity debug log not yet implemented\",\n .opencode: \"OpenCode debug log not yet implemented\",\n .alibaba: \"Alibaba Coding Plan debug log not yet implemented\",\n .factory: \"Droid debug log not yet implemented\",\n .copilot: \"Copilot debug log not yet implemented\",\n .vertexai: \"Vertex AI debug log not yet implemented\",\n .kilo: \"Kilo debug log not yet implemented\",\n .kiro: \"Kiro debug log not yet implemented\",\n .kimi: \"Kimi debug log not yet implemented\",\n .kimik2: \"Kimi K2 debug log not yet implemented\",\n .jetbrains: \"JetBrains AI debug log not yet implemented\",\n ]\n let buildText = {\n switch provider {\n case .codex:\n return await codexFetcher.debugRawRateLimits()\n case .claude:\n guard let claudeDebugConfiguration else {\n return \"Claude debug log configuration unavailable\"\n }\n return await claudeDebugExecutionContext.apply {\n await Self.debugClaudeLog(\n browserDetection: browserDetection,\n configuration: claudeDebugConfiguration)\n }\n case .zai:\n let resolution = ProviderTokenResolver.zaiResolution()\n let hasAny = resolution != nil\n let source = resolution?.source.rawValue ?? \"none\"\n return \"Z_AI_API_KEY=\\(hasAny ? \"present\" : \"missing\") source=\\(source)\"\n case .synthetic:\n let resolution = ProviderTokenResolver.syntheticResolution()\n let hasAny = resolution != nil\n let source = resolution?.source.rawValue ?? \"none\"\n return \"SYNTHETIC_API_KEY=\\(hasAny ? \"present\" : \"missing\") source=\\(source)\"\n case .cursor:\n return await Self.debugCursorLog(\n browserDetection: browserDetection,\n cursorCookieSource: cursorCookieSource,\n cursorCookieHeader: cursorCookieHeader)\n case .minimax:\n let tokenResolution = ProviderTokenResolver.minimaxTokenResolution()\n let cookieResolution = ProviderTokenResolver.minimaxCookieResolution()\n let tokenSource = tokenResolution?.source.rawValue ?? \"none\"\n let cookieSource = cookieResolution?.source.rawValue ?? \"none\"\n return \"MINIMAX_API_KEY=\\(tokenResolution == nil ? \"missing\" : \"present\") \" +\n \"source=\\(tokenSource) MINIMAX_COOKIE=\\(cookieResolution == nil ? \"missing\" : \"present\") \" +\n \"source=\\(cookieSource)\"\n case .alibaba:\n let resolution = ProviderTokenResolver.alibabaTokenResolution()\n let hasAny = resolution != nil\n let source = resolution?.source.rawValue ?? \"none\"\n return \"ALIBABA_CODING_PLAN_API_KEY=\\(hasAny ? \"present\" : \"missing\") source=\\(source)\"\n case .augment:\n return await Self.debugAugmentLog()\n case .amp:\n return await Self.debugAmpLog(\n browserDetection: browserDetection,\n ampCookieSource: ampCookieSource,\n ampCookieHeader: ampCookieHeader)\n case .ollama:\n return await Self.debugOllamaLog(\n browserDetection: browserDetection,\n ollamaCookieSource: ollamaCookieSource,\n ollamaCookieHeader: ollamaCookieHeader)\n case .openrouter:\n let resolution = ProviderTokenResolver.openRouterResolution(environment: openRouterEnvironment)\n let hasAny = resolution != nil\n let source: String = if resolution == nil {\n \"none\"\n } else if openRouterHasConfigToken, openRouterHasEnvToken {\n \"settings-config (overrides env)\"\n } else if openRouterHasConfigToken {\n \"settings-config\"\n } else {\n resolution?.source.rawValue ?? \"environment\"\n }\n return \"OPENROUTER_API_KEY=\\(hasAny ? \"present\" : \"missing\") source=\\(source)\"\n case .warp:\n let resolution = ProviderTokenResolver.warpResolution()\n let hasAny = resolution != nil\n let source = resolution?.source.rawValue ?? \"none\"\n return \"WARP_API_KEY=\\(hasAny ? \"present\" : \"missing\") source=\\(source)\"\n case .gemini, .antigravity, .opencode, .factory, .copilot, .vertexai, .kilo, .kiro, .kimi,\n .kimik2, .jetbrains:\n return unimplementedDebugLogMessages[provider] ?? \"Debug log not yet implemented\"\n }\n }\n return await claudeDebugExecutionContext.apply {\n await buildText()\n }\n }.value\n self.probeLogs[provider] = text\n return text\n }\n\n private func makeClaudeDebugConfiguration(\n fallbackUsageDataSource: ClaudeUsageDataSource,\n fallbackWebExtrasEnabled: Bool,\n fallbackCookieSource: ProviderCookieSource,\n fallbackCookieHeader: String) async -> ClaudeDebugLogConfiguration\n {\n await MainActor.run {\n let sourceMode = self.sourceMode(for: .claude)\n let snapshot = ProviderRegistry.makeSettingsSnapshot(settings: self.settings, tokenOverride: nil)\n let environment = ProviderRegistry.makeEnvironment(\n base: ProcessInfo.processInfo.environment,\n provider: .claude,\n settings: self.settings,\n tokenOverride: nil)\n let claudeSettings = snapshot.claude ?? ProviderSettingsSnapshot.ClaudeProviderSettings(\n usageDataSource: fallbackUsageDataSource,\n webExtrasEnabled: fallbackWebExtrasEnabled,\n cookieSource: fallbackCookieSource,\n manualCookieHeader: fallbackCookieHeader)\n return ClaudeDebugLogConfiguration(\n runtime: CodexBarCore.ProviderRuntime.app,\n sourceMode: sourceMode,\n environment: environment,\n webExtrasEnabled: claudeSettings.webExtrasEnabled,\n usageDataSource: claudeSettings.usageDataSource,\n cookieSource: claudeSettings.cookieSource,\n cookieHeader: claudeSettings.manualCookieHeader ?? \"\",\n keepCLISessionsAlive: snapshot.debugKeepCLISessionsAlive)\n }\n }\n\n private struct ClaudeDebugExecutionContext {\n let interaction: ProviderInteraction\n let refreshPhase: ProviderRefreshPhase\n #if DEBUG\n let keychainServiceOverride: String?\n let credentialsURLOverride: URL?\n let testingOverrides: ClaudeOAuthCredentialsStore.TestingOverridesSnapshot\n let keychainDeniedUntilStoreOverride: ClaudeOAuthKeychainAccessGate.DeniedUntilStore?\n let keychainPromptModeOverride: ClaudeOAuthKeychainPromptMode?\n let keychainReadStrategyOverride: ClaudeOAuthKeychainReadStrategy?\n let cliPathOverride: String?\n let statusFetchOverride: ClaudeStatusProbe.FetchOverride?\n #endif\n\n func apply(_ operation: () async -> T) async -> T {\n await ProviderInteractionContext.$current.withValue(self.interaction) {\n await ProviderRefreshContext.$current.withValue(self.refreshPhase) {\n #if DEBUG\n return await KeychainCacheStore.withServiceOverrideForTesting(self.keychainServiceOverride) {\n await ClaudeOAuthCredentialsStore\n .withCredentialsURLOverrideForTesting(self.credentialsURLOverride) {\n await ClaudeOAuthCredentialsStore\n .withTestingOverridesSnapshotForTask(self.testingOverrides) {\n await ClaudeOAuthKeychainAccessGate\n .withDeniedUntilStoreOverrideForTesting(self\n .keychainDeniedUntilStoreOverride)\n {\n await ClaudeOAuthKeychainPromptPreference\n .withTaskOverrideForTesting(self.keychainPromptModeOverride) {\n await ClaudeOAuthKeychainReadStrategyPreference\n .withTaskOverrideForTesting(self\n .keychainReadStrategyOverride)\n {\n await ClaudeCLIResolver\n .withResolvedBinaryPathOverrideForTesting(self\n .cliPathOverride)\n {\n await ClaudeStatusProbe\n .withFetchOverrideForTesting(self\n .statusFetchOverride)\n {\n await operation()\n }\n }\n }\n }\n }\n }\n }\n }\n #else\n return await operation()\n #endif\n }\n }\n }\n }\n\n private func currentClaudeDebugExecutionContext() -> ClaudeDebugExecutionContext {\n #if DEBUG\n ClaudeDebugExecutionContext(\n interaction: ProviderInteractionContext.current,\n refreshPhase: ProviderRefreshContext.current,\n keychainServiceOverride: KeychainCacheStore.currentServiceOverrideForTesting,\n credentialsURLOverride: ClaudeOAuthCredentialsStore.currentCredentialsURLOverrideForTesting,\n testingOverrides: ClaudeOAuthCredentialsStore.currentTestingOverridesSnapshotForTask,\n keychainDeniedUntilStoreOverride: ClaudeOAuthKeychainAccessGate.currentDeniedUntilStoreOverrideForTesting,\n keychainPromptModeOverride: ClaudeOAuthKeychainPromptPreference.currentTaskOverrideForTesting,\n keychainReadStrategyOverride: ClaudeOAuthKeychainReadStrategyPreference.currentTaskOverrideForTesting,\n cliPathOverride: ClaudeCLIResolver.currentResolvedBinaryPathOverrideForTesting,\n statusFetchOverride: ClaudeStatusProbe.currentFetchOverrideForTesting)\n #else\n ClaudeDebugExecutionContext(\n interaction: ProviderInteractionContext.current,\n refreshPhase: ProviderRefreshContext.current)\n #endif\n }\n\n private static func debugCursorLog(\n browserDetection: BrowserDetection,\n cursorCookieSource: ProviderCookieSource,\n cursorCookieHeader: String) async -> String\n {\n await runWithTimeout(seconds: 15) {\n var lines: [String] = []\n\n do {\n let probe = CursorStatusProbe(browserDetection: browserDetection)\n let snapshot: CursorStatusSnapshot = if cursorCookieSource == .manual,\n let normalizedHeader = CookieHeaderNormalizer\n .normalize(cursorCookieHeader)\n {\n try await probe.fetchWithManualCookies(normalizedHeader)\n } else {\n try await probe.fetch { msg in lines.append(\"[cursor-cookie] \\(msg)\") }\n }\n\n lines.append(\"\")\n lines.append(\"Cursor Status Summary:\")\n lines.append(\"membershipType=\\(snapshot.membershipType ?? \"nil\")\")\n lines.append(\"accountEmail=\\(snapshot.accountEmail ?? \"nil\")\")\n lines.append(\"planPercentUsed=\\(snapshot.planPercentUsed)%\")\n lines.append(\"planUsedUSD=$\\(snapshot.planUsedUSD)\")\n lines.append(\"planLimitUSD=$\\(snapshot.planLimitUSD)\")\n lines.append(\"onDemandUsedUSD=$\\(snapshot.onDemandUsedUSD)\")\n lines.append(\"onDemandLimitUSD=\\(snapshot.onDemandLimitUSD.map { \"$\\($0)\" } ?? \"nil\")\")\n if let teamUsed = snapshot.teamOnDemandUsedUSD {\n lines.append(\"teamOnDemandUsedUSD=$\\(teamUsed)\")\n }\n if let teamLimit = snapshot.teamOnDemandLimitUSD {\n lines.append(\"teamOnDemandLimitUSD=$\\(teamLimit)\")\n }\n lines.append(\"billingCycleEnd=\\(snapshot.billingCycleEnd?.description ?? \"nil\")\")\n\n if let rawJSON = snapshot.rawJSON {\n lines.append(\"\")\n lines.append(\"Raw API Response:\")\n lines.append(rawJSON)\n }\n\n return lines.joined(separator: \"\\n\")\n } catch {\n lines.append(\"\")\n lines.append(\"Cursor probe failed: \\(error.localizedDescription)\")\n return lines.joined(separator: \"\\n\")\n }\n }\n }\n\n private static func debugAugmentLog() async -> String {\n await runWithTimeout(seconds: 15) {\n let probe = AugmentStatusProbe()\n return await probe.debugRawProbe()\n }\n }\n\n private static func debugAmpLog(\n browserDetection: BrowserDetection,\n ampCookieSource: ProviderCookieSource,\n ampCookieHeader: String) async -> String\n {\n await runWithTimeout(seconds: 15) {\n let fetcher = AmpUsageFetcher(browserDetection: browserDetection)\n let manualHeader = ampCookieSource == .manual\n ? CookieHeaderNormalizer.normalize(ampCookieHeader)\n : nil\n return await fetcher.debugRawProbe(cookieHeaderOverride: manualHeader)\n }\n }\n\n private static func debugOllamaLog(\n browserDetection: BrowserDetection,\n ollamaCookieSource: ProviderCookieSource,\n ollamaCookieHeader: String) async -> String\n {\n await runWithTimeout(seconds: 15) {\n let fetcher = OllamaUsageFetcher(browserDetection: browserDetection)\n let manualHeader = ollamaCookieSource == .manual\n ? CookieHeaderNormalizer.normalize(ollamaCookieHeader)\n : nil\n return await fetcher.debugRawProbe(\n cookieHeaderOverride: manualHeader,\n manualCookieMode: ollamaCookieSource == .manual)\n }\n }\n\n private func detectVersions() {\n let implementations = ProviderCatalog.all\n let browserDetection = self.browserDetection\n Task { @MainActor [weak self] in\n let resolved = await Task.detached { () -> [UsageProvider: String] in\n var resolved: [UsageProvider: String] = [:]\n await withTaskGroup(of: (UsageProvider, String?).self) { group in\n for implementation in implementations {\n let context = ProviderVersionContext(\n provider: implementation.id,\n browserDetection: browserDetection)\n group.addTask {\n await (implementation.id, implementation.detectVersion(context: context))\n }\n }\n for await (provider, version) in group {\n guard let version, !version.isEmpty else { continue }\n resolved[provider] = version\n }\n }\n return resolved\n }.value\n self?.versions = resolved\n }\n }\n\n @MainActor\n private func schedulePathDebugInfoRefresh() {\n self.pathDebugRefreshTask?.cancel()\n self.pathDebugRefreshTask = Task { [weak self] in\n do {\n try await Task.sleep(nanoseconds: 150_000_000)\n } catch {\n return\n }\n await self?.refreshPathDebugInfo()\n }\n }\n\n private func runBackgroundSnapshot(\n _ snapshot: @escaping @Sendable () async -> PathDebugSnapshot) async\n {\n let result = await snapshot()\n await MainActor.run {\n self.pathDebugInfo = result\n }\n }\n\n private func refreshPathDebugInfo() async {\n await self.runBackgroundSnapshot {\n await PathBuilder.debugSnapshotAsync(purposes: [.rpc, .tty, .nodeTooling])\n }\n }\n\n func clearCostUsageCache() async -> String? {\n let errorMessage: String? = await Task.detached(priority: .utility) {\n let fm = FileManager.default\n let cacheDirs = [\n Self.costUsageCacheDirectory(fileManager: fm),\n ]\n\n for cacheDir in cacheDirs {\n do {\n try fm.removeItem(at: cacheDir)\n } catch let error as NSError {\n if error.domain == NSCocoaErrorDomain, error.code == NSFileNoSuchFileError { continue }\n return error.localizedDescription\n }\n }\n return nil\n }.value\n\n guard errorMessage == nil else { return errorMessage }\n\n self.tokenSnapshots.removeAll()\n self.tokenErrors.removeAll()\n self.lastTokenFetchAt.removeAll()\n self.tokenFailureGates[.codex]?.reset()\n self.tokenFailureGates[.claude]?.reset()\n return nil\n }\n\n private func refreshTokenUsage(_ provider: UsageProvider, force: Bool) async {\n guard provider == .codex || provider == .claude || provider == .vertexai else {\n self.tokenSnapshots.removeValue(forKey: provider)\n self.tokenErrors[provider] = nil\n self.tokenFailureGates[provider]?.reset()\n self.lastTokenFetchAt.removeValue(forKey: provider)\n return\n }\n\n guard self.settings.costUsageEnabled else {\n self.tokenSnapshots.removeValue(forKey: provider)\n self.tokenErrors[provider] = nil\n self.tokenFailureGates[provider]?.reset()\n self.lastTokenFetchAt.removeValue(forKey: provider)\n return\n }\n\n guard self.isEnabled(provider) else {\n self.tokenSnapshots.removeValue(forKey: provider)\n self.tokenErrors[provider] = nil\n self.tokenFailureGates[provider]?.reset()\n self.lastTokenFetchAt.removeValue(forKey: provider)\n return\n }\n\n guard !self.tokenRefreshInFlight.contains(provider) else { return }\n\n let now = Date()\n if !force,\n let last = self.lastTokenFetchAt[provider],\n now.timeIntervalSince(last) < self.tokenFetchTTL\n {\n return\n }\n self.lastTokenFetchAt[provider] = now\n self.tokenRefreshInFlight.insert(provider)\n defer { self.tokenRefreshInFlight.remove(provider) }\n\n let startedAt = Date()\n let providerText = provider.rawValue\n self.tokenCostLogger\n .debug(\"cost usage start provider=\\(providerText) force=\\(force)\")\n\n do {\n let fetcher = self.costUsageFetcher\n let timeoutSeconds = self.tokenFetchTimeout\n let snapshot = try await withThrowingTaskGroup(of: CostUsageTokenSnapshot.self) { group in\n group.addTask(priority: .utility) {\n try await fetcher.loadTokenSnapshot(\n provider: provider,\n now: now,\n forceRefresh: force,\n allowVertexClaudeFallback: !self.isEnabled(.claude))\n }\n group.addTask {\n try await Task.sleep(nanoseconds: UInt64(timeoutSeconds * 1_000_000_000))\n throw CostUsageError.timedOut(seconds: Int(timeoutSeconds))\n }\n defer { group.cancelAll() }\n guard let snapshot = try await group.next() else { throw CancellationError() }\n return snapshot\n }\n\n guard !snapshot.daily.isEmpty else {\n self.tokenSnapshots.removeValue(forKey: provider)\n self.tokenErrors[provider] = Self.tokenCostNoDataMessage(for: provider)\n self.tokenFailureGates[provider]?.recordSuccess()\n return\n }\n let duration = Date().timeIntervalSince(startedAt)\n let sessionCost = snapshot.sessionCostUSD.map(UsageFormatter.usdString) ?? \"—\"\n let monthCost = snapshot.last30DaysCostUSD.map(UsageFormatter.usdString) ?? \"—\"\n let durationText = String(format: \"%.2f\", duration)\n let message =\n \"cost usage success provider=\\(providerText) \" +\n \"duration=\\(durationText)s \" +\n \"today=\\(sessionCost) \" +\n \"30d=\\(monthCost)\"\n self.tokenCostLogger.info(message)\n self.tokenSnapshots[provider] = snapshot\n self.tokenErrors[provider] = nil\n self.tokenFailureGates[provider]?.recordSuccess()\n self.persistWidgetSnapshot(reason: \"token-usage\")\n } catch {\n if error is CancellationError { return }\n let duration = Date().timeIntervalSince(startedAt)\n let msg = error.localizedDescription\n let durationText = String(format: \"%.2f\", duration)\n let message = \"cost usage failed provider=\\(providerText) duration=\\(durationText)s error=\\(msg)\"\n self.tokenCostLogger.error(message)\n let hadPriorData = self.tokenSnapshots[provider] != nil\n let shouldSurface = self.tokenFailureGates[provider]?\n .shouldSurfaceError(onFailureWithPriorData: hadPriorData) ?? true\n if shouldSurface {\n self.tokenErrors[provider] = error.localizedDescription\n self.tokenSnapshots.removeValue(forKey: provider)\n } else {\n self.tokenErrors[provider] = nil\n }\n }\n }\n}\n" }, { "path": "Sources/CodexBar/UsageStoreSupport.swift", "content": "import CodexBarCore\nimport Foundation\n\nenum ProviderStatusIndicator: String {\n case none\n case minor\n case major\n case critical\n case maintenance\n case unknown\n\n var hasIssue: Bool {\n switch self {\n case .none: false\n default: true\n }\n }\n\n var label: String {\n switch self {\n case .none: \"Operational\"\n case .minor: \"Partial outage\"\n case .major: \"Major outage\"\n case .critical: \"Critical issue\"\n case .maintenance: \"Maintenance\"\n case .unknown: \"Status unknown\"\n }\n }\n}\n\nstruct ProviderStatus {\n let indicator: ProviderStatusIndicator\n let description: String?\n let updatedAt: Date?\n}\n\n/// Tracks consecutive failures so we can ignore a single flake when we previously had fresh data.\nstruct ConsecutiveFailureGate {\n private(set) var streak: Int = 0\n\n mutating func recordSuccess() {\n self.streak = 0\n }\n\n mutating func reset() {\n self.streak = 0\n }\n\n /// Returns true when the caller should surface the error to the UI.\n mutating func shouldSurfaceError(onFailureWithPriorData hadPriorData: Bool) -> Bool {\n self.streak += 1\n if hadPriorData, self.streak == 1 { return false }\n return true\n }\n}\n\n#if DEBUG\nextension UsageStore {\n func _setSnapshotForTesting(_ snapshot: UsageSnapshot?, provider: UsageProvider) {\n self.snapshots[provider] = snapshot?.scoped(to: provider)\n }\n\n func _setTokenSnapshotForTesting(_ snapshot: CostUsageTokenSnapshot?, provider: UsageProvider) {\n self.tokenSnapshots[provider] = snapshot\n }\n\n func _setTokenErrorForTesting(_ error: String?, provider: UsageProvider) {\n self.tokenErrors[provider] = error\n }\n\n func _setErrorForTesting(_ error: String?, provider: UsageProvider) {\n self.errors[provider] = error\n }\n\n func _setCodexHistoricalDatasetForTesting(_ dataset: CodexHistoricalDataset?, accountKey: String? = nil) {\n self.codexHistoricalDataset = dataset\n self.codexHistoricalDatasetAccountKey = accountKey\n self.historicalPaceRevision += 1\n }\n}\n#endif\n" }, { "path": "Sources/CodexBar/ZaiTokenStore.swift", "content": "import CodexBarCore\nimport Foundation\nimport Security\n\nprotocol ZaiTokenStoring: Sendable {\n func loadToken() throws -> String?\n func storeToken(_ token: String?) throws\n}\n\nenum ZaiTokenStoreError: LocalizedError {\n case keychainStatus(OSStatus)\n case invalidData\n\n var errorDescription: String? {\n switch self {\n case let .keychainStatus(status):\n \"Keychain error: \\(status)\"\n case .invalidData:\n \"Keychain returned invalid data.\"\n }\n }\n}\n\nstruct KeychainZaiTokenStore: ZaiTokenStoring {\n private static let log = CodexBarLog.logger(LogCategories.zaiTokenStore)\n\n private let service = \"com.steipete.CodexBar\"\n private let account = \"zai-api-token\"\n\n // Cache to reduce keychain access frequency\n private nonisolated(unsafe) static var cachedToken: String?\n private nonisolated(unsafe) static var cacheTimestamp: Date?\n private static let cacheLock = NSLock()\n private static let cacheTTL: TimeInterval = 1800 // 30 minutes\n\n func loadToken() throws -> String? {\n guard !KeychainAccessGate.isDisabled else {\n Self.log.debug(\"Keychain access disabled; skipping token load\")\n return nil\n }\n // Check cache first\n Self.cacheLock.lock()\n if let timestamp = Self.cacheTimestamp,\n Date().timeIntervalSince(timestamp) < Self.cacheTTL\n {\n let cached = Self.cachedToken\n Self.cacheLock.unlock()\n Self.log.debug(\"Using cached Zai token\")\n return cached\n }\n Self.cacheLock.unlock()\n var result: CFTypeRef?\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n kSecMatchLimit as String: kSecMatchLimitOne,\n kSecReturnData as String: true,\n ]\n\n if case .interactionRequired = KeychainAccessPreflight\n .checkGenericPassword(service: self.service, account: self.account)\n {\n KeychainPromptHandler.handler?(KeychainPromptContext(\n kind: .zaiToken,\n service: self.service,\n account: self.account))\n }\n\n let status = SecItemCopyMatching(query as CFDictionary, &result)\n if status == errSecItemNotFound {\n // Cache the nil result\n Self.cacheLock.lock()\n Self.cachedToken = nil\n Self.cacheTimestamp = Date()\n Self.cacheLock.unlock()\n return nil\n }\n guard status == errSecSuccess else {\n Self.log.error(\"Keychain read failed: \\(status)\")\n throw ZaiTokenStoreError.keychainStatus(status)\n }\n\n guard let data = result as? Data else {\n throw ZaiTokenStoreError.invalidData\n }\n let token = String(data: data, encoding: .utf8)?.trimmingCharacters(in: .whitespacesAndNewlines)\n let finalValue = (token?.isEmpty == false) ? token : nil\n\n // Cache the result\n Self.cacheLock.lock()\n Self.cachedToken = finalValue\n Self.cacheTimestamp = Date()\n Self.cacheLock.unlock()\n\n return finalValue\n }\n\n func storeToken(_ token: String?) throws {\n guard !KeychainAccessGate.isDisabled else {\n Self.log.debug(\"Keychain access disabled; skipping token store\")\n return\n }\n let cleaned = token?.trimmingCharacters(in: .whitespacesAndNewlines)\n if cleaned == nil || cleaned?.isEmpty == true {\n try self.deleteTokenIfPresent()\n // Invalidate cache\n Self.cacheLock.lock()\n Self.cachedToken = nil\n Self.cacheTimestamp = nil\n Self.cacheLock.unlock()\n return\n }\n\n let data = cleaned!.data(using: .utf8)!\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n ]\n let attributes: [String: Any] = [\n kSecValueData as String: data,\n kSecAttrAccessible as String: kSecAttrAccessibleAfterFirstUnlockThisDeviceOnly,\n ]\n\n let updateStatus = SecItemUpdate(query as CFDictionary, attributes as CFDictionary)\n if updateStatus == errSecSuccess {\n // Update cache\n Self.cacheLock.lock()\n Self.cachedToken = cleaned\n Self.cacheTimestamp = Date()\n Self.cacheLock.unlock()\n return\n }\n if updateStatus != errSecItemNotFound {\n Self.log.error(\"Keychain update failed: \\(updateStatus)\")\n throw ZaiTokenStoreError.keychainStatus(updateStatus)\n }\n\n var addQuery = query\n for (key, value) in attributes {\n addQuery[key] = value\n }\n let addStatus = SecItemAdd(addQuery as CFDictionary, nil)\n guard addStatus == errSecSuccess else {\n Self.log.error(\"Keychain add failed: \\(addStatus)\")\n throw ZaiTokenStoreError.keychainStatus(addStatus)\n }\n\n // Update cache\n Self.cacheLock.lock()\n Self.cachedToken = cleaned\n Self.cacheTimestamp = Date()\n Self.cacheLock.unlock()\n }\n\n private func deleteTokenIfPresent() throws {\n guard !KeychainAccessGate.isDisabled else { return }\n let query: [String: Any] = [\n kSecClass as String: kSecClassGenericPassword,\n kSecAttrService as String: self.service,\n kSecAttrAccount as String: self.account,\n ]\n let status = SecItemDelete(query as CFDictionary)\n if status == errSecSuccess || status == errSecItemNotFound {\n // Invalidate cache\n Self.cacheLock.lock()\n Self.cachedToken = nil\n Self.cacheTimestamp = nil\n Self.cacheLock.unlock()\n return\n }\n Self.log.error(\"Keychain delete failed: \\(status)\")\n throw ZaiTokenStoreError.keychainStatus(status)\n }\n}\n" }, { "path": "Sources/CodexBarCLI/CLIConfigCommand.swift", "content": "import CodexBarCore\nimport Commander\nimport Foundation\n\nextension CodexBarCLI {\n static func runConfigValidate(_ values: ParsedValues) {\n let output = CLIOutputPreferences.from(values: values)\n let config = Self.loadConfig(output: output)\n let issues = CodexBarConfigValidator.validate(config)\n let hasErrors = issues.contains(where: { $0.severity == .error })\n\n switch output.format {\n case .text:\n if issues.isEmpty {\n print(\"Config: OK\")\n } else {\n for issue in issues {\n let provider = issue.provider?.rawValue ?? \"config\"\n let field = issue.field ?? \"\"\n let prefix = \"[\\(issue.severity.rawValue.uppercased())]\"\n let suffix = field.isEmpty ? \"\" : \" (\\(field))\"\n print(\"\\(prefix) \\(provider)\\(suffix): \\(issue.message)\")\n }\n }\n case .json:\n Self.printJSON(issues, pretty: output.pretty)\n }\n\n Self.exit(code: hasErrors ? .failure : .success, output: output, kind: .config)\n }\n\n static func runConfigDump(_ values: ParsedValues) {\n let output = CLIOutputPreferences.from(values: values)\n let config = Self.loadConfig(output: output)\n Self.printJSON(config, pretty: output.pretty)\n Self.exit(code: .success, output: output, kind: .config)\n }\n}\n\nstruct ConfigOptions: CommanderParsable {\n @Flag(names: [.short(\"v\"), .long(\"verbose\")], help: \"Enable verbose logging\")\n var verbose: Bool = false\n\n @Flag(name: .long(\"json-output\"), help: \"Emit machine-readable logs\")\n var jsonOutput: Bool = false\n\n @Option(name: .long(\"log-level\"), help: \"Set log level (trace|verbose|debug|info|warning|error|critical)\")\n var logLevel: String?\n\n @Option(name: .long(\"format\"), help: \"Output format: text | json\")\n var format: OutputFormat?\n\n @Flag(name: .long(\"json\"), help: \"\")\n var jsonShortcut: Bool = false\n\n @Flag(name: .long(\"json-only\"), help: \"Emit JSON only (suppress non-JSON output)\")\n var jsonOnly: Bool = false\n\n @Flag(name: .long(\"pretty\"), help: \"Pretty-print JSON output\")\n var pretty: Bool = false\n}\n" }, { "path": "Sources/CodexBarCLI/CLICostCommand.swift", "content": "import CodexBarCore\nimport Commander\nimport Foundation\n\nextension CodexBarCLI {\n private static let costSupportedProviders: Set = [.claude, .codex]\n\n static func runCost(_ values: ParsedValues) async {\n let output = CLIOutputPreferences.from(values: values)\n let config = CodexBarCLI.loadConfig(output: output)\n let selection = CodexBarCLI.decodeProvider(from: values, config: config)\n let providers = Self.costProviders(from: selection)\n let unsupported = selection.asList.filter { !Self.costSupportedProviders.contains($0) }\n if !unsupported.isEmpty {\n let names = unsupported\n .map { ProviderDescriptorRegistry.descriptor(for: $0).metadata.displayName }\n .sorted()\n .joined(separator: \", \")\n if !output.jsonOnly {\n Self.writeStderr(\"Skipping providers without local cost usage: \\(names)\\n\")\n }\n }\n guard !providers.isEmpty else {\n Self.exit(\n code: .failure,\n message: \"Error: cost is only supported for Claude and Codex.\",\n output: output,\n kind: .args)\n }\n\n let format = output.format\n let forceRefresh = values.flags.contains(\"refresh\")\n let useColor = Self.shouldUseColor(noColor: values.flags.contains(\"noColor\"), format: format)\n\n let fetcher = CostUsageFetcher()\n var sections: [String] = []\n var payload: [CostPayload] = []\n var exitCode: ExitCode = .success\n\n for provider in providers {\n do {\n // Cost usage is local-only; it does not require web/CLI provider fetches.\n let snapshot = try await fetcher.loadTokenSnapshot(\n provider: provider,\n forceRefresh: forceRefresh)\n switch format {\n case .text:\n sections.append(Self.renderCostText(provider: provider, snapshot: snapshot, useColor: useColor))\n case .json:\n payload.append(Self.makeCostPayload(provider: provider, snapshot: snapshot, error: nil))\n }\n } catch {\n exitCode = Self.mapError(error)\n if format == .json {\n payload.append(Self.makeCostPayload(provider: provider, snapshot: nil, error: error))\n } else if !output.jsonOnly {\n Self.writeStderr(\"Error: \\(error.localizedDescription)\\n\")\n }\n }\n }\n\n switch format {\n case .text:\n if !sections.isEmpty {\n print(sections.joined(separator: \"\\n\\n\"))\n }\n case .json:\n if !payload.isEmpty {\n Self.printJSON(payload, pretty: output.pretty)\n }\n }\n\n Self.exit(code: exitCode, output: output, kind: exitCode == .success ? .runtime : .provider)\n }\n\n static func renderCostText(\n provider: UsageProvider,\n snapshot: CostUsageTokenSnapshot,\n useColor: Bool) -> String\n {\n let name = ProviderDescriptorRegistry.descriptor(for: provider).metadata.displayName\n let header = Self.costHeaderLine(\"\\(name) Cost (local)\", useColor: useColor)\n\n let todayCost = snapshot.sessionCostUSD.map { UsageFormatter.usdString($0) } ?? \"—\"\n let todayTokens = snapshot.sessionTokens.map { UsageFormatter.tokenCountString($0) }\n let todayLine = todayTokens.map { \"Today: \\(todayCost) · \\($0) tokens\" } ?? \"Today: \\(todayCost)\"\n\n let monthCost = snapshot.last30DaysCostUSD.map { UsageFormatter.usdString($0) } ?? \"—\"\n let monthTokens = snapshot.last30DaysTokens.map { UsageFormatter.tokenCountString($0) }\n let monthLine = monthTokens.map { \"Last 30 days: \\(monthCost) · \\($0) tokens\" } ?? \"Last 30 days: \\(monthCost)\"\n\n return [header, todayLine, monthLine].joined(separator: \"\\n\")\n }\n\n private static func costHeaderLine(_ header: String, useColor: Bool) -> String {\n guard useColor else { return header }\n return \"\\u{001B}[1;36m\\(header)\\u{001B}[0m\"\n }\n\n private static func costProviders(from selection: ProviderSelection) -> [UsageProvider] {\n selection.asList.filter { Self.costSupportedProviders.contains($0) }\n }\n\n private static func makeCostPayload(\n provider: UsageProvider,\n snapshot: CostUsageTokenSnapshot?,\n error: Error?) -> CostPayload\n {\n let daily = snapshot?.daily.map { entry in\n CostDailyEntryPayload(\n date: entry.date,\n inputTokens: entry.inputTokens,\n outputTokens: entry.outputTokens,\n cacheReadTokens: entry.cacheReadTokens,\n cacheCreationTokens: entry.cacheCreationTokens,\n totalTokens: entry.totalTokens,\n costUSD: entry.costUSD,\n modelsUsed: entry.modelsUsed,\n modelBreakdowns: entry.modelBreakdowns?.map { breakdown in\n CostModelBreakdownPayload(\n modelName: breakdown.modelName,\n costUSD: breakdown.costUSD,\n totalTokens: breakdown.totalTokens)\n })\n } ?? []\n\n return CostPayload(\n provider: provider.rawValue,\n source: \"local\",\n updatedAt: snapshot?.updatedAt ?? (error == nil ? nil : Date()),\n sessionTokens: snapshot?.sessionTokens,\n sessionCostUSD: snapshot?.sessionCostUSD,\n last30DaysTokens: snapshot?.last30DaysTokens,\n last30DaysCostUSD: snapshot?.last30DaysCostUSD,\n daily: daily,\n totals: snapshot.flatMap(Self.costTotals(from:)),\n error: error.map { Self.makeErrorPayload($0) })\n }\n\n private static func costTotals(from snapshot: CostUsageTokenSnapshot) -> CostTotalsPayload? {\n let entries = snapshot.daily\n guard !entries.isEmpty else {\n guard snapshot.last30DaysTokens != nil || snapshot.last30DaysCostUSD != nil else { return nil }\n return CostTotalsPayload(\n totalInputTokens: nil,\n totalOutputTokens: nil,\n cacheReadTokens: nil,\n cacheCreationTokens: nil,\n totalTokens: snapshot.last30DaysTokens,\n totalCostUSD: snapshot.last30DaysCostUSD)\n }\n\n var totalInput = 0\n var totalOutput = 0\n var totalCacheRead = 0\n var totalCacheCreation = 0\n var totalTokens = 0\n var totalCost = 0.0\n var sawInput = false\n var sawOutput = false\n var sawCacheRead = false\n var sawCacheCreation = false\n var sawTokens = false\n var sawCost = false\n\n for entry in entries {\n if let input = entry.inputTokens {\n totalInput += input\n sawInput = true\n }\n if let output = entry.outputTokens {\n totalOutput += output\n sawOutput = true\n }\n if let cacheRead = entry.cacheReadTokens {\n totalCacheRead += cacheRead\n sawCacheRead = true\n }\n if let cacheCreation = entry.cacheCreationTokens {\n totalCacheCreation += cacheCreation\n sawCacheCreation = true\n }\n if let tokens = entry.totalTokens {\n totalTokens += tokens\n sawTokens = true\n }\n if let cost = entry.costUSD {\n totalCost += cost\n sawCost = true\n }\n }\n\n // Prefer totals derived from daily rows; fall back to snapshot aggregates when rows omit fields.\n return CostTotalsPayload(\n totalInputTokens: sawInput ? totalInput : nil,\n totalOutputTokens: sawOutput ? totalOutput : nil,\n cacheReadTokens: sawCacheRead ? totalCacheRead : nil,\n cacheCreationTokens: sawCacheCreation ? totalCacheCreation : nil,\n totalTokens: sawTokens ? totalTokens : snapshot.last30DaysTokens,\n totalCostUSD: sawCost ? totalCost : snapshot.last30DaysCostUSD)\n }\n}\n\nstruct CostOptions: CommanderParsable {\n @Flag(names: [.short(\"v\"), .long(\"verbose\")], help: \"Enable verbose logging\")\n var verbose: Bool = false\n\n @Flag(name: .long(\"json-output\"), help: \"Emit machine-readable logs\")\n var jsonOutput: Bool = false\n\n @Option(name: .long(\"log-level\"), help: \"Set log level (trace|verbose|debug|info|warning|error|critical)\")\n var logLevel: String?\n\n @Option(\n name: .long(\"provider\"),\n help: ProviderHelp.optionHelp)\n var provider: ProviderSelection?\n\n @Option(name: .long(\"format\"), help: \"Output format: text | json\")\n var format: OutputFormat?\n\n @Flag(name: .long(\"json\"), help: \"\")\n var jsonShortcut: Bool = false\n\n @Flag(name: .long(\"json-only\"), help: \"Emit JSON only (suppress non-JSON output)\")\n var jsonOnly: Bool = false\n\n @Flag(name: .long(\"pretty\"), help: \"Pretty-print JSON output\")\n var pretty: Bool = false\n\n @Flag(name: .long(\"no-color\"), help: \"Disable ANSI colors in text output\")\n var noColor: Bool = false\n\n @Flag(name: .long(\"refresh\"), help: \"Force refresh by ignoring cached scans\")\n var refresh: Bool = false\n}\n\nstruct CostPayload: Encodable {\n let provider: String\n let source: String\n let updatedAt: Date?\n let sessionTokens: Int?\n let sessionCostUSD: Double?\n let last30DaysTokens: Int?\n let last30DaysCostUSD: Double?\n let daily: [CostDailyEntryPayload]\n let totals: CostTotalsPayload?\n let error: ProviderErrorPayload?\n}\n\nstruct CostDailyEntryPayload: Encodable {\n let date: String\n let inputTokens: Int?\n let outputTokens: Int?\n let cacheReadTokens: Int?\n let cacheCreationTokens: Int?\n let totalTokens: Int?\n let costUSD: Double?\n let modelsUsed: [String]?\n let modelBreakdowns: [CostModelBreakdownPayload]?\n\n private enum CodingKeys: String, CodingKey {\n case date\n case inputTokens\n case outputTokens\n case cacheReadTokens\n case cacheCreationTokens\n case totalTokens\n case costUSD = \"totalCost\"\n case modelsUsed\n case modelBreakdowns\n }\n}\n\nstruct CostModelBreakdownPayload: Encodable {\n let modelName: String\n let costUSD: Double?\n let totalTokens: Int?\n\n private enum CodingKeys: String, CodingKey {\n case modelName\n case costUSD = \"cost\"\n case totalTokens\n }\n}\n\nstruct CostTotalsPayload: Encodable {\n let totalInputTokens: Int?\n let totalOutputTokens: Int?\n let cacheReadTokens: Int?\n let cacheCreationTokens: Int?\n let totalTokens: Int?\n let totalCostUSD: Double?\n\n private enum CodingKeys: String, CodingKey {\n case totalInputTokens = \"inputTokens\"\n case totalOutputTokens = \"outputTokens\"\n case cacheReadTokens\n case cacheCreationTokens\n case totalTokens\n case totalCostUSD = \"totalCost\"\n }\n}\n\n// Intentionally empty.\n" }, { "path": "Sources/CodexBarCLI/CLIEntry.swift", "content": "import CodexBarCore\nimport Commander\n#if canImport(AppKit)\nimport AppKit\n#endif\n#if canImport(Darwin)\nimport Darwin\n#else\nimport Glibc\n#endif\nimport Foundation\n#if canImport(FoundationNetworking)\nimport FoundationNetworking\n#endif\n\n@main\nenum CodexBarCLI {\n static func main() async {\n let rawArgv = Array(CommandLine.arguments.dropFirst())\n let argv = Self.effectiveArgv(rawArgv)\n let outputPreferences = CLIOutputPreferences.from(argv: argv)\n\n // Fast path: global help/version before building descriptors.\n if let helpIndex = argv.firstIndex(where: { $0 == \"-h\" || $0 == \"--help\" }) {\n let command = helpIndex == 0 ? argv.dropFirst().first : argv.first\n Self.printHelp(for: command)\n }\n if argv.contains(\"-V\") || argv.contains(\"--version\") {\n Self.printVersion()\n }\n\n let program = Program(descriptors: Self.commandDescriptors())\n\n do {\n let invocation = try program.resolve(argv: argv)\n Self.bootstrapLogging(values: invocation.parsedValues)\n switch invocation.path {\n case [\"usage\"]:\n await self.runUsage(invocation.parsedValues)\n case [\"cost\"]:\n await self.runCost(invocation.parsedValues)\n case [\"config\", \"validate\"]:\n self.runConfigValidate(invocation.parsedValues)\n case [\"config\", \"dump\"]:\n self.runConfigDump(invocation.parsedValues)\n default:\n Self.exit(\n code: .failure,\n message: \"Unknown command\",\n output: outputPreferences,\n kind: .args)\n }\n } catch let error as CommanderProgramError {\n Self.exit(code: .failure, message: error.description, output: outputPreferences, kind: .args)\n } catch {\n Self.exit(code: .failure, message: error.localizedDescription, output: outputPreferences, kind: .runtime)\n }\n }\n\n private static func commandDescriptors() -> [CommandDescriptor] {\n let usageSignature = CommandSignature.describe(UsageOptions())\n let costSignature = CommandSignature.describe(CostOptions())\n let configSignature = CommandSignature.describe(ConfigOptions())\n\n return [\n CommandDescriptor(\n name: \"usage\",\n abstract: \"Print usage as text or JSON\",\n discussion: nil,\n signature: usageSignature),\n CommandDescriptor(\n name: \"cost\",\n abstract: \"Print local cost usage as text or JSON\",\n discussion: nil,\n signature: costSignature),\n CommandDescriptor(\n name: \"config\",\n abstract: \"Config utilities\",\n discussion: nil,\n signature: CommandSignature(),\n subcommands: [\n CommandDescriptor(\n name: \"validate\",\n abstract: \"Validate config file\",\n discussion: nil,\n signature: configSignature),\n CommandDescriptor(\n name: \"dump\",\n abstract: \"Print normalized config JSON\",\n discussion: nil,\n signature: configSignature),\n ],\n defaultSubcommandName: \"validate\"),\n ]\n }\n\n // MARK: - Helpers\n\n private static func bootstrapLogging(values: ParsedValues) {\n let isJSON = values.flags.contains(\"jsonOutput\") || values.flags.contains(\"jsonOnly\")\n let verbose = values.flags.contains(\"verbose\")\n let rawLevel = values.options[\"logLevel\"]?.last\n let level = Self.resolvedLogLevel(verbose: verbose, rawLevel: rawLevel)\n CodexBarLog.bootstrapIfNeeded(.init(destination: .stderr, level: level, json: isJSON))\n }\n\n static func resolvedLogLevel(verbose: Bool, rawLevel: String?) -> CodexBarLog.Level {\n CodexBarLog.parseLevel(rawLevel) ?? (verbose ? .debug : .error)\n }\n\n static func effectiveArgv(_ argv: [String]) -> [String] {\n guard let first = argv.first else { return [\"usage\"] }\n if first.hasPrefix(\"-\") { return [\"usage\"] + argv }\n return argv\n }\n}\n" }, { "path": "Sources/CodexBarCLI/CLIErrorReporting.swift", "content": "import CodexBarCore\nimport Foundation\n\nenum CLIErrorKind: String, Encodable {\n case args\n case config\n case provider\n case runtime\n}\n\nstruct ProviderErrorPayload: Encodable {\n let code: Int32\n let message: String\n let kind: CLIErrorKind?\n}\n\nextension CodexBarCLI {\n static func makeErrorPayload(_ error: Error, kind: CLIErrorKind? = nil) -> ProviderErrorPayload {\n ProviderErrorPayload(\n code: self.mapError(error).rawValue,\n message: error.localizedDescription,\n kind: kind)\n }\n\n static func makeErrorPayload(code: ExitCode, message: String, kind: CLIErrorKind? = nil) -> ProviderErrorPayload {\n ProviderErrorPayload(code: code.rawValue, message: message, kind: kind)\n }\n\n static func makeCLIErrorPayload(\n message: String,\n code: ExitCode,\n kind: CLIErrorKind,\n pretty: Bool) -> String?\n {\n let payload = ProviderPayload(\n providerID: \"cli\",\n account: nil,\n version: nil,\n source: \"cli\",\n status: nil,\n usage: nil,\n credits: nil,\n antigravityPlanInfo: nil,\n openaiDashboard: nil,\n error: ProviderErrorPayload(code: code.rawValue, message: message, kind: kind))\n return self.encodeJSON([payload], pretty: pretty)\n }\n\n static func makeProviderErrorPayload(\n provider: UsageProvider,\n account: String?,\n source: String,\n status: ProviderStatusPayload?,\n error: Error,\n kind: CLIErrorKind = .provider) -> ProviderPayload\n {\n ProviderPayload(\n provider: provider,\n account: account,\n version: nil,\n source: source,\n status: status,\n usage: nil,\n credits: nil,\n antigravityPlanInfo: nil,\n openaiDashboard: nil,\n error: self.makeErrorPayload(error, kind: kind))\n }\n\n static func encodeJSON(_ payload: some Encodable, pretty: Bool) -> String? {\n let encoder = JSONEncoder()\n encoder.dateEncodingStrategy = .iso8601\n encoder.outputFormatting = pretty ? [.prettyPrinted, .sortedKeys] : []\n guard let data = try? encoder.encode(payload) else { return nil }\n return String(data: data, encoding: .utf8)\n }\n\n static func printJSON(_ payload: some Encodable, pretty: Bool) {\n if let output = self.encodeJSON(payload, pretty: pretty) {\n print(output)\n }\n }\n\n static func exit(\n code: ExitCode,\n message: String? = nil,\n output: CLIOutputPreferences? = nil,\n kind: CLIErrorKind = .runtime) -> Never\n {\n if code != .success {\n if let output, output.usesJSONOutput {\n let payload = self.makeCLIErrorPayload(\n message: message ?? \"Error\",\n code: code,\n kind: kind,\n pretty: output.pretty)\n if let payload {\n print(payload)\n }\n } else if let message {\n self.writeStderr(\"\\(message)\\n\")\n }\n }\n platformExit(code.rawValue)\n }\n\n static func printError(_ error: Error, output: CLIOutputPreferences, kind: CLIErrorKind = .runtime) {\n if output.usesJSONOutput {\n let payload = ProviderPayload(\n providerID: \"cli\",\n account: nil,\n version: nil,\n source: \"cli\",\n status: nil,\n usage: nil,\n credits: nil,\n antigravityPlanInfo: nil,\n openaiDashboard: nil,\n error: self.makeErrorPayload(error, kind: kind))\n self.printJSON([payload], pretty: output.pretty)\n } else {\n self.writeStderr(\"Error: \\(error.localizedDescription)\\n\")\n }\n }\n}\n" }, { "path": "Sources/CodexBarCLI/CLIExitCode.swift", "content": "enum ExitCode: Int32 {\n case success = 0\n case failure = 1\n case binaryNotFound = 2\n case parseError = 3\n case timeout = 4\n\n init(_ rawValue: Int) {\n self = ExitCode(rawValue: Int32(rawValue)) ?? .failure\n }\n}\n" }, { "path": "Sources/CodexBarCLI/CLIHelp.swift", "content": "import CodexBarCore\nimport Foundation\n\nextension CodexBarCLI {\n static func usageHelp(version: String) -> String {\n \"\"\"\n CodexBar \\(version)\n\n Usage:\n codexbar usage [--format text|json]\n [--json]\n [--json-only]\n [--json-output] [--log-level ] [-v|--verbose]\n [--provider \\(ProviderHelp.list)]\n [--account