Move Checkout() to Repository

Closes #107

2 files changed, 24 insertions, 13 deletions

diff --git a/LibGit2Sharp/BranchCollection.cs b/LibGit2Sharp/BranchCollection.cs
index 422d7ee4..22954329 100644
--- a/LibGit2Sharp/BranchCollection.cs
+++ b/LibGit2Sharp/BranchCollection.cs
@@ -67,21 +67,10 @@ namespace LibGit2Sharp
         /// </summary>
         /// <param name = "shaOrReferenceName">The sha of the commit, a canonical reference name or the name of the branch to checkout.</param>
         /// <returns></returns>
+        [Obsolete("This method will be removed in the next release. Please use Repository.Checkout() instead.")]
         public Branch Checkout(string shaOrReferenceName)
         {
-            // TODO: This does not yet checkout (write) the working directory
-
-            Branch branch = this[shaOrReferenceName];
-
-            if (branch != null)
-            {
-                repo.Refs.UpdateTarget("HEAD", branch.CanonicalName);
-                return branch;
-            }
-
-            ObjectId commitId = repo.LookupCommit(shaOrReferenceName).Id;
-            repo.Refs.UpdateTarget("HEAD", commitId.Sha);
-            return repo.Head;
+            return repo.Checkout(shaOrReferenceName);
         }
 
         /// <summary>
diff --git a/LibGit2Sharp/Repository.cs b/LibGit2Sharp/Repository.cs
index 9aea29a9..e6f2ab39 100644
--- a/LibGit2Sharp/Repository.cs
+++ b/LibGit2Sharp/Repository.cs
@@ -342,6 +342,28 @@ namespace LibGit2Sharp
         }
 
         /// <summary>
+        ///   Checkout the specified branch, reference or SHA.
+        /// </summary>
+        /// <param name = "shaOrReferenceName">The sha of the commit, a canonical reference name or the name of the branch to checkout.</param>
+        /// <returns>The new HEAD.</returns>
+        public Branch Checkout(string shaOrReferenceName)
+        {
+            // TODO: This does not yet checkout (write) the working directory
+
+            var branch = Branches[shaOrReferenceName];
+
+            if (branch != null)
+            {
+                Refs.UpdateTarget("HEAD", branch.CanonicalName);
+                return branch;
+            }
+
+            var commitId = LookupCommit(shaOrReferenceName).Id;
+            Refs.UpdateTarget("HEAD", commitId.Sha);
+            return Head;
+        }
+
+        /// <summary>
         ///   Sets the current <see cref = "Head" /> to the specified commit and optionally resets the <see cref = "Index" /> and
         ///   the content of the working tree to match.
         /// </summary>