Extract authn & authz utils

Signed-off-by: Émile Ré <emile@getprobo.com>

pkg/server/api/authn/context.go

@@ -0,0 +1,58 @@
+// Copyright (c) 2025 Probo Inc <hello@getprobo.com>.
+//
+// Permission to use, copy, modify, and/or distribute this software for any
+// purpose with or without fee is hereby granted, provided that the above
+// copyright notice and this permission notice appear in all copies.
+//
+// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
+// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
+// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
+// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
+// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
+// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+// PERFORMANCE OF THIS SOFTWARE.
+
+package authn
+
+import (
+	"context"
+
+	"go.probo.inc/probo/pkg/coredata"
+)
+
+type (
+	ctxKey struct{ name string }
+)
+
+var (
+	identityContextKey = &ctxKey{name: "identity"}
+	sessionContextKey  = &ctxKey{name: "session"}
+	apiKeyContextKey   = &ctxKey{name: "api_key"}
+)
+
+func SessionFromContext(ctx context.Context) *coredata.Session {
+	session, _ := ctx.Value(sessionContextKey).(*coredata.Session)
+	return session
+}
+
+func ContextWithSession(ctx context.Context, session *coredata.Session) context.Context {
+	return context.WithValue(ctx, sessionContextKey, session)
+}
+
+func IdentityFromContext(ctx context.Context) *coredata.Identity {
+	identity, _ := ctx.Value(identityContextKey).(*coredata.Identity)
+	return identity
+}
+
+func ContextWithIdentity(ctx context.Context, identity *coredata.Identity) context.Context {
+	return context.WithValue(ctx, identityContextKey, identity)
+}
+
+func APIKeyFromContext(ctx context.Context) *coredata.PersonalAPIKey {
+	apiKey, _ := ctx.Value(apiKeyContextKey).(*coredata.PersonalAPIKey)
+	return apiKey
+}
+
+func ContextWithAPIKey(ctx context.Context, apiKey *coredata.PersonalAPIKey) context.Context {
+	return context.WithValue(ctx, apiKeyContextKey, apiKey)
+}