Fixes #254. Use flatMap() instead of reduce() to flatten an array.

molefrog · Feb 26, 2024 · 67abe0e · 67abe0e
1 parent c5efa2e
commit 67abe0e
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/packages/wouter/src/index.js b/packages/wouter/src/index.js
@@ -220,17 +220,12 @@ export const Link = forwardRef((props, ref) => {
     : h("a", { ...restProps, href, onClick, children, ref });
 });
 
-const flattenChildren = (children) => {
-  return Array.isArray(children)
-    ? [].concat(
-        ...children.map((c) =>
-          c && c.type === Fragment
-            ? flattenChildren(c.props.children)
-            : flattenChildren(c)
-        )
+const flattenChildren = (children) =>
+  Array.isArray(children)
+    ? children.flatMap((c) =>
+        flattenChildren(c && c.type === Fragment ? c.props.children : c)
       )
     : [children];
-};
 
 export const Switch = ({ children, location }) => {
   const router = useRouter();