<?php
if ( ! defined( 'ABSPATH' ) ) {
	exit;
}

class Amber_Mega_Menu_Render {
	/** Avoid rendering fixed overlays and live cart controls inside page builders. */
	protected static function is_builder_preview() {
		// Divi visual/theme builder shortcode preview requests.
		if ( isset( $_REQUEST['et_fb'] ) || isset( $_REQUEST['et_bfb'] ) ) { // phpcs:ignore WordPress.Security.NonceVerification.Recommended
			return true;
		}
		if ( function_exists( 'et_fb_is_enabled' ) && et_fb_is_enabled() ) {
			return true;
		}
		if ( function_exists( 'et_core_is_fb_enabled' ) && et_core_is_fb_enabled() ) {
			return true;
		}
		if ( class_exists( 'ET_Builder_Element' ) && is_callable( array( 'ET_Builder_Element', 'is_fb_enabled' ) ) && ET_Builder_Element::is_fb_enabled() ) {
			return true;
		}
		if ( class_exists( '\\Elementor\\Plugin' ) && isset( \Elementor\Plugin::$instance ) ) {
			$elementor = \Elementor\Plugin::$instance;
			if ( ( isset( $elementor->editor ) && $elementor->editor->is_edit_mode() ) || ( isset( $elementor->preview ) && $elementor->preview->is_preview_mode() ) ) {
				return true;
			}
		}
		if ( class_exists( 'FLBuilderModel' ) && FLBuilderModel::is_builder_active() ) {
			return true;
		}
		if ( function_exists( 'vc_is_inline' ) && vc_is_inline() ) {
			return true;
		}
		if ( function_exists( 'bricks_is_builder' ) && bricks_is_builder() ) {
			return true;
		}
		if ( function_exists( 'bricks_is_builder_iframe' ) && bricks_is_builder_iframe() ) {
			return true;
		}
		if ( function_exists( '\\Breakdance\\Editor\\isEditor' ) && \Breakdance\Editor\isEditor() ) {
			return true;
		}
		if ( function_exists( 'themify_builder_is_frontend_editor' ) && themify_builder_is_frontend_editor() ) {
			return true;
		}
		if ( defined( 'SHOW_CT_BUILDER' ) && SHOW_CT_BUILDER ) {
			return true;
		}

		$preview_keys = array(
			'elementor-preview',
			'elementor_library',
			'ct_builder',
			'ct_inner',
			'oxygen_iframe',
			'brizy-edit-iframe',
			'brizy-edit',
			'fl_builder',
			'fl_builder_ui',
			'bricks',
			'brickspreview',
			'breakdance',
			'breakdance_iframe',
			'vc_editable',
			'vcv-action',
			'tve',
			'tve_editor',
			'tb-preview',
			'themify_builder_infinite_scroll',
		);
		foreach ( $preview_keys as $key ) {
			// Read-only detection of page-builder editor modes; no data is changed.
			if ( isset( $_REQUEST[ $key ] ) ) { // phpcs:ignore WordPress.Security.NonceVerification.Recommended
				return true;
			}
		}

		return (bool) apply_filters( 'amber_mega_menu_is_builder_preview', false );
	}

	public static function init() {
		add_shortcode( 'amber_mega_menu', array( __CLASS__, 'shortcode' ) );
		// Backward-compatible aliases for embeds created by earlier versions.
		add_shortcode( 'category_mega_menu', array( __CLASS__, 'shortcode' ) );
		add_shortcode( 'smm_mega_menu', array( __CLASS__, 'shortcode' ) );
		add_action( 'wp_enqueue_scripts', array( __CLASS__, 'register_assets' ) );
	}

	public static function register_assets() {
		wp_register_style( 'smm-frontend', AMBER_MEGA_MENU_URL . 'assets/frontend.css', array(), AMBER_MEGA_MENU_VERSION );
		wp_register_script( 'smm-frontend', AMBER_MEGA_MENU_URL . 'assets/frontend.js', array(), AMBER_MEGA_MENU_VERSION, true );
		// Visual builders can inject shortcode HTML after wp_head has already run.
		// Load the lightweight stylesheet up front so their live canvas never shows
		// an unstyled menu while a shortcode preview is being refreshed.
		wp_enqueue_style( 'smm-frontend' );
	}

	/** English defaults, optionally replaced per menu by the premium text module. */
	protected static function frontend_text( $settings ) {
		$text = array(
			'panel_title'       => __( 'Categories', 'amber-mega-menu' ),
			'search_label'      => __( 'Search categories', 'amber-mega-menu' ),
			'search_placeholder'=> __( 'Search categories…', 'amber-mega-menu' ),
			'close_label'       => __( 'Close categories', 'amber-mega-menu' ),
			'cart_label'        => __( 'View shopping cart', 'amber-mega-menu' ),
			'empty_message'     => __( 'There are currently no visible items in this menu.', 'amber-mega-menu' ),
			'no_results'        => __( 'No categories found.', 'amber-mega-menu' ),
			'show_all'          => __( 'Show all', 'amber-mega-menu' ),
			'show_less'         => __( 'Show less', 'amber-mega-menu' ),
		);
		/** Premium distributions may replace any built-in message per menu. */
		return apply_filters( 'amber_mega_menu_frontend_text', $text, $settings );
	}

	public static function shortcode( $atts ) {
		$atts = shortcode_atts(
			array(
				'id'      => 0,
				'layout'  => '',
				'display' => '', // adaptive | modal — overrides the saved desktop opening mode
				'trigger' => '',
				'columns' => '',
				'depth'   => '',
				'open'    => '', // "1" renders the panel always open (no trigger button)
				'nav'     => '', // "1" = panel only, opened by a WordPress nav menu item
				'disable_link' => '',
				'nav_for' => '', // key of the WordPress menu item link that opens this panel
				'nav_item_id' => 0,
			),
			$atts,
			'amber_mega_menu'
		);

		$post_id = (int) $atts['id'];
		if ( ! $post_id ) {
			$found = get_posts(
				array(
					'post_type'      => AMBER_MEGA_MENU_CPT,
					'posts_per_page' => 1,
					'fields'         => 'ids',
					'orderby'        => 'ID',
					'order'          => 'ASC',
				)
			);
			$post_id = $found ? (int) $found[0] : 0;
		}
		if ( ! $post_id || AMBER_MEGA_MENU_CPT !== get_post_type( $post_id ) ) {
			return '';
		}

		$s = Amber_Mega_Menu_Source::get_settings( $post_id );
		if ( $atts['layout'] ) {
			$s['layout'] = ( 'sideways' === $atts['layout'] ) ? 'sideways' : 'columns';
		}
		if ( $atts['display'] ) {
			$s['desktop_display'] = ( 'modal' === $atts['display'] ) ? 'modal' : 'adaptive';
		}
		if ( '' !== $atts['trigger'] ) {
			$s['trigger'] = $atts['trigger'];
		}
		if ( $atts['columns'] ) {
			$s['columns'] = max( 2, min( 6, (int) $atts['columns'] ) );
		}
		if ( '' !== (string) $atts['depth'] ) {
			// 0 = unlimited; anything higher is honoured up to a generous cap.
			$s['depth'] = max( 0, min( 20, (int) $atts['depth'] ) );
		}
		$always_open  = ( '1' === (string) $atts['open'] );
		$nav_mode     = ( '1' === (string) $atts['nav'] );
		$nav_no_link  = ( '1' === (string) $atts['disable_link'] );
		$nav_for      = sanitize_html_class( (string) $atts['nav_for'] );
		$nav_item_id  = absint( $atts['nav_item_id'] );
		if ( $nav_mode ) {
			$always_open = false;
		}

		/** Final settings for this placement; premium modules may extend them. */
		$s = apply_filters( 'smm_menu_settings', $s, $post_id, $atts ); // phpcs:ignore WordPress.NamingConventions.PrefixAllGlobals.NonPrefixedHooknameFound -- legacy 'smm_' prefix kept for compatibility between the free and premium distributions.

		$builder_preview = self::is_builder_preview();
		if ( $builder_preview ) {
			// Render the real configured panel in place. The builder preview must show
			// the selected source, layout, depth, tabs and visual settings—not a proxy.
			$always_open    = true;
			$nav_mode       = false;
			$s['show_cart'] = 0;
		}

		$tree       = self::visible( Amber_Mega_Menu_Source::merged_tree( $post_id, $s ), $s );
		$empty_tree = empty( $tree );

		wp_enqueue_style( 'smm-frontend' );
		wp_enqueue_script( 'smm-frontend' );

		/** Premium distributions enqueue their own assets for this placement. */
		do_action( 'amber_mega_menu_enqueue_assets', $s, $post_id );

		$uid  = 'smm-' . $post_id . '-' . wp_rand( 1000, 9999 );
		$text = self::frontend_text( $s );

		ob_start();
		?>
		<div class="smm smm--<?php echo esc_attr( $s['layout'] ); ?> smm--desktop-<?php echo esc_attr( $s['desktop_display'] ); ?><?php echo $always_open ? ' smm--static' : ''; ?><?php echo $builder_preview ? ' smm--builder-live-preview' : ''; ?><?php echo $nav_mode ? ' smm--nav' : ''; ?><?php echo ( ! $nav_mode && ! empty( $s['show_cart'] ) && 'before' === $s['cart_position'] ) ? ' smm--cart-before' : ''; ?>"
			data-smm data-layout="<?php echo esc_attr( $s['layout'] ); ?>" data-icon-only-breakpoint="<?php echo (int) $s['icon_only_breakpoint']; ?>" data-menu-name="<?php echo esc_attr( get_the_title( $post_id ) ); ?>" data-no-results="<?php echo esc_attr( $text['no_results'] ); ?>"<?php echo $nav_mode ? ' data-smm-nav="1" data-smm-nav-disable-link="' . ( $nav_no_link ? '1' : '0' ) . '"' . ( $nav_for ? ' data-smm-nav-for="' . esc_attr( $nav_for ) . '"' : '' ) . ( $nav_item_id ? ' data-smm-nav-item-id="' . (int) $nav_item_id . '"' : '' ) : ''; ?>
			style="--smm-panel-width:<?php echo (int) $s['width']; ?>px;--smm-columns:<?php echo (int) $s['columns']; ?>;--smm-blue:<?php echo esc_attr( $s['accent_color'] ); ?>;--smm-bg:<?php echo esc_attr( $s['panel_bg'] ); ?>;--smm-sideways-left-bg:<?php echo esc_attr( $s['sideways_left_bg'] ); ?>;--smm-text:<?php echo esc_attr( $s['font_color'] ); ?>;--smm-navy:<?php echo esc_attr( $s['font_color'] ); ?>;--smm-muted:<?php echo esc_attr( $s['muted_color'] ); ?>;--smm-line:<?php echo esc_attr( $s['border_color'] ); ?>;--smm-highlight:<?php echo esc_attr( $s['highlight_color'] ); ?>;--smm-font-family:<?php echo esc_attr( $s['font_family'] ); ?>;--smm-font-size:<?php echo (int) $s['font_size']; ?>px;--smm-parent-font-size:<?php echo (int) $s['parent_font_size']; ?>px;--smm-category-font-weight:<?php echo (int) $s['category_font_weight']; ?>;--smm-radius:<?php echo (int) $s['panel_radius']; ?>px;--smm-control-radius:<?php echo (int) $s['control_radius']; ?>px;--smm-image-size:<?php echo (int) $s['image_size']; ?>px;--smm-image-radius:<?php echo (int) $s['image_radius']; ?>px;--smm-button-bg:<?php echo esc_attr( $s['button_bg'] ); ?>;--smm-button-bg-hover:<?php echo esc_attr( $s['button_bg_hover'] ); ?>;--smm-button-text:<?php echo esc_attr( $s['button_text'] ); ?>;--smm-button-text-hover:<?php echo esc_attr( $s['button_text_hover'] ); ?>;--smm-button-border:<?php echo esc_attr( $s['button_border'] ); ?>;--smm-button-border-hover:<?php echo esc_attr( $s['button_border_hover'] ); ?>;--smm-button-border-width:<?php echo (int) $s['button_border_width']; ?>px;--smm-button-radius:<?php echo (int) $s['button_radius']; ?>px;--smm-button-font-size:<?php echo (int) $s['button_font_size']; ?>px;--smm-button-font-weight:<?php echo (int) $s['button_font_weight']; ?>;--smm-button-padding-y:<?php echo (int) $s['button_padding_y']; ?>px;--smm-button-padding-x:<?php echo (int) $s['button_padding_x']; ?>px;--smm-cart-bubble-bg:<?php echo esc_attr( $s['cart_bubble_bg'] ); ?>;--smm-cart-bubble-text:<?php echo esc_attr( $s['cart_bubble_text'] ); ?>;">
			<?php if ( ! $always_open && ! $nav_mode ) : ?>
				<button type="button" class="smm__trigger" aria-expanded="false" aria-controls="<?php echo esc_attr( $uid ); ?>">
					<span class="smm__burger" aria-hidden="true"><span></span><span></span><span></span></span>
					<span class="smm__trigger-text"><?php echo esc_html( $s['trigger'] ); ?></span>
				</button>
			<?php endif; ?>
			<?php
			if ( ! $nav_mode ) {
				/** Premium distributions may output extra trigger-area controls (e.g. a cart). */
				do_action( 'amber_mega_menu_after_trigger', $s, $text );
			}
			?>

			<div class="smm__panel" id="<?php echo esc_attr( $uid ); ?>"<?php echo $always_open ? '' : ' hidden'; ?>>
				<div class="smm__toolbar">
					<strong class="smm__toolbar-title"><?php echo esc_html( $text['panel_title'] ); ?></strong>
					<label class="smm__search">
						<span class="smm__search-icon" aria-hidden="true"></span>
						<span class="screen-reader-text"><?php echo esc_html( $text['search_label'] ); ?></span>
						<input type="search" placeholder="<?php echo esc_attr( $text['search_placeholder'] ); ?>" autocomplete="off">
					</label>
					<?php if ( ! $always_open ) : ?>
						<button type="button" class="smm__close" aria-label="<?php echo esc_attr( $text['close_label'] ); ?>">&times;</button>
					<?php endif; ?>
				</div>
				<?php if ( $empty_tree ) : ?>
					<div class="smm__source-empty" role="status">
						<?php echo esc_html( $text['empty_message'] ); ?>
					</div>
				<?php endif; ?>
				<?php
				$panes = array(
					array(
						'label' => '' !== trim( (string) $s['tab_label'] ) ? $s['tab_label'] : $text['panel_title'],
						'tree'  => $tree,
					),
				);
				/** Premium distributions may add further panes (e.g. extra menus as tabs). */
				$panes = apply_filters( 'amber_mega_menu_panes', $panes, $s, $post_id );
				do_action( 'smm_panel_before', $s, $post_id ); // phpcs:ignore WordPress.NamingConventions.PrefixAllGlobals.NonPrefixedHooknameFound -- legacy 'smm_' prefix kept for compatibility between the free and premium distributions.

				if ( count( $panes ) > 1 ) :
					?>
					<div class="smm__tabs" role="tablist">
						<?php foreach ( $panes as $i => $pane ) : ?>
							<button type="button" class="smm__tab<?php echo 0 === $i ? ' is-active' : ''; ?>" role="tab" data-tab="<?php echo (int) $i; ?>" aria-selected="<?php echo 0 === $i ? 'true' : 'false'; ?>"><?php echo esc_html( $pane['label'] ); ?></button>
						<?php endforeach; ?>
					</div>
					<?php
				endif;

				foreach ( $panes as $i => $pane ) {
					echo '<div class="smm__pane' . ( 0 === $i ? ' is-active' : '' ) . '" data-smm-pane="' . (int) $i . '"' . ( 0 === $i ? '' : ' hidden' ) . '>';
					if ( 'sideways' === $s['layout'] ) {
						self::render_sideways( $pane['tree'], $s );
					} else {
					self::render_columns( $pane['tree'], $s, $text );
					}
					echo '</div>';
				}
				do_action( 'smm_panel_after', $s, $post_id ); // phpcs:ignore WordPress.NamingConventions.PrefixAllGlobals.NonPrefixedHooknameFound -- legacy 'smm_' prefix kept for compatibility between the free and premium distributions.
				?>
				<?php if ( $s['all_link'] ) : ?>
					<div class="smm__footer">
						<a class="smm__all" href="<?php echo esc_url( $s['all_link'] ); ?>"><?php echo esc_html( $s['all_label'] ); ?> <span aria-hidden="true">&rarr;</span></a>
					</div>
				<?php endif; ?>
			</div>
		</div>
		<?php
		return ob_get_clean();
	}

	public static function visible( $nodes, $s = null ) {
		$hide_empty = is_array( $s ) && ! empty( $s['hide_empty'] ) && 'menu' !== ( isset( $s['source'] ) ? $s['source'] : 'taxonomy' );
		$out        = array();
		foreach ( $nodes as $node ) {
			$children = self::visible( $node['children'], $s );
			if ( ! empty( $node['hidden'] ) ) {
				// 'self' keeps the sub-items and promotes them one level up.
				if ( 'self' === ( isset( $node['hide_mode'] ) ? $node['hide_mode'] : 'branch' ) && $children ) {
					foreach ( $children as $child ) {
						$out[] = $child;
					}
				}
				continue;
			}
			$node['children'] = $children;
			// Drop empty leaf categories; keep parents that still have visible children.
			if ( $hide_empty && empty( $node['count'] ) && empty( $node['children'] ) ) {
				continue;
			}
			$out[] = $node;
		}
		return $out;
	}

	protected static function count_label( $node, $s ) {
		if ( empty( $s['show_counts'] ) || empty( $node['count'] ) ) {
			return '';
		}
		return '<span class="smm__count">' . (int) $node['count'] . '</span>';
	}

	/** Layout 1 — grid of parent columns. */
	protected static function render_columns( $tree, $s, $text ) {
		echo '<div class="smm__grid">';
		foreach ( $tree as $parent ) {
			echo '<section class="smm__col" data-smm-group>';
			echo '<a class="smm__parent" href="' . esc_url( $parent['url'] ) . '">';
			if ( ! empty( $s['show_images'] ) && $parent['image'] ) {
				echo '<img class="smm__icon" src="' . esc_url( $parent['image'] ) . '" alt="" loading="lazy">';
			}
			echo esc_html( $parent['label'] ) . wp_kses_post( self::count_label( $parent, $s ) ) . '</a>';

			if ( self::depth_allows( $s, 1 ) && ! empty( $parent['children'] ) ) {
				echo '<ul class="smm__children">';
				foreach ( $parent['children'] as $child_index => $child ) {
					$extra_class = $child_index >= 6 ? ' smm__extra' : '';
					echo '<li class="smm__child' . esc_attr( $extra_class ) . '" data-smm-search-item><a href="' . esc_url( $child['url'] ) . '">' . esc_html( $child['label'] ) . wp_kses_post( self::count_label( $child, $s ) ) . '</a>';
					self::render_nested( $child, $s, 2 );
					echo '</li>';
				}
				echo '</ul>';
				if ( count( $parent['children'] ) > 6 ) {
					echo '<button type="button" class="smm__more" aria-expanded="false"><span class="smm__more-open">' . esc_html( $text['show_all'] ) . ' (' . (int) count( $parent['children'] ) . ')</span><span class="smm__more-close">' . esc_html( $text['show_less'] ) . '</span></button>';
				}
			}
			echo '</section>';
		}
		echo '</div>';
	}

	/** Layout 2 — left rail of parents, hover/focus reveals the sub-level panel. */
	protected static function render_sideways( $tree, $s ) {
		echo '<div class="smm__side">';

		echo '<ul class="smm__rail">';
		foreach ( $tree as $i => $parent ) {
			$has = ( self::depth_allows( $s, 1 ) && ! empty( $parent['children'] ) );
			echo '<li class="smm__rail-item' . ( $has ? ' has-children' : '' ) . ( 0 === $i ? ' is-active' : '' ) . '" data-index="' . (int) $i . '" data-smm-group>';
			echo '<a href="' . esc_url( $parent['url'] ) . '" class="smm__rail-link"' . ( $has ? ' aria-expanded="false"' : '' ) . '>';
			if ( ! empty( $s['show_images'] ) && $parent['image'] ) {
				echo '<img class="smm__icon" src="' . esc_url( $parent['image'] ) . '" alt="" loading="lazy">';
			}
			echo '<span>' . esc_html( $parent['label'] ) . '</span>' . wp_kses_post( self::count_label( $parent, $s ) );
			if ( $has ) {
				echo '<span class="smm__chev" aria-hidden="true">&rsaquo;</span>';
			}
			echo '</a>';

			// Mobile accordion copy of the sub level.
			if ( $has ) {
				echo '<div class="smm__mobile-sub">';
				self::render_sub_lists( $parent, $s );
				echo '</div>';
			}
			echo '</li>';
		}
		echo '</ul>';

		echo '<div class="smm__flyouts">';
		foreach ( $tree as $i => $parent ) {
			$has = ( self::depth_allows( $s, 1 ) && ! empty( $parent['children'] ) );
			if ( ! $has ) {
				continue;
			}
			echo '<div class="smm__flyout' . ( 0 === $i ? ' is-active' : '' ) . '" data-index="' . (int) $i . '" data-smm-flyout>';
			echo '<div class="smm__flyout-head"><a href="' . esc_url( $parent['url'] ) . '">' . esc_html( $parent['label'] ) . ' <span aria-hidden="true">&rarr;</span></a></div>';
			self::render_sub_lists( $parent, $s );
			echo '</div>';
		}
		echo '</div>';

		echo '</div>';
	}

	protected static function render_sub_lists( $parent, $s ) {
		echo '<div class="smm__sub-grid">';
		foreach ( $parent['children'] as $child ) {
			echo '<div class="smm__sub-col" data-smm-search-item>';
			echo '<a class="smm__sub-title" href="' . esc_url( $child['url'] ) . '">' . esc_html( $child['label'] ) . wp_kses_post( self::count_label( $child, $s ) ) . '</a>';
			self::render_nested( $child, $s, 2 );
			echo '</div>';
		}
		echo '</div>';
	}

	/** Whether the configured depth shows $level (1 = children, 2 = grandchildren, ...). 0 means unlimited. */
	protected static function depth_allows( $s, $level ) {
		$d = (int) $s['depth'];
		return 0 === $d || $d > $level;
	}

	/**
	 * Levels below children, rendered recursively to any depth. Deeper levels
	 * reuse the grandchildren list styles, which nest (indent + left border).
	 */
	protected static function render_nested( $node, $s, $level ) {
		if ( ! self::depth_allows( $s, $level ) || empty( $node['children'] ) ) {
			return;
		}
		echo '<ul class="smm__grandchildren">';
		foreach ( $node['children'] as $child ) {
			echo '<li><a href="' . esc_url( $child['url'] ) . '">' . esc_html( $child['label'] ) . wp_kses_post( self::count_label( $child, $s ) ) . '</a>';
			self::render_nested( $child, $s, $level + 1 );
			echo '</li>';
		}
		echo '</ul>';
	}
}