swiftspiracy-bot/cmd/bot/main.go

package main

import (
	"embed"
	"encoding/json"
	"fmt"
	"math/rand"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/bwmarrin/discordgo"
	"github.com/joho/godotenv"
)

//go:embed praises.json
var praisesFile embed.FS

//go:embed conspiracies.json
var conspiraciesFile embed.FS

var (
	praises      []string
	conspiracies []string
)

func main() {
	err := godotenv.Load()
	if err != nil {
		fmt.Println("Error loading .env file")
		return
	}

	botToken := os.Getenv("DISCORD_BOT_TOKEN")
	channelID := os.Getenv("DISCORD_CHANNEL_ID")
	if botToken == "" || channelID == "" {
		fmt.Println("Missing DISCORD_BOT_TOKEN or DISCORD_CHANNEL_ID in .env")
		return
	}

	// load messages at build time
	err = loadMessages()
	if err != nil {
		fmt.Println("Error loading embedded messages: %v", err)
		return
	}

	// initialize bot session
	dg, err := discordgo.New("Bot " + botToken)
	if err != nil {
		fmt.Println("Error creating Discord session,", err)
		return
	}

	// open WebSocket connection
	dg.AddHandler(readyHandler)
	err = dg.Open()
	if err != nil {
		fmt.Println("Error opening connection,", err)
		return
	}

	// start the scheduled messages
	go sendScheduledMessages(dg, channelID)

	// wait here until CTRL-C or other term signal is received.
	fmt.Println("Bot is now running.  Press CTRL-C to exit.")
	sc := make(chan os.Signal, 1)
	signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt)
	<-sc

	// cleanly close down the Discord session
	dg.Close()
}

func loadMessages() error {
	file, err := praisesFile.ReadFile("praises.json")
	if err != nil {
		return err
	}

	err = json.Unmarshal(file, &praises)
	if err != nil {
		return err
	}

	file, err = conspiraciesFile.ReadFile("conspiracies.json")
	if err != nil {
		return err
	}

	err = json.Unmarshal(file, &conspiracies)
	if err != nil {
		return err
	}

	return nil
}

func readyHandler(s *discordgo.Session, r *discordgo.Ready) {
	fmt.Println("Ready to rumble!")
}

func sendScheduledMessages(s *discordgo.Session, channelID string) {
	// TODO: change time after debugging
	ticker := time.NewTicker(time.Second * 10)
	defer ticker.Stop()

	for range ticker.C {
		sendMessage(praises, s, channelID)

		// 30% chance to send a conspiracy theory
		if rand.Float32() < 0.3 {
			discordMessage := sendMessage(conspiracies, s, channelID)

			// schedule message deletion after 5 minutes
			go func(messageID string) {
				// TODO: change time after debugging
				time.Sleep(5 * time.Second)
				err := s.ChannelMessageDelete(channelID, messageID)
				if err != nil {
					fmt.Printf("Error deleting conspiracy: %v\n", err)
				} else {
					fmt.Println("Conspiracy deleted!")
				}
			}(discordMessage.ID)
		}
	}
}

func sendMessage(messages []string, s *discordgo.Session, channelID string) *discordgo.Message {
	if len(messages) == 0 {
		fmt.Printf("No messages available.")
		return nil
	}
	message := messages[rand.Intn(len(messages))]

	discordMessage, err := s.ChannelMessageSend(channelID, message)
	if err != nil {
		fmt.Printf("Error sending message: %v\n", err)
		return nil
	} else {
		fmt.Println("Message sent!")
	}
	return discordMessage
}