package api import ( "net/http" "github.com/go-chi/chi" "demodesk/neko/internal/api/members" "demodesk/neko/internal/api/room" "demodesk/neko/internal/config" "demodesk/neko/internal/http/auth" "demodesk/neko/internal/types" "demodesk/neko/internal/utils" ) type ApiManagerCtx struct { sessions types.SessionManager members types.MemberManager desktop types.DesktopManager capture types.CaptureManager routers map[string]func(chi.Router) } func New( sessions types.SessionManager, members types.MemberManager, desktop types.DesktopManager, capture types.CaptureManager, conf *config.Server, ) *ApiManagerCtx { return &ApiManagerCtx{ sessions: sessions, members: members, desktop: desktop, capture: capture, routers: make(map[string]func(chi.Router)), } } func (api *ApiManagerCtx) Route(r chi.Router) { r.Post("/login", api.Login) // Authenticated area r.Group(func(r chi.Router) { r.Use(api.Authenticate) r.Post("/logout", api.Logout) r.Get("/whoami", api.Whoami) membersHandler := members.New(api.members) r.Route("/members", membersHandler.Route) r.Route("/members_bulk", membersHandler.RouteBulk) roomHandler := room.New(api.sessions, api.desktop, api.capture) r.Route("/room", roomHandler.Route) for path, router := range api.routers { r.Route(path, router) } }) r.Get("/health", func(w http.ResponseWriter, r *http.Request) { //nolint w.Write([]byte("true")) }) } func (api *ApiManagerCtx) Authenticate(next http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { session, err := api.sessions.Authenticate(r) if err != nil { if api.sessions.CookieEnabled() { api.sessions.CookieClearToken(w, r) } utils.HttpUnauthorized(w, err) return } next.ServeHTTP(w, auth.SetSession(r, session)) }) } func (api *ApiManagerCtx) AddRouter(path string, router func(chi.Router)) { api.routers[path] = router }