Пользователь Nav Nav Walker отображает текущий элемент меню детей или братьев и сестер нет детей

Я бездельничал / искал часы и все еще не могу заставить это работать, поэтому я наконец сдаюсь и прошу некоторую помощь.

Я пытаюсь написать пользовательский ходок, который показывает только текущие страницы детей, или если нет детей отображать страницы братьев и сестер.

Например, возьмите следующее дерево меню:

• 1,0
  • 1.2.0
    • 1.3.0
    • 1.3.1
    • 1.3.2
  • 1.2.1
  • 1.2.2
• 2,0

Давайте предположим, что я нахожусь на текущей странице 1.2.0. На этой странице я хочу показать, что это дети (1.3.0, 1.3.1, 1.3.2)

однако, если я нахожусь на странице 1.2.2, так как у него нет дочерних элементов, он должен отобразить своих братьев и сестер текущего уровня, поэтому он должен показать мне (1.2.0, 1.2.1, 1.2.2).

Это ходунки, которые я использовал для отображения только дочерних элементов текущего пункта меню. Или пункты меню братья и сестры, если у него нет собственных детей.

В классе есть комментарии, объясняющие каждый раздел

<?php

class SH_Child_Only_Walker extends Walker_Nav_Menu {

private $ID;
private $depth;
private $classes = array();
private $child_count = 0;
private $have_current = false;


// Don't start the top level
function start_lvl(&$output, $depth=0, $args=array()) {

    if( 0 == $depth || $this->depth != $depth )
        return;

    parent::start_lvl($output, $depth,$args);
}

// Don't end the top level
function end_lvl(&$output, $depth=0, $args=array()) {
    if( 0 == $depth || $this->depth != $depth )
        return;

    parent::end_lvl($output, $depth,$args);
}

// Don't print top-level elements
function start_el(&$output, $item, $depth=0, $args=array()) {

    $is_current = in_array('current-menu-item', $this->classes);

    if( 0 == $depth || ! $is_current )
        return;

    parent::start_el($output, $item, $depth, $args);
}

function end_el(&$output, $item, $depth=0, $args=array()) {
    if( 0 == $depth )
        return;

    parent::end_el($output, $item, $depth, $args);
}

// Only follow down one branch
function display_element( $element, &$children_elements, $max_depth, $depth=0, $args, &$output ) {

    // Check if element is in the current tree to display
    $current_element_markers = array( 'current-menu-item', 'current-menu-parent', 'current-menu-ancestor' );
    $this->classes = array_intersect( $current_element_markers, $element->classes );

    // If element has a 'current' class, it is an ancestor of the current element
    $ancestor_of_current = !empty($this->classes);

    // check if the element is the actual page element we are on.
    $is_current = in_array('current-menu-item', $this->classes);

    // if it is the current element
    if($is_current) {

        // set the count / ID / and depth to use in the other functions.
        $this->child_count = ( isset($children_elements[$element->ID]) ) ? count($children_elements[$element->ID]) : 0;
        $this->ID = $element->ID;
        $this->depth = $depth;
        $this->have_current = true;

        if($this->child_count > 0) {

            // if there are children loop through them and display the kids.
            foreach( $children_elements[$element->ID] as $child ) {
                parent::display_element( $child, $children_elements, $max_depth, $depth, $args, $output );
            }

        } else {
            // no children so loop through kids of parent item.
            foreach( $children_elements[$element->menu_item_parent] as $child ) {
                parent::display_element( $child, $children_elements, $max_depth, $depth, $args, $output );
            }

        }
    }

    // if depth is zero and not in current tree go to the next element
    if ( 0 == $depth && !$ancestor_of_current)
        return;

    // if we aren't on the current element proceed as normal
    if(! $this->have_current )
        parent::display_element( $element, $children_elements, $max_depth, $depth, $args, $output );
}
}

Прикрепите его, как если бы вы использовали любой другой пользовательский ходок в wp_nav_menu

<?php
wp_nav_menu( array(
    'menu' => 'primary-menu'
    ,'container' => 'nav'
    ,'container_class' => 'subpages'
    ,'depth' => 0
    ,'walker' => new SH_Child_Only_Walker()
 ));
?>

У меня был похожий опыт. Возможно, вы захотите подумать о том, как убрать логику страниц с бродилки. В основном, скомпилируйте текущую иерархию страниц как объект. Затем используйте параметр exclude в функции wp_nav_menu. Теперь исключенные страницы будут зависеть от того, есть ли у текущей страницы дочерние элементы. Если нет детей, покажите братьев; если дети - это дети, в конце строки показывают братьев и детей; если есть дети и внуки, исключите братьев и покажите детей и внуков.