mirror of
https://github.com/Voronsky/pterogo.git
synced 2025-12-12 16:46:34 -05:00
added global logger
This commit is contained in:
35
pterogo.go
35
pterogo.go
@@ -10,6 +10,13 @@ import (
|
|||||||
"os"
|
"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
|
// 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
|
// Its methods allow to make a request using the auth token and base url
|
||||||
type PteroRequestHeaders struct {
|
type PteroRequestHeaders struct {
|
||||||
@@ -53,12 +60,11 @@ type Server struct {
|
|||||||
// Executes the Request based on the method and route passed
|
// Executes the Request based on the method and route passed
|
||||||
func (prh PteroRequestHeaders) PteroGetRequest(route string) ([]byte, error) {
|
func (prh PteroRequestHeaders) PteroGetRequest(route string) ([]byte, error) {
|
||||||
client := &http.Client{}
|
client := &http.Client{}
|
||||||
logger := slog.New(slog.NewJSONHandler(os.Stdout, nil))
|
|
||||||
|
|
||||||
//Build Get Request
|
//Build Get Request
|
||||||
req, err := http.NewRequest("GET", route, nil)
|
req, err := http.NewRequest("GET", route, nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
slog.Error("Failed to make a new request", "Error", err)
|
slog.Error("Failed to make a new GET request", "Error", err)
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -70,7 +76,7 @@ func (prh PteroRequestHeaders) PteroGetRequest(route string) ([]byte, error) {
|
|||||||
//Issue Method request
|
//Issue Method request
|
||||||
resp, err := client.Do(req)
|
resp, err := client.Do(req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Error("An error occurred trying to issue the request", "Error", err)
|
logger.Error("An error occurred trying to issue the GET request", "Error", err)
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -102,7 +108,7 @@ func (prh PteroRequestHeaders) PteroGetRequest(route string) ([]byte, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.Info("Request successful", "RespStatusCode", resp.StatusCode)
|
logger.Debug("Request successful", "RespStatusCode", resp.StatusCode)
|
||||||
return body, nil
|
return body, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -110,14 +116,13 @@ func (prh PteroRequestHeaders) PteroGetRequest(route string) ([]byte, error) {
|
|||||||
// Executes the POST Request to the route passed
|
// Executes the POST Request to the route passed
|
||||||
func (prh PteroRequestHeaders) PteroPostRequest(route string, jsonBody []byte) (*PteroResp, error) {
|
func (prh PteroRequestHeaders) PteroPostRequest(route string, jsonBody []byte) (*PteroResp, error) {
|
||||||
client := &http.Client{}
|
client := &http.Client{}
|
||||||
logger := slog.New(slog.NewJSONHandler(os.Stdout, nil))
|
|
||||||
pResp := &PteroResp{}
|
pResp := &PteroResp{}
|
||||||
|
|
||||||
//Build Post Request
|
//Build Post Request
|
||||||
bodyReader := bytes.NewReader(jsonBody)
|
bodyReader := bytes.NewReader(jsonBody)
|
||||||
req, err := http.NewRequest("POST", route, bodyReader)
|
req, err := http.NewRequest("POST", route, bodyReader)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
slog.Error("Failed to make a new request", "Error", err)
|
slog.Error("Failed to make a new POST request", "Error", err)
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -129,7 +134,7 @@ func (prh PteroRequestHeaders) PteroPostRequest(route string, jsonBody []byte) (
|
|||||||
//Issue Method request
|
//Issue Method request
|
||||||
resp, err := client.Do(req)
|
resp, err := client.Do(req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Error("An error occurred trying to issue the request", "Error", err)
|
logger.Error("An error occurred trying to issue the POST request", "Error", err)
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -153,7 +158,7 @@ func (prh PteroRequestHeaders) PteroPostRequest(route string, jsonBody []byte) (
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.Info("Request successful", "Resp", resp.StatusCode)
|
logger.Debug("Request successful", "Resp", resp.StatusCode)
|
||||||
|
|
||||||
defer resp.Body.Close()
|
defer resp.Body.Close()
|
||||||
|
|
||||||
@@ -175,7 +180,6 @@ func (pc PterodactylClient) ListServers() (map[string]Server, error) {
|
|||||||
r := PteroResp{}
|
r := PteroResp{}
|
||||||
servers := map[string]Server{}
|
servers := map[string]Server{}
|
||||||
|
|
||||||
logger := slog.New(slog.NewJSONHandler(os.Stdout, nil))
|
|
||||||
//Build GET Request
|
//Build GET Request
|
||||||
route := fmt.Sprintf("%s/api/client", pc.Request.url)
|
route := fmt.Sprintf("%s/api/client", pc.Request.url)
|
||||||
|
|
||||||
@@ -188,11 +192,11 @@ func (pc PterodactylClient) ListServers() (map[string]Server, error) {
|
|||||||
|
|
||||||
json.Unmarshal(body, &r)
|
json.Unmarshal(body, &r)
|
||||||
|
|
||||||
logger.Info("Decoded JSON body", "pteroResp", r)
|
logger.Debug("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}
|
||||||
logger.Info("Server identifer", "Server", attrs.Attributes.Identifier)
|
logger.Debug("Server identifer", "Server", attrs.Attributes.Identifier)
|
||||||
}
|
}
|
||||||
|
|
||||||
return servers, nil
|
return servers, nil
|
||||||
@@ -200,7 +204,6 @@ func (pc PterodactylClient) ListServers() (map[string]Server, error) {
|
|||||||
|
|
||||||
// Return server details for the specific identifier
|
// Return server details for the specific identifier
|
||||||
func (pc PterodactylClient) ServerDetails(identifier string) (*Server, error) {
|
func (pc PterodactylClient) ServerDetails(identifier string) (*Server, error) {
|
||||||
logger := slog.New(slog.NewJSONHandler(os.Stdout, nil))
|
|
||||||
server := &Server{}
|
server := &Server{}
|
||||||
data := PteroData{}
|
data := PteroData{}
|
||||||
|
|
||||||
@@ -214,7 +217,7 @@ func (pc PterodactylClient) ServerDetails(identifier string) (*Server, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
json.Unmarshal(body, &data)
|
json.Unmarshal(body, &data)
|
||||||
logger.Info("Decoded JSON body", "PteroResp", data)
|
logger.Debug("Decoded JSON body", "PteroResp", data)
|
||||||
|
|
||||||
server.Name = data.Attributes.Name
|
server.Name = data.Attributes.Name
|
||||||
server.Description = data.Attributes.Description
|
server.Description = data.Attributes.Description
|
||||||
@@ -224,7 +227,6 @@ func (pc PterodactylClient) ServerDetails(identifier string) (*Server, error) {
|
|||||||
|
|
||||||
// Retrieves the a string of power state based on the server or "identifier"
|
// Retrieves the a string of power state based on the server or "identifier"
|
||||||
func (pc PterodactylClient) GetPowerState(identifier string) (string, error) {
|
func (pc PterodactylClient) GetPowerState(identifier string) (string, error) {
|
||||||
logger := slog.New(slog.NewJSONHandler(os.Stdout, nil))
|
|
||||||
pData := &PteroData{}
|
pData := &PteroData{}
|
||||||
|
|
||||||
//Build GET route and make the request
|
//Build GET route and make the request
|
||||||
@@ -237,7 +239,7 @@ func (pc PterodactylClient) GetPowerState(identifier string) (string, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
json.Unmarshal(resp, &pData)
|
json.Unmarshal(resp, &pData)
|
||||||
logger.Info("Decoded JSON body", "PteroResp", pData)
|
logger.Debug("Decoded JSON body", "PteroResp", pData)
|
||||||
|
|
||||||
return pData.Attributes.CurrentState, nil
|
return pData.Attributes.CurrentState, nil
|
||||||
|
|
||||||
@@ -245,7 +247,6 @@ func (pc PterodactylClient) GetPowerState(identifier string) (string, error) {
|
|||||||
|
|
||||||
// Returns 0 for success or -1 for failure. Pterodactyl does not provide additional information besides a status code for success
|
// 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) {
|
func (pc PterodactylClient) ChangePowerState(identifier string, state string) (int, error) {
|
||||||
logger := slog.New(slog.NewJSONHandler(os.Stdout, nil))
|
|
||||||
|
|
||||||
//Build POST route and make Request
|
//Build POST route and make Request
|
||||||
route := fmt.Sprintf("%s/api/client/servers/%s/power", pc.Request.url, identifier)
|
route := fmt.Sprintf("%s/api/client/servers/%s/power", pc.Request.url, identifier)
|
||||||
@@ -257,7 +258,7 @@ func (pc PterodactylClient) ChangePowerState(identifier string, state string) (i
|
|||||||
return -1, err
|
return -1, err
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.Info("Successful post", "StatusCode", resp.StatusCode, "Body", resp.Data)
|
logger.Debug("Successful post", "StatusCode", resp.StatusCode, "Body", resp.Data)
|
||||||
return 0, nil
|
return 0, nil
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user