Multi Database MCP Server

package config import ( "encoding/json" "fmt" "os" "path/filepath" "strconv" "github.com/joho/godotenv" "github.com/FreePeak/db-mcp-server/internal/logger" "github.com/FreePeak/db-mcp-server/pkg/db" ) // Config holds all server configuration type Config struct { ServerPort int TransportMode string LogLevel string DBConfig DatabaseConfig // Legacy single database config MultiDBConfig *db.MultiDBConfig // New multi-database config ConfigPath string // Path to the configuration file DisableLogging bool // When true, disables logging in stdio/SSE transport } // DatabaseConfig holds database configuration (legacy support) type DatabaseConfig struct { Type string Host string Port int User string Password string Name string } // LoadConfig loads the configuration from environment variables and optional JSON config func LoadConfig(logDir string) (*Config, error) { // Reinitialize logger with info level and custom log directory // This ensures consistent logging throughout config loading regardless of initial setup logger.Initialize(logger.Config{Level: "info", LogDir: logDir}) // Load .env file if it exists err := godotenv.Load() if err != nil { logger.Info("Warning: .env file not found, using environment variables only") } else { logger.Info("Loaded configuration from .env file") } port, err := strconv.Atoi(getEnv("SERVER_PORT", "9090")) if err != nil { logger.Warn("Warning: Invalid SERVER_PORT value, using default 9090") port = 9090 } dbPort, err := strconv.Atoi(getEnv("DB_PORT", "3306")) if err != nil { logger.Warn("Warning: Invalid DB_PORT value, using default 3306") dbPort = 3306 } // Get config path from environment or use default configPath := getEnv("CONFIG_PATH", "") if configPath == "" { configPath = getEnv("DB_CONFIG_FILE", "config.json") } // Resolve absolute path if relative path is provided if !filepath.IsAbs(configPath) { absPath, err := filepath.Abs(configPath) if err != nil { logger.Warn("Warning: Could not resolve absolute path for config file: %v", err) } else { configPath = absPath } } // Parse DISABLE_LOGGING env var disableLogging := false if v := getEnv("DISABLE_LOGGING", "false"); v == "true" || v == "1" { disableLogging = true } config := &Config{ ServerPort: port, TransportMode: getEnv("TRANSPORT_MODE", "sse"), LogLevel: getEnv("LOG_LEVEL", "info"), ConfigPath: configPath, DisableLogging: disableLogging, DBConfig: DatabaseConfig{ Type: getEnv("DB_TYPE", "mysql"), Host: getEnv("DB_HOST", "localhost"), Port: dbPort, User: getEnv("DB_USER", ""), Password: getEnv("DB_PASSWORD", ""), Name: getEnv("DB_NAME", ""), }, } // Try to load multi-database configuration from JSON file if _, err := os.Stat(config.ConfigPath); err == nil { logger.Info("Loading configuration from: %s", config.ConfigPath) configData, err := os.ReadFile(config.ConfigPath) if err != nil { return nil, fmt.Errorf("failed to read config file %s: %w", config.ConfigPath, err) } var multiDBConfig db.MultiDBConfig if err := json.Unmarshal(configData, &multiDBConfig); err != nil { return nil, fmt.Errorf("failed to parse config file %s: %w", config.ConfigPath, err) } config.MultiDBConfig = &multiDBConfig } else { logger.Info("Warning: Config file not found at %s, using environment variables", config.ConfigPath) // If no JSON config found, create a single connection config from environment variables config.MultiDBConfig = &db.MultiDBConfig{ Connections: []db.DatabaseConnectionConfig{ { ID: "default", Type: config.DBConfig.Type, Host: config.DBConfig.Host, Port: config.DBConfig.Port, User: config.DBConfig.User, Password: config.DBConfig.Password, Name: config.DBConfig.Name, }, }, } } return config, nil } // getEnv gets an environment variable or returns a default value func getEnv(key, defaultValue string) string { value := os.Getenv(key) if value == "" { return defaultValue } return value }