#import "TestNodes.h"

@implementation Node_Standard

@synthesize key = key;
@synthesize childKeys = childKeys;

- (id)init
{
	if ((self = [super init]))
	{
		key = [[NSUUID UUID] UUIDString];
	}
	return self;
}

- (id)initWithCoder:(NSCoder *)decoder
{
	if ((self = [super init]))
	{
		key = [decoder decodeObjectForKey:@"key"];
		childKeys = [decoder decodeObjectForKey:@"childKeys"];
	}
	return self;
}

- (void)encodeWithCoder:(NSCoder *)coder
{
	[coder encodeObject:key forKey:@"key"];
	[coder encodeObject:childKeys forKey:@"childKeys"];
}

- (NSArray *)yapDatabaseRelationshipEdges
{
	NSUInteger count = [childKeys count];
	if (count == 0) return nil;
	
	NSMutableArray *edges = [NSMutableArray arrayWithCapacity:count];
	
	for (NSString *childKey in childKeys)
	{
		YapDatabaseRelationshipEdge *edge =
		  [YapDatabaseRelationshipEdge edgeWithName:@"child"
		                             destinationKey:childKey
		                                 collection:nil
		                            nodeDeleteRules:YDB_DeleteDestinationIfSourceDeleted];
		
		[edges addObject:edge];
	}
	
	return edges;
}

@end

////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#pragma mark -
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

@implementation Node_Inverse

@synthesize key = key;
@synthesize parentKey = parentKey;

- (id)init
{
	if ((self = [super init]))
	{
		key = [[NSUUID UUID] UUIDString];
	}
	return self;
}

- (id)initWithCoder:(NSCoder *)decoder
{
	if ((self = [super init]))
	{
		key = [decoder decodeObjectForKey:@"key"];
		parentKey = [decoder decodeObjectForKey:@"parentKey"];
	}
	return self;
}

- (void)encodeWithCoder:(NSCoder *)coder
{
	[coder encodeObject:key forKey:@"key"];
	[coder encodeObject:parentKey forKey:@"parentKey"];
}

- (NSArray *)yapDatabaseRelationshipEdges
{
	if (parentKey == nil)
		return nil;
	
	YapDatabaseRelationshipEdge *edge =
	  [YapDatabaseRelationshipEdge edgeWithName:@"parent"
	                             destinationKey:parentKey
	                                 collection:nil
	                            nodeDeleteRules:YDB_DeleteSourceIfDestinationDeleted];
	
	return @[ edge ];
}

@end

////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#pragma mark -
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

@implementation Node_RetainCount

@synthesize key = key;
@synthesize retainedKey = retainedKey;

- (id)init
{
	if ((self = [super init]))
	{
		key = [[NSUUID UUID] UUIDString];
	}
	return self;
}

- (id)initWithCoder:(NSCoder *)decoder
{
	if ((self = [super init]))
	{
		key = [decoder decodeObjectForKey:@"key"];
		retainedKey = [decoder decodeObjectForKey:@"retainedKey"];
	}
	return self;
}

- (void)encodeWithCoder:(NSCoder *)coder
{
	[coder encodeObject:key forKey:@"key"];
	[coder encodeObject:retainedKey forKey:@"retainedKey"];
}

- (NSArray *)yapDatabaseRelationshipEdges
{
	if (retainedKey == nil)
		return nil;
	
	YapDatabaseRelationshipEdge *edge =
	  [YapDatabaseRelationshipEdge edgeWithName:@"retained"
	                             destinationKey:retainedKey
	                                 collection:nil
	                            nodeDeleteRules:YDB_DeleteDestinationIfAllSourcesDeleted];
	
	return @[ edge ];
}

@end

////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#pragma mark -
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

@implementation Node_InverseRetainCount

@synthesize key = key;
@synthesize retainerKeys = retainerKeys;

- (id)init
{
	if ((self = [super init]))
	{
		key = [[NSUUID UUID] UUIDString];
	}
	return self;
}

- (id)initWithCoder:(NSCoder *)decoder
{
	if ((self = [super init]))
	{
		key = [decoder decodeObjectForKey:@"key"];
		retainerKeys = [decoder decodeObjectForKey:@"retainerKeys"];
	}
	return self;
}

- (void)encodeWithCoder:(NSCoder *)coder
{
	[coder encodeObject:key forKey:@"key"];
	[coder encodeObject:retainerKeys forKey:@"retainerKeys"];
}

- (NSArray *)yapDatabaseRelationshipEdges
{
	NSUInteger count = [retainerKeys count];
	if (count == 0) return nil;
	
	NSMutableArray *edges = [NSMutableArray arrayWithCapacity:count];
	
	for (NSString *retainerKey in retainerKeys)
	{
		YapDatabaseRelationshipEdge *edge =
		  [YapDatabaseRelationshipEdge edgeWithName:@"retainer"
		                             destinationKey:retainerKey
		                                 collection:nil
		                            nodeDeleteRules:YDB_DeleteSourceIfAllDestinationsDeleted];
		
		[edges addObject:edge];
	}
	
	return edges;
}

@end

////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#pragma mark -
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

/**
 * Standard file relationship: (parent)->(filePath)
 * nodeDeleteRule = YDB_DeleteDestinationIfSourceDeleted
 * 
 * So the parent node creates the edge which points to the "child" filePath.
 * And the file should get deleted if the parent is deleted.
**/
@implementation Node_Standard_FileURL

@synthesize key = key;
@synthesize fileURL = fileURL;

- (id)init
{
	if ((self = [super init]))
	{
		key = [[NSUUID UUID] UUIDString];
	}
	return self;
}

- (id)initWithCoder:(NSCoder *)decoder
{
	if ((self = [super init]))
	{
		key = [decoder decodeObjectForKey:@"key"];
		fileURL = [decoder decodeObjectForKey:@"fileURL"];
	}
	return self;
}

- (void)encodeWithCoder:(NSCoder *)coder
{
	[coder encodeObject:key forKey:@"key"];
	[coder encodeObject:fileURL forKey:@"fileURL"];
}

- (NSArray *)yapDatabaseRelationshipEdges
{
	if (fileURL == nil) return nil;
	
	YapDatabaseRelationshipEdge *edge =
	  [YapDatabaseRelationshipEdge edgeWithName:@"random"
	                         destinationFileURL:fileURL
	                            nodeDeleteRules:YDB_DeleteDestinationIfSourceDeleted];
	
	return @[ edge ];
}

@end

////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#pragma mark -
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

/**
 * Retain count file relationship: (parent)->(filePath)
 * nodeDeleteRule = YDB_DeleteDestinationIfAllSourcesDeleted
 *
 * So the retainer node creates the edge which points the the retained file.
 * And there may be multiple retainers pointing to the same retained file.
 * And the file doesn't get deleted unless all the retainers are deleted.
 **/
@implementation Node_RetainCount_FileURL

@synthesize key = key;
@synthesize fileURL = fileURL;

- (id)init
{
	if ((self = [super init]))
	{
		key = [[NSUUID UUID] UUIDString];
	}
	return self;
}

- (id)initWithCoder:(NSCoder *)decoder
{
	if ((self = [super init]))
	{
		key = [decoder decodeObjectForKey:@"key"];
		fileURL = [decoder decodeObjectForKey:@"fileURL"];
	}
	return self;
}

- (void)encodeWithCoder:(NSCoder *)coder
{
	[coder encodeObject:key forKey:@"key"];
	[coder encodeObject:fileURL forKey:@"fileURL"];
}

- (NSArray *)yapDatabaseRelationshipEdges
{
	if (fileURL == nil) return nil;
	
	YapDatabaseRelationshipEdge *edge =
	  [YapDatabaseRelationshipEdge edgeWithName:@"shared"
	                         destinationFileURL:fileURL
	                            nodeDeleteRules:YDB_DeleteDestinationIfAllSourcesDeleted];
	
	return @[ edge ];
}

@end

////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#pragma mark -
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

@implementation Node_Notify

@synthesize key = key;
@synthesize child = child;

- (id)init
{
	if ((self = [super init]))
	{
		key = [[NSUUID UUID] UUIDString];
	}
	return self;
}

- (id)initWithCoder:(NSCoder *)decoder
{
	if ((self = [super init]))
	{
		key = [decoder decodeObjectForKey:@"key"];
		child = [decoder decodeObjectForKey:@"child"];
	}
	return self;
}

- (void)encodeWithCoder:(NSCoder *)coder
{
	[coder encodeObject:key forKey:@"key"];
	[coder encodeObject:child forKey:@"child"];
}

- (NSArray *)yapDatabaseRelationshipEdges
{
	if (child == nil) return nil;
	
	YapDatabaseRelationshipEdge *edge =
	  [YapDatabaseRelationshipEdge edgeWithName:@"child"
	                             destinationKey:child
	                                 collection:nil
	                            nodeDeleteRules:(YDB_DeleteDestinationIfSourceDeleted | YDB_NotifyIfSourceDeleted)];
	
	return @[ edge ];
}

@end

////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#pragma mark -
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

@implementation Node_NotifyCount

static NSUInteger notifyCount = 0;

@synthesize key = key;

- (id)init
{
	if ((self = [super init]))
	{
		key = [[NSUUID UUID] UUIDString];
	}
	return self;
}

- (id)initWithCoder:(NSCoder *)decoder
{
	if ((self = [super init]))
	{
		key = [decoder decodeObjectForKey:@"key"];
	}
	return self;
}

- (void)encodeWithCoder:(NSCoder *)coder
{
	[coder encodeObject:key forKey:@"key"];
}

- (NSArray<YapDatabaseRelationshipEdge *> *)yapDatabaseRelationshipEdges
{
	return nil;
}

- (id)yapDatabaseRelationshipEdgeDeleted:(YapDatabaseRelationshipEdge *)edge
                                       withReason:(YDB_NotifyReason)reason
{
	notifyCount++;
	return nil;
}

+ (NSUInteger)notifyCount
{
	return notifyCount;
}

@end