WayfinderAK
3b0bf0866a
All checks were successful
Build and Push Container / docker (push) Successful in 24m19s
84 lines
2.2 KiB
Go
84 lines
2.2 KiB
Go
package messages
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"regexp"
|
|
"strings"
|
|
neturl "net/url"
|
|
|
|
"github.com/dustinpianalto/discordgo"
|
|
)
|
|
|
|
func CleanAmazonURLs(s *discordgo.Session, message *discordgo.MessageCreate) {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
log.Println("Recovered from panic in CleanAmazonURLs")
|
|
}
|
|
}()
|
|
if message.Author.Bot {
|
|
return
|
|
}
|
|
content := message.Content
|
|
amazonRegexString := `(http|https):\/\/((?:[\w-_\.]*)amazon(?:\.[\w\-_]+)+)([\w\/\-\.,@?^=%&~\+#]*[\w\-\@?^=%&/~\+#])?`
|
|
amazonRegex := regexp.MustCompile(amazonRegexString)
|
|
urls := amazonRegex.FindAllString(message.Content, -1)
|
|
if len(urls) != 0 {
|
|
|
|
for _, url := range urls {
|
|
u, _ := neturl.Parse(url)
|
|
u.RawQuery = ""
|
|
u.Fragment = ""
|
|
if ( url != u.String() ) {
|
|
content = strings.ReplaceAll(content, url, u.String())
|
|
}
|
|
}
|
|
_, err := sendWebhook(s, message, message.ChannelID, content)
|
|
if err == nil {
|
|
s.ChannelMessageDelete(message.ChannelID, message.ID)
|
|
}
|
|
}
|
|
message.Content = content
|
|
urlRegexString := `http[s]?:\/\/(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+`
|
|
urlRegex := regexp.MustCompile(urlRegexString)
|
|
urls = urlRegex.FindAllString(message.Content, -1)
|
|
if len(urls) != 0 {
|
|
moveNewsLinks(s, message)
|
|
}
|
|
}
|
|
|
|
func moveNewsLinks(s *discordgo.Session, message *discordgo.MessageCreate) {
|
|
linkChannel := os.Getenv("LINK_CHANNEL")
|
|
chatChannel := os.Getenv("CHAT_CHANNEL")
|
|
if message.ChannelID == chatChannel {
|
|
_, err := sendWebhook(s, message, linkChannel, message.Content)
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func sendWebhook(s *discordgo.Session, message *discordgo.MessageCreate, channelID, content string) (string, error) {
|
|
webhook, err := s.WebhookCreate(channelID, message.ID, "")
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer s.WebhookDelete(webhook.ID)
|
|
var name string
|
|
if message.Member != nil && message.Member.Nick != "" {
|
|
name = message.Member.Nick
|
|
} else {
|
|
name = message.Author.Username
|
|
}
|
|
params := &discordgo.WebhookParams{
|
|
Content: content,
|
|
Username: name,
|
|
AvatarURL: message.Author.AvatarURL(""),
|
|
}
|
|
w, err := s.WebhookExecute(webhook.ID, webhook.Token, true, params)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return w.ID, nil
|
|
}
|