mirror of
https://github.com/khairul169/vaulterm.git
synced 2025-04-28 16:49:39 +07:00
36 lines
624 B
Go
36 lines
624 B
Go
package utils
|
|
|
|
import (
|
|
"github.com/gofiber/contrib/websocket"
|
|
"github.com/gofiber/fiber/v2"
|
|
"rul.sh/vaulterm/models"
|
|
)
|
|
|
|
type UserContext struct {
|
|
*models.User
|
|
IsAdmin bool `json:"isAdmin"`
|
|
}
|
|
|
|
func getUserData(user *models.User) *UserContext {
|
|
isAdmin := false
|
|
|
|
if user.Role == models.UserRoleAdmin {
|
|
isAdmin = true
|
|
}
|
|
|
|
return &UserContext{
|
|
User: user,
|
|
IsAdmin: isAdmin,
|
|
}
|
|
}
|
|
|
|
func GetUser(c *fiber.Ctx) *UserContext {
|
|
user := c.Locals("user").(*models.User)
|
|
return getUserData(user)
|
|
}
|
|
|
|
func GetUserWs(c *websocket.Conn) *UserContext {
|
|
user := c.Locals("user").(*models.User)
|
|
return getUserData(user)
|
|
}
|