pj0235-eai_agentplatform/eai_agentplatform/backend-go/internal/config/config.go

package config

import (
	"os"
	"path/filepath"
	"strconv"
	"strings"
)

// Config 全局配置，从环境变量读取（带默认值）。
// system_config 表中的值在运行时覆盖对应字段（见 service）。
type Config struct {
	Port         string
	DBPath       string // SQLite 文件路径
	JWTSecret    string
	JWTExpireMin int

	LLMBaseURL string
	LLMAPIKey  string
	LLMModel   string
	EmbedModel string

	AssetRootDir         string
	KBDataDir            string
	KnowledgeSourceDir   string
	TrainingMaterialsDir string

	// 定期备份。SQLite 是单文件、又没有任何外部主从，坏一份就是全丢，
	// 所以默认开着：启动补一次，之后每 BackupIntervalHours 小时一次，保留最近 N 份。
	BackupEnabled       bool
	BackupDir           string
	BackupKeep          int
	BackupIntervalHours int

	FileMaxDoc     int64
	FileMaxVideo   int64
	ChunkThreshold int64

	LibreOfficeBin string
	PdftotextBin   string

	KingdeeBaseURL    string
	KingdeeAccountID  string
	KingdeeUsername   string
	KingdeePassword   string
	KingdeeLCID       int
	KingdeeTimeoutSec int
}

// Load 读取环境变量构造配置。缺失的用默认值（部署时通过 .env / systemd 注入）。
func Load() *Config {
	baseDir := backendBaseDir()
	assetRootDir := getenv("ASSET_ROOT_DIR", filepath.Join(appRootDir(baseDir), "assets"))

	return &Config{
		Port:         getenv("PORT", "8080"),
		DBPath:       getenv("DB_PATH", filepath.Join(baseDir, "data", "eai_agentplatform.db")),
		JWTSecret:    getenv("JWT_SECRET", "change-this-to-a-strong-secret-in-production"),
		JWTExpireMin: getenvInt("JWT_EXPIRE_MINUTES", 480),

		LLMBaseURL: getenv("LLM_BASE_URL", "http://127.0.0.1:11434/v1"),
		LLMAPIKey:  getenv("LLM_API_KEY", "ollama"),
		LLMModel:   getenv("LLM_MODEL", "qwen2.5:7b"),
		EmbedModel: getenv("EMBED_MODEL", "bge-m3"),

		AssetRootDir:         assetRootDir,
		KBDataDir:            getenv("KB_DATA_DIR", filepath.Join(baseDir, "data", "kb_data")),
		KnowledgeSourceDir:   getenv("KNOWLEDGE_SOURCE_DIR", filepath.Join(assetRootDir, "knowledge", "source")),
		TrainingMaterialsDir: getenv("TRAINING_MATERIALS_DIR", filepath.Join(assetRootDir, "training", "materials")),

		// 备份目录必须落在 data/ 下：systemd 单元是 ProtectSystem=strict，
		// 只有 data/ 可写（ReadWritePaths），换到别处会静默失败。
		BackupEnabled:       getenvBool("BACKUP_ENABLED", true),
		BackupDir:           getenv("BACKUP_DIR", filepath.Join(baseDir, "data", "backups")),
		BackupKeep:          getenvInt("BACKUP_KEEP", 7),
		BackupIntervalHours: getenvInt("BACKUP_INTERVAL_HOURS", 24),

		FileMaxDoc:     getenvInt64("FILE_MAX_SIZE_DOC", 209715200),
		FileMaxVideo:   getenvInt64("FILE_MAX_SIZE_VIDEO", 2147483648),
		ChunkThreshold: getenvInt64("CHUNK_THRESHOLD", 104857600),

		LibreOfficeBin: getenv("LIBREOFFICE_BIN", "libreoffice"),
		PdftotextBin:   getenv("PDFTOTEXT_BIN", "pdftotext"),

		KingdeeBaseURL:    getenv("KINGDEE_BASE_URL", ""),
		KingdeeAccountID:  getenv("KINGDEE_ACCOUNT_ID", ""),
		KingdeeUsername:   getenv("KINGDEE_USERNAME", ""),
		KingdeePassword:   getenv("KINGDEE_PASSWORD", ""),
		KingdeeLCID:       getenvInt("KINGDEE_LCID", 2052),
		KingdeeTimeoutSec: getenvInt("KINGDEE_TIMEOUT_SEC", 15),
	}
}

func backendBaseDir() string {
	if exe, err := os.Executable(); err == nil && exe != "" {
		return filepath.Clean(filepath.Join(filepath.Dir(exe), ".."))
	}
	if wd, err := os.Getwd(); err == nil && wd != "" {
		return wd
	}
	return "."
}

func appRootDir(baseDir string) string {
	return filepath.Clean(filepath.Join(baseDir, ".."))
}

func getenv(key, def string) string {
	if v := os.Getenv(key); v != "" {
		return v
	}
	return def
}

// getenvBool 只认显式的假值，其余（含没设）都算真 —— 默认开着的东西，
// 不该因为写了个看不懂的值就悄悄关掉。
func getenvBool(key string, def bool) bool {
	v := strings.ToLower(strings.TrimSpace(os.Getenv(key)))
	switch v {
	case "":
		return def
	case "0", "false", "no", "off":
		return false
	default:
		return true
	}
}

func getenvInt(key string, def int) int {
	if v := os.Getenv(key); v != "" {
		if n, err := strconv.Atoi(v); err == nil {
			return n
		}
	}
	return def
}

func getenvInt64(key string, def int64) int64 {
	if v := os.Getenv(key); v != "" {
		if n, err := strconv.ParseInt(v, 10, 64); err == nil {
			return n
		}
	}
	return def
}