//
// @Link  https://github.com/bufanyun/hotgo
// @Copyright  Copyright (c) 2022 HotGo CLI
// @Author  Ms <133814250@qq.com>
// @License  https://github.com/bufanyun/hotgo/blob/master/LICENSE
//
package middleware

import (
	"github.com/bufanyun/hotgo/app/com"
	"github.com/bufanyun/hotgo/app/consts"
	"github.com/bufanyun/hotgo/app/utils"
	"github.com/gogf/gf/v2/errors/gcode"
	"github.com/gogf/gf/v2/errors/gerror"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/net/ghttp"
)

//
//  @Title  全局响应中间件
//  @Description
//  @Author  Ms <133814250@qq.com>
//  @Param   r
//
func (s *sMiddleware) HandlerResponse(r *ghttp.Request) {
	r.Middleware.Next()

	var (
		ctx         = r.Context()
		comResponse = com.Context.Get(ctx).ComResponse
		code        = gcode.CodeOK.Code()
		message     = "操作成功"
		data        interface{}
		err         error
	)

	if err := r.GetError(); err != nil {
		g.Log().Print(ctx, err)
		// 记录到自定义错误日志文件
		//g.Log("exception").Error(err)
		////返回固定的友好信息
		//r.Response.ClearBuffer()
		//r.Response.Writeln("服务器居然开小差了，请稍后再试吧！")
	}

	// TODO  已存在响应内容，且是comResponse返回的时，中断运行
	if r.Response.BufferLength() > 0 && comResponse != nil {
		return
	}

	if err = r.GetError(); err != nil {
		// TODO  记录到自定义错误日志文件
		g.Log("exception").Print(r.Context(), "exception:", err)

		code = consts.CodeInternalError
		message = "服务器居然开小差了，请稍后再试吧！"

		// TODO  是否输出错误到页面
		if debug, _ := g.Cfg().Get(ctx, "hotgo.debug", true); debug.Bool() {
			data = utils.Charset.GetStack(err)
		}

	} else if data, err = r.GetHandlerResponse(); err != nil {
		errCode := gerror.Code(err)
		if errCode == gcode.CodeNil {
			errCode = gcode.CodeInternalError
		}
		code = errCode.Code()
		message = err.Error()
	}

	// TODO  返回固定的友好信息
	com.Response.RJson(r, code, message, data)
}