mirror of
https://github.com/harness/drone.git
synced 2025-05-06 04:50:38 +08:00
8c2f900c80
This change introduces the concept of a principal (abstraction of call identity), and adds a new service account type principal. Also adds support for different tokens (session, PAT, SAT, OAuth2) and adds auth.Session which is being used to capture information about the caller and call method.
97 lines
3.0 KiB
Go
97 lines
3.0 KiB
Go
// Copyright 2021 Harness Inc. All rights reserved.
|
|
// Use of this source code is governed by the Polyform Free Trial License
|
|
// that can be found in the LICENSE.md file for this repository.
|
|
|
|
package request
|
|
|
|
// This pattern was inpired by the kubernetes request context package.
|
|
// https://github.com/kubernetes/apiserver/blob/master/pkg/endpoints/request/context.go
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/harness/gitness/internal/auth"
|
|
"github.com/harness/gitness/types"
|
|
)
|
|
|
|
type key int
|
|
|
|
const (
|
|
authSessionKey key = iota
|
|
serviceAccountKey
|
|
userKey
|
|
spaceKey
|
|
repoKey
|
|
)
|
|
|
|
// WithAuthSession returns a copy of parent in which the principal
|
|
// value is set.
|
|
func WithAuthSession(parent context.Context, v *auth.Session) context.Context {
|
|
return context.WithValue(parent, authSessionKey, v)
|
|
}
|
|
|
|
// AuthSessionFrom returns the value of the principal key on the
|
|
// context.
|
|
func AuthSessionFrom(ctx context.Context) (*auth.Session, bool) {
|
|
v, ok := ctx.Value(authSessionKey).(*auth.Session)
|
|
return v, ok && v != nil
|
|
}
|
|
|
|
// PrincipalFrom returns the principal of the authsession.
|
|
func PrincipalFrom(ctx context.Context) (*types.Principal, bool) {
|
|
v, ok := AuthSessionFrom(ctx)
|
|
if !ok {
|
|
return nil, false
|
|
}
|
|
|
|
return &v.Principal, true
|
|
}
|
|
|
|
// WithUser returns a copy of parent in which the user value is set.
|
|
func WithUser(parent context.Context, v *types.User) context.Context {
|
|
return context.WithValue(parent, userKey, v)
|
|
}
|
|
|
|
// UserFrom returns the value of the user key on the
|
|
// context - ok is true iff a non-nile value existed.
|
|
func UserFrom(ctx context.Context) (*types.User, bool) {
|
|
v, ok := ctx.Value(userKey).(*types.User)
|
|
return v, ok && v != nil
|
|
}
|
|
|
|
// WithServiceAccount returns a copy of parent in which the service account value is set.
|
|
func WithServiceAccount(parent context.Context, v *types.ServiceAccount) context.Context {
|
|
return context.WithValue(parent, serviceAccountKey, v)
|
|
}
|
|
|
|
// ServiceAccountFrom returns the value of the service account key on the
|
|
// context - ok is true iff a non-nile value existed.
|
|
func ServiceAccountFrom(ctx context.Context) (*types.ServiceAccount, bool) {
|
|
v, ok := ctx.Value(serviceAccountKey).(*types.ServiceAccount)
|
|
return v, ok && v != nil
|
|
}
|
|
|
|
// WithSpace returns a copy of parent in which the space value is set.
|
|
func WithSpace(parent context.Context, v *types.Space) context.Context {
|
|
return context.WithValue(parent, spaceKey, v)
|
|
}
|
|
|
|
// SpaceFrom returns the value of the space key on the
|
|
// context - ok is true iff a non-nile value existed.
|
|
func SpaceFrom(ctx context.Context) (*types.Space, bool) {
|
|
v, ok := ctx.Value(spaceKey).(*types.Space)
|
|
return v, ok && v != nil
|
|
}
|
|
|
|
// WithRepo returns a copy of parent in which the repo value is set.
|
|
func WithRepo(parent context.Context, v *types.Repository) context.Context {
|
|
return context.WithValue(parent, repoKey, v)
|
|
}
|
|
|
|
// RepoFrom returns the value of the repo key on the
|
|
// context - ok is true iff a non-nile value existed.
|
|
func RepoFrom(ctx context.Context) (*types.Repository, bool) {
|
|
v, ok := ctx.Value(repoKey).(*types.Repository)
|
|
return v, ok && v != nil
|
|
}
|