From 9ba81e33e389cb80b80ce114fb664506fab1859e Mon Sep 17 00:00:00 2001
From: Andrey Kutejko <andy128k@gmail.com>
Date: Wed, 20 Mar 2019 00:45:20 +0100
Subject: [PATCH] add map methods

---
 src/map.php   | 21 +++++++++++++++
 t/MapTest.php | 74 +++++++++++++++++++++++++++++++++++++--------------
 2 files changed, 75 insertions(+), 20 deletions(-)

diff --git a/src/map.php b/src/map.php
index 6af1f37..9e4dbcf 100644
--- a/src/map.php
+++ b/src/map.php
@@ -49,6 +49,27 @@ class Map
         });
     }
 
+    function update($key, $updater)
+    {
+        foreach ($this->data as &$entry) {
+            if ($this->keysAreEqual($key, $entry[0])) {
+                $entry[1] = call_user_func($updater, $entry[1]);
+                return;
+            }
+        }
+    }
+
+    function updateOrInsert($key, $updater, $valueToInsert)
+    {
+        foreach ($this->data as &$entry) {
+            if ($this->keysAreEqual($key, $entry[0])) {
+                $entry[1] = call_user_func($updater, $entry[1]);
+                return;
+            }
+        }
+        $this->data[] = [$key, $valueToInsert];
+    }
+
     function __debugInfo()
     {
         $i = [];
diff --git a/t/MapTest.php b/t/MapTest.php
index 6b11b27..d7db531 100644
--- a/t/MapTest.php
+++ b/t/MapTest.php
@@ -4,30 +4,30 @@ class MapTest extends PHPUnit_Framework_TestCase
 {
     public function testGet()
     {
-        $arr = new \PFF\Map;
-        $arr->set(['fruit', 'apple'], 100);
-        $arr->set(['fruit', 'grapefruit'], 400);
-        $arr->set(['vegetable', 'carrot'], 50);
-
-        $this->assertEquals(3, $arr->count());
-        $this->assertEquals(true, $arr->contains(['fruit', 'grapefruit']));
-        $this->assertEquals(400, $arr->get(['fruit', 'grapefruit']));
-        $this->assertEquals(null, $arr->get(['fruit', 'orange']));
-        $this->assertEquals(false, $arr->contains(['apple', 'fruit']));
-        $this->assertEquals(null, $arr->get(['apple', 'fruit']));
-
-        $this->assertEquals(50, $arr->get(['vegetable', 'carrot']));
-        $arr->set(['vegetable', 'carrot'], 150);
-        $this->assertEquals(150, $arr->get(['vegetable', 'carrot']));
-        $arr->remove(['vegetable', 'carrot']);
-        $this->assertEquals(false, $arr->contains(['vegetable', 'carrot']));
+        $map = new \PFF\Map;
+        $map->set(['fruit', 'apple'], 100);
+        $map->set(['fruit', 'grapefruit'], 400);
+        $map->set(['vegetable', 'carrot'], 50);
+
+        $this->assertEquals(3, $map->count());
+        $this->assertEquals(true, $map->contains(['fruit', 'grapefruit']));
+        $this->assertEquals(400, $map->get(['fruit', 'grapefruit']));
+        $this->assertEquals(null, $map->get(['fruit', 'orange']));
+        $this->assertEquals(false, $map->contains(['apple', 'fruit']));
+        $this->assertEquals(null, $map->get(['apple', 'fruit']));
+
+        $this->assertEquals(50, $map->get(['vegetable', 'carrot']));
+        $map->set(['vegetable', 'carrot'], 150);
+        $this->assertEquals(150, $map->get(['vegetable', 'carrot']));
+        $map->remove(['vegetable', 'carrot']);
+        $this->assertEquals(false, $map->contains(['vegetable', 'carrot']));
     }
 
     public function testDebugInfo()
     {
-        $arr = new \PFF\Map;
-        $arr->set(['fruit', 'apple'], 100);
-        $output = $this->varDumpToString($arr);
+        $map = new \PFF\Map;
+        $map->set(['fruit', 'apple'], 100);
+        $output = $this->varDumpToString($map);
         $this->assertContains('fruit', $output);
         $this->assertContains('apple', $output);
         $this->assertContains('100', $output);
@@ -41,4 +41,38 @@ class MapTest extends PHPUnit_Framework_TestCase
         ob_end_clean();
         return $output;
     }
+
+    public function testUpdate()
+    {
+        $map = new \PFF\Map;
+        $map->set('carrot', 1);
+        $map->update('carrot', function ($c) {
+            return $c + 1;
+        });
+        $map->update('tomato', function ($c) {
+            return $c + 1;
+        });
+        $this->assertEquals(2, $map->get('carrot'));
+        $this->assertEquals(null, $map->get('tomato'));
+    }
+
+    public function testUpdateOrInsert()
+    {
+        $map = new \PFF\Map;
+        $map->set('carrot', 1);
+        $map->updateOrInsert('carrot', function ($c) {return $c + 1;}, 1);
+        $map->updateOrInsert('tomato', function ($c) {return $c + 1;}, 1);
+        $this->assertEquals(2, $map->get('carrot'));
+        $this->assertEquals(1, $map->get('tomato'));
+    }
+
+    public function testUpdateOrInsertArray()
+    {
+        $map = new \PFF\Map;
+        $map->set('carrot', [1]);
+        $map->updateOrInsert('carrot', function ($c) {return array_merge($c, [1]);}, [1]);
+        $map->updateOrInsert('tomato', function ($c) {return array_merge($c, [1]);}, [1]);
+        $this->assertEquals([1, 1], $map->get('carrot'));
+        $this->assertEquals([1], $map->get('tomato'));
+    }
 }
-- 
2.49.0