2024-11-10 18:49:18 +07:00
|
|
|
package utils
|
|
|
|
|
|
|
|
import (
|
|
|
|
"github.com/gofiber/contrib/websocket"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
2024-11-16 02:34:07 +07:00
|
|
|
"rul.sh/vaulterm/server/middleware"
|
2024-11-10 18:49:18 +07:00
|
|
|
)
|
|
|
|
|
2024-11-12 17:17:10 +00:00
|
|
|
type UserContext = middleware.AuthUser
|
2024-11-10 18:49:18 +07:00
|
|
|
|
|
|
|
func GetUser(c *fiber.Ctx) *UserContext {
|
2024-11-12 17:17:10 +00:00
|
|
|
user, _ := c.Locals("user").(*UserContext)
|
|
|
|
return user
|
2024-11-10 18:49:18 +07:00
|
|
|
}
|
|
|
|
|
|
|
|
func GetUserWs(c *websocket.Conn) *UserContext {
|
2024-11-12 17:17:10 +00:00
|
|
|
user, _ := c.Locals("user").(*UserContext)
|
|
|
|
return user
|
2024-11-10 18:49:18 +07:00
|
|
|
}
|