Joren/SolMonitor
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

ImplMessageMonitor #2

Merged
Joren merged 4 commits from ImplMessageMonitor into main 2024-06-27 12:42:37 +02:00
Conversation 0 Commits 4 Files Changed 3 +28 -16
1 changed files with 28 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit c0076759a4 - Show all commits

44
main.go
View File

@ -5,13 +5,11 @@ import (
"os"
"regexp"
"github.com/BurntSushi/toml"
"github.com/bwmarrin/discordgo"
"git.directme.in/Joren/SmsHook/ringbuffer"
)
var (
Token string
Servers []ServerConfig
@ -21,14 +19,15 @@ var (
)
type ServerConfig struct {
ServerID string `toml:"SERVER_ID"`
OutputChannelID string `toml:"OUTPUT_CHANNEL_ID"`
CoinRegexes map[string]string `toml:"COIN_REGEXES"`
ServerID string `toml:"SERVER_ID"`
OutputChannelID string `toml:"OUTPUT_CHANNEL_ID"`
CoinRegexes map[string]string `toml:"COIN_REGEXES"`
ChannelBlacklist struct {
Channels []string `toml:"CHANNELS"`
} `toml:"CHANNEL_BLACKLIST"`
PrefixEnabled bool `toml:"PREFIX_ENABLED"`
AllowDuplicates bool `toml:"ALLOW_DUPLICATES"`
PrefixEnabled bool `toml:"PREFIX_ENABLED"`
AllowDuplicates bool `toml:"ALLOW_DUPLICATES"`
ServerIDEnabled bool `toml:"SERVER_ID_ENABLED"`
}
func init() {
@ -110,19 +109,25 @@ func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
return
}
checkMessageContent(s, server, m)
checkEmbeds(s, server, m)
guild, err := s.Guild(m.GuildID)
if err != nil {
fmt.Println("Error getting guild:", err)
return
}
checkMessageContent(s, server, m, guild.Name)
checkEmbeds(s, server, m, guild.Name)
return
}
}
}
func checkMessageContent(s *discordgo.Session, server ServerConfig, m *discordgo.MessageCreate) {
func checkMessageContent(s *discordgo.Session, server ServerConfig, m *discordgo.MessageCreate, serverName string) {
for coin, regex := range CoinRegexes {
match := regex.FindStringSubmatch(m.Content)
if len(match) > 0 {
matchedAddress := match[0]
message := formatMessage(server, coin, m.Author.Username, matchedAddress)
message := formatMessage(server, coin, m.Author.Username, matchedAddress, serverName)
if !server.AllowDuplicates && MessageHistory[server.ServerID].ContainsItem(message) {
return
@ -134,14 +139,14 @@ func checkMessageContent(s *discordgo.Session, server ServerConfig, m *discordgo
}
}
func checkEmbeds(s *discordgo.Session, server ServerConfig, m *discordgo.MessageCreate) {
func checkEmbeds(s *discordgo.Session, server ServerConfig, m *discordgo.MessageCreate, serverName string) {
for _, embed := range m.Message.Embeds {
if embed.Type == "rich" && embed.Description != "" {
for coin, regex := range CoinRegexes {
match := regex.FindStringSubmatch(embed.Description)
if len(match) > 0 {
matchedAddress := match[0]
message := formatMessage(server, coin, m.Author.Username, matchedAddress)
message := formatMessage(server, coin, m.Author.Username, matchedAddress, serverName)
if !server.AllowDuplicates && MessageHistory[server.ServerID].ContainsItem(message) {
return
@ -155,11 +160,18 @@ func checkEmbeds(s *discordgo.Session, server ServerConfig, m *discordgo.Message
}
}
func formatMessage(server ServerConfig, coin, username, address string) string {
func formatMessage(server ServerConfig, coin, username, address, serverName string) string {
message := address
if server.PrefixEnabled {
return fmt.Sprintf("%s: %s", coin, address)
message = fmt.Sprintf("%s: %s", coin, address)
}
return address
if server.ServerIDEnabled {
message = fmt.Sprintf("%s (Server: %s)", message, serverName)
}
return message
}
func isChannelBlacklisted(serverID, channelID string) bool {