package repository

import (
	"context"
	"time"

	"github.com/google/uuid"
	"github.com/jackc/pgx/v5/pgxpool"
	"github.com/ren/econ/backend/internal/models"
)

type ProgresoRepository struct {
	db *pgxpool.Pool
}

func NewProgresoRepository(db *pgxpool.Pool) *ProgresoRepository {
	return &ProgresoRepository{db: db}
}

func (r *ProgresoRepository) GetProgresoByUsuarioID(usuarioID uuid.UUID) ([]models.Progreso, error) {
	query := `
		SELECT id, usuario_id, modulo_numero, ejercicio_id, completado, puntuacion, intentos, ultima_vez
		FROM progreso_usuario WHERE usuario_id = $1
		ORDER BY ultima_vez DESC
	`
	rows, err := r.db.Query(context.Background(), query, usuarioID)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	var progresos []models.Progreso
	for rows.Next() {
		var p models.Progreso
		err := rows.Scan(
			&p.ID, &p.UsuarioID, &p.ModuloNumero, &p.EjercicioID,
			&p.Completado, &p.Puntuacion, &p.Intentos, &p.UltimaVez)
		if err != nil {
			return nil, err
		}
		progresos = append(progresos, p)
	}
	return progresos, nil
}

func (r *ProgresoRepository) SaveProgreso(progreso *models.Progreso) error {
	query := `
		INSERT INTO progreso_usuario (id, usuario_id, modulo_numero, ejercicio_id, completado, puntuacion, intentos, ultima_vez)
		VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
		ON CONFLICT (usuario_id, modulo_numero, ejercicio_id)
		DO UPDATE SET completado = $5, puntuacion = $6, intentos = progreso_usuario.intentos + 1, ultima_vez = $8
	`

	if progreso.ID == uuid.Nil {
		progreso.ID = uuid.New()
	}
	if progreso.UltimaVez.IsZero() {
		progreso.UltimaVez = time.Now()
	}
	if progreso.Intentos == 0 {
		progreso.Intentos = 1
	}

	_, err := r.db.Exec(context.Background(), query,
		progreso.ID, progreso.UsuarioID, progreso.ModuloNumero, progreso.EjercicioID,
		progreso.Completado, progreso.Puntuacion, progreso.Intentos, progreso.UltimaVez)
	return err
}

func (r *ProgresoRepository) GetResumen(usuarioID uuid.UUID) (*models.ResumenProgreso, error) {
	query := `
		SELECT
			COALESCE(SUM(puntuacion), 0) as puntos_totales,
			COUNT(CASE WHEN completado THEN 1 END) as ejercicios_completados,
			COUNT(*) as total_ejercicios
		FROM progreso_usuario WHERE usuario_id = $1
	`
	var resumen models.ResumenProgreso
	err := r.db.QueryRow(context.Background(), query, usuarioID).Scan(
		&resumen.PuntosTotales, &resumen.EjerciciosCompletados, &resumen.TotalEjercicios)
	if err != nil {
		return nil, err
	}

	// Alias para compatibilidad con frontend
	resumen.TotalPuntuacion = resumen.PuntosTotales

	// Calcular nivel basado en puntuación
	resumen.Nivel = calcularNivel(resumen.PuntosTotales)

	// Generar badges basados en progreso
	resumen.Badges = generarBadges(resumen.PuntosTotales, resumen.EjerciciosCompletados)

	return &resumen, nil
}

func calcularNivel(puntuacion int) string {
	if puntuacion >= 2000 {
		return "Maestro"
	}
	if puntuacion >= 1000 {
		return "Experto"
	}
	if puntuacion >= 300 {
		return "Aprendiz"
	}
	return "Novato"
}

func generarBadges(puntuacion, ejerciciosCompletados int) []models.Badge {
	badges := []models.Badge{
		{ID: "primer-ejercicio", Nombre: "Primer Ejercicio", Descripcion: "Completa tu primer ejercicio", Icono: "star", Desbloqueado: ejerciciosCompletados >= 1},
		{ID: "primer-modulo", Nombre: "Primer Módulo", Descripcion: "Completa todas las lecciones de un módulo", Icono: "award", Desbloqueado: ejerciciosCompletados >= 3},
		{ID: "aprendiz", Nombre: "Aprendiz", Descripcion: "Alcanza el nivel Aprendiz", Icono: "book", Desbloqueado: puntuacion >= 300},
		{ID: "experto", Nombre: "Experto", Descripcion: "Alcanza el nivel Experto", Icono: "trophy", Desbloqueado: puntuacion >= 1000},
		{ID: "maestro", Nombre: "Maestro", Descripcion: "Alcanza el nivel Maestro", Icono: "crown", Desbloqueado: puntuacion >= 2000},
		{ID: "puntos-500", Nombre: "500 Puntos", Descripcion: "Acumula 500 puntos", Icono: "target", Desbloqueado: puntuacion >= 500},
		{ID: "puntos-1000", Nombre: "1000 Puntos", Descripcion: "Acumula 1000 puntos", Icono: "zap", Desbloqueado: puntuacion >= 1000},
		{ID: "puntos-2000", Nombre: "2000 Puntos", Descripcion: "Acumula 2000 puntos", Icono: "flame", Desbloqueado: puntuacion >= 2000},
	}
	return badges
}