package httpserver

import (
	"net/http"

	"go-backend/internal/http/handler"
	"go-backend/internal/http/middleware"
)

func NewRouter(h *handler.Handler, jwtSecret string) http.Handler {
	mux := http.NewServeMux()
	h.Register(mux)
	mux.Handle("/system-info", h.WebSocketHandler())

	wrapped := middleware.Recover(mux)
	wrapped = middleware.JWT(middleware.AuthOptions{JWTSecret: jwtSecret})(wrapped)
	wrapped = middleware.CORS(wrapped)
	return wrapped
}