fix(gemini): Fix configuration loss issues and simplify the authentication architecture #3

Author: SaladDay

src-tauri/Cargo.lock

@@ -372,14 +372,11 @@ fn prompt_gemini_config(current: Option<&Value>) -> Result<Value, AppError> {
     let current_auth_type = detect_gemini_auth_type(current);
     let default_index = match current_auth_type.as_deref() {
         Some("oauth") => 0,
-        Some("packycode") => 1,
-        Some("generic") => 2,
-        _ => 1, // 默认 PackyCode
+        _ => 1, // 默认 Generic API Key（包括 packycode 和 generic）
     };
 
     let auth_options = vec![
         texts::google_oauth_official(),
-        texts::packycode_api_key(),
         texts::generic_api_key(),
     ];
 
@@ -391,62 +388,15 @@ fn prompt_gemini_config(current: Option<&Value>) -> Result<Value, AppError> {
 
     // Match using the translated strings
     let google_oauth = texts::google_oauth_official();
-    let packycode = texts::packycode_api_key();
 
     if auth_type == google_oauth {
         println!("{}", texts::use_google_oauth_warning().yellow());
         Ok(json!({
             "env": {},
             "config": {}
         }))
-    } else if auth_type == packycode {
-        let api_key = if let Some(current_key) = current
-            .and_then(|v| v.get("env"))
-            .and_then(|e| e.get("GEMINI_API_KEY"))
-            .and_then(|k| k.as_str())
-            .filter(|s| !s.is_empty())
-        {
-            Text::new(texts::gemini_api_key_label())
-                .with_initial_value(current_key)
-                .with_help_message(texts::packycode_api_key_help())
-                .prompt()
-                .map_err(|e| AppError::Message(texts::input_failed_error(&e.to_string())))?
-        } else {
-            Text::new(texts::gemini_api_key_label())
-                .with_placeholder("pk-...")
-                .with_help_message(texts::packycode_api_key_help())
-                .prompt()
-                .map_err(|e| AppError::Message(texts::input_failed_error(&e.to_string())))?
-        };
-
-        let base_url = if let Some(current_url) = current
-            .and_then(|v| v.get("env"))
-            .and_then(|e| e.get("GOOGLE_GEMINI_BASE_URL"))
-            .and_then(|u| u.as_str())
-            .filter(|s| !s.is_empty())
-        {
-            Text::new(texts::gemini_base_url_label())
-                .with_initial_value(current_url)
-                .with_help_message(texts::packycode_endpoint_help())
-                .prompt()
-                .map_err(|e| AppError::Message(texts::input_failed_error(&e.to_string())))?
-        } else {
-            Text::new(texts::gemini_base_url_label())
-                .with_placeholder("https://packycode.com/api")
-                .with_help_message(texts::packycode_endpoint_help())
-                .prompt()
-                .map_err(|e| AppError::Message(texts::input_failed_error(&e.to_string())))?
-        };
-
-        Ok(json!({
-            "env": {
-                "GEMINI_API_KEY": api_key.trim(),
-                "GOOGLE_GEMINI_BASE_URL": base_url.trim()
-            },
-            "config": {}
-        }))
     } else {
-        // Generic API Key
+        // Generic API Key (统一处理所有 API Key 供应商，包括 PackyCode)
         let api_key = if let Some(current_key) = current
             .and_then(|v| v.get("env"))
             .and_then(|e| e.get("GEMINI_API_KEY"))
@@ -460,15 +410,15 @@ fn prompt_gemini_config(current: Option<&Value>) -> Result<Value, AppError> {
                 .map_err(|e| AppError::Message(texts::input_failed_error(&e.to_string())))?
         } else {
             Text::new(texts::gemini_api_key_label())
-                .with_placeholder("AIza...")
+                .with_placeholder("AIza... or pk-...")
                 .with_help_message(texts::generic_api_key_help())
                 .prompt()
                 .map_err(|e| AppError::Message(texts::input_failed_error(&e.to_string())))?
         };
 
         let base_url = if let Some(current_url) = current
             .and_then(|v| v.get("env"))
-            .and_then(|e| e.get("BASE_URL"))
+            .and_then(|e| e.get("GOOGLE_GEMINI_BASE_URL"))
             .and_then(|u| u.as_str())
             .filter(|s| !s.is_empty())
         {
@@ -488,7 +438,7 @@ fn prompt_gemini_config(current: Option<&Value>) -> Result<Value, AppError> {
         Ok(json!({
             "env": {
                 "GEMINI_API_KEY": api_key.trim(),
-                "BASE_URL": base_url.trim()
+                "GOOGLE_GEMINI_BASE_URL": base_url.trim()
             },
             "config": {}
         }))

src-tauri/src/cli/commands/provider_input.rs

@@ -336,40 +336,56 @@ fn update_selected_type(selected_type: &str) -> Result<(), AppError> {
     Ok(())
 }
 
-/// 为 Packycode Gemini 供应商写入 settings.json
+/// 为 Google 官方 Gemini 供应商写入 settings.json（OAuth 模式）
 ///
 /// 设置 `~/.gemini/settings.json` 中的：
 /// ```json
 /// {
 ///   "security": {
 ///     "auth": {
-///       "selectedType": "gemini-api-key"
+///       "selectedType": "oauth-personal"
 ///     }
 ///   }
 /// }
 /// ```
 ///
 /// 保留文件中的其他所有字段。
-pub fn write_packycode_settings() -> Result<(), AppError> {
-    update_selected_type("gemini-api-key")
+pub fn write_google_oauth_settings() -> Result<(), AppError> {
+    update_selected_type("oauth-personal")
 }
 
-/// 为 Google 官方 Gemini 供应商写入 settings.json（OAuth 模式）
+/// 为通用 Gemini API Key 供应商写入 settings.json
 ///
 /// 设置 `~/.gemini/settings.json` 中的：
 /// ```json
 /// {
 ///   "security": {
 ///     "auth": {
-///       "selectedType": "oauth-personal"
+///       "selectedType": "gemini-api-key"
 ///     }
 ///   }
 /// }
 /// ```
 ///
 /// 保留文件中的其他所有字段。
-pub fn write_google_oauth_settings() -> Result<(), AppError> {
-    update_selected_type("oauth-personal")
+///
+/// 此函数适用于所有使用 API Key 认证的 Gemini 供应商，包括：
+/// - PackyCode（合作伙伴）
+/// - 其他第三方 Gemini API 服务
+pub fn write_generic_settings() -> Result<(), AppError> {
+    update_selected_type("gemini-api-key")
+}
+
+/// 为 Packycode Gemini 供应商写入 settings.json（已废弃，使用 write_generic_settings）
+///
+/// **注意**：此函数已废弃，仅为保持向后兼容性而保留。
+/// PackyCode 现在被视为普通的 API Key 供应商，请使用 `write_generic_settings()` 代替。
+#[deprecated(
+    since = "4.1.1",
+    note = "PackyCode is now treated as a generic API key provider. Use write_generic_settings() instead."
+)]
+pub fn write_packycode_settings() -> Result<(), AppError> {
+    write_generic_settings()
 }
 
 #[cfg(test)]