@ai-devkit/agent-manager 0.33.1 → 0.34.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (81) hide show
  1. package/dist/__tests__/capacity/codex.test.js +2 -1
  2. package/dist/__tests__/capacity/codex.test.js.map +1 -1
  3. package/dist/__tests__/capacity/index.test.js +6 -3
  4. package/dist/__tests__/capacity/index.test.js.map +1 -1
  5. package/dist/__tests__/capacity/zai.test.js +213 -0
  6. package/dist/__tests__/capacity/zai.test.js.map +1 -0
  7. package/dist/__tests__/providers/opencode/OpenCodeAdapter.test.js +189 -0
  8. package/dist/__tests__/providers/opencode/OpenCodeAdapter.test.js.map +1 -0
  9. package/dist/__tests__/providers/opencode/OpenCodeAgentMapper.test.js +89 -0
  10. package/dist/__tests__/providers/opencode/OpenCodeAgentMapper.test.js.map +1 -0
  11. package/dist/__tests__/providers/opencode/OpenCodeSessionLocator.test.js +94 -0
  12. package/dist/__tests__/providers/opencode/OpenCodeSessionLocator.test.js.map +1 -0
  13. package/dist/__tests__/providers/opencode/OpenCodeSessionParser.test.js +147 -0
  14. package/dist/__tests__/providers/opencode/OpenCodeSessionParser.test.js.map +1 -0
  15. package/dist/adapters/index.d.ts +1 -1
  16. package/dist/adapters/index.d.ts.map +1 -1
  17. package/dist/adapters/index.js +1 -1
  18. package/dist/adapters/index.js.map +1 -1
  19. package/dist/capacity/codex.d.ts +1 -0
  20. package/dist/capacity/codex.d.ts.map +1 -1
  21. package/dist/capacity/codex.js +9 -4
  22. package/dist/capacity/codex.js.map +1 -1
  23. package/dist/capacity/index.d.ts +8 -0
  24. package/dist/capacity/index.d.ts.map +1 -1
  25. package/dist/capacity/index.js +10 -9
  26. package/dist/capacity/index.js.map +1 -1
  27. package/dist/capacity/types.d.ts +8 -0
  28. package/dist/capacity/types.d.ts.map +1 -1
  29. package/dist/capacity/types.js.map +1 -1
  30. package/dist/capacity/zai.d.ts +14 -0
  31. package/dist/capacity/zai.d.ts.map +1 -0
  32. package/dist/capacity/zai.js +203 -0
  33. package/dist/capacity/zai.js.map +1 -0
  34. package/dist/index.d.ts +3 -3
  35. package/dist/index.d.ts.map +1 -1
  36. package/dist/index.js +2 -2
  37. package/dist/index.js.map +1 -1
  38. package/dist/{adapters → providers/opencode}/OpenCodeAdapter.d.ts +5 -12
  39. package/dist/providers/opencode/OpenCodeAdapter.d.ts.map +1 -0
  40. package/dist/providers/opencode/OpenCodeAdapter.js +86 -0
  41. package/dist/providers/opencode/OpenCodeAdapter.js.map +1 -0
  42. package/dist/providers/opencode/OpenCodeAgentMapper.d.ts +11 -0
  43. package/dist/providers/opencode/OpenCodeAgentMapper.d.ts.map +1 -0
  44. package/dist/providers/opencode/OpenCodeAgentMapper.js +50 -0
  45. package/dist/providers/opencode/OpenCodeAgentMapper.js.map +1 -0
  46. package/dist/providers/opencode/OpenCodeSessionLocator.d.ts +21 -0
  47. package/dist/providers/opencode/OpenCodeSessionLocator.d.ts.map +1 -0
  48. package/dist/providers/opencode/OpenCodeSessionLocator.js +99 -0
  49. package/dist/providers/opencode/OpenCodeSessionLocator.js.map +1 -0
  50. package/dist/providers/opencode/OpenCodeSessionParser.d.ts +18 -0
  51. package/dist/providers/opencode/OpenCodeSessionParser.d.ts.map +1 -0
  52. package/dist/providers/opencode/OpenCodeSessionParser.js +104 -0
  53. package/dist/providers/opencode/OpenCodeSessionParser.js.map +1 -0
  54. package/package.json +1 -1
  55. package/src/__tests__/capacity/codex.test.ts +94 -26
  56. package/src/__tests__/capacity/fixtures/zai-auth.json +6 -0
  57. package/src/__tests__/capacity/fixtures/zai-quota-invalid.json +7 -0
  58. package/src/__tests__/capacity/fixtures/zai-quota.json +38 -0
  59. package/src/__tests__/capacity/index.test.ts +10 -3
  60. package/src/__tests__/capacity/zai.test.ts +186 -0
  61. package/src/__tests__/providers/opencode/OpenCodeAdapter.test.ts +197 -0
  62. package/src/__tests__/providers/opencode/OpenCodeAgentMapper.test.ts +83 -0
  63. package/src/__tests__/providers/opencode/OpenCodeSessionLocator.test.ts +84 -0
  64. package/src/__tests__/providers/opencode/OpenCodeSessionParser.test.ts +148 -0
  65. package/src/adapters/index.ts +1 -1
  66. package/src/capacity/codex.ts +96 -28
  67. package/src/capacity/index.ts +29 -11
  68. package/src/capacity/types.ts +8 -0
  69. package/src/capacity/zai.ts +237 -0
  70. package/src/index.ts +8 -3
  71. package/src/providers/opencode/OpenCodeAdapter.ts +112 -0
  72. package/src/providers/opencode/OpenCodeAgentMapper.ts +61 -0
  73. package/src/providers/opencode/OpenCodeSessionLocator.ts +123 -0
  74. package/src/providers/opencode/OpenCodeSessionParser.ts +145 -0
  75. package/dist/__tests__/adapters/OpenCodeAdapter.test.js +0 -636
  76. package/dist/__tests__/adapters/OpenCodeAdapter.test.js.map +0 -1
  77. package/dist/adapters/OpenCodeAdapter.d.ts.map +0 -1
  78. package/dist/adapters/OpenCodeAdapter.js +0 -289
  79. package/dist/adapters/OpenCodeAdapter.js.map +0 -1
  80. package/src/__tests__/adapters/OpenCodeAdapter.test.ts +0 -544
  81. package/src/adapters/OpenCodeAdapter.ts +0 -373
@@ -127,7 +127,8 @@ describe("tiered Codex probing", ()=>{
127
127
  });
128
128
  expect(rpc).not.toHaveBeenCalled();
129
129
  expect(result).toMatchObject({
130
- provider: "codex",
130
+ harness: "codex",
131
+ provider: "openai",
131
132
  available: "yes",
132
133
  creditsRemaining: 12.5,
133
134
  authenticated: true
@@ -1 +1 @@
1
- {"version":3,"sources":["../../../src/__tests__/capacity/codex.test.ts"],"sourcesContent":["import { describe, expect, it, vi } from \"vitest\";\nimport {\n CODEX_APP_SERVER_ARGS,\n parseUsage,\n probeCodexCapacity,\n resolveCodexAuthPath,\n toRateWindow,\n} from \"../../capacity/codex.js\";\n\nconst checkedAt = \"2026-08-20T10:00:00.000Z\";\nconst context = { installed: true, checkedAt };\n\nfunction apiUsage(overrides: Record<string, unknown> = {}) {\n return {\n rate_limit: {\n primary_window: { used_percent: 20, limit_window_seconds: 18_000, reset_at: 1_787_220_000 },\n secondary_window: {\n used_percent: 60,\n limit_window_seconds: 604_800,\n reset_at: 1_787_824_800,\n },\n ...overrides,\n },\n credits: { balance: 12.5 },\n additional_rate_limits: [\n {\n limit_name: \"reviews\",\n rate_limit: {\n primary_window: {\n used_percent: 10,\n limit_window_seconds: 3_600,\n reset_at: 1_787_220_000,\n },\n },\n },\n ],\n };\n}\n\ndescribe(\"Codex auth resolution\", () => {\n it(\"uses CODEX_HOME before HOME\", () => {\n expect(resolveCodexAuthPath({ CODEX_HOME: \"/custom/codex\", HOME: \"/users/test\" })).toBe(\n \"/custom/codex/auth.json\",\n );\n });\n\n it(\"falls back to ~/.codex/auth.json\", () => {\n expect(resolveCodexAuthPath({ HOME: \"/users/test\" })).toBe(\"/users/test/.codex/auth.json\");\n });\n});\n\ndescribe(\"Codex API usage mapping\", () => {\n it(\"converts an API window without treating missing data as zero\", () => {\n expect(\n toRateWindow(\n { used_percent: 25, limit_window_seconds: 18_000, reset_at: 1_787_220_000 },\n \"session\",\n \"Session\",\n ),\n ).toEqual({\n id: \"session\",\n label: \"Session\",\n durationMinutes: 300,\n usedPercent: 25,\n resetsAt: \"2026-08-20T10:00:00.000Z\",\n });\n expect(toRateWindow({}, \"session\", \"Session\")).toMatchObject({ usedPercent: null });\n });\n\n it(\"maps session, weekly, credits, and extra limits\", () => {\n const snapshot = parseUsage(apiUsage(), \"pat\");\n expect(snapshot).toMatchObject({ source: \"pat\", creditsRemaining: 12.5 });\n expect(snapshot.windows).toEqual([\n expect.objectContaining({ id: \"session\", durationMinutes: 300 }),\n expect.objectContaining({ id: \"weekly\", durationMinutes: 10080 }),\n expect.objectContaining({ id: \"reviews:primary\", durationMinutes: 60 }),\n ]);\n });\n\n it(\"represents missing limits as unavailable rather than zero\", () => {\n const snapshot = parseUsage({ credits: {} }, \"oauth\");\n expect(snapshot.windows).toEqual([]);\n expect(snapshot.creditsRemaining).toBeNull();\n });\n});\n\ndescribe(\"tiered Codex probing\", () => {\n it(\"selects PAT, calls whoami then usage, and never invokes the CLI\", async () => {\n const fetch = vi\n .fn()\n .mockResolvedValueOnce(\n new Response(JSON.stringify({ chatgpt_account_id: \"acct-1\" }), { status: 200 }),\n )\n .mockResolvedValueOnce(new Response(JSON.stringify(apiUsage()), { status: 200 }));\n const rpc = vi.fn();\n const result = await probeCodexCapacity({\n ...context,\n readFile: async () =>\n JSON.stringify({\n personal_access_token: \"pat-secret\",\n tokens: { access_token: \"ignored-oauth\", account_id: \"ignored-account\" },\n }),\n fetch,\n rpc,\n });\n expect(fetch).toHaveBeenCalledTimes(2);\n expect(fetch.mock.calls[0][0]).toBe(\n \"https://auth.openai.com/api/accounts/v1/user-auth-credential/whoami\",\n );\n expect(fetch.mock.calls[1][0]).toBe(\"https://chatgpt.com/backend-api/wham/usage\");\n expect(fetch.mock.calls[1][1].headers).toMatchObject({\n Authorization: \"Bearer pat-secret\",\n \"ChatGPT-Account-Id\": \"acct-1\",\n });\n expect(rpc).not.toHaveBeenCalled();\n expect(result).toMatchObject({\n provider: \"codex\",\n available: \"yes\",\n creditsRemaining: 12.5,\n authenticated: true,\n });\n expect(result.windows.map((window) => window.id)).toEqual([\n \"session\",\n \"weekly\",\n \"reviews:primary\",\n ]);\n });\n\n it(\"selects a fresh OAuth token without calling whoami\", async () => {\n const fetch = vi\n .fn()\n .mockResolvedValue(new Response(JSON.stringify(apiUsage()), { status: 200 }));\n const result = await probeCodexCapacity({\n ...context,\n readFile: async () =>\n JSON.stringify({\n tokens: { access_token: \"oauth-secret\", account_id: \"acct-2\", expires_at: 1_800_000_000 },\n }),\n fetch,\n now: () => new Date(\"2026-08-20T10:00:00.000Z\"),\n });\n expect(fetch).toHaveBeenCalledOnce();\n expect(fetch.mock.calls[0][1].headers).toMatchObject({\n Authorization: \"Bearer oauth-secret\",\n \"ChatGPT-Account-Id\": \"acct-2\",\n });\n expect(result.available).toBe(\"yes\");\n });\n\n it.each([\n [\n \"missing auth file\",\n async () => {\n throw Object.assign(new Error(\"missing\"), { code: \"ENOENT\" });\n },\n ],\n [\n \"stale OAuth token\",\n async () =>\n JSON.stringify({\n tokens: { access_token: \"stale-secret\", account_id: \"acct\", expires_at: 1 },\n }),\n ],\n [\n \"OAuth 401\",\n async () =>\n JSON.stringify({\n tokens: { access_token: \"oauth-secret\", account_id: \"acct\", expires_at: 1_800_000_000 },\n }),\n ],\n ])(\"falls back to the CLI for %s\", async (name, readFile) => {\n const fetch = vi\n .fn()\n .mockResolvedValue(new Response(\"\", { status: name === \"OAuth 401\" ? 401 : 200 }));\n const rpc = vi.fn(async () => ({\n rateLimits: {\n rateLimits: { primary: { usedPercent: 5, windowDurationMins: 300, resetsAt: null } },\n },\n account: { account: { type: \"chatgpt\" } },\n }));\n const result = await probeCodexCapacity({\n ...context,\n readFile,\n fetch,\n rpc,\n now: () => new Date(checkedAt),\n });\n expect(rpc).toHaveBeenCalledOnce();\n expect(result.windows).toHaveLength(1);\n });\n\n it(\"falls back to CLI if PAT requests fail\", async () => {\n const rpc = vi.fn(async () => ({ rateLimits: {}, account: { account: null } }));\n const result = await probeCodexCapacity({\n ...context,\n readFile: async () => JSON.stringify({ personal_access_token: \"pat-secret\" }),\n fetch: vi.fn().mockRejectedValue(new Error(\"network failure pat-secret\")),\n rpc,\n });\n expect(rpc).toHaveBeenCalledOnce();\n expect(result.available).toBe(\"unknown\");\n });\n\n it(\"tries fresh OAuth after a PAT request fails\", async () => {\n const fetch = vi\n .fn()\n .mockRejectedValueOnce(new Error(\"PAT failed\"))\n .mockResolvedValueOnce(new Response(JSON.stringify(apiUsage()), { status: 200 }));\n const rpc = vi.fn();\n const result = await probeCodexCapacity({\n ...context,\n readFile: async () =>\n JSON.stringify({\n personal_access_token: \"pat-secret\",\n tokens: { access_token: \"oauth-secret\", account_id: \"acct\", expires_at: 1_800_000_000 },\n }),\n fetch,\n rpc,\n now: () => new Date(checkedAt),\n });\n expect(fetch).toHaveBeenCalledTimes(2);\n expect(result.available).toBe(\"yes\");\n expect(rpc).not.toHaveBeenCalled();\n });\n\n it(\"uses hardened read-only app-server arguments and both account methods\", async () => {\n const rpc = vi.fn(async () => ({\n rateLimits: {\n rateLimits: { primary: { usedPercent: 5, windowDurationMins: 300, resetsAt: null } },\n },\n account: { account: { type: \"chatgpt\" } },\n }));\n await probeCodexCapacity({ ...context, readFile: async () => \"{}\", rpc });\n const messages = rpc.mock.calls[0][0];\n expect(messages.map((message) => message.method)).toEqual([\n \"initialize\",\n \"initialized\",\n \"account/rateLimits/read\",\n \"account/read\",\n ]);\n expect(JSON.stringify(messages)).not.toMatch(/prompt|turn\\/start/);\n expect(CODEX_APP_SERVER_ARGS).toEqual([\"-s\", \"read-only\", \"-a\", \"untrusted\", \"app-server\"]);\n });\n\n it(\"uses account/read to distinguish logged-out CLI state\", async () => {\n const result = await probeCodexCapacity({\n ...context,\n readFile: async () => \"{}\",\n rpc: async () => ({ rateLimits: {}, account: { account: null } }),\n });\n expect(result).toMatchObject({ authenticated: false, available: \"unknown\" });\n });\n\n it(\"never exposes tokens or raw auth content through failures\", async () => {\n const secrets = [\"pat-secret-value\", \"oauth-secret-value\", \"refresh-secret-value\"];\n const result = await probeCodexCapacity({\n ...context,\n readFile: async () =>\n JSON.stringify({\n personal_access_token: secrets[0],\n tokens: { access_token: secrets[1], refresh_token: secrets[2] },\n }),\n fetch: vi.fn().mockRejectedValue(new Error(secrets.join(\" \"))),\n rpc: async () => {\n throw new Error(secrets.join(\" \"));\n },\n });\n const output = JSON.stringify(result);\n for (const secret of secrets) expect(output).not.toContain(secret);\n });\n});\n"],"names":["describe","expect","it","vi","CODEX_APP_SERVER_ARGS","parseUsage","probeCodexCapacity","resolveCodexAuthPath","toRateWindow","checkedAt","context","installed","apiUsage","overrides","rate_limit","primary_window","used_percent","limit_window_seconds","reset_at","secondary_window","credits","balance","additional_rate_limits","limit_name","CODEX_HOME","HOME","toBe","toEqual","id","label","durationMinutes","usedPercent","resetsAt","toMatchObject","snapshot","source","creditsRemaining","windows","objectContaining","toBeNull","fetch","fn","mockResolvedValueOnce","Response","JSON","stringify","chatgpt_account_id","status","rpc","result","readFile","personal_access_token","tokens","access_token","account_id","toHaveBeenCalledTimes","mock","calls","headers","Authorization","not","toHaveBeenCalled","provider","available","authenticated","map","window","mockResolvedValue","expires_at","now","Date","toHaveBeenCalledOnce","each","Object","assign","Error","code","name","rateLimits","primary","windowDurationMins","account","type","toHaveLength","mockRejectedValue","mockRejectedValueOnce","messages","message","method","toMatch","secrets","refresh_token","join","output","secret","toContain"],"mappings":"AAAA,SAASA,QAAQ,EAAEC,MAAM,EAAEC,EAAE,EAAEC,EAAE,QAAQ,SAAS;AAClD,SACEC,qBAAqB,EACrBC,UAAU,EACVC,kBAAkB,EAClBC,oBAAoB,EACpBC,YAAY,QACP,0BAA0B;AAEjC,MAAMC,YAAY;AAClB,MAAMC,UAAU;IAAEC,WAAW;IAAMF;AAAU;AAE7C,SAASG,SAASC,YAAqC,CAAC,CAAC;IACvD,OAAO;QACLC,YAAY;YACVC,gBAAgB;gBAAEC,cAAc;gBAAIC,sBAAsB;gBAAQC,UAAU;YAAc;YAC1FC,kBAAkB;gBAChBH,cAAc;gBACdC,sBAAsB;gBACtBC,UAAU;YACZ;YACA,GAAGL,SAAS;QACd;QACAO,SAAS;YAAEC,SAAS;QAAK;QACzBC,wBAAwB;YACtB;gBACEC,YAAY;gBACZT,YAAY;oBACVC,gBAAgB;wBACdC,cAAc;wBACdC,sBAAsB;wBACtBC,UAAU;oBACZ;gBACF;YACF;SACD;IACH;AACF;AAEAlB,SAAS,yBAAyB;IAChCE,GAAG,+BAA+B;QAChCD,OAAOM,qBAAqB;YAAEiB,YAAY;YAAiBC,MAAM;QAAc,IAAIC,IAAI,CACrF;IAEJ;IAEAxB,GAAG,oCAAoC;QACrCD,OAAOM,qBAAqB;YAAEkB,MAAM;QAAc,IAAIC,IAAI,CAAC;IAC7D;AACF;AAEA1B,SAAS,2BAA2B;IAClCE,GAAG,gEAAgE;QACjED,OACEO,aACE;YAAEQ,cAAc;YAAIC,sBAAsB;YAAQC,UAAU;QAAc,GAC1E,WACA,YAEFS,OAAO,CAAC;YACRC,IAAI;YACJC,OAAO;YACPC,iBAAiB;YACjBC,aAAa;YACbC,UAAU;QACZ;QACA/B,OAAOO,aAAa,CAAC,GAAG,WAAW,YAAYyB,aAAa,CAAC;YAAEF,aAAa;QAAK;IACnF;IAEA7B,GAAG,mDAAmD;QACpD,MAAMgC,WAAW7B,WAAWO,YAAY;QACxCX,OAAOiC,UAAUD,aAAa,CAAC;YAAEE,QAAQ;YAAOC,kBAAkB;QAAK;QACvEnC,OAAOiC,SAASG,OAAO,EAAEV,OAAO,CAAC;YAC/B1B,OAAOqC,gBAAgB,CAAC;gBAAEV,IAAI;gBAAWE,iBAAiB;YAAI;YAC9D7B,OAAOqC,gBAAgB,CAAC;gBAAEV,IAAI;gBAAUE,iBAAiB;YAAM;YAC/D7B,OAAOqC,gBAAgB,CAAC;gBAAEV,IAAI;gBAAmBE,iBAAiB;YAAG;SACtE;IACH;IAEA5B,GAAG,6DAA6D;QAC9D,MAAMgC,WAAW7B,WAAW;YAAEe,SAAS,CAAC;QAAE,GAAG;QAC7CnB,OAAOiC,SAASG,OAAO,EAAEV,OAAO,CAAC,EAAE;QACnC1B,OAAOiC,SAASE,gBAAgB,EAAEG,QAAQ;IAC5C;AACF;AAEAvC,SAAS,wBAAwB;IAC/BE,GAAG,mEAAmE;QACpE,MAAMsC,QAAQrC,GACXsC,EAAE,GACFC,qBAAqB,CACpB,IAAIC,SAASC,KAAKC,SAAS,CAAC;YAAEC,oBAAoB;QAAS,IAAI;YAAEC,QAAQ;QAAI,IAE9EL,qBAAqB,CAAC,IAAIC,SAASC,KAAKC,SAAS,CAACjC,aAAa;YAAEmC,QAAQ;QAAI;QAChF,MAAMC,MAAM7C,GAAGsC,EAAE;QACjB,MAAMQ,SAAS,MAAM3C,mBAAmB;YACtC,GAAGI,OAAO;YACVwC,UAAU,UACRN,KAAKC,SAAS,CAAC;oBACbM,uBAAuB;oBACvBC,QAAQ;wBAAEC,cAAc;wBAAiBC,YAAY;oBAAkB;gBACzE;YACFd;YACAQ;QACF;QACA/C,OAAOuC,OAAOe,qBAAqB,CAAC;QACpCtD,OAAOuC,MAAMgB,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE,EAAE/B,IAAI,CACjC;QAEFzB,OAAOuC,MAAMgB,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE,EAAE/B,IAAI,CAAC;QACpCzB,OAAOuC,MAAMgB,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE,CAACC,OAAO,EAAEzB,aAAa,CAAC;YACnD0B,eAAe;YACf,sBAAsB;QACxB;QACA1D,OAAO+C,KAAKY,GAAG,CAACC,gBAAgB;QAChC5D,OAAOgD,QAAQhB,aAAa,CAAC;YAC3B6B,UAAU;YACVC,WAAW;YACX3B,kBAAkB;YAClB4B,eAAe;QACjB;QACA/D,OAAOgD,OAAOZ,OAAO,CAAC4B,GAAG,CAAC,CAACC,SAAWA,OAAOtC,EAAE,GAAGD,OAAO,CAAC;YACxD;YACA;YACA;SACD;IACH;IAEAzB,GAAG,sDAAsD;QACvD,MAAMsC,QAAQrC,GACXsC,EAAE,GACF0B,iBAAiB,CAAC,IAAIxB,SAASC,KAAKC,SAAS,CAACjC,aAAa;YAAEmC,QAAQ;QAAI;QAC5E,MAAME,SAAS,MAAM3C,mBAAmB;YACtC,GAAGI,OAAO;YACVwC,UAAU,UACRN,KAAKC,SAAS,CAAC;oBACbO,QAAQ;wBAAEC,cAAc;wBAAgBC,YAAY;wBAAUc,YAAY;oBAAc;gBAC1F;YACF5B;YACA6B,KAAK,IAAM,IAAIC,KAAK;QACtB;QACArE,OAAOuC,OAAO+B,oBAAoB;QAClCtE,OAAOuC,MAAMgB,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE,CAACC,OAAO,EAAEzB,aAAa,CAAC;YACnD0B,eAAe;YACf,sBAAsB;QACxB;QACA1D,OAAOgD,OAAOc,SAAS,EAAErC,IAAI,CAAC;IAChC;IAEAxB,GAAGsE,IAAI,CAAC;QACN;YACE;YACA;gBACE,MAAMC,OAAOC,MAAM,CAAC,IAAIC,MAAM,YAAY;oBAAEC,MAAM;gBAAS;YAC7D;SACD;QACD;YACE;YACA,UACEhC,KAAKC,SAAS,CAAC;oBACbO,QAAQ;wBAAEC,cAAc;wBAAgBC,YAAY;wBAAQc,YAAY;oBAAE;gBAC5E;SACH;QACD;YACE;YACA,UACExB,KAAKC,SAAS,CAAC;oBACbO,QAAQ;wBAAEC,cAAc;wBAAgBC,YAAY;wBAAQc,YAAY;oBAAc;gBACxF;SACH;KACF,EAAE,gCAAgC,OAAOS,MAAM3B;QAC9C,MAAMV,QAAQrC,GACXsC,EAAE,GACF0B,iBAAiB,CAAC,IAAIxB,SAAS,IAAI;YAAEI,QAAQ8B,SAAS,cAAc,MAAM;QAAI;QACjF,MAAM7B,MAAM7C,GAAGsC,EAAE,CAAC,UAAa,CAAA;gBAC7BqC,YAAY;oBACVA,YAAY;wBAAEC,SAAS;4BAAEhD,aAAa;4BAAGiD,oBAAoB;4BAAKhD,UAAU;wBAAK;oBAAE;gBACrF;gBACAiD,SAAS;oBAAEA,SAAS;wBAAEC,MAAM;oBAAU;gBAAE;YAC1C,CAAA;QACA,MAAMjC,SAAS,MAAM3C,mBAAmB;YACtC,GAAGI,OAAO;YACVwC;YACAV;YACAQ;YACAqB,KAAK,IAAM,IAAIC,KAAK7D;QACtB;QACAR,OAAO+C,KAAKuB,oBAAoB;QAChCtE,OAAOgD,OAAOZ,OAAO,EAAE8C,YAAY,CAAC;IACtC;IAEAjF,GAAG,0CAA0C;QAC3C,MAAM8C,MAAM7C,GAAGsC,EAAE,CAAC,UAAa,CAAA;gBAAEqC,YAAY,CAAC;gBAAGG,SAAS;oBAAEA,SAAS;gBAAK;YAAE,CAAA;QAC5E,MAAMhC,SAAS,MAAM3C,mBAAmB;YACtC,GAAGI,OAAO;YACVwC,UAAU,UAAYN,KAAKC,SAAS,CAAC;oBAAEM,uBAAuB;gBAAa;YAC3EX,OAAOrC,GAAGsC,EAAE,GAAG2C,iBAAiB,CAAC,IAAIT,MAAM;YAC3C3B;QACF;QACA/C,OAAO+C,KAAKuB,oBAAoB;QAChCtE,OAAOgD,OAAOc,SAAS,EAAErC,IAAI,CAAC;IAChC;IAEAxB,GAAG,+CAA+C;QAChD,MAAMsC,QAAQrC,GACXsC,EAAE,GACF4C,qBAAqB,CAAC,IAAIV,MAAM,eAChCjC,qBAAqB,CAAC,IAAIC,SAASC,KAAKC,SAAS,CAACjC,aAAa;YAAEmC,QAAQ;QAAI;QAChF,MAAMC,MAAM7C,GAAGsC,EAAE;QACjB,MAAMQ,SAAS,MAAM3C,mBAAmB;YACtC,GAAGI,OAAO;YACVwC,UAAU,UACRN,KAAKC,SAAS,CAAC;oBACbM,uBAAuB;oBACvBC,QAAQ;wBAAEC,cAAc;wBAAgBC,YAAY;wBAAQc,YAAY;oBAAc;gBACxF;YACF5B;YACAQ;YACAqB,KAAK,IAAM,IAAIC,KAAK7D;QACtB;QACAR,OAAOuC,OAAOe,qBAAqB,CAAC;QACpCtD,OAAOgD,OAAOc,SAAS,EAAErC,IAAI,CAAC;QAC9BzB,OAAO+C,KAAKY,GAAG,CAACC,gBAAgB;IAClC;IAEA3D,GAAG,yEAAyE;QAC1E,MAAM8C,MAAM7C,GAAGsC,EAAE,CAAC,UAAa,CAAA;gBAC7BqC,YAAY;oBACVA,YAAY;wBAAEC,SAAS;4BAAEhD,aAAa;4BAAGiD,oBAAoB;4BAAKhD,UAAU;wBAAK;oBAAE;gBACrF;gBACAiD,SAAS;oBAAEA,SAAS;wBAAEC,MAAM;oBAAU;gBAAE;YAC1C,CAAA;QACA,MAAM5E,mBAAmB;YAAE,GAAGI,OAAO;YAAEwC,UAAU,UAAY;YAAMF;QAAI;QACvE,MAAMsC,WAAWtC,IAAIQ,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE;QACrCxD,OAAOqF,SAASrB,GAAG,CAAC,CAACsB,UAAYA,QAAQC,MAAM,GAAG7D,OAAO,CAAC;YACxD;YACA;YACA;YACA;SACD;QACD1B,OAAO2C,KAAKC,SAAS,CAACyC,WAAW1B,GAAG,CAAC6B,OAAO,CAAC;QAC7CxF,OAAOG,uBAAuBuB,OAAO,CAAC;YAAC;YAAM;YAAa;YAAM;YAAa;SAAa;IAC5F;IAEAzB,GAAG,yDAAyD;QAC1D,MAAM+C,SAAS,MAAM3C,mBAAmB;YACtC,GAAGI,OAAO;YACVwC,UAAU,UAAY;YACtBF,KAAK,UAAa,CAAA;oBAAE8B,YAAY,CAAC;oBAAGG,SAAS;wBAAEA,SAAS;oBAAK;gBAAE,CAAA;QACjE;QACAhF,OAAOgD,QAAQhB,aAAa,CAAC;YAAE+B,eAAe;YAAOD,WAAW;QAAU;IAC5E;IAEA7D,GAAG,6DAA6D;QAC9D,MAAMwF,UAAU;YAAC;YAAoB;YAAsB;SAAuB;QAClF,MAAMzC,SAAS,MAAM3C,mBAAmB;YACtC,GAAGI,OAAO;YACVwC,UAAU,UACRN,KAAKC,SAAS,CAAC;oBACbM,uBAAuBuC,OAAO,CAAC,EAAE;oBACjCtC,QAAQ;wBAAEC,cAAcqC,OAAO,CAAC,EAAE;wBAAEC,eAAeD,OAAO,CAAC,EAAE;oBAAC;gBAChE;YACFlD,OAAOrC,GAAGsC,EAAE,GAAG2C,iBAAiB,CAAC,IAAIT,MAAMe,QAAQE,IAAI,CAAC;YACxD5C,KAAK;gBACH,MAAM,IAAI2B,MAAMe,QAAQE,IAAI,CAAC;YAC/B;QACF;QACA,MAAMC,SAASjD,KAAKC,SAAS,CAACI;QAC9B,KAAK,MAAM6C,UAAUJ,QAASzF,OAAO4F,QAAQjC,GAAG,CAACmC,SAAS,CAACD;IAC7D;AACF"}
1
+ {"version":3,"sources":["../../../src/__tests__/capacity/codex.test.ts"],"sourcesContent":["import { describe, expect, it, vi } from \"vitest\";\nimport {\n CODEX_APP_SERVER_ARGS,\n parseUsage,\n probeCodexCapacity,\n resolveCodexAuthPath,\n toRateWindow,\n} from \"../../capacity/codex.js\";\n\nconst checkedAt = \"2026-08-20T10:00:00.000Z\";\nconst context = { installed: true, checkedAt };\n\nfunction apiUsage(overrides: Record<string, unknown> = {}) {\n return {\n rate_limit: {\n primary_window: {\n used_percent: 20,\n limit_window_seconds: 18_000,\n reset_at: 1_787_220_000,\n },\n secondary_window: {\n used_percent: 60,\n limit_window_seconds: 604_800,\n reset_at: 1_787_824_800,\n },\n ...overrides,\n },\n credits: { balance: 12.5 },\n additional_rate_limits: [\n {\n limit_name: \"reviews\",\n rate_limit: {\n primary_window: {\n used_percent: 10,\n limit_window_seconds: 3_600,\n reset_at: 1_787_220_000,\n },\n },\n },\n ],\n };\n}\n\ndescribe(\"Codex auth resolution\", () => {\n it(\"uses CODEX_HOME before HOME\", () => {\n expect(\n resolveCodexAuthPath({\n CODEX_HOME: \"/custom/codex\",\n HOME: \"/users/test\",\n }),\n ).toBe(\"/custom/codex/auth.json\");\n });\n\n it(\"falls back to ~/.codex/auth.json\", () => {\n expect(resolveCodexAuthPath({ HOME: \"/users/test\" })).toBe(\n \"/users/test/.codex/auth.json\",\n );\n });\n});\n\ndescribe(\"Codex API usage mapping\", () => {\n it(\"converts an API window without treating missing data as zero\", () => {\n expect(\n toRateWindow(\n {\n used_percent: 25,\n limit_window_seconds: 18_000,\n reset_at: 1_787_220_000,\n },\n \"session\",\n \"Session\",\n ),\n ).toEqual({\n id: \"session\",\n label: \"Session\",\n durationMinutes: 300,\n usedPercent: 25,\n resetsAt: \"2026-08-20T10:00:00.000Z\",\n });\n expect(toRateWindow({}, \"session\", \"Session\")).toMatchObject({\n usedPercent: null,\n });\n });\n\n it(\"maps session, weekly, credits, and extra limits\", () => {\n const snapshot = parseUsage(apiUsage(), \"pat\");\n expect(snapshot).toMatchObject({ source: \"pat\", creditsRemaining: 12.5 });\n expect(snapshot.windows).toEqual([\n expect.objectContaining({ id: \"session\", durationMinutes: 300 }),\n expect.objectContaining({ id: \"weekly\", durationMinutes: 10080 }),\n expect.objectContaining({ id: \"reviews:primary\", durationMinutes: 60 }),\n ]);\n });\n\n it(\"represents missing limits as unavailable rather than zero\", () => {\n const snapshot = parseUsage({ credits: {} }, \"oauth\");\n expect(snapshot.windows).toEqual([]);\n expect(snapshot.creditsRemaining).toBeNull();\n });\n});\n\ndescribe(\"tiered Codex probing\", () => {\n it(\"selects PAT, calls whoami then usage, and never invokes the CLI\", async () => {\n const fetch = vi\n .fn()\n .mockResolvedValueOnce(\n new Response(JSON.stringify({ chatgpt_account_id: \"acct-1\" }), {\n status: 200,\n }),\n )\n .mockResolvedValueOnce(\n new Response(JSON.stringify(apiUsage()), { status: 200 }),\n );\n const rpc = vi.fn();\n const result = await probeCodexCapacity({\n ...context,\n readFile: async () =>\n JSON.stringify({\n personal_access_token: \"pat-secret\",\n tokens: {\n access_token: \"ignored-oauth\",\n account_id: \"ignored-account\",\n },\n }),\n fetch,\n rpc,\n });\n expect(fetch).toHaveBeenCalledTimes(2);\n expect(fetch.mock.calls[0][0]).toBe(\n \"https://auth.openai.com/api/accounts/v1/user-auth-credential/whoami\",\n );\n expect(fetch.mock.calls[1][0]).toBe(\n \"https://chatgpt.com/backend-api/wham/usage\",\n );\n expect(fetch.mock.calls[1][1].headers).toMatchObject({\n Authorization: \"Bearer pat-secret\",\n \"ChatGPT-Account-Id\": \"acct-1\",\n });\n expect(rpc).not.toHaveBeenCalled();\n expect(result).toMatchObject({\n harness: \"codex\",\n provider: \"openai\",\n available: \"yes\",\n creditsRemaining: 12.5,\n authenticated: true,\n });\n expect(result.windows.map((window) => window.id)).toEqual([\n \"session\",\n \"weekly\",\n \"reviews:primary\",\n ]);\n });\n\n it(\"selects a fresh OAuth token without calling whoami\", async () => {\n const fetch = vi\n .fn()\n .mockResolvedValue(\n new Response(JSON.stringify(apiUsage()), { status: 200 }),\n );\n const result = await probeCodexCapacity({\n ...context,\n readFile: async () =>\n JSON.stringify({\n tokens: {\n access_token: \"oauth-secret\",\n account_id: \"acct-2\",\n expires_at: 1_800_000_000,\n },\n }),\n fetch,\n now: () => new Date(\"2026-08-20T10:00:00.000Z\"),\n });\n expect(fetch).toHaveBeenCalledOnce();\n expect(fetch.mock.calls[0][1].headers).toMatchObject({\n Authorization: \"Bearer oauth-secret\",\n \"ChatGPT-Account-Id\": \"acct-2\",\n });\n expect(result.available).toBe(\"yes\");\n });\n\n it.each([\n [\n \"missing auth file\",\n async () => {\n throw Object.assign(new Error(\"missing\"), { code: \"ENOENT\" });\n },\n ],\n [\n \"stale OAuth token\",\n async () =>\n JSON.stringify({\n tokens: {\n access_token: \"stale-secret\",\n account_id: \"acct\",\n expires_at: 1,\n },\n }),\n ],\n [\n \"OAuth 401\",\n async () =>\n JSON.stringify({\n tokens: {\n access_token: \"oauth-secret\",\n account_id: \"acct\",\n expires_at: 1_800_000_000,\n },\n }),\n ],\n ])(\"falls back to the CLI for %s\", async (name, readFile) => {\n const fetch = vi\n .fn()\n .mockResolvedValue(\n new Response(\"\", { status: name === \"OAuth 401\" ? 401 : 200 }),\n );\n const rpc = vi.fn(async () => ({\n rateLimits: {\n rateLimits: {\n primary: { usedPercent: 5, windowDurationMins: 300, resetsAt: null },\n },\n },\n account: { account: { type: \"chatgpt\" } },\n }));\n const result = await probeCodexCapacity({\n ...context,\n readFile,\n fetch,\n rpc,\n now: () => new Date(checkedAt),\n });\n expect(rpc).toHaveBeenCalledOnce();\n expect(result.windows).toHaveLength(1);\n });\n\n it(\"falls back to CLI if PAT requests fail\", async () => {\n const rpc = vi.fn(async () => ({\n rateLimits: {},\n account: { account: null },\n }));\n const result = await probeCodexCapacity({\n ...context,\n readFile: async () =>\n JSON.stringify({ personal_access_token: \"pat-secret\" }),\n fetch: vi.fn().mockRejectedValue(new Error(\"network failure pat-secret\")),\n rpc,\n });\n expect(rpc).toHaveBeenCalledOnce();\n expect(result.available).toBe(\"unknown\");\n });\n\n it(\"tries fresh OAuth after a PAT request fails\", async () => {\n const fetch = vi\n .fn()\n .mockRejectedValueOnce(new Error(\"PAT failed\"))\n .mockResolvedValueOnce(\n new Response(JSON.stringify(apiUsage()), { status: 200 }),\n );\n const rpc = vi.fn();\n const result = await probeCodexCapacity({\n ...context,\n readFile: async () =>\n JSON.stringify({\n personal_access_token: \"pat-secret\",\n tokens: {\n access_token: \"oauth-secret\",\n account_id: \"acct\",\n expires_at: 1_800_000_000,\n },\n }),\n fetch,\n rpc,\n now: () => new Date(checkedAt),\n });\n expect(fetch).toHaveBeenCalledTimes(2);\n expect(result.available).toBe(\"yes\");\n expect(rpc).not.toHaveBeenCalled();\n });\n\n it(\"uses hardened read-only app-server arguments and both account methods\", async () => {\n const rpc = vi.fn(async () => ({\n rateLimits: {\n rateLimits: {\n primary: { usedPercent: 5, windowDurationMins: 300, resetsAt: null },\n },\n },\n account: { account: { type: \"chatgpt\" } },\n }));\n await probeCodexCapacity({ ...context, readFile: async () => \"{}\", rpc });\n const messages = rpc.mock.calls[0][0];\n expect(messages.map((message) => message.method)).toEqual([\n \"initialize\",\n \"initialized\",\n \"account/rateLimits/read\",\n \"account/read\",\n ]);\n expect(JSON.stringify(messages)).not.toMatch(/prompt|turn\\/start/);\n expect(CODEX_APP_SERVER_ARGS).toEqual([\n \"-s\",\n \"read-only\",\n \"-a\",\n \"untrusted\",\n \"app-server\",\n ]);\n });\n\n it(\"uses account/read to distinguish logged-out CLI state\", async () => {\n const result = await probeCodexCapacity({\n ...context,\n readFile: async () => \"{}\",\n rpc: async () => ({ rateLimits: {}, account: { account: null } }),\n });\n expect(result).toMatchObject({\n authenticated: false,\n available: \"unknown\",\n });\n });\n\n it(\"never exposes tokens or raw auth content through failures\", async () => {\n const secrets = [\n \"pat-secret-value\",\n \"oauth-secret-value\",\n \"refresh-secret-value\",\n ];\n const result = await probeCodexCapacity({\n ...context,\n readFile: async () =>\n JSON.stringify({\n personal_access_token: secrets[0],\n tokens: { access_token: secrets[1], refresh_token: secrets[2] },\n }),\n fetch: vi.fn().mockRejectedValue(new Error(secrets.join(\" \"))),\n rpc: async () => {\n throw new Error(secrets.join(\" \"));\n },\n });\n const output = JSON.stringify(result);\n for (const secret of secrets) expect(output).not.toContain(secret);\n });\n});\n"],"names":["describe","expect","it","vi","CODEX_APP_SERVER_ARGS","parseUsage","probeCodexCapacity","resolveCodexAuthPath","toRateWindow","checkedAt","context","installed","apiUsage","overrides","rate_limit","primary_window","used_percent","limit_window_seconds","reset_at","secondary_window","credits","balance","additional_rate_limits","limit_name","CODEX_HOME","HOME","toBe","toEqual","id","label","durationMinutes","usedPercent","resetsAt","toMatchObject","snapshot","source","creditsRemaining","windows","objectContaining","toBeNull","fetch","fn","mockResolvedValueOnce","Response","JSON","stringify","chatgpt_account_id","status","rpc","result","readFile","personal_access_token","tokens","access_token","account_id","toHaveBeenCalledTimes","mock","calls","headers","Authorization","not","toHaveBeenCalled","harness","provider","available","authenticated","map","window","mockResolvedValue","expires_at","now","Date","toHaveBeenCalledOnce","each","Object","assign","Error","code","name","rateLimits","primary","windowDurationMins","account","type","toHaveLength","mockRejectedValue","mockRejectedValueOnce","messages","message","method","toMatch","secrets","refresh_token","join","output","secret","toContain"],"mappings":"AAAA,SAASA,QAAQ,EAAEC,MAAM,EAAEC,EAAE,EAAEC,EAAE,QAAQ,SAAS;AAClD,SACEC,qBAAqB,EACrBC,UAAU,EACVC,kBAAkB,EAClBC,oBAAoB,EACpBC,YAAY,QACP,0BAA0B;AAEjC,MAAMC,YAAY;AAClB,MAAMC,UAAU;IAAEC,WAAW;IAAMF;AAAU;AAE7C,SAASG,SAASC,YAAqC,CAAC,CAAC;IACvD,OAAO;QACLC,YAAY;YACVC,gBAAgB;gBACdC,cAAc;gBACdC,sBAAsB;gBACtBC,UAAU;YACZ;YACAC,kBAAkB;gBAChBH,cAAc;gBACdC,sBAAsB;gBACtBC,UAAU;YACZ;YACA,GAAGL,SAAS;QACd;QACAO,SAAS;YAAEC,SAAS;QAAK;QACzBC,wBAAwB;YACtB;gBACEC,YAAY;gBACZT,YAAY;oBACVC,gBAAgB;wBACdC,cAAc;wBACdC,sBAAsB;wBACtBC,UAAU;oBACZ;gBACF;YACF;SACD;IACH;AACF;AAEAlB,SAAS,yBAAyB;IAChCE,GAAG,+BAA+B;QAChCD,OACEM,qBAAqB;YACnBiB,YAAY;YACZC,MAAM;QACR,IACAC,IAAI,CAAC;IACT;IAEAxB,GAAG,oCAAoC;QACrCD,OAAOM,qBAAqB;YAAEkB,MAAM;QAAc,IAAIC,IAAI,CACxD;IAEJ;AACF;AAEA1B,SAAS,2BAA2B;IAClCE,GAAG,gEAAgE;QACjED,OACEO,aACE;YACEQ,cAAc;YACdC,sBAAsB;YACtBC,UAAU;QACZ,GACA,WACA,YAEFS,OAAO,CAAC;YACRC,IAAI;YACJC,OAAO;YACPC,iBAAiB;YACjBC,aAAa;YACbC,UAAU;QACZ;QACA/B,OAAOO,aAAa,CAAC,GAAG,WAAW,YAAYyB,aAAa,CAAC;YAC3DF,aAAa;QACf;IACF;IAEA7B,GAAG,mDAAmD;QACpD,MAAMgC,WAAW7B,WAAWO,YAAY;QACxCX,OAAOiC,UAAUD,aAAa,CAAC;YAAEE,QAAQ;YAAOC,kBAAkB;QAAK;QACvEnC,OAAOiC,SAASG,OAAO,EAAEV,OAAO,CAAC;YAC/B1B,OAAOqC,gBAAgB,CAAC;gBAAEV,IAAI;gBAAWE,iBAAiB;YAAI;YAC9D7B,OAAOqC,gBAAgB,CAAC;gBAAEV,IAAI;gBAAUE,iBAAiB;YAAM;YAC/D7B,OAAOqC,gBAAgB,CAAC;gBAAEV,IAAI;gBAAmBE,iBAAiB;YAAG;SACtE;IACH;IAEA5B,GAAG,6DAA6D;QAC9D,MAAMgC,WAAW7B,WAAW;YAAEe,SAAS,CAAC;QAAE,GAAG;QAC7CnB,OAAOiC,SAASG,OAAO,EAAEV,OAAO,CAAC,EAAE;QACnC1B,OAAOiC,SAASE,gBAAgB,EAAEG,QAAQ;IAC5C;AACF;AAEAvC,SAAS,wBAAwB;IAC/BE,GAAG,mEAAmE;QACpE,MAAMsC,QAAQrC,GACXsC,EAAE,GACFC,qBAAqB,CACpB,IAAIC,SAASC,KAAKC,SAAS,CAAC;YAAEC,oBAAoB;QAAS,IAAI;YAC7DC,QAAQ;QACV,IAEDL,qBAAqB,CACpB,IAAIC,SAASC,KAAKC,SAAS,CAACjC,aAAa;YAAEmC,QAAQ;QAAI;QAE3D,MAAMC,MAAM7C,GAAGsC,EAAE;QACjB,MAAMQ,SAAS,MAAM3C,mBAAmB;YACtC,GAAGI,OAAO;YACVwC,UAAU,UACRN,KAAKC,SAAS,CAAC;oBACbM,uBAAuB;oBACvBC,QAAQ;wBACNC,cAAc;wBACdC,YAAY;oBACd;gBACF;YACFd;YACAQ;QACF;QACA/C,OAAOuC,OAAOe,qBAAqB,CAAC;QACpCtD,OAAOuC,MAAMgB,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE,EAAE/B,IAAI,CACjC;QAEFzB,OAAOuC,MAAMgB,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE,EAAE/B,IAAI,CACjC;QAEFzB,OAAOuC,MAAMgB,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE,CAACC,OAAO,EAAEzB,aAAa,CAAC;YACnD0B,eAAe;YACf,sBAAsB;QACxB;QACA1D,OAAO+C,KAAKY,GAAG,CAACC,gBAAgB;QAChC5D,OAAOgD,QAAQhB,aAAa,CAAC;YAC3B6B,SAAS;YACTC,UAAU;YACVC,WAAW;YACX5B,kBAAkB;YAClB6B,eAAe;QACjB;QACAhE,OAAOgD,OAAOZ,OAAO,CAAC6B,GAAG,CAAC,CAACC,SAAWA,OAAOvC,EAAE,GAAGD,OAAO,CAAC;YACxD;YACA;YACA;SACD;IACH;IAEAzB,GAAG,sDAAsD;QACvD,MAAMsC,QAAQrC,GACXsC,EAAE,GACF2B,iBAAiB,CAChB,IAAIzB,SAASC,KAAKC,SAAS,CAACjC,aAAa;YAAEmC,QAAQ;QAAI;QAE3D,MAAME,SAAS,MAAM3C,mBAAmB;YACtC,GAAGI,OAAO;YACVwC,UAAU,UACRN,KAAKC,SAAS,CAAC;oBACbO,QAAQ;wBACNC,cAAc;wBACdC,YAAY;wBACZe,YAAY;oBACd;gBACF;YACF7B;YACA8B,KAAK,IAAM,IAAIC,KAAK;QACtB;QACAtE,OAAOuC,OAAOgC,oBAAoB;QAClCvE,OAAOuC,MAAMgB,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE,CAACC,OAAO,EAAEzB,aAAa,CAAC;YACnD0B,eAAe;YACf,sBAAsB;QACxB;QACA1D,OAAOgD,OAAOe,SAAS,EAAEtC,IAAI,CAAC;IAChC;IAEAxB,GAAGuE,IAAI,CAAC;QACN;YACE;YACA;gBACE,MAAMC,OAAOC,MAAM,CAAC,IAAIC,MAAM,YAAY;oBAAEC,MAAM;gBAAS;YAC7D;SACD;QACD;YACE;YACA,UACEjC,KAAKC,SAAS,CAAC;oBACbO,QAAQ;wBACNC,cAAc;wBACdC,YAAY;wBACZe,YAAY;oBACd;gBACF;SACH;QACD;YACE;YACA,UACEzB,KAAKC,SAAS,CAAC;oBACbO,QAAQ;wBACNC,cAAc;wBACdC,YAAY;wBACZe,YAAY;oBACd;gBACF;SACH;KACF,EAAE,gCAAgC,OAAOS,MAAM5B;QAC9C,MAAMV,QAAQrC,GACXsC,EAAE,GACF2B,iBAAiB,CAChB,IAAIzB,SAAS,IAAI;YAAEI,QAAQ+B,SAAS,cAAc,MAAM;QAAI;QAEhE,MAAM9B,MAAM7C,GAAGsC,EAAE,CAAC,UAAa,CAAA;gBAC7BsC,YAAY;oBACVA,YAAY;wBACVC,SAAS;4BAAEjD,aAAa;4BAAGkD,oBAAoB;4BAAKjD,UAAU;wBAAK;oBACrE;gBACF;gBACAkD,SAAS;oBAAEA,SAAS;wBAAEC,MAAM;oBAAU;gBAAE;YAC1C,CAAA;QACA,MAAMlC,SAAS,MAAM3C,mBAAmB;YACtC,GAAGI,OAAO;YACVwC;YACAV;YACAQ;YACAsB,KAAK,IAAM,IAAIC,KAAK9D;QACtB;QACAR,OAAO+C,KAAKwB,oBAAoB;QAChCvE,OAAOgD,OAAOZ,OAAO,EAAE+C,YAAY,CAAC;IACtC;IAEAlF,GAAG,0CAA0C;QAC3C,MAAM8C,MAAM7C,GAAGsC,EAAE,CAAC,UAAa,CAAA;gBAC7BsC,YAAY,CAAC;gBACbG,SAAS;oBAAEA,SAAS;gBAAK;YAC3B,CAAA;QACA,MAAMjC,SAAS,MAAM3C,mBAAmB;YACtC,GAAGI,OAAO;YACVwC,UAAU,UACRN,KAAKC,SAAS,CAAC;oBAAEM,uBAAuB;gBAAa;YACvDX,OAAOrC,GAAGsC,EAAE,GAAG4C,iBAAiB,CAAC,IAAIT,MAAM;YAC3C5B;QACF;QACA/C,OAAO+C,KAAKwB,oBAAoB;QAChCvE,OAAOgD,OAAOe,SAAS,EAAEtC,IAAI,CAAC;IAChC;IAEAxB,GAAG,+CAA+C;QAChD,MAAMsC,QAAQrC,GACXsC,EAAE,GACF6C,qBAAqB,CAAC,IAAIV,MAAM,eAChClC,qBAAqB,CACpB,IAAIC,SAASC,KAAKC,SAAS,CAACjC,aAAa;YAAEmC,QAAQ;QAAI;QAE3D,MAAMC,MAAM7C,GAAGsC,EAAE;QACjB,MAAMQ,SAAS,MAAM3C,mBAAmB;YACtC,GAAGI,OAAO;YACVwC,UAAU,UACRN,KAAKC,SAAS,CAAC;oBACbM,uBAAuB;oBACvBC,QAAQ;wBACNC,cAAc;wBACdC,YAAY;wBACZe,YAAY;oBACd;gBACF;YACF7B;YACAQ;YACAsB,KAAK,IAAM,IAAIC,KAAK9D;QACtB;QACAR,OAAOuC,OAAOe,qBAAqB,CAAC;QACpCtD,OAAOgD,OAAOe,SAAS,EAAEtC,IAAI,CAAC;QAC9BzB,OAAO+C,KAAKY,GAAG,CAACC,gBAAgB;IAClC;IAEA3D,GAAG,yEAAyE;QAC1E,MAAM8C,MAAM7C,GAAGsC,EAAE,CAAC,UAAa,CAAA;gBAC7BsC,YAAY;oBACVA,YAAY;wBACVC,SAAS;4BAAEjD,aAAa;4BAAGkD,oBAAoB;4BAAKjD,UAAU;wBAAK;oBACrE;gBACF;gBACAkD,SAAS;oBAAEA,SAAS;wBAAEC,MAAM;oBAAU;gBAAE;YAC1C,CAAA;QACA,MAAM7E,mBAAmB;YAAE,GAAGI,OAAO;YAAEwC,UAAU,UAAY;YAAMF;QAAI;QACvE,MAAMuC,WAAWvC,IAAIQ,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE;QACrCxD,OAAOsF,SAASrB,GAAG,CAAC,CAACsB,UAAYA,QAAQC,MAAM,GAAG9D,OAAO,CAAC;YACxD;YACA;YACA;YACA;SACD;QACD1B,OAAO2C,KAAKC,SAAS,CAAC0C,WAAW3B,GAAG,CAAC8B,OAAO,CAAC;QAC7CzF,OAAOG,uBAAuBuB,OAAO,CAAC;YACpC;YACA;YACA;YACA;YACA;SACD;IACH;IAEAzB,GAAG,yDAAyD;QAC1D,MAAM+C,SAAS,MAAM3C,mBAAmB;YACtC,GAAGI,OAAO;YACVwC,UAAU,UAAY;YACtBF,KAAK,UAAa,CAAA;oBAAE+B,YAAY,CAAC;oBAAGG,SAAS;wBAAEA,SAAS;oBAAK;gBAAE,CAAA;QACjE;QACAjF,OAAOgD,QAAQhB,aAAa,CAAC;YAC3BgC,eAAe;YACfD,WAAW;QACb;IACF;IAEA9D,GAAG,6DAA6D;QAC9D,MAAMyF,UAAU;YACd;YACA;YACA;SACD;QACD,MAAM1C,SAAS,MAAM3C,mBAAmB;YACtC,GAAGI,OAAO;YACVwC,UAAU,UACRN,KAAKC,SAAS,CAAC;oBACbM,uBAAuBwC,OAAO,CAAC,EAAE;oBACjCvC,QAAQ;wBAAEC,cAAcsC,OAAO,CAAC,EAAE;wBAAEC,eAAeD,OAAO,CAAC,EAAE;oBAAC;gBAChE;YACFnD,OAAOrC,GAAGsC,EAAE,GAAG4C,iBAAiB,CAAC,IAAIT,MAAMe,QAAQE,IAAI,CAAC;YACxD7C,KAAK;gBACH,MAAM,IAAI4B,MAAMe,QAAQE,IAAI,CAAC;YAC/B;QACF;QACA,MAAMC,SAASlD,KAAKC,SAAS,CAACI;QAC9B,KAAK,MAAM8C,UAAUJ,QAAS1F,OAAO6F,QAAQlC,GAAG,CAACoC,SAAS,CAACD;IAC7D;AACF"}
@@ -4,7 +4,8 @@ const checkedAt = "2026-08-09T10:00:00.000Z";
4
4
  describe("getCodexCapacityReport", ()=>{
5
5
  it("checks Codex installation before probing", async ()=>{
6
6
  const probe = vi.fn(async (context)=>({
7
- provider: "codex",
7
+ harness: "codex",
8
+ provider: "openai",
8
9
  generatedAt: context.checkedAt,
9
10
  authenticated: true,
10
11
  available: "yes",
@@ -24,7 +25,8 @@ describe("getCodexCapacityReport", ()=>{
24
25
  checkedAt
25
26
  });
26
27
  expect(report).toMatchObject({
27
- provider: "codex",
28
+ harness: "codex",
29
+ provider: "openai",
28
30
  generatedAt: checkedAt,
29
31
  available: "yes"
30
32
  });
@@ -38,7 +40,8 @@ describe("getCodexCapacityReport", ()=>{
38
40
  }
39
41
  });
40
42
  expect(report).toMatchObject({
41
- provider: "codex",
43
+ harness: "codex",
44
+ provider: "openai",
42
45
  available: "unknown",
43
46
  authenticated: null,
44
47
  windows: []
@@ -1 +1 @@
1
- {"version":3,"sources":["../../../src/__tests__/capacity/index.test.ts"],"sourcesContent":["import { describe, expect, it, vi } from \"vitest\";\nimport { getCodexCapacityReport } from \"../../capacity/index.js\";\n\nconst checkedAt = \"2026-08-09T10:00:00.000Z\";\n\ndescribe(\"getCodexCapacityReport\", () => {\n it(\"checks Codex installation before probing\", async () => {\n const probe = vi.fn(async (context) => ({\n provider: \"codex\",\n generatedAt: context.checkedAt,\n authenticated: true,\n available: \"yes\" as const,\n windows: [],\n creditsRemaining: null,\n }));\n\n const report = await getCodexCapacityReport({\n now: () => new Date(checkedAt),\n path: \"/usr/bin:/opt/bin\",\n access: async (target) => {\n if (target !== \"/opt/bin/codex\") throw new Error(\"missing\");\n },\n probe,\n });\n\n expect(probe).toHaveBeenCalledWith({ installed: true, checkedAt });\n expect(report).toMatchObject({ provider: \"codex\", generatedAt: checkedAt, available: \"yes\" });\n });\n\n it(\"redacts unexpected probe failures into a stable unknown result\", async () => {\n const report = await getCodexCapacityReport({\n now: () => new Date(checkedAt),\n path: \"\",\n probe: async () => {\n throw new Error(\"private provider response\");\n },\n });\n\n expect(report).toMatchObject({\n provider: \"codex\",\n available: \"unknown\",\n authenticated: null,\n windows: [],\n });\n expect(JSON.stringify(report)).not.toContain(\"private provider response\");\n });\n});\n"],"names":["describe","expect","it","vi","getCodexCapacityReport","checkedAt","probe","fn","context","provider","generatedAt","authenticated","available","windows","creditsRemaining","report","now","Date","path","access","target","Error","toHaveBeenCalledWith","installed","toMatchObject","JSON","stringify","not","toContain"],"mappings":"AAAA,SAASA,QAAQ,EAAEC,MAAM,EAAEC,EAAE,EAAEC,EAAE,QAAQ,SAAS;AAClD,SAASC,sBAAsB,QAAQ,0BAA0B;AAEjE,MAAMC,YAAY;AAElBL,SAAS,0BAA0B;IACjCE,GAAG,4CAA4C;QAC7C,MAAMI,QAAQH,GAAGI,EAAE,CAAC,OAAOC,UAAa,CAAA;gBACtCC,UAAU;gBACVC,aAAaF,QAAQH,SAAS;gBAC9BM,eAAe;gBACfC,WAAW;gBACXC,SAAS,EAAE;gBACXC,kBAAkB;YACpB,CAAA;QAEA,MAAMC,SAAS,MAAMX,uBAAuB;YAC1CY,KAAK,IAAM,IAAIC,KAAKZ;YACpBa,MAAM;YACNC,QAAQ,OAAOC;gBACb,IAAIA,WAAW,kBAAkB,MAAM,IAAIC,MAAM;YACnD;YACAf;QACF;QAEAL,OAAOK,OAAOgB,oBAAoB,CAAC;YAAEC,WAAW;YAAMlB;QAAU;QAChEJ,OAAOc,QAAQS,aAAa,CAAC;YAAEf,UAAU;YAASC,aAAaL;YAAWO,WAAW;QAAM;IAC7F;IAEAV,GAAG,kEAAkE;QACnE,MAAMa,SAAS,MAAMX,uBAAuB;YAC1CY,KAAK,IAAM,IAAIC,KAAKZ;YACpBa,MAAM;YACNZ,OAAO;gBACL,MAAM,IAAIe,MAAM;YAClB;QACF;QAEApB,OAAOc,QAAQS,aAAa,CAAC;YAC3Bf,UAAU;YACVG,WAAW;YACXD,eAAe;YACfE,SAAS,EAAE;QACb;QACAZ,OAAOwB,KAAKC,SAAS,CAACX,SAASY,GAAG,CAACC,SAAS,CAAC;IAC/C;AACF"}
1
+ {"version":3,"sources":["../../../src/__tests__/capacity/index.test.ts"],"sourcesContent":["import { describe, expect, it, vi } from \"vitest\";\nimport { getCodexCapacityReport } from \"../../capacity/index.js\";\n\nconst checkedAt = \"2026-08-09T10:00:00.000Z\";\n\ndescribe(\"getCodexCapacityReport\", () => {\n it(\"checks Codex installation before probing\", async () => {\n const probe = vi.fn(async (context) => ({\n harness: \"codex\",\n provider: \"openai\",\n generatedAt: context.checkedAt,\n authenticated: true,\n available: \"yes\" as const,\n windows: [],\n creditsRemaining: null,\n }));\n\n const report = await getCodexCapacityReport({\n now: () => new Date(checkedAt),\n path: \"/usr/bin:/opt/bin\",\n access: async (target) => {\n if (target !== \"/opt/bin/codex\") throw new Error(\"missing\");\n },\n probe,\n });\n\n expect(probe).toHaveBeenCalledWith({ installed: true, checkedAt });\n expect(report).toMatchObject({\n harness: \"codex\",\n provider: \"openai\",\n generatedAt: checkedAt,\n available: \"yes\",\n });\n });\n\n it(\"redacts unexpected probe failures into a stable unknown result\", async () => {\n const report = await getCodexCapacityReport({\n now: () => new Date(checkedAt),\n path: \"\",\n probe: async () => {\n throw new Error(\"private provider response\");\n },\n });\n\n expect(report).toMatchObject({\n harness: \"codex\",\n provider: \"openai\",\n available: \"unknown\",\n authenticated: null,\n windows: [],\n });\n expect(JSON.stringify(report)).not.toContain(\"private provider response\");\n });\n});\n"],"names":["describe","expect","it","vi","getCodexCapacityReport","checkedAt","probe","fn","context","harness","provider","generatedAt","authenticated","available","windows","creditsRemaining","report","now","Date","path","access","target","Error","toHaveBeenCalledWith","installed","toMatchObject","JSON","stringify","not","toContain"],"mappings":"AAAA,SAASA,QAAQ,EAAEC,MAAM,EAAEC,EAAE,EAAEC,EAAE,QAAQ,SAAS;AAClD,SAASC,sBAAsB,QAAQ,0BAA0B;AAEjE,MAAMC,YAAY;AAElBL,SAAS,0BAA0B;IACjCE,GAAG,4CAA4C;QAC7C,MAAMI,QAAQH,GAAGI,EAAE,CAAC,OAAOC,UAAa,CAAA;gBACtCC,SAAS;gBACTC,UAAU;gBACVC,aAAaH,QAAQH,SAAS;gBAC9BO,eAAe;gBACfC,WAAW;gBACXC,SAAS,EAAE;gBACXC,kBAAkB;YACpB,CAAA;QAEA,MAAMC,SAAS,MAAMZ,uBAAuB;YAC1Ca,KAAK,IAAM,IAAIC,KAAKb;YACpBc,MAAM;YACNC,QAAQ,OAAOC;gBACb,IAAIA,WAAW,kBAAkB,MAAM,IAAIC,MAAM;YACnD;YACAhB;QACF;QAEAL,OAAOK,OAAOiB,oBAAoB,CAAC;YAAEC,WAAW;YAAMnB;QAAU;QAChEJ,OAAOe,QAAQS,aAAa,CAAC;YAC3BhB,SAAS;YACTC,UAAU;YACVC,aAAaN;YACbQ,WAAW;QACb;IACF;IAEAX,GAAG,kEAAkE;QACnE,MAAMc,SAAS,MAAMZ,uBAAuB;YAC1Ca,KAAK,IAAM,IAAIC,KAAKb;YACpBc,MAAM;YACNb,OAAO;gBACL,MAAM,IAAIgB,MAAM;YAClB;QACF;QAEArB,OAAOe,QAAQS,aAAa,CAAC;YAC3BhB,SAAS;YACTC,UAAU;YACVG,WAAW;YACXD,eAAe;YACfE,SAAS,EAAE;QACb;QACAb,OAAOyB,KAAKC,SAAS,CAACX,SAASY,GAAG,CAACC,SAAS,CAAC;IAC/C;AACF"}
@@ -0,0 +1,213 @@
1
+ import { readFile } from "node:fs/promises";
2
+ import { fileURLToPath } from "node:url";
3
+ import { describe, expect, it, vi } from "vitest";
4
+ import { parseZaiQuota, probeZaiCapacity, resolveZaiApiKey } from "../../capacity/zai.js";
5
+ const checkedAt = "2026-08-20T10:00:00.000Z";
6
+ const fixture = (name)=>readFile(fileURLToPath(new URL(`./fixtures/${name}`, import.meta.url)), "utf8");
7
+ describe("z.ai credential resolution", ()=>{
8
+ it("prefers Z_AI_API_KEY without reading Pi auth", async ()=>{
9
+ const readAuth = vi.fn(async ()=>fixture("zai-auth.json"));
10
+ await expect(resolveZaiApiKey({
11
+ env: {
12
+ Z_AI_API_KEY: "env-zai-test-key"
13
+ },
14
+ readFile: readAuth
15
+ })).resolves.toBe("env-zai-test-key");
16
+ expect(readAuth).not.toHaveBeenCalled();
17
+ });
18
+ it("reads the Pi zai api_key credential from ~/.pi/agent/auth.json", async ()=>{
19
+ const readAuth = vi.fn(async ()=>fixture("zai-auth.json"));
20
+ await expect(resolveZaiApiKey({
21
+ env: {
22
+ HOME: "/users/test"
23
+ },
24
+ readFile: readAuth
25
+ })).resolves.toBe("pi-zai-test-key");
26
+ expect(readAuth).toHaveBeenCalledWith("/users/test/.pi/agent/auth.json", "utf8");
27
+ });
28
+ it.each([
29
+ [
30
+ "missing",
31
+ async ()=>Promise.reject(Object.assign(new Error("missing"), {
32
+ code: "ENOENT"
33
+ }))
34
+ ],
35
+ [
36
+ "invalid JSON",
37
+ async ()=>"{"
38
+ ],
39
+ [
40
+ "wrong credential type",
41
+ async ()=>JSON.stringify({
42
+ zai: {
43
+ type: "oauth",
44
+ key: "x"
45
+ }
46
+ })
47
+ ],
48
+ [
49
+ "missing key",
50
+ async ()=>JSON.stringify({
51
+ zai: {
52
+ type: "api_key"
53
+ }
54
+ })
55
+ ]
56
+ ])("fails clearly for %s credentials without exposing file content", async (_name, readAuth)=>{
57
+ await expect(resolveZaiApiKey({
58
+ env: {
59
+ HOME: "/users/test"
60
+ },
61
+ readFile: readAuth
62
+ })).rejects.toThrow(/z\.ai API key|z\.ai Pi auth file/);
63
+ });
64
+ });
65
+ describe("z.ai quota mapping", ()=>{
66
+ it("maps token, credit, and monthly MCP windows with recomputed usage", async ()=>{
67
+ const report = parseZaiQuota(JSON.parse(await fixture("zai-quota.json")), checkedAt);
68
+ expect(report).toMatchObject({
69
+ harness: "pi",
70
+ provider: "zai",
71
+ generatedAt: checkedAt,
72
+ authenticated: true,
73
+ available: "yes",
74
+ creditsRemaining: null
75
+ });
76
+ expect(report.windows).toEqual([
77
+ {
78
+ id: "zai:tokens:1",
79
+ label: "Tokens · 5 hours",
80
+ limitType: "TOKENS_LIMIT",
81
+ durationMinutes: 300,
82
+ usedPercent: 30,
83
+ resetsAt: "2026-08-20T10:00:00.000Z",
84
+ total: 1000,
85
+ current: 200,
86
+ remaining: 700
87
+ },
88
+ {
89
+ id: "zai:credit:2",
90
+ label: "Credit · 1 week",
91
+ limitType: "CREDIT_LIMIT",
92
+ durationMinutes: 10080,
93
+ usedPercent: 40,
94
+ resetsAt: "2026-08-27T10:00:00.000Z",
95
+ total: 200,
96
+ current: 80,
97
+ remaining: 150
98
+ },
99
+ {
100
+ id: "zai:time:3",
101
+ label: "MCP · monthly",
102
+ limitType: "TIME_LIMIT",
103
+ durationMinutes: 43200,
104
+ usedPercent: 20,
105
+ resetsAt: "2026-09-19T10:00:00.000Z",
106
+ total: 100,
107
+ current: 20,
108
+ remaining: 90
109
+ }
110
+ ]);
111
+ });
112
+ it("uses and clamps the server percentage when quota amounts are unavailable", ()=>{
113
+ const report = parseZaiQuota({
114
+ success: true,
115
+ code: 200,
116
+ data: {
117
+ limits: [
118
+ {
119
+ type: "TOKENS_LIMIT",
120
+ unit: 1,
121
+ number: 1,
122
+ percentage: 120
123
+ },
124
+ {
125
+ type: "CREDIT_LIMIT",
126
+ unit: 5,
127
+ number: 30,
128
+ percentage: -5
129
+ }
130
+ ]
131
+ }
132
+ }, checkedAt);
133
+ expect(report.windows.map((window)=>window.usedPercent)).toEqual([
134
+ 100,
135
+ 0
136
+ ]);
137
+ });
138
+ it.each([
139
+ [
140
+ "invalid fixture",
141
+ async ()=>JSON.parse(await fixture("zai-quota-invalid.json"))
142
+ ],
143
+ [
144
+ "unsuccessful envelope",
145
+ async ()=>({
146
+ success: false,
147
+ code: 401,
148
+ data: {
149
+ limits: []
150
+ }
151
+ })
152
+ ],
153
+ [
154
+ "invalid entry",
155
+ async ()=>({
156
+ success: true,
157
+ code: 200,
158
+ data: {
159
+ limits: [
160
+ {}
161
+ ]
162
+ }
163
+ })
164
+ ]
165
+ ])("rejects %s", async (_name, input)=>{
166
+ const value = await input();
167
+ expect(()=>parseZaiQuota(value, checkedAt)).toThrow(/z\.ai quota/);
168
+ });
169
+ });
170
+ describe("z.ai request", ()=>{
171
+ it("GETs the global quota endpoint with only Bearer authentication", async ()=>{
172
+ const body = await fixture("zai-quota.json");
173
+ const fetch = vi.fn(async ()=>new Response(body, {
174
+ status: 200
175
+ }));
176
+ const report = await probeZaiCapacity({
177
+ checkedAt,
178
+ env: {
179
+ Z_AI_API_KEY: "request-test-key"
180
+ },
181
+ fetch
182
+ });
183
+ expect(fetch).toHaveBeenCalledWith("https://api.z.ai/api/monitor/usage/quota/limit", expect.objectContaining({
184
+ method: "GET",
185
+ headers: {
186
+ Authorization: "Bearer request-test-key"
187
+ }
188
+ }));
189
+ expect(JSON.stringify(report)).not.toContain("request-test-key");
190
+ });
191
+ it("rejects non-200 and malformed JSON responses with sanitized errors", async ()=>{
192
+ await expect(probeZaiCapacity({
193
+ checkedAt,
194
+ env: {
195
+ Z_AI_API_KEY: "never-expose-this-key"
196
+ },
197
+ fetch: vi.fn(async ()=>new Response("private body", {
198
+ status: 503
199
+ }))
200
+ })).rejects.toThrow("z.ai quota request failed: HTTP 503");
201
+ await expect(probeZaiCapacity({
202
+ checkedAt,
203
+ env: {
204
+ Z_AI_API_KEY: "never-expose-this-key"
205
+ },
206
+ fetch: vi.fn(async ()=>new Response("not-json", {
207
+ status: 200
208
+ }))
209
+ })).rejects.toThrow("z.ai quota response is not valid JSON");
210
+ });
211
+ });
212
+
213
+ //# sourceMappingURL=zai.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../../../src/__tests__/capacity/zai.test.ts"],"sourcesContent":["import { readFile } from \"node:fs/promises\";\nimport { fileURLToPath } from \"node:url\";\nimport { describe, expect, it, vi } from \"vitest\";\nimport {\n parseZaiQuota,\n probeZaiCapacity,\n resolveZaiApiKey,\n} from \"../../capacity/zai.js\";\n\nconst checkedAt = \"2026-08-20T10:00:00.000Z\";\nconst fixture = (name: string) =>\n readFile(\n fileURLToPath(new URL(`./fixtures/${name}`, import.meta.url)),\n \"utf8\",\n );\n\ndescribe(\"z.ai credential resolution\", () => {\n it(\"prefers Z_AI_API_KEY without reading Pi auth\", async () => {\n const readAuth = vi.fn(async () => fixture(\"zai-auth.json\"));\n await expect(\n resolveZaiApiKey({\n env: { Z_AI_API_KEY: \"env-zai-test-key\" },\n readFile: readAuth,\n }),\n ).resolves.toBe(\"env-zai-test-key\");\n expect(readAuth).not.toHaveBeenCalled();\n });\n\n it(\"reads the Pi zai api_key credential from ~/.pi/agent/auth.json\", async () => {\n const readAuth = vi.fn(async () => fixture(\"zai-auth.json\"));\n await expect(\n resolveZaiApiKey({ env: { HOME: \"/users/test\" }, readFile: readAuth }),\n ).resolves.toBe(\"pi-zai-test-key\");\n expect(readAuth).toHaveBeenCalledWith(\n \"/users/test/.pi/agent/auth.json\",\n \"utf8\",\n );\n });\n\n it.each([\n [\n \"missing\",\n async () =>\n Promise.reject(Object.assign(new Error(\"missing\"), { code: \"ENOENT\" })),\n ],\n [\"invalid JSON\", async () => \"{\"],\n [\n \"wrong credential type\",\n async () => JSON.stringify({ zai: { type: \"oauth\", key: \"x\" } }),\n ],\n [\"missing key\", async () => JSON.stringify({ zai: { type: \"api_key\" } })],\n ])(\n \"fails clearly for %s credentials without exposing file content\",\n async (_name, readAuth) => {\n await expect(\n resolveZaiApiKey({ env: { HOME: \"/users/test\" }, readFile: readAuth }),\n ).rejects.toThrow(/z\\.ai API key|z\\.ai Pi auth file/);\n },\n );\n});\n\ndescribe(\"z.ai quota mapping\", () => {\n it(\"maps token, credit, and monthly MCP windows with recomputed usage\", async () => {\n const report = parseZaiQuota(\n JSON.parse(await fixture(\"zai-quota.json\")),\n checkedAt,\n );\n expect(report).toMatchObject({\n harness: \"pi\",\n provider: \"zai\",\n generatedAt: checkedAt,\n authenticated: true,\n available: \"yes\",\n creditsRemaining: null,\n });\n expect(report.windows).toEqual([\n {\n id: \"zai:tokens:1\",\n label: \"Tokens · 5 hours\",\n limitType: \"TOKENS_LIMIT\",\n durationMinutes: 300,\n usedPercent: 30,\n resetsAt: \"2026-08-20T10:00:00.000Z\",\n total: 1000,\n current: 200,\n remaining: 700,\n },\n {\n id: \"zai:credit:2\",\n label: \"Credit · 1 week\",\n limitType: \"CREDIT_LIMIT\",\n durationMinutes: 10080,\n usedPercent: 40,\n resetsAt: \"2026-08-27T10:00:00.000Z\",\n total: 200,\n current: 80,\n remaining: 150,\n },\n {\n id: \"zai:time:3\",\n label: \"MCP · monthly\",\n limitType: \"TIME_LIMIT\",\n durationMinutes: 43200,\n usedPercent: 20,\n resetsAt: \"2026-09-19T10:00:00.000Z\",\n total: 100,\n current: 20,\n remaining: 90,\n },\n ]);\n });\n\n it(\"uses and clamps the server percentage when quota amounts are unavailable\", () => {\n const report = parseZaiQuota(\n {\n success: true,\n code: 200,\n data: {\n limits: [\n { type: \"TOKENS_LIMIT\", unit: 1, number: 1, percentage: 120 },\n { type: \"CREDIT_LIMIT\", unit: 5, number: 30, percentage: -5 },\n ],\n },\n },\n checkedAt,\n );\n expect(report.windows.map((window) => window.usedPercent)).toEqual([\n 100, 0,\n ]);\n });\n\n it.each([\n [\n \"invalid fixture\",\n async () => JSON.parse(await fixture(\"zai-quota-invalid.json\")),\n ],\n [\n \"unsuccessful envelope\",\n async () => ({ success: false, code: 401, data: { limits: [] } }),\n ],\n [\n \"invalid entry\",\n async () => ({ success: true, code: 200, data: { limits: [{}] } }),\n ],\n ])(\"rejects %s\", async (_name, input) => {\n const value = await input();\n expect(() => parseZaiQuota(value, checkedAt)).toThrow(/z\\.ai quota/);\n });\n});\n\ndescribe(\"z.ai request\", () => {\n it(\"GETs the global quota endpoint with only Bearer authentication\", async () => {\n const body = await fixture(\"zai-quota.json\");\n const fetch = vi.fn(async () => new Response(body, { status: 200 }));\n const report = await probeZaiCapacity({\n checkedAt,\n env: { Z_AI_API_KEY: \"request-test-key\" },\n fetch,\n });\n expect(fetch).toHaveBeenCalledWith(\n \"https://api.z.ai/api/monitor/usage/quota/limit\",\n expect.objectContaining({\n method: \"GET\",\n headers: { Authorization: \"Bearer request-test-key\" },\n }),\n );\n expect(JSON.stringify(report)).not.toContain(\"request-test-key\");\n });\n\n it(\"rejects non-200 and malformed JSON responses with sanitized errors\", async () => {\n await expect(\n probeZaiCapacity({\n checkedAt,\n env: { Z_AI_API_KEY: \"never-expose-this-key\" },\n fetch: vi.fn(async () => new Response(\"private body\", { status: 503 })),\n }),\n ).rejects.toThrow(\"z.ai quota request failed: HTTP 503\");\n await expect(\n probeZaiCapacity({\n checkedAt,\n env: { Z_AI_API_KEY: \"never-expose-this-key\" },\n fetch: vi.fn(async () => new Response(\"not-json\", { status: 200 })),\n }),\n ).rejects.toThrow(\"z.ai quota response is not valid JSON\");\n });\n});\n"],"names":["readFile","fileURLToPath","describe","expect","it","vi","parseZaiQuota","probeZaiCapacity","resolveZaiApiKey","checkedAt","fixture","name","URL","url","readAuth","fn","env","Z_AI_API_KEY","resolves","toBe","not","toHaveBeenCalled","HOME","toHaveBeenCalledWith","each","Promise","reject","Object","assign","Error","code","JSON","stringify","zai","type","key","_name","rejects","toThrow","report","parse","toMatchObject","harness","provider","generatedAt","authenticated","available","creditsRemaining","windows","toEqual","id","label","limitType","durationMinutes","usedPercent","resetsAt","total","current","remaining","success","data","limits","unit","number","percentage","map","window","input","value","body","fetch","Response","status","objectContaining","method","headers","Authorization","toContain"],"mappings":"AAAA,SAASA,QAAQ,QAAQ,mBAAmB;AAC5C,SAASC,aAAa,QAAQ,WAAW;AACzC,SAASC,QAAQ,EAAEC,MAAM,EAAEC,EAAE,EAAEC,EAAE,QAAQ,SAAS;AAClD,SACEC,aAAa,EACbC,gBAAgB,EAChBC,gBAAgB,QACX,wBAAwB;AAE/B,MAAMC,YAAY;AAClB,MAAMC,UAAU,CAACC,OACfX,SACEC,cAAc,IAAIW,IAAI,CAAC,WAAW,EAAED,MAAM,EAAE,YAAYE,GAAG,IAC3D;AAGJX,SAAS,8BAA8B;IACrCE,GAAG,gDAAgD;QACjD,MAAMU,WAAWT,GAAGU,EAAE,CAAC,UAAYL,QAAQ;QAC3C,MAAMP,OACJK,iBAAiB;YACfQ,KAAK;gBAAEC,cAAc;YAAmB;YACxCjB,UAAUc;QACZ,IACAI,QAAQ,CAACC,IAAI,CAAC;QAChBhB,OAAOW,UAAUM,GAAG,CAACC,gBAAgB;IACvC;IAEAjB,GAAG,kEAAkE;QACnE,MAAMU,WAAWT,GAAGU,EAAE,CAAC,UAAYL,QAAQ;QAC3C,MAAMP,OACJK,iBAAiB;YAAEQ,KAAK;gBAAEM,MAAM;YAAc;YAAGtB,UAAUc;QAAS,IACpEI,QAAQ,CAACC,IAAI,CAAC;QAChBhB,OAAOW,UAAUS,oBAAoB,CACnC,mCACA;IAEJ;IAEAnB,GAAGoB,IAAI,CAAC;QACN;YACE;YACA,UACEC,QAAQC,MAAM,CAACC,OAAOC,MAAM,CAAC,IAAIC,MAAM,YAAY;oBAAEC,MAAM;gBAAS;SACvE;QACD;YAAC;YAAgB,UAAY;SAAI;QACjC;YACE;YACA,UAAYC,KAAKC,SAAS,CAAC;oBAAEC,KAAK;wBAAEC,MAAM;wBAASC,KAAK;oBAAI;gBAAE;SAC/D;QACD;YAAC;YAAe,UAAYJ,KAAKC,SAAS,CAAC;oBAAEC,KAAK;wBAAEC,MAAM;oBAAU;gBAAE;SAAG;KAC1E,EACC,kEACA,OAAOE,OAAOtB;QACZ,MAAMX,OACJK,iBAAiB;YAAEQ,KAAK;gBAAEM,MAAM;YAAc;YAAGtB,UAAUc;QAAS,IACpEuB,OAAO,CAACC,OAAO,CAAC;IACpB;AAEJ;AAEApC,SAAS,sBAAsB;IAC7BE,GAAG,qEAAqE;QACtE,MAAMmC,SAASjC,cACbyB,KAAKS,KAAK,CAAC,MAAM9B,QAAQ,oBACzBD;QAEFN,OAAOoC,QAAQE,aAAa,CAAC;YAC3BC,SAAS;YACTC,UAAU;YACVC,aAAanC;YACboC,eAAe;YACfC,WAAW;YACXC,kBAAkB;QACpB;QACA5C,OAAOoC,OAAOS,OAAO,EAAEC,OAAO,CAAC;YAC7B;gBACEC,IAAI;gBACJC,OAAO;gBACPC,WAAW;gBACXC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;gBACVC,OAAO;gBACPC,SAAS;gBACTC,WAAW;YACb;YACA;gBACER,IAAI;gBACJC,OAAO;gBACPC,WAAW;gBACXC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;gBACVC,OAAO;gBACPC,SAAS;gBACTC,WAAW;YACb;YACA;gBACER,IAAI;gBACJC,OAAO;gBACPC,WAAW;gBACXC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;gBACVC,OAAO;gBACPC,SAAS;gBACTC,WAAW;YACb;SACD;IACH;IAEAtD,GAAG,4EAA4E;QAC7E,MAAMmC,SAASjC,cACb;YACEqD,SAAS;YACT7B,MAAM;YACN8B,MAAM;gBACJC,QAAQ;oBACN;wBAAE3B,MAAM;wBAAgB4B,MAAM;wBAAGC,QAAQ;wBAAGC,YAAY;oBAAI;oBAC5D;wBAAE9B,MAAM;wBAAgB4B,MAAM;wBAAGC,QAAQ;wBAAIC,YAAY,CAAC;oBAAE;iBAC7D;YACH;QACF,GACAvD;QAEFN,OAAOoC,OAAOS,OAAO,CAACiB,GAAG,CAAC,CAACC,SAAWA,OAAOZ,WAAW,GAAGL,OAAO,CAAC;YACjE;YAAK;SACN;IACH;IAEA7C,GAAGoB,IAAI,CAAC;QACN;YACE;YACA,UAAYO,KAAKS,KAAK,CAAC,MAAM9B,QAAQ;SACtC;QACD;YACE;YACA,UAAa,CAAA;oBAAEiD,SAAS;oBAAO7B,MAAM;oBAAK8B,MAAM;wBAAEC,QAAQ,EAAE;oBAAC;gBAAE,CAAA;SAChE;QACD;YACE;YACA,UAAa,CAAA;oBAAEF,SAAS;oBAAM7B,MAAM;oBAAK8B,MAAM;wBAAEC,QAAQ;4BAAC,CAAC;yBAAE;oBAAC;gBAAE,CAAA;SACjE;KACF,EAAE,cAAc,OAAOzB,OAAO+B;QAC7B,MAAMC,QAAQ,MAAMD;QACpBhE,OAAO,IAAMG,cAAc8D,OAAO3D,YAAY6B,OAAO,CAAC;IACxD;AACF;AAEApC,SAAS,gBAAgB;IACvBE,GAAG,kEAAkE;QACnE,MAAMiE,OAAO,MAAM3D,QAAQ;QAC3B,MAAM4D,QAAQjE,GAAGU,EAAE,CAAC,UAAY,IAAIwD,SAASF,MAAM;gBAAEG,QAAQ;YAAI;QACjE,MAAMjC,SAAS,MAAMhC,iBAAiB;YACpCE;YACAO,KAAK;gBAAEC,cAAc;YAAmB;YACxCqD;QACF;QACAnE,OAAOmE,OAAO/C,oBAAoB,CAChC,kDACApB,OAAOsE,gBAAgB,CAAC;YACtBC,QAAQ;YACRC,SAAS;gBAAEC,eAAe;YAA0B;QACtD;QAEFzE,OAAO4B,KAAKC,SAAS,CAACO,SAASnB,GAAG,CAACyD,SAAS,CAAC;IAC/C;IAEAzE,GAAG,sEAAsE;QACvE,MAAMD,OACJI,iBAAiB;YACfE;YACAO,KAAK;gBAAEC,cAAc;YAAwB;YAC7CqD,OAAOjE,GAAGU,EAAE,CAAC,UAAY,IAAIwD,SAAS,gBAAgB;oBAAEC,QAAQ;gBAAI;QACtE,IACAnC,OAAO,CAACC,OAAO,CAAC;QAClB,MAAMnC,OACJI,iBAAiB;YACfE;YACAO,KAAK;gBAAEC,cAAc;YAAwB;YAC7CqD,OAAOjE,GAAGU,EAAE,CAAC,UAAY,IAAIwD,SAAS,YAAY;oBAAEC,QAAQ;gBAAI;QAClE,IACAnC,OAAO,CAACC,OAAO,CAAC;IACpB;AACF"}
@@ -0,0 +1,189 @@
1
+ import Database from "better-sqlite3";
2
+ import * as fs from "fs";
3
+ import * as os from "os";
4
+ import * as path from "path";
5
+ import { afterEach, beforeEach, describe, expect, it } from "vitest";
6
+ import { AgentStatus } from "../../../adapters/AgentAdapter.js";
7
+ import { OpenCodeAdapter } from "../../../providers/opencode/OpenCodeAdapter.js";
8
+ describe("OpenCodeAdapter", ()=>{
9
+ let adapter;
10
+ let tmpDir;
11
+ let dbPath;
12
+ beforeEach(()=>{
13
+ tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "opencode-adapter-test-"));
14
+ dbPath = path.join(tmpDir, "opencode.db");
15
+ adapter = new OpenCodeAdapter(dbPath);
16
+ });
17
+ afterEach(()=>{
18
+ adapter.close();
19
+ fs.rmSync(tmpDir, {
20
+ recursive: true,
21
+ force: true
22
+ });
23
+ });
24
+ it("detects opencode executables", ()=>{
25
+ expect(adapter.canHandle({
26
+ pid: 1,
27
+ command: "opencode",
28
+ cwd: "/repo"
29
+ })).toBe(true);
30
+ expect(adapter.canHandle({
31
+ pid: 2,
32
+ command: "/usr/bin/opencode.exe",
33
+ cwd: "/repo"
34
+ })).toBe(true);
35
+ expect(adapter.canHandle({
36
+ pid: 3,
37
+ command: "node /repo/opencode-plugin.js"
38
+ })).toBe(false);
39
+ });
40
+ it("returns process-only agents when the database is unavailable", async ()=>{
41
+ const agents = await adapter.detectAgents({
42
+ processes: [
43
+ {
44
+ pid: 100,
45
+ command: "opencode",
46
+ cwd: "/repo",
47
+ tty: "ttys001"
48
+ }
49
+ ]
50
+ });
51
+ expect(agents).toEqual([
52
+ expect.objectContaining({
53
+ type: "opencode",
54
+ status: AgentStatus.RUNNING,
55
+ pid: 100,
56
+ projectPath: "/repo",
57
+ sessionId: "pid-100",
58
+ summary: "OpenCode process running"
59
+ })
60
+ ]);
61
+ });
62
+ it("maps matching OpenCode sessions to agents", async ()=>{
63
+ const now = Date.now();
64
+ writeDatabase(dbPath, {
65
+ sessions: [
66
+ {
67
+ id: "sess-001",
68
+ directory: "/repo",
69
+ timeCreated: now - 60_000
70
+ }
71
+ ],
72
+ messages: [
73
+ {
74
+ id: "msg-user",
75
+ sessionId: "sess-001",
76
+ role: "user",
77
+ timeCreated: now - 50_000,
78
+ timeUpdated: now - 50_000
79
+ },
80
+ {
81
+ id: "msg-assistant",
82
+ sessionId: "sess-001",
83
+ role: "assistant",
84
+ timeCreated: now - 30_000,
85
+ timeUpdated: now - 30_000,
86
+ completed: now - 20_000
87
+ }
88
+ ],
89
+ parts: [
90
+ {
91
+ messageId: "msg-user",
92
+ sessionId: "sess-001",
93
+ type: "text",
94
+ text: "Refactor this module",
95
+ timeCreated: now - 50_000
96
+ }
97
+ ]
98
+ });
99
+ const process = {
100
+ pid: 200,
101
+ command: "opencode",
102
+ cwd: "/repo",
103
+ tty: "ttys002"
104
+ };
105
+ const [agent] = await adapter.detectAgents({
106
+ processes: [
107
+ process
108
+ ]
109
+ });
110
+ expect(agent).toMatchObject({
111
+ type: "opencode",
112
+ status: AgentStatus.WAITING,
113
+ pid: 200,
114
+ projectPath: "/repo",
115
+ sessionId: "sess-001",
116
+ summary: "Refactor this module",
117
+ sessionFilePath: `${dbPath}::sess-001`
118
+ });
119
+ });
120
+ it("returns conversation messages for encoded OpenCode session refs", ()=>{
121
+ writeDatabase(dbPath, {
122
+ sessions: [
123
+ {
124
+ id: "sess-abc",
125
+ directory: "/repo",
126
+ timeCreated: Date.now()
127
+ }
128
+ ],
129
+ messages: [
130
+ {
131
+ id: "msg-user",
132
+ sessionId: "sess-abc",
133
+ role: "user",
134
+ timeCreated: 1000,
135
+ timeUpdated: 1000
136
+ }
137
+ ],
138
+ parts: [
139
+ {
140
+ messageId: "msg-user",
141
+ sessionId: "sess-abc",
142
+ type: "text",
143
+ text: "Hello agent",
144
+ timeCreated: 1000
145
+ }
146
+ ]
147
+ });
148
+ expect(adapter.getConversation(`${dbPath}::sess-abc`)).toEqual([
149
+ {
150
+ role: "user",
151
+ content: "Hello agent"
152
+ }
153
+ ]);
154
+ expect(adapter.getConversation("/not-an-opencode-ref")).toEqual([]);
155
+ });
156
+ });
157
+ function writeDatabase(filePath, data) {
158
+ const db = new Database(filePath);
159
+ db.exec(`
160
+ CREATE TABLE session (id TEXT, directory TEXT, time_created INTEGER);
161
+ CREATE TABLE message (id TEXT, session_id TEXT, data TEXT, time_created INTEGER, time_updated INTEGER);
162
+ CREATE TABLE part (message_id TEXT, session_id TEXT, data TEXT, time_created INTEGER);
163
+ `);
164
+ const insertSession = db.prepare("INSERT INTO session (id, directory, time_created) VALUES (?, ?, ?)");
165
+ for (const session of data.sessions){
166
+ insertSession.run(session.id, session.directory, session.timeCreated);
167
+ }
168
+ const insertMessage = db.prepare("INSERT INTO message (id, session_id, data, time_created, time_updated) VALUES (?, ?, ?, ?, ?)");
169
+ for (const message of data.messages ?? []){
170
+ insertMessage.run(message.id, message.sessionId, JSON.stringify({
171
+ role: message.role,
172
+ time: message.completed === undefined ? {} : {
173
+ completed: message.completed
174
+ }
175
+ }), message.timeCreated, message.timeUpdated);
176
+ }
177
+ const insertPart = db.prepare("INSERT INTO part (message_id, session_id, data, time_created) VALUES (?, ?, ?, ?)");
178
+ for (const part of data.parts ?? []){
179
+ insertPart.run(part.messageId, part.sessionId, JSON.stringify({
180
+ type: part.type,
181
+ text: part.text,
182
+ reasoning: part.reasoning,
183
+ tool: part.tool
184
+ }), part.timeCreated);
185
+ }
186
+ db.close();
187
+ }
188
+
189
+ //# sourceMappingURL=OpenCodeAdapter.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../../../../src/__tests__/providers/opencode/OpenCodeAdapter.test.ts"],"sourcesContent":["import Database from \"better-sqlite3\";\nimport * as fs from \"fs\";\nimport * as os from \"os\";\nimport * as path from \"path\";\nimport { afterEach, beforeEach, describe, expect, it } from \"vitest\";\nimport { AgentStatus, type ProcessInfo } from \"../../../adapters/AgentAdapter.js\";\nimport { OpenCodeAdapter } from \"../../../providers/opencode/OpenCodeAdapter.js\";\n\ndescribe(\"OpenCodeAdapter\", () => {\n let adapter: OpenCodeAdapter;\n let tmpDir: string;\n let dbPath: string;\n\n beforeEach(() => {\n tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), \"opencode-adapter-test-\"));\n dbPath = path.join(tmpDir, \"opencode.db\");\n adapter = new OpenCodeAdapter(dbPath);\n });\n\n afterEach(() => {\n adapter.close();\n fs.rmSync(tmpDir, { recursive: true, force: true });\n });\n\n it(\"detects opencode executables\", () => {\n expect(adapter.canHandle({ pid: 1, command: \"opencode\", cwd: \"/repo\" })).toBe(true);\n expect(adapter.canHandle({ pid: 2, command: \"/usr/bin/opencode.exe\", cwd: \"/repo\" })).toBe(\n true,\n );\n expect(adapter.canHandle({ pid: 3, command: \"node /repo/opencode-plugin.js\" })).toBe(false);\n });\n\n it(\"returns process-only agents when the database is unavailable\", async () => {\n const agents = await adapter.detectAgents({\n processes: [{ pid: 100, command: \"opencode\", cwd: \"/repo\", tty: \"ttys001\" }],\n });\n\n expect(agents).toEqual([\n expect.objectContaining({\n type: \"opencode\",\n status: AgentStatus.RUNNING,\n pid: 100,\n projectPath: \"/repo\",\n sessionId: \"pid-100\",\n summary: \"OpenCode process running\",\n }),\n ]);\n });\n\n it(\"maps matching OpenCode sessions to agents\", async () => {\n const now = Date.now();\n writeDatabase(dbPath, {\n sessions: [{ id: \"sess-001\", directory: \"/repo\", timeCreated: now - 60_000 }],\n messages: [\n {\n id: \"msg-user\",\n sessionId: \"sess-001\",\n role: \"user\",\n timeCreated: now - 50_000,\n timeUpdated: now - 50_000,\n },\n {\n id: \"msg-assistant\",\n sessionId: \"sess-001\",\n role: \"assistant\",\n timeCreated: now - 30_000,\n timeUpdated: now - 30_000,\n completed: now - 20_000,\n },\n ],\n parts: [\n {\n messageId: \"msg-user\",\n sessionId: \"sess-001\",\n type: \"text\",\n text: \"Refactor this module\",\n timeCreated: now - 50_000,\n },\n ],\n });\n\n const process: ProcessInfo = { pid: 200, command: \"opencode\", cwd: \"/repo\", tty: \"ttys002\" };\n const [agent] = await adapter.detectAgents({ processes: [process] });\n\n expect(agent).toMatchObject({\n type: \"opencode\",\n status: AgentStatus.WAITING,\n pid: 200,\n projectPath: \"/repo\",\n sessionId: \"sess-001\",\n summary: \"Refactor this module\",\n sessionFilePath: `${dbPath}::sess-001`,\n });\n });\n\n it(\"returns conversation messages for encoded OpenCode session refs\", () => {\n writeDatabase(dbPath, {\n sessions: [{ id: \"sess-abc\", directory: \"/repo\", timeCreated: Date.now() }],\n messages: [\n {\n id: \"msg-user\",\n sessionId: \"sess-abc\",\n role: \"user\",\n timeCreated: 1000,\n timeUpdated: 1000,\n },\n ],\n parts: [\n {\n messageId: \"msg-user\",\n sessionId: \"sess-abc\",\n type: \"text\",\n text: \"Hello agent\",\n timeCreated: 1000,\n },\n ],\n });\n\n expect(adapter.getConversation(`${dbPath}::sess-abc`)).toEqual([\n { role: \"user\", content: \"Hello agent\" },\n ]);\n expect(adapter.getConversation(\"/not-an-opencode-ref\")).toEqual([]);\n });\n});\n\nfunction writeDatabase(\n filePath: string,\n data: {\n sessions: Array<{ id: string; directory: string; timeCreated: number }>;\n messages?: Array<{\n id: string;\n sessionId: string;\n role: string;\n timeCreated: number;\n timeUpdated: number;\n completed?: number;\n }>;\n parts?: Array<{\n messageId: string;\n sessionId: string;\n type: string;\n text?: string;\n reasoning?: string;\n tool?: string;\n timeCreated: number;\n }>;\n },\n): void {\n const db = new Database(filePath);\n db.exec(`\n CREATE TABLE session (id TEXT, directory TEXT, time_created INTEGER);\n CREATE TABLE message (id TEXT, session_id TEXT, data TEXT, time_created INTEGER, time_updated INTEGER);\n CREATE TABLE part (message_id TEXT, session_id TEXT, data TEXT, time_created INTEGER);\n `);\n\n const insertSession = db.prepare(\n \"INSERT INTO session (id, directory, time_created) VALUES (?, ?, ?)\",\n );\n for (const session of data.sessions) {\n insertSession.run(session.id, session.directory, session.timeCreated);\n }\n\n const insertMessage = db.prepare(\n \"INSERT INTO message (id, session_id, data, time_created, time_updated) VALUES (?, ?, ?, ?, ?)\",\n );\n for (const message of data.messages ?? []) {\n insertMessage.run(\n message.id,\n message.sessionId,\n JSON.stringify({\n role: message.role,\n time: message.completed === undefined ? {} : { completed: message.completed },\n }),\n message.timeCreated,\n message.timeUpdated,\n );\n }\n\n const insertPart = db.prepare(\n \"INSERT INTO part (message_id, session_id, data, time_created) VALUES (?, ?, ?, ?)\",\n );\n for (const part of data.parts ?? []) {\n insertPart.run(\n part.messageId,\n part.sessionId,\n JSON.stringify({\n type: part.type,\n text: part.text,\n reasoning: part.reasoning,\n tool: part.tool,\n }),\n part.timeCreated,\n );\n }\n\n db.close();\n}\n"],"names":["Database","fs","os","path","afterEach","beforeEach","describe","expect","it","AgentStatus","OpenCodeAdapter","adapter","tmpDir","dbPath","mkdtempSync","join","tmpdir","close","rmSync","recursive","force","canHandle","pid","command","cwd","toBe","agents","detectAgents","processes","tty","toEqual","objectContaining","type","status","RUNNING","projectPath","sessionId","summary","now","Date","writeDatabase","sessions","id","directory","timeCreated","messages","role","timeUpdated","completed","parts","messageId","text","process","agent","toMatchObject","WAITING","sessionFilePath","getConversation","content","filePath","data","db","exec","insertSession","prepare","session","run","insertMessage","message","JSON","stringify","time","undefined","insertPart","part","reasoning","tool"],"mappings":"AAAA,OAAOA,cAAc,iBAAiB;AACtC,YAAYC,QAAQ,KAAK;AACzB,YAAYC,QAAQ,KAAK;AACzB,YAAYC,UAAU,OAAO;AAC7B,SAASC,SAAS,EAAEC,UAAU,EAAEC,QAAQ,EAAEC,MAAM,EAAEC,EAAE,QAAQ,SAAS;AACrE,SAASC,WAAW,QAA0B,oCAAoC;AAClF,SAASC,eAAe,QAAQ,iDAAiD;AAEjFJ,SAAS,mBAAmB;IAC1B,IAAIK;IACJ,IAAIC;IACJ,IAAIC;IAEJR,WAAW;QACTO,SAASX,GAAGa,WAAW,CAACX,KAAKY,IAAI,CAACb,GAAGc,MAAM,IAAI;QAC/CH,SAASV,KAAKY,IAAI,CAACH,QAAQ;QAC3BD,UAAU,IAAID,gBAAgBG;IAChC;IAEAT,UAAU;QACRO,QAAQM,KAAK;QACbhB,GAAGiB,MAAM,CAACN,QAAQ;YAAEO,WAAW;YAAMC,OAAO;QAAK;IACnD;IAEAZ,GAAG,gCAAgC;QACjCD,OAAOI,QAAQU,SAAS,CAAC;YAAEC,KAAK;YAAGC,SAAS;YAAYC,KAAK;QAAQ,IAAIC,IAAI,CAAC;QAC9ElB,OAAOI,QAAQU,SAAS,CAAC;YAAEC,KAAK;YAAGC,SAAS;YAAyBC,KAAK;QAAQ,IAAIC,IAAI,CACxF;QAEFlB,OAAOI,QAAQU,SAAS,CAAC;YAAEC,KAAK;YAAGC,SAAS;QAAgC,IAAIE,IAAI,CAAC;IACvF;IAEAjB,GAAG,gEAAgE;QACjE,MAAMkB,SAAS,MAAMf,QAAQgB,YAAY,CAAC;YACxCC,WAAW;gBAAC;oBAAEN,KAAK;oBAAKC,SAAS;oBAAYC,KAAK;oBAASK,KAAK;gBAAU;aAAE;QAC9E;QAEAtB,OAAOmB,QAAQI,OAAO,CAAC;YACrBvB,OAAOwB,gBAAgB,CAAC;gBACtBC,MAAM;gBACNC,QAAQxB,YAAYyB,OAAO;gBAC3BZ,KAAK;gBACLa,aAAa;gBACbC,WAAW;gBACXC,SAAS;YACX;SACD;IACH;IAEA7B,GAAG,6CAA6C;QAC9C,MAAM8B,MAAMC,KAAKD,GAAG;QACpBE,cAAc3B,QAAQ;YACpB4B,UAAU;gBAAC;oBAAEC,IAAI;oBAAYC,WAAW;oBAASC,aAAaN,MAAM;gBAAO;aAAE;YAC7EO,UAAU;gBACR;oBACEH,IAAI;oBACJN,WAAW;oBACXU,MAAM;oBACNF,aAAaN,MAAM;oBACnBS,aAAaT,MAAM;gBACrB;gBACA;oBACEI,IAAI;oBACJN,WAAW;oBACXU,MAAM;oBACNF,aAAaN,MAAM;oBACnBS,aAAaT,MAAM;oBACnBU,WAAWV,MAAM;gBACnB;aACD;YACDW,OAAO;gBACL;oBACEC,WAAW;oBACXd,WAAW;oBACXJ,MAAM;oBACNmB,MAAM;oBACNP,aAAaN,MAAM;gBACrB;aACD;QACH;QAEA,MAAMc,UAAuB;YAAE9B,KAAK;YAAKC,SAAS;YAAYC,KAAK;YAASK,KAAK;QAAU;QAC3F,MAAM,CAACwB,MAAM,GAAG,MAAM1C,QAAQgB,YAAY,CAAC;YAAEC,WAAW;gBAACwB;aAAQ;QAAC;QAElE7C,OAAO8C,OAAOC,aAAa,CAAC;YAC1BtB,MAAM;YACNC,QAAQxB,YAAY8C,OAAO;YAC3BjC,KAAK;YACLa,aAAa;YACbC,WAAW;YACXC,SAAS;YACTmB,iBAAiB,GAAG3C,OAAO,UAAU,CAAC;QACxC;IACF;IAEAL,GAAG,mEAAmE;QACpEgC,cAAc3B,QAAQ;YACpB4B,UAAU;gBAAC;oBAAEC,IAAI;oBAAYC,WAAW;oBAASC,aAAaL,KAAKD,GAAG;gBAAG;aAAE;YAC3EO,UAAU;gBACR;oBACEH,IAAI;oBACJN,WAAW;oBACXU,MAAM;oBACNF,aAAa;oBACbG,aAAa;gBACf;aACD;YACDE,OAAO;gBACL;oBACEC,WAAW;oBACXd,WAAW;oBACXJ,MAAM;oBACNmB,MAAM;oBACNP,aAAa;gBACf;aACD;QACH;QAEArC,OAAOI,QAAQ8C,eAAe,CAAC,GAAG5C,OAAO,UAAU,CAAC,GAAGiB,OAAO,CAAC;YAC7D;gBAAEgB,MAAM;gBAAQY,SAAS;YAAc;SACxC;QACDnD,OAAOI,QAAQ8C,eAAe,CAAC,yBAAyB3B,OAAO,CAAC,EAAE;IACpE;AACF;AAEA,SAASU,cACPmB,QAAgB,EAChBC,IAmBC;IAED,MAAMC,KAAK,IAAI7D,SAAS2D;IACxBE,GAAGC,IAAI,CAAC,CAAC;;;;EAIT,CAAC;IAED,MAAMC,gBAAgBF,GAAGG,OAAO,CAC9B;IAEF,KAAK,MAAMC,WAAWL,KAAKnB,QAAQ,CAAE;QACnCsB,cAAcG,GAAG,CAACD,QAAQvB,EAAE,EAAEuB,QAAQtB,SAAS,EAAEsB,QAAQrB,WAAW;IACtE;IAEA,MAAMuB,gBAAgBN,GAAGG,OAAO,CAC9B;IAEF,KAAK,MAAMI,WAAWR,KAAKf,QAAQ,IAAI,EAAE,CAAE;QACzCsB,cAAcD,GAAG,CACfE,QAAQ1B,EAAE,EACV0B,QAAQhC,SAAS,EACjBiC,KAAKC,SAAS,CAAC;YACbxB,MAAMsB,QAAQtB,IAAI;YAClByB,MAAMH,QAAQpB,SAAS,KAAKwB,YAAY,CAAC,IAAI;gBAAExB,WAAWoB,QAAQpB,SAAS;YAAC;QAC9E,IACAoB,QAAQxB,WAAW,EACnBwB,QAAQrB,WAAW;IAEvB;IAEA,MAAM0B,aAAaZ,GAAGG,OAAO,CAC3B;IAEF,KAAK,MAAMU,QAAQd,KAAKX,KAAK,IAAI,EAAE,CAAE;QACnCwB,WAAWP,GAAG,CACZQ,KAAKxB,SAAS,EACdwB,KAAKtC,SAAS,EACdiC,KAAKC,SAAS,CAAC;YACbtC,MAAM0C,KAAK1C,IAAI;YACfmB,MAAMuB,KAAKvB,IAAI;YACfwB,WAAWD,KAAKC,SAAS;YACzBC,MAAMF,KAAKE,IAAI;QACjB,IACAF,KAAK9B,WAAW;IAEpB;IAEAiB,GAAG5C,KAAK;AACV"}