<?php if ( ! defined( 'ABSPATH' ) ) { die( 'Direct access forbidden.' ); }
/**
 * Register menus
 */

// Theme wp_nav_menu() locations.
register_nav_menus( array(
	'header'   => esc_html__( 'Header Navigation', 'jevelin' ),
	'header-left'   => esc_html__( 'Header Navigation (left side)', 'jevelin' ),
	'header-right'   => esc_html__( 'Header Navigation (right side)', 'jevelin' ),
	'topbar'   => esc_html__( 'Top Bar Navigation', 'jevelin' ),
));


// End page if not frontend
if( is_admin() ) :
	return ;
endif;


// Force enable mega walker
if( !class_exists( 'FW_Ext_Mega_Menu_Walker' ) && jevelin_framework() == 'redux' ) :
	class FW_Ext_Mega_Menu_Walker { }
endif;


// Imitate Unyson framework funtion
if( !function_exists( 'fw_ext_mega_menu_get_meta' ) ) :
	function fw_ext_mega_menu_get_meta( $id, $name = '' ) {
		if( $name == 'enabled' ) :
			$mega_menu = get_post_meta( $id, 'mega-menu' );
			$status = isset( $mega_menu[0]['enabled'] ) ? $mega_menu[0]['enabled'] : false;
			if( $status ) :
				return true;
			endif;
		elseif( $name == 'icon' ) :
			$mega_menu = get_post_meta( $id, 'mega-menu' );
			$icon = isset( $mega_menu[0]['icon'] ) ? $mega_menu[0]['icon'] : '';
			if( $icon ) :
				return $icon;
			endif;
		endif;
	}
endif;


// Imitate Unyson framework funtion
if( !function_exists( 'fw_ext_mega_menu_get_db_item_option' ) ) :
	function fw_ext_mega_menu_get_db_item_option( $id, $name = '' ) {
		$name = 'jevelin-' . str_replace( 'row/', '', $name );
		$name = str_replace( '_', '-', $name );
		$value = get_post_meta( $id, $name, 1 );
		if( $value ) :
			return $value;
		endif;
	}
endif;


// Replace Unyson mega menu render function
function jevelin_mega_menu_render_item( $params ) {
	extract( $params );

	$attributes_out = [];
	if( is_array( $attributes ) && count( $attributes ) ) :
		foreach( $attributes as $attribute_key => $attribute ) :
			$attributes_out[] = esc_attr( $attribute_key ) . ' = "'. esc_attr( $attribute ) .'"';
		endforeach;
	endif;

	$o = $args->before;
	$o.= '<a '.implode( ' ', $attributes_out ).'>' . ( $args->link_before . $title . $args->link_after ) . '</a>';
	$o.= $args->after;
	return $o;
}


/**
 * Menu Walker
 */
if( class_exists( 'FW_Ext_Mega_Menu_Walker' ) ) {
	class FW_Ext_Mega_Menu_Custom_Walker extends Walker_Nav_Menu {
	// class FW_Ext_Mega_Menu_Custom_Walker extends FW_Ext_Mega_Menu_Walker {

		public $mega_menu_ids = [];

		function display_element( $element, &$children_elements, $max_depth, $depth, $args, &$output ) {
			if ( !$element )
				return;
			$id_field = $this->db_fields['id'];
			$id       = $element->$id_field;
			//display this element
			$this->has_children = ! empty( $children_elements[ $id ] );
			if ( isset( $args[0] ) && is_array( $args[0] ) ) {
				$args[0]['has_children'] = $this->has_children; // Backwards compatibility.
			}
			$cb_args = array_merge( array(&$output, $element, $depth), $args);
			call_user_func_array(array($this, 'start_el'), $cb_args);
			// descend only when the depth is right and there are childrens for this element
			if ( ($max_depth == 0 || $max_depth > $depth+1 ) && isset( $children_elements[$id]) ) {
				foreach( $children_elements[ $id ] as $child ){
	# BEGIN - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
					if ($depth == 0 && fw_ext_mega_menu_get_meta($id, 'enabled') && fw_ext_mega_menu_get_meta($child, 'new-row')) {
						if (isset($newlevel) && $newlevel) {
							$cb_args = array_merge( array(&$output, $depth), $args);
							call_user_func_array(array($this, 'end_lvl'), $cb_args);
							unset($newlevel);
						}
					}
	# END - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
					if ( !isset($newlevel) ) {
						$newlevel = true;
	# BEGIN - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
						if (!isset($mega_menu_container) && $depth == 0 && fw_ext_mega_menu_get_meta($id, 'enabled')) {
							$mega_menu_container = apply_filters('fw_ext_mega_menu_container', array(
								'tag'  => 'div',
								'attr' => array( 'class' => 'mega-menu' )
							), array(
								'element' => $element,
								'children_elements' => $children_elements,
								'max_depth' => $max_depth,
								'depth' => $depth,
								'args' => $args,
							));
							//$output .= '<'. $mega_menu_container['tag'] .' '. fw_attr_to_html($mega_menu_container['attr']) .'>';
						}
						$classes = array('sub-menu' => true);
						if (isset($mega_menu_container)) {
							if ($this->row_contains_icons($element, $child, $children_elements)) {
								$classes['sub-menu-has-icons'] = true;
							}
							$classes['mega-menu-row'] = true;;
						}
						else {
							if ($this->sub_menu_contains_icons($element, $children_elements)) {
								$classes['sub-menu-has-icons'] = true;
							}
						}
						$classes = apply_filters('fw_ext_mega_menu_start_lvl_classes', $classes, array(
							'element' => $element,
							'children_elements' => $children_elements,
							'max_depth' => $max_depth,
							'depth' => $depth,
							'args' => $args,
							'mega_menu_container' => isset($mega_menu_container) ? $mega_menu_container : false
						));
						$classes = array_filter($classes);
	# END - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
						//start the child delimiter
	# BEGIN - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
						//$cb_args = array_merge( array(&$output, $depth), $args);
						$cb_args = array_merge( array(&$output, $depth), $args, array(
							implode(' ', array_keys($classes))
						));
	# END - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
						call_user_func_array(array($this, 'start_lvl'), $cb_args);
					}
					$this->display_element( $child, $children_elements, $max_depth, $depth + 1, $args, $output );
				}
				unset( $children_elements[ $id ] );
			}
			if ( isset($newlevel) && $newlevel ){
				//end the child delimiter
				$cb_args = array_merge( array(&$output, $depth), $args);
				call_user_func_array(array($this, 'end_lvl'), $cb_args);
			}
	# BEGIN - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
			/*if (isset($mega_menu_container)) {
				$output .= '</'. $mega_menu_container['tag'] .'>';
			}*/
	# END - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
			//end this element
			$cb_args = array_merge( array(&$output, $element, $depth), $args);
			call_user_func_array(array($this, 'end_el'), $cb_args);
		}


		function start_lvl( &$output, $depth = 0, $args = array(), $class = 'sub-menu' ) {
			$indent = str_repeat("\t", $depth);
			$output .= "\n$indent<ul class=\"$class\">\n";
		}

		protected function sub_menu_contains_icons($element, $children_elements) {
			$id_field = $this->db_fields['id'];
			$id = $element->$id_field;
			foreach ($children_elements[$id] as $child) {
				if (fw_ext_mega_menu_get_meta($child, 'icon')) {
					return true;
				}
			}
			return false;
		}

		protected function row_contains_icons($row, $first_column, $children_elements) {

			$id_field = $this->db_fields['id'];
			$row_id = $row->$id_field;

			reset($children_elements[$row_id]);

			// navigate to $first_column
			while ($child = next($children_elements[$row_id])) {
				if ($child->$id_field == $first_column->$id_field) {
					break;
				}
			}

			// scan row
			while (true) {
				if (fw_ext_mega_menu_get_meta($child, 'icon')) {
					return true;
				}
				$child = next($children_elements[$row_id]);
				if ($child === false || fw_ext_mega_menu_get_meta($child, 'new-row')) {
					break;
				}
			}

			return false;
		}

		public 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;

			/**
			 * Filter the CSS class(es) applied to a menu item's list item element.
			 *
			 * @since 3.0.0
			 * @since 4.1.0 The `$depth` parameter was added.
			 *
			 * @param array  $classes The CSS classes that are applied to the menu item's `<li>` element.
			 * @param object $item    The current menu item.
			 * @param array  $args    An array of {@see wp_nav_menu()} arguments.
			 * @param int    $depth   Depth of menu item. Used for padding.
			 */
			$class_names = join( ' ', apply_filters( 'nav_menu_css_class', array_filter( $classes ), $item, $args, $depth ) );

			// Mega menu row
			if( fw_ext_mega_menu_get_meta( $item->ID, 'enabled' ) ) :
				$this->mega_menu_ids[$item->ID] = true;
				$class_names.= ' menu-item-has-mega-menu';
				$get_dynamic_elements = fw_ext_mega_menu_get_db_item_option( $item->ID, 'row/dynamic_elements' );
				if( isset( $get_dynamic_elements ) && $get_dynamic_elements && $get_dynamic_elements != 'none' ) :
					$class_names.= ' menu-item-has-children';
				endif;
			endif;

			// Mega menu column
			if( isset( $this->mega_menu_ids[ $item->menu_item_parent ] ) ) :
				$class_names.= ' mega-menu-col';
			endif;

			// Icon
			$icon_class = '';
			if( $depth == 0 && fw_ext_mega_menu_get_meta( $item->ID, 'icon' ) ) :
				$icon_class = fw_ext_mega_menu_get_meta( $item->ID, 'icon' );
			endif;

			$class_names = $class_names ? ' class="' . esc_attr( $class_names ) . '"' : '';

			/**
			 * Filter the ID applied to a menu item's list item element.
			 *
			 * @since 3.0.1
			 * @since 4.1.0 The `$depth` parameter was added.
			 *
			 * @param string $menu_id The ID that is applied to the menu item's `<li>` element.
			 * @param object $item    The current menu item.
			 * @param array  $args    An array of {@see wp_nav_menu()} arguments.
			 * @param int    $depth   Depth of menu item. Used for padding.
			 */
			$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['class']  = $icon_class;
			$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        : '';

			/**
			 * Filter the HTML attributes applied to a menu item's anchor element.
			 *
			 * @since 3.6.0
			 * @since 4.1.0 The `$depth` parameter was added.
			 *
			 * @param array $atts {
			 *     The HTML attributes applied to the menu item's `<a>` element, empty strings are ignored.
			 *
			 *     @type string $title  Title attribute.
			 *     @type string $target Target attribute.
			 *     @type string $rel    The rel attribute.
			 *     @type string $href   The href attribute.
			 * }
			 * @param object $item  The current menu item.
			 * @param array  $args  An array of {@see wp_nav_menu()} arguments.
			 * @param int    $depth Depth of menu item. Used for padding.
			 */
			$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 . '"';
				}
			}

	# BEGIN - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
			// $item_output = $args->before;
			// $item_output .= '<a'. $attributes .'>';
			// /** This filter is documented in wp-includes/post-template.php */
			// $item_output .= $args->link_before . apply_filters( 'the_title', $item->title, $item->ID ) . $args->link_after;
			// $item_output .= '</a>';
			// $item_output .= $args->after;
			$title = apply_filters('the_title', $item->title, $item->ID);
			$attributes = array_filter($atts);

			if( function_exists( 'fw_ext' ) && jevelin_framework() == 'unyson' ) :
				$item_output = fw_ext('megamenu')->render_str('item-link', compact('item', 'attributes', 'title', 'args', 'depth'));
			else :
				$item_output = jevelin_mega_menu_render_item( compact('item', 'attributes', 'title', 'args', 'depth') );
			endif;
	# END - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -

			/**
			 * Filter a menu item's starting output.
			 *
			 * The menu item's starting output only includes $args->before, the opening <a>,
			 * the menu item's title, the closing </a>, and $args->after. Currently, there is
			 * no filter for modifying the opening and closing <li> for a menu item.
			 *
			 * @since 3.0.0
			 *
			 * @see wp_nav_menu()
			 *
			 * @param string $item_output The menu item's starting HTML output.
			 * @param object $item        Menu item data object.
			 * @param int    $depth       Depth of menu item. Used for padding.
			 * @param array  $args        An array of wp_nav_menu() arguments.
			 */
			$output .= apply_filters( 'walker_nav_menu_start_el', $item_output, $item, $depth, $args );
		}


	}


	// replace default walker
	{
	    remove_filter('wp_nav_menu_args', '_filter_fw_ext_mega_menu_wp_nav_menu_args');

	    /** @internal */
	    function jevelin_filter_theme_ext_mega_menu_wp_nav_menu_args($args) {
	        $args['walker'] = new FW_Ext_Mega_Menu_Custom_Walker();

	        return $args;
	    }
	    add_filter('wp_nav_menu_args', 'jevelin_filter_theme_ext_mega_menu_wp_nav_menu_args');
	}

}
