mirror of
https://github.com/farion1231/cc-switch.git
synced 2026-07-24 12:44:18 +08:00
Jason
51d6c458ff
Allow the built-in Codex official provider to participate in takeover mode while preserving Codex's native OAuth or API-key credentials instead of persisting them into provider records. Project official routing into a dedicated TOML provider, normalize inline tables, clean stale managed placeholders, and fail closed when the live configuration cannot be transformed safely. Validate forwarded authorization, make official 401/403 responses non-retryable, avoid circuit-breaker pollution, and share the first-party ChatGPT endpoint across the Codex and Claude adapters.
96 lines
3.4 KiB
Rust
96 lines
3.4 KiB
Rust
//! 官方供应商种子数据
|
||
//!
|
||
//! 启动时调用 `Database::init_default_official_providers` 把这些条目
|
||
//! 写入 `providers` 表，让所有用户都能看到一个"一键切回官方"的入口。
|
||
//!
|
||
//! 字段与前端预设保持一致，参见：
|
||
//! - `src/config/claudeProviderPresets.ts`（"Claude Official"）
|
||
//! - `src/config/codexProviderPresets.ts`（"OpenAI Official"）
|
||
//! - `src/config/geminiProviderPresets.ts`（"Google Official"）
|
||
|
||
use crate::app_config::AppType;
|
||
|
||
pub(crate) const CLAUDE_DESKTOP_OFFICIAL_PROVIDER_ID: &str = "claude-desktop-official";
|
||
pub(crate) const CODEX_OFFICIAL_PROVIDER_ID: &str = "codex-official";
|
||
|
||
/// 单条官方供应商种子定义。
|
||
pub(crate) struct OfficialProviderSeed {
|
||
pub id: &'static str,
|
||
pub app_type: AppType,
|
||
pub name: &'static str,
|
||
pub website_url: &'static str,
|
||
pub icon: &'static str,
|
||
pub icon_color: &'static str,
|
||
/// settings_config 的 JSON 字符串，每个 app 结构不同。
|
||
pub settings_config_json: &'static str,
|
||
}
|
||
|
||
/// Claude / Claude Desktop / Codex / Gemini 的官方预设。
|
||
///
|
||
/// id 固定，便于幂等检查；name 直接用英文原名（与前端预设一致），不做 i18n。
|
||
pub(crate) const OFFICIAL_SEEDS: &[OfficialProviderSeed] = &[
|
||
OfficialProviderSeed {
|
||
id: "claude-official",
|
||
app_type: AppType::Claude,
|
||
name: "Claude Official",
|
||
website_url: "https://www.anthropic.com/claude-code",
|
||
icon: "anthropic",
|
||
icon_color: "#D4915D",
|
||
// 空 env 让用户走 Claude CLI 默认认证流程
|
||
settings_config_json: r#"{"env":{}}"#,
|
||
},
|
||
OfficialProviderSeed {
|
||
id: CLAUDE_DESKTOP_OFFICIAL_PROVIDER_ID,
|
||
app_type: AppType::ClaudeDesktop,
|
||
name: "Claude Desktop Official",
|
||
website_url: "https://claude.ai/download",
|
||
icon: "anthropic",
|
||
icon_color: "#D4915D",
|
||
// 空 env 只是占位；切换该 provider 时会恢复 Claude Desktop 1P 模式
|
||
settings_config_json: r#"{"env":{}}"#,
|
||
},
|
||
OfficialProviderSeed {
|
||
id: CODEX_OFFICIAL_PROVIDER_ID,
|
||
app_type: AppType::Codex,
|
||
name: "OpenAI Official",
|
||
website_url: "https://chatgpt.com/codex",
|
||
icon: "openai",
|
||
icon_color: "#00A67E",
|
||
// 空 auth + 空 config 让用户走 ChatGPT Plus/Pro OAuth
|
||
settings_config_json: r#"{"auth":{},"config":""}"#,
|
||
},
|
||
OfficialProviderSeed {
|
||
id: "gemini-official",
|
||
app_type: AppType::Gemini,
|
||
name: "Google Official",
|
||
website_url: "https://ai.google.dev/",
|
||
icon: "gemini",
|
||
icon_color: "#4285F4",
|
||
// 空 env + 空 config 让用户走 Google OAuth
|
||
settings_config_json: r#"{"env":{},"config":{}}"#,
|
||
},
|
||
];
|
||
|
||
/// 判断给定的 provider id 是否属于内置官方种子。
|
||
///
|
||
/// 单一事实源：直接扫描 `OFFICIAL_SEEDS`，避免在多处重复维护 id 列表。
|
||
pub(crate) fn is_official_seed_id(id: &str) -> bool {
|
||
OFFICIAL_SEEDS.iter().any(|seed| seed.id == id)
|
||
}
|
||
|
||
#[cfg(test)]
|
||
mod tests {
|
||
use super::*;
|
||
|
||
#[test]
|
||
fn official_seeds_include_claude_desktop() {
|
||
let seed = OFFICIAL_SEEDS
|
||
.iter()
|
||
.find(|seed| seed.id == CLAUDE_DESKTOP_OFFICIAL_PROVIDER_ID)
|
||
.expect("claude desktop official seed");
|
||
|
||
assert_eq!(seed.app_type, AppType::ClaudeDesktop);
|
||
assert!(is_official_seed_id(CLAUDE_DESKTOP_OFFICIAL_PROVIDER_ID));
|
||
}
|
||
}
|