Added git_tree_entry_bypath method wrapper to GTTree

Author: Jan Gassen

ObjectiveGit/GTTree.h

@@ -64,6 +64,8 @@ typedef NS_ENUM(NSInteger, GTTreeEnumerationOptions) {
 /// returns a GTTreeEntry or nil if there is nothing with the specified name
 - (GTTreeEntry *)entryWithName:(NSString *)name;
 
+- (GTTreeEntry*) treeEntryByPath:(NSString*) path error:(NSError**)error;
+
 /// Enumerates the contents of the tree
 ///
 /// options -  One of `GTTreeEnumerationOptionPre` (for pre-order walks) or

ObjectiveGit/GTTree.m

@@ -68,6 +68,17 @@ - (GTTreeEntry *)entryWithName:(NSString *)name {
 	return [self createEntryWithEntry:git_tree_entry_byname(self.git_tree, name.UTF8String)];
 }
 
+- (GTTreeEntry*) treeEntryByPath:(NSString*)path error:(NSError**)error {
+	git_tree_entry *entry = NULL;
+	int gitError = git_tree_entry_bypath(&entry, self.git_tree, path.UTF8String);
+	if (error != GIT_OK) {
+		*error = [NSError git_errorFor:gitError description:@"Failed to get tree ntry %@", path];
+		return nil;
+	}
+	
+	return [GTTreeEntry entryWithEntryToFree:entry parentTree:self];
+}
+
 - (git_tree *)git_tree {
 	return (git_tree *)self.git_object;
 }

ObjectiveGit/GTTreeEntry.h

@@ -35,7 +35,9 @@
 
 /// Initializer and convience methods.
 - (instancetype)initWithEntry:(const git_tree_entry *)theEntry parentTree:(GTTree *)parent;
+- (instancetype)initWithEntryToFree:(git_tree_entry *)theEntry parentTree:(GTTree *)parent;
 + (instancetype)entryWithEntry:(const git_tree_entry *)theEntry parentTree:(GTTree *)parent;
++ (instancetype)entryWithEntryToFree:(git_tree_entry *)theEntry parentTree:(GTTree *)parent;
 
 /// The underlying `git_tree_entry`.
 - (const git_tree_entry *)git_tree_entry __attribute__((objc_returns_inner_pointer));

ObjectiveGit/GTTreeEntry.m

@@ -39,6 +39,7 @@
 
 @interface GTTreeEntry ()
 @property (nonatomic, assign, readonly) const git_tree_entry *git_tree_entry;
+@property (nonatomic, assign, readonly) git_tree_entry* entry_to_free;
 @end
 
 @implementation GTTreeEntry
@@ -64,13 +65,30 @@ - (BOOL)isEqualToEntry:(GTTreeEntry *)treeEntry {
 	return git_tree_entry_cmp(self.git_tree_entry, treeEntry.git_tree_entry) == 0 ? YES : NO;
 }
 
+- (void)dealloc {
+	if (_entry_to_free != nil) {
+		git_tree_entry_free(_entry_to_free);
+	}
+}
+
 #pragma mark API
 
 - (instancetype)initWithEntry:(const git_tree_entry *)theEntry parentTree:(GTTree *)parent {
 	NSParameterAssert(theEntry != NULL);
 	if((self = [super init])) {
 		_git_tree_entry = theEntry;
 		_tree = parent;
+		_entry_to_free = nil;
+	}
+	return self;
+}
+
+- (instancetype)initWithEntryToFree:(git_tree_entry *)theEntry parentTree:(GTTree *)parent {
+	NSParameterAssert(theEntry != NULL);
+	if((self = [super init])) {
+		_git_tree_entry = theEntry;
+		_tree = parent;
+		_entry_to_free = theEntry;
 	}
 	return self;
 }
@@ -79,6 +97,10 @@ + (instancetype)entryWithEntry:(const git_tree_entry *)theEntry parentTree:(GTTr
 	return [[self alloc] initWithEntry:theEntry parentTree:parent];
 }
 
++ (instancetype)entryWithEntryToFree:(git_tree_entry *)theEntry parentTree:(GTTree *)parent {
+	return [[self alloc] initWithEntryToFree:theEntry parentTree:parent];
+}
+
 - (NSString *)name {
 	return @(git_tree_entry_name(self.git_tree_entry));
 }