package auth

import (
	"task-track-backend/model"
	"task-track-backend/pkg/auth"

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

func (h *AuthHandler) Refresh(c *gin.Context) {
	userID, exists := c.Get("user_id")
	if !exists {
		c.JSON(401, gin.H{"code": 401, "message": "Unauthorized"})
		return
	}

	var user model.User
	if err := h.db.First(&user, userID).Error; err != nil {
		c.JSON(404, gin.H{"code": 404, "message": "User not found"})
		return
	}

	var organizationID uint = 0
	var userOrg model.UserOrganization
	if err := h.db.Where("user_id = ?", user.ID).First(&userOrg).Error; err == nil {
		organizationID = userOrg.OrganizationID
	}

	token, err := auth.GenerateToken(user.ID, user.Username, organizationID)
	if err != nil {
		c.JSON(500, gin.H{"code": 500, "message": "Failed to generate token", "error": err.Error()})
		return
	}

	c.JSON(200, gin.H{"code": 200, "message": "Token refreshed successfully", "data": gin.H{"token": token}})
}