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()
    }
  }
})


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

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

let originalModuleResolve = Module._resolveFilename

Module._resolveFilename = function (request, parent, isMain) {
  let file = cache[request];
  if (file) {
    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) {
    cache[fileObject.path] = fileObject
  }
  return fileArray
}