Simplify

sanmai · Apr 11, 2023 · 6dbd089 · 6dbd089
1 parent 4f57bed
commit 6dbd089
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/Standard.php b/src/Standard.php
@@ -204,9 +204,9 @@ private static function joinYield(iterable $left, iterable $right): iterable
      */
     public function unpack(?callable $func = null): self
     {
-        $func = $func ?? static function (...$args) {
-            yield from $args;
-        };
+        if (null === $func) {
+            return $this->flatten();
+        }
 
         return $this->map(static function (iterable $args = []) use ($func) {
             /** @psalm-suppress InvalidArgument */
@@ -219,7 +219,9 @@ public function unpack(?callable $func = null): self
      */
     public function flatten(): self
     {
-        return $this->unpack();
+        return $this->map(static function (iterable $args = []) {
+            yield from $args;
+        });
     }
 
     /**