Skip to content

Add WAL-safe Kimi Desktop monthly enrichment - #2622

Merged
steipete merged 3 commits into
mainfrom
codex/kimi-cn-monthly
Aug 3, 2026
Merged

Add WAL-safe Kimi Desktop monthly enrichment#2622
steipete merged 3 commits into
mainfrom
codex/kimi-cn-monthly

Conversation

@steipete

@steipete steipete commented Aug 3, 2026

Copy link
Copy Markdown
Owner

Summary

  • enrich Kimi Code API and CLI snapshots with the monthly membership pool when an approved web session is available
  • read the official Kimi Desktop Chromium cookie store directly through SQLite read-only mode
  • keep active WAL databases on SQLite's normal WAL-aware path and retry idle WAL headers through an immutable URI only when both sidecars are absent
  • preserve the required Code usage result when optional membership enrichment fails, while propagating cancellation
  • stop before desktop or browser resolution when Cookie Source is Off

This is the in-house Slice B takeover of #2351. The contributor implementation was mixed into a conflicted multi-provider commit, so the focused behavior is reimplemented here with contributor credit preserved.

Root cause

Kimi's Code usage endpoint exposes weekly and rate-limit data but not the monthly membership pool. The contributor branch copied only the main Chromium database file before reading it, which can miss committed rows held in an active WAL and is not a safe snapshot of a WAL database. This version follows the read-only OpenCode pattern from #2563: normal read-only access for active WAL stores, with a narrowly gated immutable fallback for an idle WAL header whose -wal and -shm files are both absent.

Verification

  • focused Kimi desktop/enrichment suites: 38 tests passed
  • active WAL fixture remains readable while its sidecar is present
  • idle WAL fixture reads through immutable fallback and creates no sidecars
  • Cookie Source Off regression proves no enrichment resolver is invoked
  • make check — passed, 0 format or lint violations
  • make test — 803 selections in 67 groups passed; 0 failed groups, retries, or timeouts
  • isolated autoreview — clean, no accepted/actionable findings

Live CN account verification was not run in this lane. After release, verification is requested from the reporter community on #2351, #2349, and #2500 for Kimi Desktop monthly-pool enrichment and real account payloads.

References #2351.

Co-authored-by: haoli <haoli@local.dev>

@chatgpt-codex-connector chatgpt-codex-connector Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

💡 Codex Review

Here are some automated review suggestions for this pull request.

Reviewed commit: e1825f0be5

ℹ️ About Codex in GitHub

Codex has been enabled to automatically review pull requests in this repo. Reviews are triggered when you

  • Open a pull request for review
  • Mark a draft as ready
  • Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

When you sign up for Codex through ChatGPT, Codex can also answer questions or update the PR, like "@codex address that feedback".


enum KimiWebEnrichmentTokenResolver {
static func resolve(_ context: ProviderFetchContext) -> String? {
guard let settings = context.settings?.kimi, settings.cookieSource != .off else { return nil }

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

P2 Badge Honor the optional-usage opt-out for Kimi enrichment

When the user disables optional credits/extra usage, ProviderRegistry still passes context.includeOptionalUsage == false, but this resolver only checks Cookie Source. In API/CLI Kimi modes that means CodexBar still resolves desktop/browser cookies, sends the monthly membership request, and renders extraRateWindows that the global opt-out is supposed to suppress; short-circuit Kimi web enrichment when includeOptionalUsage is false.

Useful? React with 👍 / 👎.

Comment on lines +50 to +54
return try await self.enrichCodeAPIUsage(
snapshot,
webAuthToken: webAuthToken,
now: now,
transport: transport)

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

P2 Badge Bound optional Kimi enrichment before returning Code usage

When a web session is available but GetSubscriptionStats stalls, this sequential await withholds an otherwise successful Code API/CLI usage snapshot until the provider HTTP timeout instead of returning the required result promptly. The web-cookie path already treats the same subscription call as optional with a short grace period, so API/CLI enrichment should use a bounded/joined optional fetch as well.

Useful? React with 👍 / 👎.

@clawsweeper clawsweeper Bot added rating: 🦐 gold shrimp Decent PR readiness signal, but merge confidence is limited. status: ⏳ waiting on author ClawSweeper has contributor-facing work open and is waiting for author action. P2 Normal priority bug or improvement with limited blast radius. merge-risk: 🚨 auth-provider 🚨 Merging this PR could break OAuth, tokens, provider routing, model choice, or credentials. labels Aug 3, 2026
@clawsweeper

clawsweeper Bot commented Aug 3, 2026

Copy link
Copy Markdown

Codex review: needs changes before merge. Reviewed August 3, 2026, 5:07 PM ET / 21:07 UTC.

ClawSweeper review

What this changes

The branch adds WAL-safe read-only Kimi Desktop cookie access and uses an approved web session to append an optional monthly membership window to Kimi Code API and CLI usage.

Merge readiness

⚠️ Ready for maintainer review - 4 items remain

This owner-authored PR remains useful and should stay open, but its API and CLI paths still bypass the global optional-usage opt-out and can delay required Kimi Code usage behind a stalled optional membership request.

Priority: P2
Reviewed head: 0a644dcdb2c60010d7a48df90bcd934e5fb28e96

Review scores

Measure Result What it means
Overall readiness 🦐 gold shrimp (3/6) Focused tests and a narrow WAL design provide useful signal, but the two functional blockers must be fixed before the patch is merge-ready.
Proof confidence 🦐 gold shrimp (3/6) Not applicable: This is an owner-authored PR, so the external-contributor proof gate does not apply; its reported automated validation remains useful, and post-release community confirmation is planned.
Patch quality 🦐 gold shrimp (3/6) 2 actionable review findings remain.

Verification

Check Result Evidence
Real behavior Not applicable Not applicable: This is an owner-authored PR, so the external-contributor proof gate does not apply; its reported automated validation remains useful, and post-release community confirmation is planned.
Evidence reviewed 5 items Optional-usage opt-out is bypassed: Both API and CLI enrichment helpers test only Kimi Cookie Source before resolving a web token; neither tests ProviderFetchContext.includeOptionalUsage.
Required result waits on optional request: The Code API path awaits enrichCodeAPIUsage after successful required usage; that helper directly awaits fetchSubscriptionStats rather than using the bounded join already used by the web-only path.
Existing bounded pattern: The Kimi web path starts the membership request independently and joins it with subscriptionGrace while required usage continues, establishing the intended best-effort behavior.
Findings 2 actionable findings [P2] Honor the optional-usage opt-out before resolving cookies
[P2] Bound optional membership enrichment before returning Code usage
Security None None.

How this fits together

CodexBar’s Kimi provider first obtains required Code quota data through API-key or CLI credentials, then may enrich that result from a Kimi web session. The combined usage snapshot feeds the menu-bar display and CLI output.

flowchart LR
  A[Provider settings] --> B[Kimi API or CLI fetch]
  B --> C[Required Code usage]
  A --> D[Optional web session resolver]
  D --> E[Monthly membership request]
  C --> F[Combined usage snapshot]
  E --> F
  F --> G[Menu bar and CLI output]
Loading

Before merge

  • Honor the optional-usage opt-out before resolving cookies (P2) - ProviderRegistry can set includeOptionalUsage to false, but this guard checks only Cookie Source. Both API and CLI paths can still inspect desktop/browser sessions, call the membership endpoint, and render an optional monthly window after the user disabled optional usage.
  • Bound optional membership enrichment before returning Code usage (P2) - fetchCodeAPIUsage awaits enrichCodeAPIUsage, which directly awaits the optional membership request. If GetSubscriptionStats stalls, a successful required Code snapshot is withheld until transport timeout instead of using the existing bounded best-effort pattern.
  • Resolve merge risk (P1) - Existing users who disable optional usage can still have Kimi Desktop or browser credentials inspected and a monthly-membership request sent.
  • Resolve merge risk (P2) - A stalled membership endpoint can make required Kimi Code usage wait for transport timeout rather than returning promptly.

Findings

  • [P2] Honor the optional-usage opt-out before resolving cookies — Sources/CodexBarCore/Providers/Kimi/KimiProviderDescriptor.swift:103-105
  • [P2] Bound optional membership enrichment before returning Code usage — Sources/CodexBarCore/Providers/Kimi/KimiUsageFetcher.swift:48-54
Agent review details

Security

None.

Review metrics

Metric Value Why it matters
Patch scope production +216, tests +248, docs +12 The PR adds a substantive credential-reading and provider-enrichment path with substantial focused test coverage.

Merge-risk options

Maintainer options:

  1. Repair optional enrichment (recommended)
    Gate both API and CLI enrichment on the shared optional-usage flag and use a bounded join for the membership request before merge.
  2. Accept changed opt-out semantics
    Merge only if maintainers intentionally want Cookie Source to override the global optional-usage setting and accept required usage waiting on membership transport timeouts.
Copy recommended automerge instruction
@clawsweeper automerge

Special instructions:
Gate Kimi API and CLI enrichment on context.includeOptionalUsage, use the existing bounded optional-membership pattern so required Code usage returns promptly while cancellation propagates, and add focused regressions.

Technical review

Best possible solution:

Keep the WAL-safe reader, but make the monthly overlay respect the global opt-out and use the existing bounded best-effort pattern so required Code usage is never held behind enrichment.

Do we have a high-confidence way to reproduce the issue?

Yes—set includeOptionalUsage false with Cookie Source enabled, invoke either Kimi API or CLI strategy, and its current enrichment helper still resolves a web token. A stubbed slow subscription endpoint can likewise show required Code usage waiting behind enrichment.

Is this the best way to solve the issue?

No—the current implementation is not yet the best solution because it bypasses the shared opt-out and does not reuse the existing bounded best-effort approach for an optional request.

Full review comments:

  • [P2] Honor the optional-usage opt-out before resolving cookies — Sources/CodexBarCore/Providers/Kimi/KimiProviderDescriptor.swift:103-105
    ProviderRegistry can set includeOptionalUsage to false, but this guard checks only Cookie Source. Both API and CLI paths can still inspect desktop/browser sessions, call the membership endpoint, and render an optional monthly window after the user disabled optional usage.
    Confidence: 0.99
  • [P2] Bound optional membership enrichment before returning Code usage — Sources/CodexBarCore/Providers/Kimi/KimiUsageFetcher.swift:48-54
    fetchCodeAPIUsage awaits enrichCodeAPIUsage, which directly awaits the optional membership request. If GetSubscriptionStats stalls, a successful required Code snapshot is withheld until transport timeout instead of using the existing bounded best-effort pattern.
    Confidence: 0.98

Overall correctness: patch is incorrect
Overall confidence: 0.99

AGENTS.md: found and applied where relevant.

Codex review notes: model internal, reasoning high; reviewed against 7002b5782053.

Labels

Label justifications:

  • P2: The remaining defects have limited but concrete provider impact: they can violate a user preference and delay normal quota output.
  • merge-risk: 🚨 auth-provider: Merging would let Kimi web credential resolution proceed despite the shared optional-usage opt-out.
  • rating: 🦐 gold shrimp: Overall readiness is 🦐 gold shrimp; proof is 🦐 gold shrimp and patch quality is 🦐 gold shrimp.
  • status: ⏳ waiting on author: ClawSweeper has contributor-facing work open and is waiting for author action. Not applicable: This is an owner-authored PR, so the external-contributor proof gate does not apply; its reported automated validation remains useful, and post-release community confirmation is planned.

Evidence

Acceptance criteria:

  • [P1] swift test --filter KimiProviderTests.
  • [P1] swift test --filter KimiDesktopAuthTokenTests.
  • [P1] make check.
  • [P1] make test.

What I checked:

Likely related people:

  • steipete: Recent main history attributes Kimi quota-routing work and the related merged idle-WAL read pattern to Peter Steinberger; GitHub context maps that author to this repository owner handle. (role: recent area contributor; confidence: high; commits: 7002b5782053, c4df5bb029d0, 0a644dcdb2c6; files: Sources/CodexBarCore/Providers/Kimi/KimiProviderDescriptor.swift, Sources/CodexBarCore/Providers/Kimi/KimiUsageFetcher.swift, Sources/CodexBarCore/Providers/Kimi/KimiDesktopAuthToken.swift)
  • rehanchrl: History identifies this contributor as the author of the original Kimi provider implementation that the changed fetch path extends. (role: feature introducer; confidence: medium; commits: c3a030429859, dfcf532b5478; files: Sources/CodexBarCore/Providers/Kimi/KimiUsageFetcher.swift)

Rank-up moves

Optional improvements that raise the rating; they are not merge blockers.

  • Add the shared includeOptionalUsage guard and cover both API and CLI paths.
  • Bound the optional membership request while preserving cancellation, with a slow-transport regression.

Rating scale

Score Internal tier Crab rank Meaning
6/6 S 🦀 challenger crab Exceptional readiness
5/6 A 🦞 diamond lobster Very strong readiness
4/6 B 🐚 platinum hermit Good normal PR; ordinary maintainer review
3/6 C 🦐 gold shrimp Useful, but confidence is limited
2/6 D 🦪 silver shellfish Proof or implementation needs work
1/6 F 🧂 unranked krab Not merge-ready
N/A NA 🌊 off-meta tidepool Rating does not apply

Overall follows the weaker of proof and patch quality.
Shiny media proof means a screenshot, video, or linked artifact directly shows the changed behavior. Runtime, network, CSP, and security claims still need visible diagnostics.

Workflow

  • ClawSweeper keeps one durable marker-backed review comment per issue or PR.
  • Re-runs edit this comment so the latest verdict, findings, and automation markers stay together instead of adding duplicate bot comments.
  • A fresh review can be triggered by eligible @clawsweeper re-review comments, exact-item GitHub events, scheduled/background review runs, or manual workflow dispatch.
  • PR/issue authors and users with repository write access can comment @clawsweeper re-review or @clawsweeper re-run on an open PR or issue to request a fresh review only.
  • Maintainers can also comment @clawsweeper review to request a fresh review only.
  • Fresh-review commands do not start repair, autofix, rebase, CI repair, or automerge.
  • Maintainer-only repair and merge flows require explicit commands such as @clawsweeper autofix, @clawsweeper automerge, @clawsweeper fix ci, or @clawsweeper address review.
  • Maintainers can comment @clawsweeper explain to ask for more context, or @clawsweeper stop to stop active automation.

History

Review history (1 earlier review cycle)
  • reviewed 2026-08-03T20:20:42.761Z sha 613af4c :: needs changes before merge. :: [P2] Honor the optional-usage opt-out before resolving cookies | [P2] Bound the optional membership request | [P3] Remove the release-owned changelog edit

@steipete
steipete merged commit 6afa672 into main Aug 3, 2026
9 checks passed
Finesssee added a commit to nesszer/Win-CodexBar that referenced this pull request Aug 9, 2026
* Port upstream 0.48.0: bound serve request heads

* Test serve deadline-driven connection recovery

* Port upstream 0.48.0: serve dashboard + snapshot contract

* Port upstream 0.48.0: codexbar dashboard command

* fix(dashboard): coordinator lost-wakeup/cancellation, daily totalCost wire, bare --output, camelCase schema

F1: coordinator.rs registers the Building waiter's Notified while the lock
is held (await_build helper), closing the notify_waiters lost-wakeup window;
a BuildGuard resets a stranded Slot::Building to Empty + wakes waiters when a
build is cancelled or panics. Adds deterministic lost-wakeup + cancel/panic
regression tests with tokio::time::timeout bounds. Single-flight, TTL,
late-result delivery, and errors-not-cached semantics preserved.

F2: data.rs emits the upstream daily wire key 'totalCost' (was cost_usd);
dashboard.html gates on 'daily.some(v => (v.totalCost || 0) > 0)' and every
per-row read uses 'd.totalCost'. Adds wire/HTML agreement and positive/zero/
empty chart behavior tests.

F3: dashboard.rs write_atomic treats an empty Path::parent as '.' so
'--output bare-name.json' writes to the working directory. Test covers bare
relative create, replacement, and no temporary leftover.

F4: AccountPayload and StatusPayload serialize camelCase (updatedAt) to
match pinned v1; golden assertions verify 'updatedAt' exists and 'updated_at'
does not.

* Port upstream 0.48.0: OpenCode Go per-model daily cost breakdown

Extract each local assistant message's modelID from opencode.db (the real
model behind the constant opencode-go Zen-proxy providerID) and group cost /
request counts by (day, model) instead of just by day, so the shared Cost
history chart shows a per-model breakdown for OpenCode Go the same way it
already does for Codex/Claude. Rows with no modelID fall back to an 'unknown'
bucket; whitespace-only ids collapse to 'unknown', whitespace-padded ids merge
with the trimmed bucket (upstream steipete#2649).

Provider-local: the SQL extraction, model normalization and (day, model)
aggregation live in the OpenCode Go local reader. The shared cost surfaces are
reused, not duplicated -- get_daily_cost_history gains an 'opencodego' arm and
CostScanner gains scan_opencodego_with_cancel that maps the provider-local
summary onto the existing CostSummary (total_cost_usd, by_model, sessions_count,
period), so the chart's local-usage panel and daily cost history treat OpenCode
Go like Codex/Claude without a bespoke chart surface.

No Codex/Claude pricing policy is copied; no duplicate chart machinery.

Tests: 11 focused regressions covering multiple models/days, same-model merge,
step-finish model inheritance, unknown/whitespace-only/whitespace-padded model
ids, zero-cost rows, malformed rows, history-window exclusion, local-day
boundary keying, deterministic (day, model) ordering, summary aggregation, and
Zen-wait independence of the pure aggregation.

* Port upstream 0.48.0: parse CommandCode rolling windows + GOAT plan

F12 (steipete#2630): parse windowLimits.fiveHour/weekly (root or nested in
credits) into primary/secondary rate windows with number-or-string
coercion and epoch-s/epoch-ms/ISO-8601 resetAt handling; monthly grant
moves to tertiary and uses the plan catalog for its total.

F13 (steipete#2706): add the individual-goat plan (0/mo) to the plan catalog,
recognize the new commandcode_prod_.session_token cookie names ahead of
the legacy better-auth family (bare tokens keep the legacy name), and
narrow pasted headers to the session cookie per upstream
CommandCodeCookieHeader.override.

Regressions: both upstream window-limits fixtures copied verbatim;
cookie priority/case-folding/family tests; plan catalog + login-method
and monthly-window mapping tests.

* Port upstream 0.48.0: derive OpenRouter key meter from server remaining

F14 (steipete#2612): the key-limit meter and its left/used text now come from the
server-reported current-period limit_remaining (clamped to [0, limit]:
negative reads exhausted, above-limit reads 0%) instead of lifetime usage.
Without a server remaining, fall back to the period usage matching the
declared limit_reset window, then cumulative usage; keys without any
usable quota source keep the meter hidden. Adds wire decoding for
limit_remaining/limit_reset.

* Port upstream 0.48.0: preserve ZoomMate browser cookie scope

F16 (steipete#2627): browser-imported cookies keep their raw host scope instead of
being merged into one header reused on host failover. Chromium/Firefox
host keys carry the scope (leading '.' = parent-domain), so each
destination host (ai.zoom.us / zoommate.zoom.us) now gets its own header
via RFC 6265 domain matching: parent-domain sessions reach both API
hosts, host-only cookies never leak onto sibling hosts, non-API hosts
are never destinations.

Upstream fixture issue-2507-cookie-scope.json copied verbatim; regression
tests cover the per-host partition, suffix-attacker rejection, empty
domains, and the hostOnly/domain matrix.

* Port upstream 0.48.0: decode Copilot AI credits counter

A15 (steipete#2593/steipete#2613): quota snapshots now decode credits_used (number or
string) for token-billed seats. The absolute counter stays off the
rate-window path — it surfaces as an informational extra window
(ai-credits), matching the existing snapshot/bridge/diagnostics pipeline
without inventing a fake quota denominator.

Upstream carriesCreditsCounter parity: zero-entitlement placeholder
snapshots still yield their counter, so a business seat with no
renderable quota window no longer blanks out — its snapshot becomes an
informational credits row (preferred premium-, then chat-classified
entries) instead of the previous hard error. Seats without any counter
keep the existing token-billing error.

* Port upstream 0.48.0: classify Claude OAuth refresh failures, terminal backoff

F3 (steipete#2650): on Windows the credential file is readable, so the upstream
touch-completes-but-unreadable state has no equivalent — the matching
provably-unrecoverable-by-retry state is the refresh endpoint rejecting
the stored refresh token (400/401/403). Those now classify as terminal:
a 5-minute per-source backoff (upstream defaultCooldownInterval) instead
of a doomed grant on every poll, and an honest re-login message without
the useless retry tail. Transient failures (network, 429, 5xx) keep a
short 20-second cooldown (upstream shortCooldownInterval) so recovery
still lands quickly. Successful refresh clears state; a re-login via the
shared credentials file is adopted past any backoff as before.

Regressions: classification matrix, long/short backoff gating + purge,
distinct terminal/cooldown user messages.

* Port upstream 0.48.0: bounded Zen balance wait in OpenCode Go usage reads

F15 (steipete#2583): the Zen balance now joins usage reads with an explicit
policy bound measured from task creation — CLI usage/serve /usage reads
(requires_optional_usage_completeness, new FetchContext field) join for
the remainder of the 5 s optional-balance budget; background/UI/guard/
diagnose hooks keep the 250 ms grace, so a slow subscription fetch can
never stack a second full wait. Local (SQLite) reads gain the same
optional enrichment as web reads, matching the upstream local strategy.

The balance fetch itself is the upstream chain: dashboard page parse
first, dedicated billing server-fn (raw 1e-8 USD behind a customerID
marker, RSC-fragment tolerant) as fallback, 25 ms start delay, bounded
per-request, and abandoned-over-budget tasks are aborted instead of
leaking. A zero-cost balance embedded in the usage page still wins
without any extra request.

A14 (per-model cost breakdown by day) intentionally NOT in this commit.

* Port upstream 0.48.0: Claude OAuth refresh gate + serve/opencodego fixes (M1-M4)

M1: from_http_status terminal iff (400|401) AND OAuth error == invalid_grant
    (case-insensitive). 403 and 400/401-without-invalid_grant are transient.
M2: terminal gate indefinite until credential fingerprint changes or success
    clears; transient base 5min flat; eliminates repeated dead-grant retries.
M3: serve /usage sets requires_optional_usage_completeness false (background
    poll grace); CLI usage remains true.
M4: abort spawned Zen balance task on usage-page/parse error before early
    return.

* Port upstream 0.48.0: Kimi/GLM/z.ai China routing (WS5)

- Kimi Desktop monthly membership pool enrichment: read-only WAL-safe
  kimi-auth token reader for the Electron Chromium store (%APPDATA%/kimi-
  desktop), AES-256-GCM via existing browser cookie crypto; Code API + CLI
  snapshots merged with Monthly + Code 7-day membership windows (steipete#2622/A10).
- Cookie Source Off disables Kimi Desktop + browser import (manual cookie
  headers keep working) (steipete#2623/A12).
- Moonshot/Kimi Open Platform: MOONSHOT_REGION + region-bound
  CODEXBAR_MOONSHOT_API_KEY(_REGION) binding so CN/intl keys stay on their
  issuing hosts; provider renamed per upstream (steipete#2621/A11).
- GLM Coding Plan: 5-hour TOKENS_LIMIT window is primary, weekly secondary,
  MCP rendered as a separate named window; plan name falls back to
  plan/plan_type/packageName/level (steipete#2621/A11).
- z.ai region routing: BIGMODEL/ZHIPU(ZHIPUAI)/GLM env aliases and
  coding-relay key files only for BigModel CN; canonical cross-region
  endpoint overrides rejected before bearer auth (steipete#2623/A12).
- Shared WAL-safe read-only SQLite helper (core::sqlite) replacing the
  OpenCode Go-local copy, reused by the Kimi Desktop reader (steipete#2544 pattern).

* Port upstream 0.48.0: unify Pi-family (pi + OMP) agent sessions (WS6)

- One dialect-aware scanner: live pi/OMP process detection (basename +
  bun-shim handling, helper filtering), session-jsonl correlation per CWD,
  PID-only rows when no transcript can be attributed, upstream OMP profile /
  PI_CONFIG_DIR / --session-dir / settings.json root resolution with the
  same fail-closed validation, upstream fixtures copied verbatim (steipete#2626/A13).
- Wire shape: AgentSession gains optional dialect + sessionName; --json
  stays legacy Codex/Claude-only (v1) while --json-v2 emits the complete
  array; SSH session discovery negotiates --json-v2 with --json fallback.
- Local scanner adopts the shared bounded directory budget; provider
  labels Pi/OMP in the sessions UI (bridge DTO + locale keys).
- Remote plumbing (RemoteSessionFetcher) moved to agent_sessions/remote.rs
  and the Pi scanner split into pi_family/{mod,parser,roots} to keep every
  file under 1000 lines.

* fix(dashboard): enable waiter notification under the decision guard (F1)

Corrects the lost-wakeup fix in 87ff071: await_build re-acquired the slot
mutex AFTER the decision guard was released, so a build completing in that
gap could fire notify_waiters (and swap the slot) before the waiter ever
registered — the waiter then slept forever on an already-fired Notify.

Now the waiter constructs and enable()s an OwnedNotified (which owns the
Arc<Notify>, Send+Sync) while still holding the SAME decision mutex that
observed Slot::Building, carries the registered future out of the critical
section, and awaits it only after the guard drops. The builder can update the
slot and notify_waiters only while holding that same mutex, so registration
is provably ordered before any wakeup attempt for this build. await_build
helper removed.

Regression: completion_in_decision_window_sets_waiter_notified drives the
waiter with manual polls and forces build completion + notify_waiters into
the exact decision->await window (slot manipulated directly; zero scheduler
dependence). Existing racing/cancel/panic regressions and single-flight, TTL,
late-result, and errors-not-cached behavior unchanged.

Also repairs two FetchContext initializers in source.rs for the integrated
requires_optional_usage_completeness field (E0063 at d2a63eb; false =
FetchContext::default, no behavior change).

Verified: cargo fmt --all --check clean; cargo clippy -p codexbar
--all-targets -- -D warnings clean; 10/10 coordinator tests pass
(2 threads).

* Port upstream 0.48.0: Codex cost-scanner robustness (F1,F2,F18,F19)

WS2 — Codex cost cache and scan robustness, ported from upstream 0.48.0.

F1 (cache bounds): add CostUsageCacheBudget module with upstream's
256 MiB save / 320 MiB load / 25 000 entry caps. Load refuses to decode
artifacts above MAX_LOAD_BYTES (cheaper to rebuild bounded). Save prunes
out-of-window entries, then trims oldest in-window entries to fit the
budget, protecting partially-parsed (growing) files so append-only resume
keeps its catch-up progress.

F2 (fork catch-up resume): validate the cached resume offset is a real
line boundary (byte at offset-1 == newline) before resuming an append-only
parse. A partial trailing-line write leaves the offset mid-line; resuming
there corrupts the first record. When the check fails, fall back to a full
re-parse from zero instead of the append-only merge.

F18 (priced + unpriced Auto Review): codex-auto-review and the model-less
sentinel are now deliberately unpriced routing rows — tokens counted,
by_model row present with 0 cost, no fallback to gpt-4o rates. Add typed
ModelPricingCompleteness (Complete | Partial{unpriced_models}) to CostSummary
so the dashboard can label a partial breakdown.

F19 (overshoot contract + predecessor keys): document the save/load overshoot
contract — save may exceed MAX_FILE_BYTES up to MAX_LOAD_BYTES when protected
entries cannot be trimmed further. Predecessor-key acceptance is N/A locally
(local cache uses filename -v1 versioning, no producer-key field); documented
as a documented divergence.

* Port upstream 0.48.0: Codex windows/pricing (F5,F6,C4)

WS3 — Codex duration classification and pricing, ported from upstream 0.48.0.

F5 (duration classification 5h/weekly/30-day): centralize duration policy in
RateWindowCadence (Session/Weekly/Monthly/Unknown) with from_minutes() and
from_seconds(). Add MONTHLY_WINDOW_MINUTES (43 200) next to the existing
SESSION/WEEKLY constants. Update codex_window_role to use RateWindowCadence
so 30-day windows classify as Monthly instead of being swallowed into Weekly.

C4 (Fast cost semantics + Terra/Luna refresh): add codex_api_fast_multiplier()
(gpt-5.4/5.4-mini/5.6-sol/5.6-terra/5.6-luna → 2.0; gpt-5.5 → 2.5; else nil)
and codex_fast_cost_usd() (standard cost × multiplier with long-context guard
at 272 000 input). Wire into codex_costs::codex_cost_usd after canonical
resolution fails but before legacy gpt-4o fallback, for fast/priority model
IDs. Refresh Terra rates (2e-6/1.2e-5, long 4e-6/1.8e-5) and Luna rates
(2e-7/1.2e-6, long 4e-7/1.8e-6). Fast detection is name-based locally (upstream
uses a priority-trace SQLite DB scan — documented divergence).

* Port upstream 0.48.0: complete WS2+WS3 follow-up (A16,F6,F8,F5)

A16 (scan completeness JSON): add historyCoverageIsEstablished to CostSummary
and surface it in the CLI cost JSON as historyCoverageIsEstablished (Bool?,
null for non-Codex providers). Set from cache freshness + catch-up state so
callers know when a re-scan is pending. Provider-native-only flag is N/A
locally (no pi/OMP mirror sessions) — documented divergence.

F6 (manual reset backfill): add codex_reset_backfill in Tauri providers.rs —
backfills missing resets_at/reset_description on fresh Codex windows from the
cached snapshot when the cached reset is still future (fresh used_percent
untouched). Wired into refresh_provider before publishing so every surface
(tray, CLI, frontend) sees the backfilled reset. Implemented through the
existing provider refresh abstraction, not a generic trait hook.

F8 (cached spend during refresh): add refreshing + stale_updated_at to
UsageSpendRow (backward-compatible optional fields). When the Codex cache
was pruned for budget (previous_report set), the spend row shows the stale
timestamp and refreshing indicator so the UI can show old data while a
re-scan rebuilds the artifact. Frontend UsageSpendTab renders the indicator
and uses the UsageSpendRefreshing locale key.

F5 (monthly cadence wiring): add Monthly to WindowRole (managed accounts)
and wire monthly through the ambient provider (normalize_array_windows
4-tuple routes monthly to UsageSnapshot.tertiary). Add tertiary_label to
bridge ProviderUsageSnapshot (duration-cadence label via RateWindowCadence).
Frontend MenuCard.tsx uses tertiaryLabel with monthly localization (ProviderMonthly).
Tray provider_status_label for Codex picks first non-informational lane
(session → weekly → monthly). CLI usage.rs appends a monthly lane line with
RateWindowCadence-based label. Test added for Monthly role classification.

* Port upstream 0.48.0: fix fmt/clippy integration issues

- Remove untracked package-lock.json (pnpm repo, npm artifacts incompatible)
- cargo fmt --all: bridge.rs, providers.rs, usage_spend.rs, tray_bridge.rs
- clippy: move constant-size budget assertion into const block (assertions_on_constants)
- clippy: add TestCache type alias to simplify test helper return type (type_complexity)
- clippy: collapse nested if-let in tray_bridge.rs codex_lane_headline_window
  using let-chains (let-chains stable since 2025 edition)

* Port upstream 0.48.0: C4 centralize fast suffix stripping (audit fix)

Extract codex_fast_base_model() that strips -fast/-priority suffixes.
Both codex_api_fast_multiplier() and codex_fast_cost_usd() now use it
so the original suffix does not leak into the Standard base lookup.
Previously codex_fast_cost_usd passed the original model name to
codex_cost_usd, which failed for suffixed IDs like gpt-5.5-fast.

Tests added:
- test_codex_fast_cost_usd_suffixed_models_resolve_to_base:
  gpt-5.5-fast -> base gpt-5.5 × 2.5, gpt-5.6-sol-priority -> base
  gpt-5.6-sol × 2.0
- test_codex_fast_base_model_unsuffixed: unsuffixed and unknown
  models resolve to themselves.

* Port upstream 0.48.0: F8 clear previous_report after full scan (audit fix)

A completed full scan rebuilds the cache for the current window, so any
prior catch-up state is no longer pending. Clear previous_report before
save_cache so the persisted artifact no longer signals stale/refreshing.
Previously previous_report was set during save-time budget pruning but
never cleared, causing a permanent Refreshing indicator.

Test: previous_report_clears_after_successful_full_scan — first scan
clears, inject previous_report to simulate trim, full scan clears it.

* Port upstream 0.48.0: A16/F18 expose coverage+completeness in CLI JSON (audit fix)

A16 historyCoverageIsEstablished and F18 modelPricingCompleteness were
added to CostSummary in the prior follow-up but never wired into the CLI
cost JSON or text output. Now:
- JSON emits historyCoverageIsEstablished (bool for Codex, null otherwise)
  and modelPricingCompleteness ("complete" or {partial:{unpriced_models}}).
- Text output labels partial pricing and partial coverage when present.
- --provider-native-only flag added, maps to CostScanOptions::include_pi_sessions
  = false, excluding pi/OMP session mirrors. Documented divergence: no
  pi/OMP mirror sessions on this Windows build so the flag is accepted but
  has no observable effect locally.

Tests: json_output_emits_a16_and_f18_fields, json_output_a16_null_for_non_codex,
provider_native_only_flag_default_false.

* Port upstream 0.48.0: F19 refuse oversized cache + fix trim double-subtraction (audit fix)

F19: save_cache now checks the encoded JSON length against MAX_LOAD_BYTES
before persisting. If the artifact still exceeds the load budget after
pruning+trimming (e.g. a single protected entry alone exceeds the limit),
the save is refused — no persist/refuse/rebuild loop. Extracted as
CostUsageCacheBudget::should_refuse_persistence() pure helper for
testability.

Trim double-subtraction fix: trim_in_window_for_budget pre-subtracted
droppable[0] from the initial estimate, then subtracted it again inside
the loop — a double count. Now the initial estimate is the full
estimated_cache_bytes and the loop subtracts each candidate once.

Tests:
- should_refuse_persistence_at_and_above_limit (boundary 1024/1025)
- trim_estimate_no_double_subtraction_of_first_entry
- trim_drops_until_target_reached_then_stops
- save_cache_persists_small_codex_artifact (no false-positive)
- save_cache_refuses_non_bounded_provider_oversize (Claude gate)

* Port upstream 0.48.0: F2 boundary helper + scan-level negative regression (audit fix)

F2 (upstream 0.48.0 steipete#2648): add boundary helper tests for all edge cases
and a scan-level negative regression proving midline/truncated rewrite
forces full parse (no resume from stale offset).

Tests:
- is_line_boundary_offset_zero_returns_true (offset 0)
- is_line_boundary_offset_at_or_past_size_returns_true (EOF)
- is_line_boundary_offset_exact_newline_returns_true (valid boundary)
- is_line_boundary_offset_midline_returns_false (fall through)
- is_line_boundary_offset_missing_file_returns_false (probe fail)
- cost_scan_midline_rewrite_forces_full_parse_not_resume (scan-level)

* Port upstream 0.48.0: F5 cadence/routing/headline regression tests (audit fix)

F5 (upstream 0.48.0): boundary table for RateWindowCadence and routing
regression for normalize_array_windows + tray headline preference.

Tests:
- cadence_boundary_session_exactly_300 (Session)
- cadence_boundary_weekly_10080 (Weekly)
- cadence_boundary_below_monthly_43199_is_weekly (boundary)
- cadence_boundary_monthly_43200 (Monthly)
- cadence_from_seconds_rounding (0/neg→Unknown, 18001s→301→Unknown)
- cadence_label_keys (session/weekly/monthly/unknown)
- f5_normalize_array_routes_session_weekly_monthly_to_lanes
- f5_normalize_array_monthly_routes_to_tertiary_not_secondary
- f5_normalize_array_empty_returns_placeholder_primary
- f5_normalize_array_unknown_windows_fall_to_code_review
- f5_headline_prefers_non_informational_primary (Tauri)
- f5_headline_falls_back_to_secondary_when_primary_informational (Tauri)
- f5_headline_falls_back_to_tertiary_when_primary_and_secondary_informational (Tauri)
- f5_headline_returns_primary_when_all_informational (Tauri)

* Port upstream 0.48.0: F6 reset-backfill regression tests (audit fix)

F6 (upstream 0.48.0 UsageStore+CodexResetBackfill): regression tests for
codex_reset_backfill covering future/stale/no-cached/non-codex/existing
paths.

Tests:
- f6_backfills_future_cached_reset (future reset backfilled, used untouched)
- f6_does_not_backfill_stale_cached_reset (past reset skipped)
- f6_does_not_overwrite_existing_resets_at (fresh reset preserved)
- f6_skips_non_codex_provider (Claude skip)
- f6_skips_when_no_cached_snapshot (None cached)

Divergence: upstream weekly-confirmation exemption N/A locally — local
backfill is the observable analog (no weekly-confirmation guard exists).

* Port upstream 0.48.0: fix clippy lint in audit-followup test code (audit fix)

Fix 4 clippy -D warnings violations in the audit-followup test code:
- collapsible_if in cli/cost.rs partial pricing label
- field_assignment_outside_initializer in cli/cost.rs test
- field_assignment_outside_initializer in jsonl_scanner.rs test
- needless_borrows_for_generic_args in cost_scanner.rs test

* Port upstream 0.48.0: F19 refusal removes preexisting destination artifact (audit fix)

The prior F19 commit refused on oversized post-encode but left any existing
destination cache file in place; a stale/oversized artifact could persist
and trip the load-refusal path on the next scan, forcing an unnecessary
full rebuild from a poisoned artifact.

- save_cache now deletes the destination file on refusal (best-effort,
  mirroring the fs-delete idiom used elsewhere in core).
- Extracted save_cache_with_limit(provider, cache, cache_root, max_load_bytes)
  as a private testable helper; save_cache delegates with the production
  MAX_LOAD_BYTES const. Production limit behavior is unchanged.

Integration regressions (jsonl_scanner.rs):
- save_cache_refusal_removes_preexisting_destination_artifact: precreate
  real destination via save_cache_with_limit(usize::MAX), trigger refusal
  with limit=1, assert destination gone, no tmp artifact with content, and
  load yields empty cache (no rebuild loop).
- save_cache_at_exact_limit_is_accepted: encoded artifact at exactly the
  injected limit is persisted (boundary).
- save_cache_one_over_limit_is_refused_and_removes_destination: one byte
  over limit is refused and destination removed.

* fix: fan out dashboard build errors

* fix: pin Windows globalization timezone module

iana-time-zone resolves the Windows system zone through WinRT's
Windows.Globalization Calendar class, but nothing keeps that DLL loaded.
COM cleanup exercised by the notification sound/toast tests unloads it,
leaving windows-core's process-static factory cache pointing into an
abandoned mapping; the next get_timezone() call then access-violates
(observed under LLDB; full lib suite crashed 3/3 single-threaded).

Load Windows.Globalization.dll from System32 and pin it for the process
lifetime before any get_timezone() call, gated behind a one-shot
LazyLock so every caller waits for load+pin to settle. On pin failure
return UTC without calling iana-time-zone (an AV is uncatchable). Route
both project call sites (claude cli_reset, sub2api) through the new
crate-internal helper. No lockfile change.

* test: make globalization pin test host-independent

The self-hosted PR runner is a stripped Windows image without
registered WinRT types (ToastNotification not registered), where
pinning Windows.Globalization.dll can fail by design; the helper then
correctly falls back to UTC. Assert the environment-independent
contract instead: the pin attempt settles exactly once and every
caller observes the same decision.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

merge-risk: 🚨 auth-provider 🚨 Merging this PR could break OAuth, tokens, provider routing, model choice, or credentials. P2 Normal priority bug or improvement with limited blast radius. rating: 🦐 gold shrimp Decent PR readiness signal, but merge confidence is limited. status: ⏳ waiting on author ClawSweeper has contributor-facing work open and is waiting for author action.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant