diff --git a/internal/router/helpers.go b/internal/router/helpers.go index 4c68734..5542514 100644 --- a/internal/router/helpers.go +++ b/internal/router/helpers.go @@ -35,12 +35,19 @@ func getTemplateUserWithTheme(c *echo.Context, cfg *Config) (templates.User, err userTheme = userDB.Theme.String } + // Extract JWT token for WebSocket authentication + token := "" + if cookie, err := c.Cookie("token"); err == nil { + token = cookie.Value + } + return templates.User{ ID: userID, Email: userEmail, Username: userUsername, Role: userRole, Theme: userTheme, + Token: token, }, nil } diff --git a/templates/types.go b/templates/types.go index e53748f..40d3eea 100644 --- a/templates/types.go +++ b/templates/types.go @@ -15,6 +15,7 @@ type User struct { FirstName string LastName string CreatedAt time.Time + Token string } type PageData struct {