mirror of
https://github.com/woodchen-ink/Q58Bot.git
synced 2025-07-18 05:42:06 +08:00
226 lines
7.3 KiB
Go
226 lines
7.3 KiB
Go
package service
|
||
|
||
// 消息处理辅助函数
|
||
import (
|
||
"fmt"
|
||
"log"
|
||
"strings"
|
||
"time"
|
||
|
||
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
|
||
"github.com/woodchen-ink/Q58Bot/core"
|
||
)
|
||
|
||
// deleteMessageAfterDelay 根据指定延迟删除消息。
|
||
func deleteMessageAfterDelay(bot *tgbotapi.BotAPI, chatID int64, messageID int, delay time.Duration) {
|
||
// 让线程暂停指定的延迟时间。
|
||
time.Sleep(delay)
|
||
|
||
// 创建一个删除消息的请求。
|
||
deleteMsg := tgbotapi.NewDeleteMessage(chatID, messageID)
|
||
|
||
// 尝试发送删除消息的请求，并检查是否有错误发生。
|
||
// 注意: 错误情况下只是记录错误，不进行其他操作。
|
||
_, err := bot.Request(deleteMsg)
|
||
if err != nil {
|
||
log.Printf("删除消息失败: %v", err)
|
||
}
|
||
}
|
||
|
||
// SendLongMessage 如有必要，可将长消息拆分为多条消息来发送
|
||
func SendLongMessage(bot *tgbotapi.BotAPI, chatID int64, prefix string, items []string) error {
|
||
const maxMessageLength = 4000 // Leave some room for Telegram's message limit
|
||
|
||
message := prefix + "\n"
|
||
for i, item := range items {
|
||
newLine := fmt.Sprintf("%d. %s\n", i+1, item)
|
||
if len(message)+len(newLine) > maxMessageLength {
|
||
msg := tgbotapi.NewMessage(chatID, message)
|
||
_, err := bot.Send(msg)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
message = ""
|
||
}
|
||
message += newLine
|
||
}
|
||
|
||
if message != "" {
|
||
msg := tgbotapi.NewMessage(chatID, message)
|
||
_, err := bot.Send(msg)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// HandleKeywordCommand 处理关键词命令
|
||
func HandleKeywordCommand(bot *tgbotapi.BotAPI, message *tgbotapi.Message, command string, args string, db *core.Database) {
|
||
switch command {
|
||
case "list":
|
||
keywords, err := db.GetAllKeywords()
|
||
if err != nil {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, "获取关键词列表时发生错误。"))
|
||
return
|
||
}
|
||
if len(keywords) == 0 {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, "关键词列表为空。"))
|
||
} else {
|
||
SendLongMessage(bot, message.Chat.ID, "当前关键词列表：", keywords)
|
||
}
|
||
case "add":
|
||
if args != "" {
|
||
keyword := args
|
||
exists, err := db.KeywordExists(keyword)
|
||
if err != nil {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, "检查关键词时发生错误。"))
|
||
return
|
||
}
|
||
if !exists {
|
||
err = db.AddKeyword(keyword)
|
||
if err != nil {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, "添加关键词时发生错误。"))
|
||
} else {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("关键词 '%s' 已添加。", keyword)))
|
||
}
|
||
} else {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("关键词 '%s' 已存在。", keyword)))
|
||
}
|
||
}
|
||
case "delete":
|
||
if args != "" {
|
||
keyword := args
|
||
err := db.RemoveKeyword(keyword)
|
||
if err != nil {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("删除关键词 '%s' 时发生错误: %v", keyword, err)))
|
||
return
|
||
}
|
||
|
||
// 检查关键词是否仍然存在
|
||
exists, err := db.KeywordExists(keyword)
|
||
if err != nil {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("检查关键词 '%s' 是否存在时发生错误: %v", keyword, err)))
|
||
return
|
||
}
|
||
|
||
if !exists {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("关键词 '%s' 已成功删除。", keyword)))
|
||
} else {
|
||
similarKeywords, err := db.SearchKeywords(keyword)
|
||
if err != nil {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, "搜索关键词时发生错误。"))
|
||
return
|
||
}
|
||
if len(similarKeywords) > 0 {
|
||
SendLongMessage(bot, message.Chat.ID, fmt.Sprintf("未能删除关键词 '%s'。\n\n以下是相似的关键词：", keyword), similarKeywords)
|
||
} else {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("未能删除关键词 '%s'，且未找到相似的关键词。", keyword)))
|
||
}
|
||
}
|
||
} else {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, "请提供要删除的关键词。"))
|
||
}
|
||
case "deletecontaining":
|
||
if args != "" {
|
||
substring := args
|
||
removedKeywords, err := db.RemoveKeywordsContaining(substring)
|
||
if err != nil {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, "删除关键词时发生错误。"))
|
||
return
|
||
}
|
||
if len(removedKeywords) > 0 {
|
||
SendLongMessage(bot, message.Chat.ID, fmt.Sprintf("已删除包含 '%s' 的以下关键词：", substring), removedKeywords)
|
||
} else {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("没有找到包含 '%s' 的关键词。", substring)))
|
||
}
|
||
}
|
||
default:
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, "无效的命令或参数。"))
|
||
}
|
||
}
|
||
|
||
func HandleWhitelistCommand(bot *tgbotapi.BotAPI, message *tgbotapi.Message, command string, args string, db *core.Database) {
|
||
switch command {
|
||
case "listwhite":
|
||
whitelist, err := db.GetAllWhitelist()
|
||
if err != nil {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("获取白名单时发生错误: %v", err)))
|
||
return
|
||
}
|
||
if len(whitelist) == 0 {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, "白名单为空。"))
|
||
} else {
|
||
SendLongMessage(bot, message.Chat.ID, "白名单域名列表：", whitelist)
|
||
}
|
||
|
||
case "addwhite":
|
||
if args == "" {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, "请提供要添加的域名。"))
|
||
return
|
||
}
|
||
domain := strings.ToLower(args)
|
||
exists, err := db.WhitelistExists(domain)
|
||
if err != nil {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("检查白名单时发生错误: %v", err)))
|
||
return
|
||
}
|
||
if exists {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("域名 '%s' 已在白名单中。", domain)))
|
||
return
|
||
}
|
||
err = db.AddWhitelist(domain)
|
||
if err != nil {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("添加到白名单时发生错误: %v", err)))
|
||
return
|
||
}
|
||
// 再次检查以确保添加成功
|
||
exists, err = db.WhitelistExists(domain)
|
||
if err != nil {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("验证添加操作时发生错误: %v", err)))
|
||
return
|
||
}
|
||
if exists {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("域名 '%s' 已成功添加到白名单。", domain)))
|
||
} else {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("未能添加域名 '%s' 到白名单。", domain)))
|
||
}
|
||
|
||
case "delwhite":
|
||
if args == "" {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, "请提供要删除的域名。"))
|
||
return
|
||
}
|
||
domain := strings.ToLower(args)
|
||
exists, err := db.WhitelistExists(domain)
|
||
if err != nil {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("检查白名单时发生错误: %v", err)))
|
||
return
|
||
}
|
||
if !exists {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("域名 '%s' 不在白名单中。", domain)))
|
||
return
|
||
}
|
||
err = db.RemoveWhitelist(domain)
|
||
if err != nil {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("从白名单删除时发生错误: %v", err)))
|
||
return
|
||
}
|
||
// 再次检查以确保删除成功
|
||
exists, err = db.WhitelistExists(domain)
|
||
if err != nil {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("验证删除操作时发生错误: %v", err)))
|
||
return
|
||
}
|
||
if !exists {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("域名 '%s' 已成功从白名单中删除。", domain)))
|
||
} else {
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, fmt.Sprintf("未能从白名单中删除域名 '%s'。", domain)))
|
||
}
|
||
|
||
default:
|
||
bot.Send(tgbotapi.NewMessage(message.Chat.ID, "无效的命令或参数。"))
|
||
}
|
||
}
|