srctree

#include <cassert>

#include <variant>

inline std::string_view dom_name(LayoutBox const &node) {

assert(node.node);

return std::get<dom::Element>(node.node->node).name;

}

 

inline std::vector<LayoutBox const *> dom_children(LayoutBox const &node) {

assert(node.node);

std::vector<LayoutBox const *> children{};

for (auto const &child : node.children) {

if (child.type == LayoutType::AnonymousBlock) {

for (auto const &inline_child : child.children) {

assert(inline_child.node);

if (!std::holds_alternative<dom::Element>(inline_child.node->node)) {

continue;

}

 

children.push_back(&inline_child);

}

continue;

}

 

assert(child.node);

if (!std::holds_alternative<dom::Element>(child.node->node)) {

continue;

}

 

children.push_back(&child);

}

return children;

}

 

etest::test("xpath", [] {

dom::Node html_node = dom::Element{"html"s};

dom::Node div_node = dom::Element{"div"s};

dom::Node p_node = dom::Element{"p"s};

dom::Node text_node = dom::Text{"hello!"s};

style::StyledNode styled_node{

.node = html_node,

.properties{{css::PropertyId::Display, "block"}},

.children{

style::StyledNode{.node = div_node, .properties{{css::PropertyId::Display, "block"}}},

style::StyledNode{.node = text_node},

style::StyledNode{.node = div_node,

.children{

style::StyledNode{.node = p_node},

style::StyledNode{.node = text_node},

}},

},

};

auto layout = layout::create_layout(styled_node, 123).value();

 

// Verify that we have a shady anon-box to deal with in here.

expect_eq(layout.children.size(), std::size_t{2});

expect_eq(layout.children.at(1).type, LayoutType::AnonymousBlock);

 

auto const &anon_block = layout.children.at(1);

 

using NodeVec = std::vector<layout::LayoutBox const *>;

expect_eq(dom::nodes_by_xpath(layout, "/html"), NodeVec{&layout});

expect_eq(dom::nodes_by_xpath(layout, "/html/div"), NodeVec{&layout.children[0], &anon_block.children[1]});

expect_eq(dom::nodes_by_xpath(layout, "/html/div/"), NodeVec{});

expect_eq(dom::nodes_by_xpath(layout, "/html/div/p"), NodeVec{&anon_block.children[1].children[0]});

expect_eq(dom::nodes_by_xpath(layout, "/htm/div"), NodeVec{});

expect_eq(dom::nodes_by_xpath(layout, "//div"), NodeVec{});

});