#include <AK/Utf8View.h>
#include <LibHTML/CSS/StyleSheet.h>
#include <LibHTML/DOM/Document.h>
#include <LibHTML/DOM/Element.h>
#include <LibHTML/DOM/Text.h>
#include <LibHTML/Dump.h>
#include <LibHTML/Layout/LayoutBlock.h>
#include <LibHTML/Layout/LayoutNode.h>
#include <LibHTML/Layout/LayoutText.h>
#include <stdio.h>

void dump_tree(const Node& node)
{
    static int indent = 0;
    for (int i = 0; i < indent; ++i)
        dbgprintf("  ");
    if (node.is_document()) {
        dbgprintf("*Document*\n");
    } else if (node.is_element()) {
        dbgprintf("<%s", static_cast<const Element&>(node).tag_name().characters());
        static_cast<const Element&>(node).for_each_attribute([](auto& name, auto& value) {
            dbgprintf(" %s=%s", name.characters(), value.characters());
        });
        dbgprintf(">\n");
    } else if (node.is_text()) {
        dbgprintf("\"%s\"\n", static_cast<const Text&>(node).data().characters());
    }
    ++indent;
    if (node.is_parent_node()) {
        static_cast<const ParentNode&>(node).for_each_child([](auto& child) {
            dump_tree(child);
        });
    }
    --indent;
}

void dump_tree(const LayoutNode& layout_node)
{
    static int indent = 0;
    for (int i = 0; i < indent; ++i)
        dbgprintf("    ");

    String tag_name;
    if (layout_node.is_anonymous())
        tag_name = "(anonymous)";
    else if (layout_node.node()->is_text())
        tag_name = "#text";
    else if (layout_node.node()->is_document())
        tag_name = "#document";
    else if (layout_node.node()->is_element())
        tag_name = static_cast<const Element&>(*layout_node.node()).tag_name();
    else
        tag_name = "???";

    dbgprintf("%s {%s} at (%d,%d) size %dx%d",
        layout_node.class_name(),
        tag_name.characters(),
        layout_node.rect().x(),
        layout_node.rect().y(),
        layout_node.rect().width(),
        layout_node.rect().height());

    // Dump the horizontal box properties
    dbgprintf(" [%d+%d+%d %d %d+%d+%d]",
        layout_node.box_model().margin().left.to_px(),
        layout_node.box_model().border().left.to_px(),
        layout_node.box_model().padding().left.to_px(),
        layout_node.rect().width(),
        layout_node.box_model().padding().right.to_px(),
        layout_node.box_model().border().right.to_px(),
        layout_node.box_model().margin().right.to_px());

    // And the vertical box properties
    dbgprintf(" [%d+%d+%d %d %d+%d+%d]",
        layout_node.box_model().margin().top.to_px(),
        layout_node.box_model().border().top.to_px(),
        layout_node.box_model().padding().top.to_px(),
        layout_node.rect().height(),
        layout_node.box_model().padding().bottom.to_px(),
        layout_node.box_model().border().bottom.to_px(),
        layout_node.box_model().margin().bottom.to_px());

    dbgprintf("\n");

    if (layout_node.is_block() && static_cast<const LayoutBlock&>(layout_node).children_are_inline()) {
        auto& block = static_cast<const LayoutBlock&>(layout_node);
        for (int i = 0; i < indent; ++i)
            dbgprintf("    ");
        dbgprintf("  Line boxes (%d):\n", block.line_boxes().size());
        for (int line_box_index = 0; line_box_index < block.line_boxes().size(); ++line_box_index) {
            auto& line_box = block.line_boxes()[line_box_index];
            for (int i = 0; i < indent; ++i)
                dbgprintf("    ");
            dbgprintf("    [%d] width: %d\n", line_box_index, line_box.width());
            for (int fragment_index = 0; fragment_index < line_box.fragments().size(); ++fragment_index) {
                auto& fragment = line_box.fragments()[fragment_index];
                for (int i = 0; i < indent; ++i)
                    dbgprintf("    ");
                dbgprintf("      [%d] layout_node: %s{%p}, start: %d, length: %d, rect: %s\n",
                    fragment_index,
                    fragment.layout_node().class_name(),
                    &fragment.layout_node(),
                    fragment.start(),
                    fragment.length(),
                    fragment.rect().to_string().characters());
                if (fragment.layout_node().is_text()) {
                    for (int i = 0; i < indent; ++i)
                        dbgprintf("    ");
                    auto& layout_text = static_cast<const LayoutText&>(fragment.layout_node());
                    dbgprintf("        text: \"%s\"\n",
                        String(Utf8View(layout_text.node().data()).substring_view(fragment.start(), fragment.length()).as_string()).characters());
                }
            }
        }
    }

    layout_node.style().for_each_property([&](auto& key, auto& value) {
        for (int i = 0; i < indent; ++i)
            dbgprintf("    ");
        dbgprintf("  (%s: %s)\n", key.characters(), value.to_string().characters());
    });

    ++indent;
    layout_node.for_each_child([](auto& child) {
        dump_tree(child);
    });
    --indent;
}

void dump_rule(const StyleRule& rule)
{
    dbgprintf("Rule:\n");
    for (auto& selector : rule.selectors()) {
        dbgprintf("  Selector:\n");
        for (auto& component : selector.components()) {
            const char* type_description = "Unknown";
            switch (component.type) {
            case Selector::Component::Type::Invalid:
                type_description = "Invalid";
                break;
            case Selector::Component::Type::Id:
                type_description = "Id";
                break;
            case Selector::Component::Type::Class:
                type_description = "Class";
                break;
            case Selector::Component::Type::TagName:
                type_description = "TagName";
                break;
            }
            const char* relation_description = "";
            switch (component.relation) {
            case Selector::Component::Relation::None:
                break;
            case Selector::Component::Relation::ImmediateChild:
                relation_description = "{ImmediateChild}";
                break;
            case Selector::Component::Relation::Descendant:
                relation_description = "{Descendant}";
                break;
            }
            dbgprintf("    %s:%s %s\n", type_description, component.value.characters(), relation_description);
        }
    }
    dbgprintf("  Declarations:\n");
    for (auto& property : rule.declaration().properties()) {
        dbgprintf("    '%s': '%s'\n", property.name.characters(), property.value->to_string().characters());
    }
}

void dump_sheet(const StyleSheet& sheet)
{
    dbgprintf("StyleSheet{%p}: %d rule(s)\n", &sheet, sheet.rules().size());

    for (auto& rule : sheet.rules()) {
        dump_rule(rule);
    }
}