✨ bootstrap

2021-10-27 22:45:36 +08:00
parent 9644cc98c3
commit 55f683b12d
18 changed files with 311 additions and 133 deletions
--- a/bootstrap/model.go
+++ b/bootstrap/model.go
@ -0,0 +1,149 @@
+package bootstrap
+
+import (
+	"fmt"
+	"github.com/Xhofe/alist/conf"
+	"github.com/Xhofe/alist/drivers"
+	"github.com/Xhofe/alist/model"
+	"github.com/Xhofe/alist/utils"
+	log "github.com/sirupsen/logrus"
+	"gorm.io/driver/mysql"
+	"gorm.io/driver/postgres"
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
+	"gorm.io/gorm/schema"
+	"strings"
+)
+
+func InitModel() {
+	log.Infof("init model...")
+	config := conf.Conf.Database
+	switch config.Type {
+	case "sqlite3":
+		{
+			if !(strings.HasSuffix(config.DBFile, ".db") && len(config.DBFile) > 3) {
+				log.Fatalf("db name error.")
+			}
+			db, err := gorm.Open(sqlite.Open(config.DBFile), &gorm.Config{
+				NamingStrategy: schema.NamingStrategy{
+					TablePrefix: config.TablePrefix,
+				},
+			})
+			if err != nil {
+				log.Fatalf("failed to connect database:%s", err.Error())
+			}
+			conf.DB = db
+		}
+	case "mysql":
+		{
+			dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
+				config.User, config.Password, config.Host, config.Port, config.Name)
+			db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
+				NamingStrategy: schema.NamingStrategy{
+					TablePrefix: config.TablePrefix,
+				},
+			})
+			if err != nil {
+				log.Fatalf("failed to connect database:%s", err.Error())
+			}
+			conf.DB = db
+		}
+	case "postgres":
+		{
+			dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%d sslmode=disable TimeZone=Asia/Shanghai",
+				config.Host, config.User, config.Password, config.Name, config.Port)
+			db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{
+				NamingStrategy: schema.NamingStrategy{
+					TablePrefix: config.TablePrefix,
+				},
+			})
+			if err != nil {
+				log.Errorf("failed to connect database:%s", err.Error())
+			}
+			conf.DB = db
+
+		}
+	default:
+		log.Fatalf("not supported database type: %s", config.Type)
+	}
+	log.Infof("auto migrate model")
+	err := conf.DB.AutoMigrate(&model.SettingItem{}, &model.Account{})
+	if err != nil {
+		log.Fatalf("failed to auto migrate")
+	}
+
+	// TODO init filetype
+	initAccounts()
+	initSettings()
+}
+
+func initAccounts() {
+	log.Infof("init accounts...")
+	var accounts []model.Account
+	if err := conf.DB.Find(&accounts).Error; err != nil {
+		log.Fatalf("failed sync init accounts")
+	}
+	for _, account := range accounts {
+		model.RegisterAccount(account)
+		driver, ok := drivers.GetDriver(account.Type)
+		if !ok {
+			log.Error("no [%s] driver", driver)
+		} else {
+			err := driver.Save(&account, nil)
+			if err != nil {
+				log.Errorf("init account [%s] error:[%s]", account.Name, err.Error())
+			}
+		}
+	}
+}
+
+func initSettings() {
+	log.Infof("init settings...")
+	version, err := model.GetSettingByKey("version")
+	if err != nil {
+		log.Debugf("first run")
+		version = &model.SettingItem{
+			Key:         "version",
+			Value:       "0.0.0",
+			Description: "version",
+			Type:        model.CONST,
+		}
+	}
+	settingsMap := map[string][]model.SettingItem{
+		"2.0.0": {
+			{
+				Key:         "title",
+				Value:       "Alist",
+				Description: "title",
+				Type:        model.PUBLIC,
+			},
+			{
+				Key:         "password",
+				Value:       "alist",
+				Description: "password",
+				Type:        model.PRIVATE,
+			},
+			{
+				Key:         "version",
+				Value:       "2.0.0",
+				Description: "version",
+				Type:        model.CONST,
+			},
+			{
+				Key:         "logo",
+				Value:       "",
+				Description: "logo",
+				Type:        model.PUBLIC,
+			},
+		},
+	}
+	for k, v := range settingsMap {
+		if utils.VersionCompare(k, version.Value) > 0 {
+			log.Infof("writing [v%s] settings", k)
+			err = model.SaveSettings(v)
+			if err != nil {
+				log.Fatalf("save settings error")
+			}
+		}
+	}
+}