Menus: Save display locations for new menus.

Fixes a bug preventing the locations saving when creating a new menu. Introduced in [50938].

Props dlh, andraganescu, mukesh27, desrosj.
Fixes #53877.



git-svn-id: https://develop.svn.wordpress.org/trunk@51688 602fd350-edb4-49c9-b593-d223f7449a82
This commit is contained in:
Peter Wilson 2021-08-30 04:11:30 +00:00
parent 8dee6dcc55
commit 485c6b83db

View File

@ -316,6 +316,7 @@ switch ( $action ) {
check_admin_referer( 'update-nav_menu', 'update-nav-menu-nonce' );
// Merge new and existing menu locations if any new ones are set.
$new_menu_locations = array();
if ( isset( $_POST['menu-locations'] ) ) {
$new_menu_locations = array_map( 'absint', $_POST['menu-locations'] );
$menu_locations = array_merge( $menu_locations, $new_menu_locations );
@ -353,6 +354,15 @@ switch ( $action ) {
break; // There should only be 1.
}
set_theme_mod( 'nav_menu_locations', $locations );
} elseif ( count( $new_menu_locations ) > 0 ) {
// If locations have been selected for the new menu, save those.
$locations = get_nav_menu_locations();
foreach ( array_keys( $new_menu_locations ) as $location ) {
$locations[ $location ] = $nav_menu_selected_id;
}
set_theme_mod( 'nav_menu_locations', $locations );
}