from flask import Blueprint
from flask_mako import render_template
from exquisite_branch.db import get_db, dict_factory
from exquisite_branch.tree import make_tree

bp = Blueprint("display", __name__, url_prefix="/display")


# @bp.route('/')
# def display():
#     db = get_db()

#     branches = db.execute(
#         "SELECT content, branch, parent, username FROM branches"
#     ).fetchall()

#     streams = []
#     for branch in branches[::-1]:
#         if branch not in flatten(streams):
#             stream = [branch]
#             parent = branch['parent']
#             while parent != 'NEW':
#                 current = next(
#                     (x for x in branches if x['branch'] == parent), None)
#                 parent = current['parent']
#                 stream.append(current)

#             streams.append(stream[::-1])

#     return render_template('display_mako.html', branches=branches, streams=streams)


# def flatten(t):
#   return [item for sublist in t for item in sublist]


# @bp.route("/")
# def display():
#     db = get_db()

#     branches = db.execute(
#         "SELECT content, branch, parent, username FROM branches"
#     ).fetchall()

#     return render_template("display_mako_bis.html", branches=branches)


@bp.route("/")
def display():
    db = get_db()

    branches = db.execute(
        "SELECT content, branch, parent, username FROM branches"
    ).fetchall()

    root = make_tree(branches)

    return render_template("display_mako_bis.html", root=root)
    # return render_template("display_linked_mako.html", branches=branches)