Пользовательский вывод меню wp nav (отображение всех дочерних элементов верхнего элемента меню в текущей ветви)

Идея

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

Этот крючок определен в /wp-includes/nav-menu-templates.php и когда он запускается, он переходит к функциям, подключающим к нему массив $sorted_menu_items, который содержит все элементы печатаемого меню, если функция изменит этот массив, добавив или удалив элементы, результирующее меню будет изменено.

Метод

Если применен фильтр к wp_nav_menu_objects напрямую, все меню будут отфильтрованы, поэтому я подумал, что лучше создать функцию, которая будет обертывать wp_nav_menu добавление фильтра перед вызовом wp_nav_menu и удаление его после : таким образом фильтруется только нужное меню.

Рабочий процесс фильтрации

1. Цикл всех элементов, переданных wp_nav_menu_objects крючок фильтра
2. Создайте 2 вспомогательных массива: один только из родительских идентификаторов, один с такими элементами, как $itemid => $parentid
3. При циклировании элементов также проверьте соответствие URL-адреса элемента текущий URL-адрес
4. если URL-адрес не совпадает, возвращайте только родительские элементы
5. если URL-адрес совпадает, используя созданные вспомогательные массивы, верните нужные элементы

Код

Код, который я написал для этого, использует 5 функций, поэтому я создаю плагин, который содержит все это, вот код:

<?php
/**
 * Plugin Name: Filtered Nav Menus
 * Author: Giuseppe  Mazzapica
 * Plugin URI: http://wordpress.stackexchange.com/questions/118720/
 * Author URI: http://wordpress.stackexchange.com/users/35541/
 */

/**
 * The API function, just a wrap for wp_nav_menu adding filter first and removing after
 */ 
function filtered_nav_menu( $args = array() ) {
  $echo = isset($args['echo']) ? (bool)$args['echo'] : true;
  $args['echo'] = false;
  add_filter( 'wp_nav_menu_objects', 'gmfnv_filter', 999 );
  $menu = wp_nav_menu( $args );
  remove_filter( 'wp_nav_menu_objects', 'gmfnv_filter', 999 );
  if ( $echo ) echo $menu;
  else return $menu;
}

/**
 * The filter callback, return the filtered elements
 */
function gmfnv_filter( $items ) {
  $found = false;
  $parents = $items_tree = $allowed = array();
  $all_items = $items;
  while ( ! empty( $items ) ) {
    $item = array_shift( $items );
    $items_tree[$item->ID] = $item->menu_item_parent;
    if ( (int) $item->menu_item_parent == 0 ) $parents[] = $item->ID;
    if ( isset($item->current) && $item->current ) $found = $item->ID;
  }
  if ( ! $found ) {
    $ids = $parents;
  } else {
    $tree = gmfnv_get_tree( $found, $all_items, $items_tree );
    $ids = array_merge( $parents, $tree );
  }
  foreach ( $all_items as $item ) {
    if ( in_array( $item->ID, $ids ) ) $allowed[] = $item;
  }
  return $allowed;
}


/**
 * Helper function: take the matched element if and the helper array and
 * return the item ancestors by gmfnv_get_parents,
 * and the children of these ancestors returned by gmfnv_get_parents
 * using gmfnv_get_parents
 */
function gmfnv_get_tree( $test, $items, $tree ) {
  $parents = gmfnv_get_parents( $test, $items );
  $parents[] = $test;
  $n = array();
  foreach ( $parents as $parent ) {
    $n = array_merge( $n, gmfnv_get_childrens( $parent, $tree ) );
  }
  return array_unique( $n );
}


/**
 * Helper function: return ancestors of an element using the helper array
 */
function gmfnv_get_parents( $test, $items ) {
  $parents = array();
  foreach( $items as $item ) {
      if (
        (isset($item->current_item_ancestor) && $item->current_item_ancestor)
        || (isset($item->current_item_ancestor) && $item->current_item_ancestor)
      ) $parents[] = $item->ID;
  }
  return $parents;
}


/**
 * Helper function: return children of an element using the helper array
 */
function gmfnv_get_childrens( $test, $tree ) {
  $children = array();
  foreach ( $tree as $child => $parent ) {
    if ( $parent == $test ) $children[] = $child;
  }
  return $children;
}

Как

Создайте файл, содержащий этот плагин, поместите его в папку plugins и активируйте.

Если вы хотите отфильтровать меню по мере необходимости, вместо этого используйте wp_nav_menu( $args ) использовать

filtered_nav_menu( $args );

Отказ от ответственности

Код предоставлен как есть, без гарантии, но просто быстро протестирован на PHP 5.4, WP 3.7 с активной темой twentytherteen и без других плагинов: в данном случае это сработало.