smartinject/ts/index.ts

import 'typings-global'

let Module = require('module')
import path = require('path')
import sourceMap = require('source-map-support')
import through2 = require('through2')
import vm = require('vm')

import { Transform } from 'stream'

sourceMap.install({
  retrieveFile: function (path) {
    if (cache[path]) {
      return cache[path].contents.toString()
    } else if (cache[path + '.js']) {
      return cache[path + '.js'].contents.toString()
    }
  }
})

let cache = {}
let originalLoader = Module._extensions['.js']

Module._extensions['.js'] = function (module, filename) {
  let file = cache[filename]
  let file2 = cache[filename + '.js']
  if (file) {
    module._compile(file.contents.toString(), filename)
  } else if (file2) {
    module._compile(file2.contents.toString(), filename)
  } else {
    originalLoader.apply(this, arguments)
  }
}

let originalModuleResolve = Module._resolveFilename

Module._resolveFilename = function (request, parent, isMain) {
  let resolvedRequest: string
  if (parent) {
    let parsedIdDir = path.parse(parent.id).dir
    resolvedRequest = path.join('/', parent.id, request)
  } else {
    resolvedRequest = request
  }
  // console.log('resolvedRequest:' + resolvedRequest)
  let file = cache[resolvedRequest]
  let file2 = cache[resolvedRequest + '.js']
  if (file || file2) {
    return request
  } else {
    return originalModuleResolve.apply(this, arguments)
  }
}

export let gulpPipe = function () {
  return through2.obj(function (file, enc, cb) {
    cache[file.path] = file
    return cb(null, file)
  })
}

export interface fileObject {
  path: string
  contents: Buffer
}

export let injectFileArray = async (fileArray: fileObject[]) => {
  for (let fileObject of fileArray) {
    // console.log('injected:' + fileObject.path)
    cache[fileObject.path] = fileObject
  }
  return fileArray
}