Highlight the current commit with an orange circle

Author: encode-zz

PBGitRepository.h

@@ -50,6 +50,7 @@ static NSString * PBStringFromBranchFilterType(PBGitXBranchFilterType type) {
 	NSMutableDictionary *refs;
 
 	PBGitRevSpecifier *_headRef; // Caching
+	NSString* _headSha;
 }
 
 - (void) cloneRepositoryToPath:(NSString *)path bare:(BOOL)isBare;

PBGitRepository.m

@@ -234,6 +234,7 @@ - (void) addRef: (PBGitRef *) ref fromParameters: (NSArray *) components
 - (void) reloadRefs
 {
 	_headRef = nil;
+	_headSha = nil;
 
 	refs = [NSMutableDictionary dictionary];
 	NSMutableArray *oldBranches = [branches mutableCopy];
@@ -287,12 +288,17 @@ - (PBGitRevSpecifier *)headRef
 	else
 		_headRef = [[PBGitRevSpecifier alloc] initWithRef:[PBGitRef refFromString:@"HEAD"]];
 
+	_headSha = [self shaForRef:[_headRef ref]];
+
 	return _headRef;
 }
 
 - (NSString *) headSHA
 {
-	return [self shaForRef:[[self headRef] ref]];
+	if (! _headSha)
+		[self headRef];
+
+	return _headSha;
 }
 
 - (PBGitCommit *) headCommit

PBGitRevisionCell.m

@@ -54,6 +54,16 @@ - (void) drawLineFromColumn: (int) from toColumn: (int) to inRect: (NSRect) r of
 
 }
 
+- (BOOL) isCurrentCommit
+{
+	NSString* thisSha = [self.objectValue realSha];
+
+	PBGitRepository* repository = [self.objectValue repository];
+	NSString* currentSha = [repository headSHA];
+
+	return [ currentSha isEqualToString : thisSha ];
+}
+
 - (void) drawCircleInRect: (NSRect) r
 {
 
@@ -71,7 +81,13 @@ - (void) drawCircleInRect: (NSRect) r
 	[path fill];
 
 	NSRect smallOval = { columnOrigin.x - 3, columnOrigin.y + r.size.height * 0.5 - 3, 6, 6};
-	[[NSColor whiteColor] set];
+
+	if ( [self isCurrentCommit ] ) {
+		[[NSColor colorWithCalibratedRed: 0Xfc/256.0 green:0Xa6/256.0 blue: 0X4f/256.0 alpha: 1.0] set];
+	} else {
+		[[NSColor whiteColor] set];
+	}
+
 	path = [NSBezierPath bezierPathWithOvalInRect:smallOval];
 	[path fill];	
 }