verification/internal/server/http.go

package server

import (
	"context"
	"fmt"
	"github.com/gofiber/contrib/fiberzap"
	"github.com/gofiber/fiber/v2"
	"github.com/gofiber/fiber/v2/middleware/recover"
	"github.com/themakers/hlog"
	"go.uber.org/zap"
)

type ServerConfig struct {
	Logger      *zap.Logger
	Controllers []Controller
	HLogger     hlog.Logger
}

type Server struct {
	Logger      *zap.Logger
	Controllers []Controller
	app         *fiber.App
}

func NewHTTP(cfg ServerConfig) *Server {
	srv := fiber.New(fiber.Config{
		AppName:      "Verification",
		ErrorHandler: fiber.DefaultErrorHandler,
		BodyLimit:    50 << 20,
	})

	srv.Use(
		recover.New(recover.Config{EnableStackTrace: true}),
		fiberzap.New(fiberzap.Config{Logger: cfg.Logger}),
		LocalJwt(),
		Jwt(),
		ContextLogger(cfg.HLogger),
	)

	s := &Server{
		Logger:      cfg.Logger,
		Controllers: cfg.Controllers,
		app:         srv,
	}

	s.registerRoutes()

	return s
}

func (s *Server) Start(addr string) error {
	if err := s.app.Listen(addr); err != nil {
		s.Logger.Error("Failed to start server", zap.Error(err))
		return err
	}
	return nil
}

func (s *Server) Shutdown(ctx context.Context) error {
	return s.app.Shutdown()
}

func (s *Server) registerRoutes() {
	for _, c := range s.Controllers {
		router := s.app.Group(c.Name())
		c.Register(router)
	}
}

type Controller interface {
	Register(router fiber.Router)
	Name() string
}

func (s *Server) ListRoutes() {
	fmt.Println("Registered routes:")
	for _, stack := range s.app.Stack() {
		for _, route := range stack {
			fmt.Printf("%s %s\n", route.Method, route.Path)
		}
	}
}