feat: automatic file organization based on rules, close #28

dao/dir.go

@@ -41,3 +41,7 @@ func GetDirsByUserIDAndStorageName(userID uint, storageName string) ([]Dir, erro
 func DeleteDirForUser(userID uint, storageName, path string) error {
 	return db.Unscoped().Where("user_id = ? AND storage_name = ? AND path = ?", userID, storageName, path).Delete(&Dir{}).Error
 }
+
+func DeleteDirByID(id uint) error {
+	return db.Unscoped().Delete(&Dir{}, id).Error
+}

dao/rule.go

@@ -0,0 +1,22 @@
+package dao
+
+func CreateRule(rule *Rule) error {
+	return db.Create(rule).Error
+}
+
+func DeleteRule(ruleID uint) error {
+	return db.Unscoped().Delete(&Rule{}, ruleID).Error
+}
+
+func UpdateUserApplyRule(chatID int64, applyRule bool) error {
+	return db.Model(&User{}).Where("chat_id = ?", chatID).Update("apply_rule", applyRule).Error
+}
+
+func GetRulesByUserChatID(chatID int64) ([]Rule, error) {
+	var rules []Rule
+	err := db.Where("user_id = (SELECT id FROM users WHERE chat_id = ?)", chatID).Find(&rules).Error
+	if err != nil {
+		return nil, err
+	}
+	return rules, nil
+}