PBGitRepository: Add method to remove branch

This adds a removeBranch method similar to the addBranch method. It
calls these methods when a branch is created/deleted in the UI, avoiding
the cost of simply calling refresh afterwards.

Signed-off-by: Johannes Gilger <[email protected]>

Author: heipei

PBGitRepository.h

@@ -51,6 +51,7 @@ extern NSString* PBGitRepositoryErrorDomain;
 
 - (void) readCurrentBranch;
 - (PBGitRevSpecifier*) addBranch: (PBGitRevSpecifier*) rev;
+- (BOOL)removeBranch:(PBGitRevSpecifier *)rev;
 
 - (NSString*) parseSymbolicReference:(NSString*) ref;
 - (NSString*) parseReference:(NSString*) ref;

PBGitRepository.m

@@ -294,6 +294,19 @@ - (PBGitRevSpecifier*) addBranch: (PBGitRevSpecifier*) rev
 	[self didChangeValueForKey:@"branches"];
 	return rev;
 }
+
+- (BOOL)removeBranch:(PBGitRevSpecifier *)rev
+{
+	for (PBGitRevSpecifier *r in branches) {
+		if ([rev isEqualTo:r]) {
+			[self willChangeValueForKey:@"branches"];
+			[branches removeObject:r];
+			[self didChangeValueForKey:@"branches"];
+			return TRUE;
+		}
+	}
+	return FALSE;
+}
 
 - (void) readCurrentBranch
 {

PBRefController.m

@@ -49,7 +49,7 @@ - (void) removeRef:(PBRefMenuItem *) sender
 			NSLog(@"Removing ref failed!");
 			return;
 		}
-
+		[historyController.repository removeBranch:[[PBGitRevSpecifier alloc] initWithRef:[sender ref]]];
 		[[sender commit] removeRef:[sender ref]];
 		[commitController rearrangeObjects];
 	}
@@ -208,7 +208,7 @@ -(void)saveSheet:(id) sender
 		[errorMessage setStringValue:@"Branch exists"];
 		return;
 	}
-
+	[historyController.repository addBranch:[[PBGitRevSpecifier alloc] initWithRef:[PBGitRef refFromString:branchName]]];
 	[self closeSheet:sender];
 	[commit addRef:[PBGitRef refFromString:branchName]];
 	[commitController rearrangeObjects];