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
94 lines
2.8 KiB
JavaScript
94 lines
2.8 KiB
JavaScript
import { buildResult, toUsageWindow, toNumber } from '../utils/index.js';
|
|
import { readManagedCredential } from '../credentials/providers.js';
|
|
|
|
export const providerId = 'ollama-cloud';
|
|
export const providerName = 'Ollama Cloud';
|
|
const aliases = ['ollama-cloud', 'ollamacloud'];
|
|
|
|
export const parseOllamaSettingsHtml = (html) => {
|
|
const windows = {};
|
|
const sessionMatch = html.match(/Session\s+usage[^0-9]*([0-9.]+)%/i);
|
|
if (sessionMatch) {
|
|
windows.session = toUsageWindow({
|
|
usedPercent: toNumber(sessionMatch[1]),
|
|
windowSeconds: null,
|
|
resetAt: null
|
|
});
|
|
}
|
|
const weeklyMatch = html.match(/Weekly\s+usage[^0-9]*([0-9.]+)%/i);
|
|
if (weeklyMatch) {
|
|
windows.weekly = toUsageWindow({
|
|
usedPercent: toNumber(weeklyMatch[1]),
|
|
windowSeconds: null,
|
|
resetAt: null
|
|
});
|
|
}
|
|
const premiumMatch = html.match(/Premium[^0-9]*([0-9]+)\s*\/\s*([0-9]+)/i);
|
|
if (premiumMatch) {
|
|
const used = toNumber(premiumMatch[1]);
|
|
const total = toNumber(premiumMatch[2]);
|
|
const usedPercent = total && used !== null ? Math.min(100, (used / total) * 100) : null;
|
|
windows.premium = toUsageWindow({
|
|
usedPercent,
|
|
windowSeconds: null,
|
|
resetAt: null,
|
|
valueLabel: `${used ?? 0} / ${total ?? 0}`
|
|
});
|
|
}
|
|
return windows;
|
|
};
|
|
|
|
export const isConfigured = () => {
|
|
return Boolean(readManagedCredential(providerId));
|
|
};
|
|
|
|
export const fetchOllamaCloudUsage = async (credential, fetchImpl = fetch) => {
|
|
const response = await fetchImpl('https://ollama.com/settings', {
|
|
method: 'GET',
|
|
headers: { Cookie: credential.cookie, '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('Ollama Cloud authentication failed');
|
|
}
|
|
if (!response.ok) throw new Error(`Ollama Cloud returned HTTP ${response.status}`);
|
|
const windows = parseOllamaSettingsHtml(await response.text());
|
|
if (Object.keys(windows).length === 0) throw new Error('Ollama Cloud usage data could not be parsed');
|
|
return windows;
|
|
};
|
|
|
|
export const fetchQuota = async () => {
|
|
const credential = readManagedCredential(providerId);
|
|
|
|
if (!credential) {
|
|
return buildResult({
|
|
providerId,
|
|
providerName,
|
|
ok: false,
|
|
configured: false,
|
|
error: 'Not configured'
|
|
});
|
|
}
|
|
|
|
try {
|
|
const windows = await fetchOllamaCloudUsage(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'
|
|
});
|
|
}
|
|
};
|