added !unraid and restructure
This commit is contained in:
30
handlers/members.go
Normal file
30
handlers/members.go
Normal file
@@ -0,0 +1,30 @@
|
||||
package handlers
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
|
||||
"github.com/bwmarrin/discordgo"
|
||||
"gitlab.com/DeveloperDurp/durpot/model"
|
||||
)
|
||||
|
||||
var (
|
||||
ChannelID = model.ChannelID
|
||||
)
|
||||
|
||||
func GuildMemberAdd(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 GuildMemberRemove(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)
|
||||
}
|
||||
}
|
||||
75
handlers/shared.go
Normal file
75
handlers/shared.go
Normal file
@@ -0,0 +1,75 @@
|
||||
package handlers
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"strings"
|
||||
|
||||
"gitlab.com/DeveloperDurp/durpot/model"
|
||||
)
|
||||
|
||||
func CallDurpAPI(url string, accesstoken string) []byte {
|
||||
req, err := http.NewRequest("GET", url, nil)
|
||||
if err != nil {
|
||||
fmt.Println("Error creating request:", err)
|
||||
return nil
|
||||
}
|
||||
|
||||
req.Header.Set("Authorization", "Bearer "+accesstoken)
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
|
||||
client := &http.Client{}
|
||||
resp, err := client.Do(req)
|
||||
if err != nil {
|
||||
fmt.Println("Error sending request:", err)
|
||||
return nil
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
body, err := ioutil.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
fmt.Println("Error reading response:", err)
|
||||
return nil
|
||||
}
|
||||
return body
|
||||
}
|
||||
|
||||
func GenerateToken(clientID string, grantType string, url string, username string, password string) model.AccessTokenResponse {
|
||||
|
||||
formData := fmt.Sprintf("grant_type=%s&client_id=%s&username=%s&password=%s",
|
||||
grantType, clientID, username, password)
|
||||
|
||||
client := &http.Client{}
|
||||
|
||||
req, err := http.NewRequest("POST", url, strings.NewReader(formData))
|
||||
if err != nil {
|
||||
fmt.Println("Error creating request:", err)
|
||||
return model.AccessTokenResponse{}
|
||||
}
|
||||
|
||||
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||||
|
||||
resp, err := client.Do(req)
|
||||
if err != nil {
|
||||
fmt.Println("Error sending request:", err)
|
||||
return model.AccessTokenResponse{}
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
body, err := ioutil.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
fmt.Println("Error reading response:", err)
|
||||
return model.AccessTokenResponse{}
|
||||
}
|
||||
|
||||
var response model.AccessTokenResponse
|
||||
err = json.Unmarshal(body, &response)
|
||||
if err != nil {
|
||||
fmt.Println("Error parsing response:", err)
|
||||
return model.AccessTokenResponse{}
|
||||
}
|
||||
|
||||
return response
|
||||
}
|
||||
124
handlers/unraid.go
Normal file
124
handlers/unraid.go
Normal file
@@ -0,0 +1,124 @@
|
||||
package handlers
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
|
||||
"github.com/bwmarrin/discordgo"
|
||||
"gitlab.com/DeveloperDurp/durpot/model"
|
||||
)
|
||||
|
||||
var (
|
||||
ClientID = model.ClientID
|
||||
TokenURL = model.TokenURL
|
||||
Username = model.Username
|
||||
Password = model.Password
|
||||
)
|
||||
|
||||
func GetUnraidUsage(s *discordgo.Session, m *discordgo.MessageCreate) {
|
||||
|
||||
token := GenerateToken(ClientID, "client_credentials", TokenURL, Username, Password)
|
||||
|
||||
url := "https://durpapi.durp.info/api/v1/unraid/powerusage"
|
||||
accessToken := token.AccessToken
|
||||
|
||||
body := CallDurpAPI(url, accessToken)
|
||||
|
||||
var response model.PowerUsageResponse
|
||||
err := json.Unmarshal(body, &response)
|
||||
if err != nil {
|
||||
fmt.Println("Error parsing response:", err)
|
||||
return
|
||||
}
|
||||
|
||||
jsonData, _ := json.MarshalIndent(response, "", " ")
|
||||
jsonStr := string(jsonData)
|
||||
s.ChannelMessageSend(m.ChannelID, "Power Usage Response:\n```json\n"+jsonStr+"\n```")
|
||||
|
||||
}
|
||||
|
||||
//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)
|
||||
// }
|
||||
//}
|
||||
Reference in New Issue
Block a user