Merge pull request '♻️ refactor(auth): move UserContextKey from pkg/greet to pkg/auth' (#90) from vibe/batch-pr-d1-move-user-context-key into main
This commit was merged in pull request #90.
This commit is contained in:
30
pkg/auth/context.go
Normal file
30
pkg/auth/context.go
Normal file
@@ -0,0 +1,30 @@
|
|||||||
|
// Package auth — context keys and helpers for authentication state.
|
||||||
|
//
|
||||||
|
// This file owns the symbols that other packages use to read/write the
|
||||||
|
// authenticated user from a request context. Previously these symbols
|
||||||
|
// lived in pkg/greet/ which was the wrong home (auth concern in greet
|
||||||
|
// package) ; moved here as part of the middleware design cleanup.
|
||||||
|
package auth
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
|
||||||
|
"dance-lessons-coach/pkg/user"
|
||||||
|
)
|
||||||
|
|
||||||
|
// contextKey is unexported to prevent collisions with other packages
|
||||||
|
// using string keys (Go convention).
|
||||||
|
type contextKey string
|
||||||
|
|
||||||
|
// UserContextKey is the key under which the authenticated user is
|
||||||
|
// stored in the request context by AuthMiddleware.
|
||||||
|
const UserContextKey contextKey = "authenticatedUser"
|
||||||
|
|
||||||
|
// GetAuthenticatedUserFromContext extracts the authenticated user from
|
||||||
|
// the request context. Returns (nil, false) if no user is present
|
||||||
|
// (which is the case for unauthenticated requests AND for requests
|
||||||
|
// that failed silent fail-through ; cf. AuthMiddleware semantics).
|
||||||
|
func GetAuthenticatedUserFromContext(ctx context.Context) (*user.User, bool) {
|
||||||
|
u, ok := ctx.Value(UserContextKey).(*user.User)
|
||||||
|
return u, ok
|
||||||
|
}
|
||||||
@@ -3,31 +3,17 @@ package greet
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
|
||||||
"dance-lessons-coach/pkg/user"
|
"dance-lessons-coach/pkg/auth"
|
||||||
|
|
||||||
"github.com/rs/zerolog/log"
|
"github.com/rs/zerolog/log"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Context key for storing authenticated user
|
|
||||||
type contextKey string
|
|
||||||
|
|
||||||
const (
|
|
||||||
// UserContextKey is the context key for storing authenticated user
|
|
||||||
UserContextKey contextKey = "authenticatedUser"
|
|
||||||
)
|
|
||||||
|
|
||||||
type Service struct{}
|
type Service struct{}
|
||||||
|
|
||||||
func NewService() *Service {
|
func NewService() *Service {
|
||||||
return &Service{}
|
return &Service{}
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetAuthenticatedUserFromContext extracts the authenticated user from context
|
|
||||||
func GetAuthenticatedUserFromContext(ctx context.Context) (*user.User, bool) {
|
|
||||||
user, ok := ctx.Value(UserContextKey).(*user.User)
|
|
||||||
return user, ok
|
|
||||||
}
|
|
||||||
|
|
||||||
// Greet returns a greeting message for the given name.
|
// Greet returns a greeting message for the given name.
|
||||||
// If name is empty, it checks for authenticated user and uses their username.
|
// If name is empty, it checks for authenticated user and uses their username.
|
||||||
// If no authenticated user and no name, it defaults to "world".
|
// If no authenticated user and no name, it defaults to "world".
|
||||||
@@ -37,7 +23,7 @@ func (s *Service) Greet(ctx context.Context, name string) string {
|
|||||||
|
|
||||||
// If no name provided, check for authenticated user
|
// If no name provided, check for authenticated user
|
||||||
if name == "" {
|
if name == "" {
|
||||||
if authenticatedUser, ok := GetAuthenticatedUserFromContext(ctx); ok {
|
if authenticatedUser, ok := auth.GetAuthenticatedUserFromContext(ctx); ok {
|
||||||
name = authenticatedUser.Username
|
name = authenticatedUser.Username
|
||||||
log.Trace().Ctx(ctx).Str("authenticated_user", name).Msg("Using authenticated username for greeting")
|
log.Trace().Ctx(ctx).Str("authenticated_user", name).Msg("Using authenticated username for greeting")
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
|
||||||
"dance-lessons-coach/pkg/greet"
|
"dance-lessons-coach/pkg/auth"
|
||||||
"dance-lessons-coach/pkg/user"
|
"dance-lessons-coach/pkg/user"
|
||||||
|
|
||||||
"github.com/rs/zerolog/log"
|
"github.com/rs/zerolog/log"
|
||||||
@@ -54,7 +54,7 @@ func (m *AuthMiddleware) Middleware(next http.Handler) http.Handler {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Add user to context
|
// Add user to context
|
||||||
ctxWithUser := context.WithValue(ctx, greet.UserContextKey, validatedUser)
|
ctxWithUser := context.WithValue(ctx, auth.UserContextKey, validatedUser)
|
||||||
r = r.WithContext(ctxWithUser)
|
r = r.WithContext(ctxWithUser)
|
||||||
|
|
||||||
// Continue to next handler
|
// Continue to next handler
|
||||||
|
|||||||
Reference in New Issue
Block a user