forked from libgit2/libgit2sharp
-
Notifications
You must be signed in to change notification settings - Fork 2
/
BranchCollection.cs
198 lines (168 loc) · 7.55 KB
/
BranchCollection.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
using System;
using System.Collections;
using System.Collections.Generic;
using System.Globalization;
using System.Linq;
using LibGit2Sharp.Core;
namespace LibGit2Sharp
{
/// <summary>
/// The collection of Branches in a <see cref = "Repository" />
/// </summary>
public class BranchCollection : IEnumerable<Branch>
{
private readonly Repository repo;
/// <summary>
/// Needed for mocking purposes.
/// </summary>
protected BranchCollection()
{ }
/// <summary>
/// Initializes a new instance of the <see cref = "BranchCollection" /> class.
/// </summary>
/// <param name = "repo">The repo.</param>
internal BranchCollection(Repository repo)
{
this.repo = repo;
}
/// <summary>
/// Gets the <see cref = "LibGit2Sharp.Branch" /> with the specified name.
/// </summary>
public virtual Branch this[string name]
{
get
{
Ensure.ArgumentNotNullOrEmptyString(name, "name");
if (LooksLikeABranchName(name))
{
return BuildFromReferenceName(name);
}
Branch branch = BuildFromReferenceName(ShortToLocalName(name));
if (branch != null)
{
return branch;
}
branch = BuildFromReferenceName(ShortToRemoteName(name));
if (branch != null)
{
return branch;
}
return BuildFromReferenceName(ShortToRefName(name));
}
}
private static string ShortToLocalName(string name)
{
return string.Format(CultureInfo.InvariantCulture, "{0}{1}", "refs/heads/", name);
}
private static string ShortToRemoteName(string name)
{
return string.Format(CultureInfo.InvariantCulture, "{0}{1}", "refs/remotes/", name);
}
private static string ShortToRefName(string name)
{
return string.Format(CultureInfo.InvariantCulture, "{0}{1}", "refs/", name);
}
private Branch BuildFromReferenceName(string canonicalName)
{
var reference = repo.Refs.Resolve<Reference>(canonicalName);
return reference == null ? null : new Branch(repo, reference, canonicalName);
}
#region IEnumerable<Branch> Members
/// <summary>
/// Returns an enumerator that iterates through the collection.
/// </summary>
/// <returns>An <see cref = "IEnumerator{T}" /> object that can be used to iterate through the collection.</returns>
public virtual IEnumerator<Branch> GetEnumerator()
{
return Libgit2UnsafeHelper
.ListAllBranchNames(repo.Handle, GitBranchType.GIT_BRANCH_LOCAL | GitBranchType.GIT_BRANCH_REMOTE)
.Select(n => this[n])
.OrderBy(b => b.CanonicalName, StringComparer.Ordinal)
.GetEnumerator();
}
/// <summary>
/// Returns an enumerator that iterates through the collection.
/// </summary>
/// <returns>An <see cref = "IEnumerator" /> object that can be used to iterate through the collection.</returns>
IEnumerator IEnumerable.GetEnumerator()
{
return GetEnumerator();
}
#endregion
/// <summary>
/// Create a new local branch with the specified name
/// </summary>
/// <param name = "name">The name of the branch.</param>
/// <param name = "shaOrReferenceName">The target which can be sha or a canonical reference name.</param>
/// <param name = "allowOverwrite">True to allow silent overwriting a potentially existing branch, false otherwise.</param>
/// <returns></returns>
public virtual Branch Add(string name, string shaOrReferenceName, bool allowOverwrite = false)
{
Ensure.ArgumentNotNullOrEmptyString(name, "name");
ObjectId commitId = repo.LookupCommit(shaOrReferenceName).Id;
using (var osw = new ObjectSafeWrapper(commitId, repo))
{
GitOid oid;
Ensure.Success(NativeMethods.git_branch_create(out oid, repo.Handle, name, osw.ObjectPtr, allowOverwrite));
}
return this[ShortToLocalName(name)];
}
/// <summary>
/// Create a new local branch with the specified name
/// </summary>
/// <param name = "name">The name of the branch.</param>
/// <param name = "shaOrReferenceName">The target which can be sha or a canonical reference name.</param>
/// <param name = "allowOverwrite">True to allow silent overwriting a potentially existing branch, false otherwise.</param>
/// <returns></returns>
[Obsolete("This method will be removed in the next release. Please use Add() instead.")]
public virtual Branch Create(string name, string shaOrReferenceName, bool allowOverwrite = false)
{
return Add(name, shaOrReferenceName, allowOverwrite);
}
/// <summary>
/// Deletes the branch with the specified name.
/// </summary>
/// <param name = "name">The name of the branch to delete.</param>
/// <param name = "isRemote">True if the provided <paramref name="name"/> is the name of a remote branch, false otherwise.</param>
public virtual void Remove(string name, bool isRemote = false)
{
Ensure.ArgumentNotNullOrEmptyString(name, "name");
int res = NativeMethods.git_branch_delete(repo.Handle, name, isRemote ? GitBranchType.GIT_BRANCH_REMOTE : GitBranchType.GIT_BRANCH_LOCAL);
if (res == (int)GitErrorCode.NotFound)
{
return;
}
Ensure.Success(res);
}
/// <summary>
/// Deletes the branch with the specified name.
/// </summary>
/// <param name = "name">The name of the branch to delete.</param>
/// <param name = "isRemote">True if the provided <paramref name="name"/> is the name of a remote branch, false otherwise.</param>
[Obsolete("This method will be removed in the next release. Please use Remove() instead.")]
public virtual void Delete(string name, bool isRemote = false)
{
Remove(name, isRemote);
}
///<summary>
/// Rename an existing local branch with a new name.
///</summary>
///<param name = "currentName">The current branch name.</param>
///<param name = "newName">The new name of the existing branch should bear.</param>
///<param name = "allowOverwrite">True to allow silent overwriting a potentially existing branch, false otherwise.</param>
///<returns></returns>
public virtual Branch Move(string currentName, string newName, bool allowOverwrite = false)
{
Ensure.ArgumentNotNullOrEmptyString(currentName, "currentName");
Ensure.ArgumentNotNullOrEmptyString(newName, "name");
Ensure.Success(NativeMethods.git_branch_move(repo.Handle, currentName, newName, allowOverwrite));
return this[newName];
}
private static bool LooksLikeABranchName(string referenceName)
{
return referenceName == "HEAD" ||
referenceName.StartsWith("refs/heads/", StringComparison.Ordinal) ||
referenceName.StartsWith("refs/remotes/", StringComparison.Ordinal);
}
}
}