<?php
namespace ABlocks\Blocks\TableOfContent;

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

use ABlocks\Classes\BlockBaseAbstract;
use ABlocks\Classes\CssGenerator;
use ABlocks\Classes\CssGeneratorV2;
use ABlocks\Helper;
use ABlocks\Controls\Typography;
use ABlocks\Controls\Background;
use ABlocks\Controls\Border;
use ABlocks\Controls\Dimensions;
use ABlocks\Controls\Alignment;
use ABlocks\Controls\Range;
use ABlocks\Controls\Color;
use ABlocks\Controls\BoxShadow;

class Block extends BlockBaseAbstract {
	protected $block_name = 'table-of-content';


	public function build_css_v1( $attributes ) {
		$css_generator = new CssGenerator( $attributes );

		$css_generator->add_class_styles(
			'{{WRAPPER}}  .ablocks-toc__header-title',
			$this->get_toc_title_css( $attributes ),
			$this->get_toc_title_css( $attributes, 'Tablet' ),
			$this->get_toc_title_css( $attributes, 'Mobile' ),
			$css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_toc_title_css( $attributes, $device ); } )
		);
		$css_generator->add_class_styles(
			'{{WRAPPER}}  .ablocks-toc__header',
			$this->get_toc_header_css( $attributes ),
			$this->get_toc_header_css( $attributes, 'Tablet' ),
			$this->get_toc_header_css( $attributes, 'Mobile' ),
			$css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_toc_header_css( $attributes, $device ); } )
		);
		$css_generator->add_class_styles(
			'{{WRAPPER}}  .ablocks-toc-body',
			$this->get_toc_body_css( $attributes ),
			$this->get_toc_body_css( $attributes, 'Tablet' ),
			$this->get_toc_body_css( $attributes, 'Mobile' ),
			$css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_toc_body_css( $attributes, $device ); } )
		);
		$css_generator->add_class_styles(
			'{{WRAPPER}}  .ablocks-toc__header-toggle-icon .ablocks-toc__show',
			$this->get_toc_header_icon_css( $attributes ),
			$this->get_toc_header_icon_css( $attributes, 'Tablet' ),
			$this->get_toc_header_icon_css( $attributes, 'Mobile' ),
			$css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_toc_header_icon_css( $attributes, $device ); } )
		);
		$css_generator->add_class_styles(
			'{{WRAPPER}}  .ablocks-toc__header-toggle-icon .ablocks-toc__show:hover',
			$this->get_toc_header_icon_hover_css( $attributes ),
			$this->get_toc_header_icon_hover_css( $attributes, 'Tablet' ),
			$this->get_toc_header_icon_hover_css( $attributes, 'Mobile' ),
			$css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_toc_header_icon_hover_css( $attributes, $device ); } )
		);
		$css_generator->add_class_styles(
			'{{WRAPPER}}  .ablocks-block-container .ablocks-toc-list,
		    {{WRAPPER}}  .ablocks-block-container .ablocks-toc-list li a',
			$this->get_list_item_css( $attributes ),
			$this->get_list_item_css( $attributes, 'Tablet' ),
			$this->get_list_item_css( $attributes, 'Mobile' ),
			$css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_list_item_css( $attributes, $device ); } )
		);

		return $css_generator->generate_css();
	}
	public function build_css_v2( $attributes ) {
		$css_generator = new CssGeneratorV2( $attributes, $this->block_name );

		$css_generator->add_class_styles(
			'{{WRAPPER}}  .ablocks-toc__header-title',
			$this->get_toc_title_css( $attributes ),
			$this->get_toc_title_css( $attributes, 'Tablet' ),
			$this->get_toc_title_css( $attributes, 'Mobile' ),
			$css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_toc_title_css( $attributes, $device ); } )
		);
		$css_generator->add_class_styles(
			'{{WRAPPER}}  .ablocks-toc__header',
			$this->get_toc_header_css( $attributes ),
			$this->get_toc_header_css( $attributes, 'Tablet' ),
			$this->get_toc_header_css( $attributes, 'Mobile' ),
			$css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_toc_header_css( $attributes, $device ); } )
		);
		$css_generator->add_class_styles(
			'{{WRAPPER}}  .ablocks-toc-body',
			$this->get_toc_body_css( $attributes ),
			$this->get_toc_body_css( $attributes, 'Tablet' ),
			$this->get_toc_body_css( $attributes, 'Mobile' ),
			$css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_toc_body_css( $attributes, $device ); } )
		);
		$css_generator->add_class_styles(
			'{{WRAPPER}}  .ablocks-toc__header-toggle-icon .ablocks-toc__show',
			$this->get_toc_header_icon_css( $attributes ),
			$this->get_toc_header_icon_css( $attributes, 'Tablet' ),
			$this->get_toc_header_icon_css( $attributes, 'Mobile' ),
			$css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_toc_header_icon_css( $attributes, $device ); } )
		);
		$css_generator->add_class_styles(
			'{{WRAPPER}}  .ablocks-toc__header-toggle-icon .ablocks-toc__show:hover',
			$this->get_toc_header_icon_hover_css( $attributes ),
			$this->get_toc_header_icon_hover_css( $attributes, 'Tablet' ),
			$this->get_toc_header_icon_hover_css( $attributes, 'Mobile' ),
			$css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_toc_header_icon_hover_css( $attributes, $device ); } )
		);
		$css_generator->add_class_styles(
			'{{WRAPPER}}  .ablocks-toc-body .ablocks-toc-list',
			$this->get_marker_list_style_css( $attributes ),
			$this->get_marker_list_style_css( $attributes, 'Tablet' ),
			$this->get_marker_list_style_css( $attributes, 'Mobile' ),
			$css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_marker_list_style_css( $attributes, $device ); } )
		);
		$css_generator->add_class_styles(
			'{{WRAPPER}} .ablocks-toc-list,
		    {{WRAPPER}} .ablocks-toc-list li a',
			$this->get_list_item_css( $attributes ),
			$this->get_list_item_css( $attributes, 'Tablet' ),
			$this->get_list_item_css( $attributes, 'Mobile' ),
			$css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_list_item_css( $attributes, $device ); } )
		);
		$css_generator->add_class_styles(
			'{{WRAPPER}} .ablocks-toc-body .ablocks-toc-list li a',
			$this->get_list_item_gap_css( $attributes ),
			$this->get_list_item_gap_css( $attributes, 'Tablet' ),
			$this->get_list_item_gap_css( $attributes, 'Mobile' ),
			$css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_list_item_gap_css( $attributes, $device ); } )
		);
		$css_generator->add_class_styles(
			'{{WRAPPER}} a.ablocks-toc-item-link.active',
			$this->get_active_list_item_css( $attributes ),
			$this->get_active_list_item_css( $attributes, 'Tablet' ),
			$this->get_active_list_item_css( $attributes, 'Mobile' ),
			$css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_active_list_item_css( $attributes, $device ); } )
		);

		return $css_generator->generate_css();
	}

	public function build_css( $attributes ) {
		if ( isset( $attributes['blockVersion'] ) && (int) $attributes['blockVersion'] === 2 ) {
			return $this->build_css_v2( $attributes );
		}
		return $this->build_css_v1( $attributes );
	}


	public function get_toc_title_css( $attributes, $device = '' ) {
		$typographyValueGlobal = ! empty( $attributes['titleTypographyGlobal'] ) ? $attributes['titleTypographyGlobal'] : array();
		$toc_title_typography_css = ! empty( $attributes['titleTypography'] ) ? Typography::get_css( $attributes['titleTypography'], '', $device, $typographyValueGlobal ) : array();
		return array_merge( $toc_title_typography_css,
		[ 'color' => Color::get_css( isset( $attributes['titleColor'] ) ? $attributes['titleColor'] : '' ) ], );
	}
	public function get_toc_header_css( $attributes, $device = '' ) {
		$headerorder = ! empty( $attributes['headerBorder'] ) ? Border::get_css( $attributes['headerBorder'], '', $device ) : array();
		$header_padding_css = ! empty( $attributes['header_padding'] ) ? Dimensions::get_css( $attributes['header_padding'], 'padding', $device ) : array();
		return array_merge(
			[ 'background' => Color::get_css( isset( $attributes['headerBG'] ) ? $attributes['headerBG'] : '' ) ],
		$headerorder, $header_padding_css );
	}
	public function get_toc_body_css( $attributes, $device = '' ) {
		$listing_padding_css = ! empty( $attributes['list_padding'] ) ? Dimensions::get_css( $attributes['list_padding'], 'padding', $device ) : array();
		return array_merge(
			[ 'background' => Color::get_css( isset( $attributes['bodyBG'] ) ? $attributes['bodyBG'] : '' ) ],
		$listing_padding_css, );
	}
	public function get_toc_header_icon_css( $attributes, $device = '' ) {
		$css = array_merge(
			[ 'fill' => Color::get_css( isset( $attributes['iconColor'] ) ? $attributes['iconColor'] : '' ) ],
			Range::get_css([
				'attributeValue' => $attributes['iconSize'] ?? null,
				'isResponsive' => false,
				'defaultValue' => 20,
				'unitDefaultValue' => 'px',
				'property' => 'font-size',
				'device' => $device,
			]),
			Dimensions::get_css( $attributes['icon_padding'] ?? [], 'padding', $device ),
			Border::get_css( $attributes['iconBorder'] ?? [], '', $device ),
			BoxShadow::get_css( $attributes['iconBoxShadow'] ?? [], $device ),
		);
		return $css;
	}

	public function get_toc_header_icon_hover_css( $attributes, $device = '' ) {
		$css = array_merge(
			Border::get_hover_css( $attributes['iconBorder'], $device ),
			BoxShadow::get_hover_css( $attributes['iconBoxShadow'], $device )
		);
		return $css;
	}

	public function get_list_item_css( $attributes, $device = '' ) {
		$typographyValueGlobal = ! empty( $attributes['contentTypographyGlobal'] ) ? $attributes['contentTypographyGlobal'] : array();
		$contentTypography = ! empty( $attributes['contentTypography'] ) ? Typography::get_css( $attributes['contentTypography'], '', $device, $typographyValueGlobal ) : array();
		return array_merge(
			[ 'color' => Color::get_css( isset( $attributes['itemColor'] ) ? $attributes['itemColor'] : '' ) ],
			$contentTypography,
		);
	}

	public function get_list_item_gap_css( $attributes, $device = '' ) {
		return array_merge(
			Range::get_css([
				'attributeValue' => $attributes['listItemGap'],
				'attribute_object_key' => 'value',
				'isResponsive' => false,
				'defaultValue' => 30,
				'unitDefaultValue' => 'px',
				'property' => 'line-height',
				'hasUnit' => true,
				'device' => $device,
			]),
		);
	}
	public function get_active_list_item_css( $attributes, $device = '' ) {
		return array_merge(
			[ 'color' => Color::get_css( isset( $attributes['activeColor'] ) ? $attributes['activeColor'] : '' ) ],
		);
	}


	public function get_marker_list_style_css( $attributes ) {
		$css = [];
		$allowed_list_types = [
			'decimal',
			'disc',
			'circle',
			'square',
			'lower-alpha',
			'lower-roman',
			'none'
		];
		if ( ! empty( $attributes['markerView'] ) && in_array( $attributes['markerView'], $allowed_list_types, true ) ) {
			$css['list-style-type'] = $attributes['markerView'];
		}
		return $css;
	}


	public function add_toc_to_post_content( $content ) {

		$content = preg_replace_callback('/<h([1-6])(.*?)>(.*?)<\/h\1>/i', function ( $matches ) {
			$level = $matches[1];  // Heading level (1-6)
			$attributes = $matches[2];  // Existing attributes (classes, styles, etc.)
			$heading = $matches[3];  // Heading text/content
			$anchor = sanitize_title( $heading );  // Generate a sanitized ID
			// Check if the id attribute is already present
			if ( strpos( $attributes, 'id=' ) === false ) {
				// Insert the id attribute after the opening <h> tag, preserving existing attributes
				return '<h' . $level . $attributes . ' id="' . $anchor . '">' . $heading . '</h' . $level . '>';
			}
			return $matches[0];  // If id is already present, return the original match
		}, $content);
		return $content;
	}


	public function render_block_content( $attributes, $content, $block_instance ) {
		$post = get_post();
		if ( ! $post ) {
			return '';
		}

		add_filter( 'the_content', [ $this, 'add_toc_to_post_content' ] );

		// Sanitize and escape icon attributes
		$open_icon_attributes = array(
			'path'      => esc_attr( $attributes['openIconSvgPath'] ),
			'viewBox'   => esc_attr( $attributes['openIconSvgViewBox'] ),
			'className' => sanitize_text_field( $attributes['openIconClass'] ),
			'width'     => '20',
			'height'    => '20',
		);
		$close_icon_attributes = array(
			'path'      => esc_attr( $attributes['closeIconSvgPath'] ),
			'viewBox'   => esc_attr( $attributes['closeIconSvgViewBox'] ),
			'className' => sanitize_text_field( $attributes['closeIconClass'] ),
			'width'     => '20',
			'height'    => '20',
		);

		$post_content = $post->post_content;
		preg_match_all( '/<h([1-6])[^>]*>(.*?)<\/h\1>/', $post_content, $matches, PREG_SET_ORDER );
		$toc = '';

		$headings = [];
		$unique_anchors = [];

		// Process headings
		foreach ( $matches as $match ) {
			$level = intval( $match[1] );
			$heading_html = trim( $match[2] );
			$heading = esc_html( wp_strip_all_tags( $heading_html ) );

			$base_anchor = strtolower( sanitize_title( $heading ) );
			$anchor = $base_anchor;
			$count = 1;

			while ( in_array( $anchor, $unique_anchors, true ) ) {
				$anchor = $base_anchor . '-' . $count;
				$count++;
			}

			if ( (
			( $level === 1 && $attributes['H1'] ) ||
			( $level === 2 && $attributes['H2'] ) ||
			( $level === 3 && $attributes['H3'] ) ||
			( $level === 4 && $attributes['H4'] ) ||
			( $level === 5 && $attributes['H5'] ) ||
			( $level === 6 && $attributes['H6'] ) ) ) {

				$headings[] = [
					'level'   => $level,
					'heading' => $heading,
					'anchor'  => esc_attr( $anchor ),
				];
				$unique_anchors[] = $anchor;
			}
		}//end foreach

		$toc_list = $this->generate_toc_list( $attributes, $headings );
		$has_toc  = $toc_list !== '';

		// Build TOC header
		if ( (bool) $attributes['hideTitle'] === true ) :
			$toc .= '<div class="ablocks-toc__header">';
			$toc .= $has_toc ? '<span class="ablocks-toc__header-title">' . esc_html( $attributes['tocTableTitle'] ) . '</span>' : '';
			if ( $attributes['collapSible'] ) :
				$toc .= '<div class="ablocks-toc__header-toggle-icon">';
				$toc .= '<span class="ablocks-toc__show">' . Helper::render_svg_icon_using_attr( $close_icon_attributes ) . '</span>';
				$toc .= '<span class="ablocks-toc__hide">' . Helper::render_svg_icon_using_attr( $open_icon_attributes ) . '</span>';
				$toc .= '</div>';
			endif;
			$toc .= '</div>';
		endif;

		// Build TOC body
		$toc .= '<div class="ablocks-toc-body">';
		$toc .= $toc_list;
		$toc .= '</div>';

		return $toc;
	}

	private function generate_toc_list( $attributes, $headings ) {
		if ( empty( $headings ) ) {
			return '';
		}

		$toc = '';
		$marker_view = 'ol';
		$current_level = 0;
		$open_lists = [];
		foreach ( $headings as $index => $heading ) {
			if ( ! isset( $heading['level'], $heading['heading'], $heading['anchor'] ) ) {
				continue;
			}

			$level = (int) $heading['level'];

			// If the first item, open the root list
			if ( $index === 0 ) {
				$toc .= '<' . esc_attr( $marker_view ) . ' class="ablocks-toc-list">';
				$open_lists[] = $marker_view;
				$current_level = $level;
			}

			// If deeper heading level, open a nested list
			while ( $level > $current_level ) {
				$toc .= '<' . esc_attr( $marker_view ) . ' class="ablocks-toc-list">';
				$open_lists[] = $marker_view;
				$current_level++;
			}

			// If shallower heading level, close open lists
			while ( $level < $current_level ) {
				$toc .= '</' . esc_attr( array_pop( $open_lists ) ) . '>';
				$current_level--;
			}

			// Close previous <li> before adding a new one (except for the first)
			if ( $index > 0 ) {
				$toc .= '</li>';
			}

			// Add list item
			$toc .= '<li class="ablocks-toc-item">';
			$toc .= '<a class="ablocks-toc-item-link" href="#' . esc_attr( $heading['anchor'] ) . '">' . esc_html( $heading['heading'] ) . '</a>';
		}//end foreach

		// Close any remaining open lists
		while ( ! empty( $open_lists ) ) {
			$toc .= '</li></' . esc_attr( array_pop( $open_lists ) ) . '>';
		}

		return $toc;
	}
}