package error

import (
	"gitee.com/yanpin-dev/propeller/app"
	"gitee.com/yanpin-dev/propeller/logger"
	"github.com/gin-gonic/gin"
)

type Order int

const (
	ValidateOrder Order = 10000
	BizOrder            = 20000
	DBOrder             = 30000
	UnknownOrder        = 99999

	ErrCode   = "errCode"
	ErrMsg    = "errMsg"
	ErrFields = "errFields"
)

// Middleware 统一异常处理
type Middleware struct {
	log      logger.LogInfoFormat
	handlers []Handler
}

func NewMiddleware(log logger.LogInfoFormat, handlers []Handler) app.Middleware {
	return &Middleware{
		log:      log,
		handlers: handlers,
	}
}

type Handler interface {
	Handle(c *gin.Context, err error) bool
	Order() Order
}

type ByOrder []Handler

func (a ByOrder) Len() int           { return len(a) }
func (a ByOrder) Less(i, j int) bool { return a[i].Order() < a[j].Order() }
func (a ByOrder) Swap(i, j int)      { a[i], a[j] = a[j], a[i] }

func (m *Middleware) NewHandler() gin.HandlerFunc {
	return func(c *gin.Context) {
		c.Next()
		if len(c.Errors) == 0 {
			return
		}

		err := c.Errors.Last().Err

		for _, h := range m.handlers {
			if h.Handle(c, err) {
				return
			}
		}
	}
}