iridium/neko
2
2
Fork 0
mirror of https://github.com/m1k1o/neko.git synced 2024-07-24 14:40:50 +12:00
Code Issues Packages Projects Releases Wiki Activity

AddRouter in API.

Browse Source
This commit is contained in:
Miroslav Šedivý 2020-12-12 17:07:45 +01:00
parent 9786c8d537
commit af463b1015
1 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
internal/api/router.go
View File

@ -14,9 +14,10 @@ import (
)
type ApiManagerCtx struct {
sessions types.SessionManager
desktop types.DesktopManager
capture types.CaptureManager
sessions types.SessionManager
desktop types.DesktopManager
capture types.CaptureManager
routers map[string]func(chi.Router)
}
func New(
@ -27,9 +28,10 @@ func New(
) *ApiManagerCtx {
return &ApiManagerCtx{
sessions: sessions,
desktop: desktop,
capture: capture,
sessions: sessions,
desktop: desktop,
capture: capture,
routers: make(map[string]func(chi.Router)),
}
}
@ -46,6 +48,10 @@ func (api *ApiManagerCtx) Route(r chi.Router) {
session := auth.GetSession(r)
utils.HttpBadRequest(w, "Hi `" + session.ID() + "`, you are authenticated.")
})
for path, router := range api.routers {
r.Route(path, router)
}
}
func (api *ApiManagerCtx) Authenticate(next http.Handler) http.Handler {
@ -58,3 +64,7 @@ func (api *ApiManagerCtx) Authenticate(next http.Handler) http.Handler {
}
})
}
func (api *ApiManagerCtx) AddRouter(path string, router func(chi.Router)) {
api.routers[path] = router
}