<?php
/**
 * Extend native menu walker.
 *
 * @package acora
 * @since acora 1.0
 */

class WalkerMenu extends Walker_Nav_Menu{
	function display_element($element, &$children_elements, $max_depth, $depth, $args, &$output){

		if(!isset($depth)){
			$depth = 0;
		}

		if(!$element) {
			return;
		}
		global $post;
		$id_field = $this->db_fields['id'];
		// Adds the "has-children" class to the current item if it has children.
		if(!empty($children_elements[$element->$id_field])) {
			array_push($element->classes, 'has-children');
		}
		// Call the default function to do the rest.
		return parent::display_element($element, $children_elements, $max_depth, $depth, $args, $output);
	}
}