From e7e969cf656fd7024f20930da8fdfe7ba7bf2ba1 Mon Sep 17 00:00:00 2001 From: bendtherules Date: Wed, 11 Sep 2019 16:54:32 +0530 Subject: [PATCH] traverse - Support traverseConfig --- src/traverse.ts | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/traverse.ts b/src/traverse.ts index 6e20e14..c6b7f47 100644 --- a/src/traverse.ts +++ b/src/traverse.ts @@ -136,8 +136,12 @@ function* traverseGenerator( * ``` * */ -function traverse(node: FiberNode, fn: (node: FiberNode) => any) { - const nodeIterator = traverseGenerator(node); +function traverse( + node: FiberNode, + fn: (node: FiberNode) => any, + traverseConfig?: TTraverseConfig +) { + const nodeIterator = traverseGenerator(node, traverseConfig); for (const tmpNode of nodeIterator) { fn.call(null, tmpNode); }