Merge pull request #55 from woocommerce/enhancement/add-ui-support

Add UI support for generating products and orders.

Author: rrennick

includes/Admin/Settings.php

@@ -49,11 +49,14 @@ public static function render_admin_page() {
 			<h2>Generate Orders</h2>
 			<p>
 				<input type="number" name="num_orders_to_generate" value="<?php echo esc_attr( self::DEFAULT_NUM_ORDERS ); ?>" min="1" />
-				Number of orders to generate over
-				<input type="number" name="order_generation_interval" value="<?php echo esc_attr( self::DEFAULT_ORDER_INTERVAL_MINUTES ); ?>" min="0" />
-				minutes.
+				Number of orders to generate.
 			</p>
 			<?php submit_button( 'Generate', 'primary', 'generate_orders' ); ?>
+
+			<h2>Cancel all scheduled generations</h2>
+			<p>
+			</p>
+			<?php submit_button( 'Cancel all', 'primary', 'cancel_all_generations' ); ?>
 		</form>
 		<?php
 	}
@@ -65,14 +68,16 @@ public static function process_page_submit() {
 		if ( ! empty( $_POST['generate_products'] ) && ! empty( $_POST['num_products_to_generate'] ) ) {
 			check_admin_referer( 'generate', 'smoothgenerator_nonce' );
 			$num_to_generate = absint( $_POST['num_products_to_generate'] );
-			// @todo kick off generation here
+			wc_smooth_generate_schedule( 'product', $num_to_generate );
 			add_action( 'admin_notices', array( __CLASS__, 'product_generating_notice' ) );
-		} else if ( ! empty( $_POST['generate_orders'] ) && ! empty( $_POST['num_orders_to_generate'] ) && ! empty( $_POST['order_generation_interval'] ) ) {
+		} else if ( ! empty( $_POST['generate_orders'] ) && ! empty( $_POST['num_orders_to_generate'] ) ) {
 			check_admin_referer( 'generate', 'smoothgenerator_nonce' );
 			$num_to_generate = absint( $_POST['num_orders_to_generate'] );
-			$order_generation_interval = absint( $_POST['order_generation_interval'] );
-			// @todo kick off generation here
+			wc_smooth_generate_schedule( 'order', $num_to_generate );
 			add_action( 'admin_notices', array( __CLASS__, 'order_generating_notice' ) );
+		} else if ( ! empty( $_POST['cancel_all_generations'] ) ) {
+			check_admin_referer( 'generate', 'smoothgenerator_nonce' );
+			wc_smooth_generate_cancel_all();
 		}
 	}
 

includes/GenerateBackgroundProcess.php

@@ -5,52 +5,51 @@
  * @package SmoothGenerator\Classes
  */
 
-namespace WC\SmoothGenerator;
-
-defined( 'ABSPATH' ) || exit;
-
-use WP_Background_Process;
+use WC\SmoothGenerator\Generator;
 
 /**
- * Background data generation and creation class.
+ * Calls generator for object type.
+ *
+ * @param string $type Type of object to generate.
+ *
+ * @return false If task was successful.
  */
-class GenerateBackgroundProcess extends WP_Background_Process {
-
-	/**
-	 * Initiate new background process.
-	 */
-	public function __construct() {
-		$this->prefix = 'wp_' . get_current_blog_id();
-		$this->action = 'wc_smoothgenerator_generate';
-		parent::__construct();
+function wc_smooth_generate_object( $type ) {
+	// Check what generation task to perform.
+	switch ( $type ) {
+		case 'order':
+			Generator\Order::generate();
+			break;
+		case 'product':
+			Generator\Product::generate();
+			break;
+		case 'customer':
+			Generator\Customer::generate();
+			break;
+		default:
+			return false;
 	}
 
-	/**
-	 * Code to execute for each item in the queue
-	 *
-	 * @param array $item Item data.
-	 * @return boolean
-	 */
-	protected function task( $item ) {
-		if ( ! is_array( $item ) && ! isset( $item['task'] ) ) {
-			return false;
-		}
+	return false;
+}
 
-		// Check what generation task to perform.
-		switch ( $item['task'] ) {
-			case 'order':
-				Generator\Order::generate();
-				break;
-			case 'product':
-				Generator\Product::generate();
-				break;
-			case 'customer':
-				Generator\Customer::generate();
-				break;
-			default:
-				return false;
-		}
+add_action( 'wc_smooth_generate_object', 'wc_smooth_generate_object' );
 
-		return false;
+/**
+ * Schedule async actions for generation of objects.
+ *
+ * @param string $type Type of object to generate.
+ * @param int    $qty  Quantity of objects.
+ */
+function wc_smooth_generate_schedule( $type, $qty ) {
+	for ( $i = 0; $i < $qty; $i++ ) {
+		as_enqueue_async_action( 'wc_smooth_generate_object', array( 'type' => $type ), 'wc_smooth_generate_object_group' );
 	}
 }
+
+/**
+ * Cancel any scheduled generation actions.
+ */
+function wc_smooth_generate_cancel_all() {
+	as_unschedule_all_actions( '', array(), 'wc_smooth_generate_object_group' );
+}

wc-smooth-generator.php

@@ -12,8 +12,10 @@
 
 defined( 'ABSPATH' ) || exit;
 
+require_once __DIR__ . '/includes/GenerateBackgroundProcess.php';
+
 // autoloader.
-if( ! class_exists( \WC\SmoothGenerator\Plugin::class ) ){
+if ( ! class_exists( \WC\SmoothGenerator\Plugin::class ) ) {
 	require __DIR__ . '/vendor/autoload.php';
 }