"""Tests for card generator."""

import pytest
import os
import tempfile
from unittest.mock import MagicMock, patch

from man_card.card_generator import PDFCardGenerator, PNGCardGenerator
from man_card.templates import TemplateLoader, Template
from man_card.man_parser import CommandInfo, Option


class TestPDFCardGenerator:
    """Tests for PDFCardGenerator class."""

    def setup_method(self):
        """Set up test fixtures."""
        template_loader = TemplateLoader()
        self.template = template_loader.load("default")
        self.generator = PDFCardGenerator(self.template)

    def test_init_with_default_template(self):
        """Test initialization with default template."""
        gen = PDFCardGenerator()
        assert gen.template is not None

    def test_init_with_custom_template(self):
        """Test initialization with custom template."""
        template = Template(name="custom")
        gen = PDFCardGenerator(template)
        assert gen.template.name == "custom"

    def test_page_sizes(self):
        """Test that page sizes are defined."""
        assert "a4" in self.generator.PAGE_SIZES
        assert "letter" in self.generator.PAGE_SIZES

    def test_generate_pdf_command_info(self):
        """Test PDF generation with CommandInfo."""
        command_info = CommandInfo(
            name="ls",
            synopsis="ls [OPTION]... [FILE]...",
            description="List directory contents",
            options=[
                Option(flag="-a", description="show all files"),
                Option(flag="-l", description="long listing format")
            ],
            examples=["ls -la", "ls -lh"]
        )

        with tempfile.TemporaryDirectory() as tmpdir:
            output_path = os.path.join(tmpdir, "test_card.pdf")
            self.generator.generate(command_info, output_path, page_size="a4")
            assert os.path.exists(output_path)
            assert os.path.getsize(output_path) > 0

    def test_generate_pdf_with_minimal_info(self):
        """Test PDF generation with minimal CommandInfo."""
        command_info = CommandInfo(name="test")

        with tempfile.TemporaryDirectory() as tmpdir:
            output_path = os.path.join(tmpdir, "minimal_card.pdf")
            self.generator.generate(command_info, output_path)
            assert os.path.exists(output_path)

    def test_hex_to_rgb_conversion(self):
        """Test hex to RGB color conversion."""
        rgb = self.generator._hex_to_rgb("#FF0000")
        assert rgb == (255, 0, 0)

        rgb = self.generator._hex_to_rgb("#00FF00")
        assert rgb == (0, 255, 0)

        rgb = self.generator._hex_to_rgb("#0000FF")
        assert rgb == (0, 0, 255)

    def test_hex_to_rgb_without_hash(self):
        """Test hex to RGB conversion without hash prefix."""
        rgb = self.generator._hex_to_rgb("FFFFFF")
        assert rgb == (255, 255, 255)


class TestPNGCardGenerator:
    """Tests for PNGCardGenerator class."""

    def setup_method(self):
        """Set up test fixtures."""
        template_loader = TemplateLoader()
        self.template = template_loader.load("default")
        self.generator = PNGCardGenerator(self.template)

    def test_init_with_default_template(self):
        """Test initialization with default template."""
        gen = PNGCardGenerator()
        assert gen.template is not None

    def test_dpi_sizes_defined(self):
        """Test that DPI sizes are defined."""
        assert 72 in self.generator.DPI_SIZES
        assert 150 in self.generator.DPI_SIZES
        assert 300 in self.generator.DPI_SIZES

    def test_generate_png_command_info(self):
        """Test PNG generation with CommandInfo."""
        command_info = CommandInfo(
            name="ls",
            synopsis="ls [OPTION]... [FILE]...",
            description="List directory contents",
            options=[
                Option(flag="-a", description="show all files"),
                Option(flag="-l", description="long listing format")
            ],
            examples=["ls -la", "ls -lh"]
        )

        with tempfile.TemporaryDirectory() as tmpdir:
            output_path = os.path.join(tmpdir, "test_card.png")
            self.generator.generate(command_info, output_path, dpi=72)
            assert os.path.exists(output_path)
            assert os.path.getsize(output_path) > 0

    def test_generate_png_with_minimal_info(self):
        """Test PNG generation with minimal CommandInfo."""
        command_info = CommandInfo(name="test")

        with tempfile.TemporaryDirectory() as tmpdir:
            output_path = os.path.join(tmpdir, "minimal_card.png")
            self.generator.generate(command_info, output_path, dpi=72)
            assert os.path.exists(output_path)

    def test_hex_to_rgb_conversion(self):
        """Test hex to RGB color conversion."""
        rgb = self.generator._hex_to_rgb("#FF0000")
        assert rgb == (255, 0, 0)


class TestCardGenerationIntegration:
    """Integration tests for card generation."""

    def test_pdf_and_png_consistency(self):
        """Test that both generators can handle the same data."""
        command_info = CommandInfo(
            name="git",
            synopsis="git [OPTIONS] COMMAND [ARGUMENTS]",
            description="the stupid content tracker",
            options=[
                Option(flag="-v", description="display version"),
                Option(flag="-h", description="display help")
            ],
            examples=["git status", "git commit -m 'message'"]
        )

        template_loader = TemplateLoader()
        template = template_loader.load("default")

        pdf_gen = PDFCardGenerator(template)
        png_gen = PNGCardGenerator(template)

        with tempfile.TemporaryDirectory() as tmpdir:
            pdf_path = os.path.join(tmpdir, "test.pdf")
            png_path = os.path.join(tmpdir, "test.png")

            pdf_gen.generate(command_info, pdf_path)
            png_gen.generate(command_info, png_path, dpi=72)

            assert os.path.exists(pdf_path)
            assert os.path.exists(png_path)
            assert os.path.getsize(pdf_path) > 0
            assert os.path.getsize(png_path) > 0