#import "GTDiffHunk.h"

#import "GTDiffLine.h"

#import "GTDiffPatch.h"

#import "NSError+Git.h"

#import "git2/errors.h"

#import "git2/patch.h"

@interface GTDiffHunk ()

@property (nonatomic, assign, readonly) const git_diff_hunk *git_hunk;

@property (nonatomic, strong, readonly) GTDiffPatch *patch;

@property (nonatomic, assign, readonly) NSUInteger hunkIndex;

@implementation GTDiffHunk

- (instancetype)init {

NSAssert(NO, @"Call to an unavailable initializer.");

return nil;

}

- (instancetype)initWithPatch:(GTDiffPatch *)patch hunkIndex:(NSUInteger)hunkIndex {

NSParameterAssert(patch != nil);

self = [super init];

if (self == nil) return nil;

size_t gitLineCount = 0;

int result = git_patch_get_hunk(&_git_hunk, &gitLineCount, patch.git_patch, hunkIndex);

if (result != GIT_OK) return nil;

_lineCount = gitLineCount;

_oldStart = self.git_hunk->old_start;

_oldLines = self.git_hunk->old_lines;

_newStart = self.git_hunk->new_start;

_newLines = self.git_hunk->new_lines;

_patch = patch;

_hunkIndex = hunkIndex;

NSString *hunkHeader = [[[NSString alloc] initWithBytes:self.git_hunk->header length:self.git_hunk->header_len encoding:NSUTF8StringEncoding] stringByTrimmingCharactersInSet:NSCharacterSet.newlineCharacterSet];

NSAssert(hunkHeader != nil, @"Failed to build hunk header");

_header = hunkHeader;

return self;

}

- (NSString *)debugDescription {

return [NSString stringWithFormat:@"%@ hunkIndex: %ld, header: %@, lineCount: %ld", super.debugDescription, (unsigned long)self.hunkIndex, self.header, (unsigned long)self.lineCount];

}

- (BOOL)enumerateLinesInHunk:(NSError **)error usingBlock:(void (^)(GTDiffLine *line, BOOL *stop))block {

NSParameterAssert(block != nil);

for (NSUInteger idx = 0; idx < self.lineCount; idx ++) {

const git_diff_line *gitLine;

int result = git_patch_get_line_in_hunk(&gitLine, self.patch.git_patch, self.hunkIndex, idx);

if (result != GIT_OK) {

if (error) *error = [NSError git_errorFor:result description:@"Extracting line from hunk failed"];

return NO;

}

GTDiffLine *line = [[GTDiffLine alloc] initWithGitLine:gitLine];

BOOL stop = NO;

block(line, &stop);

if (stop) break;

}

return YES;

}