lzt404
e4d0059efa
- Update Claude, Codex, Kimi, Antigravity, and xAI compatibility - Display plan details, subscription expiry, and dynamic reset times - Deduplicate quota requests and serialize Codex resets - Require confirmation for potentially billable xAI health probes - Add quota parsing, request, and rendering regression tests
75 lines
3.4 KiB
TypeScript
75 lines
3.4 KiB
TypeScript
import { isRecord, readString } from './managementApi';
|
|
|
|
export const decodeQuotaToken = (value: unknown): Record<string, unknown> | null => {
|
|
if (isRecord(value)) return value;
|
|
if (typeof value !== 'string' || !value.trim()) return null;
|
|
try {
|
|
const parsed = JSON.parse(value);
|
|
if (isRecord(parsed)) return parsed;
|
|
} catch { /* Try the JWT payload next. */ }
|
|
const segment = value.trim().split('.')[1];
|
|
if (!segment) return null;
|
|
try {
|
|
const normalized = segment.replace(/-/g, '+').replace(/_/g, '/')
|
|
.padEnd(Math.ceil(segment.length / 4) * 4, '=');
|
|
const bytes = Uint8Array.from(atob(normalized), (character) => character.charCodeAt(0));
|
|
const parsed = JSON.parse(new TextDecoder().decode(bytes));
|
|
return isRecord(parsed) ? parsed : null;
|
|
} catch {
|
|
return null;
|
|
}
|
|
};
|
|
|
|
export const quotaAuthRecords = (file: Record<string, unknown>) =>
|
|
[file, file.metadata, file.attributes].filter(isRecord);
|
|
|
|
export const codexMetadataFor = (file: Record<string, unknown>) => {
|
|
const records = quotaAuthRecords(file).flatMap((record) => {
|
|
const tokens = [record.id_token, record.idToken].flatMap((token) => {
|
|
const payload = decodeQuotaToken(token);
|
|
if (!payload) return [];
|
|
return [payload, payload['https://api.openai.com/auth']].filter(isRecord);
|
|
});
|
|
return [record, ...tokens];
|
|
});
|
|
const first = (...keys: string[]) => records.map((record) => readString(record, ...keys)).find(Boolean);
|
|
const expiry = first(
|
|
'chatgpt_subscription_active_until', 'chatgptSubscriptionActiveUntil',
|
|
'subscription_active_until', 'subscriptionActiveUntil',
|
|
) || records.map((record) => readString(record.subscription, 'active_until', 'activeUntil')).find(Boolean);
|
|
return {
|
|
accountId: first('chatgpt_account_id', 'chatgptAccountId', 'account_id', 'accountId') || '',
|
|
plan: first('plan_type', 'planType', 'chatgpt_plan_type', 'chatgptPlanType')?.toLowerCase(),
|
|
subscriptionActiveUntil: expiry && expiry !== '0' ? expiry : undefined,
|
|
};
|
|
};
|
|
|
|
export const antigravityProjectFor = (file: Record<string, unknown>): string => {
|
|
const records = [...quotaAuthRecords(file), file.installed, file.web].filter(isRecord);
|
|
return records.map((record) => readString(record, 'project_id', 'projectId', 'gemini_virtual_project'))
|
|
.find(Boolean) || '';
|
|
};
|
|
|
|
export const isPaidXaiFile = (file: Record<string, unknown>): boolean => {
|
|
const records: Record<string, unknown>[] = [];
|
|
const visited = new Set<Record<string, unknown>>();
|
|
const visit = (value: unknown, depth: number) => {
|
|
if (!isRecord(value) || visited.has(value) || depth > 2) return;
|
|
visited.add(value);
|
|
records.push(value);
|
|
for (const key of ['metadata', 'attributes', 'oauth', 'raw', 'credential', 'auth']) {
|
|
visit(value[key], depth + 1);
|
|
}
|
|
};
|
|
visit(file, 0);
|
|
const usingApi = records.some((record) =>
|
|
['true', '1', 'yes', 'y', 'on'].includes(String(record.using_api ?? record.usingApi).trim().toLowerCase()));
|
|
if (usingApi && records.some((record) => readString(record, 'prefix').toLowerCase() === 'paid')) return true;
|
|
return records.some((record) => ['access_token', 'accessToken', 'id_token', 'idToken', 'token'].some((key) => {
|
|
const payload = decodeQuotaToken(record[key]);
|
|
if (!payload) return false;
|
|
return Object.entries(payload).some(([claim, tier]) =>
|
|
/(?:^|[/:])tier$/i.test(claim) && (typeof tier === 'number' || typeof tier === 'string') && Number(tier) >= 1);
|
|
}));
|
|
};
|