🐶 重构部分用户数据库操作代码

2021-03-18 00:07:30 +08:00
parent d6ef8aa1db
commit 805fec4a67
50 changed files with 478 additions and 453 deletions
--- a/server/config/config.go
+++ b/server/config/config.go
@ -0,0 +1,89 @@
+package config
+
+import (
+	"strings"
+
+	"github.com/spf13/pflag"
+
+	"github.com/spf13/viper"
+)
+
+type Config struct {
+	Debug         bool
+	Demo          bool
+	DB            string
+	Server        *Server
+	Mysql         *Mysql
+	Sqlite        *Sqlite
+	ResetPassword string
+}
+
+type Mysql struct {
+	Hostname string
+	Port     int
+	Username string
+	Password string
+	Database string
+}
+
+type Sqlite struct {
+	File string
+}
+
+type Server struct {
+	Addr string
+	Cert string
+	Key  string
+}
+
+func SetupConfig() *Config {
+
+	viper.SetConfigName("config")
+	viper.SetConfigType("yml")
+	viper.AddConfigPath("/etc/next-terminal/")
+	viper.AddConfigPath("$HOME/.next-terminal")
+	viper.AddConfigPath(".")
+	viper.AutomaticEnv()
+	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
+
+	pflag.String("db", "sqlite", "db mode")
+	pflag.String("sqlite.file", "next-terminal.db", "sqlite db file")
+	pflag.String("mysql.hostname", "127.0.0.1", "mysql hostname")
+	pflag.Int("mysql.port", 3306, "mysql port")
+	pflag.String("mysql.username", "mysql", "mysql username")
+	pflag.String("mysql.password", "mysql", "mysql password")
+	pflag.String("mysql.database", "next_terminal", "mysql database")
+
+	pflag.String("server.addr", "", "server listen addr")
+	pflag.String("server.cert", "", "tls cert file")
+	pflag.String("server.key", "", "tls key file")
+	pflag.String("reset-password", "", "")
+
+	pflag.Parse()
+	_ = viper.BindPFlags(pflag.CommandLine)
+	_ = viper.ReadInConfig()
+
+	var config = &Config{
+		DB: viper.GetString("db"),
+		Mysql: &Mysql{
+			Hostname: viper.GetString("mysql.hostname"),
+			Port:     viper.GetInt("mysql.port"),
+			Username: viper.GetString("mysql.username"),
+			Password: viper.GetString("mysql.password"),
+			Database: viper.GetString("mysql.database"),
+		},
+		Sqlite: &Sqlite{
+			File: viper.GetString("sqlite.file"),
+		},
+		Server: &Server{
+			Addr: viper.GetString("server.addr"),
+			Cert: viper.GetString("server.cert"),
+			Key:  viper.GetString("server.key"),
+		},
+		ResetPassword: viper.GetString("reset-password"),
+		Debug:         viper.GetBool("debug"),
+		Demo:          viper.GetBool("demo"),
+	}
+
+	return config
+}