package errorx

import (
	"fmt"
	"net/http"
	"runtime"
	"strings"

	"github.com/gofiber/fiber/v3"
	"github.com/gofiber/fiber/v3/binder"
	"github.com/gofiber/utils/v2"
	log "github.com/sirupsen/logrus"
)

func Middleware(c fiber.Ctx) error {
	err := c.Next()
	if err != nil {
		return Wrap(err).Response(c)
	}
	return err
}

type Response struct {
	isFormat bool
	err      error
	params   []any
	sql      string
	file     string

	StatusCode int    `json:"-"              xml:"-"`
	Code       int    `json:"code"           xml:"code"`
	Message    string `json:"message"        xml:"message"`
	Data       any    `json:"data,omitempty" xml:"data"`
}

func New(code, statusCode int, message string) *Response {
	return &Response{
		isFormat:   true,
		StatusCode: statusCode,
		Code:       code,
		Message:    message,
	}
}

func (r *Response) WithMsg(msg string) *Response {
	r.Message = msg
	return r
}

func (r *Response) Sql(sql string) *Response {
	r.sql = sql
	return r
}

func (r *Response) from(err *Response) *Response {
	r.Code = err.Code
	r.Message = err.Message
	r.StatusCode = err.StatusCode
	return r
}

func (r *Response) Params(params ...any) *Response {
	r.params = params
	if _, file, line, ok := runtime.Caller(1); ok {
		r.file = fmt.Sprintf("%s:%d", file, line)
	}
	return r
}

func Wrap(err error) *Response {
	if e, ok := err.(*Response); ok {
		return e
	}
	return &Response{err: err}
}

func (r *Response) Wrap(err error) *Response {
	r.err = err
	return r
}

func (r *Response) format() {
	r.isFormat = true
	if e, ok := r.err.(*fiber.Error); ok {
		r.Code = e.Code
		r.Message = e.Message
		r.StatusCode = e.Code
		return
	}

	if r.err != nil {
		msg := r.err.Error()
		if strings.Contains(msg, "duplicate key value") || strings.Contains(msg, "unique constraint") {
			r.from(RecordDuplicated)
			return
		}

		r.Code = http.StatusInternalServerError
		r.StatusCode = http.StatusInternalServerError
		r.Message = msg
	}
	return
}

func (r *Response) Error() string {
	if !r.isFormat {
		r.format()
	}

	return fmt.Sprintf("[%d] %s", r.Code, r.Message)
}

func (r *Response) Response(ctx fiber.Ctx) error {
	if !r.isFormat {
		r.format()
	}

	contentType := utils.ToLower(utils.UnsafeString(ctx.Request().Header.ContentType()))
	contentType = binder.FilterFlags(utils.ParseVendorSpecificContentType(contentType))

	log.
		WithError(r.err).
		WithField("file", r.file).
		WithField("sql", r.sql).
		WithField("params", r.params).
		Errorf("response error: %+v", r)

	// Parse body accordingly
	switch contentType {
	case fiber.MIMETextXML, fiber.MIMEApplicationXML:
		return ctx.Status(r.StatusCode).XML(r)
	case fiber.MIMETextHTML, fiber.MIMETextPlain:
		return ctx.Status(r.StatusCode).SendString(r.Message)
	default:
		return ctx.Status(r.StatusCode).JSON(r)
	}
}

var (
	RecordDuplicated = New(1001, http.StatusBadRequest, "记录重复")
	RecordNotExists  = New(http.StatusNotFound, http.StatusNotFound, "记录不存在")
	BadRequest       = New(http.StatusBadRequest, http.StatusBadRequest, "请求错误")
	Unauthorized     = New(http.StatusUnauthorized, http.StatusUnauthorized, "未授权")
	InternalErr      = New(http.StatusInternalServerError, http.StatusInternalServerError, "内部错误")
)