WordPress菜单获取树形结构数组自定义万能菜单

虽然 WordPress 提供了 wp_nav_menu 方法来输出菜单，但是如果你想实现超级菜单（Mega Menu），单靠它就做不到了。

好在有 wp_get_nav_menu_object 和 wp_get_nav_menu_items 这两个菜单接口，可以先取得扁平化的菜单数组，然后稍稍加工一下就能得到树形结构的菜单数组了。代码如下：

function get_nav_menu_tree( $location ) {
    $locations = get_nav_menu_locations();
    $menu_id = $locations[$location] ;
    $menu_object = wp_get_nav_menu_object($menu_id);
    $menu_items = wp_get_nav_menu_items($menu_object->term_id);
 
    _wp_menu_item_classes_by_context($menu_items);
 
    $menu = array();
    $submenus = array();
 
    foreach ($menu_items as $m) {
        $m->children = array();
 
        if (!$m->menu_item_parent) {
            $menu[$m->ID] = $m;
        } else {
            $submenus[$m->ID] = $m;
 
            if (isset($menu[$m->menu_item_parent])) {
                $menu[$m->menu_item_parent]->children[$m->ID] = &$submenus[$m->ID];
            } else {
                $submenus[$m->menu_item_parent]->children[$m->ID] = $submenus[$m->ID];
            }
        }
    }
 
    return $menu;
}

调用方法：

<?php echo get_nav_menu_tree( 菜单名称 ) ?>

这样就可以输出整个菜单的对象，自己 var_dump() 即可查看全部菜单结构，自己循环遍历输出即可。