Add TreeDefinition.AddGitLink()

1 files changed, 18 insertions, 0 deletions

diff --git a/LibGit2Sharp/TreeDefinition.cs b/LibGit2Sharp/TreeDefinition.cs
index f312a9a2..950e286b 100644
--- a/LibGit2Sharp/TreeDefinition.cs
+++ b/LibGit2Sharp/TreeDefinition.cs
@@ -174,6 +174,23 @@ namespace LibGit2Sharp
             return Add(targetTreeEntryPath, ted);
         }
 
+        /// <summary>
+        ///   Adds or replaces a gitlink <see cref="TreeEntryDefinition"/>,
+        ///   referencing the commit identified by <paramref name="objectId"/>,
+        ///   at the specified <paramref name="targetTreeEntryPath"/> location.
+        /// </summary>
+        /// <param name="targetTreeEntryPath">The path within this <see cref="TreeDefinition"/>.</param>
+        /// <param name="objectId">The <see cref="ObjectId"/> of the commit to be linked at the described location.</param>
+        /// <returns>The current <see cref="TreeDefinition"/>.</returns>
+        public virtual TreeDefinition AddGitLink(string targetTreeEntryPath, ObjectId objectId)
+        {
+            Ensure.ArgumentNotNull(objectId, "objectId");
+
+            var ted = TreeEntryDefinition.From(objectId);
+
+            return Add(targetTreeEntryPath, ted);
+        }
+
         private TreeDefinition RetrieveOrBuildTreeDefinition(string treeName, bool shouldOverWrite)
         {
             TreeDefinition td;
@@ -195,6 +212,7 @@ namespace LibGit2Sharp
                         break;
 
                     case GitObjectType.Blob:
+                    case GitObjectType.Commit:
                         if (shouldOverWrite)
                         {
                             td = new TreeDefinition();