Parse refs as well

Author: tathamoddie

Logic/Commit.cs

@@ -4,5 +4,6 @@ public class Commit
     {
         public string Hash { get; set; }
         public string[] ParentHashes { get; set; }
+        public string[] Refs { get; set; }
     }
 }

Logic/LogParser.cs

@@ -1,6 +1,7 @@
 using System;
 using System.Collections.Generic;
 using System.IO;
+using System.Linq;
 using System.Text.RegularExpressions;
 
 namespace GitViz.Logic
@@ -20,7 +21,7 @@ public IEnumerable<Commit> ParseCommits(StreamReader gitLogOutput)
             gitLogOutput.Close();
         }
 
-        static readonly Regex ParseCommitRegex = new Regex(@"^(?<hash>\w{7})(?<parentHashes>( \w{7})+)?");
+        static readonly Regex ParseCommitRegex = new Regex(@"^(?<hash>\w{7})(?<parentHashes>( \w{7})+)?([ ]+\((?<refs>.*?)\))?");
 
         internal static Commit ParseCommit(string logOutputLine)
         {
@@ -30,10 +31,19 @@ internal static Commit ParseCommit(string logOutputLine)
                 ? match.Groups["parentHashes"].Value.Split(new[] {' '}, StringSplitOptions.RemoveEmptyEntries)
                 : null;
 
+            var refs = match.Groups["refs"].Success
+                ? match.Groups["refs"]
+                    .Value
+                    .Split(new[] { ',' }, StringSplitOptions.RemoveEmptyEntries)
+                    .Select(r => r.Trim())
+                    .ToArray()
+                : null;
+
             return new Commit
             {
                 Hash = match.Groups["hash"].Value,
-                ParentHashes = parentHashes
+                ParentHashes = parentHashes,
+                Refs = refs
             };
         }
     }

Tests/LogParserTests/ParseCommit.cs

@@ -10,6 +10,14 @@ public class ParseCommit
             "4be5ef1",
             Description = "Initial commit",
             Result = "{Hash:4be5ef1}")]
+        [TestCase(
+            "4be5ef1 (HEAD, master)",
+            Description = "Initial commit with head and master",
+            Result = "{Hash:4be5ef1,Refs:[HEAD,master]}")]
+        [TestCase(
+            "4be5ef1 (HEAD, origin/master, origin/HEAD, master)",
+            Description = "Initial commit with head and remote master",
+            Result = "{Hash:4be5ef1,Refs:[HEAD,origin/master,origin/HEAD,master]}")]
         [TestCase(
             "4e4224c 4be5ef1",
             Description = "Commit with one parent",

Tests/LogRetrieverTests.cs

@@ -45,6 +45,23 @@ public void ShouldReturnSingleCommitWithHashButNoParents()
             }
         }
 
+        [Test]
+        public void ShouldReturnSingleCommitWithLocalRefs()
+        {
+            using (var tempFolder = new TemporaryFolder())
+            {
+                var tempRepository = new TemporaryRepository(tempFolder);
+                tempRepository.RunCommand("init");
+                TouchFileAndCommit(tempRepository);
+
+                var executor = new GitCommandExecutor(tempFolder.Path);
+                var log = new LogRetriever(executor).GetLog().ToArray();
+
+                var commit = log.Single();
+                CollectionAssert.AreEqual(new[] { "HEAD", "master" }, commit.Refs);
+            }
+        }
+
         [Test]
         public void ShouldReturnTwoCommits()
         {