package main

import (
	"BlogFile/bootstrap"
	"BlogFile/cloud"
	"BlogFile/internal/controller"
	"BlogFile/internal/controller/admin"
	"BlogFile/internal/middleware"
	"BlogFile/third_party/database"
	"log"
	"net/http"
	"os"
	"time"

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

func main() {
	profile := os.Getenv("profile")
	log.SetFlags(log.Ldate | log.Ltime | log.Lmicroseconds | log.Ltime | log.Llongfile)
	app := iris.Default()
	app.FallbackView(iris.FallbackView("404.html"))
	// Set custom delimeters.
	if profile != "dev" {
		logFile, err := os.OpenFile("./iris.log", os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0644)
		if err != nil {
			panic("日志初始化失败")
		}
		app.Logger().SetOutput(logFile)
		log.SetOutput(logFile)
		profile = "prod"
	}
	log.Println("Go Iris 启动")
	log.Println("当前环境是:", profile)

	app.OnErrorCode(http.StatusNotFound, func(ctx iris.Context) {
		// ctx.StatusCode(http.StatusNotFound)
		// s := ctx.Path()
		// log.Println(s)
		ctx.View("404.html")
	})
	app.OnErrorCode(http.StatusInternalServerError, func(ctx iris.Context) {
		err := ctx.GetErr()
		log.Println(err)
		ctx.View("500.html")
	})

	addr := bootstrap.Config.Session.Addr
	redisDatabase := bootstrap.Config.Session.Database
	prefix := bootstrap.Config.Session.Prefix
	expires := bootstrap.Config.Session.Expires
	db := redisSession.New(redisSession.Config{
		Network:   "tcp",
		Addr:      addr,
		Timeout:   time.Duration(10) * time.Second,
		MaxActive: 10,
		Username:  "",
		Database:  redisDatabase,
		Prefix:    prefix,
		Driver:    redisSession.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)

	app.Get("/", func(ctx iris.Context) {
		ctx.View("index.html")
	})
	fileApi := app.Party("/file")

	adminFileApi := app.Party("/admin/file")
	adminFileApi.Use(middleware.Auth)

	mvc.New(fileApi).Handle(new(controller.FileController))
	mvc.New(adminFileApi).Handle(new(admin.FileController))

	go cloud.DistributedService.ServiceUp()

	host := bootstrap.Config.Cloud.Host
	app.Listen(host)

	defer func() {
		log.Println("程序结束，关闭资源")
		cloud.DistributedService.ServiceDown()
		db.Close()
		database.RedisTemplate.Close()
	}()
}