require explicit require.extensions registration

jashkenas/coffeescript#2323

Use CoffeeScript.register() or require('coffee-script-redux/register')

Author: michaelficarra

Makefile

@@ -9,7 +9,7 @@ ROOT = $(shell pwd)
 
 COFFEE = bin/coffee --js --bare
 PEGJS = node_modules/.bin/pegjs --cache --export-var 'module.exports'
-MOCHA = node_modules/.bin/mocha --compilers coffee:. -u tdd
+MOCHA = node_modules/.bin/mocha --compilers coffee:./register -u tdd
 CJSIFY = node_modules/.bin/cjsify --export CoffeeScript
 MINIFIER = node_modules/.bin/esmangle
 

lib/command.js

@@ -0,0 +1 @@
+require('./lib/register');

lib/module.js

@@ -269,6 +269,7 @@ else
 
     # --eval
     if options.eval
+      CoffeeScript.register()
       runMain input, js, jsAST, inputSource
       return
 

lib/register.js

@@ -32,8 +32,9 @@ escodegenCompactDefaults =
   semicolons: no
 
 
-module.exports =
+CoffeeScript =
 
+  CoffeeScript: CoffeeScript
   Compiler: Compiler
   Optimiser: Optimiser
   Parser: Parser
@@ -82,8 +83,7 @@ module.exports =
     CoffeeScript.js jsAST, compact: options.compact or options.minify
 
 
-CoffeeScript = module.exports.CoffeeScript = module.exports
+module.exports = CoffeeScript
 
-if (process.title == 'node')
-  noBrowserifyRequire = require
-  noBrowserifyRequire './run'
+if process.title is 'node'
+  CoffeeScript.register = -> require './register'

lib/run.js

@@ -0,0 +1,13 @@
+fs = require 'fs'
+CoffeeScript = require './module'
+{runModule} = require './run'
+
+module.exports = not require.extensions['.coffee']?
+
+require.extensions['.coffee'] ?= (module, filename) ->
+  input = fs.readFileSync filename, 'utf8'
+  csAst = CoffeeScript.parse input, raw: yes
+  jsAst = CoffeeScript.compile csAst
+  js = CoffeeScript.js jsAst
+
+  runModule module, js, jsAst, filename

register.js

@@ -1,6 +1,5 @@
 # Node.js-specific support: module loading, sourceMapping errors
 
-fs = require 'fs'
 path = require 'path'
 Module = require 'module'
 CoffeeScript = require './module'
@@ -97,32 +96,20 @@ formatSourcePosition = (frame, getSourceMapping) ->
     fileLocation
 
 # Run JavaScript as a main program - resetting process.argv and module lookup paths
-exports.runMain = (csSource, jsSource, jsAst, filename) ->
+runMain = (csSource, jsSource, jsAst, filename) ->
   mainModule = new Module '.'
   mainModule.filename = process.argv[1] = filename
-
   # Set it as the main module -- this is used for require.main
   process.mainModule = mainModule
-
   # Add the module to the cache
   Module._cache[mainModule.filename] = mainModule
-
   # Assign paths for node_modules loading
   mainModule.paths = Module._nodeModulePaths path.dirname filename
-
   runModule mainModule, jsSource, jsAst, filename
 
 runModule = (module, jsSource, jsAst, filename) ->
   do patchStackTrace
-
   Module._sourceMaps[filename] = -> "#{CoffeeScript.sourceMap jsAst, filename}"
-
   module._compile jsSource, filename
 
-require.extensions['.coffee'] = (module, filename) ->
-  input = fs.readFileSync filename, 'utf8'
-  csAst = CoffeeScript.parse input, raw: yes
-  jsAst = CoffeeScript.compile csAst
-  js = CoffeeScript.js jsAst
-
-  runModule module, js, jsAst, filename
+module.exports = {runMain, runModule}