<?php

class td_nav_menu_edit_walker extends Walker_Nav_Menu_Edit {
    public function start_el(&$output, $item, $depth = 0, $args = array(), $id = 0) {

        $control_buffy = '';

        //read the menu setting from post meta (menu id, key, single)
        $td_mega_menu_cat = get_post_meta($item->ID, 'td_mega_menu_cat', true);
        $td_mega_menu_page_id = get_post_meta($item->ID, 'td_mega_menu_page_id', true);

        //make the tree
        if ( $item->type === 'taxonomy' && $item->object !== 'category' ) {
            $td_category_tree = array(' - Not mega menu - ' => '',  ' Enable mega menu ' => $item->object_id);
            $description_txt = 'Make this a term mega menu';
        } else {
            $td_category_tree = array_merge (array(' - Not mega menu - ' => ''), td_util::get_category2id_array(false, false));
            $description_txt = 'Make this a category mega menu';
        }

        //make a new ui control ( dropdown )
        $control_buffy .= '<p class="description description-wide"><br><br>';
            $control_buffy .= '<label>';
                $control_buffy .= $description_txt;
            $control_buffy .= '</label>';
            $control_buffy .= '<select name="td_mega_menu_cat[' . $item->ID . ']" id="" class="widefat code edit-menu-item-url">';
                foreach ($td_category_tree as $category => $category_id) {
                	$disabled = '';
                	if ( '__' === $category_id && false !== strpos($category, '--')) {
                		$disabled = 'disabled';
	                }
                    $control_buffy .= '<option value="' . $category_id . '"' . selected($td_mega_menu_cat, $category_id, false) . ' ' . $disabled . '>' . $category . '</option>';
                }
            $control_buffy .= ' </select>';
        $control_buffy .= '</p>';


        if (td_api_features::is_enabled('page_mega_menu') === true){
            $control_buffy .= '<br>OR<br>';

            //make a new ui control ( dropdown )
            $control_buffy .= '<p class="description description-wide">';

                $control_buffy .= '<label>';
                    $control_buffy .= 'Load a page in the menu (enter the page ID)';
                $control_buffy .= '</label><br>';
                $control_buffy .= '<input name="td_mega_menu_page_id[' . $item->ID . ']" type="text" value="' . $td_mega_menu_page_id . '" />';
                $control_buffy .= '<span class="td-wpa-info"><strong>Just a tip:</strong> If you choose to load a mega menu or a page, please do not add submenus to this item. The mega menu and mega page menu have to be the top most menu item. <a href="http://forum.tagdiv.com/menus-newsmag/" target="_blank">Read more</a></span>';


            $control_buffy .= '</p>';
        }

        //run the parent and add in $buffy (byref) our code via regex
        $buffy = '';
        parent::start_el($buffy, $item, $depth, $args, $id);
        $buffy = preg_replace('/(?=<div.*submitbox)/', $control_buffy, $buffy);



        $output .= $buffy;
    }
}