Bohdan Triapitsyn
5917325d49
OpenCode Go now reads quota usage with a bearer API key from OpenCode auth.json Removes the old workspace ID and browser cookie credential flow Deletes legacy OpenCode Go credential files during upgrade
59 lines
2.0 KiB
JavaScript
59 lines
2.0 KiB
JavaScript
import fs from 'node:fs';
|
|
import os from 'node:os';
|
|
import path from 'node:path';
|
|
|
|
const MANAGED_QUOTA_PROVIDERS = new Set(['ollama-cloud', 'cursor']);
|
|
|
|
const credentialsDirectory = () => path.join(
|
|
process.env.OPENCHAMBER_DATA_DIR
|
|
? path.resolve(process.env.OPENCHAMBER_DATA_DIR)
|
|
: path.join(os.homedir(), '.config', 'openchamber'),
|
|
'quota',
|
|
);
|
|
|
|
const credentialPath = (providerId) => {
|
|
if (!MANAGED_QUOTA_PROVIDERS.has(providerId)) throw new Error('Unsupported credential provider');
|
|
return path.join(credentialsDirectory(), `${providerId}.json`);
|
|
};
|
|
|
|
export const readQuotaCredential = (providerId, normalize) => {
|
|
try {
|
|
return normalize(JSON.parse(fs.readFileSync(credentialPath(providerId), 'utf8')));
|
|
} catch (error) {
|
|
if (error?.code !== 'ENOENT') console.warn(`Failed to read ${providerId} quota credentials`);
|
|
return null;
|
|
}
|
|
};
|
|
|
|
export const writeQuotaCredential = (providerId, credential) => {
|
|
const target = credentialPath(providerId);
|
|
const directory = path.dirname(target);
|
|
const temporary = `${target}.${process.pid}.${Date.now()}.tmp`;
|
|
fs.mkdirSync(directory, { recursive: true, mode: 0o700 });
|
|
fs.chmodSync(directory, 0o700);
|
|
try {
|
|
fs.writeFileSync(temporary, `${JSON.stringify(credential, null, 2)}\n`, { mode: 0o600 });
|
|
fs.chmodSync(temporary, 0o600);
|
|
fs.renameSync(temporary, target);
|
|
fs.chmodSync(target, 0o600);
|
|
} finally {
|
|
try { fs.unlinkSync(temporary); } catch {}
|
|
}
|
|
};
|
|
|
|
export const deleteQuotaCredential = (providerId) => {
|
|
try { fs.unlinkSync(credentialPath(providerId)); } catch (error) {
|
|
if (error?.code !== 'ENOENT') throw error;
|
|
}
|
|
};
|
|
|
|
// OpenCode Go used to store a browser auth cookie here. Its usage API now uses
|
|
// OpenCode's auth.json API key, so remove the obsolete secret without reading it.
|
|
export const deleteLegacyOpenCodeGoCredential = () => {
|
|
try {
|
|
fs.unlinkSync(path.join(credentialsDirectory(), 'opencode-go.json'));
|
|
} catch (error) {
|
|
if (error?.code !== 'ENOENT') throw error;
|
|
}
|
|
};
|