ivan/pterogo
1
0
Fork 0
mirror of https://github.com/Voronsky/pterogo.git synced 2025-12-12 08:46:33 -05:00
Code Issues Packages Projects Releases Wiki Activity

Added server details func

Browse Source
This commit is contained in:
Voronsky
2024-06-07 13:17:19 -04:00
parent a64e9296ad
commit 41c573eb43
1 changed files with 86 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

94
pterogo.go
View File

@@ -2,7 +2,6 @@ package pterogo
import (
"encoding/json"
"errors"
"fmt"
"io"
"log/slog"
@@ -54,18 +53,31 @@ func listServers(auth_token string, url string) (map[string]Server, error) {
//Issue GET request
resp, err := client.Do(req)
if err != nil {
slog.Error("An error occurred trying to issue the request", "Error", err)
logger.Error("An error occurred trying to issue the request", "Error", err)
return nil, err
}
if resp.StatusCode >= 300 {
if resp.StatusCode >= 300 && resp.StatusCode < 400 {
// Create custom error for this
err := errors.New("request failed")
slog.Error("Non-200 status code was returned", "StatusCode", resp.StatusCode)
err := fmt.Errorf("received redirection error=%d", resp.StatusCode)
logger.Error("Redirect error code was returned", "StatusCode", resp.StatusCode)
return nil, err
}
slog.Info("Request successful", "Resp", resp)
if resp.StatusCode >= 400 && resp.StatusCode < 500 {
// Create custom error for this
err := fmt.Errorf("received client error=%d", resp.StatusCode)
logger.Error("Client error code was returned", "StatusCode", resp.StatusCode)
return nil, err
}
if resp.StatusCode >= 500 {
err := fmt.Errorf("received internal server error=%d, please report this to the github", resp.StatusCode)
logger.Error("Internal server code was returned", "StatusCode", resp.StatusCode)
return nil, err
}
logger.Info("Request successful", "Resp", resp)
servers := map[string]Server{}
r := PteroResp{}
@@ -74,13 +86,13 @@ func listServers(auth_token string, url string) (map[string]Server, error) {
body, err := io.ReadAll(resp.Body)
if err != nil {
slog.Error("Failed to read body", "Error", err)
logger.Error("Failed to read body", "Error", err)
return nil, err
}
// Decode the JSON body into the appropriate interface
json.Unmarshal(body, &r)
slog.Info("listServers()", "pteroResp", r)
slog.Info("Decoded JSON body", "pteroResp=", r)
for i := 0; i < len(r.Data); i++ {
attrs := r.Data[i]
servers[attrs.Attributes.Identifier] = Server{attrs.Attributes.Name, attrs.Attributes.Description}
@@ -90,3 +102,69 @@ func listServers(auth_token string, url string) (map[string]Server, error) {
return servers, nil
}
func ServerDetails(identifier string, auth_token string, url string) (*Server, error) {
client := &http.Client{}
logger := slog.New(slog.NewJSONHandler(os.Stdout, nil))
//Build GET Request
route := fmt.Sprintf("%s/api/client/servers/%s", url, identifier)
req, err := http.NewRequest("GET", route, nil)
server := &Server{}
data := PteroData{}
if err != nil {
slog.Error("Failed to make a new request", "Error", err)
return nil, err
}
//Add Pterodactyl Headers
req.Header.Add("Accept", "application/json")
req.Header.Add("Content-Type", "application/json")
req.Header.Add("Authorization", "Bearer "+auth_token)
//Issue GET request
resp, err := client.Do(req)
if err != nil {
logger.Error("An error occurred trying to issue the request", "Error", err)
return nil, err
}
if resp.StatusCode >= 300 && resp.StatusCode < 400 {
// Create custom error for this
err := fmt.Errorf("received redirection error=%d", resp.StatusCode)
logger.Error("Redirect error code was returned", "StatusCode", resp.StatusCode)
return nil, err
}
if resp.StatusCode >= 400 && resp.StatusCode < 500 {
// Create custom error for this
err := fmt.Errorf("received client error=%d", resp.StatusCode)
logger.Error("Client error code was returned", "StatusCode", resp.StatusCode)
return nil, err
}
if resp.StatusCode >= 500 {
err := fmt.Errorf("received internal server error=%d, please report this to the github", resp.StatusCode)
logger.Error("Internal server code was returned", "StatusCode", resp.StatusCode)
return nil, err
}
logger.Info("Request successful", "Resp", resp)
defer resp.Body.Close()
body, err := io.ReadAll(resp.Body)
if err != nil {
logger.Error("Failed to read body", "Error", err)
return nil, err
}
json.Unmarshal(body, &data)
slog.Info("Decoded JSON body", "pteroResp=", data)
server.Name = data.Attributes.Name
server.Description = data.Attributes.Description
return server, nil
}