Aleksey Shakhmatov
a90c4cd7fa
- Logging: structured slog output with method, URL, status, duration - DefaultHeaders/UserAgent: inject headers without overwriting existing - BearerAuth/BasicAuth: per-request token resolution and static credentials - Recovery: catches panics in the RoundTripper chain
34 lines
1.0 KiB
Go
34 lines
1.0 KiB
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
)
|
|
|
|
// BearerAuth returns a middleware that sets the Authorization header to a
|
|
// Bearer token obtained by calling tokenFunc on each request. If tokenFunc
|
|
// returns an error, the request is not sent and the error is returned.
|
|
func BearerAuth(tokenFunc func(ctx context.Context) (string, error)) Middleware {
|
|
return func(next http.RoundTripper) http.RoundTripper {
|
|
return RoundTripperFunc(func(req *http.Request) (*http.Response, error) {
|
|
token, err := tokenFunc(req.Context())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
req.Header.Set("Authorization", "Bearer "+token)
|
|
return next.RoundTrip(req)
|
|
})
|
|
}
|
|
}
|
|
|
|
// BasicAuth returns a middleware that sets HTTP Basic Authentication
|
|
// credentials on every outgoing request.
|
|
func BasicAuth(username, password string) Middleware {
|
|
return func(next http.RoundTripper) http.RoundTripper {
|
|
return RoundTripperFunc(func(req *http.Request) (*http.Response, error) {
|
|
req.SetBasicAuth(username, password)
|
|
return next.RoundTrip(req)
|
|
})
|
|
}
|
|
}
|