move main.go

cmd/bb/main.go

@@ -0,0 +1,238 @@
+package main
+
+import (
+	"encoding/binary"
+	"flag"
+	"fmt"
+
+	//"log"
+	crypto_rand "crypto/rand"
+	"math/rand"
+	math_rand "math/rand"
+	"os"
+	"os/signal"
+	"strings"
+	"syscall"
+
+	"git.kill0.net/chill9/beepboop/command"
+	"github.com/bwmarrin/discordgo"
+	log "github.com/sirupsen/logrus"
+
+	// log "github.com/golang/glog"
+	"github.com/spf13/pflag"
+	"github.com/spf13/viper"
+)
+
+var (
+	Token string
+
+	defaultReactions []string = []string{"👍", "🌶️", "🤣", "😂", "🍆", "🍑", "❤️", "💦", "😍", "💩", "🔥", "🍒", "🎉", "🥳", "🎊"}
+
+	commands = []*discordgo.ApplicationCommand{
+		{
+			Name:        "poop",
+			Type:        discordgo.ChatApplicationCommand,
+			Description: "Hot and steamy",
+		},
+		{
+			Name:        "ping",
+			Type:        discordgo.ChatApplicationCommand,
+			Description: "Ping the bot",
+		},
+		{
+			Name:        "roll",
+			Type:        discordgo.ChatApplicationCommand,
+			Description: "Roll a dice",
+			Options: []*discordgo.ApplicationCommandOption{
+				{
+					Type:        discordgo.ApplicationCommandOptionString,
+					Name:        "dice",
+					Description: "Dice specification e.g. d4 or 2d6",
+					Required:    true,
+				},
+			},
+		},
+		{
+			Name:        "coin",
+			Type:        discordgo.ChatApplicationCommand,
+			Description: "Flip a coin",
+		},
+	}
+
+	commandHandlers = map[string]func(s *discordgo.Session, i *discordgo.InteractionCreate){
+		"ping": command.PingCommand,
+		"poop": command.PoopCommand,
+		"roll": command.RollCommand,
+		"coin": command.CoinCommand,
+	}
+
+	config Config
+)
+
+type (
+	Config struct {
+		Handler HandlerConfig `mapstructure:"handler"`
+	}
+
+	HandlerConfig struct {
+		Reaction ReactionConfig `mapstructure:"reaction"`
+	}
+
+	ReactionConfig struct {
+		Emojis   []string
+		Channels []string
+	}
+)
+
+func main() {
+	var (
+		err error
+	)
+
+	flag.Bool("debug", false, "enable debug logging")
+	pflag.CommandLine.AddGoFlagSet(flag.CommandLine)
+	pflag.Parse()
+	viper.BindPFlags(pflag.CommandLine)
+
+	if viper.GetBool("debug") {
+		log.SetLevel(log.DebugLevel)
+	}
+
+	seedRand()
+
+	viper.SetDefault("handler.reaction.emojis", defaultReactions)
+	viper.SetEnvPrefix("BEEPBOOP")
+	viper.AutomaticEnv()
+	viper.SetConfigName("config")
+	viper.SetConfigType("toml")
+	viper.AddConfigPath(".")
+	err = viper.ReadInConfig()
+
+	if err != nil {
+		log.Fatalf("fatal error config file: %v", err)
+	}
+
+	Token, ok := viper.Get("discord_token").(string)
+
+	err = viper.Unmarshal(&config)
+	if err != nil {
+		log.Fatalf("unable to decode into struct: %v", err)
+	}
+
+	if Token == "" {
+		log.Fatalf("Discord token is not set")
+	}
+
+	if !ok {
+		log.Fatalf("Invalid type assertion")
+	}
+
+	dg, err := discordgo.New(fmt.Sprintf("Bot %s", Token))
+	if err != nil {
+		log.Fatalf("error creating Discord session: %v\n", err)
+	}
+
+	dg.AddHandler(command.PingHandler)
+	dg.AddHandler(reactionHandler)
+	dg.AddHandler(praiseHandler)
+	dg.AddHandler(command.RollHandler)
+	dg.AddHandler(command.RouletteHandler)
+
+	dg.Identify.Intents = discordgo.IntentsGuildMessages
+
+	err = dg.Open()
+	if err != nil {
+		log.Fatalf("error opening connection: %v\n", err)
+	}
+
+	for _, c := range commands {
+		_, err := dg.ApplicationCommandCreate(dg.State.User.ID, "", c)
+		if err != nil {
+			log.Errorf("Cannot create '%v' command %v", c.Name, err)
+		}
+	}
+
+	dg.AddHandler(func(s *discordgo.Session, i *discordgo.InteractionCreate) {
+		if h, ok := commandHandlers[i.ApplicationCommandData().Name]; ok {
+			h(s, i)
+		}
+	})
+
+	log.Info("The bot is now running. Press CTRL-C to exit.")
+
+	defer dg.Close()
+
+	sc := make(chan os.Signal, 1)
+	signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt, os.Kill)
+	<-sc
+
+	log.Info("Shutting down")
+}
+
+func praiseHandler(s *discordgo.Session, m *discordgo.MessageCreate) {
+	if m.Author.ID == s.State.User.ID {
+		return
+	}
+
+	if strings.Contains(m.Content, "good bot") {
+		s.ChannelMessageSend(m.ChannelID, fmt.Sprintf("<@%s> Thank you, daddy.", m.Author.ID))
+	}
+}
+
+func reactionHandler(s *discordgo.Session, m *discordgo.MessageCreate) {
+	if m.Author.ID == s.State.User.ID {
+		return
+	}
+
+	emojis := config.Handler.Reaction.Emojis
+	channels := config.Handler.Reaction.Channels
+
+	if len(emojis) == 0 {
+		log.Warning("emoji list is empty")
+		return
+	}
+
+	channel, err := s.Channel(m.ChannelID)
+	if err != nil {
+		log.Fatalf("unable to get channel name: %v", err)
+	}
+
+	if len(channels) > 0 && !contains(channels, channel.Name) {
+		return
+	}
+
+	for _, a := range m.Attachments {
+		if strings.HasPrefix(a.ContentType, "image/") {
+			r := emojis[rand.Intn(len(emojis))]
+			s.MessageReactionAdd(m.ChannelID, m.ID, r)
+		}
+	}
+
+	for range m.Embeds {
+		r := emojis[rand.Intn(len(emojis))]
+		s.MessageReactionAdd(m.ChannelID, m.ID, r)
+	}
+
+}
+
+func contains[T comparable](s []T, v T) bool {
+	for _, x := range s {
+		if x == v {
+			return true
+		}
+	}
+	return false
+}
+
+func seedRand() {
+	var b [8]byte
+
+	_, err := crypto_rand.Read(b[:])
+	if err != nil {
+		log.Panicf("cannot seed math/rand: %s", err)
+	}
+
+	log.Debugf("seeding math/rand %+v %+v", b, binary.LittleEndian.Uint64(b[:]))
+
+	math_rand.Seed(int64(binary.LittleEndian.Uint64(b[:])))
+}