完成大部分基础功能

web/service/setting.go

@@ -2,22 +2,112 @@ package service
 
 import (
 	_ "embed"
+	"errors"
+	"fmt"
+	"reflect"
 	"strconv"
 	"strings"
 	"time"
 	"x-ui/database"
 	"x-ui/database/model"
 	"x-ui/logger"
+	"x-ui/util/common"
 	"x-ui/util/random"
+	"x-ui/util/reflect_util"
+	"x-ui/web/entity"
 )
 
 //go:embed config.json
 var xrayTemplateConfig string
 
+var defaultValueMap = map[string]string{
+	"xrayTemplateConfig": xrayTemplateConfig,
+	"webListen":          "",
+	"webPort":            "65432",
+	"webCertFile":        "",
+	"webKeyFile":         "",
+	"secret":             random.Seq(32),
+	"webBasePath":        "/",
+	"timeLocation":       "Asia/Shanghai",
+}
+
 type SettingService struct {
 }
 
-func (s *SettingService) ClearSetting() error {
+func (s *SettingService) GetAllSetting() (*entity.AllSetting, error) {
+	db := database.GetDB()
+	settings := make([]*model.Setting, 0)
+	err := db.Model(model.Setting{}).Find(&settings).Error
+	if err != nil {
+		return nil, err
+	}
+	allSetting := &entity.AllSetting{}
+	t := reflect.TypeOf(allSetting).Elem()
+	v := reflect.ValueOf(allSetting).Elem()
+	fields := reflect_util.GetFields(t)
+
+	setSetting := func(key, value string) (err error) {
+		defer func() {
+			panicErr := recover()
+			if panicErr != nil {
+				err = errors.New(fmt.Sprint(panicErr))
+			}
+		}()
+
+		var found bool
+		var field reflect.StructField
+		for _, f := range fields {
+			if f.Tag.Get("json") == key {
+				field = f
+				found = true
+				break
+			}
+		}
+
+		if !found {
+			// 有些设置自动生成，不需要返回到前端给用户修改
+			return nil
+		}
+
+		fieldV := v.FieldByName(field.Name)
+		switch t := fieldV.Interface().(type) {
+		case int:
+			n, err := strconv.ParseInt(value, 10, 32)
+			if err != nil {
+				return err
+			}
+			fieldV.SetInt(n)
+		case string:
+			fieldV.SetString(value)
+		default:
+			return common.NewErrorf("unknown field %v type %v", key, t)
+		}
+		return
+	}
+
+	keyMap := map[string]bool{}
+	for _, setting := range settings {
+		err := setSetting(setting.Key, setting.Value)
+		if err != nil {
+			return nil, err
+		}
+		keyMap[setting.Key] = true
+	}
+
+	for key, value := range defaultValueMap {
+		if keyMap[key] {
+			continue
+		}
+		err := setSetting(key, value)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	return allSetting, nil
+}
+
+func (s *SettingService) ResetSettings() error {
 	db := database.GetDB()
 	return db.Delete(model.Setting{}).Error
 }
@@ -48,18 +138,22 @@ func (s *SettingService) saveSetting(key string, value string) error {
 	return db.Save(setting).Error
 }
 
-func (s *SettingService) getString(key string, defaultValue string) (string, error) {
+func (s *SettingService) getString(key string) (string, error) {
 	setting, err := s.getSetting(key)
 	if database.IsNotFound(err) {
-		return defaultValue, nil
+		value, ok := defaultValueMap[key]
+		if !ok {
+			return "", common.NewErrorf("key <%v> not in defaultValueMap", key)
+		}
+		return value, nil
 	} else if err != nil {
 		return "", err
 	}
 	return setting.Value, nil
 }
 
-func (s *SettingService) getInt(key string, defaultValue int) (int, error) {
-	str, err := s.getString(key, strconv.Itoa(defaultValue))
+func (s *SettingService) getInt(key string) (int, error) {
+	str, err := s.getString(key)
 	if err != nil {
 		return 0, err
 	}
@@ -67,29 +161,28 @@ func (s *SettingService) getInt(key string, defaultValue int) (int, error) {
 }
 
 func (s *SettingService) GetXrayConfigTemplate() (string, error) {
-	return s.getString("xray_template_config", xrayTemplateConfig)
+	return s.getString("xrayTemplateConfig")
 }
 
 func (s *SettingService) GetListen() (string, error) {
-	return s.getString("web_listen", "")
+	return s.getString("webListen")
 }
 
 func (s *SettingService) GetPort() (int, error) {
-	return s.getInt("web_port", 65432)
+	return s.getInt("webPort")
 }
 
 func (s *SettingService) GetCertFile() (string, error) {
-	return s.getString("web_cert_file", "")
+	return s.getString("webCertFile")
 }
 
 func (s *SettingService) GetKeyFile() (string, error) {
-	return s.getString("web_key_file", "")
+	return s.getString("webKeyFile")
 }
 
 func (s *SettingService) GetSecret() ([]byte, error) {
-	seq := random.Seq(32)
-	secret, err := s.getString("secret", seq)
-	if secret == seq {
+	secret, err := s.getString("secret")
+	if secret == defaultValueMap["secret"] {
 		err := s.saveSetting("secret", secret)
 		if err != nil {
 			logger.Warning("save secret failed:", err)
@@ -99,7 +192,7 @@ func (s *SettingService) GetSecret() ([]byte, error) {
 }
 
 func (s *SettingService) GetBasePath() (string, error) {
-	basePath, err := s.getString("web_base_path", "/")
+	basePath, err := s.getString("webBasePath")
 	if err != nil {
 		return "", err
 	}
@@ -113,15 +206,36 @@ func (s *SettingService) GetBasePath() (string, error) {
 }
 
 func (s *SettingService) GetTimeLocation() (*time.Location, error) {
-	defaultLocation := "Asia/Shanghai"
-	l, err := s.getString("time_location", defaultLocation)
+	l, err := s.getString("timeLocation")
 	if err != nil {
 		return nil, err
 	}
 	location, err := time.LoadLocation(l)
 	if err != nil {
+		defaultLocation := defaultValueMap["timeLocation"]
 		logger.Errorf("location <%v> not exist, using default location: %v", l, defaultLocation)
 		return time.LoadLocation(defaultLocation)
 	}
 	return location, nil
 }
+
+func (s *SettingService) UpdateAllSetting(allSetting *entity.AllSetting) error {
+	if err := allSetting.CheckValid(); err != nil {
+		return err
+	}
+
+	v := reflect.ValueOf(allSetting).Elem()
+	t := reflect.TypeOf(allSetting).Elem()
+	fields := reflect_util.GetFields(t)
+	errs := make([]error, 0)
+	for _, field := range fields {
+		key := field.Tag.Get("json")
+		fieldV := v.FieldByName(field.Name)
+		value := fmt.Sprint(fieldV.Interface())
+		err := s.saveSetting(key, value)
+		if err != nil {
+			errs = append(errs, err)
+		}
+	}
+	return common.Combine(errs...)
+}