92 lines
2.3 KiB
Go
92 lines
2.3 KiB
Go
package config
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
|
||
"gopkg.in/yaml.v3"
|
||
)
|
||
|
||
// Config 项目根目录 config/config.{env}.yaml（敏感项已 gitignore）。
|
||
type Config struct {
|
||
Feishu FeishuConfig `yaml:"feishu"`
|
||
HisenseGeo HisenseGeoConfig `yaml:"hisense_geo"`
|
||
}
|
||
|
||
// FeishuConfig 飞书应用凭证。
|
||
type FeishuConfig struct {
|
||
AppID string `yaml:"app_id"`
|
||
AppSecret string `yaml:"app_secret"`
|
||
}
|
||
|
||
// HisenseGeoConfig 海信 GEO 生产环境接口配置。
|
||
type HisenseGeoConfig struct {
|
||
BaseURL string `yaml:"base_url"` // 默认 https://geo-api.hisense.com
|
||
Token string `yaml:"token"` // x-token，勿提交仓库
|
||
}
|
||
|
||
// Load 按 APP_ENV 加载 config 目录下配置，默认 dev。
|
||
// 查找顺序：
|
||
// 1. CONFIG_PATH / FEISHU_CONFIG_PATH
|
||
// 2. 当前工作目录 ./config/config.{env}.yaml
|
||
// 3. 向上查找含 config/ 目录的项目根下的 config/config.{env}.yaml
|
||
func Load() (*Config, error) {
|
||
env := os.Getenv("APP_ENV")
|
||
if env == "" {
|
||
env = "dev"
|
||
}
|
||
name := "config." + env + ".yaml"
|
||
rel := filepath.Join("config", name)
|
||
|
||
candidates := []string{}
|
||
if p := os.Getenv("CONFIG_PATH"); p != "" {
|
||
candidates = append(candidates, p)
|
||
}
|
||
if p := os.Getenv("FEISHU_CONFIG_PATH"); p != "" {
|
||
candidates = append(candidates, p)
|
||
}
|
||
candidates = append(candidates, rel)
|
||
if root := FindProjectRoot(); root != "" {
|
||
candidates = append(candidates, filepath.Join(root, rel))
|
||
}
|
||
|
||
var lastErr error
|
||
for _, path := range candidates {
|
||
data, err := os.ReadFile(path)
|
||
if err != nil {
|
||
lastErr = err
|
||
continue
|
||
}
|
||
var cfg Config
|
||
if err := yaml.Unmarshal(data, &cfg); err != nil {
|
||
return nil, fmt.Errorf("解析配置失败 (%s): %w", path, err)
|
||
}
|
||
if cfg.HisenseGeo.BaseURL == "" {
|
||
cfg.HisenseGeo.BaseURL = "https://geo-api.hisense.com"
|
||
}
|
||
return &cfg, nil
|
||
}
|
||
return nil, fmt.Errorf("未找到配置文件 %s（放在 config/ 目录，或设置 CONFIG_PATH）: %w", rel, lastErr)
|
||
}
|
||
|
||
// FindProjectRoot 向上查找包含 config/ 目录的项目根。
|
||
func FindProjectRoot() string {
|
||
wd, err := os.Getwd()
|
||
if err != nil {
|
||
return ""
|
||
}
|
||
dir := wd
|
||
for {
|
||
cfgDir := filepath.Join(dir, "config")
|
||
if st, err := os.Stat(cfgDir); err == nil && st.IsDir() {
|
||
return dir
|
||
}
|
||
parent := filepath.Dir(dir)
|
||
if parent == dir {
|
||
return ""
|
||
}
|
||
dir = parent
|
||
}
|
||
}
|