mugit/internal/config/config.go(view raw)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 |
package config
import (
"errors"
"fmt"
"os"
"path/filepath"
"gopkg.in/yaml.v2"
)
var ErrConfigNotFound = errors.New("no config file found")
type Config struct {
Server struct {
Host string `yaml:"host"`
Port int `yaml:"port"`
} `yaml:"server"`
Meta struct {
Title string `yaml:"title"`
Description string `yaml:"description"`
Host string `yaml:"host"`
} `yaml:"meta"`
Repo struct {
Dir string `yaml:"dir"`
Readmes []string `yaml:"readmes"`
Masters []string `yaml:"masters"`
} `yaml:"repo"`
SSH struct {
Enable bool `yaml:"enable"`
Port int `yaml:"port"`
HostKey string `yaml:"host_key"`
Keys []string `yaml:"keys"`
} `yaml:"ssh"`
Mirror struct {
Enable bool `yaml:"enable"`
Interval string `yaml:"interval"`
GithubToken string `yaml:"github_token"`
} `yaml:"mirror"`
}
// Load loads configuration with the following priority:
// 1. User provided fpath (if provided and exists)
// 2. $XDG_CONFIG_HOME/mugit/config.yaml
// 3. $HOME/.config/mugit/config.yaml (fallback if XDG_CONFIG_HOME not set)
// 4. /etc/mugit/config.yaml
func Load(fpath string) (*Config, error) {
configPath, err := findConfigFile(fpath)
if err != nil {
return nil, err
}
configBytes, err := os.ReadFile(configPath)
if err != nil {
return nil, err
}
var config Config
if cerr := yaml.Unmarshal(configBytes, &config); cerr != nil {
return nil, fmt.Errorf("parsing config: %w", cerr)
}
if config.Repo.Dir, err = filepath.Abs(config.Repo.Dir); err != nil {
return nil, err
}
if verr := config.validate(); verr != nil {
return nil, verr
}
return &config, nil
}
func (c Config) validate() error {
// var errs []error
// return errors.Join(errs...)
return nil
}
func findConfigFile(userPath string) (string, error) {
if userPath != "" {
if _, err := os.Stat(userPath); err == nil {
return userPath, nil
}
}
paths := []string{}
if xdg := os.Getenv("XDG_CONFIG_HOME"); xdg != "" {
paths = append(paths, filepath.Join(xdg, "mugit", "config.yaml"))
} else if home := os.Getenv("HOME"); home != "" {
paths = append(paths, filepath.Join(home, ".config", "mugit", "config.yaml"))
}
paths = append(paths, "/etc/mugit/config.yaml")
for _, p := range paths {
if _, err := os.Stat(p); err == nil {
return p, nil
}
}
return "", ErrConfigNotFound
}
|