elementor / modules / mcp / module.php

module.php in Elementor Website Builder – more than just a page builder 4.3.4, at modules/mcp/module.php

260 lines 7.2 KB

Raw Download Zip

1	<?php
2
3	namespace Elementor\Modules\Mcp;
4
5	use Elementor\Core\Base\Module as BaseModule;
6	use Elementor\Core\Utils\Promotions\Filtered_Promotions_Manager;
7	use Elementor\MCP\Composer\Admin\McpSettingsController;
8	use Elementor\MCP\Composer\Admin\Page as Mcp_Admin_Page;
9	use Elementor\MCP\Composer\Mcp\Registry as Shared_Registry;
10	use Elementor\Modules\EditorOne\Classes\Menu_Data_Provider;
11	use Elementor\Modules\Mcp\Abilities\Abstract_Ability;
12	use Elementor\Modules\Mcp\AdminMenuItems\Editor_One_Mcp_Menu;
13	use Elementor\Modules\Mcp\Preview\Public_Preview_Handler;
14	use Elementor\Modules\Mcp\Registry\Ability_Registry;
15	use Elementor\Modules\Mcp\RestApi\Mcp_Proxy_REST_API;
16	use Elementor\Modules\Mcp\Utils\Editor_Sync_State;
17	use Elementor\Utils;
18	use WP\MCP\Core\McpAdapter;
19
20	if ( ! defined( 'ABSPATH' ) ) {
21	exit;
22	}
23
24	class Module extends BaseModule {
25
26	const ANALYTICS_REGISTRAR_HANDLE = 'elementor-mcp-analytics-registrar';
27	const PROMOTION_REGISTRAR_HANDLE = 'elementor-mcp-promotion-registrar';
28	const MCP_PROMOTION_UPGRADE_URL = 'https://go.elementor.com/go-pro-mcp-connector-page-upgrade/';
29	const MCP_PROMOTION_ALLOWED_DOMAIN = 'elementor.com';
30
31	private Ability_Registry $registry;
32
33	public function get_name() {
34	return 'mcp';
35	}
36
37	public function enqueue_analytics_registrar(): void {
38	wp_enqueue_script(
39	self::ANALYTICS_REGISTRAR_HANDLE,
40	$this->get_js_assets_url( 'mcp-analytics-registrar' ),
41	[ 'elementor-common', Mcp_Admin_Page::SCRIPT_HANDLE ],
42	ELEMENTOR_VERSION,
43	true
44	);
45	}
46
47	public function enqueue_promotion_registrar(): void {
48	if ( ! $this->should_enqueue_mcp_admin_promotion() ) {
49	return;
50	}
51
52	wp_enqueue_script(
53	self::PROMOTION_REGISTRAR_HANDLE,
54	$this->get_js_assets_url( 'mcp-promotion-registrar' ),
55	[
56	'elementor-common',
57	Mcp_Admin_Page::SCRIPT_HANDLE,
58	'react',
59	'react-dom',
60	'wp-i18n',
61	],
62	ELEMENTOR_VERSION,
63	true
64	);
65
66	wp_localize_script(
67	self::PROMOTION_REGISTRAR_HANDLE,
68	'elementorMcpPromotionConfig',
69	[
70	'upgradeUrl' => $this->get_promotion_upgrade_url(),
71	]
72	);
73
74	wp_set_script_translations( self::PROMOTION_REGISTRAR_HANDLE, 'elementor' );
75	}
76
77	private function should_enqueue_mcp_admin_promotion(): bool {
78	if ( ! current_user_can( 'manage_options' ) ) {
79	return false;
80	}
81
82	if ( ! class_exists( Utils::class ) ) {
83	return false;
84	}
85
86	if ( Utils::has_pro() && Utils::is_license_active() ) {
87	return false;
88	}
89
90	return true;
91	}
92
93	private function get_promotion_upgrade_url(): string {
94	$promotion_data = Filtered_Promotions_Manager::get_filtered_promotion_data(
95	[ 'upgrade_url' => self::MCP_PROMOTION_UPGRADE_URL ],
96	'elementor/mcp/custom_promotion',
97	'upgrade_url'
98	);
99
100	$upgrade_url = $promotion_data['upgrade_url'] ?? '';
101
102	return $this->is_allowed_promotion_url( $upgrade_url )
103	? $upgrade_url
104	: self::MCP_PROMOTION_UPGRADE_URL;
105	}
106
107	private function is_allowed_promotion_url( $url ): bool {
108	if ( ! is_string( $url ) ) {
109	return false;
110	}
111
112	$host = wp_parse_url( $url, PHP_URL_HOST );
113
114	if ( ! is_string( $host ) ) {
115	return false;
116	}
117
118	$host = strtolower( $host );
119
120	return self::MCP_PROMOTION_ALLOWED_DOMAIN === $host
121	\|\| str_ends_with( $host, '.' . self::MCP_PROMOTION_ALLOWED_DOMAIN );
122	}
123
124	public static function is_active() {
125	return class_exists( McpAdapter::class ) &&
126	function_exists( 'wp_register_ability' ) &&
127	class_exists( Shared_Registry::class );
128	}
129
130	public static function is_site_mcp_exposure_enabled(): bool {
131	return McpSettingsController::is_enabled();
132	}
133
134	public function __construct() {
135	parent::__construct();
136
137	$this->registry = self::build_core_registry();
138
139	( new Mcp_Proxy_REST_API( $this->registry ) )->register_hooks();
140	( new Public_Preview_Handler() )->register();
141	( new Editor_Sync_State() )->register_hooks();
142
143	if ( ! self::is_active() ) {
144	return;
145	}
146
147	add_action( 'init', [ $this, 'register_shared_registry_slugs' ], 5 );
148	add_action( 'elementor/editor-one/menu/register', [ $this, 'register_editor_one_menu' ], Editor_One_Mcp_Menu::REGISTER_PRIORITY_AFTER_SUBMISSIONS );
149
150	add_action( 'wp_abilities_api_categories_init', [ $this, 'register_ability_category' ] );
151
152	if ( self::is_site_mcp_exposure_enabled() ) {
153	add_action( 'wp_abilities_api_init', [ $this, 'register_abilities' ] );
154	}
155	}
156
157	public function registry(): Ability_Registry {
158	return $this->registry;
159	}
160
161	public function register_ability_category() {
162	if ( ! function_exists( 'wp_register_ability_category' ) ) {
163	return;
164	}
165
166	wp_register_ability_category(
167	'elementor',
168	[
169	'label' => __( 'Elementor', 'elementor' ),
170	'description' => __( 'Elementor page builder data, global classes, and variables.', 'elementor' ),
171	]
172	);
173	}
174
175	public function register_abilities() {
176	if ( ! function_exists( 'wp_register_ability' ) ) {
177	return;
178	}
179
180	foreach ( $this->registry->all() as $ability ) {
181	$ability->register();
182	}
183	}
184
185	public function register_shared_registry_slugs(): void {
186	$shared = Shared_Registry::instance();
187
188	$shared->register_tools( $this->collect_server_ids( $this->registry->tools() ) );
189	$shared->register_resources( $this->collect_server_ids( $this->registry->resources() ) );
190	}
191
192	public function register_editor_one_menu( Menu_Data_Provider $menu_data_provider ): void {
193	$menu_data_provider->register_menu(
194	new Editor_One_Mcp_Menu(),
195	[ 'preserve_label_casing' => true ]
196	);
197	}
198
199	public static function build_core_registry(): Ability_Registry {
200	$registry = new Ability_Registry();
201
202	foreach ( self::get_core_abilities( $registry ) as $ability ) {
203	$registry->add( $ability );
204	}
205
206	return $registry;
207	}
208
209	/** @return Abstract_Ability[] */
210	private static function get_core_abilities( Ability_Registry $registry ): array {
211	$abilities = [
212	new Abilities\Get_Structure_Ability(),
213	new Abilities\Update_Settings_Ability(),
214	new Abilities\Create_Page_Ability(),
215	new Abilities\Create_Preview_Link_Ability(),
216	new Abilities\Publish_Document_Ability(),
217	new Abilities\Style_Best_Practices_Ability(),
218	new Abilities\Wordpress_Best_Practices_Ability(),
219	new Abilities\Manage_Variable_Ability(),
220	new Abilities\Manage_Classes_Ability(),
221	new Abilities\Manage_Default_Styles_Ability(),
222	new Abilities\Get_Default_Styles_Ability(),
223	new Abilities\Reorder_Classes_Ability(),
224	new Abilities\Manage_Variable_Guide_Ability(),
225	new Abilities\Get_Widget_Schema_Ability(),
226	new Abilities\List_Widget_Schemas_Ability(),
227	new Abilities\List_Dynamic_Tags_Ability(),
228	new Abilities\Build_Composition_Ability(),
229	new Abilities\Manage_Elements_Ability(),
230	new Abilities\Global_Classes_Resource_Ability(),
231	new Abilities\List_Assets_Ability(),
232	new Abilities\Global_Variables_Resource_Ability(),
233	new Abilities\Interactions_Schema_Resource_Ability(),
234	new Abilities\List_Resources_Ability( $registry ),
235	new Abilities\Read_Resource_Ability( $registry ),
236	new Abilities\List_Components_Ability(),
237	new Abilities\Manage_Component_Ability(),
238	new Abilities\List_Posts_Ability(),
239	];
240
241	return $abilities;
242	}
243
244	/**
245	* @param Abstract_Ability[] $abilities
246	* @return string[]
247	*/
248	private function collect_server_ids( array $abilities ): array {
249	$ids = [];
250
251	foreach ( $abilities as $ability ) {
252	if ( $ability->is_exposed_on_server() ) {
253	$ids[] = $ability->get_id();
254	}
255	}
256
257	return $ids;
258	}
259	}
260