init

pkg/config/config.go

@@ -0,0 +1,5 @@
+package config
+
+const (
+	UserServiceConfigFilePath = "/etc/casaos/user-service.conf"
+)

pkg/config/init.go

@@ -0,0 +1,94 @@
+package config
+
+import (
+	"fmt"
+	"log"
+	"os"
+
+	"github.com/IceWhaleTech/CasaOS-Common/utils/constants"
+	"github.com/IceWhaleTech/CasaOS-UserService/model"
+	"gopkg.in/ini.v1"
+)
+
+// models with default values
+
+var (
+	CommonInfo = &model.CommonModel{
+		RuntimePath: constants.DefaultRuntimePath,
+	}
+
+	AppInfo = &model.APPModel{
+		DBPath:       constants.DefaultDataPath,
+		UserDataPath: constants.DefaultDataPath,
+		LogPath:      constants.DefaultLogPath,
+		LogSaveName:  "user",
+		LogFileExt:   "log",
+	}
+
+	Cfg            *ini.File
+	ConfigFilePath string
+)
+
+func InitSetup(config string, sample string) {
+	ConfigFilePath = UserServiceConfigFilePath
+	if len(config) > 0 {
+		ConfigFilePath = config
+	}
+
+	// create default config file if not exist
+	if _, err := os.Stat(ConfigFilePath); os.IsNotExist(err) {
+		fmt.Println("config file not exist, create it")
+		// create config file
+		file, err := os.Create(ConfigFilePath)
+		if err != nil {
+			panic(err)
+		}
+		defer file.Close()
+
+		// write default config
+		_, err = file.WriteString(sample)
+		if err != nil {
+			panic(err)
+		}
+	}
+
+	var err error
+
+	Cfg, err = ini.Load(ConfigFilePath)
+	if err != nil {
+		fmt.Printf("Fail to read file: %v", err)
+		os.Exit(1)
+	}
+
+	mapTo("common", CommonInfo)
+	mapTo("app", AppInfo)
+}
+
+func SaveSetup(config string) {
+	reflectFrom("common", CommonInfo)
+	reflectFrom("app", AppInfo)
+
+	configFilePath := UserServiceConfigFilePath
+	if len(config) > 0 {
+		configFilePath = config
+	}
+
+	if err := Cfg.SaveTo(configFilePath); err != nil {
+		fmt.Printf("Fail to save file: %v", err)
+		os.Exit(1)
+	}
+}
+
+func mapTo(section string, v interface{}) {
+	err := Cfg.Section(section).MapTo(v)
+	if err != nil {
+		log.Fatalf("Cfg.MapTo %s err: %v", section, err)
+	}
+}
+
+func reflectFrom(section string, v interface{}) {
+	err := Cfg.Section(section).ReflectFrom(v)
+	if err != nil {
+		log.Fatalf("Cfg.ReflectFrom %s err: %v", section, err)
+	}
+}