testing, pre-wails

api/client.go

@@ -1,27 +1,37 @@
 package api
 
-import "net/http"
+import (
+	"bytes"
+	"encoding/json"
+	"io"
+	"net/http"
+
+	"go.uber.org/zap"
+)
 
 // Client handles API communication with the Vast.ai API
 type Client struct {
 	apiKey     string
 	baseURL    string
 	httpClient *http.Client
+	logger     *zap.SugaredLogger
 }
 
 // Error represents an API error response
 type APIError struct {
-	Success bool   `json:"success"`
-	Error   string `json:"error"`
-	Msg     string `json:"msg"`
+	Success bool `json:"success"`
 }
 
 // NewClient creates a new Vast.ai API client
 func NewClient(apiKey string) *Client {
+	logger, _ := zap.NewDevelopment()
+	sugar := logger.Sugar()
+
 	return &Client{
 		apiKey:     apiKey,
 		baseURL:    "https://console.vast.ai/api/v0",
 		httpClient: &http.Client{},
+		logger:     sugar,
 	}
 }
 
@@ -32,11 +42,68 @@ func (c *Client) Apply(req *http.Request) {
 
 // makeRequest creates and sends an HTTP request with the provided method, path and body
 func (c *Client) makeRequest(method, path string, body interface{}) (*http.Response, error) {
-	req, err := http.NewRequest(method, c.baseURL+path, nil)
+	c.logger.Debugw("making request",
+		"method", method,
+		"path", path,
+		"body", body,
+	)
+
+	var bodyReader io.Reader
+	if body != nil {
+		jsonBody, err := json.Marshal(body)
+		if err != nil {
+			return nil, err
+		}
+		bodyReader = bytes.NewBuffer(jsonBody)
+	}
+
+	req, err := http.NewRequest(method, c.baseURL+path, bodyReader)
 	if err != nil {
+		c.logger.Errorw("failed to create request",
+			"error", err,
+			"method", method,
+			"path", path,
+		)
 		return nil, err
 	}
 
 	c.Apply(req)
-	return c.httpClient.Do(req)
+	res, err := c.httpClient.Do(req)
+
+	if err != nil {
+		c.logger.Errorw("failed to send request",
+			"error", err,
+			"method", method,
+			"path", path,
+		)
+		return nil, err
+	}
+
+	// debug print response
+	res_body, err := io.ReadAll(res.Body)
+	if err != nil {
+		return nil, err
+	}
+
+	if res.StatusCode != http.StatusOK {
+		var apiError APIError
+		if err := json.NewDecoder(bytes.NewBuffer(res_body)).Decode(&apiError); err != nil {
+			c.logger.Errorw("failed to decode error response",
+				"error", err,
+			)
+		} else {
+			c.logger.Errorw("api error",
+				"error", apiError,
+			)
+		}
+	} else {
+		c.logger.Debugw("response",
+			"status", res.Status,
+			"status_code", res.StatusCode,
+		)
+	}
+
+	res.Body = io.NopCloser(bytes.NewBuffer(res_body))
+
+	return res, nil
 }