feat(proxy): wrap platform adapter calls with account proxy override

Use withAccountProxyOverride to propagate account-level proxy URL
through AsyncLocalStorage for checkin, balance refresh, model
discovery, token sync, rebind-session verify, and token CRUD
operations. Also accept proxyUrl field in PUT /api/accounts/:id
with validation via parseSiteProxyUrlInput.
This commit is contained in:
Hureru
2026-03-16 01:06:40 +08:00
parent 6d2f1c4b81
commit ffc2ad039c
5 changed files with 83 additions and 35 deletions
+19 -6
View File
@@ -12,8 +12,9 @@ import {
syncTokensFromUpstream, syncTokensFromUpstream,
} from '../../services/accountTokenService.js'; } from '../../services/accountTokenService.js';
import { getAdapter } from '../../services/platforms/index.js'; import { getAdapter } from '../../services/platforms/index.js';
import { getCredentialModeFromExtraConfig, resolvePlatformUserId } from '../../services/accountExtraConfig.js'; import { getCredentialModeFromExtraConfig, getProxyUrlFromExtraConfig, resolvePlatformUserId } from '../../services/accountExtraConfig.js';
import { startBackgroundTask } from '../../services/backgroundTaskService.js'; import { startBackgroundTask } from '../../services/backgroundTaskService.js';
import { withAccountProxyOverride } from '../../services/siteProxy.js';
type AccountWithSiteRow = { type AccountWithSiteRow = {
accounts: typeof schema.accounts.$inferSelect; accounts: typeof schema.accounts.$inferSelect;
@@ -231,15 +232,18 @@ async function executeAccountTokenSync(row: AccountWithSiteRow): Promise<SyncExe
try { try {
const platformUserId = resolvePlatformUserId(row.accounts.extraConfig, row.accounts.username); const platformUserId = resolvePlatformUserId(row.accounts.extraConfig, row.accounts.username);
const accountProxyUrl = getProxyUrlFromExtraConfig(row.accounts.extraConfig);
let tokens = await withTimeout( let tokens = await withTimeout(
() => adapter.getApiTokens(row.sites.url, row.accounts.accessToken, platformUserId), () => withAccountProxyOverride(accountProxyUrl,
() => adapter.getApiTokens(row.sites.url, row.accounts.accessToken, platformUserId)),
TOKEN_SYNC_TIMEOUT_MS, TOKEN_SYNC_TIMEOUT_MS,
`token sync timeout (${Math.round(TOKEN_SYNC_TIMEOUT_MS / 1000)}s)`, `token sync timeout (${Math.round(TOKEN_SYNC_TIMEOUT_MS / 1000)}s)`,
); );
if (tokens.length === 0) { if (tokens.length === 0) {
const fallback = await withTimeout( const fallback = await withTimeout(
() => adapter.getApiToken(row.sites.url, row.accounts.accessToken, platformUserId), () => withAccountProxyOverride(accountProxyUrl,
() => adapter.getApiToken(row.sites.url, row.accounts.accessToken, platformUserId)),
TOKEN_SYNC_TIMEOUT_MS, TOKEN_SYNC_TIMEOUT_MS,
`token sync timeout (${Math.round(TOKEN_SYNC_TIMEOUT_MS / 1000)}s)`, `token sync timeout (${Math.round(TOKEN_SYNC_TIMEOUT_MS / 1000)}s)`,
); );
@@ -468,7 +472,9 @@ export async function accountTokensRoutes(app: FastifyInstance) {
} }
const platformUserId = resolvePlatformUserId(account.extraConfig, account.username); const platformUserId = resolvePlatformUserId(account.extraConfig, account.username);
const createdViaUpstream = await adapter.createApiToken( const createdViaUpstream = await withAccountProxyOverride(
getProxyUrlFromExtraConfig(account.extraConfig),
() => adapter.createApiToken(
site.url, site.url,
account.accessToken, account.accessToken,
platformUserId, platformUserId,
@@ -482,6 +488,7 @@ export async function accountTokensRoutes(app: FastifyInstance) {
modelLimitsEnabled, modelLimitsEnabled,
modelLimits: asTrimmedString(body.modelLimits), modelLimits: asTrimmedString(body.modelLimits),
}, },
),
); );
if (!createdViaUpstream) { if (!createdViaUpstream) {
return reply.code(502).send({ success: false, message: '站点创建令牌失败' }); return reply.code(502).send({ success: false, message: '站点创建令牌失败' });
@@ -536,11 +543,14 @@ export async function accountTokensRoutes(app: FastifyInstance) {
if (shouldDeleteUpstream) { if (shouldDeleteUpstream) {
const platformUserId = resolvePlatformUserId(account.extraConfig, account.username); const platformUserId = resolvePlatformUserId(account.extraConfig, account.username);
const upstreamDeleted = await adapter!.deleteApiToken( const upstreamDeleted = await withAccountProxyOverride(
getProxyUrlFromExtraConfig(account.extraConfig),
() => adapter!.deleteApiToken(
site.url, site.url,
account.accessToken, account.accessToken,
existing.token, existing.token,
platformUserId, platformUserId,
),
); );
if (!upstreamDeleted) { if (!upstreamDeleted) {
return { success: false, message: '站点删除令牌失败,本地未删除' }; return { success: false, message: '站点删除令牌失败,本地未删除' };
@@ -771,7 +781,10 @@ export async function accountTokensRoutes(app: FastifyInstance) {
try { try {
const platformUserId = resolvePlatformUserId(account.extraConfig, account.username); const platformUserId = resolvePlatformUserId(account.extraConfig, account.username);
const groups = await adapter.getUserGroups(site.url, account.accessToken, platformUserId); const groups = await withAccountProxyOverride(
getProxyUrlFromExtraConfig(account.extraConfig),
() => adapter.getUserGroups(site.url, account.accessToken, platformUserId),
);
const normalized = Array.from(new Set((groups || []).map((item) => String(item || '').trim()).filter(Boolean))); const normalized = Array.from(new Set((groups || []).map((item) => String(item || '').trim()).filter(Boolean)));
return { success: true, groups: normalized.length > 0 ? normalized : ['default'] }; return { success: true, groups: normalized.length > 0 ? normalized : ['default'] };
} catch (error: any) { } catch (error: any) {
+18 -2
View File
@@ -7,6 +7,7 @@ import { refreshModelsForAccount, rebuildTokenRoutesFromAvailability } from '../
import { ensureDefaultTokenForAccount, syncTokensFromUpstream } from '../../services/accountTokenService.js'; import { ensureDefaultTokenForAccount, syncTokensFromUpstream } from '../../services/accountTokenService.js';
import { import {
getCredentialModeFromExtraConfig, getCredentialModeFromExtraConfig,
getProxyUrlFromExtraConfig,
guessPlatformUserIdFromUsername, guessPlatformUserIdFromUsername,
getSub2ApiAuthFromExtraConfig, getSub2ApiAuthFromExtraConfig,
mergeAccountExtraConfig, mergeAccountExtraConfig,
@@ -25,7 +26,7 @@ import {
type RuntimeHealthState, type RuntimeHealthState,
} from '../../services/accountHealthService.js'; } from '../../services/accountHealthService.js';
import { appendSessionTokenRebindHint } from '../../services/alertRules.js'; import { appendSessionTokenRebindHint } from '../../services/alertRules.js';
import { withSiteRecordProxyRequestInit } from '../../services/siteProxy.js'; import { parseSiteProxyUrlInput, withAccountProxyOverride, withSiteRecordProxyRequestInit } from '../../services/siteProxy.js';
import { createRateLimitGuard } from '../../middleware/requestRateLimit.js'; import { createRateLimitGuard } from '../../middleware/requestRateLimit.js';
type AccountWithSiteRow = { type AccountWithSiteRow = {
@@ -890,7 +891,10 @@ export async function accountsRoutes(app: FastifyInstance) {
let verifyResult: any; let verifyResult: any;
try { try {
verifyResult = await adapter.verifyToken(site.url, nextAccessToken, candidatePlatformUserId); verifyResult = await withAccountProxyOverride(
getProxyUrlFromExtraConfig(account.extraConfig),
() => adapter.verifyToken(site.url, nextAccessToken, candidatePlatformUserId),
);
} catch (err: any) { } catch (err: any) {
return reply.code(400).send({ return reply.code(400).send({
success: false, success: false,
@@ -1210,6 +1214,18 @@ export async function accountsRoutes(app: FastifyInstance) {
updates.sortOrder = normalizedSortOrder; updates.sortOrder = normalizedSortOrder;
} }
if (Object.prototype.hasOwnProperty.call(body, 'proxyUrl')) {
const baseExtraConfig = typeof updates.extraConfig === 'string'
? updates.extraConfig : account.extraConfig;
const { present, valid, proxyUrl: normalizedProxy } = parseSiteProxyUrlInput(body.proxyUrl);
if (present && !valid) {
return reply.code(400).send({ message: 'Invalid proxy URL format' });
}
updates.extraConfig = mergeAccountExtraConfig(baseExtraConfig, {
proxyUrl: normalizedProxy ?? undefined,
});
}
updates.updatedAt = new Date().toISOString(); updates.updatedAt = new Date().toISOString();
await db.update(schema.accounts).set(updates).where(eq(schema.accounts.id, id)).run(); await db.update(schema.accounts).set(updates).where(eq(schema.accounts.id, id)).run();
+9 -3
View File
@@ -6,6 +6,7 @@ import { reportTokenExpired } from './alertService.js';
import { import {
getAutoReloginConfig, getAutoReloginConfig,
getCredentialModeFromExtraConfig, getCredentialModeFromExtraConfig,
getProxyUrlFromExtraConfig,
getSub2ApiAuthFromExtraConfig, getSub2ApiAuthFromExtraConfig,
mergeAccountExtraConfig, mergeAccountExtraConfig,
resolvePlatformUserId, resolvePlatformUserId,
@@ -14,7 +15,7 @@ import { decryptAccountPassword } from './accountCredentialService.js';
import { extractRuntimeHealth, setAccountRuntimeHealth } from './accountHealthService.js'; import { extractRuntimeHealth, setAccountRuntimeHealth } from './accountHealthService.js';
import { updateTodayIncomeSnapshot } from './todayIncomeRewardService.js'; import { updateTodayIncomeSnapshot } from './todayIncomeRewardService.js';
import type { BalanceInfo } from './platforms/base.js'; import type { BalanceInfo } from './platforms/base.js';
import { withSiteProxyRequestInit, withSiteRecordProxyRequestInit } from './siteProxy.js'; import { withAccountProxyOverride, withSiteProxyRequestInit, withSiteRecordProxyRequestInit } from './siteProxy.js';
function isSiteDisabled(status?: string | null): boolean { function isSiteDisabled(status?: string | null): boolean {
return (status || 'active') === 'disabled'; return (status || 'active') === 'disabled';
@@ -316,7 +317,10 @@ async function tryAutoRelogin(account: any, site: any): Promise<string | null> {
const password = decryptAccountPassword(relogin.passwordCipher); const password = decryptAccountPassword(relogin.passwordCipher);
if (!password) return null; if (!password) return null;
const loginResult = await adapter.login(site.url, relogin.username, password); const loginResult = await withAccountProxyOverride(
getProxyUrlFromExtraConfig(account.extraConfig),
() => adapter.login(site.url, relogin.username, password),
);
if (!loginResult.success || !loginResult.accessToken) return null; if (!loginResult.success || !loginResult.accessToken) return null;
await db.update(schema.accounts) await db.update(schema.accounts)
@@ -393,7 +397,9 @@ export async function refreshBalance(accountId: number) {
} }
} }
const readBalance = async (token: string) => adapter.getBalance(site.url, token, platformUserId); const accountProxyUrl = getProxyUrlFromExtraConfig(account.extraConfig);
const readBalance = async (token: string) => withAccountProxyOverride(accountProxyUrl,
() => adapter.getBalance(site.url, token, platformUserId));
const handleBalanceError = async (err: any) => { const handleBalanceError = async (err: any) => {
const message = appendSessionTokenRebindHint(err?.message || 'unknown error'); const message = appendSessionTokenRebindHint(err?.message || 'unknown error');
setAccountRuntimeHealth(account.id, { setAccountRuntimeHealth(account.id, {
+11 -3
View File
@@ -8,6 +8,7 @@ import { refreshBalance } from './balanceService.js';
import { parseCheckinRewardAmount } from './checkinRewardParser.js'; import { parseCheckinRewardAmount } from './checkinRewardParser.js';
import { import {
getAutoReloginConfig, getAutoReloginConfig,
getProxyUrlFromExtraConfig,
getPlatformUserIdFromExtraConfig, getPlatformUserIdFromExtraConfig,
guessPlatformUserIdFromUsername, guessPlatformUserIdFromUsername,
mergeAccountExtraConfig, mergeAccountExtraConfig,
@@ -16,6 +17,7 @@ import {
import { decryptAccountPassword } from './accountCredentialService.js'; import { decryptAccountPassword } from './accountCredentialService.js';
import { setAccountRuntimeHealth } from './accountHealthService.js'; import { setAccountRuntimeHealth } from './accountHealthService.js';
import { formatUtcSqlDateTime } from './localTimeService.js'; import { formatUtcSqlDateTime } from './localTimeService.js';
import { withAccountProxyOverride } from './siteProxy.js';
type CheckinExecutionStatus = 'success' | 'failed' | 'skipped'; type CheckinExecutionStatus = 'success' | 'failed' | 'skipped';
@@ -100,7 +102,10 @@ async function tryAutoRelogin(account: any, site: any): Promise<string | null> {
const password = decryptAccountPassword(relogin.passwordCipher); const password = decryptAccountPassword(relogin.passwordCipher);
if (!password) return null; if (!password) return null;
const result = await adapter.login(site.url, relogin.username, password); const result = await withAccountProxyOverride(
getProxyUrlFromExtraConfig(account.extraConfig),
() => adapter.login(site.url, relogin.username, password),
);
if (!result.success || !result.accessToken) return null; if (!result.success || !result.accessToken) return null;
await db.update(schema.accounts) await db.update(schema.accounts)
@@ -172,14 +177,17 @@ export async function checkinAccount(accountId: number, options?: { skipEvent?:
: guessPlatformUserIdFromUsername(account.username); : guessPlatformUserIdFromUsername(account.username);
const platformUserId = resolvePlatformUserId(account.extraConfig, account.username); const platformUserId = resolvePlatformUserId(account.extraConfig, account.username);
const accountProxyUrl = getProxyUrlFromExtraConfig(account.extraConfig);
let activeAccessToken = account.accessToken; let activeAccessToken = account.accessToken;
let result = await adapter.checkin(site.url, activeAccessToken, platformUserId); let result = await withAccountProxyOverride(accountProxyUrl,
() => adapter.checkin(site.url, activeAccessToken, platformUserId));
if (!result.success && shouldAttemptAutoRelogin(result.message)) { if (!result.success && shouldAttemptAutoRelogin(result.message)) {
const refreshedAccessToken = await tryAutoRelogin(account, site); const refreshedAccessToken = await tryAutoRelogin(account, site);
if (refreshedAccessToken) { if (refreshedAccessToken) {
activeAccessToken = refreshedAccessToken; activeAccessToken = refreshedAccessToken;
result = await adapter.checkin(site.url, activeAccessToken, platformUserId); result = await withAccountProxyOverride(accountProxyUrl,
() => adapter.checkin(site.url, activeAccessToken, platformUserId));
} }
} }
+9 -4
View File
@@ -2,10 +2,11 @@ import { and, eq } from 'drizzle-orm';
import { db, schema } from '../db/index.js'; import { db, schema } from '../db/index.js';
import { getAdapter } from './platforms/index.js'; import { getAdapter } from './platforms/index.js';
import { ensureDefaultTokenForAccount, getPreferredAccountToken } from './accountTokenService.js'; import { ensureDefaultTokenForAccount, getPreferredAccountToken } from './accountTokenService.js';
import { getCredentialModeFromExtraConfig, resolvePlatformUserId } from './accountExtraConfig.js'; import { getCredentialModeFromExtraConfig, getProxyUrlFromExtraConfig, resolvePlatformUserId } from './accountExtraConfig.js';
import { invalidateTokenRouterCache } from './tokenRouter.js'; import { invalidateTokenRouterCache } from './tokenRouter.js';
import { setAccountRuntimeHealth } from './accountHealthService.js'; import { setAccountRuntimeHealth } from './accountHealthService.js';
import { clearAllRouteDecisionSnapshots } from './routeDecisionSnapshotStore.js'; import { clearAllRouteDecisionSnapshots } from './routeDecisionSnapshotStore.js';
import { withAccountProxyOverride } from './siteProxy.js';
const API_TOKEN_DISCOVERY_TIMEOUT_MS = 8_000; const API_TOKEN_DISCOVERY_TIMEOUT_MS = 8_000;
const MODEL_DISCOVERY_TIMEOUT_MS = 12_000; const MODEL_DISCOVERY_TIMEOUT_MS = 12_000;
@@ -129,12 +130,14 @@ export async function refreshModelsForAccount(accountId: number) {
} }
const platformUserId = resolvePlatformUserId(account.extraConfig, account.username); const platformUserId = resolvePlatformUserId(account.extraConfig, account.username);
const accountProxyUrl = getProxyUrlFromExtraConfig(account.extraConfig);
let discoveredApiToken: string | null = null; let discoveredApiToken: string | null = null;
if (!account.apiToken && account.accessToken) { if (!account.apiToken && account.accessToken) {
try { try {
discoveredApiToken = await withTimeout( discoveredApiToken = await withTimeout(
() => adapter.getApiToken(site.url, account.accessToken, platformUserId), () => withAccountProxyOverride(accountProxyUrl,
() => adapter.getApiToken(site.url, account.accessToken, platformUserId)),
API_TOKEN_DISCOVERY_TIMEOUT_MS, API_TOKEN_DISCOVERY_TIMEOUT_MS,
`api token discovery timeout (${Math.round(API_TOKEN_DISCOVERY_TIMEOUT_MS / 1000)}s)`, `api token discovery timeout (${Math.round(API_TOKEN_DISCOVERY_TIMEOUT_MS / 1000)}s)`,
); );
@@ -200,7 +203,8 @@ export async function refreshModelsForAccount(accountId: number) {
try { try {
models = normalizeModels( models = normalizeModels(
await withTimeout( await withTimeout(
() => adapter.getModels(site.url, credential, platformUserId), () => withAccountProxyOverride(accountProxyUrl,
() => adapter.getModels(site.url, credential, platformUserId)),
MODEL_DISCOVERY_TIMEOUT_MS, MODEL_DISCOVERY_TIMEOUT_MS,
`model discovery timeout (${Math.round(MODEL_DISCOVERY_TIMEOUT_MS / 1000)}s)`, `model discovery timeout (${Math.round(MODEL_DISCOVERY_TIMEOUT_MS / 1000)}s)`,
), ),
@@ -227,7 +231,8 @@ export async function refreshModelsForAccount(accountId: number) {
try { try {
models = normalizeModels( models = normalizeModels(
await withTimeout( await withTimeout(
() => adapter.getModels(site.url, token.token, platformUserId), () => withAccountProxyOverride(accountProxyUrl,
() => adapter.getModels(site.url, token.token, platformUserId)),
MODEL_DISCOVERY_TIMEOUT_MS, MODEL_DISCOVERY_TIMEOUT_MS,
`model discovery timeout (${Math.round(MODEL_DISCOVERY_TIMEOUT_MS / 1000)}s)`, `model discovery timeout (${Math.round(MODEL_DISCOVERY_TIMEOUT_MS / 1000)}s)`,
), ),