micromdm/platform/remove/client.go

package remove

import (
	"context"
	"net/http"
	"net/url"

	"github.com/go-kit/kit/endpoint"
	"github.com/go-kit/kit/log"
	httptransport "github.com/go-kit/kit/transport/http"
)

func NewHTTPClient(instance, token string, logger log.Logger, opts ...httptransport.ClientOption) (Service, error) {
	u, err := url.Parse(instance)
	if err != nil {
		return nil, err
	}

	var blockDeviceEndpoint endpoint.Endpoint
	{
		blockDeviceEndpoint = httptransport.NewClient(
			"POST",
			copyURL(u, ""), // empty path, modified by the encodeRequest func
			encodeRequestWithToken(token, encodeBlockDeviceRequest),
			decodeBlockDeviceResponse,
			opts...,
		).Endpoint()
	}

	var unblockDeviceEndpoint endpoint.Endpoint
	{
		unblockDeviceEndpoint = httptransport.NewClient(
			"POST",
			copyURL(u, ""), //modified by encodeRequestFunc
			encodeRequestWithToken(token, encodeUnblockDeviceRequest),
			decodeUnblockDeviceResponse,
			opts...,
		).Endpoint()
	}

	return Endpoints{
		BlockDeviceEndpoint:   blockDeviceEndpoint,
		UnblockDeviceEndpoint: unblockDeviceEndpoint,
	}, nil
}

func encodeRequestWithToken(token string, next httptransport.EncodeRequestFunc) httptransport.EncodeRequestFunc {
	return func(ctx context.Context, r *http.Request, request interface{}) error {
		r.SetBasicAuth("micromdm", token)
		return next(ctx, r, request)
	}
}

func copyURL(base *url.URL, path string) *url.URL {
	next := *base
	next.Path = path
	return &next
}