Method ContextInject

coderd/agentapi/cached_workspace.go:70–82

ContextInject attempts to inject the rbac object for the cached workspace fields into the given context, either returning the wrapped context or the original.

(ctx context.Context)

// ContextInject attempts to inject the rbac object for the cached workspace fields
// into the given context, either returning the wrapped context or the original.
func (cws *CachedWorkspaceFields) ContextInject(ctx context.Context) (context.Context, error) {
	var err error
	rbacCtx := ctx
	if dbws, ok := cws.AsWorkspaceIdentity(); ok {
		rbacCtx, err = dbauthz.WithWorkspaceRBAC(ctx, dbws.RBACObject())
		if err != nil {
			// Don't error level log here, will exit the function. We want to fall back to GetWorkspaceByAgentID.
			//nolint:gocritic
			return ctx, xerrors.Errorf("Cached workspace was present but RBAC object was invalid: %w", err)
		}
	}
	return rbacCtx, nil
}