package http

import (
	"atom/container"
	"atom/providers/config"
	"atom/providers/log"
	"fmt"
	"time"

	"github.com/gin-gonic/gin"
	"go.uber.org/zap"
)

func init() {
	if err := container.Container.Provide(NewService); err != nil {
		log.Fatal(err)
	}
}

type Service struct {
	Engine *gin.Engine
	conf   *config.Config
}

func (e *Service) Serve() error {
	if e.conf.Http.Https {
		return e.Engine.RunTLS(e.conf.Http.PortString(), e.conf.Http.HttpsCert, e.conf.Http.HttpKey)
	}
	return e.Engine.Run(e.conf.Http.PortString())
}

func NewService(cfg *config.Config, logger *log.Logger) *Service {
	gin.DefaultWriter = logger.LevelWriter(zap.InfoLevel)
	gin.DefaultErrorWriter = logger.LevelWriter(zap.ErrorLevel)

	if cfg.App.IsDebug() {
		gin.SetMode(gin.DebugMode)
	} else {
		gin.SetMode(gin.ReleaseMode)
	}

	engine := gin.New()

	engine.Use(gin.LoggerWithFormatter(func(param gin.LogFormatterParams) string {
		return fmt.Sprintf(`%s - [%s] "%s %s %s %d %s '%q' %s"\n`,
			param.ClientIP,
			param.TimeStamp.Format(time.RFC1123),
			param.Method,
			param.Path,
			param.Request.Proto,
			param.StatusCode,
			param.Latency,
			param.Request.UserAgent(),
			param.ErrorMessage,
		)
	}))

	engine.Use(gin.Recovery())

	return &Service{Engine: engine, conf: cfg}
}