package gin_plugin

import (
	"fmt"
	"gitee.com/kristas/booting-go/framework/common/util/lang"
	"gitee.com/kristas/booting-go/framework/core/bean"
	"gitee.com/kristas/booting-go/framework/core/ioc/bean_repository"
	"gitee.com/kristas/booting-go/framework/core/log"
	"gitee.com/kristas/booting-go/framework/web"
	"github.com/gin-gonic/gin"
	"net/http"
	"regexp"
)

type Starter struct {
	bean.Component
	app *gin.Engine
	Log log.Logger
}

func NewStarter() *Starter {
	return &Starter{}
}

func (s *Starter) Beans() (beans bean.Components) {
	s.app = gin.Default()
	bean_repository.RegisterSingletonBeanWithName("app", s.app)
	return
}

func (s *Starter) RegisterEndpoint(endpoint web.Endpoint) {
	info := endpoint.GetApiInfo()
	s.app.Handle(info.GetHttpMethod(), pathVariableParser(info.GetURL()), func(ctx *gin.Context) {
		endpoint.Handler(newMapperAdapter(ctx), func(result interface{}) error {
			ctx.JSON(http.StatusOK, result)
			return nil
		})
	})
}

func (s *Starter) FsHandler(prefix, dir string) {
	s.app.Static(prefix, dir)
}

func (s *Starter) ListenAndServe(port int) {
	s.Log.Fatal(s.app.Run(fmt.Sprintf(":%d", port)).Error())
}

func pathVariableParser(url string) string {
	compile := regexp.MustCompile("{\\w+}")
	vars := compile.FindAllString(url, -1)
	t := lang.NewString(url)
	for _, v := range vars {
		pv := lang.NewString(v).RemovePrefix("{").RemoveSuffix("}").String()
		t = t.Replace(v, ":"+pv, 1)
	}
	return t.String()
}