1

server-rs/crates/api-server/src/config.rs

@@ -1,4 +1,4 @@
-use std::{env, net::SocketAddr};
+use std::{env, fs, net::SocketAddr, path::PathBuf};
 
 use platform_llm::{
     DEFAULT_ARK_BASE_URL, DEFAULT_MAX_RETRIES, DEFAULT_REQUEST_TIMEOUT_MS,
@@ -7,6 +7,7 @@ use platform_llm::{
 
 const DEFAULT_LLM_MODEL: &str = "doubao-1-5-pro-32k-character-250715";
 const DEFAULT_INTERNAL_API_SECRET: &str = "genarrative-dev-internal-bridge";
+const SPACETIME_LOCAL_CONFIG_FILE: &str = "spacetime.local.json";
 
 // 集中管理 api-server 的启动配置，避免入口层直接散落环境变量解析逻辑。
 #[derive(Clone, Debug)]
@@ -123,6 +124,10 @@ impl AppConfig {
     pub fn from_env() -> Self {
         let mut config = Self::default();
 
+        if let Some(local_spacetime_database) = read_local_spacetime_database() {
+            config.spacetime_database = local_spacetime_database;
+        }
+
         if let Ok(bind_host) = env::var("GENARRATIVE_API_HOST")
             && !bind_host.trim().is_empty()
         {
@@ -141,8 +146,7 @@ impl AppConfig {
             config.log_filter = log_filter;
         }
 
-        config.internal_api_secret =
-            read_first_non_empty_env(&["GENARRATIVE_INTERNAL_API_SECRET"]);
+        config.internal_api_secret = read_first_non_empty_env(&["GENARRATIVE_INTERNAL_API_SECRET"]);
 
         if let Some(jwt_issuer) =
             read_first_non_empty_env(&["GENARRATIVE_JWT_ISSUER", "JWT_ISSUER"])
@@ -361,6 +365,33 @@ fn read_first_non_empty_env(keys: &[&str]) -> Option<String> {
     })
 }
 
+fn read_local_spacetime_database() -> Option<String> {
+    let config_path = find_upward_file(SPACETIME_LOCAL_CONFIG_FILE)?;
+    let raw_text = fs::read_to_string(config_path).ok()?;
+    let parsed = serde_json::from_str::<serde_json::Value>(&raw_text).ok()?;
+    parsed
+        .get("database")
+        .and_then(|value| value.as_str())
+        .map(str::trim)
+        .filter(|value| !value.is_empty())
+        .map(ToOwned::to_owned)
+}
+
+fn find_upward_file(file_name: &str) -> Option<PathBuf> {
+    let mut current_dir = env::current_dir().ok()?;
+
+    loop {
+        let candidate = current_dir.join(file_name);
+        if candidate.is_file() {
+            return Some(candidate);
+        }
+
+        if !current_dir.pop() {
+            return None;
+        }
+    }
+}
+
 fn read_first_duration_seconds_env(keys: &[&str]) -> Option<u64> {
     keys.iter().find_map(|key| {
         env::var(key)