import json

from anytree import LevelOrderIter, LevelOrderGroupIter
from anytree.importer import DictImporter
from sys import argv


def build_rows_depth_first(node, rows, leaf_count=0):

    for child in sorted(node.children, key=lambda x: x.height, reverse=True):
        leaf_count = build_rows_depth_first(child, rows, leaf_count)

    colspan = 0
    if node.siblings:
        max_sibling_height = max([s.height for s in node.siblings])

        if max_sibling_height > node.height:
            colspan = max_sibling_height - node.height + 1

    if node.is_leaf:
        cur_row = leaf_count
        rows[cur_row].append(f'<th>{node.name}</th>')

        if colspan > 1:
            rows[cur_row].append(f'<td colspan="{colspan-1}" class="filler"></td>')

        leaf_count += 1

    else:
        child_leaves = len(node.leaves)
        cur_row = leaf_count - child_leaves
        rows[cur_row].append(f"<td rowspan='{child_leaves}' colspan='{colspan}' title='{node.description}'>{node.name}</td>")

    return leaf_count


if len(argv) < 2:
    print(f"usage: {argv[0]} <json file>")
    exit()

with open(argv[1], 'r') as f:
    recipe = json.loads(f.read())

recipe_name = recipe.get("name")
recipe_desc = recipe.get("description")
importer = DictImporter()
sub_recipes = [importer.import_(data) for data in recipe["sub_recipes"]]

print(f"<h1>{recipe_name}</h1>")
print(f"<p>{recipe_desc}</p>")
print("<h2>Needs</h2>")
print("<ul>")

for sub_recipe in sub_recipes:
    for ingredient in sub_recipe.leaves:
        print(f"  <li>{ingredient.name}</li>")

print("</ul>")

print("<h2>Steps</h2>")
print("<ol>")

for sub_recipe in sub_recipes:
    tasks = [node for node in LevelOrderIter(sub_recipe) if not node.is_leaf]
    for task in reversed(tasks):
        print(f"  <li>{task.description if task.description else task.name}</li>")

print("</ol>")

for sub_recipe in sub_recipes:
    print("<table cellspacing='0'>")

    output_rows = []

    for i in range(len(sub_recipe.leaves)):
        output_rows.append([])

    build_rows_depth_first(sub_recipe, output_rows)

    for row in output_rows:
        print("<tr>")
        for cell in row:
            print(f"  {cell}")
        print("</tr>")

    print("</table>")