Add initial support for fetching egg configuration from panel for servers

api/api.go

@@ -0,0 +1,114 @@
+package api
+
+import (
+	"bytes"
+	"errors"
+	"fmt"
+	"github.com/buger/jsonparser"
+	"github.com/pterodactyl/wings/config"
+	"go.uber.org/zap"
+	"io/ioutil"
+	"net/http"
+	"strings"
+	"time"
+)
+
+// Initalizes the requester instance.
+func NewRequester() *PanelRequest {
+	return &PanelRequest{
+		Response: nil,
+	}
+}
+
+type PanelRequest struct {
+	Response *http.Response
+}
+
+// Builds the base request instance that can be used with the HTTP client.
+func (r *PanelRequest) GetClient() *http.Client {
+	return &http.Client{Timeout: time.Second * 30}
+}
+
+func (r *PanelRequest) SetHeaders(req *http.Request) *http.Request {
+	req.Header.Set("Accept", "application/vnd.pterodactyl.v1+json")
+	req.Header.Set("Content-Type", "application/json")
+	req.Header.Set("Authorization", "Bearer " + config.Get().AuthenticationToken)
+
+	return req
+}
+
+func (r *PanelRequest) GetEndpoint(endpoint string) string {
+	return fmt.Sprintf(
+		"%s/api/remote/%s",
+		strings.TrimSuffix(config.Get().PanelLocation, "/"),
+		strings.TrimPrefix(strings.TrimPrefix(endpoint, "/"), "api/remote/"),
+	)
+}
+
+func (r *PanelRequest) Get(url string) (*http.Response, error) {
+	c := r.GetClient()
+
+	req, err := http.NewRequest(http.MethodGet, r.GetEndpoint(url), nil)
+	req = r.SetHeaders(req)
+
+	if err != nil {
+		return nil, err
+	}
+
+	zap.S().Debugw("GET request to endpoint", zap.String("endpoint", r.GetEndpoint(url)), zap.Any("headers", req.Header))
+
+	return c.Do(req)
+}
+
+// Determines if the API call encountered an error. If no request has been made
+// the response will be false.
+func (r *PanelRequest) HasError() bool {
+	if r.Response == nil {
+		return false
+	}
+
+	return r.Response.StatusCode >= 300 || r.Response.StatusCode < 200;
+}
+
+// Reads the body from the response and returns it, then replaces it on the response
+// so that it can be read again later.
+func (r *PanelRequest) ReadBody() ([]byte, error) {
+	var b []byte
+	if r.Response == nil {
+		return nil, errors.New("no response exists on interface")
+	}
+
+	if r.Response.Body != nil {
+		b, _ = ioutil.ReadAll(r.Response.Body)
+	}
+
+	r.Response.Body = ioutil.NopCloser(bytes.NewBuffer(b))
+
+	return b, nil
+}
+// Returns the error message from the API call as a string. The error message will be formatted
+// similar to the below example:
+//
+// HttpNotFoundException: The requested resource does not exist. (HTTP/404)
+func (r *PanelRequest) Error() (string, error) {
+	body, err := r.ReadBody()
+	if err != nil {
+		return "", err
+	}
+
+	zap.S().Debugw("got body", zap.ByteString("b", body))
+	_, valueType, _, err := jsonparser.Get(body, "errors")
+	if err != nil {
+		return "", err
+	}
+
+	if valueType != jsonparser.Object {
+		return "no error object present on response", nil
+	}
+
+	code, _ := jsonparser.GetString(body, "errors.0.code")
+	status, _ := jsonparser.GetString(body, "errors.0.status")
+	detail, _ := jsonparser.GetString(body, "errors.0.detail")
+
+	return fmt.Sprintf("%s: %s (HTTP/%s)", code, detail, status), nil
+}