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,69 @@
+import { readOpenCodeGoCredential } from '../opencode-go-credentials.js';
+import { buildResult, toUsageWindow } from '../utils/index.js';
+
+export const providerId = 'opencode-go';
+export const providerName = 'OpenCode Go';
+export const aliases = ['opencode-go'];
+
+const patterns = {
+  '5h': 'rollingUsage',
+  weekly: 'weeklyUsage',
+  monthly: 'monthlyUsage',
+};
+
+const captureNumber = (name, body) => {
+  const match = body.match(new RegExp(`["']?${name}["']?\\s*:\\s*["']?(-?\\d+(?:\\.\\d+)?)`));
+  const value = match ? Number(match[1]) : null;
+  return Number.isFinite(value) ? value : null;
+};
+
+export const parseOpenCodeGoUsage = (html, now = Date.now()) => {
+  if (typeof html !== 'string') return {};
+  const normalized = html.replaceAll('&quot;', '"').replaceAll('&#34;', '"').replaceAll('\\u0022', '"').replaceAll('\\"', '"');
+  const windows = {};
+  for (const [key, field] of Object.entries(patterns)) {
+    const escaped = field.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+    const match = normalized.match(new RegExp(`["']?${escaped}["']?\\s*:\\s*(?:\\$R\\[\\d+\\]\\s*=\\s*)?\\{([^{}]*)\\}`, 's'));
+    if (!match) continue;
+    const usedPercent = captureNumber('usagePercent', match[1]);
+    const resetInSec = captureNumber('resetInSec', match[1]);
+    if (usedPercent === null || resetInSec === null) continue;
+    windows[key] = toUsageWindow({
+      usedPercent: Math.min(100, Math.max(0, usedPercent)),
+      resetAt: now + Math.max(0, resetInSec) * 1000,
+      windowSeconds: null,
+    });
+  }
+  return windows;
+};
+
+export const fetchOpenCodeGoUsage = async (credential, fetchImpl = fetch) => {
+  const response = await fetchImpl(`https://opencode.ai/workspace/${encodeURIComponent(credential.workspaceId)}/go`, {
+    headers: {
+      Accept: 'text/html,application/xhtml+xml',
+      Cookie: `auth=${credential.authCookie}`,
+      'User-Agent': 'OpenChamber quota provider',
+    },
+    signal: AbortSignal.timeout(15_000),
+  });
+  if (response.status === 401 || response.status === 403 || (response.redirected && /\/auth(?:\/|$|\?)/.test(new URL(response.url).pathname))) {
+    throw new Error('OpenCode Go authentication failed');
+  }
+  if (!response.ok) throw new Error(`OpenCode Go dashboard returned HTTP ${response.status}`);
+  const windows = parseOpenCodeGoUsage(await response.text());
+  if (Object.keys(windows).length === 0) throw new Error('OpenCode Go usage data could not be parsed');
+  return windows;
+};
+
+export const isConfigured = () => Boolean(readOpenCodeGoCredential());
+
+export const fetchQuota = async () => {
+  const credential = readOpenCodeGoCredential();
+  if (!credential) return buildResult({ providerId, providerName, ok: false, configured: false, error: 'Not configured' });
+  try {
+    const windows = await fetchOpenCodeGoUsage(credential);
+    return buildResult({ providerId, providerName, ok: true, configured: true, usage: { windows } });
+  } catch (error) {
+    return buildResult({ providerId, providerName, ok: false, configured: true, error: error instanceof Error ? error.message : 'Request failed' });
+  }
+};