Add method to move links to link channel

main v0.0.5
DustyP 5 years ago
parent 311497086e
commit 70cf457e6f

@ -21,23 +21,43 @@ func CleanAmazonURLs(s *discordgo.Session, message *discordgo.MessageCreate) {
amazonRegexString := `(http|https):\/\/((?:[\w-_\.]*)amazon(?:\.[\w\-_]+)+)([\w\/\-\.,@?^=%&~\+#]*[\w\-\@?^=%&/~\+#])?` amazonRegexString := `(http|https):\/\/((?:[\w-_\.]*)amazon(?:\.[\w\-_]+)+)([\w\/\-\.,@?^=%&~\+#]*[\w\-\@?^=%&/~\+#])?`
amazonRegex := regexp.MustCompile(amazonRegexString) amazonRegex := regexp.MustCompile(amazonRegexString)
urls := amazonRegex.FindAllString(message.Content, -1) urls := amazonRegex.FindAllString(message.Content, -1)
if len(urls) == 0 { if len(urls) != 0 {
return
} for _, url := range urls {
for _, url := range urls { if strings.Contains(url, "ref=") || strings.Contains(url, "?") {
if strings.Contains(url, "ref=") || strings.Contains(url, "?") { parts := strings.Split(url, "/")
parts := strings.Split(url, "/") new := strings.Join(parts[:len(parts)-1], "/")
new := strings.Join(parts[:len(parts)-1], "/") if strings.Contains(new, "ref=") {
if strings.Contains(new, "ref=") { parts = strings.Split(new, "ref=")
parts = strings.Split(new, "ref=") new = parts[0]
new = parts[0] }
content = strings.ReplaceAll(content, url, new)
} }
content = strings.ReplaceAll(content, url, new) }
_, err := sendWebhook(s, message, message.ChannelID, content)
if err == nil {
s.ChannelMessageDelete(message.ChannelID, message.ID)
}
}
message.Content = content
moveNewsLinks(s, message)
}
func moveNewsLinks(s *discordgo.Session, message *discordgo.MessageCreate) {
linkChannel := "784855106239332352"
chatChannel := "777679317878177804"
if message.ChannelID == chatChannel {
_, err := sendWebhook(s, message, linkChannel, message.Content)
if err != nil {
log.Println(err)
} }
} }
webhook, err := s.WebhookCreate(message.ChannelID, message.ID, "") }
func sendWebhook(s *discordgo.Session, message *discordgo.MessageCreate, channelID, content string) (string, error) {
webhook, err := s.WebhookCreate(channelID, message.ID, "")
if err != nil { if err != nil {
return return "", err
} }
defer s.WebhookDelete(webhook.ID) defer s.WebhookDelete(webhook.ID)
var name string var name string
@ -51,9 +71,9 @@ func CleanAmazonURLs(s *discordgo.Session, message *discordgo.MessageCreate) {
Username: name, Username: name,
AvatarURL: message.Author.AvatarURL(""), AvatarURL: message.Author.AvatarURL(""),
} }
_, err = s.WebhookExecute(webhook.ID, webhook.Token, true, params) w, err := s.WebhookExecute(webhook.ID, webhook.Token, true, params)
if err != nil { if err != nil {
return return "", err
} }
s.ChannelMessageDelete(message.ChannelID, message.ID) return w.ID, nil
} }

Loading…
Cancel
Save