Merge remote-tracking branch 'entropy-xu/codex/ccswitch-import'

2026-10-07 18:07:47 +08:00 · 2026-05-31 20:52:26 +08:00
parent 905453d62b 7ed4b84654
commit 392353ffff
23 changed files with 1720 additions and 48 deletions
@@ -0,0 +1,163 @@
+import { describe, expect, it } from 'vitest'
+
+import {
+  base64UrlEncodeUtf8,
+  buildCcSwitchProviderImportUrl,
+  type CcSwitchTargetApp,
+} from '@/features/api-keys/utils/ccswitchImport'
+
+function decodeBase64UrlJson(value: string): unknown {
+  const padded = `${value}${'='.repeat((4 - (value.length % 4)) % 4)}`
+    .replace(/-/g, '+')
+    .replace(/_/g, '/')
+  const binary = atob(padded)
+  const bytes = Uint8Array.from(binary, char => char.charCodeAt(0))
+  return JSON.parse(new TextDecoder().decode(bytes))
+}
+
+function parseImportUrl(url: string) {
+  const parsed = new URL(url)
+  const params = parsed.searchParams
+  const config = params.get('config')
+  if (!config) throw new Error('missing config')
+  return {
+    parsed,
+    params,
+    config: decodeBase64UrlJson(config) as Record<string, unknown>,
+  }
+}
+
+const baseInput = {
+  baseUrl: 'https://aether.example.com',
+  apiKey: 'sk-user-live-1',
+  apiKeyName: 'primary',
+  siteName: 'Aether Local',
+  modelId: 'gpt-5',
+}
+
+describe('ccswitchImport', () => {
+  it('encodes UTF-8 text as base64url without padding', () => {
+    const encoded = base64UrlEncodeUtf8('Aether 中文+/=')
+
+    expect(encoded).not.toContain('+')
+    expect(encoded).not.toContain('/')
+    expect(encoded).not.toContain('=')
+    expect(new TextDecoder().decode(Uint8Array.from(atob(`${encoded}${'='.repeat((4 - (encoded.length % 4)) % 4)}`.replace(/-/g, '+').replace(/_/g, '/')), char => char.charCodeAt(0)))).toBe('Aether 中文+/=')
+  })
+
+  it('builds a Claude Code import URL with separate Claude model env fields', () => {
+    const { params, config } = parseImportUrl(buildCcSwitchProviderImportUrl({
+      ...baseInput,
+      targetApp: 'claude',
+      modelIds: {
+        haiku: 'gpt-5.4-mini',
+        sonnet: 'gpt-5',
+        opus: 'gpt-5.1-pro',
+      },
+    }))
+
+    expect(params.get('resource')).toBe('provider')
+    expect(params.get('app')).toBe('claude')
+    expect(params.get('name')).toBe('Aether Local')
+    expect(params.get('icon')).toBe('claude')
+    expect(params.get('enabled')).toBe('true')
+    expect(params.get('configFormat')).toBe('json')
+    expect(params.get('usageEnabled')).toBe('true')
+    expect(params.get('usageBaseUrl')).toBe('https://aether.example.com')
+    expect(params.get('usageApiKey')).toBe('sk-user-live-1')
+    expect(params.get('usageAutoInterval')).toBe('30')
+    expect(params.get('endpoint')).toBe('https://aether.example.com')
+    expect(params.get('apiKey')).toBe('sk-user-live-1')
+    expect(params.get('model')).toBe('gpt-5')
+    expect(params.get('haikuModel')).toBe('gpt-5.4-mini')
+    expect(params.get('sonnetModel')).toBe('gpt-5')
+    expect(params.get('opusModel')).toBe('gpt-5.1-pro')
+    const usageScript = params.get('usageScript') || ''
+    expect(usageScript).not.toBe('')
+    expect(usageScript).not.toContain('(')
+    expect(usageScript).not.toContain('+')
+    expect(usageScript).not.toContain('/')
+    const paddedUsageScript = `${usageScript}${'='.repeat((4 - (usageScript.length % 4)) % 4)}`
+      .replace(/-/g, '+')
+      .replace(/_/g, '/')
+    const decodedUsageScript = new TextDecoder().decode(
+      Uint8Array.from(atob(paddedUsageScript), char => char.charCodeAt(0)),
+    )
+    expect(decodedUsageScript).toContain('/api/ccswitch/usage')
+    expect(decodedUsageScript).toContain('Authorization')
+    expect(config).toEqual({
+      env: {
+        ANTHROPIC_AUTH_TOKEN: 'sk-user-live-1',
+        ANTHROPIC_BASE_URL: 'https://aether.example.com',
+        ANTHROPIC_MODEL: 'gpt-5',
+        ANTHROPIC_DEFAULT_HAIKU_MODEL: 'gpt-5.4-mini',
+        ANTHROPIC_DEFAULT_SONNET_MODEL: 'gpt-5',
+        ANTHROPIC_DEFAULT_OPUS_MODEL: 'gpt-5.1-pro',
+      },
+    })
+  })
+
+  it('builds a Codex import URL with auth and TOML config', () => {
+    const { params, config } = parseImportUrl(buildCcSwitchProviderImportUrl({
+      ...baseInput,
+      targetApp: 'codex',
+    }))
+
+    expect(params.get('app')).toBe('codex')
+    expect(params.get('icon')).toBe('openai')
+    expect(params.get('endpoint')).toBe('https://aether.example.com/v1')
+    expect(params.get('apiKey')).toBe('sk-user-live-1')
+    expect(params.get('model')).toBe('gpt-5')
+    expect(config.auth).toEqual({ OPENAI_API_KEY: 'sk-user-live-1' })
+    expect(config.config).toContain('model_provider = "aether"')
+    expect(config.config).toContain('model = "gpt-5"')
+    expect(config.config).toContain('base_url = "https://aether.example.com/v1"')
+    expect(config.config).toContain('wire_api = "responses"')
+  })
+
+  it.each([
+    ['gemini', {
+      GEMINI_API_KEY: 'sk-user-live-1',
+      GOOGLE_GEMINI_BASE_URL: 'https://aether.example.com',
+      GEMINI_MODEL: 'gpt-5',
+    }],
+    ['opencode', {
+      npm: '@ai-sdk/openai-compatible',
+      options: {
+        baseURL: 'https://aether.example.com/v1',
+        apiKey: 'sk-user-live-1',
+      },
+      models: {
+        'gpt-5': { name: 'gpt-5' },
+      },
+    }],
+    ['openclaw', {
+      baseUrl: 'https://aether.example.com/v1',
+      apiKey: 'sk-user-live-1',
+      api: 'openai-completions',
+      models: [{ id: 'gpt-5', name: 'gpt-5' }],
+    }],
+    ['hermes', {
+      name: 'Aether Local',
+      base_url: 'https://aether.example.com/v1',
+      api_key: 'sk-user-live-1',
+      api_mode: 'chat_completions',
+      models: [{ id: 'gpt-5', name: 'gpt-5' }],
+    }],
+  ] satisfies Array<[CcSwitchTargetApp, Record<string, unknown>]>)('builds %s import config', (targetApp, expectedConfig) => {
+    const { params, config } = parseImportUrl(buildCcSwitchProviderImportUrl({
+      ...baseInput,
+      targetApp,
+    }))
+
+    expect(params.get('app')).toBe(targetApp)
+    expect(params.get('endpoint')).toBe(
+      targetApp === 'gemini'
+        ? 'https://aether.example.com'
+        : 'https://aether.example.com/v1',
+    )
+    expect(params.get('apiKey')).toBe('sk-user-live-1')
+    expect(params.get('model')).toBe('gpt-5')
+    expect(config).toEqual(expectedConfig)
+  })
+})
@@ -0,0 +1,237 @@
+export type CcSwitchTargetApp = 'claude' | 'codex' | 'gemini' | 'opencode' | 'openclaw' | 'hermes'
+
+export interface CcSwitchTargetOption {
+  value: CcSwitchTargetApp
+  label: string
+  icon: string
+}
+
+export interface BuildCcSwitchProviderImportUrlInput {
+  targetApp: CcSwitchTargetApp
+  baseUrl: string
+  apiKey: string
+  apiKeyName: string
+  siteName?: string
+  modelId?: string
+  modelIds?: CcSwitchModelIds
+  providerName?: string
+  enabled?: boolean
+}
+
+export interface CcSwitchModelIds {
+  default?: string
+  haiku?: string
+  sonnet?: string
+  opus?: string
+}
+
+interface NormalizedCcSwitchModelIds {
+  default: string
+  haiku: string
+  sonnet: string
+  opus: string
+}
+
+export const CC_SWITCH_TARGET_OPTIONS: CcSwitchTargetOption[] = [
+  { value: 'claude', label: 'Claude Code', icon: 'claude' },
+  { value: 'codex', label: 'Codex CLI', icon: 'openai' },
+  { value: 'gemini', label: 'Gemini CLI', icon: 'gemini' },
+  { value: 'opencode', label: 'OpenCode', icon: 'opencode' },
+  { value: 'openclaw', label: 'OpenClaw', icon: 'openclaw' },
+  { value: 'hermes', label: 'Hermes', icon: 'hermes' },
+]
+
+const AETHER_USAGE_QUERY_SCRIPT = [
+  '({',
+  '  request: {',
+  '    url: "{{baseUrl}}/api/ccswitch/usage",',
+  '    method: "GET",',
+  '    headers: {',
+  '      "Authorization": "Bearer {{apiKey}}",',
+  '      "Accept": "application/json",',
+  '      "User-Agent": "cc-switch/1.0"',
+  '    }',
+  '  },',
+  '  extractor: function(response) {',
+  '    if (response && response.is_valid === false) {',
+  '      return {',
+  '        isValid: false,',
+  '        invalidMessage: response.invalid_message || "查询失败"',
+  '      };',
+  '    }',
+  '',
+  '    return {',
+  '      isValid: true,',
+  '      planName: response.plan_name || "Aether",',
+  '      remaining: response.remaining,',
+  '      used: response.used,',
+  '      total: response.total,',
+  '      unit: response.unit || "USD",',
+  '      extra: response.extra',
+  '    };',
+  '  }',
+  '})',
+].join('\n')
+
+export function base64UrlEncodeUtf8(text: string): string {
+  const bytes = new TextEncoder().encode(text)
+  let binary = ''
+  for (const byte of bytes) {
+    binary += String.fromCharCode(byte)
+  }
+
+  return btoa(binary)
+    .replace(/\+/g, '-')
+    .replace(/\//g, '_')
+    .replace(/=+$/g, '')
+}
+
+function normalizedBaseUrl(baseUrl: string): string {
+  return baseUrl.trim().replace(/\/+$/g, '')
+}
+
+function aetherV1BaseUrl(baseUrl: string): string {
+  return `${normalizedBaseUrl(baseUrl)}/v1`
+}
+
+function ccSwitchEndpointForTarget(targetApp: CcSwitchTargetApp, baseUrl: string): string {
+  if (targetApp === 'claude' || targetApp === 'gemini') {
+    return normalizedBaseUrl(baseUrl)
+  }
+  return aetherV1BaseUrl(baseUrl)
+}
+
+function quoteTomlString(value: string): string {
+  return `"${value.replace(/\\/g, '\\\\').replace(/"/g, '\\"')}"`
+}
+
+export function ccSwitchTargetLabel(targetApp: CcSwitchTargetApp): string {
+  return CC_SWITCH_TARGET_OPTIONS.find(option => option.value === targetApp)?.label || targetApp
+}
+
+export function defaultCcSwitchProviderName(
+  siteName?: string,
+): string {
+  return siteName?.trim() || 'Aether'
+}
+
+function buildCodexToml(baseUrl: string, modelId: string): string {
+  return [
+    'model_provider = "aether"',
+    `model = ${quoteTomlString(modelId)}`,
+    'model_reasoning_effort = "high"',
+    'disable_response_storage = true',
+    '',
+    '[model_providers.aether]',
+    'name = "Aether"',
+    `base_url = ${quoteTomlString(aetherV1BaseUrl(baseUrl))}`,
+    'wire_api = "responses"',
+    'requires_openai_auth = true',
+    '',
+  ].join('\n')
+}
+
+function normalizeCcSwitchModelIds(input: BuildCcSwitchProviderImportUrlInput): NormalizedCcSwitchModelIds {
+  const defaultModel = input.modelIds?.default?.trim() || input.modelId?.trim() || ''
+  const sonnet = input.modelIds?.sonnet?.trim() || defaultModel
+  const haiku = input.modelIds?.haiku?.trim() || sonnet
+  const opus = input.modelIds?.opus?.trim() || sonnet
+
+  return {
+    default: defaultModel,
+    haiku,
+    sonnet,
+    opus,
+  }
+}
+
+function buildCcSwitchConfig(input: BuildCcSwitchProviderImportUrlInput): Record<string, unknown> {
+  const baseUrl = normalizedBaseUrl(input.baseUrl)
+  const modelIds = normalizeCcSwitchModelIds(input)
+  const providerName = input.providerName?.trim() || defaultCcSwitchProviderName(input.siteName)
+
+  switch (input.targetApp) {
+    case 'claude':
+      return {
+        env: {
+          ANTHROPIC_AUTH_TOKEN: input.apiKey,
+          ANTHROPIC_BASE_URL: baseUrl,
+          ANTHROPIC_MODEL: modelIds.sonnet,
+          ANTHROPIC_DEFAULT_HAIKU_MODEL: modelIds.haiku,
+          ANTHROPIC_DEFAULT_SONNET_MODEL: modelIds.sonnet,
+          ANTHROPIC_DEFAULT_OPUS_MODEL: modelIds.opus,
+        },
+      }
+    case 'codex':
+      return {
+        auth: {
+          OPENAI_API_KEY: input.apiKey,
+        },
+        config: buildCodexToml(baseUrl, modelIds.default),
+      }
+    case 'gemini':
+      return {
+        GEMINI_API_KEY: input.apiKey,
+        GOOGLE_GEMINI_BASE_URL: baseUrl,
+        GEMINI_MODEL: modelIds.default,
+      }
+    case 'opencode':
+      return {
+        npm: '@ai-sdk/openai-compatible',
+        options: {
+          baseURL: aetherV1BaseUrl(baseUrl),
+          apiKey: input.apiKey,
+        },
+        models: {
+          [modelIds.default]: {
+            name: modelIds.default,
+          },
+        },
+      }
+    case 'openclaw':
+      return {
+        baseUrl: aetherV1BaseUrl(baseUrl),
+        apiKey: input.apiKey,
+        api: 'openai-completions',
+        models: [{ id: modelIds.default, name: modelIds.default }],
+      }
+    case 'hermes':
+      return {
+        name: providerName,
+        base_url: aetherV1BaseUrl(baseUrl),
+        api_key: input.apiKey,
+        api_mode: 'chat_completions',
+        models: [{ id: modelIds.default, name: modelIds.default }],
+      }
+  }
+}
+
+export function buildCcSwitchProviderImportUrl(input: BuildCcSwitchProviderImportUrlInput): string {
+  const providerName = input.providerName?.trim() || defaultCcSwitchProviderName(input.siteName)
+  const icon = CC_SWITCH_TARGET_OPTIONS.find(option => option.value === input.targetApp)?.icon
+  const modelIds = normalizeCcSwitchModelIds(input)
+  const params = new URLSearchParams()
+
+  params.set('resource', 'provider')
+  params.set('app', input.targetApp)
+  params.set('name', providerName)
+  params.set('endpoint', ccSwitchEndpointForTarget(input.targetApp, input.baseUrl))
+  params.set('apiKey', input.apiKey)
+  params.set('model', input.targetApp === 'claude' ? modelIds.sonnet : modelIds.default)
+  if (input.targetApp === 'claude') {
+    params.set('haikuModel', modelIds.haiku)
+    params.set('sonnetModel', modelIds.sonnet)
+    params.set('opusModel', modelIds.opus)
+  }
+  params.set('configFormat', 'json')
+  params.set('config', base64UrlEncodeUtf8(JSON.stringify(buildCcSwitchConfig(input))))
+  params.set('enabled', input.enabled === false ? 'false' : 'true')
+  params.set('usageEnabled', 'true')
+  params.set('usageBaseUrl', normalizedBaseUrl(input.baseUrl))
+  params.set('usageApiKey', input.apiKey)
+  params.set('usageAutoInterval', '30')
+  params.set('usageScript', base64UrlEncodeUtf8(AETHER_USAGE_QUERY_SCRIPT))
+  if (icon) params.set('icon', icon)
+
+  return `ccswitch://v1/import?${params.toString()}`
+}