package main

import (
	"fmt"
	_ "git.rosy.net.cn/jx-print/controllers"
	"git.rosy.net.cn/jx-print/model"
	"git.rosy.net.cn/jx-print/routers"
	"github.com/gin-contrib/sessions"
	"github.com/gin-contrib/sessions/cookie"
	"github.com/gin-gonic/gin"
	"time"
)

func main()  {
	gin.DisableConsoleColor()

	r := gin.New()
	r.Use(session(model.SessionKey))
	// LoggerWithFormatter middleware will write the logs to gin.DefaultWriter
	// By default gin.DefaultWriter = os.Stdout
	r.Use(gin.LoggerWithFormatter(func(param gin.LogFormatterParams) string {

		// your custom format
		return fmt.Sprintf("%s - [%s] \"%s %s %d \" %s\"\n",
			param.ClientIP,
			param.TimeStamp.Format(time.RFC3339),
			param.Method,
			param.Path,
			param.StatusCode,
			param.ErrorMessage,
		)
	}))
	r.Use(gin.Recovery())
	routers.Init(r)
	r.Run(":8999")
}
// 中间件，处理session
func session(keyPairs string) gin.HandlerFunc {
	store := sessionConfig()
	return sessions.Sessions(keyPairs, store)
}
func sessionConfig() sessions.Store {
	sessionMaxAge := 3600
	sessionSecret := model.SessionKey
	var store sessions.Store
	store = cookie.NewStore([]byte(sessionSecret))
	store.Options(sessions.Options{
		MaxAge: sessionMaxAge, //seconds
		Path:   "/",
	})
	return store
}