From 5a47e03f8f76f40693c8ebbc5ca6a7de48e176a0 Mon Sep 17 00:00:00 2001
From: Patrick-Beuks <code@beuks.net>
Date: Mon, 6 May 2024 22:30:56 +0200
Subject: [PATCH] Add optional entry type option for tree entries (#221)

* Add optional entry type option for tree entries

* Fix cs problems

* Update test for diff range counts

* Make each entry type a different function
---
 src/Gitonomy/Git/Tree.php             | 48 ++++++++++++++++++++++++---
 tests/Gitonomy/Git/Tests/DiffTest.php |  2 +-
 tests/Gitonomy/Git/Tests/TreeTest.php | 41 ++++++++++++++++++++++-
 3 files changed, 84 insertions(+), 7 deletions(-)

diff --git a/src/Gitonomy/Git/Tree.php b/src/Gitonomy/Git/Tree.php
index 570c8ff8..7830cfa4 100644
--- a/src/Gitonomy/Git/Tree.php
+++ b/src/Gitonomy/Git/Tree.php
@@ -24,6 +24,7 @@ class Tree
     protected $hash;
     protected $isInitialized = false;
     protected $entries;
+    protected $entriesByType;
 
     public function __construct(Repository $repository, $hash)
     {
@@ -47,31 +48,68 @@ protected function initialize()
         $parser->parse($output);
 
         $this->entries = [];
+        $this->entriesByType = [
+            'blob'   => [],
+            'tree'   => [],
+            'commit' => [],
+        ];
 
         foreach ($parser->entries as $entry) {
             list($mode, $type, $hash, $name) = $entry;
             if ($type == 'blob') {
-                $this->entries[$name] = [$mode, $this->repository->getBlob($hash)];
+                $treeEntry = [$mode, $this->repository->getBlob($hash)];
             } elseif ($type == 'tree') {
-                $this->entries[$name] = [$mode, $this->repository->getTree($hash)];
+                $treeEntry = [$mode, $this->repository->getTree($hash)];
             } else {
-                $this->entries[$name] = [$mode, new CommitReference($hash)];
+                $treeEntry = [$mode, new CommitReference($hash)];
             }
+            $this->entries[$name] = $treeEntry;
+            $this->entriesByType[$type][$name] = $treeEntry;
         }
 
         $this->isInitialized = true;
     }
 
     /**
-     * @return array An associative array name => $object
+     * @return array<string, array{string, CommitReference|Tree|Blob}> An associative array name => $object
      */
-    public function getEntries()
+    public function getEntries(): array
     {
         $this->initialize();
 
         return $this->entries;
     }
 
+    /**
+     * @return array<string, array{string, CommitReference}> An associative array of name => [mode, commit reference]
+     */
+    public function getCommitReferenceEntries(): array
+    {
+        $this->initialize();
+
+        return $this->entriesByType['commit'];
+    }
+
+    /**
+     * @return array<string, array{string, Tree}> An associative array of name => [mode, tree]
+     */
+    public function getTreeEntries(): array
+    {
+        $this->initialize();
+
+        return $this->entriesByType['tree'];
+    }
+
+    /**
+     * @return array<string, array{string, Blob}> An associative array of name => [mode, blob]
+     */
+    public function getBlobEntries(): array
+    {
+        $this->initialize();
+
+        return $this->entriesByType['blob'];
+    }
+
     public function getEntry($name)
     {
         $this->initialize();
diff --git a/tests/Gitonomy/Git/Tests/DiffTest.php b/tests/Gitonomy/Git/Tests/DiffTest.php
index e251d72e..2232345f 100644
--- a/tests/Gitonomy/Git/Tests/DiffTest.php
+++ b/tests/Gitonomy/Git/Tests/DiffTest.php
@@ -139,7 +139,7 @@ public function testDiffRangeParse($repository)
         $this->assertSame(0, $changes[0]->getRangeOldCount());
 
         $this->assertSame(1, $changes[0]->getRangeNewStart());
-        $this->assertSame(0, $changes[0]->getRangeNewCount());
+        $this->assertSame(1, $changes[0]->getRangeNewCount());
     }
 
     /**
diff --git a/tests/Gitonomy/Git/Tests/TreeTest.php b/tests/Gitonomy/Git/Tests/TreeTest.php
index 4c0476de..b7ba7d32 100644
--- a/tests/Gitonomy/Git/Tests/TreeTest.php
+++ b/tests/Gitonomy/Git/Tests/TreeTest.php
@@ -13,6 +13,7 @@
 namespace Gitonomy\Git\Tests;
 
 use Gitonomy\Git\Blob;
+use Gitonomy\Git\CommitReference;
 
 class TreeTest extends AbstractTest
 {
@@ -21,7 +22,7 @@ class TreeTest extends AbstractTest
     /**
      * @dataProvider provideFooBar
      */
-    public function testCase($repository)
+    public function testGetEntries($repository)
     {
         $tree = $repository->getCommit(self::LONGFILE_COMMIT)->getTree();
 
@@ -34,6 +35,44 @@ public function testCase($repository)
         $this->assertTrue($entries['README.md'][1] instanceof Blob, 'README.md is a Blob');
     }
 
+    /**
+     * @dataProvider provideFooBar
+     */
+    public function testGetCommitReferenceEntries($repository)
+    {
+        $tree = $repository->getCommit(self::NO_MESSAGE_COMMIT)->getTree();
+
+        $commits = $tree->getCommitReferenceEntries();
+
+        $this->assertNotEmpty($commits['barbaz'], 'barbaz is present');
+        $this->assertTrue($commits['barbaz'][1] instanceof CommitReference, 'barbaz is a Commit');
+    }
+
+    /**
+     * @dataProvider provideFooBar
+     */
+    public function testGetTreeEntries($repository)
+    {
+        $tree = $repository->getCommit(self::NO_MESSAGE_COMMIT)->getTree();
+
+        $trees = $tree->getTreeEntries();
+
+        $this->assertEmpty($trees);
+    }
+
+    /**
+     * @dataProvider provideFooBar
+     */
+    public function testGetBlobEntries($repository)
+    {
+        $tree = $repository->getCommit(self::NO_MESSAGE_COMMIT)->getTree();
+
+        $blobs = $tree->getBlobEntries();
+
+        $this->assertNotEmpty($blobs['README.md'], 'README.md is present');
+        $this->assertTrue($blobs['README.md'][1] instanceof Blob, 'README.md is a blob');
+    }
+
     /**
      * @dataProvider provideFooBar
      */