From 41c573eb43f1a95458e5f42a3d9783437d746ca6 Mon Sep 17 00:00:00 2001
From: Voronsky <ivan.e.diaz@pm.me>
Date: Fri, 7 Jun 2024 13:17:19 -0400
Subject: [PATCH] Added server details func

---
 pterogo.go | 94 +++++++++++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 86 insertions(+), 8 deletions(-)

diff --git a/pterogo.go b/pterogo.go
index 55a37d7..0368902 100644
--- a/pterogo.go
+++ b/pterogo.go
@@ -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
+}