Add extension methods to determine reference kind

Saaman · nulltoken · commit 1465bf31cdfc · 2013-04-26T20:28:32.000+02:00
diff --git a/LibGit2Sharp.Tests/ReferenceFixture.cs b/LibGit2Sharp.Tests/ReferenceFixture.cs
@@ -739,5 +739,21 @@ private static T[] SortedRefs<T>(IRepository repo, Func<Reference, T> selector)
         {
             return repo.Refs.OrderBy(r => r.CanonicalName, StringComparer.Ordinal).Select(selector).ToArray();
         }
+
+        [Fact]
+        public void CanIdentifyReferenceKind()
+        {
+            using (var repo = new Repository(StandardTestRepoWorkingDirPath))
+            {
+                Assert.True(repo.Refs["refs/heads/master"].IsLocalBranch());
+                Assert.True(repo.Refs["refs/remotes/origin/master"].IsRemoteTrackingBranch());
+                Assert.True(repo.Refs["refs/tags/lw"].IsTag());
+            }
+
+            using (var repo = new Repository(BareTestRepoPath))
+            {
+                Assert.True(repo.Refs["refs/notes/commits"].IsNote());
+            }
+        }
     }
 }
diff --git a/LibGit2Sharp/ReferenceExtensions.cs b/LibGit2Sharp/ReferenceExtensions.cs
@@ -31,5 +31,45 @@ private static bool IsPrefixedBy(this string input, string prefix)
         {
             return input.StartsWith(prefix, StringComparison.Ordinal);
         }
+
+        /// <summary>
+        ///   Determine if the current <see cref="Reference"/> is a local branch.
+        /// </summary>
+        /// <param name="reference">The <see cref="Reference"/> to test.</param>
+        /// <returns>true if the current <see cref="Reference"/> is a local branch, false otherwise.</returns>
+        public static bool IsLocalBranch(this Reference reference)
+        {
+            return reference.CanonicalName.LooksLikeLocalBranch();
+        }
+
+        /// <summary>
+        ///   Determine if the current <see cref="Reference"/> is a remote tracking branch.
+        /// </summary>
+        /// <param name="reference">The <see cref="Reference"/> to test.</param>
+        /// <returns>true if the current <see cref="Reference"/> is a remote tracking branch, false otherwise.</returns>
+        public static bool IsRemoteTrackingBranch(this Reference reference)
+        {
+            return reference.CanonicalName.LooksLikeRemoteTrackingBranch();
+        }
+
+        /// <summary>
+        ///   Determine if the current <see cref="Reference"/> is a tag.
+        /// </summary>
+        /// <param name="reference">The <see cref="Reference"/> to test.</param>
+        /// <returns>true if the current <see cref="Reference"/> is a tag, false otherwise.</returns>
+        public static bool IsTag(this Reference reference)
+        {
+            return reference.CanonicalName.LooksLikeTag();
+        }
+
+        /// <summary>
+        ///   Determine if the current <see cref="Reference"/> is a note.
+        /// </summary>
+        /// <param name="reference">The <see cref="Reference"/> to test.</param>
+        /// <returns>true if the current <see cref="Reference"/> is a note, false otherwise.</returns>
+        public static bool IsNote(this Reference reference)
+        {
+            return reference.CanonicalName.LooksLikeNote();
+        }
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -739,5 +739,21 @@ private static T[] SortedRefs<T>(IRepository repo, Func<Reference, T> selector)`
`739`	`739`	`{`
`740`	`740`	`return repo.Refs.OrderBy(r => r.CanonicalName, StringComparer.Ordinal).Select(selector).ToArray();`
`741`	`741`	`}`
	`742`	`+`
	`743`	`+ [Fact]`
	`744`	`+ public void CanIdentifyReferenceKind()`
	`745`	`+ {`
	`746`	`+ using (var repo = new Repository(StandardTestRepoWorkingDirPath))`
	`747`	`+ {`
	`748`	`+ Assert.True(repo.Refs["refs/heads/master"].IsLocalBranch());`
	`749`	`+ Assert.True(repo.Refs["refs/remotes/origin/master"].IsRemoteTrackingBranch());`
	`750`	`+ Assert.True(repo.Refs["refs/tags/lw"].IsTag());`
	`751`	`+ }`
	`752`	`+`
	`753`	`+ using (var repo = new Repository(BareTestRepoPath))`
	`754`	`+ {`
	`755`	`+ Assert.True(repo.Refs["refs/notes/commits"].IsNote());`
	`756`	`+ }`
	`757`	`+ }`
`742`	`758`	`}`
`743`	`759`	`}`