tstor/src/delivery/http.go

package delivery

import (
	"fmt"
	"log/slog"
	"net/http"

	"git.kmsign.ru/royalcat/tstor"
	"git.kmsign.ru/royalcat/tstor/src/config"
	"git.kmsign.ru/royalcat/tstor/src/host/service"
	"git.kmsign.ru/royalcat/tstor/src/host/vfs"
	"github.com/anacrolix/missinggo/v2/filecache"
	"github.com/gin-contrib/pprof"
	"github.com/gin-gonic/gin"
	"github.com/shurcooL/httpfs/html/vfstemplate"
)

func New(fc *filecache.Cache, ss *service.Stats, s *service.Service, vfs vfs.Filesystem, logPath string, cfg *config.Settings) error {
	log := slog.With()

	gin.SetMode(gin.ReleaseMode)
	r := gin.New()
	r.Use(gin.Recovery())
	r.Use(gin.ErrorLogger())
	r.Use(Logger())
	pprof.Register(r)

	r.GET("/assets/*filepath", func(c *gin.Context) {
		c.FileFromFS(c.Request.URL.Path, http.FS(tstor.Assets))
	})

	t, err := vfstemplate.ParseGlob(http.FS(tstor.Templates), nil, "/templates/*")
	if err != nil {
		return fmt.Errorf("error parsing html: %w", err)
	}

	r.SetHTMLTemplate(t)

	r.GET("/", indexHandler)
	// r.GET("/routes", routesHandler(ss))
	r.GET("/logs", logsHandler)
	r.GET("/servers", serversFoldersHandler())
	r.Any("/graphql", gin.WrapH(GraphQLHandler(s, vfs)))

	api := r.Group("/api")
	{
		api.GET("/log", apiLogHandler(logPath))
		api.GET("/status", apiStatusHandler(fc, ss))
		// api.GET("/servers", apiServersHandler(tss))
		// api.GET("/routes", apiRoutesHandler(ss))
		// api.POST("/routes/:route/torrent", apiAddTorrentHandler(s))
		// api.DELETE("/routes/:route/torrent/:torrent_hash", apiDelTorrentHandler(s))
	}

	log.Info("starting webserver", "host", fmt.Sprintf("%s:%d", cfg.WebUi.IP, cfg.WebUi.Port))

	if err := r.Run(fmt.Sprintf("%s:%d", cfg.WebUi.IP, cfg.WebUi.Port)); err != nil {
		return fmt.Errorf("error initializing server: %w", err)
	}

	return nil
}

func Logger() gin.HandlerFunc {
	l := slog.With("component", "http")
	return func(c *gin.Context) {
		path := c.Request.URL.Path
		raw := c.Request.URL.RawQuery
		c.Next()
		if raw != "" {
			path = path + "?" + raw
		}
		msg := c.Errors.String()
		if msg == "" {
			msg = "Request"
		}

		s := c.Writer.Status()
		switch {
		case s >= 400 && s < 500:
			l.Warn(msg, "path", path, "status", s)
		case s >= 500:
			l.Error(msg, "path", path, "status", s)
		default:
			l.Debug(msg, "path", path, "status", s)
		}
	}
}