mirror of
https://gitlab.durp.info/durfy/apps/durpot.git
synced 2026-05-07 16:00:33 -05:00
178 lines
3.3 KiB
Go
178 lines
3.3 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
"github.com/joho/godotenv"
|
|
)
|
|
|
|
var (
|
|
Token string
|
|
BotPrefix string
|
|
|
|
config *configStruct
|
|
)
|
|
|
|
type configStruct struct {
|
|
Token string `json : "Token"`
|
|
BotPrefix string `json : "BotPrefix"`
|
|
}
|
|
|
|
func ReadConfig() error {
|
|
|
|
err := godotenv.Load(".env")
|
|
|
|
if err != nil {
|
|
fmt.Println(err.Error())
|
|
return err
|
|
}
|
|
|
|
Token = os.Getenv("TOKEN")
|
|
BotPrefix = os.Getenv("BOTPREFIX")
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
var BotId string
|
|
var goBot *discordgo.Session
|
|
|
|
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)
|
|
|
|
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
|
|
}
|
|
|
|
var response string
|
|
baseurl := "https://kong.durp.info/"
|
|
switch m.Content {
|
|
case BotPrefix + "ping":
|
|
response = "pong"
|
|
s.ChannelMessageSend(m.ChannelID, response)
|
|
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":
|
|
response = getJson(baseurl+"yomama", "joke")
|
|
s.ChannelMessageSend(m.ChannelID, response)
|
|
case BotPrefix + "dadjoke":
|
|
response = getJson(baseurl+"dadjoke", "joke")
|
|
s.ChannelMessageSend(m.ChannelID, response)
|
|
case BotPrefix + "dog":
|
|
response = getJson(baseurl+"random-dogs", "message")
|
|
s.ChannelMessageSend(m.ChannelID, response)
|
|
case BotPrefix + "jinglebells":
|
|
response = getJson(baseurl+"foaas/jinglebells/durp", "message")
|
|
s.ChannelMessageSend(m.ChannelID, response)
|
|
case BotPrefix + "swanson":
|
|
response = getJson(baseurl+"ronswanson", "base")
|
|
s.ChannelMessageSend(m.ChannelID, response)
|
|
}
|
|
|
|
}
|
|
|
|
func main() {
|
|
err := ReadConfig()
|
|
|
|
if err != nil {
|
|
fmt.Println(err.Error())
|
|
return
|
|
}
|
|
|
|
Start()
|
|
|
|
<-make(chan struct{})
|
|
return
|
|
}
|
|
|
|
func getJson(url, value string) (output string) {
|
|
|
|
var s string
|
|
switch value {
|
|
case "base":
|
|
resp, err := http.Get(url)
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
|
|
sc := strings.Trim(string(body), "[")
|
|
sc = strings.Trim(sc, "]")
|
|
sc = strings.Trim(sc, "\"")
|
|
|
|
s = fmt.Sprintf("%v", sc)
|
|
default:
|
|
client := http.Client{
|
|
Timeout: time.Second * 2, // Timeout after 2 seconds
|
|
}
|
|
req, err := http.NewRequest(http.MethodGet, url, nil)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
req.Header.Set("Accept", "application/json")
|
|
|
|
res, err := client.Do(req)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
defer res.Body.Close()
|
|
|
|
b, err := io.ReadAll(res.Body)
|
|
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
|
|
var result map[string]interface{}
|
|
json.Unmarshal([]byte(b), &result)
|
|
s = fmt.Sprintf("%v", result[value])
|
|
}
|
|
return string(s)
|
|
}
|