package client

import (
	"net/http"
)

const (
	BearerToken   = "bearer"
	TotpToken     = "totp"
	EdgeVersion   = "edge-version"
	Authorization = "Authorization"
)

type Transport struct {
	T       http.RoundTripper
	Headers map[string]string
}

func (t *Transport) RoundTrip(req *http.Request) (*http.Response, error) {
	for key, val := range t.Headers {
		req.Header.Add(key, val)
	}
	return t.T.RoundTrip(req)
}

func NewTokenClient(headers map[string]string) *http.Client {
	return &http.Client{
		Transport: &Transport{
			T:       http.DefaultTransport,
			Headers: headers,
		},
	}
}