diff options
Diffstat (limited to 'graphql/graphqlidentity')
-rw-r--r-- | graphql/graphqlidentity/graphqlidentity.go | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/graphql/graphqlidentity/graphqlidentity.go b/graphql/graphqlidentity/graphqlidentity.go new file mode 100644 index 00000000..6851c4a8 --- /dev/null +++ b/graphql/graphqlidentity/graphqlidentity.go @@ -0,0 +1,39 @@ +// Package graphqlidentity contains helpers for managing identities within the GraphQL API. +package graphqlidentity + +import ( + "context" + + "github.com/MichaelMure/git-bug/cache" + "github.com/MichaelMure/git-bug/entity" + "github.com/MichaelMure/git-bug/identity" + "github.com/MichaelMure/git-bug/repository" +) + +// identityCtxKey is a unique context key, accessible only in this package. +var identityCtxKey = &struct{}{} + +// AttachToContext attaches an Identity to a context. +func AttachToContext(ctx context.Context, u *identity.Identity) context.Context { + return context.WithValue(ctx, identityCtxKey, u.Id()) +} + +// ForContext retrieves an IdentityCache from the context, or nil if no identity is present. +// If an error occurs while resolving the identity (e.g. I/O error), then it will be returned. +func ForContext(ctx context.Context, r *cache.RepoCache) (*cache.IdentityCache, error) { + id, ok := ctx.Value(identityCtxKey).(entity.Id) + if !ok { + return nil, nil + } + return r.ResolveIdentity(id) +} + +// ForContextUncached retrieves an Identity from the context, or nil if no identity is present. +// If an error occurs while resolving the identity (e.g. I/O error), then it will be returned. +func ForContextUncached(ctx context.Context, repo repository.Repo) (*identity.Identity, error) { + id, ok := ctx.Value(identityCtxKey).(entity.Id) + if !ok { + return nil, nil + } + return identity.ReadLocal(repo, id) +} |