package controller

import (
	"Blog/bootstrap"
	"Blog/internal/controller/adm_controller"
	"Blog/internal/controller/cli_controller"
	"Blog/internal/middleware"
	"Blog/internal/utils"
	"net/http"
	"time"

	"github.com/kataras/iris/v12"
	"github.com/kataras/iris/v12/mvc"
	"github.com/kataras/iris/v12/sessions"
	"github.com/kataras/iris/v12/sessions/sessiondb/redis"
	"github.com/sirupsen/logrus"
)

func Router() {
	profile := bootstrap.Config.Profile
	//
	app := iris.Default()
	//
	app.FallbackView(iris.FallbackView("404.html"))
	//配置资源文件位置
	app.HandleDir("/assets", iris.Dir("./assets"))
	//配置模板文件位置
	tmpl := iris.HTML("./template", ".html")
	//模板占位符
	tmpl.Delims("#{", "}")
	//是否每次都加载模板文件，默认否
	if profile == "dev" {
		tmpl.Reload(true)
	}
	app.RegisterView(tmpl)

	app.OnErrorCode(http.StatusNotFound, func(ctx iris.Context) {
		ctx.View("404.html")
	})
	app.OnErrorCode(http.StatusInternalServerError, func(ctx iris.Context) {
		err := ctx.GetErr()
		logrus.Info(err)
		ctx.View("500.html")
	})

	configSession(app)

	// app.Get("/", func(ctx iris.Context) {
	// 	ctx.View("index.html")
	// })

	mvc.Configure(app.Party("/"), func(m *mvc.Application) {
		m.Party("/").Handle(new(cli_controller.IndexController))
		m.Party("/article").Handle(new(cli_controller.ArticleController))
		m.Party("/diary").Handle(new(cli_controller.DiaryController))
		m.Party("/file").Handle(new(cli_controller.FileController))
		m.Party("/admin/login").Handle(new(adm_controller.LoginController))
		m.Router.Get("/logout", func(ctx iris.Context) {
			session := sessions.Get(ctx)
			user := utils.SessionUtil.GetUser(session)
			if user.Id != "" {
				logrus.Info("[", user.Username, "]退出登录,清除Session")
				session.Man.Destroy(ctx)
				ctx.Redirect("")
				return
			}
			logrus.Error("当前用户没有登录状态,无法退出登录")
		})

	})

	mvc.Configure(app.Party("/admin"), func(m *mvc.Application) {
		m.Router.Use(middleware.Auth)
		m.Party("/").Handle(new(adm_controller.AdminController))
		m.Party("/article").Handle(new(adm_controller.ArticleController))
		m.Party("/diary").Handle(new(adm_controller.DiaryController))
		m.Party("/version").Handle(new(adm_controller.VersionController))
		m.Party("/file").Handle(new(adm_controller.FileController))
		m.Party("/user").Handle(new(adm_controller.UserController))

	})

	app.Listen("localhost:8080")

}

func configSession(app *iris.Application) {
	addr := bootstrap.Config.Iris.Session.Address
	redisDatabase := bootstrap.Config.Iris.Session.Db
	prefix := bootstrap.Config.Iris.Session.Prefix
	expires := bootstrap.Config.Iris.Session.Expires
	db := redis.New(redis.Config{
		Network:   "tcp",
		Addr:      addr,
		Timeout:   time.Duration(10) * time.Second,
		MaxActive: 10,
		Username:  "",
		Database:  redisDatabase,
		Prefix:    prefix,
		Driver:    redis.GoRedis(), // defaults to this driver.
	})

	// using the standard `html/template` package.
	sess := sessions.New(sessions.Config{
		Cookie:       "session_id_cookie",
		Expires:      time.Duration(expires) * time.Hour,
		AllowReclaim: true,
	})

	sess.UseDatabase(db)
	app.Use(sess.Handler(), middleware.SessionViewData)
}