mirror of
https://github.com/decolua/9router.git
synced 2026-05-08 12:01:28 +00:00
198 lines
8.7 KiB
JavaScript
198 lines
8.7 KiB
JavaScript
import { BaseExecutor } from "./base.js";
|
|
import { PROVIDERS, OAUTH_ENDPOINTS } from "../config/constants.js";
|
|
|
|
export class DefaultExecutor extends BaseExecutor {
|
|
constructor(provider) {
|
|
super(provider, PROVIDERS[provider] || PROVIDERS.openai);
|
|
}
|
|
|
|
buildUrl(model, stream, urlIndex = 0, credentials = null) {
|
|
if (this.provider?.startsWith?.("openai-compatible-")) {
|
|
const baseUrl = credentials?.providerSpecificData?.baseUrl || "https://api.openai.com/v1";
|
|
const normalized = baseUrl.replace(/\/$/, "");
|
|
const path = this.provider.includes("responses") ? "/responses" : "/chat/completions";
|
|
return `${normalized}${path}`;
|
|
}
|
|
if (this.provider?.startsWith?.("anthropic-compatible-")) {
|
|
const baseUrl = credentials?.providerSpecificData?.baseUrl || "https://api.anthropic.com/v1";
|
|
const normalized = baseUrl.replace(/\/$/, "");
|
|
return `${normalized}/messages`;
|
|
}
|
|
switch (this.provider) {
|
|
case "claude":
|
|
case "glm":
|
|
case "kimi":
|
|
case "kimi-coding":
|
|
case "cline":
|
|
case "minimax":
|
|
case "minimax-cn":
|
|
return `${this.config.baseUrl}?beta=true`;
|
|
case "gemini":
|
|
return `${this.config.baseUrl}/${model}:${stream ? "streamGenerateContent?alt=sse" : "generateContent"}`;
|
|
default:
|
|
return this.config.baseUrl;
|
|
}
|
|
}
|
|
|
|
buildHeaders(credentials, stream = true) {
|
|
const headers = { "Content-Type": "application/json", ...this.config.headers };
|
|
|
|
switch (this.provider) {
|
|
case "gemini":
|
|
credentials.apiKey ? headers["x-goog-api-key"] = credentials.apiKey : headers["Authorization"] = `Bearer ${credentials.accessToken}`;
|
|
break;
|
|
case "claude":
|
|
credentials.apiKey ? headers["x-api-key"] = credentials.apiKey : headers["Authorization"] = `Bearer ${credentials.accessToken}`;
|
|
break;
|
|
case "glm":
|
|
case "kimi":
|
|
case "kimi-coding":
|
|
case "cline":
|
|
case "minimax":
|
|
case "minimax-cn":
|
|
headers["x-api-key"] = credentials.apiKey || credentials.accessToken;
|
|
break;
|
|
default:
|
|
if (this.provider?.startsWith?.("anthropic-compatible-")) {
|
|
if (credentials.apiKey) {
|
|
headers["x-api-key"] = credentials.apiKey;
|
|
} else if (credentials.accessToken) {
|
|
headers["Authorization"] = `Bearer ${credentials.accessToken}`;
|
|
}
|
|
if (!headers["anthropic-version"]) {
|
|
headers["anthropic-version"] = "2023-06-01";
|
|
}
|
|
} else {
|
|
headers["Authorization"] = `Bearer ${credentials.apiKey || credentials.accessToken}`;
|
|
}
|
|
}
|
|
|
|
if (stream) headers["Accept"] = "text/event-stream";
|
|
return headers;
|
|
}
|
|
|
|
async refreshCredentials(credentials, log) {
|
|
if (!credentials.refreshToken) return null;
|
|
|
|
const refreshers = {
|
|
claude: () => this.refreshWithJSON(OAUTH_ENDPOINTS.anthropic.token, { grant_type: "refresh_token", refresh_token: credentials.refreshToken, client_id: PROVIDERS.claude.clientId }),
|
|
codex: () => this.refreshWithForm(OAUTH_ENDPOINTS.openai.token, { grant_type: "refresh_token", refresh_token: credentials.refreshToken, client_id: PROVIDERS.codex.clientId, scope: "openid profile email offline_access" }),
|
|
qwen: () => this.refreshWithForm(OAUTH_ENDPOINTS.qwen.token, { grant_type: "refresh_token", refresh_token: credentials.refreshToken, client_id: PROVIDERS.qwen.clientId }),
|
|
iflow: () => this.refreshIflow(credentials.refreshToken),
|
|
gemini: () => this.refreshGoogle(credentials.refreshToken),
|
|
kiro: () => this.refreshKiro(credentials.refreshToken),
|
|
cline: () => this.refreshCline(credentials.refreshToken),
|
|
"kimi-coding": () => this.refreshKimiCoding(credentials.refreshToken),
|
|
kilocode: () => this.refreshKilocode(credentials.refreshToken)
|
|
};
|
|
|
|
const refresher = refreshers[this.provider];
|
|
if (!refresher) return null;
|
|
|
|
try {
|
|
const result = await refresher();
|
|
if (result) log?.info?.("TOKEN", `${this.provider} refreshed`);
|
|
return result;
|
|
} catch (error) {
|
|
log?.error?.("TOKEN", `${this.provider} refresh error: ${error.message}`);
|
|
return null;
|
|
}
|
|
}
|
|
|
|
async refreshWithJSON(url, body) {
|
|
const response = await fetch(url, {
|
|
method: "POST",
|
|
headers: { "Content-Type": "application/json", "Accept": "application/json" },
|
|
body: JSON.stringify(body)
|
|
});
|
|
if (!response.ok) return null;
|
|
const tokens = await response.json();
|
|
return { accessToken: tokens.access_token, refreshToken: tokens.refresh_token || body.refresh_token, expiresIn: tokens.expires_in };
|
|
}
|
|
|
|
async refreshWithForm(url, params) {
|
|
const response = await fetch(url, {
|
|
method: "POST",
|
|
headers: { "Content-Type": "application/x-www-form-urlencoded", "Accept": "application/json" },
|
|
body: new URLSearchParams(params)
|
|
});
|
|
if (!response.ok) return null;
|
|
const tokens = await response.json();
|
|
return { accessToken: tokens.access_token, refreshToken: tokens.refresh_token || params.refresh_token, expiresIn: tokens.expires_in };
|
|
}
|
|
|
|
async refreshIflow(refreshToken) {
|
|
const basicAuth = btoa(`${PROVIDERS.iflow.clientId}:${PROVIDERS.iflow.clientSecret}`);
|
|
const response = await fetch(OAUTH_ENDPOINTS.iflow.token, {
|
|
method: "POST",
|
|
headers: { "Content-Type": "application/x-www-form-urlencoded", "Accept": "application/json", "Authorization": `Basic ${basicAuth}` },
|
|
body: new URLSearchParams({ grant_type: "refresh_token", refresh_token: refreshToken, client_id: PROVIDERS.iflow.clientId, client_secret: PROVIDERS.iflow.clientSecret })
|
|
});
|
|
if (!response.ok) return null;
|
|
const tokens = await response.json();
|
|
return { accessToken: tokens.access_token, refreshToken: tokens.refresh_token || refreshToken, expiresIn: tokens.expires_in };
|
|
}
|
|
|
|
async refreshGoogle(refreshToken) {
|
|
const response = await fetch(OAUTH_ENDPOINTS.google.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: this.config.clientId, client_secret: this.config.clientSecret })
|
|
});
|
|
if (!response.ok) return null;
|
|
const tokens = await response.json();
|
|
return { accessToken: tokens.access_token, refreshToken: tokens.refresh_token || refreshToken, expiresIn: tokens.expires_in };
|
|
}
|
|
|
|
async refreshKiro(refreshToken) {
|
|
const response = await fetch(PROVIDERS.kiro.tokenUrl, {
|
|
method: "POST",
|
|
headers: { "Content-Type": "application/json", "Accept": "application/json", "User-Agent": "kiro-cli/1.0.0" },
|
|
body: JSON.stringify({ refreshToken })
|
|
});
|
|
if (!response.ok) return null;
|
|
const tokens = await response.json();
|
|
return { accessToken: tokens.accessToken, refreshToken: tokens.refreshToken || refreshToken, expiresIn: tokens.expiresIn };
|
|
}
|
|
|
|
async refreshCline(refreshToken) {
|
|
console.log('[DEBUG] Refreshing Cline token, refreshToken length:', refreshToken?.length);
|
|
const response = await fetch("https://api.cline.bot/api/v1/auth/refresh", {
|
|
method: "POST",
|
|
headers: { "Content-Type": "application/json", "Accept": "application/json" },
|
|
body: JSON.stringify({ refreshToken, grantType: "refresh_token", clientType: "extension" })
|
|
});
|
|
console.log('[DEBUG] Cline refresh response status:', response.status);
|
|
if (!response.ok) {
|
|
const errorText = await response.text();
|
|
console.log('[DEBUG] Cline refresh error:', errorText);
|
|
return null;
|
|
}
|
|
const payload = await response.json();
|
|
console.log('[DEBUG] Cline refresh payload:', JSON.stringify(payload).substring(0, 200));
|
|
const data = payload?.data || payload;
|
|
const expiresAtIso = data?.expiresAt;
|
|
const expiresIn = expiresAtIso ? Math.max(1, Math.floor((new Date(expiresAtIso).getTime() - Date.now()) / 1000)) : undefined;
|
|
console.log('[DEBUG] Cline refresh success, expiresIn:', expiresIn);
|
|
return { accessToken: data?.accessToken, refreshToken: data?.refreshToken || refreshToken, expiresIn };
|
|
}
|
|
|
|
async refreshKimiCoding(refreshToken) {
|
|
const response = await fetch("https://auth.kimi.com/api/oauth/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: "17e5f671-d194-4dfb-9706-5516cb48c098" })
|
|
});
|
|
if (!response.ok) return null;
|
|
const tokens = await response.json();
|
|
return { accessToken: tokens.access_token, refreshToken: tokens.refresh_token || refreshToken, expiresIn: tokens.expires_in };
|
|
}
|
|
|
|
async refreshKilocode(refreshToken) {
|
|
// Kilocode uses device code flow, no refresh token support
|
|
return null;
|
|
}
|
|
}
|
|
|
|
export default DefaultExecutor;
|