package iris_container

import (
	"fmt"
	"gitee.com/kristas/booting-go/framework"
	. "gitee.com/kristas/booting-go/framework/bean/bean_factory"
	"gitee.com/kristas/booting-go/framework/common/tool/browser"
	"github.com/kataras/iris/v12"
	"github.com/sirupsen/logrus"
	"strings"
)

func init() {
	framework.Application().Use(new(IrisStarter))
}

type IrisStarter struct {
	Log        *logrus.Logger     `bean:"log"`
	IrisConfig iris.Configuration `value:"iris"`
	Logging    framework.Logging  `value:"logging"`
	IrisApp    *iris.Application  `bean:"iris_app"`
}

func (r *IrisStarter) Bean() string {
	return "web_container"
}

func (r *IrisStarter) AfterPropertiesSet() {
	RegisterIrisApp(r.Logging.Level)
}

func (r *IrisStarter) ListenAndServe(port int) {
	if framework.OpenBrowser {
		browser.Open(fmt.Sprintf("http://localhost:%d", port))
	}
	r.Log.Fatal(
		r.IrisApp.Listen(
			fmt.Sprintf(":%d", port),
			iris.WithConfiguration(r.IrisConfig),
		),
	)
}

func RegisterIrisApp(loggingLevel string) {
	app := iris.New()
	app.Logger().SetLevel(irisLevelAdapter(loggingLevel))
	RegisterWithName("iris_app", app)
}

func irisLevelAdapter(level string) string {
	switch strings.ToLower(level) {
	case "panic":
		return "fatal"
	case "trace":
		return "debug"
	default:
		return level
	}
}