/* * MIT License * * Copyright (c) 2019 Jianhui Zhao * * Permission is hereby granted, free of charge, to any person obtaining a copy * of this software and associated documentation files (the "Software"), to deal * in the Software without restriction, including without limitation the rights * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell * copies of the Software, and to permit persons to whom the Software is * furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice shall be included in all * copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ 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 } 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, } 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) } }