api.go

package devicecheck

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
)

const (
	developmentBaseURL = "https://api.development.devicecheck.apple.com/v1"
	productionBaseURL  = "https://api.devicecheck.apple.com/v1"
)

func newBaseURL(env Environment) string {
	switch env {
	case Development:
		return developmentBaseURL
	case Production:
		return productionBaseURL
	default:
		return developmentBaseURL
	}
}

type api struct {
	client  *http.Client
	baseURL string
}

func newAPI(env Environment) api {
	return api{
		client:  http.DefaultClient,
		baseURL: newBaseURL(env),
	}
}

func newAPIWithHTTPClient(client *http.Client, env Environment) api {
	return api{
		client:  client,
		baseURL: newBaseURL(env),
	}
}

func (api api) do(ctx context.Context, jwt, path string, requestBody interface{}) (int, string, error) {
	buf := new(bytes.Buffer)
	if err := json.NewEncoder(buf).Encode(requestBody); err != nil {
		return 0, "", fmt.Errorf("json: %w", err)
	}

	req, err := http.NewRequestWithContext(ctx, http.MethodPost, api.baseURL+path, buf)
	if err != nil {
		return 0, "", fmt.Errorf("http: %w", err)
	}

	req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", jwt))
	req.Header.Set("User-Agent", "device-check-go (+https://github.com/rinchsan/device-check-go)")

	resp, err := api.client.Do(req)
	if err != nil {
		var traceID string
		if resp != nil {
			traceID = resp.Header.Get("x-b3-traceid")
		}

		return 0, "", fmt.Errorf("http: %w: x-b3-traceid: %s", err, traceID)
	}
	defer resp.Body.Close()

	respBody, err := io.ReadAll(resp.Body)
	if err != nil {
		return 0, "", fmt.Errorf("io: %w", err)
	}

	return resp.StatusCode, string(respBody), nil
}