added !unraid and restructure

This commit is contained in:
2023-05-27 16:32:59 -05:00
parent 21ce2d0609
commit b2b50d303b
6 changed files with 278 additions and 179 deletions

171
main.go
View File

@@ -3,45 +3,22 @@ package main
import (
"context"
"fmt"
"log"
"os"
"strings"
"github.com/bwmarrin/discordgo"
"github.com/joho/godotenv"
openai "github.com/sashabaranov/go-openai"
"gitlab.com/DeveloperDurp/durpot/handlers"
"gitlab.com/DeveloperDurp/durpot/model"
)
var (
Token string
BotPrefix string
ChannelID string
Token = model.Token
BotPrefix = model.BotPrefix
ChannelID = model.ChannelID
BotId string
apiKey string
goBot *discordgo.Session
config *model.ConfigStruct
ApiKey = model.ApiKey
)
func ReadConfig() error {
err := godotenv.Load(".env")
if err != nil {
fmt.Println(err.Error())
//return err
}
Token = os.Getenv("TOKEN")
BotPrefix = os.Getenv("BOTPREFIX")
ChannelID = os.Getenv("ChannelID")
apiKey = os.Getenv("OPENAI_API_KEY")
return nil
}
func Start() {
goBot, err := discordgo.New("Bot " + Token)
@@ -60,8 +37,8 @@ func Start() {
BotId = u.ID
goBot.AddHandler(messageHandler)
goBot.AddHandler(handleGuildMemberAdd)
goBot.AddHandler(handleGuildMemberRemove)
goBot.AddHandler(handlers.GuildMemberAdd)
goBot.AddHandler(handlers.GuildMemberRemove)
goBot.AddHandler(handleTag)
err = goBot.Open()
@@ -85,146 +62,18 @@ func messageHandler(s *discordgo.Session, m *discordgo.MessageCreate) {
if err != nil {
fmt.Println("Failed to send Message")
}
//case BotPrefix + "meme":
// response = getJson(baseurl+"random-meme", "url")
// s.ChannelMessageSend(m.ChannelID, response)
//case BotPrefix + "catfact":
// response = getJson(baseurl+"cat-facts/fact", "fact")
// s.ChannelMessageSend(m.ChannelID, response)
//case BotPrefix + "cat":
// response = getJson(baseurl+"random-cats", "file")
// s.ChannelMessageSend(m.ChannelID, response)
//case BotPrefix + "yomama":
// sendAPIRequest(s, m, baseurl, "yomama")
//case BotPrefix + "dadjoke":
// sendAPIRequest(s, m, baseurl, "dadjoke")
//case BotPrefix + "dog":
// response = getJson(baseurl+"random-dogs", "message")
// s.ChannelMessageSend(m.ChannelID, response)
//case BotPrefix + "jinglebells":
// sendAPIRequest(s, m, baseurl, "jinglebells")
//case BotPrefix + "swanson":
// sendAPIRequest(s, m, baseurl, "swanson")
case BotPrefix + "unraid":
handlers.GetUnraidUsage(s, m)
}
}
func main() {
err := ReadConfig()
if err != nil {
fmt.Println(err.Error())
return
}
Start()
<-make(chan struct{})
return
}
//func getSwansonQuote(s *discordgo.Session, m *discordgo.MessageCreate, url string) {
// resp, err := http.Get(url + "/ronswanson")
// if err != nil {
// errStr := err.Error()
// s.ChannelMessageSend(m.ChannelID, errStr)
// return
// }
// defer resp.Body.Close()
//
// var data []string
// err = json.NewDecoder(resp.Body).Decode(&data)
// if err != nil {
// errStr := err.Error()
// s.ChannelMessageSend(m.ChannelID, errStr)
// return
// }
//
// if len(data) == 0 {
// errStr := "No quotes found."
// s.ChannelMessageSend(m.ChannelID, errStr)
// return
// }
//
// s.ChannelMessageSend(m.ChannelID, data[0])
//}
//func sendAPIRequest(s *discordgo.Session, m *discordgo.MessageCreate, url string, endpoint string) {
// var response interface{}
// switch endpoint {
// case "dadjoke":
// url = url + "/dadjoke"
// var data model.DadJokeResponse
// response = &data
// case "jinglebells":
// url = url + "/foaas/jinglebells/durp"
// var data model.JingleBellsResponse
// response = &data
// case "yomama":
// url = url + "/yomama"
// var data model.YomamaJokeResponse
// response = &data
// case "swanson":
// getSwansonQuote(s, m, url)
// return
// default:
// s.ChannelMessageSend(m.ChannelID, "Invalid endpoint.")
// return
// }
//
// req, err := http.NewRequest("GET", url, nil)
// if err != nil {
// errStr := err.Error()
// s.ChannelMessageSend(m.ChannelID, errStr)
// return
// }
// req.Header.Set("Accept", "application/json")
//
// client := http.Client{}
// resp, err := client.Do(req)
// if err != nil {
// errStr := err.Error()
// s.ChannelMessageSend(m.ChannelID, errStr)
// return
// }
// defer resp.Body.Close()
//
// err = json.NewDecoder(resp.Body).Decode(response)
// if err != nil {
// errStr := err.Error()
// s.ChannelMessageSend(m.ChannelID, errStr)
// return
// }
//
// switch endpoint {
// case "dadjoke":
// data := response.(*model.DadJokeResponse)
// s.ChannelMessageSend(m.ChannelID, data.Joke)
// case "jinglebells":
// data := response.(*model.JingleBellsResponse)
// s.ChannelMessageSend(m.ChannelID, data.Message)
// case "yomama":
// data := response.(*model.YomamaJokeResponse)
// s.ChannelMessageSend(m.ChannelID, data.Joke)
// }
//}
func handleGuildMemberAdd(s *discordgo.Session, m *discordgo.GuildMemberAdd) {
message := fmt.Sprintf("Welcome <@%s> to our server!", m.Member.User.ID)
_, err := s.ChannelMessageSend(ChannelID, message)
if err != nil {
log.Printf("Error sending welcome message: %v\n", err)
}
}
func handleGuildMemberRemove(s *discordgo.Session, m *discordgo.GuildMemberRemove) {
message := fmt.Sprintf("Goodbye %s", m.Member.User.Username)
_, err := s.ChannelMessageSend(ChannelID, message)
if err != nil {
log.Printf("Error sending goodbye message: %v\n", err)
}
}
func handleTag(s *discordgo.Session, m *discordgo.MessageCreate) {
@@ -233,7 +82,7 @@ func handleTag(s *discordgo.Session, m *discordgo.MessageCreate) {
return
}
client := openai.NewClient(apiKey)
client := openai.NewClient(ApiKey)
// Check if bot is mentioned in message
for _, mention := range m.Mentions {