Update flattenDepth

vadimfilimonov · Jul 23, 2023 · b74dfe4 · b74dfe4
1 parent 7b12a27
commit b74dfe4
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/src/flattenDepth.js b/src/flattenDepth.js
@@ -16,11 +16,21 @@ const flattenDepth = (array, depth = 1) => {
     return [];
   }
 
-  if (depth <= 0) {
-    return array;
-  }
+  const iter = (items, currentDepth) => {
+    if (currentDepth > depth) {
+      return items;
+    }
+
+    return items.reduce((acc, item) => {
+      if (!Array.isArray(item)) {
+        return [...acc, item];
+      }
+
+      return [...acc, ...iter(item, currentDepth + 1)];
+    }, []);
+  };
 
-  return flattenDepth(array.flat(), depth - 1);
+  return iter(array, 1);
 };
 
 export default flattenDepth;