mirror of
https://github.com/khairul169/vaulterm.git
synced 2025-04-29 00:59:40 +07:00
23 lines
470 B
Go
23 lines
470 B
Go
package ws
|
|
|
|
import (
|
|
"github.com/gofiber/contrib/websocket"
|
|
"github.com/gofiber/fiber/v2"
|
|
"rul.sh/vaulterm/app/ws/stats"
|
|
"rul.sh/vaulterm/app/ws/term"
|
|
)
|
|
|
|
func Router(app fiber.Router) {
|
|
router := app.Group("/ws")
|
|
|
|
router.Use(func(c *fiber.Ctx) error {
|
|
if websocket.IsWebSocketUpgrade(c) {
|
|
return c.Next()
|
|
}
|
|
return fiber.ErrUpgradeRequired
|
|
})
|
|
|
|
router.Get("/term", websocket.New(term.HandleTerm))
|
|
router.Get("/stats", websocket.New(stats.HandleStats))
|
|
}
|