<?php
/**
 * Unified Template CPT
 *
 * Registers ONE custom post type "aafe_template" (shown as "Addronix Addons Items"
 * in Elementor Settings → Post Types) that powers all plugin template features:
 * Header & Footer, Loop Builder, and Mega Menu.
 *
 * Template type is differentiated via the `_aafe_template_type` post meta:
 *   - 'header'    → Header template
 *   - 'footer'    → Footer template
 *   - 'loop'      → Loop Builder item
 *   - 'mega-menu' → Mega Menu template
 *
 * @package AddronixAddons
 */

namespace AddronixAddons;

if ( ! defined( 'ABSPATH' ) ) {
	exit;
}

class AAFE_Template {

	/** The unified CPT slug — shown as "Addronix Addons Items" in Elementor. */
	const CPT = 'aafe_item';

	/** Meta key that stores the template type. */
	const META_TYPE = '_aafe_template_type';

	/** Allowed template types. */
	const TYPE_HEADER    = 'header';
	const TYPE_FOOTER    = 'footer';
	const TYPE_LOOP      = 'loop';
	const TYPE_MEGA_MENU = 'mega-menu';

	private static $instance = null;

	public static function instance() {
		if ( is_null( self::$instance ) ) {
			self::$instance = new self();
		}
		return self::$instance;
	}

	private function __construct() {
		add_action( 'init',              [ $this, 'register_cpt' ] );

		add_action( 'add_meta_boxes',    [ $this, 'add_type_meta_box' ] );
		add_action( 'save_post',         [ $this, 'save_type_meta' ] );
		add_action( 'pre_get_posts',     [ $this, 'filter_admin_list' ] );
		add_action( 'elementor/documents/register', [ $this, 'register_document_type' ] );
		add_action( 'admin_footer',      [ $this, 'admin_footer_js' ] );
		add_action( 'admin_enqueue_scripts', [ $this, 'enqueue_admin_assets' ] );
		add_filter( 'manage_' . self::CPT . '_posts_columns',       [ $this, 'add_type_column' ] );
		add_action( 'manage_' . self::CPT . '_posts_custom_column', [ $this, 'render_type_column' ], 10, 2 );
		add_filter( 'views_edit-' . self::CPT,                      [ $this, 'filter_views_links' ] );
		add_action( 'restrict_manage_posts',                        [ $this, 'add_hidden_type_input' ] );
		add_action( 'all_admin_notices',                            [ $this, 'render_admin_header' ] );
		add_filter( 'post_row_actions',                             [ $this, 'add_elementor_edit_link' ], 10, 2 );
	}

	/**
	 * Register the unified "Addronix Addons Items" CPT.
	 */
	public function register_cpt() {
		register_post_type( self::CPT, [
			'labels' => [
				'name'               => esc_html__( 'Addronix Addons Items', 'addronix-addons-for-elementor' ),
				'singular_name'      => esc_html__( 'Addronix Addons Item', 'addronix-addons-for-elementor' ),
				'add_new'            => esc_html__( 'Add New Template', 'addronix-addons-for-elementor' ),
				'add_new_item'       => esc_html__( 'Add New Template', 'addronix-addons-for-elementor' ),
				'edit_item'          => esc_html__( 'Edit Template', 'addronix-addons-for-elementor' ),
				'view_item'          => esc_html__( 'View Template', 'addronix-addons-for-elementor' ),
				'search_items'       => esc_html__( 'Search Templates', 'addronix-addons-for-elementor' ),
				'not_found'          => esc_html__( 'No templates found.', 'addronix-addons-for-elementor' ),
				'not_found_in_trash' => esc_html__( 'No templates found in Trash.', 'addronix-addons-for-elementor' ),
			],
			'public'              => true,
			'publicly_queryable'  => true,
			'rewrite'             => [ 'slug' => 'aafe-template', 'with_front' => false ],
			'show_ui'             => true,
			'show_in_menu'        => false,
			'show_in_nav_menus'   => false,
			'exclude_from_search' => true,
			'has_archive'         => false,
			'capability_type'     => 'page',
			'show_in_rest'        => true,
			'hierarchical'        => false,
			'supports'            => [ 'title', 'editor', 'elementor' ],
		] );

		register_post_meta( self::CPT, self::META_TYPE, [
			'show_in_rest'  => true,
			'single'        => true,
			'type'          => 'string',
			'auth_callback' => function() { return current_user_can( 'edit_posts' ); },
		] );

		// AAFE FIX: Force Elementor support for our templates so "Edit with Elementor" works.
		$cpt_support = get_option( 'elementor_cpt_support', [ 'page', 'post' ] );
		if ( ! in_array( self::CPT, $cpt_support, true ) ) {
			$cpt_support[] = self::CPT;
			update_option( 'elementor_cpt_support', $cpt_support );
		}
	}



	/**
	 * Add "Template Type" meta box to the editor sidebar.
	 */
	public function add_type_meta_box() {
		add_meta_box(
			'aafe_template_type_box',
			esc_html__( 'Template Type', 'addronix-addons-for-elementor' ),
			[ $this, 'render_type_meta_box' ],
			self::CPT,
			'side',
			'high'
		);
	}

	/**
	 * Render the "Template Type" meta box.
	 *
	 * @param \WP_Post $post
	 */
	public function render_type_meta_box( $post ) {
		wp_nonce_field( 'aafe_template_type_nonce', 'aafe_template_type_nonce_field' );

		$current = get_post_meta( $post->ID, self::META_TYPE, true );

		// Pre-select from URL param when creating a new post.
		$post_type = isset( $_GET['post_type'] ) ? sanitize_key( wp_unslash( $_GET['post_type'] ) ) : ''; // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- Reading from GET for admin UI state.

		if ( self::CPT === $post_type && isset( $_GET['aafe_type'] ) ) { // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- Reading from GET for admin UI state.
			$current = sanitize_key( wp_unslash( $_GET['aafe_type'] ) ); // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- Reading from GET for admin UI state.
		}

		$types = [
			self::TYPE_HEADER    => __( '🔝 Header', 'addronix-addons-for-elementor' ),
			self::TYPE_FOOTER    => __( '🔻 Footer', 'addronix-addons-for-elementor' ),
			self::TYPE_LOOP      => __( '🔁 Loop Item', 'addronix-addons-for-elementor' ),
			self::TYPE_MEGA_MENU => __( '🚀 Mega Menu', 'addronix-addons-for-elementor' ),
		];
		?>
		<select name="_aafe_template_type" class="aafe-type-select">
			<option value="">— <?php esc_html_e( 'Select Type', 'addronix-addons-for-elementor' ); ?> —</option>
			<?php foreach ( $types as $value => $label ) : ?>
				<option value="<?php echo esc_attr( $value ); ?>" <?php selected( $current, $value ); ?>>
					<?php echo esc_html( $label ); ?>
				</option>
			<?php endforeach; ?>
		</select>
		<span class="aafe-type-hint">
			<?php esc_html_e( 'Select the purpose of this template.', 'addronix-addons-for-elementor' ); ?>
		</span>
		<?php
	}

	/**
	 * Save template type meta on post save.
	 *
	 * @param int $post_id
	 */
	public function save_type_meta( $post_id ) {
		if ( defined( 'DOING_AUTOSAVE' ) && DOING_AUTOSAVE ) {
			return;
		}
		if ( ! current_user_can( 'edit_post', $post_id ) ) {
			return;
		}

		// Only save if our nonce is present and verified.
		if ( ! isset( $_POST['aafe_template_type_nonce_field'] ) || ! wp_verify_nonce( sanitize_key( wp_unslash( $_POST['aafe_template_type_nonce_field'] ) ), 'aafe_template_type_nonce' ) ) {
			return;
		}

		$type = '';
		if ( isset( $_POST['_aafe_template_type'] ) ) { // phpcs:ignore WordPress.Security.NonceVerification.Missing -- Nonce verified by WordPress core post save.
			$type = sanitize_key( wp_unslash( $_POST['_aafe_template_type'] ) ); // phpcs:ignore WordPress.Security.NonceVerification.Missing -- Nonce verified by WordPress core post save.
		} elseif ( isset( $_GET['aafe_type'] ) ) { // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- Reading from GET for admin UI state.
			$type = sanitize_key( wp_unslash( $_GET['aafe_type'] ) ); // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- Reading from GET for admin UI state.
		} else {
			$referer = isset( $_SERVER['HTTP_REFERER'] ) ? sanitize_text_field( wp_unslash( $_SERVER['HTTP_REFERER'] ) ) : '';
			if ( $referer && strpos( $referer, 'aafe_type=' ) !== false ) {
				wp_parse_str( (string) wp_parse_url( $referer, PHP_URL_QUERY ), $query_args );
				if ( ! empty( $query_args['aafe_type'] ) ) {
					$type = sanitize_key( $query_args['aafe_type'] );
				}
			}
		}

		$allowed = [ self::TYPE_HEADER, self::TYPE_FOOTER, self::TYPE_LOOP, self::TYPE_MEGA_MENU, '' ];

		if ( in_array( $type, $allowed, true ) ) {
			if ( $type ) {
				update_post_meta( $post_id, self::META_TYPE, $type );
			} else {
				delete_post_meta( $post_id, self::META_TYPE );
			}
		}

		// Always ensure the Elementor Canvas template is used so the theme wrapper is removed.
		update_post_meta( $post_id, '_wp_page_template', 'elementor_canvas' );
	}

	/**
	 * Register Document Type with Elementor.
	 */
	public function register_document_type( $documents_manager ) {
		$documents_manager->register_document_type(
			self::CPT,
			\Elementor\Core\DocumentTypes\Page::get_class_full_name()
		);
	}

	/**
	 * Filter the admin list to show only templates of a specific type
	 * when the `aafe_type` URL parameter is present.
	 *
	 * @param \WP_Query $query
	 */
	public function filter_admin_list( $query ) {
		if ( ! is_admin() || ! $query->is_main_query() ) {
			return;
		}

		$post_type = isset( $_GET['post_type'] ) ? sanitize_key( wp_unslash( $_GET['post_type'] ) ) : ( isset( $query->query['post_type'] ) ? $query->query['post_type'] : '' ); // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- Reading from GET for admin UI state.

		if ( self::CPT !== $post_type || empty( $_GET['aafe_type'] ) ) { // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- Reading from GET for admin UI state.
			return;
		}
		$type = sanitize_key( wp_unslash( $_GET['aafe_type'] ) ); // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- Reading from GET for admin UI state.

		if ( 'header-footer' === $type ) {
			$query->set( 'meta_query', [
				[
					'key'     => self::META_TYPE,
					'value'   => [ self::TYPE_HEADER, self::TYPE_FOOTER ],
					'compare' => 'IN',
				],
			] );
		} else {
			$query->set( 'meta_key',   self::META_TYPE );
			$query->set( 'meta_value', $type );
		}
	}

	/**
	 * Add a "Type" column to the CPT admin list table.
	 *
	 * @param array $columns
	 * @return array
	 */
	public function add_type_column( $columns ) {
		$new = [];
		foreach ( $columns as $key => $label ) {
			$new[ $key ] = $label;
			if ( 'title' === $key ) {
				$new['aafe_type'] = esc_html__( 'Type', 'addronix-addons-for-elementor' );
			}
		}
		return $new;
	}

	/**
	 * Render the "Type" column value.
	 *
	 * @param string $column
	 * @param int    $post_id
	 */
	public function render_type_column( $column, $post_id ) {
		if ( 'aafe_type' !== $column ) {
			return;
		}
		$type  = get_post_meta( $post_id, self::META_TYPE, true );
		$labels = [
			self::TYPE_HEADER    => '🔝 ' . __( 'Header', 'addronix-addons-for-elementor' ),
			self::TYPE_FOOTER    => '🔻 ' . __( 'Footer', 'addronix-addons-for-elementor' ),
			self::TYPE_LOOP      => '🔁 ' . __( 'Loop Item', 'addronix-addons-for-elementor' ),
			self::TYPE_MEGA_MENU => '🚀 ' . __( 'Mega Menu', 'addronix-addons-for-elementor' ),
		];
		echo isset( $labels[ $type ] ) ? esc_html( $labels[ $type ] ) : '<span style="opacity:.5">—</span>';
	}

	/**
	 * Append the `aafe_type` parameter to the view links (All, Published, etc.)
	 *
	 * @param array $views
	 * @return array
	 */
	public function filter_views_links( $views ) {
		if ( isset( $_GET['aafe_type'] ) ) { // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- Reading from GET for admin UI state.
			$type = sanitize_key( wp_unslash( $_GET['aafe_type'] ) ); // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- Reading from GET for admin UI state.
			foreach ( $views as $key => $view ) {
				$views[ $key ] = preg_replace( '/href=([\'"])(.*?)\1/', 'href=$1$2&aafe_type=' . $type . '$1', $view );
			}
		}
		return $views;
	}

	/**
	 * Add hidden input for `aafe_type` in the admin list filter form.
	 *
	 * @param string $post_type
	 */
	public function add_hidden_type_input( $post_type ) {
		if ( self::CPT === $post_type && isset( $_GET['aafe_type'] ) ) { // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- Reading from GET for admin UI state.
			echo '<input type="hidden" name="aafe_type" value="' . esc_attr( sanitize_key( wp_unslash( $_GET['aafe_type'] ) ) ) . '">'; // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- Reading from GET for admin UI state.
		}
	}

	/**
	 * JS: Fix the "Add New" button on filtered admin list pages
	 * to carry the `aafe_type` URL parameter into the new post URL.
	 */
	public function admin_footer_js() {
		$screen = get_current_screen();
		if ( ! $screen || 'edit' !== $screen->base || self::CPT !== $screen->post_type ) {
			return;
		}
		$type = isset( $_GET['aafe_type'] ) ? sanitize_key( wp_unslash( $_GET['aafe_type'] ) ) : ''; // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- Reading from GET for admin UI state.
		if ( ! $type ) {
			return;
		}
		// Use wp_add_inline_script instead of raw <script> tag
		$type_js = esc_js( $type );
		$js = "(function(){var links=document.querySelectorAll('a.page-title-action,a#post-new');links.forEach(function(a){var href=a.getAttribute('href');if(href&&href.indexOf('post-new.php')!==-1){a.setAttribute('href',href+'&aafe_type={$type_js}');}});}());";
		wp_add_inline_script( 'jquery', $js );
	}

	/**
	 * Get published templates filtered by type.
	 *
	 * @param  string $type One of the TYPE_* constants, or '' for all.
	 * @return \WP_Post[]
	 */
	public static function get_templates_by_type( $type = '' ) {
		$args = [
			'post_type'      => self::CPT,
			'posts_per_page' => -1,
			'post_status'    => 'publish',
		];
		if ( $type ) {
			$args['meta_query'] = [ // phpcs:ignore WordPress.DB.SlowDBQuery
				[
					'key'   => self::META_TYPE,
					'value' => $type,
				],
			];
		}
		return get_posts( $args );
	}

	/**
	 * Enqueue admin assets for meta box UI — uses wp API instead of raw style tags.
	 */
	public function enqueue_admin_assets( $hook ) {
		$screen = get_current_screen();
		if ( ! $screen || $screen->post_type !== self::CPT ) {
			return;
		}
		$css = '.aafe-type-select{width:100%;padding:6px;border-radius:4px;border:1px solid #8c8f94;margin-top:6px;}.aafe-type-hint{font-size:11px;opacity:.7;margin-top:6px;display:block;}';
		wp_add_inline_style( 'wp-admin', $css );
	}

	public function render_admin_header() {
		$screen = get_current_screen();
		if ( ! $screen || 'edit' !== $screen->base || self::CPT !== $screen->post_type ) {
			return;
		}

		$type = isset( $_GET['aafe_type'] ) ? sanitize_key( wp_unslash( $_GET['aafe_type'] ) ) : ''; // phpcs:ignore

		$title = __( 'Theme Builder', 'addronix-addons-for-elementor' );
		$desc  = __( 'Design and manage all your site parts in one central place.', 'addronix-addons-for-elementor' );

		if ( 'header-footer' === $type ) {
			$title = __( 'Header & Footer Builder', 'addronix-addons-for-elementor' );
			$desc  = __( 'Create and manage global headers and footers for your site.', 'addronix-addons-for-elementor' );
		} elseif ( 'loop' === $type ) {
			$title = __( 'Loop Builder', 'addronix-addons-for-elementor' );
			$desc  = __( 'Design custom loop items for post and product grids.', 'addronix-addons-for-elementor' );
		} elseif ( 'mega-menu' === $type ) {
			$title = __( 'Mega Menu Builder', 'addronix-addons-for-elementor' );
			$desc  = __( 'Build rich mega menus to display in your navigation.', 'addronix-addons-for-elementor' );
		}

		$add_new_url = add_query_arg( [ 'post_type' => self::CPT, 'aafe_type' => $type ], admin_url( 'post-new.php' ) );
		?>
		<div class="aafe-admin-wrap" style="min-height:auto; margin-top:20px;">
			<header class="aafe-page-hd" style="margin-right:20px;">
				<div class="aafe-page-hd-text">
					<h1 class="aafe-page-title"><?php echo esc_html( $title ); ?></h1>
					<p class="aafe-page-desc">
						<span class="aafe-page-desc-dot"></span>
						<?php echo esc_html( $desc ); ?>
					</p>
				</div>
				<div class="aafe-page-hd-meta">
					<a href="<?php echo esc_url( $add_new_url ); ?>" class="aafe-btn aafe-btn--primary">
						<span class="dashicons dashicons-plus-alt2"></span>
						<?php esc_html_e( 'Add New Template', 'addronix-addons-for-elementor' ); ?>
					</a>
				</div>
			</header>
		</div>
		<style>
			/* Hide default WP header on these pages so our custom SaaS header shines */
			.wrap > h1.wp-heading-inline,
			.wrap > a.page-title-action,
			.wrap > hr.wp-header-end { display: none !important; }
		</style>
		<?php
	}

	public function add_elementor_edit_link( $actions, $post ) {
		if ( self::CPT === $post->post_type ) {
			// Make it super easy to edit with Elementor by adding a direct link.
			$url = add_query_arg( [
				'post'   => $post->ID,
				'action' => 'elementor',
			], admin_url( 'post.php' ) );
			$actions['edit_with_elementor'] = '<a href="' . esc_url( $url ) . '" style="font-weight:600; color:#834DFB;">' . esc_html__( 'Edit with Elementor', 'addronix-addons-for-elementor' ) . '</a>';
		}
		return $actions;
	}
}