package config import ( "github.com/spf13/cobra" "github.com/spf13/viper" "demodesk/neko/internal/utils" ) type Server struct { Cert string Key string Bind string Static string CORS []string } func (Server) Init(cmd *cobra.Command) error { cmd.PersistentFlags().String("server.bind", "127.0.0.1:8080", "address/port/socket to serve neko") if err := viper.BindPFlag("server.bind", cmd.PersistentFlags().Lookup("server.bind")); err != nil { return err } cmd.PersistentFlags().String("server.cert", "", "path to the SSL cert used to secure the neko server") if err := viper.BindPFlag("server.cert", cmd.PersistentFlags().Lookup("server.cert")); err != nil { return err } cmd.PersistentFlags().String("server.key", "", "path to the SSL key used to secure the neko server") if err := viper.BindPFlag("server.key", cmd.PersistentFlags().Lookup("server.key")); err != nil { return err } cmd.PersistentFlags().String("server.static", "", "path to neko client files to serve") if err := viper.BindPFlag("server.static", cmd.PersistentFlags().Lookup("server.static")); err != nil { return err } cmd.PersistentFlags().StringSlice("server.cors", []string{"*"}, "list of allowed origins for CORS") if err := viper.BindPFlag("server.cors", cmd.PersistentFlags().Lookup("server.cors")); err != nil { return err } return nil } func (s *Server) Set() { s.Cert = viper.GetString("server.cert") s.Key = viper.GetString("server.key") s.Bind = viper.GetString("server.bind") s.Static = viper.GetString("server.static") s.CORS = viper.GetStringSlice("server.cors") in, _ := utils.ArrayIn("*", s.CORS) if len(s.CORS) == 0 || in { s.CORS = []string{"*"} } } func (s *Server) AllowOrigin(origin string) bool { in, _ := utils.ArrayIn(origin, s.CORS) return in || s.CORS[0] == "*" }