package basedataapp

import (
	"gitee.com/captials-team/ubdframe/src/apps"
	httpController "gitee.com/captials-team/ubdframe/src/apps/basedataapp/controllers/http"
	"gitee.com/captials-team/ubdframe/src/apps/basedataapp/docs"
	"gitee.com/captials-team/ubdframe/src/common"
	"gitee.com/captials-team/ubdframe/src/domain/configstc"
	"gitee.com/captials-team/ubdframe/src/domain/interfaces"
	"gitee.com/captials-team/ubdframe/src/pkg/gin_http"
	v1log "gitee.com/captials-team/ubdframe/src/pkg/logs"
	"github.com/gin-gonic/gin"
	"go.uber.org/dig"
)

type ApiServer struct {
	*apps.ApiServer

	di   *dig.Container
	conf *configstc.BaseDataAppConfig

	gin_http.SwaggerOption //swagger相关选项配置

}

func (s *ApiServer) Start() error {
	if !s.Module() {
		s.InitRouter()
	}
	return s.ApiServer.Start()
}

func (s *ApiServer) Stop() error {
	return s.ApiServer.Stop()
}

func (s *ApiServer) InitRouter() {
	s.Engine().GET("ping", gin_http.PingHandler)
	s._regRouter(s.Engine())
}

func (s *ApiServer) InitRouterForGin(engine *gin.Engine) {
	s._regRouter(engine)
}

// regRouter 注册路由
func (s *ApiServer) _regRouter(engine *gin.Engine) {
	var g = engine.Group("/")
	if len(s.conf.RoutePrefix) > 0 {
		g = engine.Group(s.conf.RoutePrefix)
	}

	//注册swagger
	s.SwaggerRouter(g)

	s.router(g)

	return
}

func (s *ApiServer) router(g gin.IRouter) {
	g.Use(
		gin_http.PanicHandler,
		gin_http.QPSLimiterHandler(10, 10),
	)

	//验证码
	common.ErrPanic(s.di.Invoke(func(ctr interfaces.ItfBaseDataController) {
		g.POST("/data/regions", ctr.Regions)
	}))
}

func NewApiServer(di *dig.Container, conf *configstc.BaseDataAppConfig, logger v1log.ILog) *ApiServer {
	logger.Info("HELLO %s", conf.Name)

	common.ErrPanic(di.Provide(httpController.NewBaseDataController, dig.As(new(interfaces.ItfBaseDataController))))

	//重置swagger的api host
	docs.SwaggerInfobasedataservice.Host = conf.ApiServer.HostAddr() + conf.RoutePrefix

	s := &ApiServer{
		di:   di,
		conf: conf,
		SwaggerOption: gin_http.SwaggerOption{
			Enable:  conf.DocsEnable,
			Name:    docs.SwaggerInfobasedataservice.InstanceName(),
			Swagger: docs.SwaggerInfobasedataservice,
		},
		ApiServer: apps.NewApiServer(gin.Default(), conf.ApiServer),
	}
	s.ApiServer.WithName("baseData")
	s.ApiServer.WithCors()
	return s
}