package network

import (
	"github.com/gin-gonic/gin"
)

// 通用路由处理函数
type RouterHandler[P any, R any] func(*gin.Context, P) (R, error)

// 通用路由处理的选项
type RouterOptions struct {
	FileJsonKey string // 有值时表示是文件路由，记录Json字段存放的PostForm键名称
	NoRequest   bool   // 无需解析请求，默认值false
	NoResponse  bool   // 无需返回应答，默认值false
}

// 路由封装函数
func Router[P any, R any](
	handler RouterHandler[P, R],
	opt *RouterOptions,
) gin.HandlerFunc {
	return func(c *gin.Context) {
		var params P
		if opt == nil {
			opt = &RouterOptions{
				FileJsonKey: "",
				NoRequest:   false,
				NoResponse:  false,
			}
		}
		if !opt.NoRequest {
			if len(opt.FileJsonKey) > 0 { // 文件，FORM表单
				fileJsonStr, ok := c.GetPostForm(opt.FileJsonKey)
				if !ok {
					FailureGeneric(c, "文件请求参数%s不存在", opt.FileJsonKey)
					return
				}
				if e := jsonUnmarshal([]byte(fileJsonStr), &params); e != nil {
					FailureGeneric(c, "文件请求参数解析失败 %s", e.Error())
					return
				}
			} else { // 普通API，JSON请求
				if e := c.ShouldBindJSON(&params); e != nil {
					FailureGeneric(c, "请求参数解析失败 %s", e.Error())
					return
				}
			}
		}
		rsp, err := handler(c, params)
		if !opt.NoResponse {
			if err != nil {
				if e, ok := err.(interface{ Code() int }); ok {
					Failure(c, e.Code(), err.Error())
				} else {
					FailureGeneric(c, err.Error())
				}
			} else {
				Success[R](c, rsp)
			}
		}
	}
}