From 108b485fb68e1bfc2ccafcafce7750aa54637a11 Mon Sep 17 00:00:00 2001
From: Taylor Lovett <admin@taylorlovett.com>
Date: Thu, 25 Apr 2013 02:57:17 +0000
Subject: [PATCH] Remove empty subcommands args; move empty terms/taxes, posts,
 and comments; insert default terms/taxes

---
 php/commands/blog.php | 133 +++++++++++++++++++++++++++++++-------------------
 1 file changed, 84 insertions(+), 49 deletions(-)

diff --git a/php/commands/blog.php b/php/commands/blog.php
index 7b046fb8..32229c68 100644
--- a/php/commands/blog.php
+++ b/php/commands/blog.php
@@ -8,30 +8,32 @@
 class Blog_Command extends WP_CLI_Command {
 
 	/**
-	 * Empty a blog
-	 *
-	 * @subcommand empty
-	 * @synopsis [--post_type=<post-type>] [--keep_terms] [--keep_comments]
+	 * Delete comments
 	 */
-	public function _empty( $args, $assoc_args ) {
-
-		$assoc_args = wp_parse_args( $assoc_args, array(
-			'post_type' => 'all',
-			'keep_terms' => 0,
-			'keep_comments' => 0,
-		) );
+	private function _empty_comments() {
+		global $wpdb;
 
-		WP_CLI::confirm( 'Are you sure you want to empty the blog at ' . site_url() . '?', $assoc_args );
+		// Empty comments and comment cache
+		$comment_ids = $wpdb->get_col( "SELECT comment_ID FROM $wpdb->comments" );
+		foreach ( $comment_ids as $comment_id ) {
+			wp_cache_delete( $comment_id, 'comment' );
+			wp_cache_delete( $comment_id, 'comment_meta' );
+		}
+		$wpdb->query( "TRUNCATE $wpdb->comments" );
+		$wpdb->query( "TRUNCATE $wpdb->commentmeta" );
+	}
 
+	/**
+	 * Delete all posts
+	 */
+	private function _empty_posts() {
 		global $wpdb;
 
 		// Empty posts and post cache
 		$posts_query = "SELECT ID FROM $wpdb->posts";
-		if ( 'all' != $assoc_args['post_type'] )
-			$posts_query .= " WHERE post_type='$assoc_args[post_type]'";
+		$posts = new WP_CLI\Iterators\Query( $posts_query, 10000 );
 
 		$taxonomies = get_taxonomies();
-		$posts = new WP_CLI\Iterators\Query( $posts_query, 10000 );
 
 		while ( $posts->valid() ) {
 			$post_id = $posts->current()->ID;
@@ -45,47 +47,80 @@ class Blog_Command extends WP_CLI_Command {
 			$posts->next();
 		}
 		$wpdb->query( "TRUNCATE $wpdb->posts" );
+		$wpdb->query( "TRUNCATE $wpdb->postmeta" );
+	}
 
-		if ( 'all' == $assoc_args['post_type'] )
-			$wpdb->query( "TRUNCATE $wpdb->postmeta" );
-
+	/**
+	 * Delete terms, taxonomies, and tax relationships
+	 */
+	private function _empty_taxonomies() {
+		global $wpdb;
 
-		// Empty comments and comment cache
-		if ( 0 == $assoc_args['keep_comments'] ) {
-			$comment_ids = $wpdb->get_col( "SELECT comment_ID FROM $wpdb->comments" );
-			foreach ( $comment_ids as $comment_id ) {
-				wp_cache_delete( $comment_id, 'comment' );
-				wp_cache_delete( $comment_id, 'comment_meta' );
-			}
-			$wpdb->query( "TRUNCATE $wpdb->comments" );
-			$wpdb->query( "TRUNCATE $wpdb->commentmeta" );
+		// Empty taxonomies and terms
+		$terms = $wpdb->get_results( "SELECT term_id, taxonomy FROM $wpdb->term_taxonomy" );
+		$ids = array();
+		foreach ( (array) $terms as $term ) {
+			$taxonomies[] = $term->taxonomy;
+			$ids[] = $term->term_id;
+			wp_cache_delete( $term->term_id, $term->taxonomy );
+		}
+		
+		$taxonomies = array_unique( $taxonomies );
+		foreach ( $taxonomies as $taxonomy ) {
+			if ( isset( $cleaned[$taxonomy] ) )
+				continue;
+			$cleaned[$taxonomy] = true;
+		
+			wp_cache_delete( 'all_ids', $taxonomy );
+			wp_cache_delete( 'get', $taxonomy );
+			delete_option( "{$taxonomy}_children" );
 		}
+		$wpdb->query( "TRUNCATE $wpdb->terms" );
+		$wpdb->query( "TRUNCATE $wpdb->term_taxonomy" );
+		$wpdb->query( "TRUNCATE $wpdb->term_relationships" );
+	}
 
-		// Empty taxonomies and terms
-		if ( 0 == $assoc_args['keep_terms'] ) {
-			$terms = $wpdb->get_results( "SELECT term_id, taxonomy FROM $wpdb->term_taxonomy" );
-			$ids = array();
-			foreach ( (array) $terms as $term ) {
-				$taxonomies[] = $term->taxonomy;
-				$ids[] = $term->term_id;
-				wp_cache_delete( $term->term_id, $term->taxonomy );
-			}
-		 
-			$taxonomies = array_unique( $taxonomies );
-			foreach ( $taxonomies as $taxonomy ) {
-				if ( isset( $cleaned[$taxonomy] ) )
-					continue;
-				$cleaned[$taxonomy] = true;
-		 
-				wp_cache_delete( 'all_ids', $taxonomy );
-				wp_cache_delete( 'get', $taxonomy );
-				delete_option( "{$taxonomy}_children" );
+	/**
+	 * Insert default terms
+	 */
+	private function _insert_default_terms() {
+		global $wpdb;
+
+		// Default category
+		$cat_name = __( 'Uncategorized' );
+
+		/* translators: Default category slug */
+		$cat_slug = sanitize_title( _x( 'Uncategorized', 'Default category slug' ) );
+
+		if ( global_terms_enabled() ) {
+			$cat_id = $wpdb->get_var( $wpdb->prepare( "SELECT cat_ID FROM {$wpdb->sitecategories} WHERE category_nicename = %s", $cat_slug ) );
+			if ( $cat_id == null ) {
+				$wpdb->insert( $wpdb->sitecategories, array('cat_ID' => 0, 'cat_name' => $cat_name, 'category_nicename' => $cat_slug, 'last_updated' => current_time('mysql', true)) );
+				$cat_id = $wpdb->insert_id;
 			}
-			$wpdb->query( "TRUNCATE $wpdb->terms" );
-			$wpdb->query( "TRUNCATE $wpdb->term_taxonomy" );
-			$wpdb->query( "TRUNCATE $wpdb->term_relationships" );
+			update_option('default_category', $cat_id);
+		} else {
+			$cat_id = 1;
 		}
 
+		$wpdb->insert( $wpdb->terms, array('term_id' => $cat_id, 'name' => $cat_name, 'slug' => $cat_slug, 'term_group' => 0) );
+		$wpdb->insert( $wpdb->term_taxonomy, array('term_id' => $cat_id, 'taxonomy' => 'category', 'description' => '', 'parent' => 0, 'count' => 1));
+	}
+
+	/**
+	 * Empty a blog
+	 *
+	 * @subcommand empty
+	 */
+	public function _empty( $args, $assoc_args ) {
+
+		WP_CLI::confirm( 'Are you sure you want to empty the blog at ' . site_url() . '?', $assoc_args );
+
+		$this->_empty_posts();
+		$this->_empty_comments();
+		$this->_empty_taxonomies();
+		$this->_insert_default_terms();
+
 		WP_CLI::success( 'The blog at ' . site_url() . ' was emptied.' );
 	}
 }
-- 
2.11.0