Bohdan Triapitsyn
b09614fd68
- add shared owner-only credential storage for OpenCode Go, Ollama Cloud, and Cursor - validate credentials before atomic writes using 0700 directories and 0600 files - replace provider-specific credential routes with an allowlisted lifecycle API - stop automatically reading Ollama's legacy cookie file - stop reading or modifying Cursor's database during regular quota requests - add explicit one-time Cursor credential import without mutating Cursor storage - persist refreshed Cursor credentials only in OpenChamber-managed storage - add Ollama Cloud and Cursor credential controls to provider settings - preserve OpenCode Go tracking through the shared credential flow - add VS Code credential management and Cursor quota parity - reject authentication redirects, enforce request timeouts, and fail on unparseable usage pages - mask stored secrets in API responses and extend quota security coverage - update quota provider documentation
71 lines
3.0 KiB
JavaScript
71 lines
3.0 KiB
JavaScript
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('"', '"').replaceAll('"', '"').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',
|
|
},
|
|
redirect: 'manual',
|
|
signal: AbortSignal.timeout(15_000),
|
|
});
|
|
if (response.status === 401 || response.status === 403 || (response.status >= 300 && response.status < 400)) {
|
|
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' });
|
|
}
|
|
};
|