Merged in feature/MAW-855-import-code-into-aws (pull request #2)

code import from pantheon

* code import from pantheon

wp/wp-content/plugins/woocommerce/packages/woocommerce-blocks/src/BlockPatterns.php

@@ -1,10 +1,14 @@
 <?php
 namespace Automattic\WooCommerce\Blocks;
 
+use Automattic\WooCommerce\Blocks\AI\Connection;
+use Automattic\WooCommerce\Blocks\Images\Pexels;
 use Automattic\WooCommerce\Blocks\Domain\Package;
+use Automattic\WooCommerce\Blocks\Patterns\PatternUpdater;
+use Automattic\WooCommerce\Blocks\Patterns\ProductUpdater;
 
 /**
- * Registers patterns under the `./patterns/` directory.
+ * Registers patterns under the `./patterns/` directory and updates their content.
  * Each pattern is defined as a PHP file and defines its metadata using plugin-style headers.
  * The minimum required definition is:
  *
@@ -48,6 +52,36 @@ class BlockPatterns {
 		$this->patterns_path = $package->get_path( 'patterns' );
 
 		add_action( 'init', array( $this, 'register_block_patterns' ) );
+		add_action( 'update_option_woo_ai_describe_store_description', array( $this, 'schedule_on_option_update' ), 10, 2 );
+		add_action( 'update_option_woo_ai_describe_store_description', array( $this, 'update_ai_connection_allowed_option' ), 10, 2 );
+		add_action( 'upgrader_process_complete', array( $this, 'schedule_on_plugin_update' ), 10, 2 );
+		add_action( 'woocommerce_update_patterns_content', array( $this, 'update_patterns_content' ) );
+	}
+
+	/**
+	 * Make sure the 'woocommerce_blocks_allow_ai_connection' option is set to true if the site is connected to AI.
+	 *
+	 * @param string $option The option name.
+	 * @param string $value The option value.
+	 *
+	 * @return bool
+	 */
+	public function update_ai_connection_allowed_option( $option, $value ): bool {
+		$ai_connection = new Connection();
+
+		$site_id = $ai_connection->get_site_id();
+
+		if ( is_wp_error( $site_id ) ) {
+			return update_option( 'woocommerce_blocks_allow_ai_connection', false );
+		}
+
+		$token = $ai_connection->get_jwt_token( $site_id );
+
+		if ( is_wp_error( $token ) ) {
+			return update_option( 'woocommerce_blocks_allow_ai_connection', false );
+		}
+
+		return update_option( 'woocommerce_blocks_allow_ai_connection', true );
 	}
 
 	/**
@@ -198,4 +232,109 @@ class BlockPatterns {
 			register_block_pattern( $pattern_data['slug'], $pattern_data );
 		}
 	}
+
+	/**
+	 * Update the patterns content when the store description is changed.
+	 *
+	 * @param string $option The option name.
+	 * @param string $value The option value.
+	 */
+	public function schedule_on_option_update( $option, $value ) {
+		$this->schedule_patterns_content_update( $value );
+	}
+
+	/**
+	 * Update the patterns content when the WooCommerce Blocks plugin is updated.
+	 *
+	 * @param \WP_Upgrader $upgrader_object  WP_Upgrader instance.
+	 * @param array        $options  Array of bulk item update data.
+	 */
+	public function schedule_on_plugin_update( $upgrader_object, $options ) {
+		if ( 'update' === $options['action'] && 'plugin' === $options['type'] ) {
+			foreach ( $options['plugins'] as $plugin ) {
+				if ( str_contains( $plugin, 'woocommerce-gutenberg-products-block.php' ) || str_contains( $plugin, 'woocommerce.php' ) ) {
+					$business_description = get_option( 'woo_ai_describe_store_description' );
+
+					if ( $business_description ) {
+						$this->schedule_patterns_content_update( $business_description );
+					}
+				}
+			}
+		}
+	}
+
+	/**
+	 * Update the patterns content when the store description is changed.
+	 *
+	 * @param string $business_description The business description.
+	 */
+	public function schedule_patterns_content_update( $business_description ) {
+		if ( ! class_exists( 'WooCommerce' ) ) {
+			return;
+		}
+
+		$action_scheduler = WP_PLUGIN_DIR . '/woocommerce/packages/action-scheduler/action-scheduler.php';
+
+		if ( ! file_exists( $action_scheduler ) ) {
+			return;
+		}
+
+		require_once $action_scheduler;
+
+		as_schedule_single_action( time(), 'woocommerce_update_patterns_content', array( $business_description ) );
+	}
+
+	/**
+	 * Update the patterns content.
+	 *
+	 * @param string $value The new value saved for the add_option_woo_ai_describe_store_description option.
+	 *
+	 * @return bool|string|\WP_Error
+	 */
+	public function update_patterns_content( $value ) {
+		$allow_ai_connection = get_option( 'woocommerce_blocks_allow_ai_connection' );
+
+		if ( ! $allow_ai_connection ) {
+			return new \WP_Error(
+				'ai_connection_not_allowed',
+				__( 'AI content generation is not allowed on this store. Update your store settings if you wish to enable this feature.', 'woocommerce' )
+			);
+		}
+
+		$ai_connection = new Connection();
+
+		$site_id = $ai_connection->get_site_id();
+
+		if ( is_wp_error( $site_id ) ) {
+			return $site_id->get_error_message();
+		}
+
+		$token = $ai_connection->get_jwt_token( $site_id );
+
+		if ( is_wp_error( $token ) ) {
+			return $token->get_error_message();
+		}
+
+		$business_description = get_option( 'woo_ai_describe_store_description' );
+
+		$images = ( new Pexels() )->get_images( $ai_connection, $token, $business_description );
+
+		if ( is_wp_error( $images ) ) {
+			return $images->get_error_message();
+		}
+
+		$populate_patterns = ( new PatternUpdater() )->generate_content( $ai_connection, $token, $images, $business_description );
+
+		if ( is_wp_error( $populate_patterns ) ) {
+			return $populate_patterns->get_error_message();
+		}
+
+		$populate_products = ( new ProductUpdater() )->generate_content( $ai_connection, $token, $images, $business_description );
+
+		if ( is_wp_error( $populate_products ) ) {
+			return $populate_products->get_error_message();
+		}
+
+		return true;
+	}
 }