import (
	"errors"
	"fmt"
	"github.com/gin-gonic/gin"
	uuid "github.com/satori/go.uuid"
	"net"
	"net/http"
	"os"
	"soul_station/src/common/response"
	"soul_station/src/global"
	"soul_station/src/startup/initialize/router"
	"strings"
)

func Router() {
	gin.SetMode(gin.ReleaseMode)
	apiEngine := newEngine()

	// API group
	api := apiEngine.Group("/api")
	//管理端
	router.AdminClient(api)
	//用户端
	router.UserClient(api)

	// 启动、监听端口
	_ = apiEngine.Run(fmt.Sprintf("localhost:%s", global.Config.Server.Port))

}

func newEngine() *gin.Engine {
	engine := gin.Default()
	//去掉默认handlers，去掉gin自带的日志打印
	engine.Handlers = nil
	//拼接TraceId
	engine.Use(traceId)
	//自定义日志打印
	engine.Use(preLog)
	engine.Use(afterLog)
	engine.Use(customRecovery)
	// 开启跨域
	engine.Use(cors)
	//404
	engine.NoRoute(func(c *gin.Context) {
		c.String(http.StatusNotFound, "404 not found")
	})
	return engine
}

// Cors 处理跨域请求
func cors(c *gin.Context) {
	method := c.Request.Method
	origin := c.Request.Header.Get("*.dra-m.com")
	c.Header("Access-Control-Allow-Origin", origin)
	c.Header("Access-Control-Allow-Headers", "Content-Type,AccessToken,X-CSRF-Token, Authorization, Token,X-Token,X-User-Id")
	c.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS,DELETE,PUT")
	c.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers, Content-Type")
	c.Header("Access-Control-Allow-Credentials", "true")

	// 放行所有OPTIONS方法
	if method == "OPTIONS" {
		c.AbortWithStatus(http.StatusNoContent)
	}

	// 处理请求
	c.Next()
}
func preLog(c *gin.Context) {
	value, _ := c.Get("Trace")
	remoteIp := c.GetHeader("X-Real-IP")
	if remoteIp == "" {
		remoteIp = c.Request.RemoteAddr
	}

	global.Log.Infof("[%s] >>>>>>>>>>> [%s] %s From %s", value, c.Request.Method, c.Request.RequestURI, remoteIp)
	c.Next()
}
func afterLog(c *gin.Context) {
	defer func() {
		value, _ := c.Get("Trace")
		remoteIp := c.GetHeader("X-Real-IP")
		if remoteIp == "" {
			remoteIp = c.Request.RemoteAddr
		}
		global.Log.Infof("[%s] <<<<< [%3d] [%s] %s To %s", value, c.Writer.Status(), c.Request.Method, c.Request.RequestURI, remoteIp)
	}()
	c.Next()

}
func customRecovery(c *gin.Context) {
	defer func() {
		if err := recover(); err != nil {
			// Check for a broken connection, as it is not really a
			// condition that warrants a panic stack trace.
			var brokenPipe bool
			if ne, ok := err.(*net.OpError); ok {
				var se *os.SyscallError
				if errors.As(ne, &se) {
					if strings.Contains(strings.ToLower(se.Error()), "broken pipe") || strings.Contains(strings.ToLower(se.Error()), "connection reset by peer") {
						brokenPipe = true
					}
				}
			}
			value, _ := c.Get("Trace")
			global.Log.Warnf("[%s] %s", value, err)

			if brokenPipe {
				// If the connection is dead, we can't write a status to it.
				c.Error(err.(error)) // nolint:
				c.Abort()
			} else {
				switch err.(type) {
				case string:
					response.ErrorWithInfo(err.(string)).Write(c)
					break
				case error:
					response.ErrorWithInfo(err.(error).Error()).Write(c)
					break
				default:
					response.ErrorWithInfo("发生不明错误").Write(c)
					break
				}

			}
		}
	}()
	c.Next()

}

func traceId(c *gin.Context) {
	trace := uuid.NewV4().String()
	c.Set("Trace", trace)
	c.Header("Trace", trace)
	c.Next()
}