<?php

/**
 * Add custom fields to $item nav object
 * in order to be used in custom Walker
 *
 * @access      public
 * @since       1.0 
 * @return      void
*/
function mts_add_custom_nav_fields( $menu_item ) {
    $menu_item->color = get_post_meta( $menu_item->ID, '_menu_item_color', true );
    $menu_item->icon = get_post_meta( $menu_item->ID, '_menu_item_icon', true );
    return $menu_item;
}
add_filter( 'wp_setup_nav_menu_item', 'mts_add_custom_nav_fields' );


/**
 * Save menu custom fields
 *
 * @access      public
 * @since       1.0 
 * @return      void
*/
function mts_update_custom_nav_fields( $menu_id, $menu_item_db_id, $args ) {
    // Check if element is properly sent
    if ( !empty($_REQUEST['menu-item-color']) && is_array( $_REQUEST['menu-item-color']) ) {
        $color_value = $_REQUEST['menu-item-color'][$menu_item_db_id];
        update_post_meta( $menu_item_db_id, '_menu_item_color', $color_value );
    }
    if ( !empty($_REQUEST['menu-item-icon']) && is_array( $_REQUEST['menu-item-icon']) ) {
        $icon_value = $_REQUEST['menu-item-icon'][$menu_item_db_id];
        update_post_meta( $menu_item_db_id, '_menu_item_icon', $icon_value );
    }
}
add_action( 'wp_update_nav_menu_item','mts_update_custom_nav_fields', 10, 3 );

/**
 * Define new Walker edit
 *
 * @access      public
 * @since       1.0 
 * @return      void
*/
function mts_edit_walker($walker,$menu_id) {
    return 'Walker_Nav_Menu_Edit_Custom';
}
add_filter( 'wp_edit_nav_menu_walker', 'mts_edit_walker', 10, 2 );

/**
 *  /!\ This is a copy of Walker_Nav_Menu_Edit class in core
 * 
 * Create HTML list of nav menu input items.
 *
 * @package WordPress
 * @since 3.0.0
 * @uses Walker_Nav_Menu
 */
if( ! class_exists( 'Walker_Nav_Menu_Edit_Custom' ) ) {
	class Walker_Nav_Menu_Edit_Custom extends Walker_Nav_Menu  {
		/**
		 * @see Walker_Nav_Menu::start_lvl()
		 * @since 3.0.0
		 *
		 * @param string $output Passed by reference.
		 */
		function start_lvl(&$output, $depth = 0, $args = array()) {	
		}
		
		/**
		 * @see Walker_Nav_Menu::end_lvl()
		 * @since 3.0.0
		 *
		 * @param string $output Passed by reference.
		 */
		function end_lvl(&$output, $depth = 0, $args = array()) {
		}

		/**
		 * @see Walker::start_el()
		 * @since 3.0.0
		 *
		 * @param string $output Passed by reference. Used to append additional content.
		 * @param object $item Menu item data object.
		 * @param int $depth Depth of menu item. Used for padding.
		 * @param array|object $args
		 * @param int $id
		 */
		function start_el(&$output, $item, $depth = 0, $args = array(), $id = 0) {
		    global $_wp_nav_menu_max_depth;
		   
		    $_wp_nav_menu_max_depth = $depth > $_wp_nav_menu_max_depth ? $depth : $_wp_nav_menu_max_depth;
		
		    $indent = ( $depth ) ? str_repeat( "\t", $depth ) : '';
		
		    ob_start();
		    $item_id = esc_attr( $item->ID );
		    $removed_args = array(
		        'action',
		        'customlink-tab',
		        'edit-menu-item',
		        'menu-item',
		        'page-tab',
		        '_wpnonce',
		    );
		
		    $original_title = '';
		    if ( 'taxonomy' == $item->type ) {
		        $original_title = get_term_field( 'name', $item->object_id, $item->object, 'raw' );
		        if ( is_wp_error( $original_title ) )
		            $original_title = false;
		    } elseif ( 'post_type' == $item->type ) {
		        $original_object = get_post( $item->object_id );
		        $original_title = $original_object->post_title;
		    }
		
		    $classes = array(
		        'menu-item menu-item-depth-' . $depth,
		        'menu-item-' . esc_attr( $item->object ),
		        'menu-item-edit-' . ( ( isset( $_GET['edit-menu-item'] ) && $item_id == $_GET['edit-menu-item'] ) ? 'active' : 'inactive'),
		    );
		
		    $title = $item->title;
		
		    if ( ! empty( $item->_invalid ) ) {
		        $classes[] = 'menu-item-invalid';
		        /* translators: %s: title of menu item which is invalid */
		        $title = sprintf( __( '%s (Invalid)', 'pointpro' ), $item->title );
		    } elseif ( isset( $item->post_status ) && 'draft' == $item->post_status ) {
		        $classes[] = 'pending';
		        /* translators: %s: title of menu item in draft status */
		        $title = sprintf( __('%s (Pending)', 'pointpro' ), $item->title );
		    }
		
		    $title = empty( $item->label ) ? $title : $item->label;
		
		    ?>
		    <li id="menu-item-<?php echo $item_id; ?>" class="<?php echo implode(' ', $classes ); ?>">
		        <dl class="menu-item-bar">
		            <dt class="menu-item-handle">
		                <span class="item-title"><?php echo esc_html( $title ); ?></span>
		                <span class="item-controls">
		                    <span class="item-type"><?php echo esc_html( $item->type_label ); ?></span>
		                    <span class="item-order hide-if-js">
		                        <a href="<?php
		                            echo wp_nonce_url(
		                                add_query_arg(
		                                    array(
		                                        'action' => 'move-up-menu-item',
		                                        'menu-item' => $item_id,
		                                    ),
		                                    remove_query_arg($removed_args, admin_url( 'nav-menus.php' ) )
		                                ),
		                                'move-menu_item'
		                            );
		                        ?>" class="item-move-up"><abbr title="<?php esc_attr_e( 'Move up', 'pointpro' ); ?>">&#8593;</abbr></a>
		                        |
		                        <a href="<?php
		                            echo wp_nonce_url(
		                                add_query_arg(
		                                    array(
		                                        'action' => 'move-down-menu-item',
		                                        'menu-item' => $item_id,
		                                    ),
		                                    remove_query_arg($removed_args, admin_url( 'nav-menus.php' ) )
		                                ),
		                                'move-menu_item'
		                            );
		                        ?>" class="item-move-down"><abbr title="<?php esc_attr_e( 'Move down', 'pointpro' ); ?>">&#8595;</abbr></a>
		                    </span>
		                    <a class="item-edit" id="edit-<?php echo $item_id; ?>" title="<?php esc_attr_e( 'Edit Menu Item', 'pointpro' ); ?>" href="<?php
		                        echo ( isset( $_GET['edit-menu-item'] ) && $item_id == $_GET['edit-menu-item'] ) ? admin_url( 'nav-menus.php' ) : add_query_arg( 'edit-menu-item', $item_id, remove_query_arg( $removed_args, admin_url( 'nav-menus.php#menu-item-settings-' . $item_id ) ) );
		                    ?>"><?php _e( 'Edit Menu Item', 'pointpro' ); ?></a>
		                </span>
		            </dt>
		        </dl>
		
		        <div class="menu-item-settings" id="menu-item-settings-<?php echo $item_id; ?>">
		            <?php if( 'custom' == $item->type ) : ?>
		                <p class="field-url description description-wide">
		                    <label for="edit-menu-item-url-<?php echo $item_id; ?>">
		                        <?php _e( 'URL', 'pointpro' ); ?><br />
		                        <input type="text" id="edit-menu-item-url-<?php echo $item_id; ?>" class="widefat code edit-menu-item-url" name="menu-item-url[<?php echo $item_id; ?>]" value="<?php echo esc_url( $item->url ); ?>" />
		                    </label>
		                </p>
		            <?php endif; ?>
		            <p class="description description-thin">
		                <label for="edit-menu-item-title-<?php echo $item_id; ?>">
		                    <?php _e( 'Navigation Label', 'pointpro' ); ?><br />
		                    <input type="text" id="edit-menu-item-title-<?php echo $item_id; ?>" class="widefat edit-menu-item-title" name="menu-item-title[<?php echo $item_id; ?>]" value="<?php echo esc_attr( $item->title ); ?>" />
		                </label>
		            </p>
		            <p class="description description-thin">
		                <label for="edit-menu-item-attr-title-<?php echo $item_id; ?>">
		                    <?php _e( 'Title Attribute', 'pointpro' ); ?><br />
		                    <input type="text" id="edit-menu-item-attr-title-<?php echo $item_id; ?>" class="widefat edit-menu-item-attr-title" name="menu-item-attr-title[<?php echo $item_id; ?>]" value="<?php echo esc_attr( $item->post_excerpt ); ?>" />
		                </label>
		            </p>
		            <p class="field-link-target description">
		                <label for="edit-menu-item-target-<?php echo $item_id; ?>">
		                    <input type="checkbox" id="edit-menu-item-target-<?php echo $item_id; ?>" value="_blank" name="menu-item-target[<?php echo $item_id; ?>]"<?php checked( $item->target, '_blank' ); ?> />
		                    <?php _e( 'Open link in a new window/tab', 'pointpro' ); ?>
		                </label>
		            </p>
		            <p class="field-css-classes description description-thin">
		                <label for="edit-menu-item-classes-<?php echo $item_id; ?>">
		                    <?php _e( 'CSS Classes (optional)', 'pointpro' ); ?><br />
		                    <input type="text" id="edit-menu-item-classes-<?php echo $item_id; ?>" class="widefat code edit-menu-item-classes" name="menu-item-classes[<?php echo $item_id; ?>]" value="<?php echo esc_attr( implode(' ', $item->classes ) ); ?>" />
		                </label>
		            </p>
		            <p class="field-xfn description description-thin">
		                <label for="edit-menu-item-xfn-<?php echo $item_id; ?>">
		                    <?php _e( 'Link Relationship (XFN)', 'pointpro' ); ?><br />
		                    <input type="text" id="edit-menu-item-xfn-<?php echo $item_id; ?>" class="widefat code edit-menu-item-xfn" name="menu-item-xfn[<?php echo $item_id; ?>]" value="<?php echo esc_attr( $item->xfn ); ?>" />
		                </label>
		            </p>
		            <p class="field-description description description-wide">
		                <label for="edit-menu-item-description-<?php echo $item_id; ?>">
		                    <?php _e( 'Description', 'pointpro' ); ?><br />
		                    <textarea id="edit-menu-item-description-<?php echo $item_id; ?>" class="widefat edit-menu-item-description" rows="3" cols="20" name="menu-item-description[<?php echo $item_id; ?>]"><?php echo esc_html( $item->description ); // textarea_escaped ?></textarea>
		                    <span class="description"><?php _e('The description will be displayed in the menu if the current theme supports it.', 'pointpro' ); ?></span>
		                </label>
		            </p>        
		            <?php
		            /* New fields insertion starts here */
		            ?>

	                <?php $mts_icons = mts_get_icons(); ?>
	                <p class="field-custom description description-thin">
		                <label for="edit-menu-item-icon-<?php echo $item_id; ?>">
		                    <?php _e( 'Icon (optional)', 'pointpro' ); ?><br />
		                    <?php 
	                        echo '<select id="edit-menu-item-icon-'.$item_id.'" name="menu-item-icon['.$item_id.']" style="width: 100%; max-width: 240px;" class="menu-item-icon-select">';
	                		echo '<option value=""'.selected($item->icon, '', false).'>'.__('No Icon', 'pointpro' ).'</option>';
	                        foreach ( $mts_icons as $icon_category => $icons ) {
	                            echo '<optgroup label="'.$icon_category.'">';
	                            foreach ($icons as $icon) {
	                                echo '<option value="'.$icon.'"'.selected($item->icon, $icon, false).'>'.ucwords(str_replace('-', ' ', $icon)).'</option>';
	                            }
	                            echo '</optgroup>';
	                		}
	                
	                		echo '</select>';
	                        ?>
		                </label>
		            </p>
	                
		            <p class="field-custom description description-thin">
		                <label for="edit-menu-item-color-<?php echo $item_id; ?>" style="position:relative;">
		                    <?php _e( 'Color (optional)', 'pointpro' ); ?><br />
		                    <input type="text" id="edit-menu-item-color-<?php echo $item_id; ?>" class="widefat code edit-menu-item-custom edit-menu-color" name="menu-item-color[<?php echo $item_id; ?>]" value="<?php echo esc_attr( $item->color ); ?>" />
		                </label>
		            </p>
	                
		            <?php
		            /* New fields insertion ends here */
		            ?>
		            <div class="menu-item-actions description-wide submitbox">
		                <?php if( 'custom' != $item->type && $original_title !== false ) : ?>
		                    <p class="link-to-original">
		                        <?php printf( __('Original: %s', 'pointpro' ), '<a href="' . esc_url( $item->url ) . '">' . esc_html( $original_title ) . '</a>' ); ?>
		                    </p>
		                <?php endif; ?>
		                <a class="item-delete submitdelete deletion" id="delete-<?php echo $item_id; ?>" href="<?php
		                echo wp_nonce_url(
		                    add_query_arg(
		                        array(
		                            'action' => 'delete-menu-item',
		                            'menu-item' => $item_id,
		                        ),
		                        remove_query_arg($removed_args, admin_url( 'nav-menus.php' ) )
		                    ),
		                    'delete-menu_item_' . $item_id
		                ); ?>"><?php _e('Remove', 'pointpro' ); ?></a> <span class="meta-sep"> | </span> <a class="item-cancel submitcancel" id="cancel-<?php echo $item_id; ?>" href="<?php echo esc_url( add_query_arg( array('edit-menu-item' => $item_id, 'cancel' => time()), remove_query_arg( $removed_args, admin_url( 'nav-menus.php' ) ) ) );
		                    ?>#menu-item-settings-<?php echo $item_id; ?>"><?php _e('Cancel', 'pointpro' ); ?></a>
		            </div>
		
		            <input class="menu-item-data-db-id" type="hidden" name="menu-item-db-id[<?php echo $item_id; ?>]" value="<?php echo $item_id; ?>" />
		            <input class="menu-item-data-object-id" type="hidden" name="menu-item-object-id[<?php echo $item_id; ?>]" value="<?php echo esc_attr( $item->object_id ); ?>" />
		            <input class="menu-item-data-object" type="hidden" name="menu-item-object[<?php echo $item_id; ?>]" value="<?php echo esc_attr( $item->object ); ?>" />
		            <input class="menu-item-data-parent-id" type="hidden" name="menu-item-parent-id[<?php echo $item_id; ?>]" value="<?php echo esc_attr( $item->menu_item_parent ); ?>" />
		            <input class="menu-item-data-position" type="hidden" name="menu-item-position[<?php echo $item_id; ?>]" value="<?php echo esc_attr( $item->menu_order ); ?>" />
		            <input class="menu-item-data-type" type="hidden" name="menu-item-type[<?php echo $item_id; ?>]" value="<?php echo esc_attr( $item->type ); ?>" />
	                
	                <script type="text/javascript">
	                jQuery(document).ready(function($) {
	                    $('#edit-menu-item-icon-<?php echo esc_js( $item_id ); ?>').select2({
	                        formatResult: function(state) {
	                            if (!state.id) return state.text; // optgroup
	                            return '<i class="fa fa-' + state.id + '"></i>&nbsp;&nbsp;' + state.text;
	                        },
	                        formatSelection: function(state) {
	                            if (!state.id) return state.text; // optgroup
	                            return '<i class="fa fa-' + state.id + '"></i>&nbsp;&nbsp;' + state.text;
	                        },
	                        escapeMarkup: function(m) { return m; }
	                    });
	                    $('#edit-menu-item-color-<?php echo $item_id; ?>').wpColorPicker();
	                });
	                </script>
	                
	            </div><!-- .menu-item-settings-->
		        <ul class="menu-item-transport"></ul>
		    <?php
		    
		    $output .= ob_get_clean();

		    }
	}
}

/**
 * Custom Walker
 *
 * @access      public
 * @since       1.0 
 * @return      void
*/
if( ! class_exists( 'mts_menu_walker' ) ) {
	class mts_menu_walker extends Walker_Nav_Menu {

		function start_el( &$output, $item, $depth = 0, $args = array(), $id = 0 ) {

			$indent = ( $depth ) ? str_repeat( "\t", $depth ) : '';

			$classes = empty( $item->classes ) ? array() : (array) $item->classes;
			$classes[] = 'menu-item-' . $item->ID;

			$class_names = join( ' ', apply_filters( 'nav_menu_css_class', array_filter( $classes ), $item, $args, $depth ) );
			$class_names = $class_names ? ' class="' . esc_attr( $class_names ) . '"' : '';

			$id = apply_filters( 'nav_menu_item_id', 'menu-item-'. $item->ID, $item, $args, $depth );
			$id = $id ? ' id="' . esc_attr( $id ) . '"' : '';

			$output .= $indent . '<li' . $id . $class_names .'>';

			$atts = array();
			$atts['title']  = ! empty( $item->attr_title ) ? $item->attr_title : '';
			$atts['target'] = ! empty( $item->target )     ? $item->target     : '';
			$atts['rel']    = ! empty( $item->xfn )        ? $item->xfn        : '';
			$atts['href']   = ! empty( $item->url )        ? $item->url        : '';

			$atts = apply_filters( 'nav_menu_link_attributes', $atts, $item, $args, $depth );

			$attributes = '';
			foreach ( $atts as $attr => $value ) {
				if ( ! empty( $value ) ) {
					$value = ( 'href' === $attr ) ? esc_url( $value ) : esc_attr( $value );
					$attributes .= ' ' . $attr . '="' . $value . '"';
				}
			}

			$item_output = $args->before;
			$item_output .= '<a'. $attributes;

			// Color
	        if (!empty($item->color)) 
	            $item_output .= ' style="color: '.$item->color.';"';
	        $item_output .= '>';
	        // Icon
	        if (!empty($item->icon)) {
	            $item_output .= '<i class="fa fa-'.$item->icon.'"></i> ';
	        }

			$item_output .= $args->link_before . apply_filters( 'the_title', $item->title, $item->ID ) . $args->link_after;
			$item_output .= '</a>';
			$item_output .= $args->after;

			$output .= apply_filters( 'walker_nav_menu_start_el', $item_output, $item, $depth, $args );
		}
	}
}	

// Enqueue iris color picker.
function mts_custom_menu_admin_scripts($hook) {
    if ($hook != 'nav-menus.php')
        return;
        
    wp_enqueue_script('wp-color-picker');
	wp_enqueue_style('wp-color-picker');

    wp_enqueue_script(
		'select2', 
		get_template_directory_uri().'/options/js/select2.min.js', 
		array('jquery'),
		null,
		true
	);
    wp_enqueue_style(
		'select2', 
		get_template_directory_uri().'/options/css/select2.css', 
		array(),
		null,
		'all'
	);
    wp_enqueue_style(
		'font-awesome', 
		get_template_directory_uri().'/css/font-awesome.min.css',
		array(),
		null,
		'all'
	);
}
add_action('admin_enqueue_scripts', 'mts_custom_menu_admin_scripts');

?>