70ba4d5e3d
* Add registration route * add route registration * remove ui * 0.3.6 build scaffold (#473) * wip * wip * wip * Merge Branches * Update helper.sh * add goreleaser * fix migration not working for LEGACY_WITHOUT_VERSION (#479) * wip * wip * add error handling to migration script * add migration tool * Adjusting shell script location * update disk user authentication method * delayed port modification * Update system.go Co-authored-by: Tiger Wang (王豫) <tigerwang@outlook.com>
117 lines
2.3 KiB
Go
117 lines
2.3 KiB
Go
/*
|
|
* @Author: LinkLeong link@icewhale.org
|
|
* @Date: 2022-08-23 18:09:11
|
|
* @LastEditors: LinkLeong
|
|
* @LastEditTime: 2022-08-31 14:17:51
|
|
* @FilePath: /CasaOS/cmd/migration-tool/main.go
|
|
* @Description:
|
|
* @Website: https://www.casaos.io
|
|
* Copyright (c) 2022 by icewhale, All Rights Reserved.
|
|
*/
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
|
|
interfaces "github.com/IceWhaleTech/CasaOS-Common"
|
|
"github.com/IceWhaleTech/CasaOS-Common/utils/systemctl"
|
|
"github.com/IceWhaleTech/CasaOS-Gateway/common"
|
|
"github.com/IceWhaleTech/CasaOS/pkg/config"
|
|
"github.com/IceWhaleTech/CasaOS/pkg/sqlite"
|
|
"github.com/IceWhaleTech/CasaOS/service"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
const (
|
|
casaosServiceName = "casaos.service"
|
|
)
|
|
|
|
var _logger *Logger
|
|
var sqliteDB *gorm.DB
|
|
|
|
var configFlag = ""
|
|
var dbFlag = ""
|
|
|
|
func init() {
|
|
config.InitSetup(configFlag)
|
|
config.UpdateSetup()
|
|
|
|
if len(dbFlag) == 0 {
|
|
dbFlag = config.AppInfo.DBPath + "/db"
|
|
}
|
|
|
|
sqliteDB = sqlite.GetDb(dbFlag)
|
|
//gredis.GetRedisConn(config.RedisInfo),
|
|
|
|
service.MyService = service.NewService(sqliteDB, "")
|
|
}
|
|
func main() {
|
|
versionFlag := flag.Bool("v", false, "version")
|
|
debugFlag := flag.Bool("d", true, "debug")
|
|
forceFlag := flag.Bool("f", true, "force")
|
|
flag.Parse()
|
|
_logger = NewLogger()
|
|
if *versionFlag {
|
|
fmt.Println(common.Version)
|
|
os.Exit(0)
|
|
}
|
|
|
|
if os.Getuid() != 0 {
|
|
os.Exit(1)
|
|
}
|
|
|
|
if *debugFlag {
|
|
_logger.DebugMode = true
|
|
}
|
|
|
|
if !*forceFlag {
|
|
serviceEnabled, err := systemctl.IsServiceEnabled(casaosServiceName)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if serviceEnabled {
|
|
_logger.Info("%s is already enabled. If migration is still needed, try with -f.", casaosServiceName)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
migrationTools := []interfaces.MigrationTool{
|
|
NewMigrationToolFor_035(),
|
|
}
|
|
|
|
var selectedMigrationTool interfaces.MigrationTool
|
|
|
|
// look for the right migration tool matching current version
|
|
for _, tool := range migrationTools {
|
|
migrationNeeded, err := tool.IsMigrationNeeded()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if migrationNeeded {
|
|
selectedMigrationTool = tool
|
|
break
|
|
}
|
|
}
|
|
|
|
if selectedMigrationTool == nil {
|
|
_logger.Error("selectedMigrationTool is null")
|
|
return
|
|
}
|
|
|
|
if err := selectedMigrationTool.PreMigrate(); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if err := selectedMigrationTool.Migrate(); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
selectedMigrationTool.PostMigrate()
|
|
_logger.Info("casaos migration ok")
|
|
//panic(err)
|
|
|
|
}
|