@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.
- package/dist/__tests__/capacity/codex.test.js +2 -1
- package/dist/__tests__/capacity/codex.test.js.map +1 -1
- package/dist/__tests__/capacity/index.test.js +6 -3
- package/dist/__tests__/capacity/index.test.js.map +1 -1
- package/dist/__tests__/capacity/zai.test.js +213 -0
- package/dist/__tests__/capacity/zai.test.js.map +1 -0
- package/dist/__tests__/providers/opencode/OpenCodeAdapter.test.js +189 -0
- package/dist/__tests__/providers/opencode/OpenCodeAdapter.test.js.map +1 -0
- package/dist/__tests__/providers/opencode/OpenCodeAgentMapper.test.js +89 -0
- package/dist/__tests__/providers/opencode/OpenCodeAgentMapper.test.js.map +1 -0
- package/dist/__tests__/providers/opencode/OpenCodeSessionLocator.test.js +94 -0
- package/dist/__tests__/providers/opencode/OpenCodeSessionLocator.test.js.map +1 -0
- package/dist/__tests__/providers/opencode/OpenCodeSessionParser.test.js +147 -0
- package/dist/__tests__/providers/opencode/OpenCodeSessionParser.test.js.map +1 -0
- package/dist/adapters/index.d.ts +1 -1
- package/dist/adapters/index.d.ts.map +1 -1
- package/dist/adapters/index.js +1 -1
- package/dist/adapters/index.js.map +1 -1
- package/dist/capacity/codex.d.ts +1 -0
- package/dist/capacity/codex.d.ts.map +1 -1
- package/dist/capacity/codex.js +9 -4
- package/dist/capacity/codex.js.map +1 -1
- package/dist/capacity/index.d.ts +8 -0
- package/dist/capacity/index.d.ts.map +1 -1
- package/dist/capacity/index.js +10 -9
- package/dist/capacity/index.js.map +1 -1
- package/dist/capacity/types.d.ts +8 -0
- package/dist/capacity/types.d.ts.map +1 -1
- package/dist/capacity/types.js.map +1 -1
- package/dist/capacity/zai.d.ts +14 -0
- package/dist/capacity/zai.d.ts.map +1 -0
- package/dist/capacity/zai.js +203 -0
- package/dist/capacity/zai.js.map +1 -0
- package/dist/index.d.ts +3 -3
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +2 -2
- package/dist/index.js.map +1 -1
- package/dist/{adapters → providers/opencode}/OpenCodeAdapter.d.ts +5 -12
- package/dist/providers/opencode/OpenCodeAdapter.d.ts.map +1 -0
- package/dist/providers/opencode/OpenCodeAdapter.js +86 -0
- package/dist/providers/opencode/OpenCodeAdapter.js.map +1 -0
- package/dist/providers/opencode/OpenCodeAgentMapper.d.ts +11 -0
- package/dist/providers/opencode/OpenCodeAgentMapper.d.ts.map +1 -0
- package/dist/providers/opencode/OpenCodeAgentMapper.js +50 -0
- package/dist/providers/opencode/OpenCodeAgentMapper.js.map +1 -0
- package/dist/providers/opencode/OpenCodeSessionLocator.d.ts +21 -0
- package/dist/providers/opencode/OpenCodeSessionLocator.d.ts.map +1 -0
- package/dist/providers/opencode/OpenCodeSessionLocator.js +99 -0
- package/dist/providers/opencode/OpenCodeSessionLocator.js.map +1 -0
- package/dist/providers/opencode/OpenCodeSessionParser.d.ts +18 -0
- package/dist/providers/opencode/OpenCodeSessionParser.d.ts.map +1 -0
- package/dist/providers/opencode/OpenCodeSessionParser.js +104 -0
- package/dist/providers/opencode/OpenCodeSessionParser.js.map +1 -0
- package/package.json +1 -1
- package/src/__tests__/capacity/codex.test.ts +94 -26
- package/src/__tests__/capacity/fixtures/zai-auth.json +6 -0
- package/src/__tests__/capacity/fixtures/zai-quota-invalid.json +7 -0
- package/src/__tests__/capacity/fixtures/zai-quota.json +38 -0
- package/src/__tests__/capacity/index.test.ts +10 -3
- package/src/__tests__/capacity/zai.test.ts +186 -0
- package/src/__tests__/providers/opencode/OpenCodeAdapter.test.ts +197 -0
- package/src/__tests__/providers/opencode/OpenCodeAgentMapper.test.ts +83 -0
- package/src/__tests__/providers/opencode/OpenCodeSessionLocator.test.ts +84 -0
- package/src/__tests__/providers/opencode/OpenCodeSessionParser.test.ts +148 -0
- package/src/adapters/index.ts +1 -1
- package/src/capacity/codex.ts +96 -28
- package/src/capacity/index.ts +29 -11
- package/src/capacity/types.ts +8 -0
- package/src/capacity/zai.ts +237 -0
- package/src/index.ts +8 -3
- package/src/providers/opencode/OpenCodeAdapter.ts +112 -0
- package/src/providers/opencode/OpenCodeAgentMapper.ts +61 -0
- package/src/providers/opencode/OpenCodeSessionLocator.ts +123 -0
- package/src/providers/opencode/OpenCodeSessionParser.ts +145 -0
- package/dist/__tests__/adapters/OpenCodeAdapter.test.js +0 -636
- package/dist/__tests__/adapters/OpenCodeAdapter.test.js.map +0 -1
- package/dist/adapters/OpenCodeAdapter.d.ts.map +0 -1
- package/dist/adapters/OpenCodeAdapter.js +0 -289
- package/dist/adapters/OpenCodeAdapter.js.map +0 -1
- package/src/__tests__/adapters/OpenCodeAdapter.test.ts +0 -544
- 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
|
-
|
|
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
|
-
|
|
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
|
-
|
|
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
|
-
|
|
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
|
|
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"}
|