package context

import (
	"context"

	"github.com/influxdata/platform"
	"github.com/influxdata/platform/kit/errors"
)

type contextKey string

const (
	authorizerCtxKey = contextKey("influx/authorizer/v1")
)

// SetAuthorizer sets an authorizer on context.
func SetAuthorizer(ctx context.Context, a platform.Authorizer) context.Context {
	return context.WithValue(ctx, authorizerCtxKey, a)
}

// GetAuthorizer retrieves an authorizer from context.
func GetAuthorizer(ctx context.Context) (platform.Authorizer, error) {
	a, ok := ctx.Value(authorizerCtxKey).(platform.Authorizer)
	if !ok {
		return nil, errors.InternalErrorf("authorizer not found on context")
	}

	return a, nil
}