package main

import (
	"fmt"
	"log/slog"
	"net"
	"net/http"
	"net/http/httputil"
	"os"
	"runtime/debug"
	"strings"
	"time"

	"joylink.club/bj-rtsts-server/db/dbquery"
	"joylink.club/bj-rtsts-server/dto"
	"joylink.club/bj-rtsts-server/sys_error"

	"github.com/gin-contrib/cors"
	"github.com/gin-gonic/gin"
	sloggin "github.com/samber/slog-gin"
	"joylink.club/bj-rtsts-server/config"
	"joylink.club/bj-rtsts-server/db"
	"joylink.club/bj-rtsts-server/logger"
)

func InitServer() *gin.Engine {
	config.LoadConfig()
	err := logger.InitLogger()
	if err != nil {
		panic(err)
	}

	err = db.InitDb()
	if err != nil {
		panic(err)
	}
	dbquery.SetDefault(db.DB)

	engine := gin.New()

	conf := cors.DefaultConfig()
	conf.AllowHeaders = []string{"*"}
	conf.AllowAllOrigins = true
	engine.Use(sloggin.New(slog.Default()))
	engine.Use(cors.New(conf))
	// gin panic 异常处理，默认处理为
	engine.Use(CustomRecoveryWithSlog(slog.Default(), true, func(c *gin.Context, e interface{}) {
		be, ok := e.(*sys_error.BusinessError)
		if !ok {
			e, ok := e.(error)
			if ok {
				be = sys_error.New("未知错误", e)
			} else {
				be = sys_error.New("未知错误", fmt.Errorf("%v", e))
			}
		}
		c.JSON(http.StatusInternalServerError, &dto.ErrorDto{
			Tip:     be.UserMsg,
			Message: be.Error(),
		})
		c.Writer.WriteHeaderNow()
		c.Abort()
	}))
	return engine
}

func CustomRecoveryWithSlog(logger *slog.Logger, stack bool, recovery gin.RecoveryFunc) gin.HandlerFunc {
	return func(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 {
					if se, ok := ne.Err.(*os.SyscallError); ok {
						if strings.Contains(strings.ToLower(se.Error()), "broken pipe") || strings.Contains(strings.ToLower(se.Error()), "connection reset by peer") {
							brokenPipe = true
						}
					}
				}

				httpRequest, _ := httputil.DumpRequest(c.Request, false)
				if brokenPipe {
					logger.Error(c.Request.URL.Path,
						"error", err,
						"request", string(httpRequest),
					)
					// If the connection is dead, we can't write a status to it.
					c.Error(err.(error)) // nolint: errcheck
					c.Abort()
					return
				}

				if stack {
					logger.Error("请求处理Panic异常",
						"time", time.Now(),
						"error", err,
						"request", string(httpRequest),
						"stack", string(debug.Stack()),
					)
					debug.PrintStack()
				} else {
					logger.Error("请求处理Panic异常",
						"time", time.Now(),
						"error", err,
						"request", string(httpRequest),
					)
				}
				recovery(c, err)
			}
		}()
		c.Next()
	}
}