package config import ( "bytes" "fmt" "io/ioutil" "runtime" "strings" "github.com/spf13/viper" ) var defaultConf = []byte(` core: enabled: true # enable httpd server address: "" # ip address to bind (default: any) shutdown_timeout: 30 # default is 30 second port: "8088" # ignore this port number if auto_tls is enabled (listen 443). worker_num: 0 # default worker number is runtime.NumCPU() queue_num: 0 # default queue number is 8192 max_notification: 100 sync: false # set true if you need get error message from fail push notification in API response. feedback_hook_url: "" # set webhook url if you need get error message asynchronously from fail push notification in API response. feedback_timeout: 10 # default is 10 second mode: "release" ssl: false cert_path: "cert.pem" key_path: "key.pem" cert_base64: "" key_base64: "" http_proxy: "" pid: enabled: false path: "gorush.pid" override: true auto_tls: enabled: false # Automatically install TLS certificates from Let's Encrypt. folder: ".cache" # folder for storing TLS certificates host: "" # which domains the Let's Encrypt will attempt grpc: enabled: false # enable gRPC server port: 9000 api: push_uri: "/api/push" stat_go_uri: "/api/stat/go" stat_app_uri: "/api/stat/app" config_uri: "/api/config" sys_stat_uri: "/sys/stats" metric_uri: "/metrics" health_uri: "/healthz" android: enabled: true apikey: "YOUR_API_KEY" max_retry: 0 # resend fail notification, default value zero is disabled huawei: enabled: false appsecret: "YOUR_APP_SECRET" appid: "YOUR_APP_ID" max_retry: 0 # resend fail notification, default value zero is disabled queue: engine: "local" # support "local", "nsq", default value is "local" nsq: addr: 127.0.0.1:4150 topic: gorush channel: gorush nats: addr: 127.0.0.1:4222 subj: gorush queue: gorush ios: enabled: false key_path: "" key_base64: "" # load iOS key from base64 input key_type: "pem" # could be pem, p12 or p8 type password: "" # certificate password, default as empty string. production: false max_concurrent_pushes: 100 # just for push ios notification max_retry: 0 # resend fail notification, default value zero is disabled key_id: "" # KeyID from developer account (Certificates, Identifiers & Profiles -> Keys) team_id: "" # TeamID from developer account (View Account -> Membership) log: format: "string" # string or json access_log: "stdout" # stdout: output to console, or define log path like "log/access_log" access_level: "debug" error_log: "stderr" # stderr: output to console, or define log path like "log/error_log" error_level: "error" hide_token: true stat: engine: "memory" # support memory, redis, boltdb, buntdb or leveldb redis: cluster: false addr: "localhost:6379" # if cluster is true, you may set this to "localhost:6379,localhost:6380,localhost:6381" password: "" db: 0 boltdb: path: "bolt.db" bucket: "gorush" buntdb: path: "bunt.db" leveldb: path: "level.db" badgerdb: path: "badger.db" `) // ConfYaml is config structure. type ConfYaml struct { Core SectionCore `yaml:"core"` API SectionAPI `yaml:"api"` Android SectionAndroid `yaml:"android"` Huawei SectionHuawei `yaml:"huawei"` Ios SectionIos `yaml:"ios"` Queue SectionQueue `yaml:"queue"` Log SectionLog `yaml:"log"` Stat SectionStat `yaml:"stat"` GRPC SectionGRPC `yaml:"grpc"` } // SectionCore is sub section of config. type SectionCore struct { Enabled bool `yaml:"enabled"` Address string `yaml:"address"` ShutdownTimeout int64 `yaml:"shutdown_timeout"` Port string `yaml:"port"` MaxNotification int64 `yaml:"max_notification"` WorkerNum int64 `yaml:"worker_num"` QueueNum int64 `yaml:"queue_num"` Mode string `yaml:"mode"` Sync bool `yaml:"sync"` SSL bool `yaml:"ssl"` CertPath string `yaml:"cert_path"` KeyPath string `yaml:"key_path"` CertBase64 string `yaml:"cert_base64"` KeyBase64 string `yaml:"key_base64"` HTTPProxy string `yaml:"http_proxy"` FeedbackURL string `yaml:"feedback_hook_url"` FeedbackTimeout int64 `yaml:"feedback_timeout"` PID SectionPID `yaml:"pid"` AutoTLS SectionAutoTLS `yaml:"auto_tls"` } // SectionAutoTLS support Let's Encrypt setting. type SectionAutoTLS struct { Enabled bool `yaml:"enabled"` Folder string `yaml:"folder"` Host string `yaml:"host"` } // SectionAPI is sub section of config. type SectionAPI struct { PushURI string `yaml:"push_uri"` StatGoURI string `yaml:"stat_go_uri"` StatAppURI string `yaml:"stat_app_uri"` ConfigURI string `yaml:"config_uri"` SysStatURI string `yaml:"sys_stat_uri"` MetricURI string `yaml:"metric_uri"` HealthURI string `yaml:"health_uri"` } // SectionAndroid is sub section of config. type SectionAndroid struct { Enabled bool `yaml:"enabled"` APIKey string `yaml:"apikey"` MaxRetry int `yaml:"max_retry"` } // SectionHuawei is sub section of config. type SectionHuawei struct { Enabled bool `yaml:"enabled"` AppSecret string `yaml:"appsecret"` AppID string `yaml:"appid"` MaxRetry int `yaml:"max_retry"` } // SectionIos is sub section of config. type SectionIos struct { Enabled bool `yaml:"enabled"` KeyPath string `yaml:"key_path"` KeyBase64 string `yaml:"key_base64"` KeyType string `yaml:"key_type"` Password string `yaml:"password"` Production bool `yaml:"production"` MaxConcurrentPushes uint `yaml:"max_concurrent_pushes"` MaxRetry int `yaml:"max_retry"` KeyID string `yaml:"key_id"` TeamID string `yaml:"team_id"` } // SectionLog is sub section of config. type SectionLog struct { Format string `yaml:"format"` AccessLog string `yaml:"access_log"` AccessLevel string `yaml:"access_level"` ErrorLog string `yaml:"error_log"` ErrorLevel string `yaml:"error_level"` HideToken bool `yaml:"hide_token"` } // SectionStat is sub section of config. type SectionStat struct { Engine string `yaml:"engine"` Redis SectionRedis `yaml:"redis"` BoltDB SectionBoltDB `yaml:"boltdb"` BuntDB SectionBuntDB `yaml:"buntdb"` LevelDB SectionLevelDB `yaml:"leveldb"` BadgerDB SectionBadgerDB `yaml:"badgerdb"` } // SectionQueue is sub section of config. type SectionQueue struct { Engine string `yaml:"engine"` NSQ SectionNSQ `yaml:"nsq"` NATS SectionNATS `yaml:"nats"` } // SectionNSQ is sub section of config. type SectionNSQ struct { Addr string `yaml:"addr"` Topic string `yaml:"topic"` Channel string `yaml:"channel"` } // SectionNATS is sub section of config. type SectionNATS struct { Addr string `yaml:"addr"` Subj string `yaml:"subj"` Queue string `yaml:"queue"` } // SectionRedis is sub section of config. type SectionRedis struct { Cluster bool `yaml:"cluster"` Addr string `yaml:"addr"` Password string `yaml:"password"` DB int `yaml:"db"` } // SectionBoltDB is sub section of config. type SectionBoltDB struct { Path string `yaml:"path"` Bucket string `yaml:"bucket"` } // SectionBuntDB is sub section of config. type SectionBuntDB struct { Path string `yaml:"path"` } // SectionLevelDB is sub section of config. type SectionLevelDB struct { Path string `yaml:"path"` } // SectionBadgerDB is sub section of config. type SectionBadgerDB struct { Path string `yaml:"path"` } // SectionPID is sub section of config. type SectionPID struct { Enabled bool `yaml:"enabled"` Path string `yaml:"path"` Override bool `yaml:"override"` } // SectionGRPC is sub section of config. type SectionGRPC struct { Enabled bool `yaml:"enabled"` Port string `yaml:"port"` } func setDefault() { viper.SetDefault("ios.max_concurrent_pushes", uint(100)) } // LoadConf load config from file and read in environment variables that match func LoadConf(confPath ...string) (*ConfYaml, error) { conf := &ConfYaml{} // load default values setDefault() viper.SetConfigType("yaml") viper.AutomaticEnv() // read in environment variables that match viper.SetEnvPrefix("gorush") // will be uppercased automatically viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_")) if len(confPath) > 0 && confPath[0] != "" { content, err := ioutil.ReadFile(confPath[0]) if err != nil { return conf, err } if err := viper.ReadConfig(bytes.NewBuffer(content)); err != nil { return conf, err } } else { // Search config in home directory with name ".gorush" (without extension). viper.AddConfigPath("/etc/gorush/") viper.AddConfigPath("$HOME/.gorush") viper.AddConfigPath(".") viper.SetConfigName("config") // If a config file is found, read it in. if err := viper.ReadInConfig(); err == nil { fmt.Println("Using config file:", viper.ConfigFileUsed()) } else if err := viper.ReadConfig(bytes.NewBuffer(defaultConf)); err != nil { // load default config return conf, err } } // Core conf.Core.Address = viper.GetString("core.address") conf.Core.Port = viper.GetString("core.port") conf.Core.ShutdownTimeout = int64(viper.GetInt("core.shutdown_timeout")) conf.Core.Enabled = viper.GetBool("core.enabled") conf.Core.WorkerNum = int64(viper.GetInt("core.worker_num")) conf.Core.QueueNum = int64(viper.GetInt("core.queue_num")) conf.Core.Mode = viper.GetString("core.mode") conf.Core.Sync = viper.GetBool("core.sync") conf.Core.FeedbackURL = viper.GetString("core.feedback_hook_url") conf.Core.FeedbackTimeout = int64(viper.GetInt("core.feedback_timeout")) conf.Core.SSL = viper.GetBool("core.ssl") conf.Core.CertPath = viper.GetString("core.cert_path") conf.Core.KeyPath = viper.GetString("core.key_path") conf.Core.CertBase64 = viper.GetString("core.cert_base64") conf.Core.KeyBase64 = viper.GetString("core.key_base64") conf.Core.MaxNotification = int64(viper.GetInt("core.max_notification")) conf.Core.HTTPProxy = viper.GetString("core.http_proxy") conf.Core.PID.Enabled = viper.GetBool("core.pid.enabled") conf.Core.PID.Path = viper.GetString("core.pid.path") conf.Core.PID.Override = viper.GetBool("core.pid.override") conf.Core.AutoTLS.Enabled = viper.GetBool("core.auto_tls.enabled") conf.Core.AutoTLS.Folder = viper.GetString("core.auto_tls.folder") conf.Core.AutoTLS.Host = viper.GetString("core.auto_tls.host") // Api conf.API.PushURI = viper.GetString("api.push_uri") conf.API.StatGoURI = viper.GetString("api.stat_go_uri") conf.API.StatAppURI = viper.GetString("api.stat_app_uri") conf.API.ConfigURI = viper.GetString("api.config_uri") conf.API.SysStatURI = viper.GetString("api.sys_stat_uri") conf.API.MetricURI = viper.GetString("api.metric_uri") conf.API.HealthURI = viper.GetString("api.health_uri") // Android conf.Android.Enabled = viper.GetBool("android.enabled") conf.Android.APIKey = viper.GetString("android.apikey") conf.Android.MaxRetry = viper.GetInt("android.max_retry") // Huawei conf.Huawei.Enabled = viper.GetBool("huawei.enabled") conf.Huawei.AppSecret = viper.GetString("huawei.appsecret") conf.Huawei.AppID = viper.GetString("huawei.appid") conf.Huawei.MaxRetry = viper.GetInt("huawei.max_retry") // iOS conf.Ios.Enabled = viper.GetBool("ios.enabled") conf.Ios.KeyPath = viper.GetString("ios.key_path") conf.Ios.KeyBase64 = viper.GetString("ios.key_base64") conf.Ios.KeyType = viper.GetString("ios.key_type") conf.Ios.Password = viper.GetString("ios.password") conf.Ios.Production = viper.GetBool("ios.production") conf.Ios.MaxConcurrentPushes = viper.GetUint("ios.max_concurrent_pushes") conf.Ios.MaxRetry = viper.GetInt("ios.max_retry") conf.Ios.KeyID = viper.GetString("ios.key_id") conf.Ios.TeamID = viper.GetString("ios.team_id") // log conf.Log.Format = viper.GetString("log.format") conf.Log.AccessLog = viper.GetString("log.access_log") conf.Log.AccessLevel = viper.GetString("log.access_level") conf.Log.ErrorLog = viper.GetString("log.error_log") conf.Log.ErrorLevel = viper.GetString("log.error_level") conf.Log.HideToken = viper.GetBool("log.hide_token") // Queue Engine conf.Queue.Engine = viper.GetString("queue.engine") conf.Queue.NSQ.Addr = viper.GetString("queue.nsq.addr") conf.Queue.NSQ.Topic = viper.GetString("queue.nsq.topic") conf.Queue.NSQ.Channel = viper.GetString("queue.nsq.channel") conf.Queue.NATS.Addr = viper.GetString("queue.nats.addr") conf.Queue.NATS.Subj = viper.GetString("queue.nats.subj") conf.Queue.NATS.Queue = viper.GetString("queue.nats.queue") // Stat Engine conf.Stat.Engine = viper.GetString("stat.engine") conf.Stat.Redis.Cluster = viper.GetBool("stat.redis.cluster") conf.Stat.Redis.Addr = viper.GetString("stat.redis.addr") conf.Stat.Redis.Password = viper.GetString("stat.redis.password") conf.Stat.Redis.DB = viper.GetInt("stat.redis.db") conf.Stat.BoltDB.Path = viper.GetString("stat.boltdb.path") conf.Stat.BoltDB.Bucket = viper.GetString("stat.boltdb.bucket") conf.Stat.BuntDB.Path = viper.GetString("stat.buntdb.path") conf.Stat.LevelDB.Path = viper.GetString("stat.leveldb.path") conf.Stat.BadgerDB.Path = viper.GetString("stat.badgerdb.path") // gRPC Server conf.GRPC.Enabled = viper.GetBool("grpc.enabled") conf.GRPC.Port = viper.GetString("grpc.port") if conf.Core.WorkerNum == int64(0) { conf.Core.WorkerNum = int64(runtime.NumCPU()) } if conf.Core.QueueNum == int64(0) { conf.Core.QueueNum = int64(8192) } return conf, nil }