OAuth2 Bearer Token with Customer HttpClient

[shanhui1]

For our use case, we need to pass in our custom HttpClient and make request with a different OAuth2 Bearer Token. We can't use the oauth2 lib as suggested. Anyway to get access to the req.headers before sending mutation queries?

I have a custom HttpClient such as this:

type AddHeaderTransport struct {
    T http.RoundTripper
}

func (adt *AddHeaderTransport) RoundTrip(req *http.Request) (*http.Response, error) {
    // TODO: bearer_token needs to be changed in each request cycle as there are many accounts that need to be called
        req.Header.Add("Authorization", "Bearer token_here")
    req.Header.Add("User-Agent", "go")
    return adt.T.RoundTrip(req)
}

func NewAddHeaderTransport(T http.RoundTripper) *AddHeaderTransport {
    if T == nil {
        T = http.DefaultTransport
    }
    return &AddHeaderTransport{T}
}

func NewClientWithAccessToken(timeout, tcpConnTimeout, tlsConnTimeout time.Duration) *http.Client {
    transport := &http.Transport{
        DialContext: (&net.Dialer{
            Timeout: tcpConnTimeout,
        }).DialContext,
        TLSHandshakeTimeout: tlsConnTimeout,
    }
    cc := http.Client{
        Transport: NewAddHeaderTransport(transport),
        Timeout:   timeout,
    }
    return &cc
}

func DefaultClientWithAccessToken() *http.Client {
    return NewClientWithAccessToken(
        DEFAULT_TIMEOUT_MS*time.Millisecond,
        DEFAULT_CONNECTION_TIMEOUT_MS*time.Millisecond,
        DEFAULT_CONNECTION_TIMEOUT_MS*time.Millisecond,
    )
}

[onurkose]

I was looking for a solution to make self signed certificates work for local development. But I believe this will solve your problem too.

tr := &http.Transport{
    TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
}
sslcli := &http.Client{Transport: tr}
ctx := context.TODO()
ctx = context.WithValue(ctx, oauth2.HTTPClient, sslcli)
src := oauth2.StaticTokenSource(
    &oauth2.Token{AccessToken: "...put your token without the Bearer prefix here..."},
)