commit d745d449d5e0c173351c70eaff06a52537f24e4d Author: Guillaume Dorce Date: Sun Sep 22 18:07:21 2024 +0200 batman diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..a703a4a --- /dev/null +++ b/.gitignore @@ -0,0 +1,2 @@ +.env +tmp/ diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..d39bca9 --- /dev/null +++ b/go.mod @@ -0,0 +1,22 @@ +module polynux/disgoroq + +go 1.23.0 + +toolchain go1.23.1 + +require ( + github.com/bahlo/generic-list-go v0.2.0 // indirect + github.com/buger/jsonparser v1.1.1 // indirect + github.com/bwmarrin/discordgo v0.28.1 // indirect + github.com/conneroisu/groq-go v0.9.2 // indirect + github.com/gorilla/websocket v1.4.2 // indirect + github.com/joho/godotenv v1.5.1 // indirect + github.com/mailru/easyjson v0.7.7 // indirect + github.com/mattn/go-colorable v0.1.13 // indirect + github.com/mattn/go-isatty v0.0.20 // indirect + github.com/rs/zerolog v1.33.0 // indirect + github.com/wk8/go-ordered-map/v2 v2.1.8 // indirect + golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b // indirect + golang.org/x/sys v0.25.0 // indirect + gopkg.in/yaml.v3 v3.0.1 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..8dc765d --- /dev/null +++ b/go.sum @@ -0,0 +1,45 @@ +github.com/bahlo/generic-list-go v0.2.0 h1:5sz/EEAK+ls5wF+NeqDpk5+iNdMDXrh3z3nPnH1Wvgk= +github.com/bahlo/generic-list-go v0.2.0/go.mod h1:2KvAjgMlE5NNynlg/5iLrrCCZ2+5xWbdbCW3pNTGyYg= +github.com/buger/jsonparser v1.1.1 h1:2PnMjfWD7wBILjqQbt530v576A/cAbQvEW9gGIpYMUs= +github.com/buger/jsonparser v1.1.1/go.mod h1:6RYKKt7H4d4+iWqouImQ9R2FZql3VbhNgx27UK13J/0= +github.com/bwmarrin/discordgo v0.28.1 h1:gXsuo2GBO7NbR6uqmrrBDplPUx2T3nzu775q/Rd1aG4= +github.com/bwmarrin/discordgo v0.28.1/go.mod h1:NJZpH+1AfhIcyQsPeuBKsUtYrRnjkyu0kIVMCHkZtRY= +github.com/conneroisu/groq-go v0.9.2 h1:84kA30fYN5EIzeXBYWJjcmDHs06MpA+88fR1d5wgQUg= +github.com/conneroisu/groq-go v0.9.2/go.mod h1:lgL4jd3iBuw+hAJlbdFEelUsPAdLLt5N0ir1E2x18zc= +github.com/coreos/go-systemd/v22 v22.5.0/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc= +github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA= +github.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc= +github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE= +github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0= +github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4= +github.com/josharian/intern v1.0.0/go.mod h1:5DoeVV0s6jJacbCEi61lwdGj/aVlrQvzHFFd8Hwg//Y= +github.com/mailru/easyjson v0.7.7 h1:UGYAvKxe3sBsEDzO8ZeWOSlIQfWFlxbzLZe7hwFURr0= +github.com/mailru/easyjson v0.7.7/go.mod h1:xzfreul335JAWq5oZzymOObrkdz5UnU4kGfJJLY9Nlc= +github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA= +github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg= +github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM= +github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= +github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= +github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= +github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= +github.com/rs/xid v1.5.0/go.mod h1:trrq9SKmegXys3aeAKXMUTdJsYXVwGY3RLcfgqegfbg= +github.com/rs/zerolog v1.33.0 h1:1cU2KZkvPxNyfgEmhHAz/1A9Bz+llsdYzklWFzgp0r8= +github.com/rs/zerolog v1.33.0/go.mod h1:/7mN4D5sKwJLZQ2b/znpjC3/GQWY/xaDXUM0kKWRHss= +github.com/wk8/go-ordered-map/v2 v2.1.8 h1:5h/BUHu93oj4gIdvHHHGsScSTMijfx5PeYkE/fJgbpc= +github.com/wk8/go-ordered-map/v2 v2.1.8/go.mod h1:5nJHM5DyteebpVlHnWMV0rPz6Zp7+xBAnxjb1X5vnTw= +golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b h1:7mWr3k41Qtv8XlltBkDkl8LoP3mpSgBW8BUoxtEdbXg= +golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b/go.mod h1:T9bdIzuCu7OtxOm1hfPfRQxPLYneinmdGuTeoZ9dtd4= +golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68 h1:nxC68pudNYkKU6jWhgrqdreuFiOQWj1Fs7T3VrH4Pjw= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.12.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.25.0 h1:r+8e+loiHxRqhXVl6ML1nO3l1+oFoWbnlu2Ehimmi34= +golang.org/x/sys v0.25.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= +golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/main.go b/main.go new file mode 100644 index 0000000..53ef5e3 --- /dev/null +++ b/main.go @@ -0,0 +1,165 @@ +package main + +import ( + "fmt" + "context" + "github.com/bwmarrin/discordgo" + "os" + "log" + "os/signal" + "syscall" + "math/rand" + "github.com/conneroisu/groq-go" + "github.com/joho/godotenv" +) + +var ( + Token string + GroqKey string +) + +func init() { + err := godotenv.Load() + if err != nil { + log.Fatal("Error loading .env file") + } + + Token = os.Getenv("DISCORD_TOKEN") + GroqKey = os.Getenv("GROQ_API_KEY") + + if Token == "" { + log.Fatal("No discord token found in .env file") + } + + if GroqKey == "" { + log.Fatal("No Groq key found in .env file") + } +} + +func main() { + dg, err := discordgo.New("Bot " + Token) + if err != nil { + log.Fatal("Error creating Discord session,", err) + return + } + + dg.AddHandler(messageCreate) + + dg.Identify.Intents = discordgo.IntentsGuildMessages + + err = dg.Open() + if err != nil { + log.Fatal("Error opening connection,", err) + return + } + + log.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 + + dg.Close() +} + +func getMessages(s *discordgo.Session, channelID string, num int) ([]*discordgo.Message, error) { + if (num <= 100) { + messages, err := s.ChannelMessages(channelID, num, "", "", "") + if err != nil { + log.Println("error getting messages,", err) + return nil, err + } + return messages, nil + } + + messages := []*discordgo.Message{} + for num > 0 { + var toGet int + if num > 100 { + toGet = 100 + } else { + toGet = num + } + lastMessage := "" + if len(messages) > 0 { + lastMessage = messages[len(messages)-1].ID + } + newMessages, err := s.ChannelMessages(channelID, toGet, lastMessage, "", "") + if err != nil { + fmt.Println("error getting messages,", err) + return nil, err + } + messages = append(messages, newMessages...) + num -= toGet + } + return messages, nil +} + +func botMentioned(s *discordgo.Session, m *discordgo.MessageCreate) bool { + for i := range m.Mentions { + if m.Mentions[i].ID == s.State.User.ID { + return true + } + } + return false +} + +func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) { + if m.Author.ID == s.State.User.ID { + return + } + + rand := rand.Intn(100) + + if rand > 10 && !botMentioned(s, m){ + return + } + + messages, err := getMessages(s, m.ChannelID, 100) + if err != nil { + fmt.Println("error getting messages,", err) + return + } + messagesFormatted := "" + for i := range messages { + if messages[i].Author.ID == s.State.User.ID { + continue + } + messagesFormatted = messagesFormatted + "<" + messages[i].Author.ID + "> " + messages[i].Content + "\n" + } + + question := "Tu fais partie d'une conversation. Tu n'as pas conscience d'etre un bot. Reponds de maniere goofy. Ta reponse doit etre tres courte, une phrase ou deux.\n" + question += "\n" + messagesFormatted + "\n" + s.ChannelMessageSend(m.ChannelID, "Thinking...") + response, err := askGroq(context.Background(), question) + if err != nil { + s.ChannelMessageSend(m.ChannelID, "There was an error getting the response.") + return + } + s.ChannelMessageSend(m.ChannelID, response) +} + +func askGroq(ctx context.Context, message string) (string,error) { + client, err := groq.NewClient(GroqKey) + if err != nil { + fmt.Println("error creating Groq client,", err) + return "", err + } + + resp, err := client.CreateChatCompletion(ctx, groq.ChatCompletionRequest{ + Model: groq.Llama318BInstant, + Messages: []groq.ChatCompletionMessage{ + { + Role: groq.ChatMessageRoleUser, + Content: message, + }, + }, + MaxTokens: 100, + }) + if err != nil { + fmt.Println("error creating Groq completion,", err) + return "", err + } + + + return string(resp.Choices[0].Message.Content), nil +}