iknowyou/back/internal/api/router.go

package api

import (
	"time"

	"github.com/go-chi/chi/v5"
	chimiddleware "github.com/go-chi/chi/v5/middleware"
	"golang.org/x/time/rate"

	"github.com/anotherhadi/iknowyou/internal/api/handler"
	ikymiddleware "github.com/anotherhadi/iknowyou/internal/api/middleware"
	"github.com/anotherhadi/iknowyou/internal/search"
	"github.com/anotherhadi/iknowyou/internal/tools"
)

func NewRouter(
	manager *search.Manager,
	factories []func() tools.ToolRunner,
	configPath string,
	frontDir string,
	demo bool,
) *chi.Mux {
	r := chi.NewRouter()

	r.Use(chimiddleware.Logger)
	r.Use(chimiddleware.Recoverer)
	r.Use(chimiddleware.RequestID)

	searchHandler := handler.NewSearchHandler(manager, demo)
	toolsHandler := handler.NewToolsHandler(factories)
	configHandler := handler.NewConfigHandler(configPath, factories, demo)

	searchLimiter := ikymiddleware.New(rate.Every(10*time.Second), 3)

	r.Route("/api", func(r chi.Router) {
		r.Route("/searches", func(r chi.Router) {
			r.With(searchLimiter.Handler).Post("/", searchHandler.Create)
			r.Get("/", searchHandler.List)
			r.Get("/{id}", searchHandler.Get)
			r.Delete("/{id}", searchHandler.Delete)
		})

		r.Route("/tools", func(r chi.Router) {
			r.Get("/", toolsHandler.List)
			r.Get("/{name}", toolsHandler.Get)
		})

		r.Route("/config", func(r chi.Router) {
			r.Get("/", configHandler.Get)

			r.Route("/tools", func(r chi.Router) {
				r.Patch("/{toolName}", configHandler.UpdateToolConfig)
				r.Delete("/{toolName}", configHandler.DeleteToolConfig)
			})

			r.Route("/profiles", func(r chi.Router) {
				r.Get("/", configHandler.ListProfiles)
				r.Post("/", configHandler.CreateProfile)
				r.Get("/{name}", configHandler.GetProfile)
				r.Patch("/{name}", configHandler.UpdateProfile)
				r.Delete("/{name}", configHandler.DeleteProfile)
				r.Patch("/{name}/tools/{toolName}", configHandler.UpdateProfileToolConfig)
				r.Delete("/{name}/tools/{toolName}", configHandler.DeleteProfileToolConfig)
			})
		})
	})

	if frontDir != "" {
		r.Handle("/*", newStaticHandler(frontDir))
	}

	return r
}