package http import ( "fmt" "net/http" "git.kmsign.ru/royalcat/tstor" "git.kmsign.ru/royalcat/tstor/src/config" "git.kmsign.ru/royalcat/tstor/src/host/torrent" "github.com/anacrolix/missinggo/v2/filecache" "github.com/gin-contrib/pprof" "github.com/gin-gonic/gin" "github.com/rs/zerolog/log" "github.com/shurcooL/httpfs/html/vfstemplate" ) func New(fc *filecache.Cache, ss *torrent.Stats, s *torrent.Service, logPath string, cfg *config.Config) error { 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()) 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().Str("host", fmt.Sprintf("%s:%d", cfg.WebUi.IP, cfg.WebUi.Port)).Msg("starting webserver") 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 := log.Logger.With().Str("component", "http").Logger() 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().Str("path", path).Int("status", s).Msg(msg) case s >= 500: l.Error().Str("path", path).Int("status", s).Msg(msg) default: l.Debug().Str("path", path).Int("status", s).Msg(msg) } } }