Add cli flag use-existing-terms

Author: coreymckrill

includes/CLI.php

@@ -263,7 +263,7 @@ function () use ( $progress ) {
 
 WP_CLI::add_command( 'wc generate products', array( 'WC\SmoothGenerator\CLI', 'products' ), array(
 	'shortdesc' => 'Generate products.',
-	'synopsis' => array(
+	'synopsis'  => array(
 		array(
 			'name'        => 'amount',
 			'type'        => 'positional',
@@ -278,8 +278,14 @@ function () use ( $progress ) {
 			'optional'    => true,
 			'options'     => array( 'simple', 'variable' ),
 		),
+		array(
+			'name'        => 'use-existing-terms',
+			'type'        => 'flag',
+			'description' => 'Only apply existing categories and tags to products, rather than generating new ones.',
+			'optional'    => true,
+		),
 	),
-	'longdesc' => "## EXAMPLES\n\nwc generate products 10\n\nwc generate products 20 --type=variable",
+	'longdesc'  => "## EXAMPLES\n\nwc generate products 10\n\nwc generate products 20 --type=variable --use-existing-terms",
 ) );
 
 WP_CLI::add_command( 'wc generate orders', array( 'WC\SmoothGenerator\CLI', 'orders' ), array(

includes/Generator/Product.php

@@ -133,6 +133,11 @@ public static function batch( $amount, array $args = array() ) {
 			return $amount;
 		}
 
+		$use_existing_terms = ! empty( $args['use-existing-terms'] );
+		if ( ! $use_existing_terms ) {
+			self::maybe_generate_terms( $amount );
+		}
+
 		$product_ids = array();
 
 		for ( $i = 1; $i <= $amount; $i ++ ) {
@@ -419,6 +424,41 @@ protected static function generate_simple_product() {
 		return $product;
 	}
 
+	/**
+	 * Maybe generate a number of terms for use with products, if there aren't enough existing terms.
+	 *
+	 * Number of terms is determined by the number of products that will be generated.
+	 *
+	 * @param int $product_amount The number of products that will be generated.
+	 *
+	 * @return void
+	 */
+	protected static function maybe_generate_terms( int $product_amount ): void {
+		if ( $product_amount < 10 ) {
+			$cats      = 5;
+			$cat_depth = 1;
+			$tags      = 10;
+		} elseif ( $product_amount < 50 ) {
+			$cats      = 10;
+			$cat_depth = 2;
+			$tags      = 20;
+		} else {
+			$cats      = 20;
+			$cat_depth = 3;
+			$tags      = 40;
+		}
+
+		$existing_cats = count( self::get_term_ids( 'product_cat', $cats ) );
+		if ( $existing_cats < $cats ) {
+			Term::batch( $cats - $existing_cats, 'product_cat', array( 'max-depth' => $cat_depth ) );
+		}
+
+		$existing_tags = count( self::get_term_ids( 'product_tag', $tags ) );
+		if ( $existing_tags < $tags ) {
+			Term::batch( $tags - $existing_tags, 'product_tag' );
+		}
+	}
+
 	/**
 	 * Get a number of random term IDs for a specific taxonomy.
 	 *
@@ -427,7 +467,7 @@ protected static function generate_simple_product() {
 	 *
 	 * @return array
 	 */
-	protected static function get_term_ids( $taxonomy, $limit ) {
+	protected static function get_term_ids( string $taxonomy, int $limit ): array {
 		if ( $limit <= 0 ) {
 			return array();
 		}