module.js 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131
  1. // thanks to airbnb/hypernova
  2. var NativeModule = require('module')
  3. var path = require('path')
  4. var assert = require('assert')
  5. var vm = require('vm')
  6. var NativeModules = process.binding('natives')
  7. var moduleExtensions = Object.assign({}, NativeModule._extensions)
  8. function isNativeModule (id) {
  9. return Object.prototype.hasOwnProperty.call(NativeModules, id)
  10. }
  11. // Creates a sandbox so we don't share globals across different runs.
  12. function createContext () {
  13. var sandbox = {
  14. Buffer,
  15. clearImmediate,
  16. clearInterval,
  17. clearTimeout,
  18. setImmediate,
  19. setInterval,
  20. setTimeout,
  21. console,
  22. process
  23. }
  24. sandbox.global = sandbox
  25. return sandbox
  26. }
  27. function Module (id, parent, isBundle) {
  28. var cache = parent ? parent.cache : null
  29. this.id = id
  30. this.exports = {}
  31. this.cache = cache || {}
  32. this.parent = parent
  33. this.filename = null
  34. this.loaded = false
  35. this.context = parent ? parent.context : createContext()
  36. this.isBundle = isBundle
  37. }
  38. Module.prototype.load = function (filename) {
  39. assert.ok(!this.loaded)
  40. this.filename = filename
  41. this.paths = NativeModule._nodeModulePaths(path.dirname(filename))
  42. }
  43. Module.prototype.run = function (filename) {
  44. var ext = path.extname(filename)
  45. var extension = moduleExtensions[ext] ? ext : '.js'
  46. moduleExtensions[extension](this, filename)
  47. this.loaded = true
  48. }
  49. Module.prototype.require = function (filePath) {
  50. assert.ok(typeof filePath === 'string', 'path must be a string')
  51. return Module.loadFile(filePath, this)
  52. }
  53. Module.prototype._compile = function (content, filename) {
  54. var self = this
  55. function r (filePath) {
  56. return self.require(filePath)
  57. }
  58. r.resolve = request => NativeModule._resolveFilename(request, this)
  59. r.main = process.mainModule
  60. r.extensions = moduleExtensions
  61. r.cache = this.cache
  62. var dirname = path.dirname(filename)
  63. // create wrapper function
  64. var wrapper = NativeModule.wrap(content)
  65. var options = {
  66. filename,
  67. displayErrors: true
  68. }
  69. var compiledWrapper = vm.runInNewContext(wrapper, this.context, options)
  70. return compiledWrapper.call(this.exports, this.exports, r, this, filename, dirname)
  71. }
  72. Module.load = function (id, filename) {
  73. var m = new Module(id)
  74. filename = filename || id
  75. m.load(filename)
  76. m.run(filename)
  77. return m
  78. }
  79. Module.loadFile = function (file, parent) {
  80. var filename = NativeModule._resolveFilename(file, parent)
  81. if (parent) {
  82. var cachedModule = parent.cache[filename]
  83. if (cachedModule) return cachedModule.exports
  84. }
  85. if (parent.isBundle || isNativeModule(filename)) {
  86. return require(filename)
  87. }
  88. var m = new Module(filename, parent)
  89. m.cache[filename] = m
  90. var hadException = true
  91. try {
  92. m.load(filename)
  93. m.run(filename)
  94. hadException = false
  95. } finally {
  96. if (hadException) {
  97. delete m.cache[filename]
  98. }
  99. }
  100. return m.exports
  101. }
  102. Module.addExtension = function (ext, f) {
  103. moduleExtensions[ext] = f
  104. }
  105. module.exports = Module