fix(codex): durable OAuth refresh lifecycle

Add shared OAuth credential lifecycle manager with provider-aware refresh
decisions. Implement CodexExecutor.refreshCredentials so 401/403 retry
refresh works for Codex, track lastRefreshAt and refresh before the
upstream stale-token window, preserve omitted idToken, and add
per-connection single-flight refresh to avoid refresh-token rotation races.

Merged from PR #1664.

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
Kevin Le
2026-06-06 11:04:36 +07:00
committed by decolua
co-authored by Cursor
parent 38b73bfc6b
commit c233c7c8fc
15 changed files with 484 additions and 140 deletions
+2 -3
View File
@@ -1,4 +1,5 @@
import { HTTP_STATUS, RETRY_CONFIG, DEFAULT_RETRY_CONFIG, resolveRetryEntry, FETCH_CONNECT_TIMEOUT_MS } from "../config/runtimeConfig.js"; import { HTTP_STATUS, RETRY_CONFIG, DEFAULT_RETRY_CONFIG, resolveRetryEntry, FETCH_CONNECT_TIMEOUT_MS } from "../config/runtimeConfig.js";
import { shouldRefreshCredentials } from "../services/oauthCredentialManager.js";
import { proxyAwareFetch } from "../utils/proxyFetch.js"; import { proxyAwareFetch } from "../utils/proxyFetch.js";
import { dbg } from "../utils/debugLog.js"; import { dbg } from "../utils/debugLog.js";
@@ -87,9 +88,7 @@ export class BaseExecutor {
} }
needsRefresh(credentials) { needsRefresh(credentials) {
if (!credentials.expiresAt) return false; return shouldRefreshCredentials(this.provider, credentials);
const expiresAtMs = new Date(credentials.expiresAt).getTime();
return expiresAtMs - Date.now() < 5 * 60 * 1000;
} }
parseError(response, bodyText) { parseError(response, bodyText) {
+13
View File
@@ -2,6 +2,10 @@ import { createHash } from "crypto";
import { BaseExecutor } from "./base.js"; import { BaseExecutor } from "./base.js";
import { CODEX_DEFAULT_INSTRUCTIONS } from "../config/codexInstructions.js"; import { CODEX_DEFAULT_INSTRUCTIONS } from "../config/codexInstructions.js";
import { PROVIDERS } from "../config/providers.js"; import { PROVIDERS } from "../config/providers.js";
import {
refreshProviderCredentials,
shouldRefreshCredentials,
} from "../services/oauthCredentialManager.js";
import { normalizeResponsesInput } from "../translator/helpers/responsesApiHelper.js"; import { normalizeResponsesInput } from "../translator/helpers/responsesApiHelper.js";
import { fetchImageAsBase64 } from "../translator/helpers/imageHelper.js"; import { fetchImageAsBase64 } from "../translator/helpers/imageHelper.js";
import { getModelUpstreamId } from "../config/providerModels.js"; import { getModelUpstreamId } from "../config/providerModels.js";
@@ -212,6 +216,15 @@ export class CodexExecutor extends BaseExecutor {
return this._isCompact ? `${base}/compact` : base; return this._isCompact ? `${base}/compact` : base;
} }
async refreshCredentials(credentials, log) {
if (!credentials?.refreshToken) return null;
return refreshProviderCredentials("codex", credentials, log);
}
needsRefresh(credentials) {
return shouldRefreshCredentials("codex", credentials);
}
/** /**
* Prefetch remote image URLs and inline them as base64 data URIs. * Prefetch remote image URLs and inline them as base64 data URIs.
* Runs before execute() because Codex backend cannot fetch remote images. * Runs before execute() because Codex backend cannot fetch remote images.
+8
View File
@@ -59,6 +59,14 @@ export {
refreshTokenByProvider refreshTokenByProvider
} from "./services/tokenRefresh.js"; } from "./services/tokenRefresh.js";
export {
CODEX_MAX_REFRESH_AGE_MS,
shouldRefreshCredentials,
refreshProviderCredentials,
mergeRefreshedCredentials,
mergeProviderSpecificData,
} from "./services/oauthCredentialManager.js";
// Handlers // Handlers
export { handleChatCore, isTokenExpiringSoon } from "./handlers/chatCore.js"; export { handleChatCore, isTokenExpiringSoon } from "./handlers/chatCore.js";
export { createStreamController, pipeWithDisconnect, createDisconnectAwareStream } from "./utils/streamHandler.js"; export { createStreamController, pipeWithDisconnect, createDisconnectAwareStream } from "./utils/streamHandler.js";
+151
View File
@@ -0,0 +1,151 @@
import {
getRefreshLeadMs,
isUnrecoverableRefreshError,
refreshTokenByProvider,
} from "./tokenRefresh.js";
export const CODEX_MAX_REFRESH_AGE_MS = 8 * 24 * 60 * 60 * 1000;
const refreshLocks = new Map();
function parseTimeMs(value) {
if (value === undefined || value === null || value === "") return null;
if (typeof value === "number") {
return value < 1e12 ? value * 1000 : value;
}
const parsed = new Date(value).getTime();
return Number.isFinite(parsed) ? parsed : null;
}
function toExpiresAt(expiresIn, nowMs = Date.now()) {
if (!expiresIn) return null;
return new Date(nowMs + expiresIn * 1000).toISOString();
}
export function getCredentialExpiryMs(credentials) {
return parseTimeMs(credentials?.expiresAt ?? credentials?.tokenExpiresAt);
}
export function getCredentialLastRefreshMs(credentials) {
return parseTimeMs(
credentials?.lastRefreshAt ??
credentials?.lastRefresh ??
credentials?.providerSpecificData?.lastRefreshAt
);
}
export function isCodexRefreshStale(credentials, nowMs = Date.now()) {
const lastRefreshMs = getCredentialLastRefreshMs(credentials);
return !lastRefreshMs || nowMs - lastRefreshMs >= CODEX_MAX_REFRESH_AGE_MS;
}
export function shouldRefreshCredentials(provider, credentials, nowMs = Date.now()) {
if (!credentials) return false;
const expiresAtMs = getCredentialExpiryMs(credentials);
if (expiresAtMs !== null && expiresAtMs - nowMs < getRefreshLeadMs(provider)) {
return true;
}
if (provider === "codex" && credentials.refreshToken && isCodexRefreshStale(credentials, nowMs)) {
return true;
}
return false;
}
export function mergeProviderSpecificData(existing, next) {
if (!next || typeof next !== "object") return existing;
return {
...(existing || {}),
...next,
};
}
export function mergeRefreshedCredentials(provider, currentCredentials, refreshedCredentials, nowMs = Date.now()) {
if (!refreshedCredentials) return null;
if (isUnrecoverableRefreshError(refreshedCredentials)) return refreshedCredentials;
const next = {};
const nowIso = new Date(nowMs).toISOString();
if (refreshedCredentials.accessToken) next.accessToken = refreshedCredentials.accessToken;
if (refreshedCredentials.apiKey) next.apiKey = refreshedCredentials.apiKey;
if (refreshedCredentials.token) next.token = refreshedCredentials.token;
const refreshToken = refreshedCredentials.refreshToken ?? currentCredentials?.refreshToken;
if (refreshToken) next.refreshToken = refreshToken;
const idToken = refreshedCredentials.idToken ?? currentCredentials?.idToken;
if (idToken) next.idToken = idToken;
if (refreshedCredentials.expiresIn) {
next.expiresIn = refreshedCredentials.expiresIn;
next.expiresAt = toExpiresAt(refreshedCredentials.expiresIn, nowMs);
} else if (refreshedCredentials.expiresAt) {
next.expiresAt = refreshedCredentials.expiresAt;
}
if (refreshedCredentials.projectId) next.projectId = refreshedCredentials.projectId;
if (refreshedCredentials.providerSpecificData) {
next.providerSpecificData = mergeProviderSpecificData(
currentCredentials?.providerSpecificData,
refreshedCredentials.providerSpecificData
);
}
if (refreshedCredentials.copilotToken) next.copilotToken = refreshedCredentials.copilotToken;
if (refreshedCredentials.copilotTokenExpiresAt) {
next.copilotTokenExpiresAt = refreshedCredentials.copilotTokenExpiresAt;
}
if (
provider === "codex" ||
next.accessToken ||
next.apiKey ||
next.token ||
next.refreshToken ||
next.copilotToken
) {
next.lastRefreshAt = refreshedCredentials.lastRefreshAt || nowIso;
}
return next;
}
function getRefreshLockKey(provider, credentials) {
const stableId =
credentials?.connectionId ||
credentials?.id ||
credentials?.email ||
credentials?.name ||
credentials?.refreshToken?.slice?.(-16) ||
"default";
return `${provider}:${stableId}`;
}
export async function withCredentialRefreshLock(provider, credentials, refreshFn) {
const key = getRefreshLockKey(provider, credentials);
const existing = refreshLocks.get(key);
if (existing) return existing;
const pending = Promise.resolve()
.then(refreshFn)
.finally(() => {
refreshLocks.delete(key);
});
refreshLocks.set(key, pending);
return pending;
}
export async function refreshProviderCredentials(provider, credentials, log) {
if (!credentials) return null;
return withCredentialRefreshLock(provider, credentials, async () => {
const refreshed = await refreshTokenByProvider(provider, credentials, log);
return mergeRefreshedCredentials(provider, credentials, refreshed);
});
}
+71 -59
View File
@@ -277,6 +277,27 @@ export async function refreshQwenToken(refreshToken, log) {
}, log); }, log);
} }
export function classifyOAuthRefreshError(errorText = "", status = 0) {
let parsed = null;
try {
parsed = errorText ? JSON.parse(errorText) : null;
} catch {
parsed = null;
}
const code = parsed?.error?.code || parsed?.error || parsed?.error_code || "";
const description = parsed?.error_description || parsed?.message || errorText || "";
const combined = `${code} ${description}`.toLowerCase();
const permanent = [
"refresh_token_expired",
"refresh_token_reused",
"refresh_token_invalidated",
"invalid_grant",
].some((marker) => combined.includes(marker));
return { status, code, description, permanent };
}
/** /**
* Specialized refresh for Codex (OpenAI) OAuth tokens. * Specialized refresh for Codex (OpenAI) OAuth tokens.
* OpenAI uses rotating (one-time-use) refresh tokens. * OpenAI uses rotating (one-time-use) refresh tokens.
@@ -286,68 +307,59 @@ export async function refreshQwenToken(refreshToken, log) {
export async function refreshCodexToken(refreshToken, log) { export async function refreshCodexToken(refreshToken, log) {
if (!refreshToken) return null; if (!refreshToken) return null;
return dedupRefresh("codex", refreshToken, async () => { return dedupRefresh("codex", refreshToken, async () => {
try {
const response = await fetch(OAUTH_ENDPOINTS.openai.token, {
method: "POST",
headers: {
"Content-Type": "application/x-www-form-urlencoded",
Accept: "application/json",
},
body: new URLSearchParams({
grant_type: "refresh_token",
refresh_token: refreshToken,
client_id: PROVIDERS.codex.clientId,
scope: "openid profile email offline_access",
}),
});
if (!response.ok) {
const errorText = await response.text();
// Detect unrecoverable errors (token reused/expired) — Auth0 revokes whole family on retry
let errorCode = null;
try { try {
const parsed = JSON.parse(errorText); const response = await fetch(OAUTH_ENDPOINTS.openai.token, {
errorCode = parsed?.error?.code || (typeof parsed?.error === "string" ? parsed.error : null); method: "POST",
} catch {} headers: {
"Content-Type": "application/json",
if ( Accept: "application/json",
errorCode === "refresh_token_reused" || },
errorCode === "invalid_grant" || body: JSON.stringify({
errorCode === "token_expired" || client_id: PROVIDERS.codex.clientId,
errorCode === "invalid_token" grant_type: "refresh_token",
) { refresh_token: refreshToken,
log?.error?.("TOKEN_REFRESH", "Codex refresh token already used or invalid. Re-auth required.", { }),
status: response.status,
errorCode,
}); });
return { error: "unrecoverable_refresh_error", code: errorCode };
if (!response.ok) {
const errorText = await response.text();
const failure = classifyOAuthRefreshError(errorText, response.status);
if (failure.permanent) {
log?.error?.("TOKEN_REFRESH", "Codex refresh token already used or invalid. Re-auth required.", {
status: response.status,
code: failure.code,
});
return { error: "unrecoverable_refresh_error", code: failure.code };
}
log?.error?.("TOKEN_REFRESH", "Failed to refresh Codex token", {
status: response.status,
error: errorText,
code: failure.code,
permanent: failure.permanent,
});
return null;
}
const tokens = await response.json();
log?.info?.("TOKEN_REFRESH", "Successfully refreshed Codex token", {
hasNewAccessToken: !!tokens.access_token,
hasNewRefreshToken: !!tokens.refresh_token,
hasIdToken: !!tokens.id_token,
expiresIn: tokens.expires_in,
});
return {
accessToken: tokens.access_token,
refreshToken: tokens.refresh_token || refreshToken,
idToken: tokens.id_token,
expiresIn: tokens.expires_in,
};
} catch (error) {
log?.error?.("TOKEN_REFRESH", `Network error refreshing Codex token: ${error.message}`);
return null;
} }
log?.error?.("TOKEN_REFRESH", "Failed to refresh Codex token", {
status: response.status,
error: errorText,
});
return null;
}
const tokens = await response.json();
log?.info?.("TOKEN_REFRESH", "Successfully refreshed Codex token", {
hasNewAccessToken: !!tokens.access_token,
hasNewRefreshToken: !!tokens.refresh_token,
expiresIn: tokens.expires_in,
});
return {
accessToken: tokens.access_token,
refreshToken: tokens.refresh_token || refreshToken,
expiresIn: tokens.expires_in,
};
} catch (error) {
log?.error?.("TOKEN_REFRESH", `Network error refreshing Codex token: ${error.message}`);
return null;
}
}, log); }, log);
} }
+19 -19
View File
@@ -5,10 +5,13 @@ import { isOpenAICompatibleProvider, isAnthropicCompatibleProvider } from "@/sha
import { PROVIDER_ENDPOINTS } from "@/shared/constants/config"; import { PROVIDER_ENDPOINTS } from "@/shared/constants/config";
import { getDefaultModel } from "open-sse/config/providerModels.js"; import { getDefaultModel } from "open-sse/config/providerModels.js";
import { resolveOllamaLocalHost } from "open-sse/config/providers.js"; import { resolveOllamaLocalHost } from "open-sse/config/providers.js";
import {
refreshProviderCredentials,
shouldRefreshCredentials,
} from "open-sse/services/oauthCredentialManager.js";
import { import {
GEMINI_CONFIG, GEMINI_CONFIG,
ANTIGRAVITY_CONFIG, ANTIGRAVITY_CONFIG,
CODEX_CONFIG,
KIRO_CONFIG, KIRO_CONFIG,
QWEN_CONFIG, QWEN_CONFIG,
CLAUDE_CONFIG, CLAUDE_CONFIG,
@@ -126,18 +129,7 @@ async function refreshOAuthToken(connection) {
} }
if (provider === "codex") { if (provider === "codex") {
const response = await fetch(CODEX_CONFIG.tokenUrl, { return await refreshProviderCredentials(provider, connection, console);
method: "POST",
headers: { "Content-Type": "application/x-www-form-urlencoded" },
body: new URLSearchParams({
grant_type: "refresh_token",
client_id: CODEX_CONFIG.clientId,
refresh_token: refreshToken,
}),
});
if (!response.ok) return null;
const data = await response.json();
return { accessToken: data.access_token, expiresIn: data.expires_in, refreshToken: data.refresh_token || refreshToken };
} }
if (provider === "claude") { if (provider === "claude") {
@@ -227,10 +219,7 @@ async function refreshOAuthToken(connection) {
} }
function isTokenExpired(connection) { function isTokenExpired(connection) {
if (!connection.expiresAt) return false; return shouldRefreshCredentials(connection.provider, connection);
const expiresAt = new Date(connection.expiresAt).getTime();
const buffer = 5 * 60 * 1000;
return expiresAt <= Date.now() + buffer;
} }
async function testOAuthConnection(connection, effectiveProxy = null) { async function testOAuthConnection(connection, effectiveProxy = null) {
@@ -673,14 +662,25 @@ export async function testSingleConnection(id) {
}; };
if (result.refreshed && result.newTokens) { if (result.refreshed && result.newTokens) {
updateData.accessToken = result.newTokens.accessToken; if (result.newTokens.accessToken) updateData.accessToken = result.newTokens.accessToken;
if (result.newTokens.refreshToken) updateData.refreshToken = result.newTokens.refreshToken; if (result.newTokens.refreshToken) updateData.refreshToken = result.newTokens.refreshToken;
if (result.newTokens.idToken) updateData.idToken = result.newTokens.idToken;
if (result.newTokens.lastRefreshAt) updateData.lastRefreshAt = result.newTokens.lastRefreshAt;
if (result.newTokens.expiresIn) updateData.expiresIn = result.newTokens.expiresIn;
if (result.newTokens.expiresIn) { if (result.newTokens.expiresIn) {
updateData.expiresAt = new Date(Date.now() + result.newTokens.expiresIn * 1000).toISOString(); updateData.expiresAt = new Date(Date.now() + result.newTokens.expiresIn * 1000).toISOString();
} else if (result.newTokens.expiresAt) {
updateData.expiresAt = result.newTokens.expiresAt;
}
if (result.newTokens.providerSpecificData) {
updateData.providerSpecificData = {
...(connection.providerSpecificData || {}),
...result.newTokens.providerSpecificData,
};
} }
} }
await updateProviderConnection(id, updateData); await updateProviderConnection(id, updateData);
return { valid: result.valid, error: result.error, latencyMs, testedAt: new Date().toISOString() }; return { valid: result.valid, error: result.error, refreshed: !!result.refreshed, latencyMs, testedAt: new Date().toISOString() };
} }
+40 -4
View File
@@ -1,5 +1,36 @@
import { getProviderConnections } from "@/lib/localDb.js"; import { getProviderConnections, updateProviderConnection } from "@/lib/localDb.js";
import { getExecutor, refreshTokenByProvider } from "open-sse/index.js"; import { getExecutor } from "open-sse/index.js";
async function persistRefreshedCredentials(connection, newCredentials) {
const updateData = {};
if (newCredentials.accessToken) updateData.accessToken = newCredentials.accessToken;
if (newCredentials.refreshToken) updateData.refreshToken = newCredentials.refreshToken;
if (newCredentials.idToken) updateData.idToken = newCredentials.idToken;
if (newCredentials.lastRefreshAt) updateData.lastRefreshAt = newCredentials.lastRefreshAt;
if (newCredentials.expiresIn) {
updateData.expiresIn = newCredentials.expiresIn;
updateData.expiresAt = new Date(Date.now() + newCredentials.expiresIn * 1000).toISOString();
} else if (newCredentials.expiresAt) {
updateData.expiresAt = newCredentials.expiresAt;
}
const providerSpecificUpdates = {
...(newCredentials.providerSpecificData || {}),
...(newCredentials.copilotToken ? { copilotToken: newCredentials.copilotToken } : {}),
...(newCredentials.copilotTokenExpiresAt ? { copilotTokenExpiresAt: newCredentials.copilotTokenExpiresAt } : {}),
};
if (Object.keys(providerSpecificUpdates).length > 0) {
updateData.providerSpecificData = {
...(connection.providerSpecificData || {}),
...providerSpecificUpdates,
};
}
if (Object.keys(updateData).length > 0) {
await updateProviderConnection(connection.id, updateData);
}
}
export async function POST(request) { export async function POST(request) {
try { try {
@@ -19,7 +50,11 @@ export async function POST(request) {
apiKey: connection.apiKey, apiKey: connection.apiKey,
accessToken: connection.accessToken, accessToken: connection.accessToken,
refreshToken: connection.refreshToken, refreshToken: connection.refreshToken,
copilotToken: connection.copilotToken, idToken: connection.idToken,
lastRefreshAt: connection.lastRefreshAt,
connectionId: connection.id,
copilotToken: connection.providerSpecificData?.copilotToken,
copilotTokenExpiresAt: connection.providerSpecificData?.copilotTokenExpiresAt,
projectId: connection.projectId, projectId: connection.projectId,
providerSpecificData: connection.providerSpecificData providerSpecificData: connection.providerSpecificData
}; };
@@ -31,9 +66,10 @@ export async function POST(request) {
// Auto-refresh token on 401/403 and retry (same as chatCore.js) // Auto-refresh token on 401/403 and retry (same as chatCore.js)
if (response.status === 401 || response.status === 403) { if (response.status === 401 || response.status === 403) {
const newCredentials = await refreshTokenByProvider(provider, credentials); const newCredentials = await executor.refreshCredentials(credentials, console);
if (newCredentials?.accessToken || newCredentials?.copilotToken) { if (newCredentials?.accessToken || newCredentials?.copilotToken) {
Object.assign(credentials, newCredentials); Object.assign(credentials, newCredentials);
await persistRefreshedCredentials(connection, newCredentials);
({ response } = await executor.execute({ model, body, stream, credentials })); ({ response } = await executor.execute({ model, body, stream, credentials }));
} }
} }
+21 -4
View File
@@ -27,7 +27,10 @@ async function refreshAndUpdateCredentials(connection, force = false, proxyOptio
const credentials = { const credentials = {
accessToken: connection.accessToken, accessToken: connection.accessToken,
refreshToken: connection.refreshToken, refreshToken: connection.refreshToken,
idToken: connection.idToken,
expiresAt: connection.expiresAt || connection.tokenExpiresAt, expiresAt: connection.expiresAt || connection.tokenExpiresAt,
lastRefreshAt: connection.lastRefreshAt,
connectionId: connection.id,
providerSpecificData: connection.providerSpecificData, providerSpecificData: connection.providerSpecificData,
// For GitHub // For GitHub
copilotToken: connection.providerSpecificData?.copilotToken, copilotToken: connection.providerSpecificData?.copilotToken,
@@ -68,19 +71,32 @@ async function refreshAndUpdateCredentials(connection, force = false, proxyOptio
updateData.refreshToken = refreshResult.refreshToken; updateData.refreshToken = refreshResult.refreshToken;
} }
if (refreshResult.idToken) {
updateData.idToken = refreshResult.idToken;
}
if (refreshResult.lastRefreshAt) {
updateData.lastRefreshAt = refreshResult.lastRefreshAt;
}
// Update token expiry // Update token expiry
if (refreshResult.expiresIn) { if (refreshResult.expiresIn) {
updateData.expiresAt = new Date(Date.now() + refreshResult.expiresIn * 1000).toISOString(); updateData.expiresAt = new Date(Date.now() + refreshResult.expiresIn * 1000).toISOString();
updateData.expiresIn = refreshResult.expiresIn;
} else if (refreshResult.expiresAt) { } else if (refreshResult.expiresAt) {
updateData.expiresAt = refreshResult.expiresAt; updateData.expiresAt = refreshResult.expiresAt;
} }
// Handle provider-specific data (copilotToken for GitHub, etc.) // Handle provider-specific data (copilotToken for GitHub, etc.)
if (refreshResult.copilotToken || refreshResult.copilotTokenExpiresAt) { const providerSpecificUpdates = {
...(refreshResult.providerSpecificData || {}),
...(refreshResult.copilotToken ? { copilotToken: refreshResult.copilotToken } : {}),
...(refreshResult.copilotTokenExpiresAt ? { copilotTokenExpiresAt: refreshResult.copilotTokenExpiresAt } : {}),
};
if (Object.keys(providerSpecificUpdates).length > 0) {
updateData.providerSpecificData = { updateData.providerSpecificData = {
...connection.providerSpecificData, ...(connection.providerSpecificData || {}),
copilotToken: refreshResult.copilotToken, ...providerSpecificUpdates,
copilotTokenExpiresAt: refreshResult.copilotTokenExpiresAt,
}; };
} }
@@ -91,6 +107,7 @@ async function refreshAndUpdateCredentials(connection, force = false, proxyOptio
const updatedConnection = { const updatedConnection = {
...connection, ...connection,
...updateData, ...updateData,
providerSpecificData: updateData.providerSpecificData || connection.providerSpecificData,
}; };
return { return {
+4 -1
View File
@@ -225,9 +225,12 @@ const PROVIDERS = {
const mapped = { const mapped = {
accessToken: tokens.access_token, accessToken: tokens.access_token,
refreshToken: tokens.refresh_token, refreshToken: tokens.refresh_token,
idToken: tokens.id_token,
expiresIn: tokens.expires_in, expiresIn: tokens.expires_in,
lastRefreshAt: new Date().toISOString(),
}; };
if (info.email) mapped.email = info.email; const email = info.email || extractEmailFromAccessToken(tokens.access_token);
if (email) mapped.email = email;
if (info.chatgptAccountId || info.chatgptPlanType) { if (info.chatgptAccountId || info.chatgptPlanType) {
mapped.providerSpecificData = { mapped.providerSpecificData = {
chatgptAccountId: info.chatgptAccountId, chatgptAccountId: info.chatgptAccountId,
+1 -1
View File
@@ -54,6 +54,7 @@ export class CodexService extends OAuthService {
accessToken: tokens.access_token, accessToken: tokens.access_token,
refreshToken: tokens.refresh_token, refreshToken: tokens.refresh_token,
expiresIn: tokens.expires_in, expiresIn: tokens.expires_in,
lastRefreshAt: new Date().toISOString(),
}), }),
}); });
@@ -141,4 +142,3 @@ export class CodexService extends OAuthService {
} }
} }
} }
+2 -3
View File
@@ -218,9 +218,8 @@ async function handleSingleModelChat(body, modelStr, clientRawRequest = null, re
sourceFormatOverride: request?.url ? detectFormatByEndpoint(new URL(request.url).pathname, body) : null, sourceFormatOverride: request?.url ? detectFormatByEndpoint(new URL(request.url).pathname, body) : null,
onCredentialsRefreshed: async (newCreds) => { onCredentialsRefreshed: async (newCreds) => {
await updateProviderCredentials(credentials.connectionId, { await updateProviderCredentials(credentials.connectionId, {
accessToken: newCreds.accessToken, ...newCreds,
refreshToken: newCreds.refreshToken, existingProviderSpecificData: credentials.providerSpecificData,
providerSpecificData: newCreds.providerSpecificData,
testStatus: "active" testStatus: "active"
}); });
}, },
+2 -3
View File
@@ -114,9 +114,8 @@ export async function handleEmbeddings(request) {
log, log,
onCredentialsRefreshed: async (newCreds) => { onCredentialsRefreshed: async (newCreds) => {
await updateProviderCredentials(credentials.connectionId, { await updateProviderCredentials(credentials.connectionId, {
accessToken: newCreds.accessToken, ...newCreds,
refreshToken: newCreds.refreshToken, existingProviderSpecificData: credentials.providerSpecificData,
providerSpecificData: newCreds.providerSpecificData,
testStatus: "active" testStatus: "active"
}); });
}, },
+4
View File
@@ -163,6 +163,10 @@ export async function getProviderCredentials(provider, excludeConnectionIds = nu
apiKey: connection.apiKey, apiKey: connection.apiKey,
accessToken: connection.accessToken, accessToken: connection.accessToken,
refreshToken: connection.refreshToken, refreshToken: connection.refreshToken,
idToken: connection.idToken,
expiresAt: connection.expiresAt,
expiresIn: connection.expiresIn,
lastRefreshAt: connection.lastRefreshAt,
projectId: connection.projectId, projectId: connection.projectId,
connectionName: connection.displayName || connection.name || connection.email || connection.id, connectionName: connection.displayName || connection.name || connection.email || connection.id,
copilotToken: connection.providerSpecificData?.copilotToken, copilotToken: connection.providerSpecificData?.copilotToken,
+47 -33
View File
@@ -23,6 +23,10 @@ import {
refreshKiroToken as _refreshKiroToken, refreshKiroToken as _refreshKiroToken,
getRefreshLeadMs as _getRefreshLeadMs getRefreshLeadMs as _getRefreshLeadMs
} from "open-sse/services/tokenRefresh.js"; } from "open-sse/services/tokenRefresh.js";
import {
refreshProviderCredentials as _refreshProviderCredentials,
shouldRefreshCredentials as _shouldRefreshCredentials,
} from "open-sse/services/oauthCredentialManager.js";
export const TOKEN_EXPIRY_BUFFER_MS = BUFFER_MS; export const TOKEN_EXPIRY_BUFFER_MS = BUFFER_MS;
@@ -67,6 +71,9 @@ export const formatProviderCredentials = (provider, credentials) =>
export const getAllAccessTokens = (userInfo) => export const getAllAccessTokens = (userInfo) =>
_getAllAccessTokens(userInfo, log); _getAllAccessTokens(userInfo, log);
export const shouldRefreshCredentials = (provider, credentials) =>
_shouldRefreshCredentials(provider, credentials);
// ─── Lifecycle hook ─────────────────────────────────────────────────────────── // ─── Lifecycle hook ───────────────────────────────────────────────────────────
/** /**
@@ -158,6 +165,9 @@ export async function updateProviderCredentials(connectionId, newCredentials) {
if (newCredentials.accessToken) updates.accessToken = newCredentials.accessToken; if (newCredentials.accessToken) updates.accessToken = newCredentials.accessToken;
if (newCredentials.refreshToken) updates.refreshToken = newCredentials.refreshToken; if (newCredentials.refreshToken) updates.refreshToken = newCredentials.refreshToken;
if (newCredentials.idToken) updates.idToken = newCredentials.idToken;
if (newCredentials.lastRefreshAt) updates.lastRefreshAt = newCredentials.lastRefreshAt;
if (newCredentials.expiresAt) updates.expiresAt = newCredentials.expiresAt;
if (newCredentials.expiresIn) { if (newCredentials.expiresIn) {
updates.expiresAt = toExpiresAt(newCredentials.expiresIn); updates.expiresAt = toExpiresAt(newCredentials.expiresIn);
updates.expiresIn = newCredentials.expiresIn; updates.expiresIn = newCredentials.expiresIn;
@@ -174,6 +184,13 @@ export async function updateProviderCredentials(connectionId, newCredentials) {
...newCredentials.providerSpecificData, ...newCredentials.providerSpecificData,
}; };
} }
if (newCredentials.copilotToken || newCredentials.copilotTokenExpiresAt) {
updates.providerSpecificData = {
...(updates.providerSpecificData || newCredentials.existingProviderSpecificData || {}),
...(newCredentials.copilotToken ? { copilotToken: newCredentials.copilotToken } : {}),
...(newCredentials.copilotTokenExpiresAt ? { copilotTokenExpiresAt: newCredentials.copilotTokenExpiresAt } : {}),
};
}
if (newCredentials.projectId) updates.projectId = newCredentials.projectId; if (newCredentials.projectId) updates.projectId = newCredentials.projectId;
const result = await updateProviderConnection(connectionId, updates); const result = await updateProviderConnection(connectionId, updates);
@@ -205,44 +222,41 @@ export async function checkAndRefreshToken(provider, credentials) {
let creds = { ...credentials }; let creds = { ...credentials };
// ── 1. Regular access-token expiry ──────────────────────────────────────── // ── 1. Regular access-token expiry ────────────────────────────────────────
if (creds.expiresAt) { if (_shouldRefreshCredentials(provider, creds)) {
const expiresAt = new Date(creds.expiresAt).getTime(); const expiresAt = creds.expiresAt ? new Date(creds.expiresAt).getTime() : null;
const now = Date.now(); const remaining = expiresAt ? expiresAt - Date.now() : null;
const remaining = expiresAt - now;
const refreshLead = _getRefreshLeadMs(provider); const refreshLead = _getRefreshLeadMs(provider);
if (remaining < refreshLead) {
log.info("TOKEN_REFRESH", "Token expiring soon, refreshing proactively", {
provider,
expiresIn: Math.round(remaining / 1000),
refreshLeadMs: refreshLead,
});
const newCreds = await getAccessToken(provider, creds); log.info("TOKEN_REFRESH", "Refreshing provider credentials proactively", {
if (newCreds?.accessToken) { provider,
const mergedCreds = { expiresIn: remaining === null ? null : Math.round(remaining / 1000),
...newCreds, refreshLeadMs: refreshLead,
existingProviderSpecificData: creds.providerSpecificData, lastRefreshAt: creds.lastRefreshAt || null,
}; });
// Persist to DB (non-blocking path continues below) const newCreds = await _refreshProviderCredentials(provider, creds, log);
await updateProviderCredentials(creds.connectionId, mergedCreds); if (newCreds?.accessToken || newCreds?.apiKey || newCreds?.copilotToken) {
const mergedCreds = {
...newCreds,
existingProviderSpecificData: creds.providerSpecificData,
};
creds = { // Persist to DB (non-blocking path continues below)
...creds, await updateProviderCredentials(creds.connectionId, mergedCreds);
accessToken: newCreds.accessToken,
refreshToken: newCreds.refreshToken ?? creds.refreshToken,
providerSpecificData: newCreds.providerSpecificData
? { ...creds.providerSpecificData, ...newCreds.providerSpecificData }
: creds.providerSpecificData,
expiresAt: newCreds.expiresIn
? toExpiresAt(newCreds.expiresIn)
: normalizeExpiresAt(newCreds.expiresAt) || creds.expiresAt,
};
// Non-blocking: refresh projectId with the new access token creds = {
_refreshProjectId(provider, creds.connectionId, creds.accessToken); ...creds,
} ...newCreds,
expiresAt: newCreds.expiresIn
? toExpiresAt(newCreds.expiresIn)
: normalizeExpiresAt(newCreds.expiresAt) || newCreds.expiresAt || creds.expiresAt,
providerSpecificData: newCreds.providerSpecificData
? { ...creds.providerSpecificData, ...newCreds.providerSpecificData }
: creds.providerSpecificData,
};
// Non-blocking: refresh projectId with the new access token
_refreshProjectId(provider, creds.connectionId, creds.accessToken);
} }
} }
+99 -10
View File
@@ -14,21 +14,30 @@ const originalFetch = global.fetch;
describe("Codex Refresh Token", () => { describe("Codex Refresh Token", () => {
beforeEach(() => { beforeEach(() => {
vi.clearAllMocks(); vi.clearAllMocks();
vi.resetModules();
global.fetch = originalFetch;
}); });
afterEach(() => { afterEach(() => {
global.fetch = originalFetch; global.fetch = originalFetch;
}); });
function mockFetchWithJson(payload) {
const fetchMock = vi.fn().mockResolvedValue({
ok: true,
json: () => Promise.resolve(payload),
});
global.fetch = fetchMock;
return fetchMock;
}
describe("refreshCodexToken", () => { describe("refreshCodexToken", () => {
it("should return new refresh_token when server provides one (token rotation)", async () => { it("should return new refresh_token when server provides one (token rotation)", async () => {
global.fetch = vi.fn().mockResolvedValue({ const fetchMock = mockFetchWithJson({
ok: true,
json: () => Promise.resolve({
access_token: "new-access", access_token: "new-access",
refresh_token: "rotated-refresh-token", refresh_token: "rotated-refresh-token",
id_token: "new-id-token",
expires_in: 3600, expires_in: 3600,
}),
}); });
const { refreshCodexToken } = await import("../../open-sse/services/tokenRefresh.js"); const { refreshCodexToken } = await import("../../open-sse/services/tokenRefresh.js");
@@ -36,21 +45,101 @@ describe("Codex Refresh Token", () => {
expect(result.refreshToken).toBe("rotated-refresh-token"); expect(result.refreshToken).toBe("rotated-refresh-token");
expect(result.accessToken).toBe("new-access"); expect(result.accessToken).toBe("new-access");
expect(result.idToken).toBe("new-id-token");
expect(fetchMock).toHaveBeenCalledWith(
"https://auth.openai.com/oauth/token",
expect.objectContaining({
method: "POST",
headers: expect.objectContaining({
"Content-Type": "application/json",
Accept: "application/json",
}),
body: JSON.stringify({
client_id: "app_EMoamEEZ73f0CkXaXp7hrann",
grant_type: "refresh_token",
refresh_token: "old-refresh-token",
}),
})
);
}); });
it("should keep old refresh_token when server does not return new one", async () => { it("should keep old refresh_token when server does not return new one", async () => {
global.fetch = vi.fn().mockResolvedValue({ mockFetchWithJson({
ok: true,
json: () => Promise.resolve({
access_token: "new-access", access_token: "new-access",
expires_in: 3600, expires_in: 3600,
}),
}); });
const { refreshCodexToken } = await import("../../open-sse/services/tokenRefresh.js"); const { refreshCodexToken } = await import("../../open-sse/services/tokenRefresh.js");
const result = await refreshCodexToken("old-refresh-token", null); const result = await refreshCodexToken("old-refresh-token-without-rotation", null);
expect(result.refreshToken).toBe("old-refresh-token"); expect(result.refreshToken).toBe("old-refresh-token-without-rotation");
});
});
describe("CodexExecutor credential lifecycle", () => {
it("should refresh Codex credentials and preserve omitted id_token", async () => {
mockFetchWithJson({
access_token: "new-access",
refresh_token: "rotated-refresh-token",
expires_in: 3600,
});
const { CodexExecutor } = await import("../../open-sse/executors/codex.js");
const executor = new CodexExecutor();
const result = await executor.refreshCredentials({
connectionId: "codex-1",
refreshToken: "old-refresh-token",
idToken: "old-id-token",
}, null);
expect(result.accessToken).toBe("new-access");
expect(result.refreshToken).toBe("rotated-refresh-token");
expect(result.idToken).toBe("old-id-token");
expect(result.lastRefreshAt).toBeTruthy();
expect(result.expiresAt).toBeTruthy();
});
it("should refresh Codex when lastRefreshAt is older than the upstream stale window", async () => {
const { CodexExecutor } = await import("../../open-sse/executors/codex.js");
const executor = new CodexExecutor();
const farFuture = new Date(Date.now() + 30 * 24 * 60 * 60 * 1000).toISOString();
const staleRefresh = new Date(Date.now() - 9 * 24 * 60 * 60 * 1000).toISOString();
const recentRefresh = new Date(Date.now() - 1 * 24 * 60 * 60 * 1000).toISOString();
expect(executor.needsRefresh({
refreshToken: "refresh-token",
expiresAt: farFuture,
lastRefreshAt: staleRefresh,
})).toBe(true);
expect(executor.needsRefresh({
refreshToken: "refresh-token",
expiresAt: farFuture,
lastRefreshAt: recentRefresh,
})).toBe(false);
});
it("should de-duplicate concurrent refreshes for the same Codex connection", async () => {
const fetchMock = mockFetchWithJson({
access_token: "new-access",
refresh_token: "rotated-refresh-token",
expires_in: 3600,
});
const { refreshProviderCredentials } = await import("../../open-sse/services/oauthCredentialManager.js");
const credentials = {
connectionId: "codex-single-flight",
refreshToken: "old-refresh-token",
};
const [first, second] = await Promise.all([
refreshProviderCredentials("codex", credentials, null),
refreshProviderCredentials("codex", credentials, null),
]);
expect(first.accessToken).toBe("new-access");
expect(second.accessToken).toBe("new-access");
expect(fetchMock).toHaveBeenCalledTimes(1);
}); });
}); });