/* SPDX-License-Identifier: MIT */ /* * Author: Jianhui Zhao */ package main import ( "fmt" "github.com/kylelemons/go-gypsy/yaml" "github.com/urfave/cli/v3" ) type Config struct { AddrDev string AddrUser string AddrHttpProxy string HttpProxyRedirURL string HttpProxyRedirDomain string Token string DevHookUrl string UserHookUrl string LocalAuth bool Password string AllowOrigins bool PprofAddr string SslCert string SslKey string CaCert string } func (cfg *Config) Parse(c *cli.Command) error { var yamlCfg *yaml.File var err error conf := c.String("conf") if conf != "" { yamlCfg, err = yaml.ReadFile(conf) if err != nil { return fmt.Errorf(`read config file: %s`, err.Error()) } } fields := map[string]any{ "addr-dev": &cfg.AddrDev, "addr-user": &cfg.AddrUser, "addr-http-proxy": &cfg.AddrHttpProxy, "http-proxy-redir-url": &cfg.HttpProxyRedirURL, "http-proxy-redir-domain": &cfg.HttpProxyRedirDomain, "token": &cfg.Token, "dev-hook-url": &cfg.DevHookUrl, "user-hook-url": &cfg.UserHookUrl, "local-auth": &cfg.LocalAuth, "password": &cfg.Password, "allow-origins": &cfg.AllowOrigins, "pprof": &cfg.PprofAddr, "sslcert": &cfg.SslCert, "sslkey": &cfg.SslKey, "cacert": &cfg.CaCert, } for name, opt := range fields { if yamlCfg != nil { if err = getConfigOpt(yamlCfg, name, opt); err != nil { return err } } getFlagOpt(c, name, opt) } return nil } func getConfigOpt(yamlCfg *yaml.File, name string, opt any) error { var err error switch opt := opt.(type) { case *string: var val string val, err = yamlCfg.Get(name) if err == nil { *opt = val } case *int: var val int64 val, err = yamlCfg.GetInt(name) if err == nil { *opt = int(val) } case *bool: var val bool val, err = yamlCfg.GetBool(name) if err == nil { *opt = val } default: return fmt.Errorf("unsupported type for option %s", name) } if err != nil { if _, ok := err.(*yaml.NodeNotFound); ok { return nil } return fmt.Errorf(`invalud "%s": %w`, name, err) } return nil } func getFlagOpt(c *cli.Command, name string, opt any) { if !c.IsSet(name) { return } switch opt := opt.(type) { case *string: *opt = c.String(name) case *int: *opt = c.Int(name) case *bool: *opt = c.Bool(name) } }