learnyounode/exercises/my_first_io/wrap.js

Learn You The Node.js For Much Win! An intro to Node.js via a set of self-guided workshops.

var fs = require('fs')


function wrap (ctx) {
  ctx.fsCalls = {}

  // wrap app fs calls
  Object.keys(fs).forEach(function (m) {
    var orig = fs[m]

    fs[m] = function () {
      // $captureStack is a utility to capture a stacktrace array
      var stack = ctx.$captureStack(fs[m])

      // inspect the first callsite of the stacktrace and see if the
      // filename matches the mainProgram we're running, if so, then
      // the user has used the method in question
      // the substring() is necessary as the user doesn't have to provide
      // a .js extension to make it work

      if (stack[0].getFileName().substring(0, ctx.mainProgram.length) == ctx.mainProgram) {
        if (!ctx.fsCalls[m])
          ctx.fsCalls[m] = 1
        else
          ctx.fsCalls[m]++
      }

      // call the real fs.readFileSync

      return orig.apply(this, arguments)
    }
  })
}


module.exports = wrap