mirror of https://github.com/Xhofe/alist
150 lines
3.5 KiB
Go
150 lines
3.5 KiB
Go
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{},&model.Meta{})
|
|
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.Errorf("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")
|
|
}
|
|
}
|
|
}
|
|
}
|