Kiểm tra xem trang có trong một menu nhất định không

Tôi có một khách hàng muốn có hai "menu chính" và chúng tôi cần hiển thị theo chương trình bất kỳ menu nào có chứa trang hiện tại. Làm thế nào tôi có thể làm điều đó?

Ví dụ: Chúng tôi đang ở trang "Giới thiệu". Chỉ ra menu nào trong hai menu chứa trang "Giới thiệu" và hiển thị cái đó.

Đây là chức năng tôi đã viết để tìm ra điều này. Bạn cung cấp cho nó một slug / name / ID và ID bài đăng / trang và nó sẽ trả về TRUEnếu bài đăng / trang đó nằm trong menu được chỉ định và FALSEnếu không. Sau đó, nó chỉ đơn giản là vấn đề của một câu lệnh if / other nhanh chóng để kiểm tra hai menu và hiển thị chính xác.

/**
 * Check if post is in a menu
 *
 * @param $menu menu name, id, or slug
 * @param $object_id int post object id of page
 * @return bool true if object is in menu
 */
function cms_is_in_menu( $menu = null, $object_id = null ) {

    // get menu object
    $menu_object = wp_get_nav_menu_items( esc_attr( $menu ) );

    // stop if there isn't a menu
    if( ! $menu_object )
        return false;

    // get the object_id field out of the menu object
    $menu_items = wp_list_pluck( $menu_object, 'object_id' );

    // use the current post if object_id is not specified
    if( !$object_id ) {
        global $post;
        $object_id = get_queried_object_id();
    }

    // test if the specified page is in the menu or not. return true or false.
    return in_array( (int) $object_id, $menu_items );

}

Cách sử dụng ví dụ:

if( cms_is_in_menu( 'main-menu' ) ) {
    // do something like wp_nav_menu( $args );
}