<?php

/**

 * WordPress Administration for Navigation Menus

 * Interface functions

 *

 * @version 2.0.0

 *

 * @package WordPress

 * @subpackage Administration

 */

/** Load WordPress Administration Bootstrap */

require_once( './admin.php' );

// Load all the nav menu interface functions

require_once( ABSPATH . 'wp-admin/includes/nav-menu.php' );

if ( ! current_theme_supports( 'menus' ) && ! current_theme_supports( 'widgets' ) )

	wp_die( __( 'Your theme does not support navigation menus or widgets.' ) );

// Permissions Check

if ( ! current_user_can('edit_theme_options') )

	wp_die( __( 'Cheatin’ uh?' ) );

wp_enqueue_script( 'nav-menu' );

if ( wp_is_mobile() )

	wp_enqueue_script( 'jquery-touch-punch' );

// Container for any messages displayed to the user

$messages = array();

// Container that stores the name of the active menu

$nav_menu_selected_title = '';

// The menu id of the current menu being edited

$nav_menu_selected_id = isset( $_REQUEST['menu'] ) ? (int) $_REQUEST['menu'] : 0;

// Allowed actions: add, update, delete

$action = isset( $_REQUEST['action'] ) ? $_REQUEST['action'] : 'edit';

switch ( $action ) {

	case 'add-menu-item':

		check_admin_referer( 'add-menu_item', 'menu-settings-column-nonce' );

		if ( isset( $_REQUEST['nav-menu-locations'] ) )

			set_theme_mod( 'nav_menu_locations', array_map( 'absint', $_REQUEST['menu-locations'] ) );

		elseif ( isset( $_REQUEST['menu-item'] ) )

			wp_save_nav_menu_items( $nav_menu_selected_id, $_REQUEST['menu-item'] );

		break;

	case 'move-down-menu-item' :

		// moving down a menu item is the same as moving up the next in order

		check_admin_referer( 'move-menu_item' );

		$menu_item_id = isset( $_REQUEST['menu-item'] ) ? (int) $_REQUEST['menu-item'] : 0;

		if ( is_nav_menu_item( $menu_item_id ) ) {

			$menus = isset( $_REQUEST['menu'] ) ? array( (int) $_REQUEST['menu'] ) : wp_get_object_terms( $menu_item_id, 'nav_menu', array( 'fields' => 'ids' ) );

			if ( ! is_wp_error( $menus ) && ! empty( $menus[0] ) ) {

				$menu_id = (int) $menus[0];

				$ordered_menu_items = wp_get_nav_menu_items( $menu_id );

				$menu_item_data = (array) wp_setup_nav_menu_item( get_post( $menu_item_id ) );

				// set up the data we need in one pass through the array of menu items

				$dbids_to_orders = array();

				$orders_to_dbids = array();

				foreach( (array) $ordered_menu_items as $ordered_menu_item_object ) {

					if ( isset( $ordered_menu_item_object->ID ) ) {

						if ( isset( $ordered_menu_item_object->menu_order ) ) {

							$dbids_to_orders[$ordered_menu_item_object->ID] = $ordered_menu_item_object->menu_order;

							$orders_to_dbids[$ordered_menu_item_object->menu_order] = $ordered_menu_item_object->ID;

						}

					}

				}

				// get next in order

				if (

					isset( $orders_to_dbids[$dbids_to_orders[$menu_item_id] + 1] )

				) {

					$next_item_id = $orders_to_dbids[$dbids_to_orders[$menu_item_id] + 1];

					$next_item_data = (array) wp_setup_nav_menu_item( get_post( $next_item_id ) );

					// if not siblings of same parent, bubble menu item up but keep order

					if (

						! empty( $menu_item_data['menu_item_parent'] ) &&

						(

							empty( $next_item_data['menu_item_parent'] ) ||

							$next_item_data['menu_item_parent'] != $menu_item_data['menu_item_parent']

						)

					) {

						$parent_db_id = in_array( $menu_item_data['menu_item_parent'], $orders_to_dbids ) ? (int) $menu_item_data['menu_item_parent'] : 0;

						$parent_object = wp_setup_nav_menu_item( get_post( $parent_db_id ) );

						if ( ! is_wp_error( $parent_object ) ) {

							$parent_data = (array) $parent_object;

							$menu_item_data['menu_item_parent'] = $parent_data['menu_item_parent'];

							update_post_meta( $menu_item_data['ID'], '_menu_item_menu_item_parent', (int) $menu_item_data['menu_item_parent'] );

						}

					// make menu item a child of its next sibling

					} else {

						$next_item_data['menu_order'] = $next_item_data['menu_order'] - 1;

						$menu_item_data['menu_order'] = $menu_item_data['menu_order'] + 1;

						$menu_item_data['menu_item_parent'] = $next_item_data['ID'];

						update_post_meta( $menu_item_data['ID'], '_menu_item_menu_item_parent', (int) $menu_item_data['menu_item_parent'] );

						wp_update_post($menu_item_data);

						wp_update_post($next_item_data);

					}

				// the item is last but still has a parent, so bubble up

				} elseif (

					! empty( $menu_item_data['menu_item_parent'] ) &&

					in_array( $menu_item_data['menu_item_parent'], $orders_to_dbids )

				) {

					$menu_item_data['menu_item_parent'] = (int) get_post_meta( $menu_item_data['menu_item_parent'], '_menu_item_menu_item_parent', true);

					update_post_meta( $menu_item_data['ID'], '_menu_item_menu_item_parent', (int) $menu_item_data['menu_item_parent'] );

				}

			}

		}

		break;

	case 'move-up-menu-item' :

		check_admin_referer( 'move-menu_item' );

		$menu_item_id = isset( $_REQUEST['menu-item'] ) ? (int) $_REQUEST['menu-item'] : 0;

		if ( is_nav_menu_item( $menu_item_id ) ) {

			$menus = isset( $_REQUEST['menu'] ) ? array( (int) $_REQUEST['menu'] ) : wp_get_object_terms( $menu_item_id, 'nav_menu', array( 'fields' => 'ids' ) );

			if ( ! is_wp_error( $menus ) && ! empty( $menus[0] ) ) {

				$menu_id = (int) $menus[0];

				$ordered_menu_items = wp_get_nav_menu_items( $menu_id );

				$menu_item_data = (array) wp_setup_nav_menu_item( get_post( $menu_item_id ) );

				// set up the data we need in one pass through the array of menu items

				$dbids_to_orders = array();

				$orders_to_dbids = array();

				foreach( (array) $ordered_menu_items as $ordered_menu_item_object ) {

					if ( isset( $ordered_menu_item_object->ID ) ) {

						if ( isset( $ordered_menu_item_object->menu_order ) ) {

							$dbids_to_orders[$ordered_menu_item_object->ID] = $ordered_menu_item_object->menu_order;

							$orders_to_dbids[$ordered_menu_item_object->menu_order] = $ordered_menu_item_object->ID;

						}

					}

				}

				// if this menu item is not first

				if ( ! empty( $dbids_to_orders[$menu_item_id] ) && ! empty( $orders_to_dbids[$dbids_to_orders[$menu_item_id] - 1] ) ) {

					// if this menu item is a child of the previous

					if (

						! empty( $menu_item_data['menu_item_parent'] ) &&

						in_array( $menu_item_data['menu_item_parent'], array_keys( $dbids_to_orders ) ) &&

						isset( $orders_to_dbids[$dbids_to_orders[$menu_item_id] - 1] ) &&

						( $menu_item_data['menu_item_parent'] == $orders_to_dbids[$dbids_to_orders[$menu_item_id] - 1] )

					) {

						$parent_db_id = in_array( $menu_item_data['menu_item_parent'], $orders_to_dbids ) ? (int) $menu_item_data['menu_item_parent'] : 0;

						$parent_object = wp_setup_nav_menu_item( get_post( $parent_db_id ) );

						if ( ! is_wp_error( $parent_object ) ) {

							$parent_data = (array) $parent_object;

							// if there is something before the parent and parent a child of it, make menu item a child also of it

							if (

								! empty( $dbids_to_orders[$parent_db_id] ) &&

								! empty( $orders_to_dbids[$dbids_to_orders[$parent_db_id] - 1] ) &&

								! empty( $parent_data['menu_item_parent'] )

							) {

								$menu_item_data['menu_item_parent'] = $parent_data['menu_item_parent'];

							// else if there is something before parent and parent not a child of it, make menu item a child of that something's parent

							} elseif (

								! empty( $dbids_to_orders[$parent_db_id] ) &&

								! empty( $orders_to_dbids[$dbids_to_orders[$parent_db_id] - 1] )

							) {

								$_possible_parent_id = (int) get_post_meta( $orders_to_dbids[$dbids_to_orders[$parent_db_id] - 1], '_menu_item_menu_item_parent', true);

								if ( in_array( $_possible_parent_id, array_keys( $dbids_to_orders ) ) )

									$menu_item_data['menu_item_parent'] = $_possible_parent_id;

								else

									$menu_item_data['menu_item_parent'] = 0;

							// else there isn't something before the parent

							} else {

								$menu_item_data['menu_item_parent'] = 0;

							}

							// set former parent's [menu_order] to that of menu-item's

							$parent_data['menu_order'] = $parent_data['menu_order'] + 1;

							// set menu-item's [menu_order] to that of former parent

							$menu_item_data['menu_order'] = $menu_item_data['menu_order'] - 1;

							// save changes

							update_post_meta( $menu_item_data['ID'], '_menu_item_menu_item_parent', (int) $menu_item_data['menu_item_parent'] );

							wp_update_post($menu_item_data);

							wp_update_post($parent_data);

						}

					// else this menu item is not a child of the previous

					} elseif (

						empty( $menu_item_data['menu_order'] ) ||

						empty( $menu_item_data['menu_item_parent'] ) ||

						! in_array( $menu_item_data['menu_item_parent'], array_keys( $dbids_to_orders ) ) ||

						empty( $orders_to_dbids[$dbids_to_orders[$menu_item_id] - 1] ) ||

						$orders_to_dbids[$dbids_to_orders[$menu_item_id] - 1] != $menu_item_data['menu_item_parent']

					) {

						// just make it a child of the previous; keep the order

						$menu_item_data['menu_item_parent'] = (int) $orders_to_dbids[$dbids_to_orders[$menu_item_id] - 1];

						update_post_meta( $menu_item_data['ID'], '_menu_item_menu_item_parent', (int) $menu_item_data['menu_item_parent'] );

						wp_update_post($menu_item_data);

					}

				}

			}

		}

		break;

	case 'delete-menu-item':

		$menu_item_id = (int) $_REQUEST['menu-item'];

		check_admin_referer( 'delete-menu_item_' . $menu_item_id );

		if ( is_nav_menu_item( $menu_item_id ) && wp_delete_post( $menu_item_id, true ) )

			$messages[] = '<div id="message" class="updated"><p>' . __('The menu item has been successfully deleted.') . '</p></div>';

		break;

	case 'delete':

		check_admin_referer( 'delete-nav_menu-' . $nav_menu_selected_id );

		if ( is_nav_menu( $nav_menu_selected_id ) ) {

			$deleted_nav_menu = wp_get_nav_menu_object( $nav_menu_selected_id );

			$delete_nav_menu = wp_delete_nav_menu( $nav_menu_selected_id );

			if ( is_wp_error($delete_nav_menu) ) {

				$messages[] = '<div id="message" class="error"><p>' . $delete_nav_menu->get_error_message() . '</p></div>';

			} else {

				// Remove this menu from any locations.

				$locations = get_theme_mod( 'nav_menu_locations' );

				foreach ( (array) $locations as $location => $menu_id ) {

					if ( $menu_id == $nav_menu_selected_id )

						$locations[ $location ] = 0;

				}

				set_theme_mod( 'nav_menu_locations', $locations );

				$messages[] = '<div id="message" class="updated"><p>' . __('The menu has been successfully deleted.') . '</p></div>';

				// Select the next available menu

				$nav_menu_selected_id = 0;

				$_nav_menus = wp_get_nav_menus( array('orderby' => 'name') );

				foreach( $_nav_menus as $index => $_nav_menu ) {

					if ( strcmp( $_nav_menu->name, $deleted_nav_menu->name ) >= 0

					 || $index == count( $_nav_menus ) - 1 ) {

						$nav_menu_selected_id = $_nav_menu->term_id;

						break;

					}

				}

			}

			unset( $delete_nav_menu, $deleted_nav_menu, $_nav_menus );

		} else {

			// Reset the selected menu

			$nav_menu_selected_id = 0;

			unset( $_REQUEST['menu'] );

		}

		break;

	case 'update':

		check_admin_referer( 'update-nav_menu', 'update-nav-menu-nonce' );

		// Update menu theme locations

		if ( isset( $_POST['menu-locations'] ) )

			set_theme_mod( 'nav_menu_locations', array_map( 'absint', $_POST['menu-locations'] ) );

		// Add Menu

		if ( 0 == $nav_menu_selected_id ) {

			$new_menu_title = trim( esc_html( $_POST['menu-name'] ) );

			if ( $new_menu_title ) {

				$_nav_menu_selected_id = wp_update_nav_menu_object( 0, array('menu-name' => $new_menu_title) );

				if ( is_wp_error( $_nav_menu_selected_id ) ) {

					$messages[] = '<div id="message" class="error"><p>' . $_nav_menu_selected_id->get_error_message() . '</p></div>';

				} else {

					$_menu_object = wp_get_nav_menu_object( $_nav_menu_selected_id );

					$nav_menu_selected_id = $_nav_menu_selected_id;

					$nav_menu_selected_title = $_menu_object->name;

					$messages[] = '<div id="message" class="updated"><p>' . sprintf( __('The <strong>%s</strong> menu has been successfully created.'), $nav_menu_selected_title ) . '</p></div>';

				}

			} else {

				$messages[] = '<div id="message" class="error"><p>' . __('Please enter a valid menu name.') . '</p></div>';

			}

		// update existing menu

		} else {

			$_menu_object = wp_get_nav_menu_object( $nav_menu_selected_id );

			$menu_title = trim( esc_html( $_POST['menu-name'] ) );

			if ( ! $menu_title ) {

				$messages[] = '<div id="message" class="error"><p>' . __('Please enter a valid menu name.') . '</p></div>';

				$menu_title = $_menu_object->name;

			}

			if ( ! is_wp_error( $_menu_object ) ) {

				$_nav_menu_selected_id = wp_update_nav_menu_object( $nav_menu_selected_id, array( 'menu-name' => $menu_title ) );

				if ( is_wp_error( $_nav_menu_selected_id ) ) {

					$_menu_object = $_nav_menu_selected_id;

					$messages[] = '<div id="message" class="error"><p>' . $_nav_menu_selected_id->get_error_message() . '</p></div>';

				} else {

					$_menu_object = wp_get_nav_menu_object( $_nav_menu_selected_id );

					$nav_menu_selected_title = $_menu_object->name;

				}

			}

			// Update menu items

			if ( ! is_wp_error( $_menu_object ) ) {

				$unsorted_menu_items = wp_get_nav_menu_items( $nav_menu_selected_id, array('orderby' => 'ID', 'output' => ARRAY_A, 'output_key' => 'ID', 'post_status' => 'draft,publish') );

				$menu_items = array();

				// Index menu items by db ID

				foreach( $unsorted_menu_items as $_item )

					$menu_items[$_item->db_id] = $_item;

				$post_fields = array( 'menu-item-db-id', 'menu-item-object-id', 'menu-item-object', 'menu-item-parent-id', 'menu-item-position', 'menu-item-type', 'menu-item-title', 'menu-item-url', 'menu-item-description', 'menu-item-attr-title', 'menu-item-target', 'menu-item-classes', 'menu-item-xfn' );

				wp_defer_term_counting(true);

				// Loop through all the menu items' POST variables

				if ( ! empty( $_POST['menu-item-db-id'] ) ) {

					foreach( (array) $_POST['menu-item-db-id'] as $_key => $k ) {

						// Menu item title can't be blank

						if ( empty( $_POST['menu-item-title'][$_key] ) )

							continue;

						$args = array();

						foreach ( $post_fields as $field )

							$args[$field] = isset( $_POST[$field][$_key] ) ? $_POST[$field][$_key] : '';

						$menu_item_db_id = wp_update_nav_menu_item( $nav_menu_selected_id, ( $_POST['menu-item-db-id'][$_key] != $_key ? 0 : $_key ), $args );

						if ( is_wp_error( $menu_item_db_id ) )

							$messages[] = '<div id="message" class="error"><p>' . $menu_item_db_id->get_error_message() . '</p></div>';

						elseif ( isset( $menu_items[$menu_item_db_id] ) )

							unset( $menu_items[$menu_item_db_id] );

					}

				}

				// Remove menu items from the menu that weren't in $_POST

				if ( ! empty( $menu_items ) ) {

					foreach ( array_keys( $menu_items ) as $menu_item_id ) {

						if ( is_nav_menu_item( $menu_item_id ) ) {

							wp_delete_post( $menu_item_id );

						}

					}

				}

				// Store 'auto-add' pages.

				$auto_add = ! empty( $_POST['auto-add-pages'] );

				$nav_menu_option = (array) get_option( 'nav_menu_options' );

				if ( ! isset( $nav_menu_option['auto_add'] ) )

					$nav_menu_option['auto_add'] = array();

				if ( $auto_add ) {

					if ( ! in_array( $nav_menu_selected_id, $nav_menu_option['auto_add'] ) )

						$nav_menu_option['auto_add'][] = $nav_menu_selected_id;

				} else {

					if ( false !== ( $key = array_search( $nav_menu_selected_id, $nav_menu_option['auto_add'] ) ) )

						unset( $nav_menu_option['auto_add'][$key] );

				}

				// Remove nonexistent/deleted menus

				$nav_menu_option['auto_add'] = array_intersect( $nav_menu_option['auto_add'], wp_get_nav_menus( array( 'fields' => 'ids' ) ) );

				update_option( 'nav_menu_options', $nav_menu_option );

				wp_defer_term_counting(false);

				do_action( 'wp_update_nav_menu', $nav_menu_selected_id );

				$messages[] = '<div id="message" class="updated"><p>' . sprintf( __('The <strong>%s</strong> menu has been updated.'), $nav_menu_selected_title ) . '</p></div>';

				unset( $menu_items, $unsorted_menu_items );

			}

		}

		break;

}

// Get all nav menus

$nav_menus = wp_get_nav_menus( array('orderby' => 'name') );

// Get recently edited nav menu

$recently_edited = (int) get_user_option( 'nav_menu_recently_edited' );

// If there was no recently edited menu, and $nav_menu_selected_id is a nav menu, update recently edited menu.

if ( !$recently_edited && is_nav_menu( $nav_menu_selected_id ) ) {