← All changes | extensions/blocks/ai-assistant/ai-assistant.php +66 -21 12.7.3 → 16.3-a.5 View file →

		@@ -13,11 +13,20 @@
13	13	use Automattic\Jetpack\Status;
14	14	use Automattic\Jetpack\Status\Host;
15	15	use Jetpack_Gutenberg;
16	16
17		-const FEATURE_NAME = 'ai-assistant';
18		-const BLOCK_NAME = 'jetpack/' . FEATURE_NAME;
	17	+if ( ! defined( 'ABSPATH' ) ) {
	18	+ exit( 0 );
	19	+}
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	+
20	29	/**
21	30	* Registers our block for use in Gutenberg
22	31	* This is done via an action so that we can disable
23	32	* registration if we need to.
		@@ -23,13 +32,15 @@
23	32	* registration if we need to.
24	33	*/
25	34	function register_block() {
26	35	if (
27		- ( new Host() )->is_wpcom_simple()
	36	+ ( ( new Host() )->is_wpcom_simple()
28	37	|| ! ( new Status() )->is_offline_mode()
	38	+ ) && \Jetpack_AI_Settings::is_ai_enabled()
	39	+ && \Jetpack_AI_Settings::is_feature_enabled( 'writing_assistant' )
29	40	) {
30	41	Blocks::jetpack_register_block(
31		- BLOCK_NAME,
	42	+ __DIR__,
32	43	array( 'render_callback' => __NAMESPACE__ . '\load_assets' )
33	44	);
34	45	}
35	46	}
		@@ -46,52 +57,86 @@
46	57	function load_assets( $attr, $content ) {
47	58	/*
48	59	* Enqueue necessary scripts and styles.
49	60	*/
50		- Jetpack_Gutenberg::load_assets_as_required( FEATURE_NAME );
	61	+ Jetpack_Gutenberg::load_assets_as_required( __DIR__ );
51	62
52	63	return sprintf(
53	64	'<div class="%1$s">%2$s</div>',
54		- esc_attr( Blocks::classes( FEATURE_NAME, $attr ) ),
	65	+ esc_attr( Blocks::classes( Blocks::get_block_feature( __DIR__ ), $attr ) ),
55	66	$content
56	67	);
57	68	}
58	69
59	70	/**
60		- * Register the `ai-assistant-support` extension.
	71	+ * Register extensions.
61	72	*/
62	73	add_action(
63	74	'jetpack_register_gutenberg_extensions',
64	75	function () {
65		- \Jetpack_Gutenberg::set_extension_available( 'ai-assistant-support' );
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		- \Jetpack_Gutenberg::set_extension_available( 'ai-assistant-form-support' );
	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' );
	88	+
	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	+ }
	112	+ }
76	113	}
77	114	);
78	115
79	116	/**
80		- * Register the `ai-content-lens` extension.
	117	+ * Register the `ai-list-to-table-transform` extension.
81	118	*/
82	119	add_action(
83	120	'jetpack_register_gutenberg_extensions',
84	121	function () {
85		- \Jetpack_Gutenberg::set_extension_available( 'ai-content-lens' );
	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' );
	126	+ }
86	127	}
87	128	);
88	129
89	130	/**
90		- * Register the `ai-assistant-backend-prompts` extension.
	131	+ * Register the `ai-response-feedback` extension.
91	132	*/
92	133	add_action(
93	134	'jetpack_register_gutenberg_extensions',
94	135	function () {
95		- \Jetpack_Gutenberg::set_extension_available( 'ai-assistant-backend-prompts' );
	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' );
	140	+ }
96	141	}
97	142	);