support coffee-script 1.7

db-migrate · Feb 17, 2014 · 901eba6 · 901eba6
2 parents 8580515 + 950f07c
commit 901eba6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/migration.js b/lib/migration.js
@@ -7,8 +7,10 @@ var log = require('./log');
 
 var filesRegEx = /\.js$/;
 var coffeeSupported = false;
+var coffeeModule = null;
 try {
-  require('coffee-script');
+  coffeeModule = require('coffee-script');
+  if (coffeeModule && coffeeModule.register) coffeeModule.register();
   coffeeSupported = true;
   filesRegEx = /\.(js|coffee)$/;
 } catch (e) {}