// Package third_party 关于所有第三方框架的二次封装
package third_party

import (
	"git.icechen.cn/monorepo/backend/pkg/net"
	"github.com/gofiber/fiber/v2"
)

// FiberErrHandler fiber 错误处理的中间件
func FiberErrHandler(ctx *fiber.Ctx, err error) error {
	code := fiber.StatusInternalServerError
	msg := ""

	if e, ok := err.(*fiber.Error); ok {
		code = e.Code
		msg = e.Message
	}
	if err != nil {
		msg = err.Error()
	}

	return ctx.Status(code).JSON(net.Response{
		Code:    code,
		Message: msg,
		Data:    nil,
	})
}

// FibberRespHandler fiber resp的统一处理
func FibberRespHandler(c *fiber.Ctx) error {
	if respContentType := string(c.Response().Header.ContentType()); respContentType == fiber.MIMEApplicationJSON {
		resp := net.Response{
			Code:    0,
			Message: "",
			Data:    c.Response().Body(),
		}
		return c.JSON(resp)
	}

	return c.Next()
}

//5 funcs for err 2 fiberErr

func NewFError(msg string) error {
	return fiber.NewError(fiber.StatusInternalServerError, msg)
}

func NewFErrorWithCode(code int, msg string) error {
	return fiber.NewError(code, msg)
}

func WarpFError(err error) error {
	if err == nil {
		return nil
	}
	return fiber.NewError(fiber.StatusInternalServerError, err.Error())
}

func WarpFErrorWithCode(code int, err error) error {
	if err == nil {
		return nil
	}
	return fiber.NewError(code, err.Error())
}