remake_bot/quotely/quotely.go

package quotely

import (
	"fmt"
	tele "gopkg.in/telebot.v3"
	"strings"
	"unicode"
)

func QuoteReply(bot *tele.Bot, message *tele.Message) (replyMsg string) {
	if len(message.Text) < 2 {
		return
	}
	if !strings.HasPrefix(message.Text, "/") || (isASCII(message.Text[:2]) && !strings.HasPrefix(message.Text, "/$")) {
		if !strings.HasPrefix(message.Text, "\\") || (isASCII(message.Text[:2]) && !strings.HasPrefix(message.Text, "\\$")) {
			return
		}
	}

	keywords := strings.SplitN(EscapeText(strings.Replace(message.Text, "$", "", 1)[1:]), " ", 2)
	if len(keywords) == 0 {
		return
	}

	senderName := EscapeText(message.Sender.FirstName + " " + message.Sender.LastName)
	senderURI := fmt.Sprintf("tg://user?id=%d", message.Sender.ID)
	replyToName := ""
	replyToURI := ""

	if message.SenderChat != nil {
		senderName = EscapeText(message.SenderChat.Title)
		senderURI = fmt.Sprintf("https://t.me/%s", message.SenderChat.Username)
	}

	if message.ReplyTo != nil && message.TopicMessage {
		if message.ReplyTo.ID == message.ThreadID {
			message.ReplyTo = nil
		}
	}

	if message.ReplyTo != nil {
		replyToName = EscapeText(message.ReplyTo.Sender.FirstName + " " + message.ReplyTo.Sender.LastName)
		replyToURI = fmt.Sprintf("tg://user?id=%d", message.ReplyTo.Sender.ID)

		if message.ReplyTo.Sender.IsBot && len(message.ReplyTo.Entities) != 0 {
			if message.ReplyTo.Entities[0].Type == "text_mention" {
				replyToName = EscapeText(message.ReplyTo.Entities[0].User.FirstName + " " + message.ReplyTo.Entities[0].User.LastName)
				replyToURI = fmt.Sprintf("tg://user?id=%d", message.ReplyTo.Entities[0].User.ID)
			}
		}

		if message.ReplyTo.SenderChat != nil {
			replyToName = EscapeText(message.ReplyTo.SenderChat.Title)
			replyToURI = fmt.Sprintf("https://t.me/%s", message.ReplyTo.SenderChat.Username)
		}

		if strings.HasPrefix(message.Text, "\\") {
			senderName, replyToName = replyToName, senderName
			senderURI, replyToURI = replyToURI, senderURI
		}
	} else {
		textNoCommand := strings.TrimPrefix(strings.TrimPrefix(keywords[0], "/"), "$")
		if text := strings.Split(textNoCommand, "@"); len(text) > 1 {
			name := getUserByUsername(bot, text[1])
			if name != "" {
				keywords[0] = text[0]
				replyToName = EscapeText(name)
				replyToURI = fmt.Sprintf("https://t.me/%s", text[1])
			}
		}
		if replyToName == "" {
			replyToName = "自己"
			replyToURI = senderURI
		}
	}
	if len(keywords) < 2 {
		return fmt.Sprintf("[%s](%s) %s了 [%s](%s)！", senderName, senderURI, keywords[0], replyToName, replyToURI)
	} else {
		return fmt.Sprintf("[%s](%s) %s [%s](%s) %s！", senderName, senderURI, keywords[0], replyToName, replyToURI, keywords[1])
	}
}

func isASCII(s string) bool {
	for _, r := range s {
		if r > unicode.MaxASCII {
			return false
		}
	}
	return true
}

// EscapeText source: https://github.com/go-telegram-bot-api/telegram-bot-api/blob/4126fa611266940425a9dfd37e0c92ba47881718/bot.go#L729
func EscapeText(text string) string {
	replacer := strings.NewReplacer(
		"_", "\\_", "*", "\\*", "[", "\\[", "]", "\\]", "(",
		"\\(", ")", "\\)", "~", "\\~", "`", "\\`", ">", "\\>",
		"#", "\\#", "+", "\\+", "-", "\\-", "=", "\\=", "|",
		"\\|", "{", "\\{", "}", "\\}", ".", "\\.", "!", "\\!",
	)

	return replacer.Replace(text)
}

func getUserByUsername(bot *tele.Bot, username string) string {
	user, err := bot.ChatByUsername(username)
	if err != nil {
		return ""
	}
	return user.FirstName + " " + user.LastName
}