Compare commits
3 Commits
afada3e970
...
50513cf3cd
Author | SHA1 | Date | |
---|---|---|---|
50513cf3cd | |||
22d05c4807 | |||
96a33f61d7 |
84
main.go
84
main.go
@ -60,6 +60,8 @@ func main() {
|
||||
// Register messageCreate as a callback for the messageCreate events
|
||||
dg.AddHandler(messageCreate)
|
||||
|
||||
dg.AddHandler(message2Create)
|
||||
|
||||
// Open a websocket connection to Discord and begin listening
|
||||
err = dg.Open()
|
||||
if err != nil {
|
||||
@ -114,35 +116,62 @@ func scrapeNews() string {
|
||||
return html
|
||||
}
|
||||
|
||||
// Function to scrape the webpage and send a message if the date matches
|
||||
func sendNotifications(session *discordgo.Session, channelID string, html string) {
|
||||
// Parse the HTML content
|
||||
// Add a function to extract relevant tags from the HTML content
|
||||
func extractTags(html string) []string {
|
||||
var tags []string
|
||||
doc, err := goquery.NewDocumentFromReader(strings.NewReader(html))
|
||||
if err != nil {
|
||||
fmt.Println("Error parsing HTML:", err)
|
||||
return
|
||||
}
|
||||
|
||||
// Find and extract the date from the webpage
|
||||
var foundDate bool
|
||||
doc.Find("span.news-module-feed-item-details-date").Each(func(i int, s *goquery.Selection) {
|
||||
dateString := strings.TrimSpace(s.Text())
|
||||
fmt.Println("Found date:", dateString)
|
||||
if dateString == time.Now().Format("January 2, 2006") {
|
||||
foundDate = true
|
||||
return tags
|
||||
}
|
||||
// Find and extract tags with class name "news-module-feed-item-details-tag"
|
||||
doc.Find("span.news-module-feed-item-details-tag").Each(func(i int, s *goquery.Selection) {
|
||||
tag := strings.TrimSpace(s.Text())
|
||||
tags = append(tags, tag)
|
||||
})
|
||||
return tags
|
||||
}
|
||||
|
||||
// If today's date was found on the webpage, send a message to the Discord channel
|
||||
if foundDate {
|
||||
_, err := session.ChannelMessageSend(channelID, "Today's date was found on the Fallout news page!")
|
||||
// Define a map to associate each desired tag with its corresponding Discord channel ID
|
||||
var tagChannelMap = map[string]string{
|
||||
"Patch Notes": "556093419341086749",
|
||||
"Atomic Shop": "590581442392621067",
|
||||
"News": "558335339018846228",
|
||||
// Add more tags and corresponding channel IDs as needed
|
||||
}
|
||||
|
||||
func sendNotifications(session *discordgo.Session, channelID string, html string) {
|
||||
// Extract tags from the HTML content
|
||||
tags := extractTags(html)
|
||||
|
||||
// Define today's date string
|
||||
today := time.Now().Format("January 2, 2006")
|
||||
|
||||
// Iterate over extracted tags
|
||||
for _, tag := range tags {
|
||||
// Check if the tag is in the desired tags list
|
||||
if tagChannelID, ok := tagChannelMap[tag]; ok {
|
||||
// Check if today's date matches the tag
|
||||
if tag == today {
|
||||
// Send a message to the corresponding Discord channel
|
||||
message := fmt.Sprintf("Today's date matches the tag '%s' on the Fallout news page!", tag)
|
||||
_, err := session.ChannelMessageSend(tagChannelID, message)
|
||||
if err != nil {
|
||||
fmt.Println("Error sending message to Discord:", err)
|
||||
return
|
||||
fmt.Printf("Error sending message to Discord channel %s: %s\n", tagChannelID, err)
|
||||
continue
|
||||
}
|
||||
fmt.Println("Message sent to Discord channel")
|
||||
fmt.Println("Message sent to Discord channel:", tagChannelID)
|
||||
} else {
|
||||
fmt.Println("Today's date not found on the Fallout news page")
|
||||
// Send a different message indicating the tag was found
|
||||
message := fmt.Sprintf("Tag '%s' found on the Fallout news page, but it's not today's date.", tag)
|
||||
_, err := session.ChannelMessageSend(tagChannelID, message)
|
||||
if err != nil {
|
||||
fmt.Printf("Error sending message to Discord channel %s: %s\n", tagChannelID, err)
|
||||
continue
|
||||
}
|
||||
fmt.Println("Message sent to Discord channel:", tagChannelID)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@ -154,7 +183,20 @@ func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
|
||||
}
|
||||
|
||||
// If the message content is "!hello", respond with "Hello!"
|
||||
if m.Content == "!hello" {
|
||||
if m.Content == "!hello1" {
|
||||
_, _ = s.ChannelMessageSend(m.ChannelID, "Hello!")
|
||||
}
|
||||
}
|
||||
|
||||
// This function will be called whenever a new message is created
|
||||
func message2Create(s *discordgo.Session, m *discordgo.MessageCreate) {
|
||||
// Ignore messages created by the bot itself
|
||||
if m.Author.ID == s.State.User.ID {
|
||||
return
|
||||
}
|
||||
|
||||
// If the message content is "!hello", respond with "Hello!"
|
||||
if m.Content == "!hello" {
|
||||
_, _ = s.ChannelMessageSend(m.ChannelID, "You are SUCH a NERD")
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user