ficsit-cli-flake/ficsit/root.go

package ficsit

import (
	"net/http"

	"github.com/Khan/genqlient/graphql"
	"github.com/pkg/errors"
	"github.com/spf13/viper"
)

type AuthedTransport struct {
	Wrapped http.RoundTripper
}

func (t *AuthedTransport) RoundTrip(req *http.Request) (*http.Response, error) {
	key := viper.GetString("api-key")
	if key != "" {
		req.Header.Set("Authorization", key)
	}

	rt, err := t.Wrapped.RoundTrip(req)
	return rt, errors.Wrap(err, "failed roundtrip")
}

func InitAPI() graphql.Client {
	httpClient := http.Client{
		Transport: &AuthedTransport{
			Wrapped: http.DefaultTransport,
		},
	}

	return graphql.NewClient(viper.GetString("api-base")+viper.GetString("graphql-api"), &httpClient)
}