package server

import (
	"leafdev.top/Leaf/leaf-library-3/internal/pkg/response"
	"net/http"
	"strings"

	"leafdev.top/Leaf/leaf-library-3/internal/errs"

	"github.com/ansrivas/fiberprometheus/v2"
	"github.com/bytedance/sonic"
	"github.com/gofiber/fiber/v2"
	"github.com/gofiber/fiber/v2/middleware/cors"
	"github.com/gofiber/fiber/v2/middleware/recover"
	httpApi "leafdev.top/Leaf/leaf-library-3/internal/api/http"
	"leafdev.top/Leaf/leaf-library-3/internal/base/conf"
	"leafdev.top/Leaf/leaf-library-3/internal/base/logger"
	"leafdev.top/Leaf/leaf-library-3/internal/router"
	"leafdev.top/Leaf/leaf-library-3/internal/services/auth"
)

type HttpServer struct {
	config        *conf.Config
	Fiber         *fiber.App
	apiRouter     *router.Api
	swaggerRouter *router.SwaggerRouter
	middleware    *httpApi.Middleware
	authService   *auth.Service
}

// NewHTTPServer new http server.
func NewHTTPServer(
	config *conf.Config,
	apiRouter *router.Api,
	swaggerRouter *router.SwaggerRouter,
	middleware *httpApi.Middleware,
	logger *logger.Logger,
) *HttpServer {
	app := fiber.New(fiber.Config{
		JSONEncoder: sonic.Marshal,
		JSONDecoder: sonic.Unmarshal,
		ErrorHandler: func(ctx *fiber.Ctx, err error) error {
			return errorConverter(logger, ctx, err)
		},
	})
	app.Use(recover.New())
	app.Use(middleware.Logger.Handler())
	app.Use(middleware.JSONResponse.Handler())

	return &HttpServer{
		config:        config,
		Fiber:         app,
		apiRouter:     apiRouter,
		swaggerRouter: swaggerRouter,
		middleware:    middleware,
	}
}

func (hs *HttpServer) AllowAllCors() {
	if hs.config.Http.Cors.Enabled {
		var config = cors.Config{
			AllowOrigins:     strings.Join(hs.config.Http.Cors.AllowedOrigins, ","),
			AllowMethods:     strings.Join(hs.config.Http.Cors.AllowMethods, ","),
			AllowHeaders:     strings.Join(hs.config.Http.Cors.AllowHeaders, ","),
			AllowCredentials: hs.config.Http.Cors.AllowCredentials,
			ExposeHeaders:    strings.Join(hs.config.Http.Cors.ExposeHeaders, ","),
			MaxAge:           hs.config.Http.Cors.MaxAge,
		}

		hs.Fiber.Use(cors.New(config))
	}
}

func (hs *HttpServer) BizRouter() *fiber.App {
	hs.AllowAllCors()

	rootGroup := hs.Fiber.Group("")

	// Swagger
	hs.swaggerRouter.Register(rootGroup)

	// 注册路由
	hs.apiRouter.V1(rootGroup)

	// 404 Route
	hs.Fiber.Use(func(ctx *fiber.Ctx) error {
		return response.Ctx(ctx).Status(fiber.StatusNotFound).Error(errs.RouteNotFound).Send()
	})

	return hs.Fiber
}

func (hs *HttpServer) MetricRouter() *fiber.App {
	app := fiber.New()

	app.Use(recover.New())

	metricGroup := app.Group("")

	prometheus := fiberprometheus.New(hs.config.App.Name)

	prometheus.RegisterAt(app, "/metrics")

	metricGroup.Use(prometheus.Middleware)

	metricGroup.Get("/healthz", func(ctx *fiber.Ctx) error {
		return ctx.Status(http.StatusOK).SendString("OK")
	})

	return app
}