<?php
/**
 * Core class: applies block restrictions on the front-end editor.
 *
 * @package BlockCurator
 */

defined( 'ABSPATH' ) || exit;

/**
 * Class Block_Curator
 */
class Blocur_Core {

	/**
	 * All known core/embed variation slugs (providerNameSlug).
	 * These match WordPress core's registered embed variations.
	 *
	 * @var string[]
	 */
	public static $embed_variations = array(
		'amazon-kindle', 'animoto', 'bluesky', 'cloudup', 'crowdsignal',
		'dailymotion', 'flickr', 'imgur', 'issuu', 'kickstarter',
		'mixcloud', 'pinterest', 'pocket-casts', 'reddit', 'reverbnation',
		'scribd', 'smugmug', 'soundcloud', 'speaker-deck', 'spotify', 'ted',
		'tiktok', 'tumblr', 'twitter', 'videopress', 'vimeo',
		'wolfram-cloud', 'wordpress', 'wordpress-tv', 'youtube',
	);

	/**
	 * Register hooks.
	 */
	public function init() {
		add_filter( 'allowed_block_types_all', array( $this, 'filter_allowed_blocks' ), 10, 2 );
		// [3] Inject CSS to hide disabled embed variations in the block inserter.
		add_action( 'enqueue_block_editor_assets', array( $this, 'inject_embed_variation_css' ) );
		// [4] JS-side enforcement: re-applies block restrictions after all client-side
		//     blocks (e.g. WooCommerce) are registered — supplements the PHP filter.
		add_action( 'enqueue_block_editor_assets', array( $this, 'enqueue_editor_restrict_script' ) );
	}

	/**
	 * [3] Output inline CSS that hides disabled embed variations in the block inserter.
	 * This is a visual-only (CSS) restriction, not a functional block removal.
	 * WordPress renders each variation as a list item with data-slug attribute.
	 */
	public function inject_embed_variation_css() {
		// Respect admin exemption.
		$exempt_admin = get_option( 'blocur_exempt_admin', '1' );
		if ( '1' === $exempt_admin && current_user_can( 'manage_options' ) ) {
			return;
		}

		$disabled_variations = get_option( 'blocur_disabled_embed_variations', array() );
		if ( empty( $disabled_variations ) ) {
			return;
		}

		$selectors = array();
		foreach ( $disabled_variations as $slug ) {
			$slug = sanitize_text_field( $slug );
			// DOM inspection confirmed: Gutenberg renders each embed variation button with
			// class "editor-block-list-item-embed/{slug}" e.g. "editor-block-list-item-embed/youtube".
			// The "/" must be escaped as "\/" in CSS selectors.
			$selectors[] = '.editor-block-list-item-embed\\/' . esc_attr( $slug );
		}

		if ( empty( $selectors ) ) {
			return;
		}

		$css = implode( ",\n", $selectors ) . " { display: none !important; }\n";
		wp_register_style( 'blocur-embed-hide', false ); // phpcs:ignore WordPress.WP.EnqueuedResourceParameters
		wp_enqueue_style( 'blocur-embed-hide' );
		wp_add_inline_style( 'blocur-embed-hide', wp_strip_all_tags( $css ) );
	}

	/**
	 * [4] Enqueue JS that re-applies block restrictions client-side after all
	 * third-party blocks (e.g. WooCommerce) have registered themselves.
	 *
	 * The PHP `allowed_block_types_all` filter covers WordPress core blocks
	 * reliably, but blocks registered via JavaScript after the editor
	 * initialises (late client-side registration) can bypass it. This script
	 * runs on wp.domReady and updates the block-editor store's allowedBlockTypes
	 * to enforce the same disabled list.
	 */
	public function enqueue_editor_restrict_script() {
		// Respect admin exemption — same guard as the PHP filter.
		$exempt_admin = get_option( 'blocur_exempt_admin', '1' );
		if ( '1' === $exempt_admin && current_user_can( 'manage_options' ) ) {
			return;
		}

		$disabled = get_option( 'blocur_disabled_blocks', array() );
		if ( empty( $disabled ) ) {
			return;
		}

		wp_enqueue_script(
			'16deza-block-curator-editor-restrict',
			BLOCUR_PLUGIN_URL . 'includes/js/block-editor-restrict.js',
			array( 'wp-blocks', 'wp-dom-ready', 'wp-data' ),
			BLOCUR_VERSION,
			true
		);

		wp_localize_script(
			'16deza-block-curator-editor-restrict',
			'blocurEditorData',
			array(
				'disabledBlocks' => array_values( $disabled ),
			)
		);
	}

	/**
	 * Filter allowed block types for the editor.
	 *
	 * @param bool|array               $allowed_block_types  Array of block type slugs, or true for all.
	 * @param WP_Block_Editor_Context  $editor_context       Editor context.
	 * @return bool|array
	 */
	public function filter_allowed_blocks( $allowed_block_types, $editor_context ) {

		// Check if admin exemption is enabled.
		$exempt_admin = get_option( 'blocur_exempt_admin', '1' );
		if ( '1' === $exempt_admin && current_user_can( 'manage_options' ) ) {
			return $allowed_block_types; // Admins see all blocks.
		}

		$disabled = get_option( 'blocur_disabled_blocks', array() );

		// If nothing is disabled, return as-is.
		if ( empty( $disabled ) ) {
			return $allowed_block_types;
		}

		// Get the full list of registered blocks.
		$registered = WP_Block_Type_Registry::get_instance()->get_all_registered();
		$all_blocks  = array_keys( $registered );

		// If $allowed_block_types is already a specific array (set by another plugin),
		// respect that and only further remove our disabled ones.
		if ( is_array( $allowed_block_types ) ) {
			$all_blocks = $allowed_block_types;
		}

		// Remove disabled blocks.
		$allowed = array_values(
			array_filter(
				$all_blocks,
				function( $block_name ) use ( $disabled ) {
					return ! in_array( $block_name, $disabled, true );
				}
			)
		);

		return $allowed;
	}

	/**
	 * Return all registered blocks grouped by category.
	 * Used by the admin settings page.
	 *
	 * @return array  Associative array: category_slug => [ 'label' => string, 'blocks' => [] ]
	 */
	public static function get_blocks_by_category() {
		if ( ! function_exists( 'get_block_categories' ) ) {
			return array();
		}

		// Retrieve categories. Pass a dummy post for context.
		$dummy_post = new WP_Post( (object) array( 'ID' => 0, 'post_type' => 'post' ) );
		$categories = get_block_categories( $dummy_post );

		// Build category map.
		$category_map = array();
		foreach ( $categories as $cat ) {
			$category_map[ $cat['slug'] ] = array(
				'label'  => $cat['title'],
				'blocks' => array(),
			);
		}
		// Fallback for uncategorised.
		$category_map['uncategorized'] = array(
			'label'  => __( 'Uncategorized', '16deza-block-curator' ),
			'blocks' => array(),
		);

		$registered = WP_Block_Type_Registry::get_instance()->get_all_registered();

		foreach ( $registered as $block_name => $block_type ) {
			$cat_slug = isset( $block_type->category ) && ! empty( $block_type->category )
				? $block_type->category
				: 'uncategorized';

			if ( ! isset( $category_map[ $cat_slug ] ) ) {
				$category_map[ $cat_slug ] = array(
					'label'  => ucfirst( str_replace( '-', ' ', $cat_slug ) ),
					'blocks' => array(),
				);
			}

			// Determine human-readable block title.
			$title = '';
			if ( ! empty( $block_type->title ) ) {
				$title = $block_type->title;
			} else {
				// Derive a readable name from the block slug.
				$parts = explode( '/', $block_name );
				$title = ucwords( str_replace( '-', ' ', end( $parts ) ) );
			}

			$category_map[ $cat_slug ]['blocks'][] = array(
				'name'  => $block_name,
				'title' => $title,
			);
		}

		// Remove empty categories.
		$category_map = array_filter(
			$category_map,
			function( $cat ) {
				return ! empty( $cat['blocks'] );
			}
		);

		// Sort blocks within each category alphabetically by title.
		foreach ( $category_map as $slug => &$cat ) {
			usort(
				$cat['blocks'],
				function( $a, $b ) {
					return strcmp( $a['title'], $b['title'] );
				}
			);
		}
		unset( $cat );

		return $category_map;
	}
}