"""
Vykreslí Kakuro puzzle do PDF z dat v MySQL tabulce puzzles.
"""

import json
import os
import sys
from pathlib import Path

sys.stdout.reconfigure(encoding="utf-8")
sys.path.insert(0, str(Path(__file__).parent.parent.parent / "Knihovny"))

from reportlab.lib import colors
from reportlab.lib.pagesizes import A4
from reportlab.lib.units import cm
from reportlab.pdfbase import pdfmetrics
from reportlab.pdfbase.ttfonts import TTFont
from reportlab.pdfgen.canvas import Canvas

from mysql_db import connect_mysql

_fonts_dir = os.path.join(os.environ.get("WINDIR", r"C:\Windows"), "Fonts")
pdfmetrics.registerFont(TTFont("Arial", os.path.join(_fonts_dir, "arial.ttf")))
pdfmetrics.registerFont(TTFont("ArialBold", os.path.join(_fonts_dir, "arialbd.ttf")))

OUTPUT = Path(__file__).parent / "test_kakuro.pdf"


def parse_grid(data_str: str) -> list[str]:
    rows = ["0" + r for r in data_str.split(",")]
    rows.insert(0, "0" * len(rows[0]))
    return rows


def sum_hor(grid, x, y):
    s = 0
    w = len(grid[0])
    for i in range(x + 1, w):
        if grid[y][i] == "0":
            break
        s += int(grid[y][i])
    return s if s > 0 else 0


def sum_vert(grid, x, y):
    s = 0
    h = len(grid)
    for i in range(y + 1, h):
        if grid[i][x] == "0":
            break
        s += int(grid[i][x])
    return s if s > 0 else 0


def draw_kakuro(c: Canvas, x0: float, y0: float, cell: float,
                data_str: str, title: str = "", show_solution: bool = False):
    grid = parse_grid(data_str)
    h = len(grid)
    w = len(grid[0])

    clue_font = max(cell * 0.3, 5)
    num_font = max(cell * 0.5, 6)

    if title:
        c.setFont("ArialBold", 12)
        c.drawString(x0, y0 + 5, title)

    for gy in range(h):
        for gx in range(w):
            cx = x0 + gx * cell
            cy = y0 - (gy + 1) * cell
            ch = grid[gy][gx]

            if ch == "0":
                # Černá buňka
                c.setFillColor(colors.Color(0.15, 0.15, 0.15))
                c.rect(cx, cy, cell, cell, fill=1, stroke=0)

                sv = sum_vert(grid, gx, gy)
                sh = sum_hor(grid, gx, gy)

                if sv or sh:
                    # Diagonála
                    c.setStrokeColor(colors.white)
                    c.setLineWidth(0.5)
                    c.line(cx, cy + cell, cx + cell, cy)
                    c.setStrokeColor(colors.black)

                    c.setFillColor(colors.white)
                    c.setFont("ArialBold", clue_font)

                    if sh:
                        # Součet doprava — horní pravý trojúhelník
                        c.drawString(cx + cell * 0.52, cy + cell * 0.55, str(sh))

                    if sv:
                        # Součet dolů — dolní levý trojúhelník
                        c.drawString(cx + cell * 0.08, cy + cell * 0.08, str(sv))

            else:
                # Bílá buňka
                if show_solution:
                    c.setFillColor(colors.black)
                    c.setFont("ArialBold", num_font)
                    c.drawCentredString(cx + cell / 2, cy + cell * 0.3, ch)

    # Mřížka
    c.setStrokeColor(colors.black)
    for i in range(h + 1):
        c.setLineWidth(0.8)
        c.line(x0, y0 - i * cell, x0 + w * cell, y0 - i * cell)
    for i in range(w + 1):
        c.setLineWidth(0.8)
        c.line(x0 + i * cell, y0, x0 + i * cell, y0 - h * cell)


def main():
    conn = connect_mysql(database="puzzle")
    cur = conn.cursor()
    cur.execute(
        "SELECT difficulty, puzzle, extra FROM puzzles "
        "WHERE game_type='kakuro' AND puzzle_date='2026-05-08' "
        "ORDER BY FIELD(difficulty, 'easy', 'medium', 'hard', 'expert') "
        "LIMIT 1"
    )
    row = cur.fetchone()
    cur.close()
    conn.close()

    if not row:
        print("Žádná data.")
        return

    difficulty, data_str, extra = row
    grid = parse_grid(data_str)
    grid_size = len(grid[0])

    page_w, page_h = A4
    board_cm = 11
    cell = board_cm * cm / grid_size
    board = grid_size * cell

    c = Canvas(str(OUTPUT), pagesize=A4)

    # Zadání (bez čísel)
    x0 = (page_w - board) / 2
    y0 = page_h - 2 * cm
    draw_kakuro(c, x0, y0, cell, data_str,
                f"Kakuro {difficulty.capitalize()} — 2026-05-08", show_solution=False)

    # Řešení (s čísly)
    y0_sol = y0 - board - 3 * cm
    draw_kakuro(c, x0, y0_sol, cell, data_str,
                f"Řešení", show_solution=True)

    c.save()
    print(f"PDF uloženo: {OUTPUT}")


if __name__ == "__main__":
    main()