feat: Add custom server configuration fields (env vars, args, headers) (#54)

Co-authored-by: Claude <noreply@anthropic.com>

Author: its-mash

Cargo.lock

@@ -25,7 +25,7 @@ futures = "0.3"
 
 # Serialization
 serde = { version = "1.0", features = ["derive"] }
-serde_json = "1.0"
+serde_json = { version = "1.0", features = ["preserve_order"] }
 
 # Error handling
 anyhow = "1.0"

Cargo.toml

@@ -1,7 +1,7 @@
 //! Tauri commands for server log management
 
 use crate::state::AppState;
-use mcpmux_core::{LogLevel, ServerLog};
+use mcpmux_core::{AppSettingsService, LogLevel, ServerLog};
 use serde::Serialize;
 use tauri::State;
 use tracing::{info, warn};
@@ -107,3 +107,33 @@ pub async fn get_server_log_file(
 
     Ok(path.to_string_lossy().to_string())
 }
+
+/// Get log retention period in days (0 = keep forever)
+#[tauri::command]
+pub async fn get_log_retention_days(state: State<'_, AppState>) -> Result<u32, String> {
+    let settings = AppSettingsService::new(state.settings_repository.clone());
+    Ok(settings.get_log_retention_days().await)
+}
+
+/// Set log retention period in days (0 = keep forever)
+#[tauri::command]
+pub async fn set_log_retention_days(days: u32, state: State<'_, AppState>) -> Result<(), String> {
+    info!("[Logs] Setting log retention to {} days", days);
+
+    let settings = AppSettingsService::new(state.settings_repository.clone());
+    settings
+        .set_log_retention_days(days)
+        .await
+        .map_err(|e| format!("Failed to save log retention setting: {}", e))?;
+
+    // Run cleanup immediately with the new setting if retention is enabled
+    if days > 0 {
+        match state.server_log_manager.cleanup_logs_older_than(days).await {
+            Ok(n) if n > 0 => info!("[Logs] Cleaned up {} old log file(s)", n),
+            Ok(_) => {}
+            Err(e) => warn!("[Logs] Cleanup after setting change failed: {}", e),
+        }
+    }
+
+    Ok(())
+}

apps/desktop/src-tauri/src/commands/logs.rs

@@ -133,6 +133,9 @@ pub async fn save_server_inputs(
     id: String,
     input_values: HashMap<String, String>,
     space_id: String,
+    env_overrides: Option<HashMap<String, String>>,
+    args_append: Option<Vec<String>>,
+    extra_headers: Option<HashMap<String, String>>,
 ) -> Result<InstalledServer, String> {
     let service_lock = app_service.read().await;
     let service = service_lock
@@ -142,7 +145,14 @@ pub async fn save_server_inputs(
     let space_uuid = uuid::Uuid::parse_str(&space_id).map_err(|e| e.to_string())?;
 
     service
-        .update_config(space_uuid, &id, input_values)
+        .update_config(
+            space_uuid,
+            &id,
+            input_values,
+            env_overrides,
+            args_append,
+            extra_headers,
+        )
         .await
         .map_err(|e| e.to_string())
 }

apps/desktop/src-tauri/src/commands/server.rs

@@ -6,7 +6,7 @@ use mcpmux_core::branding;
 use std::sync::Arc;
 use tauri::{Emitter, Manager};
 use tokio::sync::RwLock;
-use tracing::{error, info, warn};
+use tracing::{debug, error, info, warn};
 
 mod commands;
 mod services;
@@ -519,6 +519,53 @@ pub fn run() {
                 });
             }
 
+            // Start periodic log cleanup task
+            {
+                let log_manager = app_state.server_log_manager.clone();
+                let settings_repo_for_cleanup = app_state.settings_repository.clone();
+
+                tauri::async_runtime::spawn(async move {
+                    use mcpmux_core::AppSettingsService;
+
+                    let settings = AppSettingsService::new(settings_repo_for_cleanup);
+
+                    // Run cleanup once at startup
+                    let retention_days = settings.get_log_retention_days().await;
+                    if retention_days > 0 {
+                        info!(
+                            "[LogCleanup] Running startup cleanup (retention: {} days)",
+                            retention_days
+                        );
+                        match log_manager.cleanup_logs_older_than(retention_days).await {
+                            Ok(n) if n > 0 => {
+                                info!("[LogCleanup] Startup cleanup removed {} file(s)", n)
+                            }
+                            Ok(_) => debug!("[LogCleanup] No old log files to clean up"),
+                            Err(e) => warn!("[LogCleanup] Startup cleanup failed: {}", e),
+                        }
+                    }
+
+                    // Then run every 24 hours
+                    let mut interval =
+                        tokio::time::interval(std::time::Duration::from_secs(24 * 60 * 60));
+                    interval.tick().await; // skip the first immediate tick (already ran above)
+
+                    loop {
+                        interval.tick().await;
+                        let days = settings.get_log_retention_days().await;
+                        if days > 0 {
+                            match log_manager.cleanup_logs_older_than(days).await {
+                                Ok(n) if n > 0 => {
+                                    info!("[LogCleanup] Periodic cleanup removed {} file(s)", n)
+                                }
+                                Ok(_) => {}
+                                Err(e) => warn!("[LogCleanup] Periodic cleanup failed: {}", e),
+                            }
+                        }
+                    }
+                });
+            }
+
             // Setup system tray
             tray::setup_tray(app.handle())?;
 
@@ -749,6 +796,8 @@ pub fn run() {
             commands::get_server_logs,
             commands::clear_server_logs,
             commands::get_server_log_file,
+            commands::get_log_retention_days,
+            commands::set_log_retention_days,
             // App log commands
             get_logs_path,
             open_logs_folder,

apps/desktop/src-tauri/src/lib.rs

@@ -465,6 +465,7 @@ function DashboardView() {
             <div className="relative">
               <pre className="bg-mcpmux-dark text-primary-100 p-4 rounded-lg text-sm overflow-x-auto font-mono">
 {`"mcpmux": {
+  "type": "http",
   "url": "${gatewayStatus.url || 'http://localhost:3100'}/mcp"
 }`}
               </pre>
@@ -473,7 +474,7 @@ function DashboardView() {
                 size="sm"
                 className="absolute top-2 right-2"
                 onClick={async () => {
-                  const config = `"mcpmux": {\n  "url": "${gatewayStatus.url || 'http://localhost:3100'}/mcp"\n}`;
+                  const config = `"mcpmux": {\n  "type": "http",\n  "url": "${gatewayStatus.url || 'http://localhost:3100'}/mcp"\n}`;
                   await navigator.clipboard.writeText(config);
                   setExportSuccess('Config copied to clipboard!');
                   setTimeout(() => setExportSuccess(null), 2000);

apps/desktop/src/App.tsx

@@ -165,7 +165,7 @@ export function ConfigEditorModal({ spaceId, spaceName, onClose, onSaved }: Conf
         <div className="flex items-center justify-between p-4 border-b border-[rgb(var(--border))]">
           <div>
             <h3 className="text-lg font-semibold flex items-center gap-2">
-              Add Server Manually
+              Add Custom Server
             </h3>
             <p className="text-sm text-[rgb(var(--muted))]">
               Edit the JSON configuration for space: {spaceName}