diff --git a/internal/service/telegram_mgo_admin.go b/internal/service/telegram_mgo_admin.go
new file mode 100644
index 0000000..728d50f
--- /dev/null
+++ b/internal/service/telegram_mgo_admin.go
@@ -0,0 +1,159 @@
+package service
+
+import (
+ "context"
+ "encoding/json"
+ "fmt"
+ "sort"
+ "strconv"
+ "strings"
+
+ "github.com/ShukeBta/MediaStationGo/internal/model"
+)
+
+func (s *TelegramBotService) cmdMgoAdminRole(ctx context.Context, args []string) telegramCommandReply {
+ if len(args) < 2 {
+ return telegramCommandReply{Text: "用法：/embyadmin 用户名 on|off"}
+ }
+ user := s.findMgoBotUser(ctx, args[0])
+ if user == nil {
+ return telegramCommandReply{Text: "未找到用户。"}
+ }
+ enable := parseOnOff(args[1])
+ if enable == nil {
+ return telegramCommandReply{Text: "第二个参数请使用 on/off。"}
+ }
+ if !*enable {
+ if first, _ := s.repo.User.FirstAdmin(ctx); first != nil && first.ID == user.ID {
+ return telegramCommandReply{Text: "默认管理员不可降级。"}
+ }
+ }
+ role := "user"
+ if *enable {
+ role = "admin"
+ }
+ if err := s.repo.User.UpdateFields(ctx, user.ID, map[string]any{"role": role}); err != nil {
+ return telegramCommandReply{Text: "更新失败：" + err.Error()}
+ }
+ return telegramCommandReply{Text: fmt.Sprintf("已将 %s 角色设置为 %s。", user.Username, role)}
+}
+
+func (s *TelegramBotService) cmdMgoMediaAccessAll(ctx context.Context, allow bool) telegramCommandReply {
+ users, err := s.repo.User.List(ctx)
+ if err != nil {
+ return telegramCommandReply{Text: "读取用户失败：" + err.Error()}
+ }
+ updated := 0
+ for _, user := range users {
+ if user.Role == "admin" {
+ continue
+ }
+ perm, err := s.repo.Permission.FindByUserID(ctx, user.ID)
+ if err != nil {
+ continue
+ }
+ if perm == nil {
+ perm = DefaultPermissions(user.ID)
+ perm.CanPlayMedia = allow
+ if err := s.repo.Permission.Create(ctx, perm); err != nil {
+ continue
+ }
+ }
+ if err := s.repo.DB.WithContext(ctx).Model(&model.UserPermission{}).
+ Where("user_id = ?", user.ID).
+ Update("can_play_media", allow).Error; err == nil {
+ updated++
+ }
+ }
+ state := "关闭"
+ if allow {
+ state = "开启"
+ }
+ return telegramCommandReply{Text: fmt.Sprintf("已为普通用户%s媒体播放权限：%d 个。", state, updated)}
+}
+
+func (s *TelegramBotService) cmdMgoBotAdmin(ctx context.Context, channel *model.NotifyChannel, args []string, add bool) telegramCommandReply {
+ if channel == nil {
+ return telegramCommandReply{Text: "Telegram 渠道不存在。"}
+ }
+ if len(args) == 0 {
+ return telegramCommandReply{Text: "用法：/proadmin TelegramID 或 /revadmin TelegramID"}
+ }
+ tgID := strings.TrimPrefix(strings.TrimSpace(args[0]), "tg:")
+ if _, err := strconv.ParseInt(tgID, 10, 64); err != nil {
+ return telegramCommandReply{Text: "TelegramID 必须是数字。"}
+ }
+ cfg := s.telegramChannelConfig(channel)
+ ids := telegramConfiguredUserIDs(cfg["admin_user_ids"])
+ seen := make(map[string]bool, len(ids)+1)
+ var next []string
+ for _, id := range ids {
+ if id == tgID {
+ seen[id] = true
+ if add {
+ next = append(next, id)
+ }
+ continue
+ }
+ if id != "" {
+ next = append(next, id)
+ }
+ }
+ if add && !seen[tgID] {
+ next = append(next, tgID)
+ }
+ cfg["admin_user_ids"] = strings.Join(next, ",")
+ raw, _ := json.Marshal(cfg)
+ updated := *channel
+ updated.Config = string(raw)
+ if s.crypto != nil {
+ updated.Config = s.crypto.Encrypt(updated.Config)
+ }
+ if err := s.repo.NotifyChannel.Update(ctx, &updated); err != nil {
+ return telegramCommandReply{Text: "更新管理员列表失败：" + err.Error()}
+ }
+ if add {
+ return telegramCommandReply{Text: "已添加 Bot 管理员：" + tgID + ""}
+ }
+ return telegramCommandReply{Text: "已移除 Bot 管理员：" + tgID + ""}
+}
+
+func (s *TelegramBotService) cmdMgoProtectedUser(ctx context.Context, args []string, protect bool) telegramCommandReply {
+ if len(args) == 0 || strings.EqualFold(args[0], "list") {
+ return s.cmdMgoProtectedUserList(ctx)
+ }
+ user := s.findMgoBotUser(ctx, args[0])
+ if user == nil {
+ return telegramCommandReply{Text: "未找到用户。"}
+ }
+ ids := ProtectedUserIDSet(ctx, s.repo)
+ if protect {
+ ids[user.ID] = struct{}{}
+ if err := SaveProtectedUserIDSet(ctx, s.repo, ids); err != nil {
+ return telegramCommandReply{Text: "保存保护名单失败：" + err.Error()}
+ }
+ return telegramCommandReply{Text: fmt.Sprintf("已加入保护名单：%s。\n该用户不会被 Bot 自动清理、批量禁用或删除。", user.Username)}
+ }
+ delete(ids, user.ID)
+ if err := SaveProtectedUserIDSet(ctx, s.repo, ids); err != nil {
+ return telegramCommandReply{Text: "保存保护名单失败：" + err.Error()}
+ }
+ return telegramCommandReply{Text: fmt.Sprintf("已移出保护名单：%s。", user.Username)}
+}
+
+func (s *TelegramBotService) cmdMgoProtectedUserList(ctx context.Context) telegramCommandReply {
+ ids := ProtectedUserIDSet(ctx, s.repo)
+ if len(ids) == 0 {
+ return telegramCommandReply{Text: "保护名单为空。管理员和默认管理员始终自动保护。"}
+ }
+ names := make([]string, 0, len(ids))
+ for id := range ids {
+ if user, _ := s.repo.User.FindByID(ctx, id); user != nil {
+ names = append(names, user.Username)
+ } else {
+ names = append(names, id+"(用户不存在)")
+ }
+ }
+ sort.Strings(names)
+ return telegramCommandReply{Text: fmt.Sprintf("保护名单：%d 个。\n%s", len(names), telegramInlineCodeList(names))}
+}
diff --git a/internal/service/telegram_mgo_audit.go b/internal/service/telegram_mgo_audit.go
new file mode 100644
index 0000000..b696852
--- /dev/null
+++ b/internal/service/telegram_mgo_audit.go
@@ -0,0 +1,133 @@
+package service
+
+import (
+ "context"
+ "fmt"
+ "strings"
+ "time"
+
+ "github.com/ShukeBta/MediaStationGo/internal/model"
+)
+
+func (s *TelegramBotService) cmdMgoSyncUnbound(ctx context.Context, args []string) telegramCommandReply {
+ var users []model.User
+ if err := s.repo.DB.WithContext(ctx).
+ Where("role <> ?", "admin").
+ Where("NOT EXISTS (SELECT 1 FROM telegram_bindings WHERE telegram_bindings.user_id = users.id AND telegram_bindings.deleted_at IS NULL)").
+ Order("created_at asc").Find(&users).Error; err != nil {
+ return telegramCommandReply{Text: "查询失败：" + err.Error()}
+ }
+ if len(args) >= 2 && strings.EqualFold(args[0], "delete") && strings.EqualFold(args[1], "confirm") {
+ deleted := 0
+ for _, user := range users {
+ if UserIsProtectedAccount(ctx, s.repo, &user) {
+ continue
+ }
+ _ = s.repo.UserDevice.DeleteByUser(ctx, user.ID)
+ if err := s.repo.User.Delete(ctx, user.ID); err == nil {
+ deleted++
+ }
+ }
+ return telegramCommandReply{Text: fmt.Sprintf("已删除未绑定 Bot 的普通用户：%d 个。", deleted)}
+ }
+ if len(users) == 0 {
+ return telegramCommandReply{Text: "没有未绑定 Bot 的普通用户。"}
+ }
+ names := make([]string, 0, minInt(len(users), 20))
+ for i, user := range users {
+ if i >= 20 {
+ break
+ }
+ names = append(names, user.Username)
+ }
+ return telegramCommandReply{Text: fmt.Sprintf("未绑定 Bot 的普通用户：%d 个。\n%s\n\n如需删除：/syncunbound delete confirm", len(users), telegramInlineCodeList(names))}
+}
+
+func (s *TelegramBotService) cmdMgoCheckExpired(ctx context.Context, args []string) telegramCommandReply {
+ now := time.Now()
+ var users []model.User
+ if err := s.repo.DB.WithContext(ctx).Where("expired_at IS NOT NULL AND expired_at < ?", now).Order("expired_at asc").Find(&users).Error; err != nil {
+ return telegramCommandReply{Text: "查询失败：" + err.Error()}
+ }
+ if len(args) >= 2 && strings.EqualFold(args[0], "disable") && strings.EqualFold(args[1], "confirm") {
+ disabled := 0
+ for _, user := range users {
+ if UserIsProtectedAccount(ctx, s.repo, &user) {
+ continue
+ }
+ if err := s.repo.User.UpdateFields(ctx, user.ID, map[string]any{"is_active": false}); err == nil {
+ disabled++
+ }
+ }
+ return telegramCommandReply{Text: fmt.Sprintf("已禁用过期普通用户：%d 个。", disabled)}
+ }
+ if len(users) == 0 {
+ return telegramCommandReply{Text: "没有过期用户。"}
+ }
+ lines := make([]string, 0, minInt(len(users), 20))
+ for i, user := range users {
+ if i >= 20 {
+ break
+ }
+ lines = append(lines, fmt.Sprintf("%s（%s）", user.Username, formatExpiry(user.ExpiredAt)))
+ }
+ return telegramCommandReply{Text: fmt.Sprintf("过期用户：%d 个。\n%s\n\n如需禁用：/check_ex disable confirm", len(users), telegramInlineCodeList(lines))}
+}
+
+func (s *TelegramBotService) cmdMgoScanNames(ctx context.Context) telegramCommandReply {
+ var rows []struct {
+ Username string
+ Count int64
+ }
+ if err := s.repo.DB.WithContext(ctx).Table("users").
+ Select("LOWER(username) AS username, COUNT(*) AS count").
+ Group("LOWER(username)").Having("COUNT(*) > 1").Scan(&rows).Error; err != nil {
+ return telegramCommandReply{Text: "扫描失败：" + err.Error()}
+ }
+ if len(rows) == 0 {
+ return telegramCommandReply{Text: "未发现同名用户记录。"}
+ }
+ var out []string
+ for _, row := range rows {
+ out = append(out, fmt.Sprintf("%s x%d", row.Username, row.Count))
+ }
+ return telegramCommandReply{Text: "同名用户记录\n" + telegramInlineCodeList(out)}
+}
+
+func (s *TelegramBotService) cmdMgoRanks(ctx context.Context, window time.Duration, byDuration bool) telegramCommandReply {
+ since := time.Now().Add(-window)
+ title := "播放次数排行"
+ selectExpr := "COUNT(*) AS score"
+ if byDuration {
+ title = "观影时长排行"
+ selectExpr = "COALESCE(SUM(position_ms), 0) AS score"
+ }
+ q := s.repo.DB.WithContext(ctx).Table("playback_histories").
+ Select("users.username, " + selectExpr).
+ Joins("JOIN users ON users.id = playback_histories.user_id").
+ Group("users.username").
+ Order("score DESC").
+ Limit(10)
+ if window > 0 {
+ q = q.Where("playback_histories.watched_at >= ?", since)
+ }
+ var rows []struct {
+ Username string
+ Score int64
+ }
+ if err := q.Scan(&rows).Error; err != nil {
+ return telegramCommandReply{Text: "排行查询失败：" + err.Error()}
+ }
+ if len(rows) == 0 {
+ return telegramCommandReply{Text: "暂无排行数据。"}
+ }
+ var out []string
+ for i, row := range rows {
+ score := fmt.Sprintf("%d 次", row.Score)
+ if byDuration {
+ score = humanDurationFromMillis(row.Score)
+ }
+ out = append(out, fmt.Sprintf("%d. %s — %s", i+1, row.Username, score))
+ }
+ return telegramCommandReply{Text: "" + title + "\n\n" + strings.Join(out, "\n") + ""}
+}
diff --git a/internal/service/telegram_mgo_backup.go b/internal/service/telegram_mgo_backup.go
new file mode 100644
index 0000000..7ba58c9
--- /dev/null
+++ b/internal/service/telegram_mgo_backup.go
@@ -0,0 +1,49 @@
+package service
+
+import (
+ "context"
+ "fmt"
+ "strings"
+)
+
+func (s *TelegramBotService) cmdMgoBackupDB(ctx context.Context) telegramCommandReply {
+ if s.backup == nil {
+ return telegramCommandReply{Text: "备份服务暂不可用。"}
+ }
+ info, err := s.backup.Create(ctx)
+ if err != nil {
+ return telegramCommandReply{Text: "数据库备份失败：" + err.Error()}
+ }
+ return telegramCommandReply{Text: fmt.Sprintf("数据库备份完成：%s\n大小：%d bytes", info.Filename, info.Size)}
+}
+
+func (s *TelegramBotService) cmdMgoRestoreDB(ctx context.Context, args []string) telegramCommandReply {
+ if s.backup == nil {
+ return telegramCommandReply{Text: "备份服务暂不可用。"}
+ }
+ if len(args) == 0 || strings.EqualFold(args[0], "list") {
+ items, err := s.backup.List()
+ if err != nil {
+ return telegramCommandReply{Text: "读取备份列表失败：" + err.Error()}
+ }
+ if len(items) == 0 {
+ return telegramCommandReply{Text: "暂无数据库备份。可先使用 /backup_db 创建。"}
+ }
+ lines := make([]string, 0, minInt(len(items), 10))
+ for i, item := range items {
+ if i >= 10 {
+ break
+ }
+ lines = append(lines, fmt.Sprintf("%s（%d bytes）", item.Filename, item.Size))
+ }
+ return telegramCommandReply{Text: "可恢复备份：\n" + telegramInlineCodeList(lines) + "\n\n恢复需要确认：/restore_from_db 文件名 confirm"}
+ }
+ if len(args) < 2 || !strings.EqualFold(args[len(args)-1], "confirm") {
+ return telegramCommandReply{Text: "恢复数据库会覆盖当前数据，需要确认：/restore_from_db 文件名 confirm"}
+ }
+ filename := strings.TrimSpace(args[0])
+ if err := s.backup.Restore(ctx, filename); err != nil {
+ return telegramCommandReply{Text: "恢复失败：" + err.Error()}
+ }
+ return telegramCommandReply{Text: "数据库已从备份恢复，请重启 MediaStationGo 后生效。"}
+}
diff --git a/internal/service/telegram_mgo_bulk.go b/internal/service/telegram_mgo_bulk.go
new file mode 100644
index 0000000..e06f554
--- /dev/null
+++ b/internal/service/telegram_mgo_bulk.go
@@ -0,0 +1,94 @@
+package service
+
+import (
+ "context"
+ "fmt"
+ "strconv"
+ "strings"
+
+ "github.com/ShukeBta/MediaStationGo/internal/model"
+)
+
+func (s *TelegramBotService) cmdMgoRenewAll(ctx context.Context, args []string) telegramCommandReply {
+ if len(args) < 2 || !strings.EqualFold(args[len(args)-1], "confirm") {
+ return telegramCommandReply{Text: "批量续期需要确认：/renewall 天数 confirm"}
+ }
+ days, err := strconv.Atoi(args[0])
+ if err != nil || days < 0 {
+ return telegramCommandReply{Text: "天数必须是非负整数，0 表示永久。"}
+ }
+ users, err := s.repo.User.List(ctx)
+ if err != nil {
+ return telegramCommandReply{Text: "读取用户失败：" + err.Error()}
+ }
+ var count int
+ for _, user := range users {
+ if user.Role == "admin" {
+ continue
+ }
+ if err := s.applyRenewal(ctx, user.ID, days); err == nil {
+ count++
+ }
+ }
+ return telegramCommandReply{Text: fmt.Sprintf("批量续期完成：%d 个普通用户。", count)}
+}
+
+func (s *TelegramBotService) cmdMgoBanAll(ctx context.Context, active bool, args []string) telegramCommandReply {
+ if len(args) == 0 || !strings.EqualFold(args[len(args)-1], "confirm") {
+ action := "banall"
+ if active {
+ action = "unbanall"
+ }
+ return telegramCommandReply{Text: fmt.Sprintf("批量操作需要确认：/%s confirm", action)}
+ }
+ users, err := s.repo.User.List(ctx)
+ if err != nil {
+ return telegramCommandReply{Text: "读取用户失败：" + err.Error()}
+ }
+ var count int
+ for _, user := range users {
+ if !active && UserIsProtectedAccount(ctx, s.repo, &user) {
+ continue
+ }
+ if active && user.Role == "admin" {
+ continue
+ }
+ updates := map[string]any{"is_active": active}
+ if active {
+ updates["share_warnings"] = 0
+ updates["last_share_warn_at"] = nil
+ }
+ if err := s.repo.User.UpdateFields(ctx, user.ID, updates); err == nil {
+ _ = s.repo.UserDevice.SetKickedByUser(ctx, user.ID, !active)
+ count++
+ }
+ }
+ if active {
+ return telegramCommandReply{Text: fmt.Sprintf("已解禁普通用户：%d 个。", count)}
+ }
+ return telegramCommandReply{Text: fmt.Sprintf("已禁用普通用户：%d 个。", count)}
+}
+
+func (s *TelegramBotService) cmdMgoCallAll(ctx context.Context, channel *model.NotifyChannel, args []string) telegramCommandReply {
+ message := strings.TrimSpace(strings.Join(args, " "))
+ if message == "" {
+ return telegramCommandReply{Text: "用法：/callall 消息内容"}
+ }
+ if strings.TrimSpace(s.telegramChannelConfig(channel)["bot_token"]) == "" {
+ return telegramCommandReply{Text: "当前 Telegram 渠道未配置 bot_token，无法群发。"}
+ }
+ var bindings []model.TelegramBinding
+ if err := s.repo.DB.WithContext(ctx).Find(&bindings).Error; err != nil {
+ return telegramCommandReply{Text: "读取绑定失败：" + err.Error()}
+ }
+ sent := 0
+ for _, binding := range bindings {
+ if binding.ChatID == 0 {
+ continue
+ }
+ if err := s.reply(ctx, channel, int(binding.ChatID), telegramCommandReply{Text: message}); err == nil {
+ sent++
+ }
+ }
+ return telegramCommandReply{Text: fmt.Sprintf("群发完成：成功发送 %d 个绑定用户。", sent)}
+}
diff --git a/internal/service/telegram_mgo_compat.go b/internal/service/telegram_mgo_compat.go
index bd5a928..46f2188 100644
--- a/internal/service/telegram_mgo_compat.go
+++ b/internal/service/telegram_mgo_compat.go
@@ -2,9 +2,7 @@ package service
import (
"context"
- "encoding/json"
"fmt"
- "sort"
"strconv"
"strings"
"time"
@@ -12,485 +10,6 @@ import (
"github.com/ShukeBta/MediaStationGo/internal/model"
)
-func (s *TelegramBotService) cmdMgoRenewAll(ctx context.Context, args []string) telegramCommandReply {
- if len(args) < 2 || !strings.EqualFold(args[len(args)-1], "confirm") {
- return telegramCommandReply{Text: "批量续期需要确认：/renewall 天数 confirm"}
- }
- days, err := strconv.Atoi(args[0])
- if err != nil || days < 0 {
- return telegramCommandReply{Text: "天数必须是非负整数，0 表示永久。"}
- }
- users, err := s.repo.User.List(ctx)
- if err != nil {
- return telegramCommandReply{Text: "读取用户失败：" + err.Error()}
- }
- var count int
- for _, user := range users {
- if user.Role == "admin" {
- continue
- }
- if err := s.applyRenewal(ctx, user.ID, days); err == nil {
- count++
- }
- }
- return telegramCommandReply{Text: fmt.Sprintf("批量续期完成：%d 个普通用户。", count)}
-}
-
-func (s *TelegramBotService) cmdMgoBanAll(ctx context.Context, active bool, args []string) telegramCommandReply {
- if len(args) == 0 || !strings.EqualFold(args[len(args)-1], "confirm") {
- action := "banall"
- if active {
- action = "unbanall"
- }
- return telegramCommandReply{Text: fmt.Sprintf("批量操作需要确认：/%s confirm", action)}
- }
- users, err := s.repo.User.List(ctx)
- if err != nil {
- return telegramCommandReply{Text: "读取用户失败：" + err.Error()}
- }
- var count int
- for _, user := range users {
- if !active && UserIsProtectedAccount(ctx, s.repo, &user) {
- continue
- }
- if active && user.Role == "admin" {
- continue
- }
- updates := map[string]any{"is_active": active}
- if active {
- updates["share_warnings"] = 0
- updates["last_share_warn_at"] = nil
- }
- if err := s.repo.User.UpdateFields(ctx, user.ID, updates); err == nil {
- _ = s.repo.UserDevice.SetKickedByUser(ctx, user.ID, !active)
- count++
- }
- }
- if active {
- return telegramCommandReply{Text: fmt.Sprintf("已解禁普通用户：%d 个。", count)}
- }
- return telegramCommandReply{Text: fmt.Sprintf("已禁用普通用户：%d 个。", count)}
-}
-
-func (s *TelegramBotService) cmdMgoCallAll(ctx context.Context, channel *model.NotifyChannel, args []string) telegramCommandReply {
- message := strings.TrimSpace(strings.Join(args, " "))
- if message == "" {
- return telegramCommandReply{Text: "用法：/callall 消息内容"}
- }
- if strings.TrimSpace(s.telegramChannelConfig(channel)["bot_token"]) == "" {
- return telegramCommandReply{Text: "当前 Telegram 渠道未配置 bot_token，无法群发。"}
- }
- var bindings []model.TelegramBinding
- if err := s.repo.DB.WithContext(ctx).Find(&bindings).Error; err != nil {
- return telegramCommandReply{Text: "读取绑定失败：" + err.Error()}
- }
- sent := 0
- for _, binding := range bindings {
- if binding.ChatID == 0 {
- continue
- }
- if err := s.reply(ctx, channel, int(binding.ChatID), telegramCommandReply{Text: message}); err == nil {
- sent++
- }
- }
- return telegramCommandReply{Text: fmt.Sprintf("群发完成：成功发送 %d 个绑定用户。", sent)}
-}
-
-func (s *TelegramBotService) cmdMgoSyncUnbound(ctx context.Context, args []string) telegramCommandReply {
- var users []model.User
- if err := s.repo.DB.WithContext(ctx).
- Where("role <> ?", "admin").
- Where("NOT EXISTS (SELECT 1 FROM telegram_bindings WHERE telegram_bindings.user_id = users.id AND telegram_bindings.deleted_at IS NULL)").
- Order("created_at asc").Find(&users).Error; err != nil {
- return telegramCommandReply{Text: "查询失败：" + err.Error()}
- }
- if len(args) >= 2 && strings.EqualFold(args[0], "delete") && strings.EqualFold(args[1], "confirm") {
- deleted := 0
- for _, user := range users {
- if UserIsProtectedAccount(ctx, s.repo, &user) {
- continue
- }
- _ = s.repo.UserDevice.DeleteByUser(ctx, user.ID)
- if err := s.repo.User.Delete(ctx, user.ID); err == nil {
- deleted++
- }
- }
- return telegramCommandReply{Text: fmt.Sprintf("已删除未绑定 Bot 的普通用户：%d 个。", deleted)}
- }
- if len(users) == 0 {
- return telegramCommandReply{Text: "没有未绑定 Bot 的普通用户。"}
- }
- names := make([]string, 0, minInt(len(users), 20))
- for i, user := range users {
- if i >= 20 {
- break
- }
- names = append(names, user.Username)
- }
- return telegramCommandReply{Text: fmt.Sprintf("未绑定 Bot 的普通用户：%d 个。\n%s\n\n如需删除：/syncunbound delete confirm", len(users), telegramInlineCodeList(names))}
-}
-
-func (s *TelegramBotService) cmdMgoCheckExpired(ctx context.Context, args []string) telegramCommandReply {
- now := time.Now()
- var users []model.User
- if err := s.repo.DB.WithContext(ctx).Where("expired_at IS NOT NULL AND expired_at < ?", now).Order("expired_at asc").Find(&users).Error; err != nil {
- return telegramCommandReply{Text: "查询失败：" + err.Error()}
- }
- if len(args) >= 2 && strings.EqualFold(args[0], "disable") && strings.EqualFold(args[1], "confirm") {
- disabled := 0
- for _, user := range users {
- if UserIsProtectedAccount(ctx, s.repo, &user) {
- continue
- }
- if err := s.repo.User.UpdateFields(ctx, user.ID, map[string]any{"is_active": false}); err == nil {
- disabled++
- }
- }
- return telegramCommandReply{Text: fmt.Sprintf("已禁用过期普通用户：%d 个。", disabled)}
- }
- if len(users) == 0 {
- return telegramCommandReply{Text: "没有过期用户。"}
- }
- lines := make([]string, 0, minInt(len(users), 20))
- for i, user := range users {
- if i >= 20 {
- break
- }
- lines = append(lines, fmt.Sprintf("%s（%s）", user.Username, formatExpiry(user.ExpiredAt)))
- }
- return telegramCommandReply{Text: fmt.Sprintf("过期用户：%d 个。\n%s\n\n如需禁用：/check_ex disable confirm", len(users), telegramInlineCodeList(lines))}
-}
-
-func (s *TelegramBotService) cmdMgoScanNames(ctx context.Context) telegramCommandReply {
- var rows []struct {
- Username string
- Count int64
- }
- if err := s.repo.DB.WithContext(ctx).Table("users").
- Select("LOWER(username) AS username, COUNT(*) AS count").
- Group("LOWER(username)").Having("COUNT(*) > 1").Scan(&rows).Error; err != nil {
- return telegramCommandReply{Text: "扫描失败：" + err.Error()}
- }
- if len(rows) == 0 {
- return telegramCommandReply{Text: "未发现同名用户记录。"}
- }
- var out []string
- for _, row := range rows {
- out = append(out, fmt.Sprintf("%s x%d", row.Username, row.Count))
- }
- return telegramCommandReply{Text: "同名用户记录\n" + telegramInlineCodeList(out)}
-}
-
-func (s *TelegramBotService) cmdMgoRanks(ctx context.Context, window time.Duration, byDuration bool) telegramCommandReply {
- since := time.Now().Add(-window)
- title := "播放次数排行"
- selectExpr := "COUNT(*) AS score"
- if byDuration {
- title = "观影时长排行"
- selectExpr = "COALESCE(SUM(position_ms), 0) AS score"
- }
- q := s.repo.DB.WithContext(ctx).Table("playback_histories").
- Select("users.username, " + selectExpr).
- Joins("JOIN users ON users.id = playback_histories.user_id").
- Group("users.username").
- Order("score DESC").
- Limit(10)
- if window > 0 {
- q = q.Where("playback_histories.watched_at >= ?", since)
- }
- var rows []struct {
- Username string
- Score int64
- }
- if err := q.Scan(&rows).Error; err != nil {
- return telegramCommandReply{Text: "排行查询失败：" + err.Error()}
- }
- if len(rows) == 0 {
- return telegramCommandReply{Text: "暂无排行数据。"}
- }
- var out []string
- for i, row := range rows {
- score := fmt.Sprintf("%d 次", row.Score)
- if byDuration {
- score = humanDurationFromMillis(row.Score)
- }
- out = append(out, fmt.Sprintf("%d. %s — %s", i+1, row.Username, score))
- }
- return telegramCommandReply{Text: "" + title + "\n\n" + strings.Join(out, "\n") + ""}
-}
-
-func (s *TelegramBotService) cmdMgoAdminRole(ctx context.Context, args []string) telegramCommandReply {
- if len(args) < 2 {
- return telegramCommandReply{Text: "用法：/embyadmin 用户名 on|off"}
- }
- user := s.findMgoBotUser(ctx, args[0])
- if user == nil {
- return telegramCommandReply{Text: "未找到用户。"}
- }
- enable := parseOnOff(args[1])
- if enable == nil {
- return telegramCommandReply{Text: "第二个参数请使用 on/off。"}
- }
- if !*enable {
- if first, _ := s.repo.User.FirstAdmin(ctx); first != nil && first.ID == user.ID {
- return telegramCommandReply{Text: "默认管理员不可降级。"}
- }
- }
- role := "user"
- if *enable {
- role = "admin"
- }
- if err := s.repo.User.UpdateFields(ctx, user.ID, map[string]any{"role": role}); err != nil {
- return telegramCommandReply{Text: "更新失败：" + err.Error()}
- }
- return telegramCommandReply{Text: fmt.Sprintf("已将 %s 角色设置为 %s。", user.Username, role)}
-}
-
-func (s *TelegramBotService) cmdMgoMediaAccessAll(ctx context.Context, allow bool) telegramCommandReply {
- users, err := s.repo.User.List(ctx)
- if err != nil {
- return telegramCommandReply{Text: "读取用户失败：" + err.Error()}
- }
- updated := 0
- for _, user := range users {
- if user.Role == "admin" {
- continue
- }
- perm, err := s.repo.Permission.FindByUserID(ctx, user.ID)
- if err != nil {
- continue
- }
- if perm == nil {
- perm = DefaultPermissions(user.ID)
- perm.CanPlayMedia = allow
- if err := s.repo.Permission.Create(ctx, perm); err != nil {
- continue
- }
- }
- if err := s.repo.DB.WithContext(ctx).Model(&model.UserPermission{}).
- Where("user_id = ?", user.ID).
- Update("can_play_media", allow).Error; err == nil {
- updated++
- }
- }
- state := "关闭"
- if allow {
- state = "开启"
- }
- return telegramCommandReply{Text: fmt.Sprintf("已为普通用户%s媒体播放权限：%d 个。", state, updated)}
-}
-
-func (s *TelegramBotService) cmdMgoBotAdmin(ctx context.Context, channel *model.NotifyChannel, args []string, add bool) telegramCommandReply {
- if channel == nil {
- return telegramCommandReply{Text: "Telegram 渠道不存在。"}
- }
- if len(args) == 0 {
- return telegramCommandReply{Text: "用法：/proadmin TelegramID 或 /revadmin TelegramID"}
- }
- tgID := strings.TrimPrefix(strings.TrimSpace(args[0]), "tg:")
- if _, err := strconv.ParseInt(tgID, 10, 64); err != nil {
- return telegramCommandReply{Text: "TelegramID 必须是数字。"}
- }
- cfg := s.telegramChannelConfig(channel)
- ids := telegramConfiguredUserIDs(cfg["admin_user_ids"])
- seen := make(map[string]bool, len(ids)+1)
- var next []string
- for _, id := range ids {
- if id == tgID {
- seen[id] = true
- if add {
- next = append(next, id)
- }
- continue
- }
- if id != "" {
- next = append(next, id)
- }
- }
- if add && !seen[tgID] {
- next = append(next, tgID)
- }
- cfg["admin_user_ids"] = strings.Join(next, ",")
- raw, _ := json.Marshal(cfg)
- updated := *channel
- updated.Config = string(raw)
- if s.crypto != nil {
- updated.Config = s.crypto.Encrypt(updated.Config)
- }
- if err := s.repo.NotifyChannel.Update(ctx, &updated); err != nil {
- return telegramCommandReply{Text: "更新管理员列表失败：" + err.Error()}
- }
- if add {
- return telegramCommandReply{Text: "已添加 Bot 管理员：" + tgID + ""}
- }
- return telegramCommandReply{Text: "已移除 Bot 管理员：" + tgID + ""}
-}
-
-func (s *TelegramBotService) cmdMgoProtectedUser(ctx context.Context, args []string, protect bool) telegramCommandReply {
- if len(args) == 0 || strings.EqualFold(args[0], "list") {
- return s.cmdMgoProtectedUserList(ctx)
- }
- user := s.findMgoBotUser(ctx, args[0])
- if user == nil {
- return telegramCommandReply{Text: "未找到用户。"}
- }
- ids := ProtectedUserIDSet(ctx, s.repo)
- if protect {
- ids[user.ID] = struct{}{}
- if err := SaveProtectedUserIDSet(ctx, s.repo, ids); err != nil {
- return telegramCommandReply{Text: "保存保护名单失败：" + err.Error()}
- }
- return telegramCommandReply{Text: fmt.Sprintf("已加入保护名单：%s。\n该用户不会被 Bot 自动清理、批量禁用或删除。", user.Username)}
- }
- delete(ids, user.ID)
- if err := SaveProtectedUserIDSet(ctx, s.repo, ids); err != nil {
- return telegramCommandReply{Text: "保存保护名单失败：" + err.Error()}
- }
- return telegramCommandReply{Text: fmt.Sprintf("已移出保护名单：%s。", user.Username)}
-}
-
-func (s *TelegramBotService) cmdMgoProtectedUserList(ctx context.Context) telegramCommandReply {
- ids := ProtectedUserIDSet(ctx, s.repo)
- if len(ids) == 0 {
- return telegramCommandReply{Text: "保护名单为空。管理员和默认管理员始终自动保护。"}
- }
- names := make([]string, 0, len(ids))
- for id := range ids {
- if user, _ := s.repo.User.FindByID(ctx, id); user != nil {
- names = append(names, user.Username)
- } else {
- names = append(names, id+"(用户不存在)")
- }
- }
- sort.Strings(names)
- return telegramCommandReply{Text: fmt.Sprintf("保护名单：%d 个。\n%s", len(names), telegramInlineCodeList(names))}
-}
-
-func (s *TelegramBotService) cmdMgoBackupDB(ctx context.Context) telegramCommandReply {
- if s.backup == nil {
- return telegramCommandReply{Text: "备份服务暂不可用。"}
- }
- info, err := s.backup.Create(ctx)
- if err != nil {
- return telegramCommandReply{Text: "数据库备份失败：" + err.Error()}
- }
- return telegramCommandReply{Text: fmt.Sprintf("数据库备份完成：%s\n大小：%d bytes", info.Filename, info.Size)}
-}
-
-func (s *TelegramBotService) cmdMgoRestoreDB(ctx context.Context, args []string) telegramCommandReply {
- if s.backup == nil {
- return telegramCommandReply{Text: "备份服务暂不可用。"}
- }
- if len(args) == 0 || strings.EqualFold(args[0], "list") {
- items, err := s.backup.List()
- if err != nil {
- return telegramCommandReply{Text: "读取备份列表失败：" + err.Error()}
- }
- if len(items) == 0 {
- return telegramCommandReply{Text: "暂无数据库备份。可先使用 /backup_db 创建。"}
- }
- lines := make([]string, 0, minInt(len(items), 10))
- for i, item := range items {
- if i >= 10 {
- break
- }
- lines = append(lines, fmt.Sprintf("%s（%d bytes）", item.Filename, item.Size))
- }
- return telegramCommandReply{Text: "可恢复备份：\n" + telegramInlineCodeList(lines) + "\n\n恢复需要确认：/restore_from_db 文件名 confirm"}
- }
- if len(args) < 2 || !strings.EqualFold(args[len(args)-1], "confirm") {
- return telegramCommandReply{Text: "恢复数据库会覆盖当前数据，需要确认：/restore_from_db 文件名 confirm"}
- }
- filename := strings.TrimSpace(args[0])
- if err := s.backup.Restore(ctx, filename); err != nil {
- return telegramCommandReply{Text: "恢复失败：" + err.Error()}
- }
- return telegramCommandReply{Text: "数据库已从备份恢复，请重启 MediaStationGo 后生效。"}
-}
-
-func (s *TelegramBotService) cmdMgoSyncGroup(ctx context.Context, channel *model.NotifyChannel, args []string) telegramCommandReply {
- chatIDs := s.telegramMembershipChatIDs(channel)
- if len(chatIDs) == 0 {
- return telegramCommandReply{Text: "未配置可校验成员的群组/频道 ID。请在 Telegram 通知渠道设置 group_chat_id 或 channel_chat_id。"}
- }
- if strings.TrimSpace(s.telegramChannelConfig(channel)["bot_token"]) == "" {
- return telegramCommandReply{Text: "当前 Telegram 渠道未配置 bot_token，无法校验群成员。"}
- }
- var bindings []model.TelegramBinding
- if err := s.repo.DB.WithContext(ctx).Find(&bindings).Error; err != nil {
- return telegramCommandReply{Text: "读取绑定失败：" + err.Error()}
- }
- type staleBinding struct {
- User model.User
- Binding model.TelegramBinding
- }
- var stale []staleBinding
- for _, binding := range bindings {
- if binding.TelegramUserID == 0 || binding.UserID == "" {
- continue
- }
- user, _ := s.repo.User.FindByID(ctx, binding.UserID)
- if user == nil || UserIsProtectedAccount(ctx, s.repo, user) {
- continue
- }
- // 仅当所有绑定群组/频道都「查实不是成员」时才判定为可清理；
- // getChatMember 出错（membershipUnknown）时保守跳过，避免误删。
- confirmedNo := true
- for _, chatID := range chatIDs {
- if s.telegramChatMembership(ctx, channel, chatID, int(binding.TelegramUserID)) != membershipNo {
- confirmedNo = false
- break
- }
- }
- if confirmedNo {
- stale = append(stale, staleBinding{User: *user, Binding: binding})
- }
- }
- if len(stale) == 0 {
- return telegramCommandReply{Text: "所有已绑定账号都仍在配置的群组/频道中。"}
- }
- if len(args) >= 2 && strings.EqualFold(args[0], "delete") && strings.EqualFold(args[1], "confirm") {
- deleted := 0
- for _, item := range stale {
- _ = s.repo.UserDevice.DeleteByUser(ctx, item.User.ID)
- if err := s.repo.User.Delete(ctx, item.User.ID); err == nil {
- deleted++
- }
- }
- return telegramCommandReply{Text: fmt.Sprintf("已删除不在群组/频道中的普通账号：%d 个。", deleted)}
- }
- names := make([]string, 0, minInt(len(stale), 20))
- for i, item := range stale {
- if i >= 20 {
- break
- }
- names = append(names, fmt.Sprintf("%s(tg:%d)", item.User.Username, item.Binding.TelegramUserID))
- }
- return telegramCommandReply{Text: fmt.Sprintf("不在配置群组/频道中的绑定账号：%d 个。\n%s\n\n删除需确认：/syncgroupm delete confirm", len(stale), telegramInlineCodeList(names))}
-}
-
-func (s *TelegramBotService) telegramMembershipChatIDs(channel *model.NotifyChannel) []string {
- cfg := s.telegramChannelConfig(channel)
- seen := map[string]struct{}{}
- var out []string
- for _, key := range []string{"group_chat_id", "channel_chat_id", "command_chat_id"} {
- value := strings.TrimSpace(cfg[key])
- if value == "" {
- continue
- }
- if _, ok := seen[value]; !ok {
- seen[value] = struct{}{}
- out = append(out, value)
- }
- }
- if len(out) == 0 {
- if value := strings.TrimSpace(cfg["chat_id"]); strings.HasPrefix(value, "-") {
- out = append(out, value)
- }
- }
- return out
-}
-
func (s *TelegramBotService) cmdMgoUnsupported(name, replacement string) telegramCommandReply {
text := fmt.Sprintf("%s 已识别，但当前 Telegram Bot API 无法完整复刻该行为。", name)
if replacement != "" {
diff --git a/internal/service/telegram_mgo_membership.go b/internal/service/telegram_mgo_membership.go
new file mode 100644
index 0000000..0804483
--- /dev/null
+++ b/internal/service/telegram_mgo_membership.go
@@ -0,0 +1,92 @@
+package service
+
+import (
+ "context"
+ "fmt"
+ "strings"
+
+ "github.com/ShukeBta/MediaStationGo/internal/model"
+)
+
+func (s *TelegramBotService) cmdMgoSyncGroup(ctx context.Context, channel *model.NotifyChannel, args []string) telegramCommandReply {
+ chatIDs := s.telegramMembershipChatIDs(channel)
+ if len(chatIDs) == 0 {
+ return telegramCommandReply{Text: "未配置可校验成员的群组/频道 ID。请在 Telegram 通知渠道设置 group_chat_id 或 channel_chat_id。"}
+ }
+ if strings.TrimSpace(s.telegramChannelConfig(channel)["bot_token"]) == "" {
+ return telegramCommandReply{Text: "当前 Telegram 渠道未配置 bot_token，无法校验群成员。"}
+ }
+ var bindings []model.TelegramBinding
+ if err := s.repo.DB.WithContext(ctx).Find(&bindings).Error; err != nil {
+ return telegramCommandReply{Text: "读取绑定失败：" + err.Error()}
+ }
+ type staleBinding struct {
+ User model.User
+ Binding model.TelegramBinding
+ }
+ var stale []staleBinding
+ for _, binding := range bindings {
+ if binding.TelegramUserID == 0 || binding.UserID == "" {
+ continue
+ }
+ user, _ := s.repo.User.FindByID(ctx, binding.UserID)
+ if user == nil || UserIsProtectedAccount(ctx, s.repo, user) {
+ continue
+ }
+ // 仅当所有绑定群组/频道都「查实不是成员」时才判定为可清理；
+ // getChatMember 出错（membershipUnknown）时保守跳过，避免误删。
+ confirmedNo := true
+ for _, chatID := range chatIDs {
+ if s.telegramChatMembership(ctx, channel, chatID, int(binding.TelegramUserID)) != membershipNo {
+ confirmedNo = false
+ break
+ }
+ }
+ if confirmedNo {
+ stale = append(stale, staleBinding{User: *user, Binding: binding})
+ }
+ }
+ if len(stale) == 0 {
+ return telegramCommandReply{Text: "所有已绑定账号都仍在配置的群组/频道中。"}
+ }
+ if len(args) >= 2 && strings.EqualFold(args[0], "delete") && strings.EqualFold(args[1], "confirm") {
+ deleted := 0
+ for _, item := range stale {
+ _ = s.repo.UserDevice.DeleteByUser(ctx, item.User.ID)
+ if err := s.repo.User.Delete(ctx, item.User.ID); err == nil {
+ deleted++
+ }
+ }
+ return telegramCommandReply{Text: fmt.Sprintf("已删除不在群组/频道中的普通账号：%d 个。", deleted)}
+ }
+ names := make([]string, 0, minInt(len(stale), 20))
+ for i, item := range stale {
+ if i >= 20 {
+ break
+ }
+ names = append(names, fmt.Sprintf("%s(tg:%d)", item.User.Username, item.Binding.TelegramUserID))
+ }
+ return telegramCommandReply{Text: fmt.Sprintf("不在配置群组/频道中的绑定账号：%d 个。\n%s\n\n删除需确认：/syncgroupm delete confirm", len(stale), telegramInlineCodeList(names))}
+}
+
+func (s *TelegramBotService) telegramMembershipChatIDs(channel *model.NotifyChannel) []string {
+ cfg := s.telegramChannelConfig(channel)
+ seen := map[string]struct{}{}
+ var out []string
+ for _, key := range []string{"group_chat_id", "channel_chat_id", "command_chat_id"} {
+ value := strings.TrimSpace(cfg[key])
+ if value == "" {
+ continue
+ }
+ if _, ok := seen[value]; !ok {
+ seen[value] = struct{}{}
+ out = append(out, value)
+ }
+ }
+ if len(out) == 0 {
+ if value := strings.TrimSpace(cfg["chat_id"]); strings.HasPrefix(value, "-") {
+ out = append(out, value)
+ }
+ }
+ return out
+}