Full setup/reset by running setup.js

Author: scottgonzalez

package.json

@@ -14,7 +14,8 @@
     "mysql": "0.9.5",
     "semver": "1.0.12",
     "sqlite3": "2.1.1",
-    "step": "0.0.5"
+    "step": "0.0.5",
+    "rimraf": "1.0.9"
   },
   "engines": {
     "node": "0.6.5"

src/pluginsdb.js

@@ -88,5 +88,45 @@ var pluginsDb = module.exports = {
 
 	getAllActions: auto(function( fn ) {
 		db.all( "SELECT * FROM actions", fn );
-	})
+	}),
+
+	_reset: function( fn ) {
+		var fs = require( "fs" ),
+			Step = require( "step" );
+
+		Step(
+			function() {
+				fs.unlink( config.pluginsDb, this );
+			},
+
+			function( error ) {
+				if ( !error || error.code === "ENOENT" ) {
+					return connect( this );
+				}
+
+				fn( error );
+			},
+
+			function( error ) {
+				if ( error ) {
+					return fn( error );
+				}
+
+				db.run( "CREATE TABLE owners (" +
+					"plugin TEXT PRIMARY KEY, " +
+					"owner TEXT " +
+				")", this.parallel() );
+
+				db.run( "CREATE TABLE actions (" +
+					"id INTEGER PRIMARY KEY AUTOINCREMENT, " +
+					"action TEXT, " +
+					"data TEXT " +
+				")", this.parallel() );
+			},
+
+			function( error ) {
+				fn( error );
+			}
+		);
+	}
 };

src/setup.js

@@ -1,67 +1,23 @@
-var fs = require( "fs" ),
-	sqlite = require( "sqlite3" ),
-	config = require( "./config" ),
-	success = true;
-
-try {
-	fs.unlinkSync( config.pluginsDb );
-} catch( error ) {
-	if ( error.code !== "ENOENT" ) {
-		return logError( "Could not check status of, or delete, plugin.db " );
-	}
-}
-
-var db = new sqlite.Database( config.pluginsDb, function( error ) {
-	if ( error ) {
-		return logError( error, "Could not open database." );
-	}
-
-	setup();
-});
-
-function setup() {
-	createOwnersTable(function( error ) {
-		if ( error ) {
-			return logError( error, "Could not create owners table." );
-		}
-
-		console.log( "Created owners table." );
-	});
-
-	createActionsTable(function( error ) {
+var Step = require( "step" ),
+	rimraf = require( "rimraf" ),
+	pluginsDb = require( "./pluginsdb" ),
+	wordpress = require( "./wordpress" ),
+	config = require( "./config" );
+
+Step(
+	function() {
+		pluginsDb._reset( this.parallel() );
+		wordpress._reset( this.parallel() );
+		rimraf( config.repoDir, this.parallel() );
+	},
+
+	function( error ) {
 		if ( error ) {
-			return logError( error, "Could not create actions table." );
+			console.log( "ERROR", "Setup failed." );
+			console.log( error, error.stack );
+			return;
 		}
 
-		console.log( "Created actions table." );
-	});
-}
-
-function logError( error, msg ) {
-	console.log( "ERROR!", msg );
-	console.log( error );
-	success = false;
-}
-
-function createOwnersTable( fn ) {
-	db.run( "CREATE TABLE owners (" +
-		"plugin TEXT PRIMARY KEY, " +
-		"owner TEXT " +
-	");", fn );
-}
-
-function createActionsTable( fn ) {
-	db.run( "CREATE TABLE actions (" +
-		"id INTEGER PRIMARY KEY AUTOINCREMENT, " +
-		"action TEXT, " +
-		"data TEXT " +
-	");", fn );
-}
-
-process.on( "exit", function() {
-	if ( success ) {
-		console.log( "SUCCESS!", "Setup complete." );
-	} else {
-		console.log( "ERROR!", "Setup failed." );
+		console.log( "SUCCESS", "Setup complete." );
 	}
-});
+);

src/wordpress.js

@@ -261,5 +261,20 @@ var wordpress = module.exports = {
 			db.end();
 			db = null;
 		}
-	}
+	},
+
+	_reset: auto(function( fn ) {
+		Step(
+			function() {
+				db.query( "TRUNCATE TABLE `" + postsTable + "`", this.parallel() );
+				db.query( "TRUNCATE TABLE `" + postmetaTable + "`", this.parallel() );
+				wordpress.flush( this.parallel() );
+			},
+
+			function( error ) {
+				wordpress.end();
+				fn( error );
+			}
+		);
+	})
 };