refactor(auth): share oauth result builders and token expiry checks

This commit is contained in:
Peter Steinberger
2026-02-18 18:54:40 +00:00
parent 8d4ffe350e
commit 79cc4aec80
2 changed files with 77 additions and 14 deletions

View File

@@ -66,6 +66,24 @@ function buildApiKeyProfileResult(params: { apiKey: string; provider: string; em
};
}
function buildOAuthProfileResult(params: {
provider: string;
credentials: OAuthCredentials;
email?: string;
}) {
return buildApiKeyProfileResult({
apiKey: buildOAuthApiKey(params.provider, params.credentials),
provider: params.provider,
email: params.email,
});
}
function isExpiredCredential(expires: number | undefined): boolean {
return (
typeof expires === "number" && Number.isFinite(expires) && expires > 0 && Date.now() >= expires
);
}
async function refreshOAuthTokenWithLock(params: {
profileId: string;
agentDir?: string;
@@ -148,9 +166,9 @@ async function tryResolveOAuthProfile(params: {
}
if (Date.now() < cred.expires) {
return buildApiKeyProfileResult({
apiKey: buildOAuthApiKey(cred.provider, cred),
return buildOAuthProfileResult({
provider: cred.provider,
credentials: cred,
email: cred.email,
});
}
@@ -205,20 +223,15 @@ export async function resolveApiKeyForProfile(params: {
if (!token) {
return null;
}
if (
typeof cred.expires === "number" &&
Number.isFinite(cred.expires) &&
cred.expires > 0 &&
Date.now() >= cred.expires
) {
if (isExpiredCredential(cred.expires)) {
return null;
}
return buildApiKeyProfileResult({ apiKey: token, provider: cred.provider, email: cred.email });
}
if (Date.now() < cred.expires) {
return buildApiKeyProfileResult({
apiKey: buildOAuthApiKey(cred.provider, cred),
return buildOAuthProfileResult({
provider: cred.provider,
credentials: cred,
email: cred.email,
});
}
@@ -240,9 +253,9 @@ export async function resolveApiKeyForProfile(params: {
const refreshedStore = ensureAuthProfileStore(params.agentDir);
const refreshed = refreshedStore.profiles[profileId];
if (refreshed?.type === "oauth" && Date.now() < refreshed.expires) {
return buildApiKeyProfileResult({
apiKey: buildOAuthApiKey(refreshed.provider, refreshed),
return buildOAuthProfileResult({
provider: refreshed.provider,
credentials: refreshed,
email: refreshed.email ?? cred.email,
});
}
@@ -282,9 +295,9 @@ export async function resolveApiKeyForProfile(params: {
agentDir: params.agentDir,
expires: new Date(mainCred.expires).toISOString(),
});
return buildApiKeyProfileResult({
apiKey: buildOAuthApiKey(mainCred.provider, mainCred),
return buildOAuthProfileResult({
provider: mainCred.provider,
credentials: mainCred,
email: mainCred.email,
});
}