From b5985debb8e165de8949248ae7ac7d4426281cfd Mon Sep 17 00:00:00 2001
From: Christian Tellnes <christian@tellnes.no>
Date: Sat, 20 Feb 2016 19:31:55 +0100
Subject: [PATCH] always set context in transformFn to the transform stream

---
 src/transformTools.coffee | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/transformTools.coffee b/src/transformTools.coffee
index 46e7308..c69a395 100644
--- a/src/transformTools.coffee
+++ b/src/transformTools.coffee
@@ -172,6 +172,7 @@ exports.makeFalafelTransform = (transformName, options={}, transformFn) ->
     falafelOptions = options.falafelOptions ? {}
 
     transform = exports.makeStringTransform transformName, options, (content, transformOptions, done) ->
+        tr = @
         transformErr = null
         pending = 1 # We'll decrement this to zero at the end to prevent premature call of `done`.
         transformed = null
@@ -191,7 +192,7 @@ exports.makeFalafelTransform = (transformName, options={}, transformFn) ->
         transformed = falafel content, falafelOptions, (node) ->
             pending++
             try
-                transformFn node, transformOptions, transformCb
+                transformFn.call tr, node, transformOptions, transformCb
             catch err
                 transformCb err
 
@@ -268,7 +269,7 @@ exports.makeRequireTransform = (transformName, options={}, transformFn) ->
             else
                 args = (arg.source() for arg in node.arguments)
 
-            transformFn args, transformOptions, (err, transformed) ->
+            transformFn.call @, args, transformOptions, (err, transformed) ->
                 return done err if err
                 if transformed? then node.update(transformed)
                 done()