package auth

import "context"

type contextKey string

const (
	userContextKey       contextKey = "user"
	publicUserContextKey contextKey = "public_user"
)

// GetUserFromContext retrieves admin user claims from context.
func GetUserFromContext(ctx context.Context) (*Claims, bool) {
	claims, ok := ctx.Value(userContextKey).(*Claims)
	return claims, ok
}

// GetPublicUserFromContext retrieves public user claims from context.
func GetPublicUserFromContext(ctx context.Context) (*PublicClaims, bool) {
	claims, ok := ctx.Value(publicUserContextKey).(*PublicClaims)
	return claims, ok
}

// WithUser adds admin user claims to context.
// Used by CMS auth middleware — not typically called by plugins.
func WithUser(ctx context.Context, claims *Claims) context.Context {
	return context.WithValue(ctx, userContextKey, claims)
}

// WithPublicUser adds public user claims to context.
// Used by CMS auth middleware — not typically called by plugins.
func WithPublicUser(ctx context.Context, claims *PublicClaims) context.Context {
	return context.WithValue(ctx, publicUserContextKey, claims)
}