bootstrap

This commit is contained in:
微凉
2021-10-27 22:45:36 +08:00
parent 9644cc98c3
commit 55f683b12d
18 changed files with 311 additions and 133 deletions

View File

@ -2,7 +2,6 @@ package model
import (
"github.com/Xhofe/alist/conf"
log "github.com/sirupsen/logrus"
)
type Account struct {
@ -22,6 +21,7 @@ type Account struct {
var accountsMap = map[string]Account{}
// SaveAccount save account to database
func SaveAccount(account Account) error {
if err := conf.DB.Save(account).Error; err != nil {
return err
@ -80,14 +80,3 @@ func GetAccounts() []*Account {
return accounts
}
func initAccounts() {
log.Infof("init accounts...")
var accounts []Account
if err := conf.DB.Find(&accounts).Error; err != nil {
log.Fatalf("failed sync init accounts")
}
for _, account := range accounts {
RegisterAccount(account)
}
log.Debugf("accounts:%+v", accountsMap)
}

View File

@ -1,75 +0,0 @@
package model
import (
"fmt"
"github.com/Xhofe/alist/conf"
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(&SettingItem{},&Account{})
if err != nil {
log.Fatalf("failed to auto migrate")
}
// TODO init filetype
initAccounts()
initSettings()
}

View File

@ -2,8 +2,6 @@ package model
import (
"github.com/Xhofe/alist/conf"
"github.com/Xhofe/alist/utils"
log "github.com/sirupsen/logrus"
)
const (
@ -39,47 +37,3 @@ func GetSettingByKey(key string) (*SettingItem, error) {
return &items, nil
}
func initSettings() {
log.Infof("init settings...")
version, err := GetSettingByKey("version")
if err != nil {
log.Debugf("first run")
version = &SettingItem{
Key: "version",
Value: "0.0.0",
Description: "version",
Type: CONST,
}
}
settingsMap := map[string][]SettingItem{
"2.0.0": {
{
Key: "title",
Value: "Alist",
Description: "title",
Type: PUBLIC,
},
{
Key: "password",
Value: "alist",
Description: "password",
Type: PRIVATE,
},
{
Key: "version",
Value: "2.0.0",
Description: "version",
Type: CONST,
},
},
}
for k, v := range settingsMap {
if utils.VersionCompare(k, version.Value) > 0 {
log.Infof("writing [v%s] settings",k)
err = SaveSettings(v)
if err != nil {
log.Fatalf("save settings error")
}
}
}
}