<?php
/**
 * Translate WP menus.
 *
 * @category Class
 * @package  WPM/Includes
 * @author   Valentyn Riaboshtan
 */

namespace WPM\Includes;

if ( ! defined( 'ABSPATH' ) ) {
	exit;
}

/**
 * WPM_Menus Class.
 */
class WPM_Menus {

	/**
	 * WPM_Menus constructor.
	 */
	public function __construct() {
		add_filter( 'wp_setup_nav_menu_item', array( $this, 'translate_menu_item' ), ( 'POST' === $_SERVER['REQUEST_METHOD'] ? 99 : 5 ) );
		add_filter( 'wp_setup_nav_menu_item', array( $this, 'translate_menu_url' ) );
		add_filter( 'customize_nav_menu_available_items', array( $this, 'filter_menus' ), 5 );
		add_filter( 'customize_nav_menu_searched_items', array( $this, 'filter_menus' ), 5 );
		add_filter( 'wp_nav_menu_objects', array( $this, 'add_languages_to_menu' ), 5 );
	}

	/**
	 * Translate title in menu item
	 *
	 * @param $items
	 *
	 * @return mixed
	 */
	public function filter_menus( $items ) {
		foreach ( (array) $items as &$item ) {
			$item['title'] = wpm_translate_string( $item['title'] );
		}

		return $items;
	}


	/**
	 * Translate menu item
	 *
	 * @param $menu_item
	 *
	 * @return mixed
	 */
	public function translate_menu_item( $menu_item ) {
		$menu_item = wpm_translate_object( $menu_item, wpm_get_language() );

		if ( isset( $menu_item->post_type ) ) {
			if ( 'nav_menu_item' === $menu_item->post_type ) {

				if ( 'post_type' === $menu_item->type ) {

					$object = get_post_type_object( $menu_item->object );

					if ( $object ) {
						$menu_item->type_label = $object->labels->singular_name;
					} else {
						$menu_item->type_label = $menu_item->object;
					}

					$original_object = get_post( $menu_item->object_id );

					/** This filter is documented in wp-includes/post-template.php */
					$original_title = apply_filters( 'the_title', $original_object->post_title, $original_object->ID );

					if ( '' === $original_title ) {
						/* translators: %d: ID of a post */
						$original_title = sprintf( __( '#%d (no title)' ), $original_object->ID );
					}

					$menu_item->title = '' === $menu_item->post_title ? $original_title : $menu_item->post_title;

				} elseif ( 'post_type_archive' === $menu_item->type ) {

					$object = get_post_type_object( $menu_item->object );

					if ( $object ) {
						$menu_item->title = '' === $menu_item->post_title ? $object->labels->archives : $menu_item->post_title;
					} else {
						$menu_item->_invalid = true;
					}

					$menu_item->type_label = __( 'Post Type Archive' );
					$menu_item->url        = get_post_type_archive_link( $menu_item->object );

				} elseif ( 'taxonomy' === $menu_item->type ) {

					$object = get_taxonomy( $menu_item->object );

					if ( $object ) {
						$menu_item->type_label = $object->labels->singular_name;
					} else {
						$menu_item->type_label = $menu_item->object;
					}

					$original_title = get_term_field( 'name', $menu_item->object_id, $menu_item->object, 'raw' );

					if ( is_wp_error( $original_title ) ) {
						$original_title = false;
					}

					$menu_item->title = '' === $menu_item->post_title ? $original_title : $menu_item->post_title;

				} else {

					$menu_item->type_label = __( 'Custom Link' );
					$menu_item->title      = $menu_item->post_title;
				}// End if().

				$menu_item->attr_title = ! isset( $menu_item->attr_title ) ? apply_filters( 'nav_menu_attr_title', $menu_item->post_excerpt ) : $menu_item->attr_title;
				if ( ! isset( $menu_item->description ) ) {
					$menu_item->description = apply_filters( 'nav_menu_description', wp_trim_words( $menu_item->post_content, 200 ) );
				}
			} else {

				$object                = get_post_type_object( $menu_item->post_type );
				$menu_item->object     = $object->name;
				$menu_item->type_label = $object->labels->singular_name;

				if ( '' === $menu_item->post_title ) {
					/* translators: %d: ID of a post */
					$menu_item->post_title = sprintf( __( '#%d (no title)' ), $menu_item->ID );
				}

				$menu_item->title = $menu_item->post_title;

				/** This filter is documented in wp-includes/nav-menu.php */
				$menu_item->attr_title = apply_filters( 'nav_menu_attr_title', '' );

				/** This filter is documented in wp-includes/nav-menu.php */
				$menu_item->description = apply_filters( 'nav_menu_description', '' );
			}// End if().
		} elseif ( isset( $menu_item->taxonomy ) ) {

			$object                = get_taxonomy( $menu_item->taxonomy );
			$menu_item->type_label = $object->labels->singular_name;

			$menu_item->title       = $menu_item->name;
			$menu_item->attr_title  = '';
			$menu_item->description = get_term_field( 'description', $menu_item->term_id, $menu_item->taxonomy );

		}// End if().

		$menu_item->title = '' === $menu_item->title ? sprintf( __( '#%d (no title)' ), $menu_item->ID ) : $menu_item->title;

		return $menu_item;
	}


	/**
	 * Translation custom menu link
	 *
	 * @param $menu_item
	 *
	 * @return mixed
	 */
	public function translate_menu_url( $menu_item ) {

		if ( 'custom' === $menu_item->object && ! is_admin() ) {
			$menu_item->url = wpm_translate_url( $menu_item->url );
		}

		return $menu_item;
	}

	/**
	 * Add languages to menu
	 *
	 * @param $items
	 *
	 * @return string
	 */
	public function add_languages_to_menu( $items ) {

		foreach ( $items as $key => $item ) {

			if ( '#wpm-languages' === $item->url ) {

				$languages       = wpm_get_languages();
				$count_languages = count( $languages );
				$lang            = wpm_get_language();
				$new_items       = array();
				$show_type       = get_post_meta( $item->ID, '_menu_item_languages_show', true );
				$item_type       = get_post_meta( $item->ID, '_menu_item_languages_type', true );
				$i               = 1;
				$was_current     = false;
				$first_lang      = null;

				foreach ( $languages as $code => $language ) {
					$new_item                      = new \stdClass();
					$new_item->ID                  = 'language-' . $code;
					$new_item->db_id               = 'language-' . $code;
					$new_item->classes             = $item->classes;
					$new_item->type                = 'language';
					$new_item->menu_item_parent    = $item->menu_item_parent;
					$new_item->parent_submenu_type = $item->parent_submenu_type;
					$new_item->xfn                 = '';
					$new_item->target              = '';
					$language_title                = '';

					if ( $lang === $code ) {
						$was_current = true;
						$new_item->classes[] = 'active-language';
						$new_item->url       = '#';
                        $new_item->current   = true;
					} else {
						$new_item->url = esc_url( wpm_translate_current_url( $code ) );
                        $new_item->current   = false;
					}

					if ( ( ( 'flag' === $show_type ) || ( 'both' === $show_type ) ) && ( $language['flag'] ) ) {
						$language_title = '<img src="' . esc_url( wpm_get_flag_url( $language['flag'] ) ) . '" alt="' . esc_attr( $language['name'] ) . '">';
					}

					if ( ( 'name' === $show_type ) || ( 'both' === $show_type ) ) {
						$language_title .= '<span>' . esc_html( $language['name'] ) . '</span>';
					}

					$new_item->title = $language_title;

					if ( 'dropdown' === $item_type ) {
						if ( $lang === $code ) {
							$new_item->classes[] = 'menu-item-has-children';
						} else {
							$new_item->menu_item_parent = 'language-' . $lang;
						}
					}

					if ( 'single' === $item_type ) {
						if ( 1 === $i ) {
							$first_lang = $new_item;
						}

						if ( ( $i === $count_languages ) && ( $lang === $code ) && $first_lang ) {
							$new_items[] = $first_lang;
							break;
						}

						if ( $lang !== $code && $was_current ) {
							$new_items[] = $new_item;
							break;
						}

						$i++;
						continue;
					}

					$new_items[] = $new_item;
				}// End foreach().

				$items        = wpm_array_insert_after( $items, $key, $new_items );
				$sorted_array = array();

				foreach ( $items as $k => $v ) {
					$order                  = $k + 1;
					$v->menu_order          = $order;
					$sorted_array[ $order ] = $v;
				}

				$items = $sorted_array;
				unset( $items[ $key ] );
				break;
			}// End if().
		}// End foreach().

		return $items;
	}
}