feat: (WIP) migrate storage configuration to user-specific models and remove deprecated storage loading

dao/db.go

@@ -36,9 +36,117 @@ func Init() {
 		os.Exit(1)
 	}
 	logger.L.Debug("Database connected")
-	db.AutoMigrate(&types.ReceivedFile{}, &types.User{})
+	if err := db.AutoMigrate(&types.ReceivedFile{}, &types.User{}, &types.StorageModel{}); err != nil {
+		logger.L.Fatal("迁移数据库失败, 如果您从旧版本升级, 建议手动删除数据库文件后重试: ", err)
+	}
 
 	for _, admin := range config.Cfg.Telegram.Admins {
 		CreateUser(int64(admin))
 	}
+
+	logger.L.Infof("Migrating config storages to users")
+	storageCfg := config.Cfg.Storage
+
+	allUsers, err := GetAllUsers()
+	if err != nil {
+		logger.L.Fatalf("Failed to get all users: %v", err)
+	} else {
+		for _, user := range allUsers {
+			found := false
+			for _, admin := range config.Cfg.Telegram.Admins {
+				if user.ChatID == int64(admin) {
+					found = true
+					break
+				}
+			}
+			if !found {
+				logger.L.Debugf("Deleting user %d", user.ChatID)
+				if err := DeleteUser(&user); err != nil {
+					logger.L.Fatalf("Failed to delete user %d: %v", user.ChatID, err)
+				}
+			}
+		}
+	}
+	// TODO: refactor this
+	for _, admin := range config.Cfg.Telegram.Admins {
+		user, err := GetUserByChatID(int64(admin))
+		if err != nil {
+			logger.L.Fatalf("Failed to get user by chat ID %d: %v", admin, err)
+			continue
+		}
+		if len(user.Storages) > 0 {
+			logger.L.Debugf("User %d already has storages", admin)
+			continue
+		}
+		if storageCfg.Alist.Enable {
+			alistStorage := &types.StorageModel{
+				Type:   string(types.StorageTypeAlist),
+				Active: true,
+				Config: storageCfg.Alist.ToJSON(),
+			}
+			hash := alistStorage.GenHash()
+			alistStorage.Hash = hash
+			if storagedb, err := GetStorageByHash(hash); err == nil {
+				logger.L.Debugf("Alist storage already exists")
+				user.Storages = append(user.Storages, storagedb)
+			} else {
+				id, err := CreateStorage(alistStorage)
+				if err != nil {
+					logger.L.Fatalf("Failed to create storage: %v", err)
+				} else {
+					storagedb := &types.StorageModel{}
+					storagedb.ID = id
+					user.Storages = append(user.Storages, storagedb)
+				}
+			}
+		}
+		if storageCfg.Local.Enable {
+			localStorage := &types.StorageModel{
+				Type:   string(types.StorageTypeLocal),
+				Active: true,
+				Config: storageCfg.Local.ToJSON(),
+			}
+			hash := localStorage.GenHash()
+			localStorage.Hash = hash
+			if storagedb, err := GetStorageByHash(hash); err == nil {
+				logger.L.Debugf("Local storage already exists")
+				user.Storages = append(user.Storages, storagedb)
+			} else {
+				id, err := CreateStorage(localStorage)
+				if err != nil {
+					logger.L.Fatalf("Failed to create storage: %v", err)
+				} else {
+					storagedb := &types.StorageModel{}
+					storagedb.ID = id
+					user.Storages = append(user.Storages, storagedb)
+				}
+			}
+		}
+		if storageCfg.Webdav.Enable {
+			webdavStorage := &types.StorageModel{
+				Type:   string(types.StorageTypeWebdav),
+				Active: true,
+				Config: storageCfg.Webdav.ToJSON(),
+			}
+			hash := webdavStorage.GenHash()
+			webdavStorage.Hash = hash
+			if storagedb, err := GetStorageByHash(hash); err == nil {
+				logger.L.Debugf("Webdav storage already exists")
+				user.Storages = append(user.Storages, storagedb)
+			} else {
+				id, err := CreateStorage(webdavStorage)
+				if err != nil {
+					logger.L.Fatalf("Failed to create storage: %v", err)
+				} else {
+					storagedb := &types.StorageModel{}
+					storagedb.ID = id
+					user.Storages = append(user.Storages, storagedb)
+				}
+			}
+		}
+		if err := UpdateUser(user); err != nil {
+			logger.L.Fatalf("Failed to update user with storages: %v", err)
+		}
+	}
+	logger.L.Infof("Migration done")
 }

dao/storage.go

@@ -12,15 +12,36 @@ func GetActiveStorages() ([]types.StorageModel, error) {
 	return storageModels, err
 }
 
+func GetStorageByHash(hash string) (*types.StorageModel, error) {
+	var storageModel types.StorageModel
+	err := db.Where("hash = ?", hash).First(&storageModel).Error
+	return &storageModel, err
+}
+
 func GetStorageByID(id uint) (*types.StorageModel, error) {
 	var storageModel types.StorageModel
 	err := db.Preload("Users").First(&storageModel, id).Error
 	return &storageModel, err
 }
 
-func CreateStorage(model *types.StorageModel) error {
-	if model.Name == "" {
-		model.Name = fmt.Sprintf("%s_%d", model.Type, model.ID)
+func CreateStorage(model *types.StorageModel) (uint, error) {
+	if model.Hash == "" {
+		model.Hash = model.GenHash()
 	}
-	return db.Create(model).Error
+	getModel, err := GetStorageByHash(model.Hash)
+	if err == nil {
+		return getModel.ID, nil
+	}
+	tx := db.Create(model)
+	if tx.Error != nil {
+		return 0, tx.Error
+	}
+	if model.Name == "" {
+		model.Name = fmt.Sprintf("%s - %d", model.Type, model.ID)
+		tx = db.Save(model)
+		if tx.Error != nil {
+			return 0, tx.Error
+		}
+	}
+	return model.ID, nil
 }

dao/user.go

@@ -11,6 +11,12 @@ func CreateUser(chatID int64) error {
 	return db.Create(&types.User{ChatID: chatID}).Error
 }
 
+func GetAllUsers() ([]types.User, error) {
+	var users []types.User
+	err := db.Find(&users).Error
+	return users, err
+}
+
 // GetUserByUserID gets a user by their telegram user ID
 //
 // Return with active storages
@@ -29,3 +35,7 @@ func GetUserWithAllStoragesByChatID(chatID int64) (*types.User, error) {
 func UpdateUser(user *types.User) error {
 	return db.Save(user).Error
 }
+
+func DeleteUser(user *types.User) error {
+	return db.Select("Storages").Delete(user).Error
+}