pterogo/pterogo.go

package pterogo

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io"
	"log/slog"
	"net/http"
	"os"
)

var (
	opts = &slog.HandlerOptions{
		Level: slog.LevelInfo,
	}
	logger = slog.New(slog.NewJSONHandler(os.Stdout, opts))
)

// PteroRequestHeaders keeps track of the auth token and base url for all requests
// Its methods allow to make a request using the auth token and base url
type PteroRequestHeaders struct {
	Auth_token string
	Url        string
}

// A PterodactylClient implements methods for all client API routes
type PterodactylClient struct {
	Request PteroRequestHeaders //underlying PteroRequestHeaders needed for client Requests
}

// PteroResp will hold the JSON decoded body sent from the Pterodactyl Server.
type PteroResp struct {
	StatusCode int
	Object     string      `json:"object,omitempty"`
	Data       []PteroData `json:"data,omitempty"`
}

// PteroData holds all the JSON decoded the nested Pterodactyl 'object' and data found in the Response
type PteroData struct {
	Object     string     `json:"object"`
	Attributes Attributes `json:"attributes"`
}

// Attributes holds the attributes of the Pterodactly Object found in the Data JSON object
type Attributes struct {
	Name         string `json:"name,omitempty"`
	Identifier   string `json:"identifier,omitempty"`
	Description  string `json:"description,omitempty"`
	CurrentState string `json:"current_state,omitempty"`
}

// Holds necessary information about a server
type Server struct {
	Name        string
	Description string
}

// Builds the custom headers needed for Pterodactyl API routes
// Executes the Request based on the method and route passed
func (prh PteroRequestHeaders) PteroGetRequest(route string) ([]byte, error) {
	client := &http.Client{}

	//Build Get Request
	req, err := http.NewRequest("GET", route, nil)
	if err != nil {
		slog.Error("Failed to make a new GET 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 "+prh.Auth_token)

	//Issue Method request
	resp, err := client.Do(req)
	if err != nil {
		logger.Error("An error occurred trying to issue the GET 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 trace to the github", resp.StatusCode)
		logger.Error("Internal server code was returned", "StatusCode", resp.StatusCode)
		return nil, err
	}

	defer resp.Body.Close()

	body, err := io.ReadAll(resp.Body)
	if err != nil {
		logger.Error("Failed to read body", "Error", err)
		return nil, err
	}

	logger.Debug("Request successful", "RespStatusCode", resp.StatusCode)
	return body, nil
}

// Builds the custom headers needed for Pterodactyl API routes
// Executes the POST Request to the route passed
func (prh PteroRequestHeaders) PteroPostRequest(route string, jsonBody []byte) (*PteroResp, error) {
	client := &http.Client{}
	pResp := &PteroResp{}

	//Build Post Request
	bodyReader := bytes.NewReader(jsonBody)
	req, err := http.NewRequest("POST", route, bodyReader)
	if err != nil {
		slog.Error("Failed to make a new POST 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 "+prh.Auth_token)

	//Issue Method request
	resp, err := client.Do(req)
	if err != nil {
		logger.Error("An error occurred trying to issue the POST 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 trace to the github", resp.StatusCode)
		logger.Error("Internal server code was returned", "StatusCode", resp.StatusCode)
		return nil, err
	}

	logger.Debug("Request successful", "Resp", resp.StatusCode)

	defer resp.Body.Close()

	body, err := io.ReadAll(resp.Body)
	if err != nil {
		logger.Error("Failed to read body", "Error", err)
		return nil, err
	}
	pResp.StatusCode = resp.StatusCode
	json.Unmarshal(body, &pResp)
	return pResp, nil
}

// Grabs the list of servers from Pterodactyl
// Taken from the Pterodactyl API page. This will return an error if it fails at any point
// Otherwise, it will return a map of unique servers , based off their identifier
// A Bearer Auth token is required
func (pc PterodactylClient) ListServers() (map[string]Server, error) {
	r := PteroResp{}
	servers := map[string]Server{}

	//Build GET Request
	route := fmt.Sprintf("%s/api/client", pc.Request.Url)

	// Decode the JSON body into the appropriate interface
	body, err := pc.Request.PteroGetRequest(route)
	if err != nil {
		logger.Error("Error received making request to Pterodactyl", "Error", err)
		return nil, err
	}

	json.Unmarshal(body, &r)

	logger.Debug("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}
		logger.Debug("Server identifer", "Server", attrs.Attributes.Identifier)
	}

	return servers, nil
}

// Return server details for the specific identifier
func (pc PterodactylClient) ServerDetails(identifier string) (*Server, error) {
	server := &Server{}
	data := PteroData{}

	//Build GET route and make Request
	route := fmt.Sprintf("%s/api/client/servers/%s", pc.Request.Url, identifier)

	body, err := pc.Request.PteroGetRequest(route)
	if err != nil {
		logger.Error("Error received making request to Pterodactyl", "Error", err)
		return nil, err
	}

	json.Unmarshal(body, &data)
	logger.Debug("Decoded JSON body", "PteroResp", data)

	server.Name = data.Attributes.Name
	server.Description = data.Attributes.Description

	return server, nil
}

// Retrieves the a string of power state based on the server or "identifier"
func (pc PterodactylClient) GetPowerState(identifier string) (string, error) {
	pData := &PteroData{}

	//Build GET route and make the request
	route := fmt.Sprintf("%s/api/client/servers/%s/resources", pc.Request.Url, identifier)

	resp, err := pc.Request.PteroGetRequest(route)
	if err != nil {
		logger.Error("Error received making request to Pterodactyl", "Error", err)
		return "", err
	}

	json.Unmarshal(resp, &pData)
	logger.Debug("Decoded JSON body", "PteroResp", pData)

	return pData.Attributes.CurrentState, nil

}

// Returns 0 for success or -1 for failure. Pterodactyl does not provide additional information besides a status code for success
func (pc PterodactylClient) ChangePowerState(identifier string, state string) (int, error) {

	//Build POST route and make Request
	route := fmt.Sprintf("%s/api/client/servers/%s/power", pc.Request.Url, identifier)

	jsonBody := []byte(fmt.Sprintf(`{ "signal": "%s"}`, state))
	resp, err := pc.Request.PteroPostRequest(route, jsonBody)
	if err != nil {
		logger.Error("Error received making POST request to Pterodactyl", "Error", err)
		return -1, err
	}

	logger.Debug("Successful post", "StatusCode", resp.StatusCode, "Body", resp.Data)
	return 0, nil

}