Daniele Stefano Ferru
76f7836419
* Provisioning: correctly use resource clients in controllers * better names on fields * fix struct initialisation * updating roundtripper tests
47 lines
1.3 KiB
Go
47 lines
1.3 KiB
Go
package auth
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/grafana/authlib/authn"
|
|
utilnet "k8s.io/apimachinery/pkg/util/net"
|
|
)
|
|
|
|
// tokenExchanger abstracts the token exchange client for testability.
|
|
type tokenExchanger interface {
|
|
Exchange(ctx context.Context, req authn.TokenExchangeRequest) (*authn.TokenExchangeResponse, error)
|
|
}
|
|
|
|
// RoundTripper injects an exchanged access token for the provisioning API into outgoing requests.
|
|
type RoundTripper struct {
|
|
client tokenExchanger
|
|
transport http.RoundTripper
|
|
audience string
|
|
}
|
|
|
|
// NewRoundTripper constructs a RoundTripper that exchanges the provided token per request
|
|
// and forwards the request to the provided base transport.
|
|
func NewRoundTripper(tokenExchangeClient tokenExchanger, base http.RoundTripper, audience string) *RoundTripper {
|
|
return &RoundTripper{
|
|
client: tokenExchangeClient,
|
|
transport: base,
|
|
audience: audience,
|
|
}
|
|
}
|
|
|
|
func (t *RoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
tokenResponse, err := t.client.Exchange(req.Context(), authn.TokenExchangeRequest{
|
|
Audiences: []string{t.audience},
|
|
Namespace: "*",
|
|
})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to exchange token: %w", err)
|
|
}
|
|
|
|
req = utilnet.CloneRequest(req)
|
|
req.Header.Set("X-Access-Token", "Bearer "+tokenResponse.Token)
|
|
return t.transport.RoundTrip(req)
|
|
}
|