144 lines
3.2 KiB
Go
144 lines
3.2 KiB
Go
package tg
|
|
|
|
import (
|
|
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
|
|
//"strings"
|
|
re "regexp"
|
|
"fmt"
|
|
)
|
|
type Message = tgbotapi.Message
|
|
|
|
// Simple text message component type.
|
|
type MessageCompo struct {
|
|
Message Message
|
|
ParseMode string
|
|
Text string
|
|
}
|
|
|
|
var (
|
|
escapeRe = re.MustCompile(`([_*\[\]()~`+"`"+`>#+-=|{}.!])`)
|
|
NewRawMessage = tgbotapi.NewMessage
|
|
)
|
|
|
|
// Escape special characters in Markdown 2 and return the
|
|
// resulting string.
|
|
func Escape2(str string) string {
|
|
return string(escapeRe.ReplaceAll([]byte(str), []byte("\\$1")))
|
|
}
|
|
|
|
// Call the function after the message was sent.
|
|
func (compo *MessageCompo) Update(c Context) error {
|
|
edit := tgbotapi.NewEditMessageText(
|
|
c.Session.Id.ToApi(),
|
|
compo.Message.MessageID,
|
|
compo.Text,
|
|
)
|
|
msg, err := c.bot.api.Send(edit)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
compo.Message = msg
|
|
return nil
|
|
}
|
|
|
|
func (compo *MessageCompo) Delete(c *Context) {
|
|
cfg := tgbotapi.NewDeleteMessage(c.Session.Id.ToApi(), compo.Message.MessageID)
|
|
c.Bot.Api.Send(cfg)
|
|
//c.Sendf("%q", err)
|
|
}
|
|
|
|
// Is only implemented to make it sendable and so we can put it
|
|
// return of rendering functions.
|
|
func (compo *MessageCompo) SetMessage(msg Message) {
|
|
compo.Message = msg
|
|
}
|
|
|
|
// Return new message with the specified text.
|
|
func Messagef(format string, v ...any) MessageCompo {
|
|
ret := MessageCompo{}
|
|
ret.Text = fmt.Sprintf(format, v...)
|
|
ret.ParseMode = tgbotapi.ModeMarkdown
|
|
return ret
|
|
}
|
|
|
|
// Return message with the specified parse mode.
|
|
func (msg MessageCompo) withParseMode(mode string) MessageCompo {
|
|
msg.ParseMode = mode
|
|
return msg
|
|
}
|
|
|
|
// Set the default Markdown parsing mode.
|
|
func (msg MessageCompo) MD() MessageCompo {
|
|
return msg.withParseMode(tgbotapi.ModeMarkdown)
|
|
}
|
|
|
|
// Set the Markdown 2 parsing mode.
|
|
func (msg MessageCompo) MD2() MessageCompo {
|
|
return msg.withParseMode(tgbotapi.ModeMarkdownV2)
|
|
}
|
|
|
|
// Set the HTML parsing mode.
|
|
func (msg MessageCompo) HTML() MessageCompo {
|
|
return msg.withParseMode(tgbotapi.ModeHTML)
|
|
}
|
|
|
|
// Transform the message component into one with reply keyboard.
|
|
func (msg MessageCompo) Inline(inline Inline) InlineCompo {
|
|
return InlineCompo{
|
|
Inline: inline,
|
|
MessageCompo: msg,
|
|
}
|
|
}
|
|
|
|
// Transform the message component into one with reply keyboard.
|
|
func (msg MessageCompo) Reply(reply Reply) ReplyCompo {
|
|
return ReplyCompo{
|
|
Reply: reply,
|
|
MessageCompo: msg,
|
|
}
|
|
}
|
|
|
|
// Transform the message component into the location one.
|
|
func (msg MessageCompo) Location(
|
|
lat, long float64,
|
|
) LocationCompo {
|
|
ret := &LocationCompo{
|
|
MessageCompo: msg,
|
|
Location: Location{
|
|
Latitude: lat,
|
|
Longitude: long,
|
|
},
|
|
}
|
|
return ret
|
|
}
|
|
|
|
// Implementing the Sendable interface.
|
|
func (config MessageCompo) SendConfig(
|
|
sid SessionId, bot *Bot,
|
|
) (SendConfig) {
|
|
var (
|
|
ret SendConfig
|
|
text string
|
|
)
|
|
|
|
if config.Text == "" {
|
|
text = ">"
|
|
} else {
|
|
text = config.Text
|
|
}
|
|
|
|
msg := tgbotapi.NewMessage(sid.ToApi(), text)
|
|
msg.ParseMode = config.ParseMode
|
|
ret.Chattable = msg
|
|
|
|
return ret
|
|
}
|
|
|
|
// Empty serving to use messages in rendering.
|
|
func (compo *MessageCompo) Serve(c Context) {}
|
|
|
|
// Filter that skips everything. Messages cannot do anything with updates.
|
|
func (compo *MessageCompo) Filter(_ Update) bool {
|
|
// Skip everything
|
|
return true
|
|
}
|