forked from carrydela/mygoTgChanBot
134 lines
3.2 KiB
Go
134 lines
3.2 KiB
Go
package telegram
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
|
|
tele "gopkg.in/telebot.v3"
|
|
)
|
|
|
|
func (b *Bot) handleEntryDel(c tele.Context) error {
|
|
id := strings.TrimSpace(c.Message().Payload)
|
|
if id == "" {
|
|
return c.Reply("用法: /del <ID>")
|
|
}
|
|
|
|
entry, err := b.storage.GetEntry(id)
|
|
if err != nil {
|
|
return c.Reply(fmt.Sprintf("❌ %v", err))
|
|
}
|
|
|
|
// 尝试删除频道消息
|
|
msgDeleted := false
|
|
if msgID := parseMessageIDFromLink(entry.Link); msgID != 0 {
|
|
channel := &tele.Chat{ID: b.cfg.Channel.ID}
|
|
msg := &tele.Message{ID: msgID, Chat: channel}
|
|
if err := b.bot.Delete(msg); err == nil {
|
|
msgDeleted = true
|
|
}
|
|
}
|
|
|
|
if err := b.storage.DeleteEntry(id); err != nil {
|
|
return c.Reply(fmt.Sprintf("❌ 删除失败: %v", err))
|
|
}
|
|
|
|
b.toc.TriggerUpdate()
|
|
|
|
if msgDeleted {
|
|
return c.Reply(fmt.Sprintf("✅ 已删除: [%s] %s\n📨 频道消息已同步删除", entry.Category, entry.Title))
|
|
}
|
|
return c.Reply(fmt.Sprintf("✅ 已删除: [%s] %s", entry.Category, entry.Title))
|
|
}
|
|
|
|
func (b *Bot) handleEntryEdit(c tele.Context) error {
|
|
args := strings.SplitN(c.Message().Payload, " ", 2)
|
|
if len(args) < 2 {
|
|
return c.Reply("用法: /edit <ID> <新标题>")
|
|
}
|
|
|
|
id := args[0]
|
|
newTitle := strings.TrimSpace(args[1])
|
|
|
|
if err := b.storage.UpdateEntryTitle(id, newTitle); err != nil {
|
|
return c.Reply(fmt.Sprintf("❌ %v", err))
|
|
}
|
|
|
|
b.toc.TriggerUpdate()
|
|
return c.Reply(fmt.Sprintf("✅ 标题已更新为: %s", newTitle))
|
|
}
|
|
|
|
func (b *Bot) handleEntryMove(c tele.Context) error {
|
|
args := strings.Fields(c.Message().Payload)
|
|
if len(args) < 2 {
|
|
return c.Reply("用法: /move <ID> <新分类>")
|
|
}
|
|
|
|
id := args[0]
|
|
newCategory := args[1]
|
|
|
|
if !b.storage.CategoryExists(newCategory) {
|
|
return c.Reply(fmt.Sprintf("❌ 分类 [%s] 不存在", newCategory))
|
|
}
|
|
|
|
if err := b.storage.UpdateEntryCategory(id, newCategory); err != nil {
|
|
return c.Reply(fmt.Sprintf("❌ %v", err))
|
|
}
|
|
|
|
b.toc.TriggerUpdate()
|
|
return c.Reply(fmt.Sprintf("✅ 已移动到分类: %s", newCategory))
|
|
}
|
|
|
|
func (b *Bot) handleEntryList(c tele.Context) error {
|
|
category := strings.TrimSpace(c.Message().Payload)
|
|
|
|
entries, err := b.storage.ListEntries(category)
|
|
if err != nil {
|
|
return c.Reply(fmt.Sprintf("❌ %v", err))
|
|
}
|
|
|
|
if len(entries) == 0 {
|
|
if category != "" {
|
|
return c.Reply(fmt.Sprintf("📭 分类 [%s] 暂无条目", category))
|
|
}
|
|
return c.Reply("📭 暂无条目")
|
|
}
|
|
|
|
var sb strings.Builder
|
|
if category != "" {
|
|
sb.WriteString(fmt.Sprintf("📋 分类 [%s] 的条目:\n\n", category))
|
|
} else {
|
|
sb.WriteString("📋 所有条目:\n\n")
|
|
}
|
|
|
|
currentCat := ""
|
|
for _, e := range entries {
|
|
if category == "" && e.Category != currentCat {
|
|
currentCat = e.Category
|
|
sb.WriteString(fmt.Sprintf("\n【%s】\n", currentCat))
|
|
}
|
|
sb.WriteString(fmt.Sprintf("• [%s] %s\n", e.ID, e.Title))
|
|
}
|
|
|
|
return c.Reply(sb.String())
|
|
}
|
|
|
|
func (b *Bot) handleRefresh(c tele.Context) error {
|
|
b.toc.TriggerUpdate()
|
|
return c.Reply("🔄 目录刷新已触发")
|
|
}
|
|
|
|
// parseMessageIDFromLink 从链接中解析消息ID
|
|
// 支持格式: https://t.me/c/123456/789 或 https://t.me/username/789
|
|
func parseMessageIDFromLink(link string) int {
|
|
parts := strings.Split(link, "/")
|
|
if len(parts) < 1 {
|
|
return 0
|
|
}
|
|
msgID, err := strconv.Atoi(parts[len(parts)-1])
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
return msgID
|
|
}
|