← All changes | extensions/blocks/ai-assistant/ai-assistant.php +92 -10 12.4.2 → 16.3-a.7 View file →

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