Update FoldLeft/FoldRight transformations.

loophp · Aug 15, 2020 · cece391 · cece391
1 parent ddde5cc
commit cece391
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/Transformation/FoldLeft.php b/src/Transformation/FoldLeft.php
@@ -18,7 +18,7 @@ final class FoldLeft implements Transformation
 {
     /**
      * @var callable
-     * @psalm-var callable(T|null, T, TKey):(T|null)
+     * @psalm-var callable(T|null, T, TKey, \Iterator<TKey, T>):(T|null)
      */
     private $callback;
 
@@ -29,7 +29,7 @@ final class FoldLeft implements Transformation
     private $initial;
 
     /**
-     * @psalm-param callable(T|null, T, TKey):(T|null) $callback
+     * @psalm-param callable(T|null, T, TKey, \Iterator<TKey, T>):(T|null) $callback
      *
      * @param mixed|null $initial
      * @psalm-param T|null $initial
@@ -52,7 +52,7 @@ public function __invoke(Iterator $collection)
         $initial = $this->initial;
 
         foreach ($collection as $key => $value) {
-            $initial = $callback($initial, $value, $key);
+            $initial = $callback($initial, $value, $key, $collection);
         }
 
         return $initial;

diff --git a/src/Transformation/FoldRight.php b/src/Transformation/FoldRight.php
@@ -19,7 +19,7 @@ final class FoldRight implements Transformation
 {
     /**
      * @var callable
-     * @psalm-var callable(T|null, T|null, TKey):(T|null)
+     * @psalm-var callable(T|null, T|null, TKey, \Iterator<TKey, T>):(T|null)
      */
     private $callback;
 
@@ -30,7 +30,7 @@ final class FoldRight implements Transformation
     private $initial;
 
     /**
-     * @psalm-param callable(T|null, T|null, TKey):(T|null) $callback
+     * @psalm-param callable(T|null, T|null, TKey, \Iterator<TKey, T>):(T|null) $callback
      *
      * @param mixed|null $initial
      * @psalm-param T|null $initial