提交qiming-mcp-proxy

qiming-mcp-proxy/mcp-proxy/src/config.rs

@@ -0,0 +1,106 @@
+use anyhow::Result;
+use serde::Deserialize;
+use std::env;
+use std::fs::File;
+use std::path::Path;
+
+/// The default config file
+const DEFAULT_CONFIG_YAML: &str = include_str!("../config.yml");
+
+/// config.yml 中 mirror 段的结构
+#[allow(dead_code)]
+#[derive(Debug, Deserialize, Clone, Default)]
+pub struct MirrorYamlConfig {
+    #[serde(default)]
+    pub npm_registry: String,
+    #[serde(default)]
+    pub pypi_index_url: String,
+}
+
+#[allow(dead_code)]
+#[derive(Debug, Deserialize, Clone)]
+pub struct AppConfig {
+    pub server: ServerConfig,
+    pub log: LogConfig,
+    #[serde(default)]
+    pub mirror: MirrorYamlConfig,
+}
+#[allow(dead_code)]
+#[derive(Debug, Deserialize, Clone)]
+pub struct ServerConfig {
+    /// The port to listen on for incoming connections
+    pub port: u16,
+}
+#[allow(dead_code)]
+#[derive(Debug, Deserialize, Clone)]
+pub struct LogConfig {
+    /// The log level to use
+    pub level: String,
+    /// The path to the log file
+    pub path: String,
+    /// The number of log files to retain (default: 20)
+    #[serde(default = "default_retain_days")]
+    pub retain_days: u32,
+}
+
+/// Default log files to retain
+fn default_retain_days() -> u32 {
+    5
+}
+
+#[allow(dead_code)]
+impl AppConfig {
+    /// Load the config file from the following sources:
+    /// 1. /app/config.yml
+    /// 2. config.yml
+    /// 3. BOT_SERVER_CONFIG environment variable
+    ///
+    /// Environment variables can override config values:
+    /// - MCP_PROXY_PORT: Override server port
+    /// - MCP_PROXY_LOG_DIR: Override log directory path
+    /// - MCP_PROXY_LOG_LEVEL: Override log level
+    pub fn load_config() -> Result<Self> {
+        let mut config = match (
+            File::open("/app/config.yml"),
+            File::open("config.yml"),
+            env::var("BOT_SERVER_CONFIG"),
+        ) {
+            (Ok(file), _, _) => serde_yaml::from_reader(file),
+            (_, Ok(file), _) => serde_yaml::from_reader(file),
+            (_, _, Ok(file_path)) => serde_yaml::from_reader(File::open(file_path)?),
+            _ => {
+                // 如果都没有，则使用默认配置
+                serde_yaml::from_str::<AppConfig>(DEFAULT_CONFIG_YAML)
+            }
+        }?;
+
+        // 环境变量覆盖配置（优先级最高）
+        if let Ok(port) = env::var("MCP_PROXY_PORT")
+            && let Ok(port_num) = port.parse::<u16>()
+        {
+            config.server.port = port_num;
+        }
+
+        if let Ok(log_dir) = env::var("MCP_PROXY_LOG_DIR") {
+            config.log.path = log_dir;
+        }
+
+        if let Ok(log_level) = env::var("MCP_PROXY_LOG_LEVEL") {
+            config.log.level = log_level;
+        }
+
+        Ok(config)
+    }
+
+    pub fn log_path_init(&self) -> Result<()> {
+        let log_path = &self.log.path;
+
+        // 获取日志文件的父目录
+        if let Some(parent) = Path::new(log_path).parent()
+            && !parent.exists()
+        {
+            std::fs::create_dir_all(parent)?;
+        }
+        Ok(())
+    }
+}