package handlers

import (
	"encoding/json"
	"fmt"
	"io"
	"log"
	"net/http"
	"os"
	"time"

	"lphub/database"
	"lphub/models"

	"github.com/gin-gonic/gin"
	"github.com/golang-jwt/jwt/v4"
	"github.com/pektezol/steam_go"
)

type LoginResponse struct {
	Token string `json:"token"`
}

// Login
//
//	@Description	Get (redirect) login page for Steam auth.
//	@Tags			login
//	@Accept			json
//	@Produce		json
//	@Success		200	{object}	models.Response{data=LoginResponse}
//	@Router			/login [get]
func Login(c *gin.Context) {
	log.Println(c.Request.Host)
	openID := steam_go.NewOpenID(c.Request, true)
	switch openID.Mode() {
	case "":
		c.Redirect(http.StatusMovedPermanently, openID.AuthUrl())
	case "cancel":
		c.Redirect(http.StatusMovedPermanently, "/")
	default:
		steamID, err := openID.ValidateAndGetID()
		if err != nil {
			c.JSON(http.StatusOK, models.ErrorResponse(err.Error()))
			return
		}
		// Create user if new
		var checkSteamID int64
		database.DB.QueryRow("SELECT steam_id FROM users WHERE steam_id = $1", steamID).Scan(&checkSteamID)
		// User does not exist
		if checkSteamID == 0 {
			user, err := GetPlayerSummaries(steamID, os.Getenv("API_KEY"))
			if err != nil {
				c.JSON(http.StatusOK, models.ErrorResponse(err.Error()))
				return
			}
			// Empty country code check
			if user.LocCountryCode == "" || user.LocCountryCode == "  " {
				user.LocCountryCode = "XX"
			}
			// Insert new user to database
			_, err = database.DB.Exec(`INSERT INTO users (steam_id, user_name, avatar_link, country_code)
			VALUES ($1, $2, $3, $4)`, steamID, user.PersonaName, user.AvatarFull, user.LocCountryCode)
			if err != nil {
				c.JSON(http.StatusOK, models.ErrorResponse(err.Error()))
				return
			}
		}
		moderator := false
		rows, _ := database.DB.Query("SELECT title_name FROM titles t INNER JOIN user_titles ut ON t.id=ut.title_id WHERE ut.user_id = $1", steamID)
		for rows.Next() {
			var title string
			rows.Scan(&title)
			if title == "Moderator" {
				moderator = true
			}
		}
		// Generate JWT token
		token := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.MapClaims{
			"sub": steamID,
			"exp": time.Now().Add(time.Hour * 24 * 30).Unix(),
			"mod": moderator,
		})
		// Sign and get the complete encoded token as a string using the secret
		tokenString, err := token.SignedString([]byte(os.Getenv("SECRET_KEY")))
		if err != nil {
			c.JSON(http.StatusOK, models.ErrorResponse("Failed to generate token."))
			return
		}
		c.SetCookie("token", tokenString, 3600*24*30, "/", "", true, true)
		// CreateLog(steamID, LogTypeUser, LogDescriptionUserLoginSuccess)
		c.Redirect(http.StatusTemporaryRedirect, "/")
		// c.JSON(http.StatusOK, models.Response{
		// 	Success: true,
		// 	Message: "Successfully generated token.",
		// 	Data: LoginResponse{
		// 		Token: tokenString,
		// 	},
		// })
		return
	}
}

// GET Token
//
//	@Description	Gets the token cookie value from the user.
//	@Tags			auth
//	@Produce		json
//
//	@Success		200	{object}	models.Response{data=LoginResponse}
//	@Failure		404	{object}	models.Response
//	@Router			/token [get]
func GetCookie(c *gin.Context) {
	cookie, err := c.Cookie("token")
	if err != nil {
		c.JSON(http.StatusOK, models.ErrorResponse("No token cookie found."))
		return
	}
	c.JSON(http.StatusOK, models.Response{
		Success: true,
		Message: "Token cookie successfully retrieved.",
		Data: LoginResponse{
			Token: cookie,
		},
	})
}

// DELETE Token
//
//	@Description	Deletes the token cookie from the user.
//	@Tags			auth
//	@Produce		json
//
//	@Success		200	{object}	models.Response{data=LoginResponse}
//	@Failure		404	{object}	models.Response
//	@Router			/token [delete]
func DeleteCookie(c *gin.Context) {
	cookie, err := c.Cookie("token")
	if err != nil {
		c.JSON(http.StatusOK, models.ErrorResponse("No token cookie found."))
		return
	}
	c.SetCookie("token", "", -1, "/", "", true, true)
	c.JSON(http.StatusOK, models.Response{
		Success: true,
		Message: "Token cookie successfully deleted.",
		Data: LoginResponse{
			Token: cookie,
		},
	})
}

func GetPlayerSummaries(steamId, apiKey string) (models.PlayerSummaries, error) {
	url := fmt.Sprintf("http://api.steampowered.com/ISteamUser/GetPlayerSummaries/v2/?key=%s&steamids=%s", apiKey, steamId)
	resp, err := http.Get(url)
	if err != nil {
		return models.PlayerSummaries{}, err
	}
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return models.PlayerSummaries{}, err
	}

	type Result struct {
		Response struct {
			Players []models.PlayerSummaries `json:"players"`
		} `json:"response"`
	}
	var data Result
	if err := json.Unmarshal(body, &data); err != nil {
		return models.PlayerSummaries{}, err
	}
	return data.Response.Players[0], err
}