<?php
namespace Elementor\Modules\Mcp;
use Elementor\Core\Base\Module as BaseModule;
use Elementor\Core\Utils\Promotions\Filtered_Promotions_Manager;
use Elementor\MCP\Composer\Admin\McpSettingsController;
use Elementor\MCP\Composer\Admin\Page as Mcp_Admin_Page;
use Elementor\MCP\Composer\Mcp\Registry as Shared_Registry;
use Elementor\Modules\EditorOne\Classes\Menu_Data_Provider;
use Elementor\Modules\Mcp\Abilities\Abstract_Ability;
use Elementor\Modules\Mcp\AdminMenuItems\Editor_One_Mcp_Menu;
use Elementor\Modules\Mcp\Preview\Public_Preview_Handler;
use Elementor\Modules\Mcp\Registry\Ability_Registry;
use Elementor\Modules\Mcp\RestApi\Mcp_Proxy_REST_API;
use Elementor\Modules\Mcp\Utils\Editor_Sync_State;
use Elementor\Utils;
use WP\MCP\Core\McpAdapter;
if ( ! defined( 'ABSPATH' ) ) {
exit;
}
class Module extends BaseModule {
const ANALYTICS_REGISTRAR_HANDLE = 'elementor-mcp-analytics-registrar';
const PROMOTION_REGISTRAR_HANDLE = 'elementor-mcp-promotion-registrar';
const MCP_PROMOTION_UPGRADE_URL = 'https://go.elementor.com/go-pro-mcp-connector-page-upgrade/';
const MCP_PROMOTION_ALLOWED_DOMAIN = 'elementor.com';
private Ability_Registry $registry;
public function get_name() {
return 'mcp';
}
public function enqueue_analytics_registrar(): void {
wp_enqueue_script(
self::ANALYTICS_REGISTRAR_HANDLE,
$this->get_js_assets_url( 'mcp-analytics-registrar' ),
[ 'elementor-common', Mcp_Admin_Page::SCRIPT_HANDLE ],
ELEMENTOR_VERSION,
true
);
}
public function enqueue_promotion_registrar(): void {
if ( ! $this->should_enqueue_mcp_admin_promotion() ) {
return;
}
wp_enqueue_script(
self::PROMOTION_REGISTRAR_HANDLE,
$this->get_js_assets_url( 'mcp-promotion-registrar' ),
[
'elementor-common',
Mcp_Admin_Page::SCRIPT_HANDLE,
'react',
'react-dom',
'wp-i18n',
],
ELEMENTOR_VERSION,
true
);
wp_localize_script(
self::PROMOTION_REGISTRAR_HANDLE,
'elementorMcpPromotionConfig',
[
'upgradeUrl' => $this->get_promotion_upgrade_url(),
]
);
wp_set_script_translations( self::PROMOTION_REGISTRAR_HANDLE, 'elementor' );
}
private function should_enqueue_mcp_admin_promotion(): bool {
if ( ! current_user_can( 'manage_options' ) ) {
return false;
}
if ( ! class_exists( Utils::class ) ) {
return false;
}
if ( Utils::has_pro() && Utils::is_license_active() ) {
return false;
}
return true;
}
private function get_promotion_upgrade_url(): string {
$promotion_data = Filtered_Promotions_Manager::get_filtered_promotion_data(
[ 'upgrade_url' => self::MCP_PROMOTION_UPGRADE_URL ],
'elementor/mcp/custom_promotion',
'upgrade_url'
);
$upgrade_url = $promotion_data['upgrade_url'] ?? '';
return $this->is_allowed_promotion_url( $upgrade_url )
? $upgrade_url
: self::MCP_PROMOTION_UPGRADE_URL;
}
private function is_allowed_promotion_url( $url ): bool {
if ( ! is_string( $url ) ) {
return false;
}
$host = wp_parse_url( $url, PHP_URL_HOST );
if ( ! is_string( $host ) ) {
return false;
}
$host = strtolower( $host );
return self::MCP_PROMOTION_ALLOWED_DOMAIN === $host
|| str_ends_with( $host, '.' . self::MCP_PROMOTION_ALLOWED_DOMAIN );
}
public static function is_active() {
return class_exists( McpAdapter::class ) &&
function_exists( 'wp_register_ability' ) &&
class_exists( Shared_Registry::class );
}
public static function is_site_mcp_exposure_enabled(): bool {
return McpSettingsController::is_enabled();
}
public function __construct() {
parent::__construct();
$this->registry = self::build_core_registry();
( new Mcp_Proxy_REST_API( $this->registry ) )->register_hooks();
( new Public_Preview_Handler() )->register();
( new Editor_Sync_State() )->register_hooks();
if ( ! self::is_active() ) {
return;
}
add_action( 'init', [ $this, 'register_shared_registry_slugs' ], 5 );
add_action( 'elementor/editor-one/menu/register', [ $this, 'register_editor_one_menu' ], Editor_One_Mcp_Menu::REGISTER_PRIORITY_AFTER_SUBMISSIONS );
add_action( 'wp_abilities_api_categories_init', [ $this, 'register_ability_category' ] );
if ( self::is_site_mcp_exposure_enabled() ) {
add_action( 'wp_abilities_api_init', [ $this, 'register_abilities' ] );
}
}
public function registry(): Ability_Registry {
return $this->registry;
}
public function register_ability_category() {
if ( ! function_exists( 'wp_register_ability_category' ) ) {
return;
}
wp_register_ability_category(
'elementor',
[
'label' => __( 'Elementor', 'elementor' ),
'description' => __( 'Elementor page builder data, global classes, and variables.', 'elementor' ),
]
);
}
public function register_abilities() {
if ( ! function_exists( 'wp_register_ability' ) ) {
return;
}
foreach ( $this->registry->all() as $ability ) {
$ability->register();
}
}
public function register_shared_registry_slugs(): void {
$shared = Shared_Registry::instance();
$shared->register_tools( $this->collect_server_ids( $this->registry->tools() ) );
$shared->register_resources( $this->collect_server_ids( $this->registry->resources() ) );
}
public function register_editor_one_menu( Menu_Data_Provider $menu_data_provider ): void {
$menu_data_provider->register_menu(
new Editor_One_Mcp_Menu(),
[ 'preserve_label_casing' => true ]
);
}
public static function build_core_registry(): Ability_Registry {
$registry = new Ability_Registry();
foreach ( self::get_core_abilities( $registry ) as $ability ) {
$registry->add( $ability );
}
return $registry;
}
/** @return Abstract_Ability[] */
private static function get_core_abilities( Ability_Registry $registry ): array {
$abilities = [
new Abilities\Get_Structure_Ability(),
new Abilities\Update_Settings_Ability(),
new Abilities\Create_Page_Ability(),
new Abilities\Create_Preview_Link_Ability(),
new Abilities\Publish_Document_Ability(),
new Abilities\Style_Best_Practices_Ability(),
new Abilities\Wordpress_Best_Practices_Ability(),
new Abilities\Manage_Variable_Ability(),
new Abilities\Manage_Classes_Ability(),
new Abilities\Manage_Default_Styles_Ability(),
new Abilities\Get_Default_Styles_Ability(),
new Abilities\Reorder_Classes_Ability(),
new Abilities\Manage_Variable_Guide_Ability(),
new Abilities\Get_Widget_Schema_Ability(),
new Abilities\List_Widget_Schemas_Ability(),
new Abilities\List_Dynamic_Tags_Ability(),
new Abilities\Build_Composition_Ability(),
new Abilities\Manage_Elements_Ability(),
new Abilities\Global_Classes_Resource_Ability(),
new Abilities\List_Assets_Ability(),
new Abilities\Global_Variables_Resource_Ability(),
new Abilities\Interactions_Schema_Resource_Ability(),
new Abilities\List_Resources_Ability( $registry ),
new Abilities\Read_Resource_Ability( $registry ),
new Abilities\List_Components_Ability(),
new Abilities\Manage_Component_Ability(),
new Abilities\List_Posts_Ability(),
];
return $abilities;
}
/**
* @param Abstract_Ability[] $abilities
* @return string[]
*/
private function collect_server_ids( array $abilities ): array {
$ids = [];
foreach ( $abilities as $ability ) {
if ( $ability->is_exposed_on_server() ) {
$ids[] = $ability->get_id();
}
}
return $ids;
}
}