File size: 3,681 Bytes
a21c316 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 | use std::fs;
use serde_json;
use crate::models::AppConfig;
use super::account::get_data_dir;
use tracing::warn;
const CONFIG_FILE: &str = "gui_config.json";
/// Load application configuration
pub fn load_app_config() -> Result<AppConfig, String> {
let data_dir = get_data_dir()?;
let config_path = data_dir.join(CONFIG_FILE);
if !config_path.exists() {
let config = AppConfig::new();
// [FIX #1460] Persist initial config to prevent new API Key on every refresh
let _ = save_app_config(&config);
return Ok(config);
}
let content = fs::read_to_string(&config_path)
.map_err(|e| format!("failed_to_read_config_file: {}", e))?;
let mut v: serde_json::Value = serde_json::from_str(&content)
.map_err(|e| format!("failed_to_parse_config_file: {}", e))?;
let mut modified = false;
// Migration logic
if let Some(proxy) = v.get_mut("proxy") {
// [FIX #1738] Enhanced type checking for custom_mapping
// Ensures the field is always parsed as an object, preventing type mismatch errors
let mut custom_mapping = match proxy.get("custom_mapping") {
Some(m) if m.is_object() => m.as_object().unwrap().clone(),
Some(m) => {
// If custom_mapping is not an object type (e.g., string), log warning and reset to empty
tracing::warn!("Invalid custom_mapping type (expected object, got {:?}), resetting to empty", m);
serde_json::Map::new()
}
None => serde_json::Map::new(),
};
// Migrate Anthropic mapping
if let Some(anthropic) = proxy.get_mut("anthropic_mapping").and_then(|m| m.as_object_mut()) {
for (k, v) in anthropic.iter() {
// Only move non-series fields, as series fields are now handled by Preset logic or builtin tables
if !k.ends_with("-series") {
if !custom_mapping.contains_key(k) {
custom_mapping.insert(k.clone(), v.clone());
}
}
}
// Remove old field
proxy.as_object_mut().unwrap().remove("anthropic_mapping");
modified = true;
}
// Migrate OpenAI mapping
if let Some(openai) = proxy.get_mut("openai_mapping").and_then(|m| m.as_object_mut()) {
for (k, v) in openai.iter() {
if !k.ends_with("-series") {
if !custom_mapping.contains_key(k) {
custom_mapping.insert(k.clone(), v.clone());
}
}
}
// Remove old field
proxy.as_object_mut().unwrap().remove("openai_mapping");
modified = true;
}
if modified {
proxy.as_object_mut().unwrap().insert("custom_mapping".to_string(), serde_json::Value::Object(custom_mapping));
}
}
let config: AppConfig = serde_json::from_value(v)
.map_err(|e| format!("failed_to_convert_config_after_migration: {}", e))?;
// If migration occurred, auto-save once to clean up the file
if modified {
let _ = save_app_config(&config);
}
Ok(config)
}
/// Save application configuration
pub fn save_app_config(config: &AppConfig) -> Result<(), String> {
let data_dir = get_data_dir()?;
let config_path = data_dir.join(CONFIG_FILE);
let content = serde_json::to_string_pretty(config)
.map_err(|e| format!("failed_to_serialize_config: {}", e))?;
fs::write(&config_path, content)
.map_err(|e| format!("failed_to_save_config: {}", e))
}
|