← All changes | extensions/blocks/ai-assistant/ai-assistant.php +149 -0 16.2-beta → 16.3 View file →

		@@ -1,0 +1,149 @@
	1	+<?php
	2	+/**
	3	+ * Jetpack AI Assistant Block.
	4	+ *
	5	+ * @since 12.2
	6	+ *
	7	+ * @package automattic/jetpack
	8	+ */
	9	+
	10	+namespace Automattic\Jetpack\Extensions\AIAssistant;
	11	+
	12	+use Automattic\Jetpack\Blocks;
	13	+use Automattic\Jetpack\Status;
	14	+use Automattic\Jetpack\Status\Host;
	15	+use Jetpack_Gutenberg;
	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	+
	29	+/**
	30	+ * Registers our block for use in Gutenberg
	31	+ * This is done via an action so that we can disable
	32	+ * registration if we need to.
	33	+ */
	34	+function register_block() {
	35	+ if ( ! ( new Host() )->is_wpcom_simple() && ( new Status() )->is_offline_mode() ) {
	36	+ return;
	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	+ );
	53	+}
	54	+add_action( 'init', __NAMESPACE__ . '\register_block' );
	55	+
	56	+/**
	57	+ * Jetpack AI Assistant block registration/dependency declaration.
	58	+ *
	59	+ * @param array $attr Array containing the Jetpack AI Assistant block attributes.
	60	+ * @param string $content String containing the Jetpack AI Assistant block content.
	61	+ *
	62	+ * @return string
	63	+ */
	64	+function load_assets( $attr, $content ) {
	65	+ /*
	66	+ * Enqueue necessary scripts and styles.
	67	+ */
	68	+ Jetpack_Gutenberg::load_assets_as_required( __DIR__ );
	69	+
	70	+ return sprintf(
	71	+ '<div class="%1$s">%2$s</div>',
	72	+ esc_attr( Blocks::classes( Blocks::get_block_feature( __DIR__ ), $attr ) ),
	73	+ $content
	74	+ );
	75	+}
	76	+
	77	+/**
	78	+ * Register extensions.
	79	+ */
	80	+add_action(
	81	+ 'jetpack_register_gutenberg_extensions',
	82	+ function () {
	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' );
	87	+
	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' );
	95	+
	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	+ }
	102	+
	103	+ if ( apply_filters( 'ai_correct_spelling_enabled', false ) ) {
	104	+ Jetpack_Gutenberg::set_extension_available( 'ai-correct-spelling' );
	105	+ }
	106	+ }
	107	+
	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	+ }
	119	+ }
	120	+ }
	121	+);
	122	+
	123	+/**
	124	+ * Register the `ai-list-to-table-transform` extension.
	125	+ */
	126	+add_action(
	127	+ 'jetpack_register_gutenberg_extensions',
	128	+ function () {
	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' );
	133	+ }
	134	+ }
	135	+);
	136	+
	137	+/**
	138	+ * Register the `ai-response-feedback` extension.
	139	+ */
	140	+add_action(
	141	+ 'jetpack_register_gutenberg_extensions',
	142	+ function () {
	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' );
	147	+ }
	148	+ }
	149	+);