58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
|
package bot
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/mymmrac/telego"
|
||
|
|
"github.com/mymmrac/telego/telegoutil"
|
||
|
|
"log/slog"
|
||
|
|
"net/url"
|
||
|
|
"slices"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
allowedUrlSchemes = []string{"http", "https"}
|
||
|
|
)
|
||
|
|
|
||
|
|
func (b *Bot) reply(originalMessage *telego.Message, newMessage *telego.SendMessageParams) *telego.SendMessageParams {
|
||
|
|
return newMessage.WithReplyParameters(&telego.ReplyParameters{
|
||
|
|
MessageID: originalMessage.MessageID,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Bot) sendTyping(chatId telego.ChatID) {
|
||
|
|
slog.Debug("Setting 'typing' chat action")
|
||
|
|
|
||
|
|
err := b.api.SendChatAction(telegoutil.ChatAction(chatId, "typing"))
|
||
|
|
if err != nil {
|
||
|
|
slog.Error("Cannot set chat action", "error", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Bot) trySendReplyError(message *telego.Message) {
|
||
|
|
if message == nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
_, _ = b.api.SendMessage(b.reply(message, telegoutil.Message(
|
||
|
|
telegoutil.ID(message.Chat.ID),
|
||
|
|
"Error occurred while trying to send reply.",
|
||
|
|
)))
|
||
|
|
}
|
||
|
|
|
||
|
|
func isValidAndAllowedUrl(text string) bool {
|
||
|
|
u, err := url.ParseRequestURI(text)
|
||
|
|
if err != nil {
|
||
|
|
slog.Debug("Provided text is not an URL", "text", text)
|
||
|
|
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
if !slices.Contains(allowedUrlSchemes, strings.ToLower(u.Scheme)) {
|
||
|
|
slog.Debug("Provided URL has disallowed scheme", "scheme", u.Scheme, "allowed-schemes", allowedUrlSchemes)
|
||
|
|
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
return true
|
||
|
|
}
|