Initial commit

2026-05-20 16:13:58 +08:00
commit ceffbe382e
500 changed files with 157469 additions and 0 deletions
@@ -0,0 +1,254 @@
+package tgbot
+
+import (
+	"context"
+	"errors"
+	"strconv"
+	"strings"
+	"sync"
+
+	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
+)
+
+type CommandType int32
+
+const (
+	CommandTypeUnknown  CommandType = 0
+	CommandTypeCommand  CommandType = 1
+	CommandTypeCallback CommandType = 2
+)
+
+type TgMessageList struct {
+	lock sync.Mutex
+	list []tgbotapi.Chattable
+}
+
+func (s *TgMessageList) Push(msg tgbotapi.Chattable) {
+	s.lock.Lock()
+	defer s.lock.Unlock()
+	if s.list == nil {
+		s.list = make([]tgbotapi.Chattable, 0)
+	}
+	s.list = append(s.list, msg)
+}
+
+func (s *TgMessageList) Pop() tgbotapi.Chattable {
+	s.lock.Lock()
+	defer s.lock.Unlock()
+	var ret tgbotapi.Chattable
+	if len(s.list) > 0 {
+		ret = s.list[0]
+		s.list = s.list[1:]
+	}
+	return ret
+}
+
+type TgCommands struct {
+	commands map[string]*Command
+	lock     sync.Mutex
+}
+
+func (s *TgCommands) GetCommand(cmd string) *Command {
+	s.lock.Lock()
+	defer s.lock.Unlock()
+	return s.commands[cmd]
+}
+
+func (s *TgCommands) SetCommand(cmd *Command) {
+	s.lock.Lock()
+	defer s.lock.Unlock()
+	s.commands[cmd.Cmd] = cmd
+}
+
+type CommandFunc func(ctx context.Context, cmd *Command) error
+
+type Command struct {
+	Cmd       string      // 命令
+	Help      string      // 命令帮助
+	Desc      string      // 命令描述
+	CmdType   CommandType // 命令类型
+	Process   CommandFunc // 处理函数
+	Params    []string    // 参数
+	botNames  []string    // @
+	skip      bool        // 是否跳过处理
+	owner     *TgBot
+	ParseMode string
+	PhotoPath string
+	FilePath  string
+	message   *tgbotapi.Message
+}
+
+func (c *Command) Message() *tgbotapi.Message {
+	return c.message
+}
+
+func (c *Command) HasFlag(flag string) bool {
+	for _, arg := range c.Params {
+		if strings.Contains(arg, "=") {
+			ss := strings.Split(arg, "=")
+			if len(ss) > 0 {
+				if ss[0] == flag {
+					return true
+				}
+			}
+		} else {
+			if arg == flag {
+				return true
+			}
+		}
+	}
+	return false
+}
+
+func (c *Command) GetFlagStringValue(flag string) string {
+	for index, arg := range c.Params {
+		pos := strings.Index(arg, "=")
+		if pos > 0 {
+			cmd := arg[:pos]
+			if cmd == flag {
+				content := arg[pos+1:]
+				if strings.HasPrefix(content, `"`) && strings.HasSuffix(content, `"`) {
+					if text, err := strconv.Unquote(content); err != nil {
+						return content
+					} else {
+						return text
+					}
+				} else {
+					return content
+				}
+			}
+		} else {
+			if arg == flag {
+				if index+1 < len(c.Params) {
+					return c.Params[index+1]
+				}
+			}
+		}
+	}
+	return ""
+}
+
+func (c *Command) GetFlagInt64Value(flag string) (int64, error) {
+	s := c.GetFlagStringValue(flag)
+	if s != "" {
+		return strconv.ParseInt(s, 10, 64)
+	}
+	return 0, errors.New("flag value is empty")
+}
+
+func (c *Command) GetFlagInt32Value(flag string) (int32, error) {
+	s := c.GetFlagStringValue(flag)
+	if s != "" {
+		i, err := strconv.ParseInt(s, 10, 64)
+		return int32(i), err
+	}
+	return 0, errors.New("flag value is empty")
+}
+
+func (c *Command) GetFlagStringSlice(flag string) []string {
+	s := c.GetFlagStringValue(flag)
+	return strings.Split(s, ",")
+}
+
+func (c *Command) GetFlagInt64Slice(flag string) []int64 {
+	ret := make([]int64, 0)
+	s := c.GetFlagStringValue(flag)
+	ss := strings.Split(s, ",")
+	for _, v := range ss {
+		if i, err := strconv.ParseInt(v, 10, 64); err == nil {
+			ret = append(ret, i)
+		}
+	}
+	return ret
+}
+
+func (c *Command) GetFlagInt32Slice(flag string) []int32 {
+	ret := make([]int32, 0)
+	s := c.GetFlagStringValue(flag)
+	ss := strings.Split(s, ",")
+	for _, v := range ss {
+		if i, err := strconv.ParseInt(v, 10, 64); err == nil {
+			ret = append(ret, int32(i))
+		}
+	}
+	return ret
+}
+
+func (c *Command) SendBack(text string) error {
+	if c.message != nil {
+		if c.CmdType == CommandTypeCallback {
+			if c.PhotoPath != "" || c.FilePath != "" {
+				return c.SendReply(text, nil)
+			} else {
+				return c.SendEdit(text, nil)
+			}
+		} else if c.CmdType == CommandTypeCommand {
+			return c.SendReply(text, nil)
+		} else {
+			return c.SendText(text)
+		}
+	} else {
+		return errors.New("tg message is nil")
+	}
+}
+
+func (c *Command) SendEntity(ent *EntityList) error {
+	if c == nil {
+		return errors.New("bot cmd can not be nil")
+	}
+	if ent == nil {
+		return errors.New("entity can not be nil")
+	}
+	if c.message == nil {
+		return errors.New("tg message is nil")
+	}
+	chat := c.message.Chat.ID
+	entities, text := ent.ToEntities()
+	msg := tgbotapi.NewMessage(chat, text)
+	msg.Entities = entities
+	c.owner.messages.Push(msg)
+	return nil
+}
+
+func (c *Command) SendReply(text string, replyMarkup interface{}) error {
+	if c.message != nil {
+		return replyMessage(c.owner, c.message, text, replyMarkup, c.ParseMode, c.PhotoPath, c.FilePath)
+	} else {
+		return errors.New("tg message is nil")
+	}
+}
+
+func (c *Command) SendEdit(text string, replyMarkup interface{}) error {
+	if c.message != nil {
+		return editMessage(c.owner, c.message, text, replyMarkup, c.ParseMode)
+	} else {
+		return errors.New("tg message is nil")
+	}
+}
+
+func (c *Command) SendText(text string) error {
+	if c.message != nil {
+		return sendMessageToChat(c.owner, text, c.message.Chat.ID, c.ParseMode, c.PhotoPath, c.FilePath)
+	} else {
+		return errors.New("tg message is nil")
+	}
+}
+
+func (c *Command) SendFile(text string, fileName string) error {
+	if c.message != nil {
+		c.FilePath = fileName
+		return sendMessageToChat(c.owner, text, c.message.Chat.ID, c.ParseMode, c.PhotoPath, c.FilePath)
+	} else {
+		return errors.New("tg message is nil")
+	}
+}
+
+func (c *Command) SendDelete() error {
+	if c.message != nil {
+		msg := tgbotapi.NewDeleteMessage(c.message.Chat.ID, c.message.MessageID)
+		_, err := c.owner.bot.Request(msg)
+		return err
+	} else {
+		return errors.New("tg message is nil")
+	}
+}