60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
package context
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/influxdata/influxdb"
|
|
)
|
|
|
|
type contextKey string
|
|
|
|
const (
|
|
authorizerCtxKey contextKey = "influx/authorizer/v1"
|
|
)
|
|
|
|
// SetAuthorizer sets an authorizer on context.
|
|
func SetAuthorizer(ctx context.Context, a influxdb.Authorizer) context.Context {
|
|
return context.WithValue(ctx, authorizerCtxKey, a)
|
|
}
|
|
|
|
// GetAuthorizer retrieves an authorizer from context.
|
|
func GetAuthorizer(ctx context.Context) (influxdb.Authorizer, error) {
|
|
a, ok := ctx.Value(authorizerCtxKey).(influxdb.Authorizer)
|
|
if !ok {
|
|
return nil, &influxdb.Error{
|
|
Msg: "authorizer not found on context",
|
|
Code: influxdb.EInternal,
|
|
}
|
|
}
|
|
if a == nil {
|
|
return nil, &influxdb.Error{
|
|
Code: influxdb.EInternal,
|
|
Msg: "unexpected invalid authorizer",
|
|
}
|
|
}
|
|
|
|
return a, nil
|
|
}
|
|
|
|
// GetToken retrieves a token from the context; errors if no token.
|
|
func GetToken(ctx context.Context) (string, error) {
|
|
a, ok := ctx.Value(authorizerCtxKey).(influxdb.Authorizer)
|
|
if !ok {
|
|
return "", &influxdb.Error{
|
|
Msg: "authorizer not found on context",
|
|
Code: influxdb.EInternal,
|
|
}
|
|
}
|
|
|
|
auth, ok := a.(*influxdb.Authorization)
|
|
if !ok {
|
|
return "", &influxdb.Error{
|
|
Msg: fmt.Sprintf("authorizer not an authorization but a %T", a),
|
|
Code: influxdb.EInternal,
|
|
}
|
|
}
|
|
|
|
return auth.Token, nil
|
|
}
|