📁 Captain File Manager

✏️ Editing: wpr-templates-library.php

<?php

if ( ! defined( 'ABSPATH' ) ) {
	exit; // Exit if accessed directly.
}

use WprAddons\Admin\Includes\WPR_Render_Templates;
use WprAddons\Admin\Includes\WPR_Templates_Shortcode;
use WprAddons\Admin\Includes\WPR_Templates_Modal_Popups;
use WprAddons\Admin\Includes\WPR_Templates_Actions;
use WprAddons\Admin\Templates\Library\WPR_Templates_Library_Blocks;
use WprAddons\Admin\Templates\Library\WPR_Templates_Library_Sections;
use WprAddons\Admin\Templates\Library\WPR_Templates_Library_Pages;
use WprAddons\Admin\Templates\Library\WPR_Templates_Library_Popups;
use WprAddons\Classes\Utilities;

/**
 * WPR_Templates_Library setup
 *
 * @since 1.0
 */
class WPR_Templates_Library {

/**
	** Constructor
	*/
	public function __construct() {

// Register CPTs
		add_action( 'init', [ $this, 'register_templates_library_cpt' ] );
		add_action( 'template_redirect', [ $this, 'block_template_frontend' ] );
		add_action( 'current_screen', [ $this, 'redirect_to_options_page' ] );

// Templates Shortcode
		new WPR_Templates_Shortcode();

// Init Popups
		new WPR_Templates_Modal_Popups();

// Init Theme Builder
		new WPR_Render_Templates();

// Template Actions
		new WPR_Templates_Actions();

// Add Blocks to Library
		new WPR_Templates_Library_Blocks();

// Add Sections to Library
		new WPR_Templates_Library_Sections();

// Add Pages to Library
		new WPR_Templates_Library_Pages();

// Add Popups to Library
		new WPR_Templates_Library_Popups();

// Enable Elementor for 'wpr_templates'
		$this->add_elementor_cpt_support();

}

/**
	** Register Templates Library
	*/
	public function redirect_to_options_page() {
		if ( get_current_screen()->post_type == 'wpr_templates' && isset($_GET['action']) && $_GET['action'] == 'edit' ) {
			$elementor_template_type = isset($_GET['post']) ? sanitize_text_field(wp_unslash($_GET['post'])) : '';

if ( 'wpr-popups' === Utilities::get_elementor_template_type( $elementor_template_type ) ) {
				wp_redirect('admin.php?page=wpr-popups');
			} else {
				wp_redirect('admin.php?page=wpr-theme-builder');
			}
		}
	}

public function register_templates_library_cpt() {

$args = array(
			'label'				  => esc_html__( 'Royal Templates', 'wpr-addons' ),
			'public'              => true,
			'rewrite'             => false,
			'show_ui'             => true,
			'show_in_menu'        => false,
			'show_in_nav_menus'   => false,
			'exclude_from_search' => true,
			'capability_type'     => 'post',
			'hierarchical'        => false,
		);

register_post_type( 'wpr_templates', $args );

$tax_args = [
			'hierarchical' 		=> true,
			'show_ui' 			=> true,
			'show_in_nav_menus' => false,
			'show_admin_column' => true,
			'query_var' 		=> is_admin(),
			'rewrite' 			=> false,
			'public' 			=> false,
		];
		
	    register_taxonomy( 'wpr_template_type', 'wpr_templates', $tax_args );

}

/**
	** Don't display on the frontend for non edit_posts capable users
	*/
	public function block_template_frontend() {
		if ( is_singular( 'wpr_templates' ) && ! current_user_can( 'edit_posts' ) ) {
			wp_redirect( site_url(), 301 );
			die;
		}
	}

/**
	*** Add elementor support for wpr_templates.
	**/
	function add_elementor_cpt_support() {
		if ( ! is_admin() ) {
			return;
		}

$cpt_support = get_option( 'elementor_cpt_support' );
		
		if ( ! $cpt_support ) {
		    update_option( 'elementor_cpt_support', ['post', 'page', 'wpr_templates'] );
		} elseif ( ! in_array( 'wpr_templates', $cpt_support ) ) {
		    $cpt_support[] = 'wpr_templates';
		    update_option( 'elementor_cpt_support', $cpt_support );
		}
	}

}

new WPR_Templates_Library();