import sys
from typing import Iterable
from PIL import Image, ImageDraw, ImageFont

datei_endung = ".png"

img = None
output_img = None
font = None


def NSD_init(x: float, y: float):
    #get input_img
    global img, output_img, font
    #img = Image.open(input_dir + file_name + datei_endung)
    img = Image.new("RGB", (x, y), "white")
    output_img = ImageDraw.Draw(img)
    #font = ImageFont.load_default()
    font = ImageFont.truetype("res/fonts/SpaceGrotesk-Light.ttf", 14)

def get_text_size(text: str):
    if not font:
        raise Exception("Output image was not initialized! Make sure to call NSD_init first")
    return font.getsize(text, direction="ltr")

def draw_debug_tile(x, y, x_sz, y_sz):
    from random import randint

    output_img.rectangle((x, y) + (x + x_sz, y + y_sz), fill=(randint(0, 255)))
    return x, y + y_sz

def draw_generic_instruction(instruction: str, x, y, xsize, ysize) -> Iterable[float]:
    if not output_img:
        raise Exception("Output image was not initialized! Make sure to call NSD_init first")
    
    #size shit
    text_y_size = font.getsize(instruction, direction="ltr")[1]
    ysize = max(text_y_size, ysize) # ensure the bounding box is alway at least big enought to fit the text

    #draw shit
    output_img.rectangle((x,y) + (x + xsize, y + ysize), outline=(0), width=1)

    #text shit
    output_img.multiline_text((x + 5, y + ysize * .5), instruction, font=font, anchor="lm", align="right", fill=(0))

    return x, y + ysize




def draw_if_statement(condition: str, x: int, y: int, xsize: int, ysize: int):
    """Draw an if statement into the NSD"""
    if not output_img:
        raise Exception("Output image was not initialized! Make sure to call NSD_init first")

    text_y_size = font.getsize(condition)[1]

    output_img.line((x,y) + (x + xsize / 2, y + text_y_size), fill=(0))
    output_img.line((x + xsize, y) + (x + xsize / 2, y + text_y_size), fill=(0))
    output_img.rectangle((x, y + text_y_size) + (x + xsize, y + ysize), outline=(0), width=1)
    output_img.rectangle((x, y) + (x + xsize, y + text_y_size), outline=(0), width=1)
    output_img.line((x + xsize / 2, y + text_y_size) + (x + xsize / 2, y + ysize), fill=(0))

    # condition text
    output_img.multiline_text((x + xsize / 2, y + text_y_size), condition, fill=(0), font=font, anchor="md", spacing=4, align='right')

    # true / false
    output_img.text((x + 5, y + text_y_size), "true", font = font, fill = (0), anchor="ld")
    output_img.text((x + xsize - 5, y + text_y_size), "false", font = font, fill = (0), anchor="rd")

    #first x,y,xsize,ysize of first box then of second first true and then false
    return x, y + text_y_size, xsize / 2, ysize - text_y_size, x + xsize / 2, y + text_y_size, xsize / 2, ysize - text_y_size

def draw_while_loop_front(condition: str, x: int, y: int, xsize: int, ysize: int):
    
    if not output_img:
        raise Exception("Output image was not initialized! Make sure to call NSD_init first")

    text_y_sz = font.getsize(condition)[1]

    #the box
    output_img.line((x,y) + (x + xsize, y), fill=(0))
    output_img.line((x,y) + (x, y + ysize), fill=(0))
    output_img.line((x + xsize * .1, y + text_y_sz) + (x + xsize, y + text_y_sz), fill=(0))
    output_img.line((x + xsize, y) + (x + xsize, y + text_y_sz), fill=(0))
    output_img.line((x, y + ysize) + (x + xsize * .1, y + ysize ), fill=(0))
    output_img.line((x + xsize * .1, y + ysize) + (x + xsize * .1, y + text_y_sz), fill=(0))

    #the text
    output_img.text((x + xsize * .1, y + text_y_sz * 0.5), condition, font = font, fill = (0), anchor="lm")

    #the x, y offset then the x,y draw size (the canvas)
    return x + xsize * .1, y + text_y_sz, xsize * .9

def draw_while_loop_back(condition: str, x: int, y: int, xsize: int, ysize: int):
    
    if not output_img:
        raise Exception("Output image was not initialized! Make sure to call NSD_init first")

    #ole #TODO

    #the box
    output_img.line((x,y) + (x + xsize * .1, y), fill=0)
    output_img.line((x + xsize * .1, y) + (x + xsize * .1, y + ysize * .9), fill=0)
    output_img.line((x + xsize * .1, y + ysize * .9) + (x + xsize, y + ysize * .9), fill=0)
    output_img.line((x + xsize, y + ysize * .9) + (x + xsize, y + ysize), fill=0)
    output_img.line((x,y + ysize) + (x + xsize, y + ysize), fill=0)
    output_img.line((x,y) + (x, y + ysize), fill=0)

    #the text
    output_img.text((x + xsize * .1, y + ysize * .95), condition, font = font, fill = (0), anchor="lm")

    #the x, y offset then the x,y draw size (the canvas)
    return x + xsize * .1, y, xsize * .9, ysize * .9

def NSD_save(filename: str):
    """Save the created file"""
    img.save(filename + datei_endung ,"PNG")

#x_offset , y_offset, x_size, y_size = draw_while_loop("lol", 0, 0, 100, 200)
if __name__ == "__main__":
    """Debugging :^)"""
    NSD_init(300, 500)
    #draw_if_statement("wenn das dann mach das", 0, 0, 100, 200)
    #print(x,y,xsize,ysize)
    #draw_generic_instruction(r"""Wolfgang.fuck("Nina")""", x, y, xsize, ysize)
    NSD_save("testink")