LaPaoPao/BoosAPI
1
0
Fork 0
forked from boos/BoosAPI
Code Pull Requests 1 Activity

feat: surface oauth quota details

Browse Source
This commit is contained in:
ciat-777
2026-03-18 12:42:31 +08:00
parent cd37b11722
commit f902fe1cf6
9 changed files with 578 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/server/routes/api/oauth.test.ts
+125
View File
@@ -812,6 +812,31 @@ describe('oauth routes', { timeout: 15_000 }, () => {
accountId: 'chatgpt-account-123',
email: 'codex-user@example.com',
planType: 'team',
idToken: buildJwt({
email: 'codex-user@example.com',
'https://api.openai.com/auth': {
chatgpt_account_id: 'chatgpt-account-123',
chatgpt_plan_type: 'team',
chatgpt_subscription_active_start: '2026-03-01T00:00:00.000Z',
chatgpt_subscription_active_until: '2026-04-01T00:00:00.000Z',
},
}),
quota: {
status: 'supported',
source: 'reverse_engineered',
lastSyncAt: '2026-03-17T08:00:00.000Z',
lastLimitResetAt: '2026-03-17T13:00:00.000Z',
windows: {
fiveHour: {
supported: false,
message: 'official 5h quota window is not exposed by current codex oauth artifacts',
},
sevenDay: {
supported: false,
message: 'official 7d quota window is not exposed by current codex oauth artifacts',
},
},
},
modelDiscoveryStatus: 'abnormal',
lastModelSyncAt: '2026-03-17T08:00:00.000Z',
lastModelSyncError: 'Codex 模型获取失败（HTTP 403: forbidden）',
@@ -853,6 +878,16 @@ describe('oauth routes', { timeout: 15_000 }, () => {
provider: 'codex',
email: 'codex-user@example.com',
status: 'abnormal',
quota: expect.objectContaining({
status: 'supported',
source: 'reverse_engineered',
lastLimitResetAt: '2026-03-17T13:00:00.000Z',
subscription: expect.objectContaining({
planType: 'team',
activeStart: '2026-03-01T00:00:00.000Z',
activeUntil: '2026-04-01T00:00:00.000Z',
}),
}),
routeChannelCount: 1,
lastModelSyncAt: '2026-03-17T08:00:00.000Z',
lastModelSyncError: 'Codex 模型获取失败（HTTP 403: forbidden）',
@@ -874,6 +909,96 @@ describe('oauth routes', { timeout: 15_000 }, () => {
expect(accounts).toEqual([]);
});
it('refreshes oauth quota snapshots and marks unsupported providers explicitly', async () => {
const codexSite = await db.insert(schema.sites).values({
name: 'ChatGPT Codex OAuth',
url: 'https://chatgpt.com/backend-api/codex',
platform: 'codex',
status: 'active',
}).returning().get();
const antigravitySite = await db.insert(schema.sites).values({
name: 'Antigravity OAuth',
url: 'https://example.com/antigravity',
platform: 'antigravity',
status: 'active',
}).returning().get();
const codexAccount = await db.insert(schema.accounts).values({
siteId: codexSite.id,
username: 'codex-user@example.com',
accessToken: 'oauth-access-token',
status: 'active',
oauthProvider: 'codex',
oauthAccountKey: 'chatgpt-account-123',
extraConfig: JSON.stringify({
credentialMode: 'session',
oauth: {
provider: 'codex',
accountId: 'chatgpt-account-123',
email: 'codex-user@example.com',
planType: 'plus',
idToken: buildJwt({
email: 'codex-user@example.com',
'https://api.openai.com/auth': {
chatgpt_account_id: 'chatgpt-account-123',
chatgpt_plan_type: 'plus',
chatgpt_subscription_active_start: '2026-03-01T00:00:00.000Z',
chatgpt_subscription_active_until: '2026-04-01T00:00:00.000Z',
},
}),
},
}),
}).returning().get();
const antigravityAccount = await db.insert(schema.accounts).values({
siteId: antigravitySite.id,
username: 'ag-user@example.com',
accessToken: 'oauth-access-token',
status: 'active',
oauthProvider: 'antigravity',
oauthAccountKey: 'ag-account-123',
extraConfig: JSON.stringify({
credentialMode: 'session',
oauth: {
provider: 'antigravity',
accountId: 'ag-account-123',
email: 'ag-user@example.com',
planType: 'pro',
},
}),
}).returning().get();
const codexRefresh = await app.inject({
method: 'POST',
url: `/api/oauth/connections/${codexAccount.id}/quota/refresh`,
});
expect(codexRefresh.statusCode).toBe(200);
expect(codexRefresh.json()).toMatchObject({
success: true,
quota: expect.objectContaining({
status: 'supported',
subscription: expect.objectContaining({
planType: 'plus',
activeStart: '2026-03-01T00:00:00.000Z',
activeUntil: '2026-04-01T00:00:00.000Z',
}),
}),
});
const antigravityRefresh = await app.inject({
method: 'POST',
url: `/api/oauth/connections/${antigravityAccount.id}/quota/refresh`,
});
expect(antigravityRefresh.statusCode).toBe(200);
expect(antigravityRefresh.json()).toMatchObject({
success: true,
quota: expect.objectContaining({
status: 'unsupported',
providerMessage: 'current local CLIProxyAPI evidence does not expose official quota windows for antigravity oauth',
}),
});
});
it('keeps multiple codex team workspaces with the same email as separate oauth connections', async () => {
const buildTokenExchange = (accountId: string) => ({
ok: true,
src/server/routes/api/oauth.ts
+17
View File
@@ -6,6 +6,7 @@ import {
handleOauthCallback,
listOauthConnections,
listOauthProviders,
refreshOauthConnectionQuota,
startOauthProviderFlow,
startOauthRebindFlow,
submitOauthManualCallback,
@@ -234,6 +235,22 @@ export async function oauthRoutes(app: FastifyInstance) {
},
);
app.post<{ Params: { accountId: string } }>(
'/api/oauth/connections/:accountId/quota/refresh',
{ preHandler: [limitOauthConnectionMutate] },
async (request, reply) => {
const accountId = parsePositiveInteger(request.params.accountId);
if (accountId === null) {
return reply.code(400).send({ message: 'invalid account id' });
}
try {
return await refreshOauthConnectionQuota(accountId);
} catch (error: any) {
return reply.code(404).send({ message: error?.message || 'oauth account not found' });
}
},
);
app.get<{ Params: { provider: string }; Querystring: { state?: string; code?: string; error?: string } }>(
'/api/oauth/callback/:provider',
{ preHandler: [limitOauthCallback] },