93 lines
2.9 KiB
TypeScript
93 lines
2.9 KiB
TypeScript
|
export type ProviderId = "xai" | "openai" | "anthropic" | "google" | "opencode-go";
|
|||
|
|
|
|||
|
|
export type ProviderApiKeys = Partial<Record<ProviderId, string>>;
|
|||
|
|
|
|||
|
|
const ENV_MAP: Record<ProviderId, string> = {
|
|||
|
|
xai: "XAI_API_KEY",
|
|||
|
|
openai: "OPENAI_API_KEY",
|
|||
|
|
anthropic: "ANTHROPIC_API_KEY",
|
|||
|
|
google: "GOOGLE_GENERATIVE_AI_API_KEY",
|
|||
|
|
"opencode-go": "OPENCODE_GO_API_KEY",
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
export const PROVIDER_KEY_LABELS: Record<
|
|||
|
|
ProviderId,
|
|||
|
|
{ label: string; hint: string; docsUrl?: string }
|
|||
|
|
> = {
|
|||
|
|
"opencode-go": {
|
|||
|
|
label: "OpenCode Go",
|
|||
|
|
hint: "從 opencode.ai/auth 訂閱 Go 方案後取得",
|
|||
|
|
docsUrl: "https://opencode.ai/docs/go/",
|
|||
|
|
},
|
|||
|
|
xai: { label: "Grok (xAI)", hint: "xAI Console API key" },
|
|||
|
|
openai: { label: "OpenAI", hint: "platform.openai.com API key" },
|
|||
|
|
anthropic: { label: "Anthropic", hint: "console.anthropic.com API key" },
|
|||
|
|
google: { label: "Google Gemini", hint: "Google AI Studio API key" },
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
export function parseProviderApiKeys(raw: string | null | undefined): ProviderApiKeys {
|
|||
|
|
if (!raw) return {};
|
|||
|
|
try {
|
|||
|
|
return JSON.parse(raw) as ProviderApiKeys;
|
|||
|
|
} catch {
|
|||
|
|
return {};
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export function serializeProviderApiKeys(keys: ProviderApiKeys): string {
|
|||
|
|
return JSON.stringify(keys);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export function maskApiKey(key: string | undefined | null): string | null {
|
|||
|
|
if (!key) return null;
|
|||
|
|
if (key.length <= 4) return "••••";
|
|||
|
|
return `••••${key.slice(-4)}`;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export function isMaskedKey(value: string): boolean {
|
|||
|
|
return value.startsWith("••••");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/** 只讀使用者自己存在 DB 的 key，不 fallback .env，避免新帳號看到舊設定。 */
|
|||
|
|
export function resolveApiKey(provider: ProviderId, keys: ProviderApiKeys): string | undefined {
|
|||
|
|
return keys[provider]?.trim() || undefined;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export function getApiKeyStatus(keys: ProviderApiKeys): Record<ProviderId, boolean> {
|
|||
|
|
return (Object.keys(ENV_MAP) as ProviderId[]).reduce(
|
|||
|
|
(acc, id) => {
|
|||
|
|
acc[id] = !!keys[id]?.trim();
|
|||
|
|
return acc;
|
|||
|
|
},
|
|||
|
|
{} as Record<ProviderId, boolean>
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export function mergeProviderApiKeys(
|
|||
|
|
existing: ProviderApiKeys,
|
|||
|
|
incoming: ProviderApiKeys
|
|||
|
|
): ProviderApiKeys {
|
|||
|
|
const merged = { ...existing };
|
|||
|
|
for (const [provider, value] of Object.entries(incoming) as [ProviderId, string][]) {
|
|||
|
|
const trimmed = value?.trim();
|
|||
|
|
if (!trimmed || isMaskedKey(trimmed)) continue;
|
|||
|
|
merged[provider] = trimmed;
|
|||
|
|
}
|
|||
|
|
return merged;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export function getMaskedProviderApiKeys(keys: ProviderApiKeys): ProviderApiKeys {
|
|||
|
|
return (Object.keys(ENV_MAP) as ProviderId[]).reduce((acc, id) => {
|
|||
|
|
const masked = maskApiKey(keys[id]);
|
|||
|
|
if (masked) acc[id] = masked;
|
|||
|
|
return acc;
|
|||
|
|
}, {} as ProviderApiKeys);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export function assertProviderApiKey(provider: ProviderId, keys: ProviderApiKeys): void {
|
|||
|
|
const key = resolveApiKey(provider, keys);
|
|||
|
|
if (!key) {
|
|||
|
|
const label = PROVIDER_KEY_LABELS[provider].label;
|
|||
|
|
throw new Error(`尚未設定 ${label} API key，請到 AI 設定頁填入`);
|
|||
|
|
}
|
|||
|
|
}
|