package llamactl

import (
	"github.com/go-chi/chi/v5"
	"github.com/go-chi/chi/v5/middleware"
	httpSwagger "github.com/swaggo/http-swagger"

	_ "llamactl/docs"
)

func SetupRouter(handler *Handler) *chi.Mux {
	r := chi.NewRouter()
	r.Use(middleware.Logger)

	r.Get("/swagger/*", httpSwagger.Handler(
		httpSwagger.URL("/swagger/doc.json"),
	))

	// Define routes
	r.Route("/api/v1", func(r chi.Router) {
		r.Get("/server/help", handler.HelpHandler())
		r.Get("/server/version", handler.VersionHandler())
		r.Get("/server/devices", handler.ListDevicesHandler())

		// Launch and stop handlers
		// r.Post("/server/launch/{model}", launchHandler)
		// r.Post("/server/stop/{model}", stopHandler)
	})

	return r
}