From f1f42164fcf212defa7cadab307151061cb9668c Mon Sep 17 00:00:00 2001 From: Sebastian McKenzie Date: Sun, 8 Nov 2015 03:41:00 -0800 Subject: [PATCH] add toString to allow reflection of original traversal methods when wrapping for state --- packages/babel-traverse/src/visitors.js | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/packages/babel-traverse/src/visitors.js b/packages/babel-traverse/src/visitors.js index 39b780528b..81df24e88d 100644 --- a/packages/babel-traverse/src/visitors.js +++ b/packages/babel-traverse/src/visitors.js @@ -174,9 +174,11 @@ function wrapWithState(oldVisitor, state) { if (!Array.isArray(fns)) continue; fns = fns.map(function (fn) { - return function (path) { + let newFn = function (path) { return fn.call(state, path, state); }; + newFn.toString = () => fn.toString(); + return newFn; }); newVisitor[key] = fns;