135 lines
2.9 KiB
Go
135 lines
2.9 KiB
Go
package tg
|
|
|
|
import (
|
|
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
|
|
//"strings"
|
|
re "regexp"
|
|
)
|
|
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")))
|
|
}
|
|
|
|
func (compo *MessageCompo) Update(c *Context) {
|
|
edit := tgbotapi.NewEditMessageText(
|
|
c.Session.Id.ToApi(),
|
|
compo.Message.MessageID,
|
|
compo.Text,
|
|
)
|
|
msg, _ := c.Bot.Api.Send(edit)
|
|
compo.Message = &msg
|
|
}
|
|
|
|
// 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 NewMessage(text string) *MessageCompo {
|
|
ret := &MessageCompo{}
|
|
ret.Text = text
|
|
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
|
|
}
|
|
|
|
//text = strings.ReplaceAll(text, "-", "\\-")
|
|
|
|
msg := tgbotapi.NewMessage(sid.ToApi(), text)
|
|
ret.Message = &msg
|
|
ret.Message.ParseMode = config.ParseMode
|
|
|
|
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
|
|
}
|