83 lines
3.0 KiB
Go
83 lines
3.0 KiB
Go
package auth
|
|
|
|
import (
|
|
"context"
|
|
|
|
"allaboutapps.dev/aw/go-starter/internal/data/dto"
|
|
"allaboutapps.dev/aw/go-starter/internal/util"
|
|
"github.com/go-openapi/swag"
|
|
"github.com/labstack/echo/v4"
|
|
)
|
|
|
|
// EnrichContextWithCredentials stores the provided credentials in the form of user and access token used for authentication
|
|
// in the give context and updates the logger associated with ctx to include the user's ID.
|
|
func EnrichContextWithCredentials(ctx context.Context, result Result) context.Context {
|
|
// Retrieve current logger associated with context and extend it ID of authenticated user
|
|
l := util.LogFromContext(ctx).With().Str("userID", result.User.ID).Logger()
|
|
ctx = l.WithContext(ctx)
|
|
|
|
// Store authenticated user's instance in context
|
|
ctx = context.WithValue(ctx, util.CTXKeyUser, result.User)
|
|
// Store access token used for authentication in context
|
|
ctx = context.WithValue(ctx, util.CTXKeyAccessToken, result.Token)
|
|
|
|
return ctx
|
|
}
|
|
|
|
// EnrichEchoContextWithCredentials stores the provided credentials in the form of user and access token user for authentication
|
|
// in the given echo context's request and updates the logger associated with c to include the user's ID.
|
|
func EnrichEchoContextWithCredentials(c echo.Context, result Result) echo.Context {
|
|
// Get current context and enrich it with credentials
|
|
req := c.Request()
|
|
ctx := EnrichContextWithCredentials(req.Context(), result)
|
|
|
|
// Set updated request with enriched context in echo context
|
|
c.SetRequest(req.WithContext(ctx))
|
|
|
|
return c
|
|
}
|
|
|
|
// UserFromContext returns the user model of the currently authenticated user from a context. If no authentication was provided
|
|
// or the current context does not carry any user information, nil will be returned instead.
|
|
func UserFromContext(ctx context.Context) *dto.User {
|
|
u := ctx.Value(util.CTXKeyUser)
|
|
if u == nil {
|
|
return nil
|
|
}
|
|
|
|
user, ok := u.(*dto.User)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
|
|
return user
|
|
}
|
|
|
|
// UserFromEchoContext returns the user model of the currently authenticated user from an echo context. If no authentication was
|
|
// provided or the current echo context does not carry any user information, nil will be returned instead.
|
|
func UserFromEchoContext(c echo.Context) *dto.User {
|
|
return UserFromContext(c.Request().Context())
|
|
}
|
|
|
|
// AccessTokenFromContext returns the access token model of the token used to authentication from a context. If no authentication was
|
|
// provided or the current context does not carry any access token information, nil will be returned instead.
|
|
func AccessTokenFromContext(ctx context.Context) *string {
|
|
t := ctx.Value(util.CTXKeyAccessToken)
|
|
if t == nil {
|
|
return nil
|
|
}
|
|
|
|
token, ok := t.(string)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
|
|
return swag.String(token)
|
|
}
|
|
|
|
// AccessTokenFromEchoContext returns the access token model of the token used to authentication from an echo context. If no authentication
|
|
// was provided or the current context does not carry any access token information, nil will be returned instead.
|
|
func AccessTokenFromEchoContext(c echo.Context) *string {
|
|
return AccessTokenFromContext(c.Request().Context())
|
|
}
|