feat(quota): add OpenCode Go usage tracking (#2155)

* feat(quota): add OpenCode Go usage tracking * fix(quota): align OpenCode Go VS Code parsing
2026-07-12 14:48:50 +03:00
parent b4f50e0a01
commit 3d90eddcaf
26 changed files with 658 additions and 1 deletions
@@ -0,0 +1,32 @@
+import { afterAll, afterEach, describe, expect, it } from 'bun:test';
+import fs from 'node:fs';
+import os from 'node:os';
+import path from 'node:path';
+import { deleteOpenCodeGoCredential, getOpenCodeGoCredentialStatus, readOpenCodeGoCredential, writeOpenCodeGoCredential } from './opencode-go-credentials.js';
+
+const previousDataDir = process.env.OPENCHAMBER_DATA_DIR;
+const temporaryDirectory = fs.mkdtempSync(path.join(os.tmpdir(), 'openchamber-go-'));
+process.env.OPENCHAMBER_DATA_DIR = temporaryDirectory;
+
+afterEach(() => deleteOpenCodeGoCredential());
+
+describe('OpenCode Go credential store', () => {
+  it('normalizes, masks, and stores credentials with owner-only permissions', () => {
+    const status = writeOpenCodeGoCredential({ workspaceId: ' wrk_test ', authCookie: ' auth=secret ' });
+    expect(status).toEqual({ configured: true, workspaceId: 'wrk_test', authCookieMasked: '••••••••' });
+    expect(readOpenCodeGoCredential()).toEqual({ workspaceId: 'wrk_test', authCookie: 'secret' });
+    expect(fs.statSync(path.join(temporaryDirectory, 'quota', 'opencode-go.json')).mode & 0o777).toBe(0o600);
+  });
+
+  it('removes credentials without exposing prior values', () => {
+    writeOpenCodeGoCredential({ workspaceId: 'wrk_test', authCookie: 'secret' });
+    deleteOpenCodeGoCredential();
+    expect(getOpenCodeGoCredentialStatus()).toEqual({ configured: false });
+  });
+});
+
+afterAll(() => {
+  if (previousDataDir === undefined) delete process.env.OPENCHAMBER_DATA_DIR;
+  else process.env.OPENCHAMBER_DATA_DIR = previousDataDir;
+  fs.rmSync(temporaryDirectory, { recursive: true, force: true });
+});