refactor: refactor eiblog

pkg/config/config.go

@@ -4,172 +4,51 @@ package config
 import (
 	"os"
 	"path/filepath"
-	"strings"
-
-	"gopkg.in/yaml.v3"
 )
 
-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
+// RunMode 列表
+const (
+	RunModeDev  RunMode = "dev" // 开发环境
+	RunModeProd RunMode = "pro" // 生产环境
 )
 
-// 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"`
+// RunMode 运行模式
+type RunMode string
+
+// IsReleaseMode 是否
+func (mode RunMode) IsReleaseMode() bool {
+	return mode == RunModeProd
 }
 
-// Database sql database
-type Database struct {
-	Driver string `yaml:"driver"`
-	Source string `yaml:"source"`
+// IsDevMode  是否时开发模式
+func (mode RunMode) IsDevMode() bool {
+	return mode == RunModeDev
 }
 
-// General common
-type General struct {
-	PageNum    int    `yaml:"pagenum"`    // 前台每页文章数量
-	PageSize   int    `yaml:"pagesize"`   // 后台每页文章数量
-	StartID    int    `yaml:"startid"`    // 文章启始ID
-	DescPrefix string `yaml:"descprefix"` // 文章描述前缀
-	Identifier string `yaml:"identifier"` // 文章截取标识
-	Length     int    `yaml:"length"`     // 文章预览长度
-	Timezone   string `yaml:"timezone"`   // 时区
+// IsRunMode 是否是runmode
+func (mode RunMode) IsRunMode() bool {
+	return mode == RunModeDev || mode == RunModeProd
 }
 
-// Disqus comments
-type Disqus struct {
-	ShortName   string `yaml:"shortname"`
-	PublicKey   string `yaml:"publickey"`
-	AccessToken string `yaml:"accesstoken"`
-}
-
-// 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"`
-	AdSense string `yaml:"adsense"`
-}
-
-// 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"`
-}
-
-// EiBlogApp config
-type EiBlogApp 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"`
-}
-
-// BackupApp config
-type BackupApp struct {
-	Mode
-
-	BackupTo string `yaml:"backupto"`
-	Interval string `yaml:"interval"` // circle backup, default: 7d
-	Validity int    `yaml:"validity"` // storage days, default: 60
-	Qiniu    Qiniu  `yaml:"qiniu"`    // qiniu config
-}
-
-// Config app config
-type Config struct {
-	RunMode   string    `yaml:"runmode"`
-	AppName   string    `yaml:"appname"`
-	Database  Database  `yaml:"database"`
-	ESHost    string    `yaml:"eshost"`
-	EiBlogApp EiBlogApp `yaml:"eiblogapp"`
-	BackupApp BackupApp `yaml:"backupapp"`
-}
-
-// load config file
-func init() {
-	// compatibility linux and windows
-	var err error
-	WorkDir = workDir()
-	path := filepath.Join(WorkDir, "conf", "app.yml")
-
-	data, err := os.ReadFile(path)
+// WalkWorkDir walk work dir
+func WalkWorkDir() (string, error) {
+	gopath := os.Getenv("GOPATH")
+	workDir, err := os.Getwd()
 	if err != nil {
-		panic(err)
+		return "", err
 	}
-	err = yaml.Unmarshal(data, &Conf)
-	if err != nil {
-		panic(err)
-	}
-	// read run mode from env
-	Conf.RunMode = ModeDev
-	if runmode := os.Getenv("RUN_MODE"); runmode == ModeProd {
-		Conf.RunMode = runmode
-	}
-	// read env
-	readDBEnv()
-}
+	// find work dir, try 3 times
+	for gopath != workDir && workDir != "/" {
+		dir := filepath.Join(workDir, "etc")
 
-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
+		_, err := os.Stat(dir)
+		if err == nil {
+			break
+		}
+		if !os.IsNotExist(err) {
+			return "", err
+		}
+		workDir = filepath.Dir(workDir)
 	}
+	return workDir, nil
 }