singleprogressbar

Author: opicron

facetwp-wp-cli.php

@@ -31,57 +31,76 @@ class CLI extends WP_CLI_Command {
      * Indexes all posts
      *
      * ## OPTIONS
+     * [--post-type=<name>]
+     * : post type, 'any' if not defined
      *
      * ## EXAMPLES
      *
      *     wp facet index
+     *     wp facet index --post-type=product
      *
      * @synopsis
      */
     function index( $args, $assoc_args ) {
 
         error_reporting(0);
 
-        $post_type = 'any';
-
-        if ( isset( $args[0] ) ) {
-            $post_type = $args[0];
-        }
+        if ( empty( $assoc_args['post-type'] ) )
+		$post_type = 'any';
+	else
+		$post_type = $assoc_args['post-type'];
 
         $posts_per_page = 100;
         $page = 1;
 
-        do {
-            $post_ids = get_posts( array(
+	$args = array(
                 'posts_per_page'    => $posts_per_page,
                 'paged'             => $page,
                 'post_type'         => $post_type,
                 'post_status'       => 'publish',
                 'fields'            => 'ids',
                 'orderby'           => 'ID',
                 'cache_results'     => false,
-            ));
-
-            // Do stuff
-
-            $progress_bar = WP_CLI\Utils\make_progress_bar('Indexing', count( $post_ids ));
-
-            $indexer = new FacetWP_Indexer;
-            $indexer->is_overridden = true;
+        );
 
-            foreach( $post_ids as $post_id ){
-                $progress_bar->tick();
-                $indexer->index( $post_id );
-            }
+	$total = 0;
 
-            $progress_bar->finish();
-
-            $page++;
-
-            // Free up memory
-            $this->stop_the_insanity();
-
-        } while ( count( $post_ids ) );
+        do {
+		$args['paged'] = $page;
+		$my_query = new WP_Query( $args );
+
+		if ($my_query->have_posts())
+		{
+			if ($page == 1)
+			{
+				$total = $my_query->found_posts;
+            			WP_CLI::line( 'Found '.$total.' posts of type "'.$post_type.'"' );
+				$progress_bar = WP_CLI\Utils\make_progress_bar('Indexing', $total );
+			}
+
+			foreach ( $my_query->posts as $key => $post_id )
+			{
+            			//WP_CLI::line( $post_id );
+	               		$progress_bar->tick();
+				FWP()->indexer->index( $post_id );
+			}
+
+	                // Free up memory
+        	        $this->stop_the_insanity();
+
+	                $page++;
+		}
+
+
+        } while ( $my_query->have_posts() );
+
+	if ($total > 0)
+	{
+		$progress_bar->finish();
+		WP_CLI::success( 'All posts indexed' );
+	} else {
+		WP_CLI::error( 'No posts of type "'.$post_type.'" found!' );
+	}
     }
 
     /*
@@ -102,5 +121,4 @@ protected function stop_the_insanity() {
 
 }
 
-
 WP_CLI::add_command( 'facet', __NAMESPACE__ . '\\CLI' );