package main

import (
	"auditReport/config"
	"auditReport/middleware"
	"github.com/sirupsen/logrus"

	"auditReport/routes"
	"auditReport/utils"
	"fmt"
	"github.com/gin-gonic/gin"
	"io"
	"os"
)

func main() {

	{
		logrus.SetFormatter(&logrus.TextFormatter{ // 输出格式 logfmt 风格
			DisableColors:   true,
			TimestampFormat: "2006-01-02 15:04:05",
		})

		logrus.SetReportCaller(true)
	}
	{
		gin.DisableConsoleColor()
		gin.SetMode(gin.ReleaseMode)
		f, _ := os.Create("auditReport.log")
		gin.DefaultWriter = io.MultiWriter(f)
		gin.DefaultWriter = io.MultiWriter(f, os.Stdout)
	}
	cfg := config.LoadConfig()
	// 初始化数据库
	db := utils.NewDatabase(cfg.Database)
	// 初始化 Gin
	r := gin.Default()

	r.MaxMultipartMemory = 100 << 20
	_ = r.SetTrustedProxies([]string{"127.0.0.1"})
	r.Use(middleware.CorsMiddleware())
	// 注册路由
	routes.RegisterRoutes(r, db, cfg)
	err := r.Run(fmt.Sprint(cfg.Server.Host, ":", cfg.Server.Port))
	if err != nil {
		logrus.Error(err)
		return
	}
}