Resolve driver from local projects

Fixes #318

Author: wzrdtales

lib/driver/index.js

@@ -8,7 +8,8 @@ var log = internals.mod.log;
 var tunnel = require('tunnel-ssh'),
     Promise = require('bluebird'),
     SeederInterface = require('../interface/seederInterface.js'),
-    MigratorInterface = require('../interface/migratorInterface.js');
+    MigratorInterface = require('../interface/migratorInterface.js'),
+    resolve = require( 'resolve' );
 
 var ShadowProto = {
 
@@ -44,6 +45,7 @@ exports.connect = function (config, intern, callback) {
     log.verbose('require:', config.driver.require);
     driver = require(config.driver.require);
 
+    _connect();
   }
   else {
 
@@ -58,37 +60,54 @@ exports.connect = function (config, intern, callback) {
       break;
     }
 
-    try {
+    req = 'db-migrate-' + config.driver;
+    log.verbose('require:', req);
 
-      req = 'db-migrate-' + config.driver;
-      log.verbose('require:', req);
+    resolve(req, {
 
-      try {
+    basedir: process.cwd()
+    }, function (error, localModule) {
 
-        driver = require(req);
+      try {
+        if ( error ) {
+          try {
+
+            driver = require(req);
+          }
+          catch (Exception) {
+
+            driver = require('../../../' + req);
+          }
+        }
+        else {
+          driver = require( localModule );
+        }
       }
       catch (Exception) {
 
-        driver = require('../../../' + req);
+        try {
+
+          //Fallback to internal drivers, while moving drivers to new repos
+          req = './' + config.driver;
+          log.verbose('require:', req);
+          driver = require(req);
+        }
+        catch (exception) {
+
+          return callback( {
+            stack: 'No such driver found, please try to install it via ' +
+            'npm install db-migrate-' + config.driver + ' or ' +
+            'npm install -g db-migrate-' + config.driver
+          } );
+        }
       }
-    }
-    catch (Exception) {
 
-      try {
+      _connect();
+    });
+  }
 
-        //Fallback to internal drivers, while moving drivers to new repos
-        req = './' + config.driver;
-        log.verbose('require:', req);
-        driver = require(req);
-      }
-      catch (Exception) {
 
-        return callback( { stack: 'No such driver found, please try to install it via ' +
-          'npm install db-migrate-' + config.driver + ' or ' +
-          'npm install -g db-migrate-' + config.driver  } );
-      }
-    }
-  }
+function _connect() {
 
   log.verbose('connecting');
 
@@ -136,4 +155,5 @@ exports.connect = function (config, intern, callback) {
   else {
     connect(config);
   }
+}
 };