add migration logic for 0.2.x (#8)

build/scripts/migration/service.d/user-service/migration.list

@@ -1,2 +1,2 @@
-LEGACY_WITHOUT_VERSION v0.3.6-alpha2
-v0.3.5 v0.3.6-alpha2
+LEGACY_WITHOUT_VERSION v0.3.6-alpha3
+v0.3.5 v0.3.6-alpha3

build/sysroot/etc/casaos/user-service.conf.sample

@@ -5,5 +5,5 @@ RuntimePath=/var/run/casaos
 LogPath = /var/log/casaos
 LogSaveName = user-service
 LogFileExt = log
-DBPath     = /var/lib/casaos
+DBPath     = /var/lib/casaos/db
 UserDataPath = /var/lib/casaos

cmd/migration-tool/main.go

@@ -40,19 +40,20 @@ func main() {
 	}
 
 	if !*forceFlag {
-		serviceEnabled, err := systemctl.IsServiceEnabled(userServiceName)
+		isRunning, err := systemctl.IsServiceRunning(userServiceName)
 		if err != nil {
-			_logger.Error("Failed to check if %s is enabled", userServiceName)
+			_logger.Error("Failed to check if %s is running", userServiceName)
 			panic(err)
 		}
 
-		if serviceEnabled {
-			_logger.Info("%s is already enabled. If migration is still needed, try with -f.", userServiceName)
+		if isRunning {
+			_logger.Info("%s is running. If migration is still needed, try with -f.", userServiceName)
 			os.Exit(1)
 		}
 	}
 
 	migrationTools := []interfaces.MigrationTool{
+		NewMigrationToolFor032AndOlder(),
 		NewMigrationToolFor033_034_035(),
 	}
 
@@ -85,6 +86,6 @@ func main() {
 	}
 
 	if err := selectedMigrationTool.PostMigrate(); err != nil {
-		panic(err)
+		_logger.Error("Migration succeeded, but post-migration failed: %s", err)
 	}
 }

cmd/migration-tool/migration_032_and_older.go

@@ -0,0 +1,191 @@
+package main
+
+import (
+	"os"
+	"path/filepath"
+	"strconv"
+	"strings"
+	"time"
+
+	interfaces "github.com/IceWhaleTech/CasaOS-Common"
+	"github.com/IceWhaleTech/CasaOS-Common/utils/version"
+	"github.com/IceWhaleTech/CasaOS-UserService/pkg/config"
+	"github.com/IceWhaleTech/CasaOS-UserService/pkg/sqlite"
+	"github.com/IceWhaleTech/CasaOS-UserService/pkg/utils/encryption"
+	"github.com/IceWhaleTech/CasaOS-UserService/pkg/utils/file"
+	"github.com/IceWhaleTech/CasaOS-UserService/service"
+	"github.com/IceWhaleTech/CasaOS-UserService/service/model"
+	"gopkg.in/ini.v1"
+)
+
+type migrationTool1 struct{}
+
+func (u *migrationTool1) IsMigrationNeeded() (bool, error) {
+	if _, err := os.Stat(version.LegacyCasaOSConfigFilePath); err != nil {
+		_logger.Info("`%s` not found, migration is not needed.", version.LegacyCasaOSConfigFilePath)
+		return false, nil
+	}
+
+	majorVersion, minorVersion, patchVersion, err := version.DetectLegacyVersion()
+	if err != nil {
+		if err == version.ErrLegacyVersionNotFound {
+			return false, nil
+		}
+
+		return false, err
+	}
+
+	if majorVersion != 0 {
+		return false, nil
+	}
+
+	if minorVersion == 2 {
+		_logger.Info("Migration is needed for a CasaOS version 0.2.x...")
+		return true, nil
+	}
+
+	if minorVersion == 3 && patchVersion < 2 {
+		_logger.Info("Migration is needed for a CasaOS version between 0.3.0 and 0.3.2...")
+		return true, nil
+	}
+
+	return false, nil
+}
+
+func (u *migrationTool1) PreMigrate() error {
+	_logger.Info("Copying %s to %s if it doesn't exist...", userServiceConfigSampleFilePath, config.UserServiceConfigFilePath)
+	if err := file.CopySingleFile(userServiceConfigSampleFilePath, config.UserServiceConfigFilePath, "skip"); err != nil {
+		return err
+	}
+
+	extension := "." + time.Now().Format("20060102") + ".bak"
+
+	_logger.Info("Creating a backup %s if it doesn't exist...", version.LegacyCasaOSConfigFilePath+extension)
+	return file.CopySingleFile(version.LegacyCasaOSConfigFilePath, version.LegacyCasaOSConfigFilePath+extension, "skip")
+}
+
+func (u *migrationTool1) Migrate() error {
+	_logger.Info("Loading legacy %s...", version.LegacyCasaOSConfigFilePath)
+	legacyConfigFile, err := ini.Load(version.LegacyCasaOSConfigFilePath)
+	if err != nil {
+		return err
+	}
+
+	migrateConfigurationFile1(legacyConfigFile)
+
+	return migrateUser1(legacyConfigFile)
+}
+
+func (u *migrationTool1) PostMigrate() error {
+	_logger.Info("Deleting legacy `user` section in %s...", version.LegacyCasaOSConfigFilePath)
+
+	legacyConfigFile, err := ini.Load(version.LegacyCasaOSConfigFilePath)
+	if err != nil {
+		return err
+	}
+
+	legacyConfigFile.DeleteSection("user")
+
+	return legacyConfigFile.SaveTo(version.LegacyCasaOSConfigFilePath)
+}
+
+func NewMigrationToolFor032AndOlder() interfaces.MigrationTool {
+	return &migrationTool1{}
+}
+
+func migrateConfigurationFile1(legacyConfigFile *ini.File) {
+	_logger.Info("Updating %s with settings from legacy configuration...", config.UserServiceConfigFilePath)
+	config.InitSetup(config.UserServiceConfigFilePath)
+
+	// LogPath
+	if logPath, err := legacyConfigFile.Section("app").GetKey("LogPath"); err == nil {
+		_logger.Info("[app] LogPath = %s", logPath.Value())
+		config.AppInfo.LogPath = logPath.Value()
+	}
+
+	if logPath, err := legacyConfigFile.Section("app").GetKey("LogSavePath"); err == nil {
+		_logger.Info("[app] LogSavePath = %s", logPath.Value())
+		config.AppInfo.LogPath = logPath.Value()
+	}
+
+	// LogFileExt
+	if logFileExt, err := legacyConfigFile.Section("app").GetKey("LogFileExt"); err == nil {
+		_logger.Info("[app] LogFileExt = %s", logFileExt.Value())
+		config.AppInfo.LogFileExt = logFileExt.Value()
+	}
+
+	// UserDataPath
+	if userDataPath, err := legacyConfigFile.Section("app").GetKey("UserDataPath"); err == nil {
+		_logger.Info("[app] UserDataPath = %s", userDataPath.Value())
+		config.AppInfo.UserDataPath = userDataPath.Value()
+	}
+
+	_logger.Info("Saving %s...", config.UserServiceConfigFilePath)
+	config.SaveSetup(config.UserServiceConfigFilePath)
+}
+
+func migrateUser1(legacyConfigFile *ini.File) error {
+	_logger.Info("Migrating user from configuration file to database...")
+
+	user := model.UserDBModel{Role: "admin"}
+
+	// UserName
+	if userName, err := legacyConfigFile.Section("user").GetKey("UserName"); err == nil {
+		_logger.Info("[user] UserName = %s", userName.Value())
+		user.Username = userName.Value()
+	}
+
+	// Email
+	if userEmail, err := legacyConfigFile.Section("user").GetKey("Email"); err == nil {
+		_logger.Info("[user] Email = %s", userEmail.Value())
+		user.Email = userEmail.Value()
+	}
+
+	// NickName
+	if userNickName, err := legacyConfigFile.Section("user").GetKey("NickName"); err == nil {
+		_logger.Info("[user] NickName = %s", userNickName.Value())
+		user.Nickname = userNickName.Value()
+	}
+
+	// Password
+	if userPassword, err := legacyConfigFile.Section("user").GetKey("PWD"); err == nil {
+		_logger.Info("[user] Password = %s", strings.Repeat("*", len(userPassword.Value())))
+		user.Password = encryption.GetMD5ByStr(userPassword.Value())
+	}
+
+	newDB := sqlite.GetDb(config.AppInfo.DBPath)
+	userService := service.NewUserService(newDB)
+
+	if len(user.Username) == 0 {
+		_logger.Info("No user found in legacy configuration file. Skipping...")
+		return nil
+	}
+
+	if userService.GetUserInfoByUserName(user.Username).Id > 0 {
+		_logger.Info("User `%s` already exists in user database at %s. Skipping...", user.Username, config.AppInfo.DBPath)
+		return nil
+	}
+
+	_logger.Info("Creating user %s in database at %s...", user.Username, config.AppInfo.DBPath)
+	user = userService.CreateUser(user)
+	if user.Id > 0 {
+		userPath := config.AppInfo.UserDataPath + "/" + strconv.Itoa(user.Id)
+		_logger.Info("Creating user data path: %s", userPath)
+		if err := file.MkDir(userPath); err != nil {
+			return err
+		}
+
+		if legacyProjectPath, err := legacyConfigFile.Section("app").GetKey("ProjectPath"); err == nil {
+			appOrderJSONFile := filepath.Join(legacyProjectPath.Value(), "app_order.json")
+
+			if _, err := os.Stat(appOrderJSONFile); err == nil {
+				_logger.Info("Moving %s to %s...", appOrderJSONFile, userPath)
+				if err := os.Rename(appOrderJSONFile, filepath.Join(userPath, "app_order.json")); err != nil {
+					return err
+				}
+			}
+		}
+	}
+
+	return nil
+}