Added wp-restore.js to recover from a disk failure.

Author: scottgonzalez

src/pluginsdb.js

@@ -123,6 +123,18 @@ var pluginsDb = module.exports = {
 		db.get( "SELECT * FROM actions WHERE id > " + id + " ORDER BY id ASC LIMIT 1", fn );
 	}),
 
+	getAllRepos: auto(function( fn ) {
+		db.all( "SELECT DISTINCT(repo) FROM repos", function( error, rows ) {
+			if ( error ) {
+				return fn( error );
+			}
+
+			fn( null, rows.map(function( row ) {
+				return row.repo;
+			}));
+		});
+	}),
+
 	_reset: function( fn ) {
 		var fs = require( "fs" ),
 			Step = require( "step" );

src/service/github.js

@@ -57,7 +57,7 @@ extend( GithubRepo.prototype, {
 		Step(
 			// fetch the repo
 			function() {
-				repo.fetchRepo( this );
+				repo.fetch( this );
 			},
 
 			// get the tags
@@ -106,12 +106,16 @@ extend( GithubRepo.prototype, {
 
 			fn( null, new Date( stdout ) );
 		});
+	},
+
+	restore: function( fn ) {
+		this.fetch( fn );
 	}
 });
 
 // internals
 extend( GithubRepo.prototype, {
-	fetchRepo: function( fn ) {
+	fetch: function( fn ) {
 		var repo = this;
 
 		Step(
@@ -133,7 +137,7 @@ extend( GithubRepo.prototype, {
 			function( error ) {
 				// repo already exists
 				if ( !error ) {
-					exec( "git fetch -t", { cwd: repo.getPath() }, this );
+					return exec( "git fetch -t", { cwd: repo.getPath() }, this );
 				}
 
 				// error other than repo not existing

src/setup.js

@@ -6,6 +6,26 @@ var Step = require( "step" ),
 	config = require( "./config" );
 
 Step(
+	function() {
+		console.log( "Running setup will erase any existing data, including:\n" +
+			"* Plugins Database\n" +
+			"* WordPress Database\n" +
+			"* Local Repositories\n" );
+		console.log( "Are you sure you want to continue? (y/n)" );
+		process.stdin.resume();
+		process.stdin.setEncoding( "utf8" );
+		process.stdin.on( "data", this );
+	},
+
+	function( response ) {
+		process.stdin.destroy();
+		if ( response.trim().toLowerCase() !== "y" ) {
+			console.log( "Aborting setup. Nothing has been erased." );
+			process.exit();
+		}
+		this();
+	},
+
 	function() {
 		pluginsDb._reset( this.parallel() );
 		wordpress._reset( this.parallel() );
@@ -17,7 +37,7 @@ Step(
 	function( error ) {
 		if ( error ) {
 			console.log( "ERROR", "Setup failed." );
-			console.log( error, error.stack );
+			console.log( error.stack );
 			return;
 		}
 

src/wp-restore.js

@@ -0,0 +1,54 @@
+var Step = require( "step" ),
+	rimraf = require( "rimraf" ),
+	pluginsDb = require( "./pluginsdb" ),
+	wordpress = require( "./wordpress" ),
+	service = require( "./service" );
+
+Step(
+	function() {
+		console.log( "The WordPress restore script is for restoring the WordPress site.\n" +
+			"It will clear out the WordPress Database and restore the local repositories.\n" +
+			"You must have an existing Plugins Database for this to be useful.\n" );
+		console.log( "Are you sure you want to continue? (y/n)" );
+		process.stdin.resume();
+		process.stdin.setEncoding( "utf8" );
+		process.stdin.on( "data", this );
+	},
+
+	function( response ) {
+		process.stdin.destroy();
+		if ( response.trim().toLowerCase() !== "y" ) {
+			console.log( "Aborting restore." );
+			process.exit();
+		}
+		this();
+	},
+
+	function() {
+		pluginsDb.getAllRepos( this.parallel() );
+		wordpress._reset( this.parallel() );
+		rimraf( "last-action", this.parallel() );
+	},
+
+	function( error, repos ) {
+		if ( error ) {
+			throw error;
+		}
+
+		var group = this.group();
+		repos.forEach(function( repo ) {
+			service.getRepoById( repo ).restore( group() );
+		});
+	},
+
+	function( error ) {
+		if ( error ) {
+			console.log( "ERROR", "Restore failed." );
+			console.log( error.stack );
+			return;
+		}
+
+		console.log( "SUCCESS", "Restore complete." );
+		console.log( "Please run wp-update.js to start processing actions." );
+	}
+);