use return values to auto-update

ljharb · Sep 4, 2010 · bc68fa5 · bc68fa5
1 parent 78edd30
commit bc68fa5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lib/traverse.js b/lib/traverse.js
@@ -82,7 +82,10 @@ function Traverse (refObj) {
             state.notLeaf = !state.isLeaf;
             state.notRoot = !state.isRoot;
 
-            f.call(state, node);
+            // use return values to update if defined
+            var ret = f.call(state, node);
+            if (ret !== undefined) state.update(ret);
+
             if (typeof state.node == 'object'
             && state.node !== null && !state.circular) {
                 parents.push(state);