help_command.go

package tgbot

import (
	"fmt"
	"strings"

	"github.com/nekomeowww/insights-bot/pkg/types/telegram"
	"github.com/samber/lo"
)

type Command struct {
	Command     string
	HelpMessage string
	Handler     Handler
}

type commandGroup struct {
	name     string
	commands []Command
}

type helpCommandHandler struct {
	defaultGroup  commandGroup
	commandGroups []commandGroup
}

func newHelpCommandHandler() *helpCommandHandler {
	h := &helpCommandHandler{
		commandGroups: make([]commandGroup, 0),
	}

	return h
}

func (h *helpCommandHandler) Command() string {
	return "help"
}

func (h *helpCommandHandler) CommandHelp() string {
	return "获取帮助"
}

func (h *helpCommandHandler) handle(c *Context) (Response, error) {
	is, err := c.IsBotAdministrator()
	if err != nil {
		c.Logger.Error("failed to check if bot is administrator")
	}
	if is &&
		c.Update.Message != nil &&
		c.Update.Message.Chat != nil &&
		lo.Contains([]telegram.ChatType{telegram.ChatTypeGroup, telegram.ChatTypeSuperGroup}, telegram.ChatType(c.Update.Message.Chat.Type)) &&
		!lo.Contains([]string{
			fmt.Sprintf("%s@%s", h.Command(), c.Bot.Self.UserName),
			fmt.Sprintf("%s@%s", "start", c.Bot.Self.UserName),
		}, c.Update.Message.CommandWithAt()) {
		return nil, nil
	}

	helpMessage := strings.Builder{}
	helpMessage.WriteString("你好，欢迎使用 Insights Bot！\n\n")
	helpMessage.WriteString("我当前支持这些命令：\n\n")

	commandGroupHelpMessages := make([]string, 0)

	if len(h.defaultGroup.commands) > 0 {
		h.commandGroups = append(h.commandGroups, h.defaultGroup)
	}

	for _, group := range h.commandGroups {
		commandHelpMessages := make([]string, 0)

		for _, cmd := range group.commands {
			commandHelpMessage := strings.Builder{}

			commandHelpMessage.WriteString("/")
			commandHelpMessage.WriteString(cmd.Command)
			commandHelpMessage.WriteString("@")
			commandHelpMessage.WriteString(c.Bot.Self.UserName)

			if cmd.HelpMessage != "" {
				commandHelpMessage.WriteString(" - ")
				commandHelpMessage.WriteString(cmd.HelpMessage)
			}

			commandHelpMessages = append(commandHelpMessages, commandHelpMessage.String())
		}

		commandGroupHelpMessages = append(commandGroupHelpMessages, fmt.Sprintf("%s%s", lo.Ternary(group.name != "", fmt.Sprintf("<b>%s</b>\n\n", EscapeHTMLSymbols(group.name)), ""), strings.Join(commandHelpMessages, "\n")))
	}

	helpMessage.WriteString(strings.Join(commandGroupHelpMessages, "\n\n"))

	return c.NewMessageReplyTo(helpMessage.String(), c.Update.Message.MessageID).WithParseModeHTML(), nil
}