from pathlib import Path
from PIL import Image, ImageDraw, ImageFont
import random
import json

TEXT_FONT = Path(__file__).parent / "assets" / "FreeSerifBold.otf"

CHARSET = tuple("ABCDEFGHIJKLMNOPQRSTUVWXYZ")
DIGITS = tuple("0123456789")
WORD_LENGTH = 5

INPUT_TYPE = "text"
INPUT_HINT = "enter text decoded from the number"

NAME = "Decoding"
INSTRUCTIONS = "This is the third task. You will see a sequence of tables and the task is to convert the numbers into letters using the decoding dictionary provided. The first row shows the letters and the second row the numbers corresponding to each letter. You need to convert the five number code into a five letter word. You have {} minutes to decode as many words as possible. Each incorrect answer yields a reward of {}. Below is an example of how the task will look like."
EXAMPLE = "images/decoding.png"


def generate_puzzle_fields():
    """Create new puzzle for a player"""

    chars = random.sample(CHARSET, len(DIGITS))
    digits = random.sample(DIGITS, len(DIGITS))

    lookup = dict(zip(digits, chars))

    coded_word = "".join(random.sample(DIGITS, WORD_LENGTH))
    solution = "".join(lookup[digit] for digit in coded_word)

    return dict(
        text=json.dumps(dict(rows=[chars, digits], coded_word=coded_word)),
        solution=solution,
    )


def is_correct(response, puzzle):
    return puzzle.solution.lower() == response.lower()


TEXT_SIZE = 32
TEXT_PADDING = TEXT_SIZE
CELL_DIM = TEXT_SIZE + TEXT_PADDING * 2
MID = CELL_DIM * 0.5


def render_image(puzzle):
    data = json.loads(puzzle.text)

    font = ImageFont.truetype(str(TEXT_FONT), TEXT_SIZE)
    img_w = CELL_DIM * len(DIGITS)
    # 4 because 2 rows + blank space + row for coded word
    img_h = CELL_DIM * 4
    image = Image.new("RGB", (img_w, img_h))
    draw = ImageDraw.Draw(image)

    for rownum, row in enumerate(data["rows"]):
        for colnum, char in enumerate(row):
            x = colnum * CELL_DIM
            y = rownum * CELL_DIM
            draw.rectangle([x, y, x + CELL_DIM, y + CELL_DIM])
            draw.text((x + MID, y + MID), char, font=font, anchor="mm")

    coded_word = data["coded_word"]
    w, h = draw.textsize(coded_word)
    draw.text(
        ((img_w - w) / 2, image.height - MID),
        data["coded_word"],
        font=font,
        anchor="mm",
    )

    return image