package config import ( "deploy/pkg/set" "deploy/pkg/tools" "fmt" "strings" errors "github.com/pkg/errors" confLoader "github.com/otakukaze/config-loader" ) type Config struct { Server Server `yaml:"server"` Listens []Listen `yaml:"listens"` } type Server struct { Port int `yaml:"port" env:"SERVER_PORT" default:"10230"` } type Listen struct { HTTP *HTTPListen `yaml:"http"` } type HTTPListen struct { Method string `yaml:"method"` Path string `yaml:"path"` Headers map[string]string `yaml:"headers"` Script string `yaml:"script"` } func Load(p ...string) (*Config, error) { cfg := &Config{} opts := &confLoader.LoadOptions{ FromEnv: true, } // check path exists if len(p) > 0 && p[0] != "" { if !tools.CheckExists(p[0], false) { return nil, errors.New("config not exist") } opts.ConfigFile = &confLoader.ConfigFile{ Type: confLoader.ConfigFileTypeYAML, Path: tools.ParsePath(p[0]), } } if err := confLoader.Load(&cfg, opts); err != nil { return nil, err } // check all listen path and script pathSet := set.NewSet() for _, v := range cfg.Listens { if v.HTTP == nil { continue } key := fmt.Sprintf("%s-%s", v.HTTP.Method, v.HTTP.Path) if pathSet.Has(key) { return nil, errors.WithStack(fmt.Errorf("path is duplicate: %s\n", v.HTTP.Path)) } v.HTTP.Method = strings.ToUpper(v.HTTP.Method) // check method support switch v.HTTP.Method { case "GET": case "POST": break default: return nil, errors.WithStack(fmt.Errorf("http method (%s) not support", v.HTTP.Method)) } pathSet.Add(key) // check script exists if !tools.CheckExists(v.HTTP.Script, false) { return nil, errors.WithStack(fmt.Errorf("path (%s) script not exists: %s\n", v.HTTP.Path, v.HTTP.Script)) } } return cfg, nil }