2026-06-18 22:48:16 +08:00
|
|
|
package config
|
|
|
|
|
|
|
|
|
|
import (
|
2026-06-20 21:57:46 +08:00
|
|
|
"crypto/rand"
|
|
|
|
|
"encoding/hex"
|
2026-06-18 22:48:16 +08:00
|
|
|
"os"
|
|
|
|
|
"path/filepath"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
type Config struct {
|
2026-06-20 21:57:46 +08:00
|
|
|
Port string
|
|
|
|
|
GinMode string
|
|
|
|
|
DataDir string
|
|
|
|
|
SessionSecret string
|
2026-06-18 22:48:16 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func Load() *Config {
|
|
|
|
|
cfg := &Config{
|
2026-06-20 21:57:46 +08:00
|
|
|
Port: getEnv("PORT", "8080"),
|
|
|
|
|
GinMode: getEnv("GIN_MODE", "debug"),
|
|
|
|
|
DataDir: getEnv("DATA_DIR", "data"),
|
|
|
|
|
SessionSecret: getEnv("SESSION_SECRET", ""),
|
|
|
|
|
}
|
|
|
|
|
// Generate random session secret if not provided
|
|
|
|
|
if cfg.SessionSecret == "" {
|
|
|
|
|
b := make([]byte, 32)
|
|
|
|
|
rand.Read(b)
|
|
|
|
|
cfg.SessionSecret = hex.EncodeToString(b)
|
2026-06-18 22:48:16 +08:00
|
|
|
}
|
|
|
|
|
// Ensure data directories exist
|
|
|
|
|
os.MkdirAll(cfg.DataDir, 0o755)
|
|
|
|
|
os.MkdirAll(filepath.Join(cfg.DataDir, "repos"), 0o755)
|
|
|
|
|
return cfg
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (c *Config) DBPath() string {
|
|
|
|
|
return filepath.Join(c.DataDir, "pr-helper.db")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (c *Config) ReposDir() string {
|
|
|
|
|
return filepath.Join(c.DataDir, "repos")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func getEnv(key, fallback string) string {
|
|
|
|
|
if v := os.Getenv(key); v != "" {
|
|
|
|
|
return v
|
|
|
|
|
}
|
|
|
|
|
return fallback
|
|
|
|
|
}
|