feat(traverse): visitor can return a leave callback

src/object/cloneDeep.ts

@@ -0,0 +1,53 @@
+import { traverse, TraverseContext } from 'radashi'
+
+export function cloneDeep<T extends object>(
+  root: T,
+  map: (obj: object, context: TraverseContext) => object,
+  filter: (obj: object, context: TraverseContext) => boolean = () => true,
+): T {
+  const out = map(root)
+  traverse(root, (value, key, parent, context) => {
+    if (value && typeof value === 'object') {
+    }
+  })
+}
+
+const complexTypes = [
+  Map,
+  Set,
+  Date,
+  RegExp,
+  Error,
+  WeakMap,
+  WeakSet,
+  ArrayBuffer,
+  SharedArrayBuffer,
+  DataView,
+  Int8Array,
+  Uint8Array,
+  Uint8ClampedArray,
+  Int16Array,
+  Uint16Array,
+  Int32Array,
+  Uint32Array,
+  Float32Array,
+  Float64Array,
+  BigInt64Array,
+  BigUint64Array,
+  Promise,
+  // Boxed primitive types
+  String,
+  Number,
+  Boolean,
+]
+
+function isInstance(
+  obj: object,
+  constructors: (new (...args: any[]) => any)[],
+) {
+  return constructors.some(constructor => obj instanceof constructor)
+}
+
+function isComplexObject(obj: object) {
+  return isInstance(obj, complexTypes)
+}

src/object/traverse.ts

@@ -1,4 +1,4 @@
-import { isArray, isIterable, isPlainObject, last } from 'radashi'
+import { isArray, isFunction, isIterable, isPlainObject, last } from 'radashi'
 
 /**
  * Iterate an object's properties and those of any nested objects.
@@ -66,14 +66,14 @@ export function traverse(
     }
 
     context.path.push(key)
-    if (
-      visitor(
-        (context.value = value),
-        (context.key = key),
-        context.parent,
-        context,
-      ) === false
-    ) {
+    const result = visitor(
+      (context.value = value),
+      (context.key = key),
+      context.parent,
+      context,
+    )
+
+    if (result === false) {
       return (ok = false)
     }
 
@@ -87,6 +87,10 @@ export function traverse(
       !context.parents.includes(value)
     ) {
       traverse(value)
+
+      if (isFunction(result)) {
+        result()
+      }
     }
 
     context.path.pop()
@@ -147,7 +151,7 @@ export type TraverseVisitor = (
   key: keyof any,
   parent: object,
   context: TraverseContext,
-) => boolean | void
+) => (() => void) | boolean | void
 
 /**
  * The context object for the `traverse` function.

tests/object/traverse.test.ts

@@ -301,4 +301,20 @@ describe('traverse', () => {
     expect(paths).toEqual([['a'], ['a', 'b']])
     expect(parents).toEqual([[obj], [obj, obj.a]])
   })
+
+  test('visitor can return a leave callback', () => {
+    const obj = { a: { b: 2 } }
+    const visited: [keyof any, unknown][] = []
+    _.traverse(obj, (value, key) => {
+      visited.push([key, value])
+      return () => {
+        visited.push([key, value])
+      }
+    })
+    expect(visited).toEqual([
+      ['a', { b: 2 }],
+      ['b', 2],
+      ['a', { b: 2 }],
+    ])
+  })
 })