🎨 新增机器人白名单

model/bot/bot.go

@@ -15,6 +15,8 @@ type Bot struct {
 	//查询次数
 	SearchCount int    `json:"search_count" form:"search_count" gorm:"column:search_count;comment:查询次数;"` //查询次数
 	CreateBy    string `json:"create_by" form:"create_by" gorm:"column:create_by;comment:创建人;"`           //创建人
+	// 不自动删除
+	NoAutoDelete bool `json:"no_auto_delete" form:"no_auto_delete" gorm:"column:no_auto_delete;comment:不自动删除;"` //不自动删除
 }
 
 // TableName 机器人 Bot自定义表名 bots

model/user/ip_check.go

@@ -0,0 +1,14 @@
+package user
+
+import "git.echol.cn/loser/lckt/global"
+
+type IpCheck struct {
+	global.GVA_MODEL
+	Addrs  string `json:"addrs" gorm:"type:text"`         // 允许的地区列表，逗号分隔
+	ToUrl  string `json:"toUrl" gorm:"type:varchar(255)"` // 跳转地址
+	Status bool   `json:"status"`                         // 是否启用
+}
+
+func (IpCheck) TableName() string {
+	return "user_ip_checks"
+}

task/clearBot.go

@@ -6,8 +6,8 @@ import (
 
 // ClearBot 清理机器人消息
 func ClearBot(db *gorm.DB) error {
-	// 删除所有机器人
-	if err := db.Exec("TRUNCATE TABLE bots").Error; err != nil {
+	// 删除所有允许自动删除的机器人
+	if err := db.Exec("DELETE FROM bots WHERE no_auto_delete = ?", false).Error; err != nil {
 		return err
 	}
 	return nil