← All changes | extensions/blocks/ai-assistant/ai-assistant.php +59 -32 12.8.3 → 16.3-a.5 View file →

		@@ -13,8 +13,20 @@
13	13	use Automattic\Jetpack\Status;
14	14	use Automattic\Jetpack\Status\Host;
15	15	use Jetpack_Gutenberg;
16	16
	17	+if ( ! defined( 'ABSPATH' ) ) {
	18	+ exit( 0 );
	19	+}
	20	+
	21	+// Required directly rather than relying on the plugin bootstrap: on
	22	+// WordPress.com Simple the extension files load through wpcom's own loader
	23	+// and load-jetpack.php never runs.
	24	+// This file loads before PHPUnit starts collecting coverage.
	25	+// @codeCoverageIgnoreStart
	26	+require_once __DIR__ . '/../../../_inc/lib/class-jetpack-ai-settings.php';
	27	+// @codeCoverageIgnoreEnd
	28	+
17	29	/**
18	30	* Registers our block for use in Gutenberg
19	31	* This is done via an action so that we can disable
20	32	* registration if we need to.
		@@ -22,9 +34,10 @@
22	34	function register_block() {
23	35	if (
24	36	( ( new Host() )->is_wpcom_simple()
25	37	|| ! ( new Status() )->is_offline_mode()
26		- ) && apply_filters( 'jetpack_ai_enabled', true )
	38	+ ) && \Jetpack_AI_Settings::is_ai_enabled()
	39	+ && \Jetpack_AI_Settings::is_feature_enabled( 'writing_assistant' )
27	40	) {
28	41	Blocks::jetpack_register_block(
29	42	__DIR__,
30	43	array( 'render_callback' => __NAMESPACE__ . '\load_assets' )
		@@ -54,62 +67,76 @@
54	67	);
55	68	}
56	69
57	70	/**
58		- * Register the `ai-assistant-support` extension.
	71	+ * Register extensions.
59	72	*/
60	73	add_action(
61	74	'jetpack_register_gutenberg_extensions',
62	75	function () {
63		- if ( apply_filters( 'jetpack_ai_enabled', true ) ) {
64		- \Jetpack_Gutenberg::set_extension_available( 'ai-assistant-support' );
65		- }
66		- }
67		-);
	76	+ if ( \Jetpack_AI_Settings::is_ai_enabled() ) {
	77	+ // The usage meter reports on every AI feature, so it follows the
	78	+ // master switch alone.
	79	+ Jetpack_Gutenberg::set_extension_available( 'ai-assistant-usage-panel' );
68	80
69		-/**
70		- * Register the `ai-assistant-form-support` extension.
71		- */
72		-add_action(
73		- 'jetpack_register_gutenberg_extensions',
74		- function () {
75		- if ( apply_filters( 'jetpack_ai_enabled', true ) ) {
76		- \Jetpack_Gutenberg::set_extension_available( 'ai-assistant-form-support' );
77		- }
78		- }
79		-);
	81	+ if ( \Jetpack_AI_Settings::is_feature_enabled( 'writing_assistant' ) ) {
	82	+ Jetpack_Gutenberg::set_extension_available( 'ai-content-lens' );
	83	+ Jetpack_Gutenberg::set_extension_available( 'ai-assistant-support' );
	84	+ Jetpack_Gutenberg::set_extension_available( 'ai-assistant-form-support' );
	85	+ Jetpack_Gutenberg::set_extension_available( 'ai-assistant-backend-prompts' );
	86	+ Jetpack_Gutenberg::set_extension_available( 'ai-title-optimization' );
	87	+ Jetpack_Gutenberg::set_extension_available( 'ai-title-optimization-keywords-support' );
80	88
81		-/**
82		- * Register the `ai-content-lens` extension.
83		- */
84		-add_action(
85		- 'jetpack_register_gutenberg_extensions',
86		- function () {
87		- if ( apply_filters( 'jetpack_ai_enabled', true ) ) {
88		- \Jetpack_Gutenberg::set_extension_available( 'ai-content-lens' );
	89	+ $site_locale = get_locale();
	90	+ // Only enable Write Brief for sites with an English locale.
	91	+ // Disabled by default; set the 'breve_enabled' filter to true to re-enable.
	92	+ if ( str_starts_with( $site_locale, 'en' ) && apply_filters( 'breve_enabled', false ) ) {
	93	+ Jetpack_Gutenberg::set_extension_available( 'ai-proofread-breve' );
	94	+ }
	95	+
	96	+ if ( apply_filters( 'ai_correct_spelling_enabled', false ) ) {
	97	+ Jetpack_Gutenberg::set_extension_available( 'ai-correct-spelling' );
	98	+ }
	99	+ }
	100	+
	101	+ if ( \Jetpack_AI_Settings::is_feature_enabled( 'image_editor' ) ) {
	102	+ Jetpack_Gutenberg::set_extension_available( 'ai-featured-image-generator' );
	103	+ Jetpack_Gutenberg::set_extension_available( 'ai-assistant-experimental-image-generation-support' );
	104	+ Jetpack_Gutenberg::set_extension_available( 'ai-general-purpose-image-generator' );
	105	+ Jetpack_Gutenberg::set_extension_available( 'ai-assistant-site-logo-support' );
	106	+ Jetpack_Gutenberg::set_extension_available( 'ai-assistant-image-extension' );
	107	+ }
	108	+
	109	+ if ( apply_filters( 'ai_seo_enhancer_enabled', true ) ) {
	110	+ Jetpack_Gutenberg::set_availability_for_plan( 'ai-seo-enhancer' );
	111	+ }
89	112	}
90	113	}
91	114	);
92	115
93	116	/**
94		- * Register the `ai-assistant-backend-prompts` extension.
	117	+ * Register the `ai-list-to-table-transform` extension.
95	118	*/
96	119	add_action(
97	120	'jetpack_register_gutenberg_extensions',
98	121	function () {
99		- if ( apply_filters( 'jetpack_ai_enabled', true ) ) {
100		- \Jetpack_Gutenberg::set_extension_available( 'ai-assistant-backend-prompts' );
	122	+ if ( \Jetpack_AI_Settings::is_ai_enabled() &&
	123	+ apply_filters( 'list_to_table_transform_enabled', false )
	124	+ ) {
	125	+ \Jetpack_Gutenberg::set_extension_available( 'ai-list-to-table-transform' );
101	126	}
102	127	}
103	128	);
104	129
105	130	/**
106		- * Register the `ai-assistant-usage-panel` extension.
	131	+ * Register the `ai-response-feedback` extension.
107	132	*/
108	133	add_action(
109	134	'jetpack_register_gutenberg_extensions',
110	135	function () {
111		- if ( apply_filters( 'jetpack_ai_enabled', true ) ) {
112		- \Jetpack_Gutenberg::set_extension_available( 'ai-assistant-usage-panel' );
	136	+ if ( \Jetpack_AI_Settings::is_ai_enabled() &&
	137	+ apply_filters( 'ai_response_feedback_enabled', true )
	138	+ ) {
	139	+ \Jetpack_Gutenberg::set_extension_available( 'ai-response-feedback' );
113	140	}
114	141	}
115	142	);