package main

import (
	"log"
	"net/http"
	"strconv"

	"git.0x0001f346.de/andreas/gin-oauth2-demo/core"
	"git.0x0001f346.de/andreas/gin-oauth2-demo/middleware"
	"github.com/gin-gonic/gin"
)

const urlRoot string = "/"
const urlSecret string = "/secret"
const urlSecretWithNumber string = "/secret/:num"

func main() {
	router := gin.Default()
	router.Use(gin.Recovery())

	router.ForwardedByClientIP = true
	router.SetTrustedProxies([]string{"127.0.0.1"})

	router.NoRoute(func(c *gin.Context) { c.Redirect(http.StatusSeeOther, "/") })
	router.NoMethod(func(c *gin.Context) { c.Redirect(http.StatusSeeOther, "/") })

	router.Use(middleware.Auth())
	groupAuth := router.Group(middleware.URLPrefix)
	middleware.SetupRoutes(groupAuth)
	middleware.ProtectURL(urlSecret)
	middleware.ProtectURL(urlSecretWithNumber)

	router.GET(urlRoot, viewRoot)
	router.GET(urlSecret, viewSecret)
	router.GET(urlSecretWithNumber, viewSecretWithNumber)

	log.Fatal(router.Run(":9000"))
}

func viewRoot(c *gin.Context) {
	c.JSON(http.StatusOK, gin.H{
		"user": c.MustGet("user").(core.User),
	})
}

func viewSecret(c *gin.Context) {
	c.JSON(http.StatusOK, gin.H{
		"secret text": "try to add a number to this URL",
	})
}

func viewSecretWithNumber(c *gin.Context) {
	num, err := strconv.Atoi(c.Param("num"))
	if err != nil {
		c.Redirect(http.StatusSeeOther, urlSecret)
		return
	}

	c.JSON(http.StatusOK, gin.H{
		"secret id": num,
	})
}