mirror of
https://github.com/Voronsky/pterogo.git
synced 2025-12-12 16:46:34 -05:00
Added server details func
This commit is contained in:
94
pterogo.go
94
pterogo.go
@@ -2,7 +2,6 @@ package pterogo
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"log/slog"
|
"log/slog"
|
||||||
@@ -54,18 +53,31 @@ func listServers(auth_token string, url string) (map[string]Server, error) {
|
|||||||
//Issue GET request
|
//Issue GET request
|
||||||
resp, err := client.Do(req)
|
resp, err := client.Do(req)
|
||||||
if err != nil {
|
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
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
if resp.StatusCode >= 300 {
|
if resp.StatusCode >= 300 && resp.StatusCode < 400 {
|
||||||
// Create custom error for this
|
// Create custom error for this
|
||||||
err := errors.New("request failed")
|
err := fmt.Errorf("received redirection error=%d", resp.StatusCode)
|
||||||
slog.Error("Non-200 status code was returned", "StatusCode", resp.StatusCode)
|
logger.Error("Redirect error code was returned", "StatusCode", resp.StatusCode)
|
||||||
return nil, err
|
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{}
|
servers := map[string]Server{}
|
||||||
r := PteroResp{}
|
r := PteroResp{}
|
||||||
@@ -74,13 +86,13 @@ func listServers(auth_token string, url string) (map[string]Server, error) {
|
|||||||
|
|
||||||
body, err := io.ReadAll(resp.Body)
|
body, err := io.ReadAll(resp.Body)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
slog.Error("Failed to read body", "Error", err)
|
logger.Error("Failed to read body", "Error", err)
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
// Decode the JSON body into the appropriate interface
|
// Decode the JSON body into the appropriate interface
|
||||||
json.Unmarshal(body, &r)
|
json.Unmarshal(body, &r)
|
||||||
slog.Info("listServers()", "pteroResp", r)
|
slog.Info("Decoded JSON body", "pteroResp=", r)
|
||||||
for i := 0; i < len(r.Data); i++ {
|
for i := 0; i < len(r.Data); i++ {
|
||||||
attrs := r.Data[i]
|
attrs := r.Data[i]
|
||||||
servers[attrs.Attributes.Identifier] = Server{attrs.Attributes.Name, attrs.Attributes.Description}
|
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
|
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
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user