123 lines
2.4 KiB
Go
123 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
openai "github.com/sashabaranov/go-openai"
|
|
"gitlab.com/DeveloperDurp/durpot/handlers"
|
|
"gitlab.com/DeveloperDurp/durpot/model"
|
|
)
|
|
|
|
var (
|
|
Token = model.Token
|
|
BotPrefix = model.BotPrefix
|
|
ChannelID = model.ChannelID
|
|
BotId string
|
|
ApiKey = model.ApiKey
|
|
)
|
|
|
|
func Start() {
|
|
goBot, err := discordgo.New("Bot " + Token)
|
|
|
|
if err != nil {
|
|
fmt.Println(err.Error())
|
|
return
|
|
}
|
|
|
|
u, err := goBot.User("@me")
|
|
|
|
if err != nil {
|
|
fmt.Println(err.Error())
|
|
return
|
|
}
|
|
|
|
BotId = u.ID
|
|
|
|
goBot.AddHandler(messageHandler)
|
|
goBot.AddHandler(handlers.GuildMemberAdd)
|
|
goBot.AddHandler(handlers.GuildMemberRemove)
|
|
goBot.AddHandler(handleTag)
|
|
|
|
err = goBot.Open()
|
|
|
|
if err != nil {
|
|
fmt.Println(err.Error())
|
|
return
|
|
}
|
|
fmt.Println("Bot is running!")
|
|
}
|
|
|
|
func messageHandler(s *discordgo.Session, m *discordgo.MessageCreate) {
|
|
if m.Author.ID == BotId {
|
|
return
|
|
}
|
|
|
|
//baseurl := "https://kong.durp.info/"
|
|
switch m.Content {
|
|
case BotPrefix + "ping":
|
|
_, err := s.ChannelMessageSend(m.ChannelID, "pong")
|
|
if err != nil {
|
|
fmt.Println("Failed to send Message")
|
|
}
|
|
case BotPrefix + "unraid":
|
|
handlers.GetUnraidUsage(s, m)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
func main() {
|
|
|
|
Start()
|
|
|
|
<-make(chan struct{})
|
|
}
|
|
|
|
func handleTag(s *discordgo.Session, m *discordgo.MessageCreate) {
|
|
// Ignore messages sent by the bot itself
|
|
if m.Author.ID == s.State.User.ID {
|
|
return
|
|
}
|
|
|
|
client := openai.NewClient(ApiKey)
|
|
|
|
// Check if bot is mentioned in message
|
|
for _, mention := range m.Mentions {
|
|
if mention.ID == s.State.User.ID {
|
|
// Remove mention from message content
|
|
content := strings.Replace(m.ContentWithMentionsReplaced(), "<@"+s.State.User.ID+">", "", -1)
|
|
content = strings.Replace(content, "<@!"+s.State.User.ID+">", "", -1)
|
|
content = strings.TrimSpace(content)
|
|
|
|
resp, err := client.CreateChatCompletion(
|
|
context.Background(),
|
|
openai.ChatCompletionRequest{
|
|
Model: openai.GPT3Dot5Turbo,
|
|
Messages: []openai.ChatCompletionMessage{
|
|
{
|
|
Role: openai.ChatMessageRoleUser,
|
|
Content: content,
|
|
},
|
|
},
|
|
},
|
|
)
|
|
|
|
if err != nil {
|
|
fmt.Printf("ChatCompletion error: %v\n", err)
|
|
return
|
|
}
|
|
|
|
fmt.Println(resp.Choices[0].Message.Content)
|
|
|
|
// Send generated response back to Discord
|
|
_, err = s.ChannelMessageSend(m.ChannelID, resp.Choices[0].Message.Content)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
}
|
|
}
|
|
}
|