Skip to content
This repository was archived by the owner on Dec 5, 2024. It is now read-only.

Commit 3e4f470

Browse files
StanleyGoldmanStanleyGoldman
committed
Merge branch 'enhancements/tree-improvements' into enhancements/changes-tree-view
2 parents 2cec819 + 172d5d0 commit 3e4f470

File tree

18 files changed

+423
-138
lines changed

18 files changed

+423
-138
lines changed

‎src/GitHub.Api/Cache/CacheInterfaces.cs

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,9 @@ public interface IGitUserCache : IManagedCache
5555

5656
public interface IGitStatusCache : IManagedCache
5757
{
58-
GitStatus GitStatus { get; set; }
58+
int Ahead { get; set; }
59+
int Behind { get; set; }
60+
List<GitStatusEntry> Entries { get; set; }
5961
}
6062

6163
public interface ILocalConfigBranchDictionary : IDictionary<string, ConfigBranch>

‎src/GitHub.Api/Extensions/GitStatusExtensions.cs

Lines changed: 0 additions & 13 deletions
This file was deleted.

‎src/GitHub.Api/Git/GitAheadBehindStatus.cs

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
using System;
2+
3+
namespace GitHub.Unity
4+
{
5+
[Serializable]
6+
public struct GitAheadBehindStatus
7+
{
8+
public static GitAheadBehindStatus Default = new GitAheadBehindStatus();
9+
10+
public int ahead;
11+
public int behind;
12+
13+
public GitAheadBehindStatus(int ahead, int behind)
14+
{
15+
this.ahead = ahead;
16+
this.behind = behind;
17+
}
18+
19+
public int Ahead => ahead;
20+
21+
public int Behind => behind;
22+
}
23+
}

‎src/GitHub.Api/Git/GitClient.cs

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,9 @@ public interface IGitClient
1515

1616
ITask LfsInstall(IOutputProcessor<string> processor = null);
1717

18+
ITask<GitAheadBehindStatus> AheadBehindStatus(string gitRef, string otherRef,
19+
IOutputProcessor<GitAheadBehindStatus> processor = null);
20+
1821
ITask<GitStatus> Status(IOutputProcessor<GitStatus> processor = null);
1922

2023
ITask<string> GetConfig(string key, GitConfigSource configSource,
@@ -219,6 +222,14 @@ public ITask<GitStatus> Status(IOutputProcessor<GitStatus> processor = null)
219222
.Configure(processManager);
220223
}
221224

225+
public ITask<GitAheadBehindStatus> AheadBehindStatus(string gitRef, string otherRef, IOutputProcessor<GitAheadBehindStatus> processor = null)
226+
{
227+
Logger.Trace("AheadBehindStatus");
228+
229+
return new GitAheadBehindStatusTask(gitRef, otherRef, cancellationToken, processor)
230+
.Configure(processManager);
231+
}
232+
222233
public ITask<List<GitLogEntry>> Log(BaseOutputListProcessor<GitLogEntry> processor = null)
223234
{
224235
Logger.Trace("Log");

‎src/GitHub.Api/Git/GitRemote.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ public enum GitRemoteFunction
1414
[Serializable]
1515
public struct GitRemote
1616
{
17-
public static GitRemote Default = new GitRemote();
17+
public static GitRemote Default = new GitRemote(String.Empty, String.Empty, String.Empty, GitRemoteFunction.Unknown, string.Empty, string.Empty, string.Empty);
1818

1919
public string name;
2020
public string url;

‎src/GitHub.Api/Git/IRepository.cs

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,9 @@ public interface IRepository : IEquatable<IRepository>
5757
/// Gets the current branch of the repository.
5858
/// </summary>
5959
GitBranch? CurrentBranch { get; }
60-
GitStatus CurrentStatus { get; }
60+
int CurrentAhead { get; }
61+
int CurrentBehind { get; }
62+
List<GitStatusEntry> CurrentChanges { get; }
6163
GitRemote[] Remotes { get; }
6264
GitBranch[] LocalBranches { get; }
6365
GitBranch[] RemoteBranches { get; }

‎src/GitHub.Api/Git/Repository.cs

Lines changed: 52 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,9 @@ public void Initialize(IRepositoryManager initRepositoryManager)
4949
repositoryManager = initRepositoryManager;
5050
repositoryManager.CurrentBranchUpdated += RepositoryManagerOnCurrentBranchUpdated;
5151
repositoryManager.GitStatusUpdated += RepositoryManagerOnGitStatusUpdated;
52+
repositoryManager.GitAheadBehindStatusUpdated += RepositoryManagerOnGitAheadBehindStatusUpdated;
5253
repositoryManager.GitLogUpdated += RepositoryManagerOnGitLogUpdated;
54+
repositoryManager.GitLocksUpdated += RepositoryManagerOnGitLocksUpdated;
5355
repositoryManager.LocalBranchesUpdated += RepositoryManagerOnLocalBranchesUpdated;
5456
repositoryManager.RemoteBranchesUpdated += RepositoryManagerOnRemoteBranchesUpdated;
5557
}
@@ -267,7 +269,7 @@ private void CacheContainer_OnCacheInvalidated(CacheType cacheType)
267269
break;
268270

269271
case CacheType.GitLocksCache:
270-
repositoryManager?.UpdateLocks();
272+
UpdateLocks();
271273
break;
272274

273275
case CacheType.GitUserCache:
@@ -353,26 +355,36 @@ private void RepositoryManagerOnCurrentBranchUpdated(ConfigBranch? branch, Confi
353355
new ActionTask(CancellationToken.None, () => {
354356
if (!Nullable.Equals(CurrentConfigBranch, branch))
355357
{
356-
var currentBranch = branch != null ? (GitBranch?)GetLocalGitBranch(branch.Value) : null;
358+
var currentBranch = branch != null ? (GitBranch?)GetLocalGitBranch(branch.Value) : null;
357359

358-
CurrentConfigBranch = branch;
359-
CurrentBranch = currentBranch;
360-
UpdateLocalBranches();
360+
CurrentConfigBranch = branch;
361+
CurrentBranch = currentBranch;
362+
UpdateLocalBranches();
361363
}
362364

363365
if (!Nullable.Equals(CurrentConfigRemote, remote))
364366
{
365-
CurrentConfigRemote = remote;
366-
CurrentRemote = GetGitRemote(remote.Value);
367-
ClearRepositoryInfo();
367+
CurrentConfigRemote = remote;
368+
CurrentRemote = remote.HasValue ? (GitRemote?)GetGitRemote(remote.Value) : null;
369+
ClearRepositoryInfo();
368370
}
369371
}) { Affinity = TaskAffinity.UI }.Start();
370372
}
371373

372374
private void RepositoryManagerOnGitStatusUpdated(GitStatus gitStatus)
373375
{
374376
new ActionTask(CancellationToken.None, () => {
375-
CurrentStatus = gitStatus;
377+
CurrentChanges = gitStatus.Entries;
378+
CurrentAhead = gitStatus.Ahead;
379+
CurrentBehind = gitStatus.Behind;
380+
}) { Affinity = TaskAffinity.UI }.Start();
381+
}
382+
383+
private void RepositoryManagerOnGitAheadBehindStatusUpdated(GitAheadBehindStatus aheadBehindStatus)
384+
{
385+
new ActionTask(CancellationToken.None, () => {
386+
CurrentAhead = aheadBehindStatus.Ahead;
387+
CurrentBehind = aheadBehindStatus.Behind;
376388
}) { Affinity = TaskAffinity.UI }.Start();
377389
}
378390

@@ -383,6 +395,14 @@ private void RepositoryManagerOnGitLogUpdated(List<GitLogEntry> gitLogEntries)
383395
}) { Affinity = TaskAffinity.UI }.Start();
384396
}
385397

398+
private void RepositoryManagerOnGitLocksUpdated(List<GitLock> gitLocks)
399+
{
400+
new ActionTask(CancellationToken.None, () => {
401+
CurrentLocks = gitLocks;
402+
})
403+
{ Affinity = TaskAffinity.UI }.Start();
404+
}
405+
386406
private void RepositoryManagerOnRemoteBranchesUpdated(Dictionary<string, ConfigRemote> remotes,
387407
Dictionary<string, Dictionary<string, ConfigBranch>> branches)
388408
{
@@ -401,6 +421,14 @@ private void RepositoryManagerOnLocalBranchesUpdated(Dictionary<string, ConfigBr
401421
}) { Affinity = TaskAffinity.UI }.Start();
402422
}
403423

424+
private void UpdateLocks()
425+
{
426+
if (CurrentRemote.HasValue)
427+
{
428+
repositoryManager?.UpdateLocks();
429+
}
430+
}
431+
404432
private void UpdateLocalBranches()
405433
{
406434
LocalBranches = LocalConfigBranches.Values.Select(GetLocalGitBranch).ToArray();
@@ -469,10 +497,22 @@ private ConfigRemote? CurrentConfigRemote
469497
set { cacheContainer.BranchCache.CurrentConfigRemote = value; }
470498
}
471499

472-
public GitStatus CurrentStatus
500+
public int CurrentAhead
501+
{
502+
get { return cacheContainer.GitStatusCache.Ahead; }
503+
private set { cacheContainer.GitStatusCache.Ahead = value; }
504+
}
505+
506+
public int CurrentBehind
507+
{
508+
get { return cacheContainer.GitStatusCache.Behind; }
509+
private set { cacheContainer.GitStatusCache.Behind = value; }
510+
}
511+
512+
public List<GitStatusEntry> CurrentChanges
473513
{
474-
get { return cacheContainer.GitStatusCache.GitStatus; }
475-
private set { cacheContainer.GitStatusCache.GitStatus = value; }
514+
get { return cacheContainer.GitStatusCache.Entries; }
515+
private set { cacheContainer.GitStatusCache.Entries = value; }
476516
}
477517

478518
public GitBranch? CurrentBranch

‎src/GitHub.Api/Git/RepositoryManager.cs

Lines changed: 58 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -35,12 +35,14 @@ public interface IRepositoryManager : IDisposable
3535
ITask UnlockFile(string file, bool force);
3636
void UpdateGitLog();
3737
void UpdateGitStatus();
38+
void UpdateGitAheadBehindStatus();
3839
void UpdateLocks();
3940
int WaitForEvents();
4041

4142
IGitConfig Config { get; }
4243
IGitClient GitClient { get; }
4344
bool IsBusy { get; }
45+
event Action<GitAheadBehindStatus> GitAheadBehindStatusUpdated;
4446
}
4547

4648
interface IRepositoryPathConfiguration
@@ -101,6 +103,7 @@ class RepositoryManager : IRepositoryManager
101103
public event Action<ConfigBranch?, ConfigRemote?> CurrentBranchUpdated;
102104
public event Action<bool> IsBusyChanged;
103105
public event Action<GitStatus> GitStatusUpdated;
106+
public event Action<GitAheadBehindStatus> GitAheadBehindStatusUpdated;
104107
public event Action<List<GitLock>> GitLocksUpdated;
105108
public event Action<List<GitLogEntry>> GitLogUpdated;
106109
public event Action<Dictionary<string, ConfigBranch>> LocalBranchesUpdated;
@@ -209,25 +212,13 @@ public ITask RemoteAdd(string remote, string url)
209212
{
210213
var task = GitClient.RemoteAdd(remote, url);
211214
task = HookupHandlers(task, true, false);
212-
if (!platform.Environment.IsWindows)
213-
{
214-
task.Then(_ => {
215-
UpdateConfigData(true);
216-
});
217-
}
218215
return task;
219216
}
220217

221218
public ITask RemoteRemove(string remote)
222219
{
223220
var task = GitClient.RemoteRemove(remote);
224221
task = HookupHandlers(task, true, false);
225-
if (!platform.Environment.IsWindows)
226-
{
227-
task.Then(_ => {
228-
UpdateConfigData(true);
229-
});
230-
}
231222
return task;
232223
}
233224

@@ -258,13 +249,13 @@ public ITask CreateBranch(string branch, string baseBranch)
258249
public ITask LockFile(string file)
259250
{
260251
var task = GitClient.Lock(file);
261-
return HookupHandlers(task, true, false);
252+
return HookupHandlers(task, true, false).Then(UpdateLocks);
262253
}
263254

264255
public ITask UnlockFile(string file, bool force)
265256
{
266257
var task = GitClient.Unlock(file, force);
267-
return HookupHandlers(task, true, false);
258+
return HookupHandlers(task, true, false).Then(UpdateLocks);
268259
}
269260

270261
public void UpdateGitLog()
@@ -293,6 +284,33 @@ public void UpdateGitStatus()
293284
}).Start();
294285
}
295286

287+
public void UpdateGitAheadBehindStatus()
288+
{
289+
ConfigBranch? configBranch;
290+
ConfigRemote? configRemote;
291+
GetCurrentBranchAndRemote(out configBranch, out configRemote);
292+
293+
if (configBranch.HasValue && configBranch.Value.Remote.HasValue)
294+
{
295+
var name = configBranch.Value.Name;
296+
var trackingName = configBranch.Value.IsTracking ? configBranch.Value.Remote.Value.Name + "/" + name : "[None]";
297+
298+
var task = GitClient.AheadBehindStatus(name, trackingName);
299+
task = HookupHandlers(task, true, false);
300+
task.Then((success, status) =>
301+
{
302+
if (success)
303+
{
304+
GitAheadBehindStatusUpdated?.Invoke(status);
305+
}
306+
}).Start();
307+
}
308+
else
309+
{
310+
GitAheadBehindStatusUpdated?.Invoke(GitAheadBehindStatus.Default);
311+
}
312+
}
313+
296314
public void UpdateLocks()
297315
{
298316
var task = GitClient.ListLocks(false);
@@ -357,15 +375,33 @@ private void SetupWatcher()
357375

358376
private void UpdateHead()
359377
{
360-
var head = repositoryPaths.DotGitHead.ReadAllLines().FirstOrDefault();
361-
Logger.Trace("UpdateHead: {0}", head ?? "[NULL]");
362-
UpdateCurrentBranchAndRemote(head);
378+
Logger.Trace("UpdateHead");
379+
UpdateCurrentBranchAndRemote();
380+
UpdateGitLog();
381+
}
382+
383+
private string GetCurrentHead()
384+
{
385+
return repositoryPaths.DotGitHead.ReadAllLines().FirstOrDefault();
386+
}
387+
388+
private void UpdateCurrentBranchAndRemote()
389+
{
390+
ConfigBranch? branch;
391+
ConfigRemote? remote;
392+
GetCurrentBranchAndRemote(out branch, out remote);
393+
394+
Logger.Trace("CurrentBranch: {0}", branch.HasValue ? branch.Value.ToString() : "[NULL]");
395+
Logger.Trace("CurrentRemote: {0}", remote.HasValue ? remote.Value.ToString() : "[NULL]");
396+
CurrentBranchUpdated?.Invoke(branch, remote);
363397
}
364398

365-
private void UpdateCurrentBranchAndRemote(string head)
399+
private void GetCurrentBranchAndRemote(out ConfigBranch? branch, out ConfigRemote? remote)
366400
{
367-
ConfigBranch? branch = null;
401+
branch = null;
402+
remote = null;
368403

404+
var head = GetCurrentHead();
369405
if (head.StartsWith("ref:"))
370406
{
371407
var branchName = head.Substring(head.IndexOf("refs/heads/") + "refs/heads/".Length);
@@ -378,7 +414,6 @@ private void UpdateCurrentBranchAndRemote(string head)
378414
}
379415

380416
var defaultRemote = "origin";
381-
ConfigRemote? remote = null;
382417

383418
if (branch.HasValue && branch.Value.IsTracking)
384419
{
@@ -398,10 +433,6 @@ private void UpdateCurrentBranchAndRemote(string head)
398433
remote = configRemotes.FirstOrDefault();
399434
}
400435
}
401-
402-
Logger.Trace("CurrentBranch: {0}", branch.HasValue ? branch.Value.ToString() : "[NULL]");
403-
Logger.Trace("CurrentRemote: {0}", remote.HasValue ? remote.Value.ToString() : "[NULL]");
404-
CurrentBranchUpdated?.Invoke(branch, remote);
405436
}
406437

407438
private void WatcherOnRemoteBranchesChanged()
@@ -420,6 +451,7 @@ private void WatcherOnRepositoryCommitted()
420451
{
421452
Logger.Trace("WatcherOnRepositoryCommitted");
422453
UpdateGitLog();
454+
UpdateGitAheadBehindStatus();
423455
}
424456

425457
private void WatcherOnRepositoryChanged()
@@ -518,6 +550,8 @@ private void UpdateRemoteBranches()
518550

519551
Logger.Trace("OnRemoteBranchListUpdated {0} remotes", remotes.Count);
520552
RemoteBranchesUpdated?.Invoke(remotes, remoteBranches);
553+
554+
UpdateGitAheadBehindStatus();
521555
}
522556

523557
private bool disposed;

0 commit comments

Comments
 (0)