859669ca by Chris Boden

Added class to items that have children

1 parent 840f94ca
......@@ -40,6 +40,7 @@ class HierarchicalMenu implements Countable, Iterator {
$last_id = $item->ID;
} elseif ($item->menu_item_parent == $last_id) {
$this->contents[$this->id_lookup[$last_id]]->_children = true;
$this->contents[$this->id_lookup[$last_id]]->classes[] = 'menu-item-children';
$nest[] = $last_id;
$last_id = $item->ID;
......@@ -51,9 +52,10 @@ class HierarchicalMenu implements Countable, Iterator {
}
}
$me = __CLASS__;
foreach ($this->contents as $key => &$item) {
if (true === $item->_children) {
$item->HierarchicalMenu = new HierarchicalMenu($menu, $item->ID);
$item->{str_replace(Array(__NAMESPACE__, '\\'), '', $me)} = new $me($menu, $item->ID);
}
unset($this->contents[$key]->_children);
}
......