admin.go

package main

import (
	"fmt"
	"strconv"
	"strings"
	"time"

	"github.com/mymmrac/telego"
	th "github.com/mymmrac/telego/telegohandler"
	tu "github.com/mymmrac/telego/telegoutil"
)

func uptimeStr(start time.Time) string {
	return time.Since(start).Round(time.Second).String()
}

func (v *Verifier) stateText() string {
	if v.isEnabled() {
		return "开启"
	}
	return "关闭"
}

func (v *Verifier) onPing(ctx *th.Context, update telego.Update) error {
	return v.memberCmd(ctx, update, func() string {
		return fmt.Sprintf("🏓 pong | %s | 运行 %s | 验证:%s", version, uptimeStr(v.startTime), v.stateText())
	})
}

// onStart: in a private chat it's the deep-link verification entry; in a group it's the admin toggle.
func (v *Verifier) onStart(ctx *th.Context, update telego.Update) error {
	msg := update.Message
	if msg != nil && msg.Chat.Type == "private" {
		if msg.From != nil {
			v.sendDMChallenge(ctx.Context(), ctx.Bot(), msg.From.ID)
		}
		return nil
	}
	return v.adminCmd(ctx, update, func() string {
		v.setEnabled(true)
		return "✅ 入群验证已开启。"
	})
}

func (v *Verifier) onStop(ctx *th.Context, update telego.Update) error {
	return v.adminCmd(ctx, update, func() string {
		v.setEnabled(false)
		return "⏸ 入群验证已关闭。新入群申请将不再自动验证(留给人工审批)。"
	})
}

func (v *Verifier) onStats(ctx *th.Context, update telego.Update) error {
	return v.memberCmd(ctx, update, func() string {
		date, ap, de := v.stats()
		return fmt.Sprintf("📊 今日(%s)\n✅ 通过:%d 人\n❌ 拒绝:%d 人\n验证:%s | 运行 %s",
			date, ap, de, v.stateText(), uptimeStr(v.startTime))
	})
}

// onRich (admin) toggles rich-message output for /pkg and /use at runtime.
func (v *Verifier) onRich(ctx *th.Context, update telego.Update) error {
	return v.adminCmd(ctx, update, func() string {
		if v.toggleRich() {
			return "🎨 富文本输出已开启(/pkg、/use 用富消息;旧版/第三方客户端可能渲染不佳)。"
		}
		return "📄 富文本输出已关闭,/pkg、/use 改用纯文本。"
	})
}

// onSpoiler toggles whether a new member's display name is hidden behind a Telegram spoiler in the
// in-group verification challenge — defeats spammers who set their name to an advert. Persisted.
func (v *Verifier) onSpoiler(ctx *th.Context, update telego.Update) error {
	return v.adminCmd(ctx, update, func() string {
		if v.toggleNameSpoiler() {
			return "🫥 已开启:新成员的名字会用「防剧透」遮盖,不点开看不到 —— 防止有人拿广告当名字在群里刷屏。"
		}
		return "👁 已关闭:新成员的名字将正常显示(不再遮盖)。"
	})
}

// parseAutoDelArg interprets an /autodel argument (already trimmed + lower-cased) into an
// action: "show" (empty), "off", "on", "set" (with a ttl of 1–1440 minutes), or "" for an
// invalid argument. Pure (no state) so it's unit-tested directly.
func parseAutoDelArg(arg string) (action string, ttl time.Duration) {
	switch arg {
	case "":
		return "show", 0
	case "off":
		return "off", 0
	case "on":
		return "on", 0
	}
	if n, err := strconv.Atoi(arg); err == nil && n >= 1 && n <= 1440 {
		return "set", time.Duration(n) * time.Minute
	}
	return "", 0 // invalid
}

// onAutoDel handles /autodel — toggle/adjust auto-deletion of lookup commands + answers:
// no arg shows the state, "on"/"off" enable/disable, "<minutes>" sets the delay (1–1440).
func (v *Verifier) onAutoDel(ctx *th.Context, update telego.Update) error {
	return v.adminCmd(ctx, update, func() string {
		action, ttl := parseAutoDelArg(strings.ToLower(strings.TrimSpace(commandArg(update.Message.Text))))
		switch action {
		case "show":
			if cur, on := v.lookupAutoDelete(); on {
				return fmt.Sprintf("🧹 查询结果自动删除:已开启,%d 分钟后连同提问一起删除。\n用法:/autodel off 关闭;/autodel <分钟> 调整时间。", int(cur/time.Minute))
			}
			return "查询结果自动删除:已关闭。\n用法:/autodel on 开启;/autodel <分钟> 设定时间并开启。"
		case "off":
			v.setLookupAutoDelete(0, false)
			return "已关闭查询结果自动删除(查询命令 /pkg、/use、/bug、/news、/wiki、/bbs、/pkgs、/arm、/armpkgs 的回复将保留)。"
		case "on":
			v.setLookupAutoDelete(0, true)
			cur, _ := v.lookupAutoDelete()
			return fmt.Sprintf("🧹 已开启:查询结果 %d 分钟后连同提问一起删除。", int(cur/time.Minute))
		case "set":
			v.setLookupAutoDelete(ttl, true)
			return fmt.Sprintf("🧹 已设定:查询结果 %d 分钟后连同提问一起删除。", int(ttl/time.Minute))
		default:
			return "用法:/autodel on|off,或 /autodel <分钟数>(1–1440)。"
		}
	})
}

// onHelp lists commands (admins also see the moderation/admin commands).
func (v *Verifier) onHelp(ctx *th.Context, update telego.Update) error {
	msg := update.Message
	if msg == nil || msg.From == nil || !v.dmOrGroup(msg) { // /help is free (no external request)
		return nil
	}
	bot := ctx.Bot()
	c := ctx.Context()
	chatID := msg.Chat.ID
	inGroup := v.cfg.IsGroup(chatID)
	help := "🤖 可用指令:\n" +
		"/pkg <包名> — 搜索 Gentoo 包(官方树/gentoo-zh/guru)\n" +
		"/use <包名> — 某个包的 USE 标志 + 信息\n" +
		"/bug <编号> — 查询 Gentoo Bugzilla\n" +
		"/news [关键词] — 查看/搜索 Gentoo 新闻\n" +
		"/wiki <关键词> — 搜索 Gentoo / Arch wiki(含中文页)\n" +
		"/bbs <关键词> — 搜各大 Linux 论坛(中文优先)\n" +
		"/pkgs <包名> — 跨发行版查包版本(Gentoo/Debian/Ubuntu/Fedora/Arch/openSUSE 等)\n" +
		"/arm <包名> — 查该包的 arm64 (aarch64) Gentoo keyword 状态\n" +
		"/armpkgs <包名> — 跨发行版查 arm64 支持\n" +
		"/ping — 机器人状态 / 运行时长\n" +
		"/stats — 今日通过 / 拒绝人数\n" +
		"/help — 显示本帮助"
	if inGroup && v.isGroupAdmin(c, bot, chatID, msg.From.ID) {
		help += "\n\n👮 管理员(回复某条消息使用):\n" +
			"/mute — 禁言(留群不能发言,到期自动解除);默认1h,可 /mute 30m\n" +
			"/unmute — 解除禁言\n" +
			"/ban — 封禁(踢出群,仅删该条消息)\n" +
			"/sb — 封禁并清空消息(踢出群 + 清除其全部消息)\n" +
			fmt.Sprintf("/warn — 警告(满 %d 次自动踢出);/clearwarn — 清除警告\n", v.cfg.WarnLimit) +
			"\n其它管理指令:\n" +
			"/bantime — 设定封禁时长(0=永久;如 7d/12h/30m)\n" +
			"/autodel — 查询结果自动删除开关(/autodel on|off|<分钟>)\n" +
			"/rich — 开关富文本输出(/pkg /use)\n" +
			"/spoiler — 开关:遮盖新成员名字(防广告;默认开)\n" +
			"/bc — 频道马甲封禁开关;/bc allow|deny <频道id> 管白名单\n" +
			"/start /stop — 开启 / 关闭入群验证"
	}
	if inGroup {
		_ = bot.DeleteMessage(c, &telego.DeleteMessageParams{ChatID: tu.ID(chatID), MessageID: msg.MessageID})
		v.notify(c, bot, chatID, help)
		return nil
	}
	help += "\n\n(以上查询命令私聊也能用,每分钟限次;审核/管理命令仅在群里有效。)"
	// Plain text (no HTML parse mode): the help lists literal <包名> placeholders that would
	// otherwise be misread as HTML tags and rejected by Telegram. The group path uses notify,
	// which is also plain.
	_, _ = bot.SendMessage(c, tu.Message(tu.ID(chatID), help))
	return nil
}

// memberCmd runs a cheap informational command (no external request) usable by ANY member:
// in a guarded group (the result auto-deletes and the trigger is removed) or in a private
// chat (replied plainly). NOT rate-limited — only the API-hitting lookups are. No admin check.
func (v *Verifier) memberCmd(ctx *th.Context, update telego.Update, fn func() string) error {
	msg := update.Message
	if msg == nil || !v.dmOrGroup(msg) {
		return nil
	}
	bot := ctx.Bot()
	c := ctx.Context()
	if v.cfg.IsGroup(msg.Chat.ID) {
		_ = bot.DeleteMessage(c, &telego.DeleteMessageParams{ChatID: tu.ID(msg.Chat.ID), MessageID: msg.MessageID})
		v.notify(c, bot, msg.Chat.ID, fn())
		return nil
	}
	_, _ = bot.SendMessage(c, tu.Message(tu.ID(msg.Chat.ID), fn())) // DM: reply as plain text, no delete
	return nil
}

// adminCmd runs fn only for a group admin in a guarded group, posts the result as
// a transient (auto-deleting) message, and removes the command message.
func (v *Verifier) adminCmd(ctx *th.Context, update telego.Update, fn func() string) error {
	msg := update.Message
	if msg == nil || msg.From == nil || !v.cfg.IsGroup(msg.Chat.ID) {
		return nil
	}
	bot := ctx.Bot()
	c := ctx.Context()
	gid := msg.Chat.ID
	defer func() {
		_ = bot.DeleteMessage(c, &telego.DeleteMessageParams{ChatID: tu.ID(gid), MessageID: msg.MessageID})
	}()
	if !v.isGroupAdmin(c, bot, gid, msg.From.ID) {
		v.notify(c, bot, gid, "⛔ 该命令仅群管理员可用。")
		return nil
	}
	v.notify(c, bot, gid, fn())
	return nil
}