refactor: eiblog

pkg/config/config.go

@@ -0,0 +1,173 @@
+// Package config provides ...
+package config
+
+import (
+	"io/ioutil"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"gopkg.in/yaml.v2"
+)
+
+var (
+	// Conf config instance
+	Conf Config
+
+	// ModeDev run mode as development
+	ModeDev = "dev"
+	// ModeProd run mode as production
+	ModeProd = "prod"
+	// WorkDir workspace dir
+	WorkDir string
+)
+
+// Mode run mode
+type Mode struct {
+	Name       string `yaml:"name"`
+	EnableHTTP bool   `yaml:"enablehttp"`
+	HTTPPort   int    `yaml:"httpport"`
+	EnableGRPC bool   `yaml:"enablegrpc"`
+	GRPCPort   int    `yaml:"grpcport"`
+	Host       string `yaml:"host"`
+}
+
+// Database sql database
+type Database struct {
+	Driver string `yaml:"driver"`
+	Source string `yaml:"source"`
+}
+
+// General common
+type General struct {
+	PageNum    int    `yaml:"pagenum"`    // 前台每页文章数量
+	PageSize   int    `yaml:"pagesize"`   // 后台每页文章数量
+	StartID    int32  `yaml:"startid"`    // 文章启始ID
+	DescPrefix string `yaml:"descprefix"` // 文章描述前缀
+	Identifier string `yaml:"identifier"` // 文章截取标识
+	Length     int    `yaml:"length"`     // 文章预览长度
+	Trash      int    `yaml:"trash"`      // 回收箱文章保留时间
+	Clean      int    `yaml:"clean"`      // 清理回收箱频率
+}
+
+// Disqus comments
+type Disqus struct {
+	ShortName   string `yaml:"shortname"`
+	PublicKey   string `yaml:"publickey"`
+	AccessToken string `yaml:"accesstoken"`
+	Interval    int    `yaml:"interval"` // 获取评论数量间隔
+}
+
+// Twitter card
+type Twitter struct {
+	Card    string `yaml:"card"`
+	Site    string `yaml:"site"`
+	Image   string `yaml:"image"`
+	Address string `yaml:"address"`
+}
+
+// Google analytics
+type Google struct {
+	URL string `yaml:"url"`
+	Tid string `yaml:"tid"`
+	V   string `yaml:"v"`
+	T   string `yaml:"t"`
+}
+
+// Qiniu oss
+type Qiniu struct {
+	Bucket    string `yaml:"bucket"`
+	Domain    string `yaml:"domain"`
+	AccessKey string `yaml:"accesskey"`
+	SecretKey string `yaml:"secretkey"`
+}
+
+// FeedRPC feedr
+type FeedRPC struct {
+	FeedrURL string   `yaml:"feedrurl"`
+	PingRPC  []string `yaml:"pingrpc"`
+}
+
+// Account info
+type Account struct {
+	Username    string `yaml:"username"` // *
+	Password    string `yaml:"password"` // *
+	Email       string `yaml:"email"`
+	PhoneNumber string `yaml:"phonenumber"`
+	Address     string `yaml:"address"`
+}
+
+// Blogger info
+type Blogger struct {
+	BlogName  string `yaml:"blogname"`
+	SubTitle  string `yaml:"subtitle"`
+	BeiAn     string `yaml:"beian"`
+	BTitle    string `yaml:"btitle"`
+	Copyright string `yaml:"copyright"`
+}
+
+// BlogApp config
+type BlogApp struct {
+	Mode
+
+	StaticVersion int      `yaml:"staticversion"`
+	HotWords      []string `yaml:"hotwords"`
+	General       General  `yaml:"general"`
+	Disqus        Disqus   `yaml:"disqus"`
+	Google        Google   `yaml:"google"`
+	Qiniu         Qiniu    `yaml:"qiniu"`
+	Twitter       Twitter  `yaml:"twitter"`
+	FeedRPC       FeedRPC  `yaml:"feedrpc"`
+	Account       Account  `yaml:"account"`
+	Blogger       Blogger  `yaml:"blogger"`
+}
+
+// Config app config
+type Config struct {
+	RunMode   string   `yaml:"runmode"`
+	AppName   string   `yaml:"appname"`
+	Database  Database `yaml:"database"`
+	ESHost    string   `yaml:"eshost"`
+	BlogApp   BlogApp  `yaml:"blogapp"`
+	BackupApp Mode     `yaml:"backupapp"`
+}
+
+// load config file
+func init() {
+	// compatibility linux and windows
+	var err error
+	WorkDir, err = os.Getwd()
+	if err != nil {
+		panic(err)
+	}
+	path := filepath.Join(WorkDir, "conf", "app.yml")
+
+	data, err := ioutil.ReadFile(path)
+	if err != nil {
+		panic(err)
+	}
+	err = yaml.Unmarshal(data, &Conf)
+	if err != nil {
+		panic(err)
+	}
+	// read run mode from env
+	if runmode := os.Getenv("RUN_MODE"); runmode != "" {
+		if runmode != ModeDev && runmode != ModeProd {
+			panic("invalid RUN_MODE from env: " + runmode)
+		}
+		Conf.RunMode = runmode
+	}
+	// read env
+	readDBEnv()
+}
+
+func readDBEnv() {
+	key := strings.ToUpper(Conf.AppName) + "_DB_DRIVER"
+	if d := os.Getenv(key); d != "" {
+		Conf.Database.Driver = d
+	}
+	key = strings.ToUpper(Conf.AppName) + "_DB_SOURCE"
+	if s := os.Getenv(key); s != "" {
+		Conf.Database.Source = s
+	}
+}