<?php

namespace QuadLayers\WPMI;

class Menu_Item_Custom_Fields_Walker extends \Walker_Nav_Menu_Edit {

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

		parent::start_el( $item_output, $item, $depth, $args, $id );

		$item_output = preg_replace(
				// NOTE: Check this regex from time to time!
			'/(?=<(div)[^>]+class="[^"]*menu-item-handle)/',
			$this->title( $item, $depth, $args ),
			$item_output
		);

		$output .= preg_replace(
				// NOTE: Check this regex from time to time!
			'/(?=<(fieldset|p)[^>]+class="[^"]*field-move)/',
			$this->fields( $item, $depth, $args ),
			$item_output
		);
	}

	protected function title( $item, $depth, $args = array(), $id = 0 ) {
		ob_start();

		do_action( 'wp_nav_menu_item_custom_title', $item->ID, $item, $depth, $args, $id );

		return ob_get_clean();
	}

	protected function fields( $item, $depth, $args = array(), $id = 0 ) {
		ob_start();

		do_action( 'wp_nav_menu_item_custom_fields', $item->ID, $item, $depth, $args, $id );

		return ob_get_clean();
	}
}
