mirror of
https://gitlab.durp.info/durfy/apps/durpot.git
synced 2026-05-07 16:00:33 -05:00
test update
This commit is contained in:
parent
588b6d9fa3
commit
21ce2d0609
3 changed files with 135 additions and 132 deletions
2
go.mod
2
go.mod
|
|
@ -1,4 +1,4 @@
|
||||||
module golang-discord-bot
|
module "gitlab.com/DeveloperDurp/durpot"
|
||||||
|
|
||||||
go 1.19
|
go 1.19
|
||||||
|
|
||||||
|
|
|
||||||
242
main.go
242
main.go
|
|
@ -2,16 +2,15 @@ package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"log"
|
"log"
|
||||||
"net/http"
|
|
||||||
"os"
|
"os"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/bwmarrin/discordgo"
|
"github.com/bwmarrin/discordgo"
|
||||||
"github.com/joho/godotenv"
|
"github.com/joho/godotenv"
|
||||||
openai "github.com/sashabaranov/go-openai"
|
openai "github.com/sashabaranov/go-openai"
|
||||||
|
"gitlab.com/DeveloperDurp/durpot/model"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
|
|
@ -22,31 +21,9 @@ var (
|
||||||
apiKey string
|
apiKey string
|
||||||
goBot *discordgo.Session
|
goBot *discordgo.Session
|
||||||
|
|
||||||
config *configStruct
|
config *model.ConfigStruct
|
||||||
)
|
)
|
||||||
|
|
||||||
type configStruct struct {
|
|
||||||
Token string `json : "Token"`
|
|
||||||
BotPrefix string `json : "BotPrefix"`
|
|
||||||
ChannelID string `json : "ChannelID"`
|
|
||||||
apiKey string `json : "OPEN_API_KEY"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type jingleBellsResponse struct {
|
|
||||||
Message string `json:"message"`
|
|
||||||
Subtitle string `json:"subtitle"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type dadJokeResponse struct {
|
|
||||||
ID string `json:"id"`
|
|
||||||
Joke string `json:"joke"`
|
|
||||||
Status int `json:"status"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type yomamaJokeResponse struct {
|
|
||||||
Joke string `json:"joke"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func ReadConfig() error {
|
func ReadConfig() error {
|
||||||
|
|
||||||
err := godotenv.Load(".env")
|
err := godotenv.Load(".env")
|
||||||
|
|
@ -101,30 +78,33 @@ func messageHandler(s *discordgo.Session, m *discordgo.MessageCreate) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
baseurl := "https://kong.durp.info/"
|
//baseurl := "https://kong.durp.info/"
|
||||||
switch m.Content {
|
switch m.Content {
|
||||||
case BotPrefix + "ping":
|
case BotPrefix + "ping":
|
||||||
s.ChannelMessageSend(m.ChannelID, "pong")
|
_, err := s.ChannelMessageSend(m.ChannelID, "pong")
|
||||||
//case BotPrefix + "meme":
|
if err != nil {
|
||||||
// response = getJson(baseurl+"random-meme", "url")
|
fmt.Println("Failed to send Message")
|
||||||
// s.ChannelMessageSend(m.ChannelID, response)
|
}
|
||||||
//case BotPrefix + "catfact":
|
//case BotPrefix + "meme":
|
||||||
// response = getJson(baseurl+"cat-facts/fact", "fact")
|
// response = getJson(baseurl+"random-meme", "url")
|
||||||
// s.ChannelMessageSend(m.ChannelID, response)
|
// s.ChannelMessageSend(m.ChannelID, response)
|
||||||
//case BotPrefix + "cat":
|
//case BotPrefix + "catfact":
|
||||||
// response = getJson(baseurl+"random-cats", "file")
|
// response = getJson(baseurl+"cat-facts/fact", "fact")
|
||||||
// s.ChannelMessageSend(m.ChannelID, response)
|
// s.ChannelMessageSend(m.ChannelID, response)
|
||||||
case BotPrefix + "yomama":
|
//case BotPrefix + "cat":
|
||||||
sendAPIRequest(s, m, baseurl, "yomama")
|
// response = getJson(baseurl+"random-cats", "file")
|
||||||
case BotPrefix + "dadjoke":
|
// s.ChannelMessageSend(m.ChannelID, response)
|
||||||
sendAPIRequest(s, m, baseurl, "dadjoke")
|
//case BotPrefix + "yomama":
|
||||||
//case BotPrefix + "dog":
|
// sendAPIRequest(s, m, baseurl, "yomama")
|
||||||
// response = getJson(baseurl+"random-dogs", "message")
|
//case BotPrefix + "dadjoke":
|
||||||
// s.ChannelMessageSend(m.ChannelID, response)
|
// sendAPIRequest(s, m, baseurl, "dadjoke")
|
||||||
case BotPrefix + "jinglebells":
|
//case BotPrefix + "dog":
|
||||||
sendAPIRequest(s, m, baseurl, "jinglebells")
|
// response = getJson(baseurl+"random-dogs", "message")
|
||||||
case BotPrefix + "swanson":
|
// s.ChannelMessageSend(m.ChannelID, response)
|
||||||
sendAPIRequest(s, m, baseurl, "swanson")
|
//case BotPrefix + "jinglebells":
|
||||||
|
// sendAPIRequest(s, m, baseurl, "jinglebells")
|
||||||
|
//case BotPrefix + "swanson":
|
||||||
|
// sendAPIRequest(s, m, baseurl, "swanson")
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -144,91 +124,91 @@ func main() {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func getSwansonQuote(s *discordgo.Session, m *discordgo.MessageCreate, url string) {
|
//func getSwansonQuote(s *discordgo.Session, m *discordgo.MessageCreate, url string) {
|
||||||
resp, err := http.Get(url + "/ronswanson")
|
// resp, err := http.Get(url + "/ronswanson")
|
||||||
if err != nil {
|
// if err != nil {
|
||||||
errStr := err.Error()
|
// errStr := err.Error()
|
||||||
s.ChannelMessageSend(m.ChannelID, errStr)
|
// s.ChannelMessageSend(m.ChannelID, errStr)
|
||||||
return
|
// return
|
||||||
}
|
// }
|
||||||
defer resp.Body.Close()
|
// 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])
|
||||||
|
//}
|
||||||
|
|
||||||
var data []string
|
//func sendAPIRequest(s *discordgo.Session, m *discordgo.MessageCreate, url string, endpoint string) {
|
||||||
err = json.NewDecoder(resp.Body).Decode(&data)
|
// var response interface{}
|
||||||
if err != nil {
|
// switch endpoint {
|
||||||
errStr := err.Error()
|
// case "dadjoke":
|
||||||
s.ChannelMessageSend(m.ChannelID, errStr)
|
// url = url + "/dadjoke"
|
||||||
return
|
// var data model.DadJokeResponse
|
||||||
}
|
// response = &data
|
||||||
|
// case "jinglebells":
|
||||||
if len(data) == 0 {
|
// url = url + "/foaas/jinglebells/durp"
|
||||||
errStr := "No quotes found."
|
// var data model.JingleBellsResponse
|
||||||
s.ChannelMessageSend(m.ChannelID, errStr)
|
// response = &data
|
||||||
return
|
// case "yomama":
|
||||||
}
|
// url = url + "/yomama"
|
||||||
|
// var data model.YomamaJokeResponse
|
||||||
s.ChannelMessageSend(m.ChannelID, data[0])
|
// response = &data
|
||||||
}
|
// case "swanson":
|
||||||
|
// getSwansonQuote(s, m, url)
|
||||||
func sendAPIRequest(s *discordgo.Session, m *discordgo.MessageCreate, url string, endpoint string) {
|
// return
|
||||||
var response interface{}
|
// default:
|
||||||
switch endpoint {
|
// s.ChannelMessageSend(m.ChannelID, "Invalid endpoint.")
|
||||||
case "dadjoke":
|
// return
|
||||||
url = url + "/dadjoke"
|
// }
|
||||||
var data dadJokeResponse
|
//
|
||||||
response = &data
|
// req, err := http.NewRequest("GET", url, nil)
|
||||||
case "jinglebells":
|
// if err != nil {
|
||||||
url = url + "/foaas/jinglebells/durp"
|
// errStr := err.Error()
|
||||||
var data jingleBellsResponse
|
// s.ChannelMessageSend(m.ChannelID, errStr)
|
||||||
response = &data
|
// return
|
||||||
case "yomama":
|
// }
|
||||||
url = url + "/yomama"
|
// req.Header.Set("Accept", "application/json")
|
||||||
var data yomamaJokeResponse
|
//
|
||||||
response = &data
|
// client := http.Client{}
|
||||||
case "swanson":
|
// resp, err := client.Do(req)
|
||||||
getSwansonQuote(s, m, url)
|
// if err != nil {
|
||||||
return
|
// errStr := err.Error()
|
||||||
default:
|
// s.ChannelMessageSend(m.ChannelID, errStr)
|
||||||
s.ChannelMessageSend(m.ChannelID, "Invalid endpoint.")
|
// return
|
||||||
return
|
// }
|
||||||
}
|
// defer resp.Body.Close()
|
||||||
|
//
|
||||||
req, err := http.NewRequest("GET", url, nil)
|
// err = json.NewDecoder(resp.Body).Decode(response)
|
||||||
if err != nil {
|
// if err != nil {
|
||||||
errStr := err.Error()
|
// errStr := err.Error()
|
||||||
s.ChannelMessageSend(m.ChannelID, errStr)
|
// s.ChannelMessageSend(m.ChannelID, errStr)
|
||||||
return
|
// return
|
||||||
}
|
// }
|
||||||
req.Header.Set("Accept", "application/json")
|
//
|
||||||
|
// switch endpoint {
|
||||||
client := http.Client{}
|
// case "dadjoke":
|
||||||
resp, err := client.Do(req)
|
// data := response.(*model.DadJokeResponse)
|
||||||
if err != nil {
|
// s.ChannelMessageSend(m.ChannelID, data.Joke)
|
||||||
errStr := err.Error()
|
// case "jinglebells":
|
||||||
s.ChannelMessageSend(m.ChannelID, errStr)
|
// data := response.(*model.JingleBellsResponse)
|
||||||
return
|
// s.ChannelMessageSend(m.ChannelID, data.Message)
|
||||||
}
|
// case "yomama":
|
||||||
defer resp.Body.Close()
|
// data := response.(*model.YomamaJokeResponse)
|
||||||
|
// s.ChannelMessageSend(m.ChannelID, data.Joke)
|
||||||
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.(*dadJokeResponse)
|
|
||||||
s.ChannelMessageSend(m.ChannelID, data.Joke)
|
|
||||||
case "jinglebells":
|
|
||||||
data := response.(*jingleBellsResponse)
|
|
||||||
s.ChannelMessageSend(m.ChannelID, data.Message)
|
|
||||||
case "yomama":
|
|
||||||
data := response.(*yomamaJokeResponse)
|
|
||||||
s.ChannelMessageSend(m.ChannelID, data.Joke)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func handleGuildMemberAdd(s *discordgo.Session, m *discordgo.GuildMemberAdd) {
|
func handleGuildMemberAdd(s *discordgo.Session, m *discordgo.GuildMemberAdd) {
|
||||||
message := fmt.Sprintf("Welcome <@%s> to our server!", m.Member.User.ID)
|
message := fmt.Sprintf("Welcome <@%s> to our server!", m.Member.User.ID)
|
||||||
|
|
|
||||||
23
model/model.go
Normal file
23
model/model.go
Normal file
|
|
@ -0,0 +1,23 @@
|
||||||
|
package model
|
||||||
|
|
||||||
|
type ConfigStruct struct {
|
||||||
|
Token string `json:"Token"`
|
||||||
|
BotPrefix string `json:"BotPrefix"`
|
||||||
|
ChannelID string `json:"ChannelID"`
|
||||||
|
ApiKey string `json:"OPEN_API_KEY"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type JingleBellsResponse struct {
|
||||||
|
Message string `json:"message"`
|
||||||
|
Subtitle string `json:"subtitle"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type DadJokeResponse struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
Joke string `json:"joke"`
|
||||||
|
Status int `json:"status"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type YomamaJokeResponse struct {
|
||||||
|
Joke string `json:"joke"`
|
||||||
|
}
|
||||||
Loading…
Add table
Add a link
Reference in a new issue