#import "GTEnumerator.h"

#import "GTCommit.h"

#import "NSError+Git.h"

#import "NSString+Git.h"

#import "GTRepository.h"

#import "GTRepository+Private.h"

#import "GTOID.h"

#import "git2/errors.h"

@interface GTEnumerator ()

@property (nonatomic, assign, readonly) git_revwalk *walk;

@property (nonatomic, assign, readwrite) GTEnumeratorOptions options;

@implementation GTEnumerator

#pragma mark Lifecycle

- (instancetype)init {

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

return nil;

}

- (git_revwalk *)git_revwalk {

return self.walk;

}

- (instancetype)initWithRepository:(GTRepository *)repo error:(NSError **)error {

NSParameterAssert(repo != nil);

self = [super init];

if (self == nil) return nil;

_repository = repo;

_options = GTEnumeratorOptionsNone;

int gitError = git_revwalk_new(&_walk, self.repository.git_repository);

if (gitError != GIT_OK) {

if (error != NULL) *error = [NSError git_errorFor:gitError description:@"Failed to initialize rev walker."];

return nil;

}

return self;

}

- (void)dealloc {

if (_walk != NULL) {

git_revwalk_free(_walk);

_walk = NULL;

}

}

#pragma mark Pushing and Hiding

- (BOOL)pushSHA:(NSString *)sha error:(NSError **)error {

NSParameterAssert(sha != nil);

GTOID *oid = [[GTOID alloc] initWithSHA:sha error:error];

if (oid == nil) return NO;

int gitError = git_revwalk_push(self.walk, oid.git_oid);

if (gitError != GIT_OK) {

if (error != NULL) *error = [NSError git_errorFor:gitError description:@"Failed to push SHA %@ onto rev walker.", sha];

return NO;

}

return YES;

}

- (BOOL)pushGlob:(NSString *)refGlob error:(NSError **)error {

NSParameterAssert(refGlob != nil);

int gitError = git_revwalk_push_glob(self.walk, refGlob.UTF8String);

if (gitError != GIT_OK) {

if (error != NULL) *error = [NSError git_errorFor:gitError description:@"Failed to push glob %@ onto rev walker.", refGlob];

return NO;

}

return YES;

}

- (BOOL)pushHEAD:(NSError **)error {

int gitError = git_revwalk_push_head(self.walk);

if (gitError != GIT_OK) {

if (error != NULL) *error = [NSError git_errorFor:gitError description:@"Failed to push HEAD onto rev walker."];

return NO;

}

return YES;

}

- (BOOL)pushReferenceName:(NSString *)refName error:(NSError **)error {

NSParameterAssert(refName != nil);

int gitError = git_revwalk_push_ref(self.walk, refName.UTF8String);

if (gitError != 0) {

if (error) *error = [NSError git_errorFor:gitError description:@"Failed to push reference %@", refName];

return NO;

}

return YES;

}

- (BOOL)hideSHA:(NSString *)sha error:(NSError **)error {

NSParameterAssert(sha != nil);

GTOID *oid = [[GTOID alloc] initWithSHA:sha error:error];

if (oid == nil) return NO;

int gitError = git_revwalk_hide(self.walk, oid.git_oid);

if (gitError != GIT_OK) {

if (error != NULL) *error = [NSError git_errorFor:gitError description:@"Failed to hide SHA %@ on rev walker.", sha];

return NO;

}

return YES;

}

- (BOOL)hideGlob:(NSString *)refGlob error:(NSError **)error {

NSParameterAssert(refGlob != nil);

int gitError = git_revwalk_hide_glob(self.walk, refGlob.UTF8String);

if (gitError != GIT_OK) {

if (error != NULL) *error = [NSError git_errorFor:gitError description:@"Failed to hide glob %@ in rev walker.", refGlob];

return NO;

}

return YES;

}

- (BOOL)hideHEAD:(NSError **)error {

int gitError = git_revwalk_hide_head(self.walk);

if (gitError != GIT_OK) {

if (error != NULL) *error = [NSError git_errorFor:gitError description:@"Failed to hide HEAD onto rev walker."];

return NO;

}

return YES;

}

- (BOOL)hideReferenceName:(NSString *)refName error:(NSError **)error {

NSParameterAssert(refName != nil);

int gitError = git_revwalk_hide_ref(self.walk, refName.UTF8String);

if (gitError != 0) {

if (error) *error = [NSError git_errorFor:gitError description:@"Failed to hide reference %@", refName];

return NO;

}

return YES;

}

#pragma mark Resetting

- (void)resetWithOptions:(GTEnumeratorOptions)options {

self.options = options;

// This will also reset the walker.

git_revwalk_sorting(self.walk, self.options);

}

#pragma mark Enumerating

- (GTOID *)nextOIDWithSuccess:(BOOL *)success error:(NSError **)error {

git_oid oid;

int gitError = git_revwalk_next(&oid, self.walk);

if (gitError == GIT_ITEROVER) {

if (success != NULL) *success = YES;

return nil;

}

if (gitError != GIT_OK) {

if (success != NULL) *success = NO;

if (error != NULL) *error = [NSError git_errorFor:gitError description:@"Enumeration failed"];

return nil;

}

if (success != NULL) *success = YES;

return [GTOID oidWithGitOid:&oid];

}

- (GTCommit *)nextObjectWithSuccess:(BOOL *)success error:(NSError **)error {

GTOID *oid = [self nextOIDWithSuccess:success error:error];

if (oid == nil) {

// We don't care whether the iteration completed, or an error occurred,

// there's nothing to lookup.

return nil;

}

// Ignore error if we can't lookup object and just return nil.

GTCommit *commit = [self.repository lookUpObjectByOID:oid objectType:GTObjectTypeCommit error:error];

if (success != NULL) *success = (commit != nil);

return commit;

}

- (NSArray *)allObjectsWithError:(NSError **)error {

NSMutableArray *array = [NSMutableArray array];

GTCommit *object;

do {

BOOL success;

object = [self nextObjectWithSuccess:&success error:error];

if (!success) return nil;

if (object != nil) {

[array addObject:object];

}

} while (object != nil);

return array;

}

- (NSUInteger)countRemainingObjects:(NSError **)error {

git_oid oid;

int gitError;

NSUInteger count = 0;

while ((gitError = git_revwalk_next(&oid, self.walk)) == GIT_OK) {

count++;

}

if (gitError == GIT_ITEROVER) {

return count;

} else {

if (error != NULL) *error = [NSError git_errorFor:gitError description:@"Failed to get next SHA with rev walker."];

return NSNotFound;

}

}

#pragma mark NSEnumerator

- (NSArray *)allObjects {

NSArray *objects = [self allObjectsWithError:NULL];

return objects ? objects : [NSArray array];

}

- (id)nextObject {

return [self nextObjectWithSuccess:NULL error:NULL];

}

#pragma mark NSObject

- (NSString *)description {

return [NSString stringWithFormat:@"<%@: %p> repository: %@", self.class, self, self.repository];

}