package digitalocean

import (
	"context"
	"fmt"
	"net/http"

	"congo.gg/pkg/platform"
	"github.com/digitalocean/godo"
)

type backend struct {
	client *godo.Client
	ctx    context.Context
}

// bearerTransport adds an Authorization header to every request.
type bearerTransport struct {
	token string
}

func (t *bearerTransport) RoundTrip(req *http.Request) (*http.Response, error) {
	r := req.Clone(req.Context())
	r.Header.Set("Authorization", "Bearer "+t.token)
	return http.DefaultTransport.RoundTrip(r)
}

// New creates a DigitalOcean platform.
func New(token string) (*platform.Platform, error) {
	if token == "" {
		return nil, fmt.Errorf("token required")
	}

	httpClient := &http.Client{Transport: &bearerTransport{token: token}}

	return &platform.Platform{
		Backend: &backend{
			client: godo.NewClient(httpClient),
			ctx:    context.Background(),
		},
	}, nil
}