$value ) {
register_nav_menu( $key, $value );
}
}
add_action( "after_setup_theme", "wpgrade_register_custom_menus" );
/*
* Function for displaying The Main Header Menu
*/
function wpgrade_main_nav() {
// test if there are menu locations to prevent errors
$theme_locations = get_nav_menu_locations();
$args = array(
'theme_location' => 'main_menu',
'menu' => '',
'container' => '',
'container_id' => '',
'menu_class' => 'nav nav--main nav--items-menu',
'menu_id' => '',
'fallback_cb' => 'wpgrade_please_select_a_menu',
'items_wrap' => '',
);
wp_nav_menu( $args );
// }
}
/*
* Function for displaying The Footer Menu
*/
function wpgrade_footer_nav($before = '', $after = '') {
$theme_locations = get_nav_menu_locations();
if ( isset( $theme_locations["footer_menu"] ) && ( $theme_locations["footer_menu"] != 0 ) ) {
$args = array(
'theme_location' => 'footer_menu',
'menu' => '',
'container' => '',
'container_id' => '',
// 'menu_class' => 'site-navigation site-navigation--footer site-navigation--secondary flush--bottom',
'menu_class' => 'nav--footer',
'fallback_cb' => false,
'menu_id' => '',
'depth' => 1,
'items_wrap' => $before . '' . $after,
);
wp_nav_menu( $args );
}
}
function wpgrade_please_select_a_menu() {
echo '
';
}