package router

import (
	"encoding/json"
	"errors"
	"geo/pkg"
	"geo/tmpl/errcode"
	"reflect"
	"strings"

	"github.com/go-playground/validator/v10"
	"github.com/gofiber/fiber/v2"
)

type Module interface {
	Register(r fiber.Router)
}
type RouterServer struct {
	modules []Module
}

func NewRouterServer(
	app *AppModule,
) *RouterServer {
	return &RouterServer{
		modules: []Module{app},
	}
}

// SetupRoutes 设置路由
func SetupRoutes(app *fiber.App, routerServer *RouterServer) {
	app.Use(func(c *fiber.Ctx) error {
		// 设置 CORS 头
		c.Set("Access-Control-Allow-Origin", "*")
		c.Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
		c.Set("Access-Control-Allow-Headers", "Content-Type, Authorization")

		// 如果是预检请求（OPTIONS），直接返回 204
		if c.Method() == "OPTIONS" {
			return c.SendStatus(fiber.StatusNoContent) // 204
		}

		// 继续处理后续中间件或路由
		return c.Next()
	})

	registerResponse(app)

	// 注册所有模块
	for _, module := range routerServer.modules {
		module.Register(app)
	}
}

func registerResponse(router fiber.Router) {
	// 自定义返回
	router.Use(func(c *fiber.Ctx) error {
		err := c.Next()
		return registerCommon(c, err)
	})

}

func registerCommon(c *fiber.Ctx, err error) error {
	// 调用下一个中间件或路由处理函数

	// 如果有错误发生
	if err != nil {
		var businessErr *errcode.BusinessErr
		var fiberError *fiber.Error
		switch {
		case errors.As(err, &businessErr):
		case errors.As(err, &fiberError):
			errors.As(err, &fiberError)
			businessErr = errcode.NewBusinessErr(fiberError.Code, fiberError.Message)
		default:
			businessErr = errcode.SystemError
		}
		// 返回自定义错误响应
		return c.JSON(fiber.Map{
			"message": businessErr.Error(),
			"code":    businessErr.Code(),
			"data":    nil,
		})
	}
	// 如果没有错误发生，继续处理请求
	var data interface{}
	json.Unmarshal(c.Response().Body(), &data)
	return c.JSON(fiber.Map{
		"data":    data,
		"message": errcode.Success.Error(),
		"code":    errcode.Success.Code(),
	})
}

func vali[T any](handler func(*fiber.Ctx, *T) error, _ *T) fiber.Handler {
	return func(c *fiber.Ctx) error {
		var data T

		// 解析请求
		if err := c.BodyParser(&data); err != nil {
			return errcode.ParamErr(err.Error())
		}

		// 创建验证器
		validate := validator.New()

		// 注册中文标签
		validate.RegisterTagNameFunc(func(fld reflect.StructField) string {
			name := fld.Tag.Get("zh")
			if name == "" {
				name = fld.Tag.Get("json")
			}
			return name
		})

		// 验证
		if err := validate.Struct(&data); err != nil {
			er := make([]string, len(err.(validator.ValidationErrors)))
			for k, e := range err.(validator.ValidationErrors) {
				er[k] = pkg.GetErr(e.Tag(), e.Field(), e.Param())
			}
			return errcode.ParamErr(strings.Join(er, ","))
		}

		return handler(c, &data)
	}
}