package routers

import (
	"duoduo/apis/forward"
	"duoduo/apis/pdd"
	"duoduo/apis/user"

	"github.com/gin-gonic/gin"
	"github.com/unrolled/secure"
)

func InitRouter() {

	r := gin.New()

	r.GET("/", func(c *gin.Context) { c.File("./root.txt") })
	r.GET("/root.txt", func(c *gin.Context) { c.File("./root.txt") })
	r2 := r.Group("/")
	{
		pdd.InitPddDdkRouter(r2)
		pdd.InitSetRouter(r2)
		user.InitUserRouter(r2)
		forward.InitForwardRouter(r2)
	}

	_ = r.Run(":8082")

}

func TlsHandler() gin.HandlerFunc {
	return func(c *gin.Context) {
		secureMiddleware := secure.New(secure.Options{
			SSLRedirect: true,
			SSLHost:     "localhost:8080",
		})
		err := secureMiddleware.Process(c.Writer, c.Request)

		// If there was an error, do not continue.
		if err != nil {
			return
		}

		c.Next()
	}
}