Redmage/server/routes/routes.go

package routes

import (
	"io/fs"
	"net/http"
	"os"

	"github.com/go-chi/chi/v5"
	chimiddleware "github.com/go-chi/chi/v5/middleware"
	"github.com/riandyrn/otelchi"
	"github.com/tigorlazuardi/redmage/api"
	"github.com/tigorlazuardi/redmage/config"
	"github.com/tigorlazuardi/redmage/server/routes/middleware"
)

type Routes struct {
	API       *api.API
	Config    *config.Config
	PublicDir fs.FS
}

func (routes *Routes) Register(router chi.Router) {
	router.Use(chimiddleware.Compress(5, "text/html", "text/css", "application/javascript", "application/json", "image/svg+xml", "image/x-icon"))

	router.HandleFunc("/ping", routes.HealthCheck)
	router.HandleFunc("/health", routes.HealthCheck)
	if routes.Config.Bool("http.hotreload") {
		router.Get("/hot_reload", routes.CreateHotReloadRoute())
	}

	router.Group(routes.registerWWWRoutes)
	router.Route("/api/v1", routes.registerV1APIRoutes)
}

func (routes *Routes) registerV1APIRoutes(router chi.Router) {
	router.Use(otelchi.Middleware("redmage"))
	router.Use(chimiddleware.RequestLogger(middleware.ChiLogger{}))
	router.Use(chimiddleware.SetHeader("Content-Type", "application/json"))

	router.Post("/subreddits/start", routes.SubredditStartDownloadAPI)
	router.Get("/subreddits", routes.SubredditsListAPI)
	router.Post("/subreddits", routes.SubredditsCreateAPI)
	router.Post("/subreddits/check", routes.SubredditsCheckAPI)

	router.Get("/devices", routes.APIDeviceList)
	router.Post("/devices", routes.APIDeviceCreate)
	router.Patch("/devices/{id}", routes.APIDeviceUpdate)

	router.Get("/images", routes.ImagesListAPI)

	router.Get("/events", routes.EventsAPI)
}

func (routes *Routes) registerWWWRoutes(router chi.Router) {
	router.Mount("/public", http.StripPrefix("/public", http.FileServer(http.FS(routes.PublicDir))))

	imagesDir := http.FS(os.DirFS(routes.Config.String("download.directory")))

	router.Mount("/img", http.StripPrefix("/img", http.FileServer(imagesDir)))

	router.Group(func(r chi.Router) {
		r.Use(otelchi.Middleware("redmage"))
		r.Use(chimiddleware.RequestID)
		r.Use(chimiddleware.RequestLogger(middleware.ChiLogger{}))
		r.Use(chimiddleware.SetHeader("Content-Type", "text/html; charset=utf-8"))
		r.Get("/", routes.PageHome)
	})
}