← All changes | extensions/blocks/ai-assistant/ai-assistant.php +72 -62 13.3.3 → 16.3-beta 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.
		@@ -19,18 +31,26 @@
19	31	* This is done via an action so that we can disable
20	32	* registration if we need to.
21	33	*/
22	34	function register_block() {
23		- if (
24		- ( ( new Host() )->is_wpcom_simple()
25		- || ! ( new Status() )->is_offline_mode()
26		- ) && apply_filters( 'jetpack_ai_enabled', true )
27		- ) {
28		- Blocks::jetpack_register_block(
29		- __DIR__,
30		- array( 'render_callback' => __NAMESPACE__ . '\load_assets' )
31		- );
	35	+ if ( ! ( new Host() )->is_wpcom_simple() && ( new Status() )->is_offline_mode() ) {
	36	+ return;
32	37	}
	38	+
	39	+ if ( ! \Jetpack_AI_Settings::is_ai_enabled() ) {
	40	+ Jetpack_Gutenberg::set_extension_unavailable( 'ai-assistant', 'ai_disabled' );
	41	+ return;
	42	+ }
	43	+
	44	+ if ( ! \Jetpack_AI_Settings::is_feature_enabled( 'writing_assistant' ) ) {
	45	+ Jetpack_Gutenberg::set_extension_unavailable( 'ai-assistant', 'ai_disabled', array( 'feature' => 'writing_assistant' ) );
	46	+ return;
	47	+ }
	48	+
	49	+ Blocks::jetpack_register_block(
	50	+ __DIR__,
	51	+ array( 'render_callback' => __NAMESPACE__ . '\load_assets' )
	52	+ );
33	53	}
34	54	add_action( 'init', __NAMESPACE__ . '\register_block' );
35	55
36	56	/**
		@@ -54,86 +74,76 @@
54	74	);
55	75	}
56	76
57	77	/**
58		- * Register the `ai-assistant-support` extension.
	78	+ * Register extensions.
59	79	*/
60	80	add_action(
61	81	'jetpack_register_gutenberg_extensions',
62	82	function () {
63		- if ( apply_filters( 'jetpack_ai_enabled', true ) ) {
64		- \Jetpack_Gutenberg::set_extension_available( 'ai-assistant-support' );
65		- }
66		- }
67		-);
	83	+ if ( \Jetpack_AI_Settings::is_ai_enabled() ) {
	84	+ // The usage meter reports on every AI feature, so it follows the
	85	+ // master switch alone.
	86	+ Jetpack_Gutenberg::set_extension_available( 'ai-assistant-usage-panel' );
68	87
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		-);
	88	+ if ( \Jetpack_AI_Settings::is_feature_enabled( 'writing_assistant' ) ) {
	89	+ Jetpack_Gutenberg::set_extension_available( 'ai-content-lens' );
	90	+ Jetpack_Gutenberg::set_extension_available( 'ai-assistant-support' );
	91	+ Jetpack_Gutenberg::set_extension_available( 'ai-assistant-form-support' );
	92	+ Jetpack_Gutenberg::set_extension_available( 'ai-assistant-backend-prompts' );
	93	+ Jetpack_Gutenberg::set_extension_available( 'ai-title-optimization' );
	94	+ Jetpack_Gutenberg::set_extension_available( 'ai-title-optimization-keywords-support' );
80	95
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		- }
90		- }
91		-);
	96	+ $site_locale = get_locale();
	97	+ // Only enable Write Brief for sites with an English locale.
	98	+ // Disabled by default; set the 'breve_enabled' filter to true to re-enable.
	99	+ if ( str_starts_with( $site_locale, 'en' ) && apply_filters( 'breve_enabled', false ) ) {
	100	+ Jetpack_Gutenberg::set_extension_available( 'ai-proofread-breve' );
	101	+ }
92	102
93		-/**
94		- * Register the `ai-assistant-backend-prompts` extension.
95		- */
96		-add_action(
97		- 'jetpack_register_gutenberg_extensions',
98		- function () {
99		- if ( apply_filters( 'jetpack_ai_enabled', true ) ) {
100		- \Jetpack_Gutenberg::set_extension_available( 'ai-assistant-backend-prompts' );
101		- }
102		- }
103		-);
	103	+ if ( apply_filters( 'ai_correct_spelling_enabled', false ) ) {
	104	+ Jetpack_Gutenberg::set_extension_available( 'ai-correct-spelling' );
	105	+ }
	106	+ }
104	107
105		-/**
106		- * Register the `ai-assistant-usage-panel` extension.
107		- */
108		-add_action(
109		- 'jetpack_register_gutenberg_extensions',
110		- function () {
111		- if ( apply_filters( 'jetpack_ai_enabled', true ) ) {
112		- \Jetpack_Gutenberg::set_extension_available( 'ai-assistant-usage-panel' );
	108	+ if ( \Jetpack_AI_Settings::is_feature_enabled( 'image_editor' ) ) {
	109	+ Jetpack_Gutenberg::set_extension_available( 'ai-featured-image-generator' );
	110	+ Jetpack_Gutenberg::set_extension_available( 'ai-assistant-experimental-image-generation-support' );
	111	+ Jetpack_Gutenberg::set_extension_available( 'ai-general-purpose-image-generator' );
	112	+ Jetpack_Gutenberg::set_extension_available( 'ai-assistant-site-logo-support' );
	113	+ Jetpack_Gutenberg::set_extension_available( 'ai-assistant-image-extension' );
	114	+ }
	115	+
	116	+ if ( apply_filters( 'ai_seo_enhancer_enabled', true ) ) {
	117	+ Jetpack_Gutenberg::set_availability_for_plan( 'ai-seo-enhancer' );
	118	+ }
113	119	}
114	120	}
115	121	);
116	122
117	123	/**
118		- * Register the `ai-logo-generator` extension.
	124	+ * Register the `ai-list-to-table-transform` extension.
119	125	*/
120	126	add_action(
121	127	'jetpack_register_gutenberg_extensions',
122	128	function () {
123		- if ( apply_filters( 'jetpack_ai_enabled', true ) ) {
124		- \Jetpack_Gutenberg::set_extension_available( 'ai-logo-generator' );
	129	+ if ( \Jetpack_AI_Settings::is_ai_enabled() &&
	130	+ apply_filters( 'list_to_table_transform_enabled', false )
	131	+ ) {
	132	+ \Jetpack_Gutenberg::set_extension_available( 'ai-list-to-table-transform' );
125	133	}
126	134	}
127	135	);
128	136
129	137	/**
130		- * Register the `ai-featured-image-generator` extension.
	138	+ * Register the `ai-response-feedback` extension.
131	139	*/
132	140	add_action(
133	141	'jetpack_register_gutenberg_extensions',
134	142	function () {
135		- if ( apply_filters( 'jetpack_ai_enabled', true ) ) {
136		- \Jetpack_Gutenberg::set_extension_available( 'ai-featured-image-generator' );
	143	+ if ( \Jetpack_AI_Settings::is_ai_enabled() &&
	144	+ apply_filters( 'ai_response_feedback_enabled', true )
	145	+ ) {
	146	+ \Jetpack_Gutenberg::set_extension_available( 'ai-response-feedback' );
137	147	}
138	148	}
139	149	);