X-Agents/server/internal/config/config.go

package config

import (
	"fmt"
	"log"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"

	"github.com/spf13/viper"
)

type Config struct {
	Port              string
	JWTSecret         string
	DatabaseHost      string
	DatabasePort      string
	DatabaseUser      string
	DatabasePassword  string
	DatabaseName      string
	DatabaseURL       string // 拼接后的完整连接字符串
	PythonServiceURL  string
	AICoreServiceAddr string // AI-Core gRPC 服务地址，如 "localhost:50051"
	// 文件上传配置
	UploadMode        string // "local" 或 "minio"
	UploadLocalPath   string // 本地存储路径，如 "resource/files"
	ServerBaseURL     string // 服务器基础 URL，用于生成本地文件 URL
	MarkdownLocalPath string // Markdown 文件存储路径，如 "resource/markdown"
	// MinIO 配置
	MinIOEndpoint   string
	MinIOAccessKey  string
	MinIOSecretKey  string
	MinIOBucket     string
	MinIOUseSSL    bool
}

func Load() *Config {
	viper.SetConfigName("config")
	viper.SetConfigType("yaml")
	viper.AddConfigPath("./config")
	viper.AddConfigPath("../config")
	viper.AddConfigPath("../../config")

	// 默认值
	viper.SetDefault("port", "8080")
	viper.SetDefault("jwt_secret", "your-secret-key-change-in-production")
	viper.SetDefault("python_service_url", "http://localhost:8081")
	viper.SetDefault("ai_core_service_addr", "localhost:50051")
	// 数据库默认配置
	viper.SetDefault("database_host", "localhost")
	viper.SetDefault("database_port", "3306")
	viper.SetDefault("database_user", "root")
	viper.SetDefault("database_password", "root")
	viper.SetDefault("database_name", "x_agents")
	// 文件上传默认配置
	viper.SetDefault("upload_mode", "local")
	viper.SetDefault("upload_local_path", "resource/files")
	viper.SetDefault("server_base_url", "http://localhost:8080")
	viper.SetDefault("markdown_local_path", "resource/markdown")
	viper.SetDefault("minio_endpoint", "localhost:9000")
	viper.SetDefault("minio_access_key", "")
	viper.SetDefault("minio_secret_key", "")
	viper.SetDefault("minio_bucket", "x-agents")
	viper.SetDefault("minio_use_ssl", false)

	if err := viper.ReadInConfig(); err != nil {
		log.Printf("Using default config: %v", err)
	}

	// 拼接数据库连接字符串
	dbHost := viper.GetString("database_host")
	dbPort := viper.GetString("database_port")
	dbUser := viper.GetString("database_user")
	dbPassword := viper.GetString("database_password")
	dbName := viper.GetString("database_name")
	databaseURL := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		dbUser, dbPassword, dbHost, dbPort, dbName)

	return &Config{
		Port:              viper.GetString("port"),
		JWTSecret:         viper.GetString("jwt_secret"),
		DatabaseURL:       databaseURL,
		DatabaseHost:      dbHost,
		DatabasePort:      dbPort,
		DatabaseUser:      dbUser,
		DatabasePassword:  dbPassword,
		DatabaseName:      dbName,
		PythonServiceURL:  viper.GetString("python_service_url"),
		AICoreServiceAddr:  viper.GetString("ai_core_service_addr"),
		// 文件上传配置
		UploadMode:        viper.GetString("upload_mode"),
		UploadLocalPath:   viper.GetString("upload_local_path"),
		ServerBaseURL:     viper.GetString("server_base_url"),
		MarkdownLocalPath: viper.GetString("markdown_local_path"),
		// MinIO 配置
		MinIOEndpoint:  viper.GetString("minio_endpoint"),
		MinIOAccessKey: viper.GetString("minio_access_key"),
		MinIOSecretKey: viper.GetString("minio_secret_key"),
		MinIOBucket:    viper.GetString("minio_bucket"),
		MinIOUseSSL:    viper.GetBool("minio_use_ssl"),
	}
}

func InitDB(cfg *Config) (*gorm.DB, error) {
	dsn := cfg.DatabaseURL
	if dsn == "" {
		return nil, fmt.Errorf("database URL is empty")
	}

	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
		Logger: logger.Default.LogMode(logger.Silent),
	})
	if err != nil {
		return nil, fmt.Errorf("failed to connect database: %w", err)
	}

	log.Println("Database connected successfully")
	return db, nil
}