RestKit/Code/ObjectMapping/RKObjectConnectionMapping.m

//
//  RKObjectConnectionMapping.m
//  RestKit
//
//  Created by Charlie Savage on 5/15/12.
//  Copyright (c) 2009-2012 RestKit. All rights reserved.
//
//  Licensed under the Apache License, Version 2.0 (the "License");
//  you may not use this file except in compliance with the License.
//  You may obtain a copy of the License at
//
//  http://www.apache.org/licenses/LICENSE-2.0
//
//  Unless required by applicable law or agreed to in writing, software
//  distributed under the License is distributed on an "AS IS" BASIS,
//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//  See the License for the specific language governing permissions and
//  limitations under the License.
//

#import "RKObjectConnectionMapping.h"
#import "NSManagedObject+ActiveRecord.h"
#import "RKManagedObjectMapping.h"
#import "RKObjectManager.h"
#import "RKManagedObjectCaching.h"
#import "RKDynamicObjectMappingMatcher.h"

@interface RKObjectConnectionMapping()
@property (nonatomic, retain) NSString * relationshipName;
@property (nonatomic, retain) NSString * destinationKeyPath;
@property (nonatomic, retain) RKObjectMappingDefinition * mapping;
@property (nonatomic, retain) RKDynamicObjectMappingMatcher* matcher;
@property (nonatomic, retain) NSString * sourceKeyPath;
@end

@implementation RKObjectConnectionMapping

@synthesize relationshipName;
@synthesize destinationKeyPath;
@synthesize mapping;
@synthesize matcher;
@synthesize sourceKeyPath;

+ (RKObjectConnectionMapping*)mapping:(NSString *)relationshipName fromKeyPath:(NSString*)sourceKeyPath toKeyPath:(NSString*)destinationKeyPath withMapping:(RKObjectMappingDefinition *)objectOrDynamicMapping {
    RKObjectConnectionMapping *mapping = [[self alloc] init:relationshipName fromKeyPath:sourceKeyPath toKeyPath:destinationKeyPath matcher:nil withMapping:objectOrDynamicMapping];
    return [mapping autorelease];
}

+ (RKObjectConnectionMapping*)mapping:(NSString *)relationshipName fromKeyPath:(NSString*)sourceKeyPath toKeyPath:(NSString*)destinationKeyPath matcher:(RKDynamicObjectMappingMatcher *)matcher withMapping:(RKObjectMappingDefinition *)objectOrDynamicMapping {
    RKObjectConnectionMapping *mapping = [[self alloc] init:relationshipName fromKeyPath:sourceKeyPath toKeyPath:destinationKeyPath matcher:matcher withMapping:objectOrDynamicMapping];
    return [mapping autorelease];
}

- (id)init:(NSString *)aRelationshipName fromKeyPath:(NSString*)aSourceKeyPath toKeyPath:(NSString*)aDestinationKeyPath matcher:(RKDynamicObjectMappingMatcher *)aMatcher withMapping:(RKObjectMappingDefinition *)aObjectOrDynamicMapping {
    self = [super init];
    self.relationshipName = aRelationshipName;
    self.sourceKeyPath = aSourceKeyPath;
    self.destinationKeyPath = aDestinationKeyPath;
    self.mapping = aObjectOrDynamicMapping;
    self.matcher = aMatcher;
    return self;
}

- (id)copyWithZone:(NSZone *)zone {
    return [[[self class] allocWithZone:zone] init:self.relationshipName fromKeyPath:self.sourceKeyPath toKeyPath:self.destinationKeyPath matcher:self.matcher withMapping:mapping];
}

- (void)dealloc {
    self.relationshipName = nil;
    self.destinationKeyPath = nil;
    self.mapping = nil;
    self.matcher = nil;
    self.sourceKeyPath = nil;
    [super dealloc];
}

- (BOOL)isToMany:(NSManagedObject *)source {
    NSEntityDescription *entity = [source entity];
    NSDictionary *relationships = [entity relationshipsByName];
    NSRelationshipDescription *relationship = [relationships objectForKey:self.relationshipName];
    return relationship.isToMany;
}

- (NSManagedObject*)findOneConnected:(NSManagedObject *)source sourceValue:(id)sourceValue  {
    RKManagedObjectMapping* objectMapping = (RKManagedObjectMapping *)self.mapping;
    NSObject<RKManagedObjectCaching> *cache = [[objectMapping objectStore] cacheStrategy];
    NSManagedObjectContext *context = [[objectMapping objectStore] managedObjectContextForCurrentThread];
    return [cache findInstanceOfEntity:objectMapping.entity withPrimaryKeyAttribute:self.destinationKeyPath value:sourceValue inManagedObjectContext:context];
}

- (NSMutableSet*)findAllConnected:(NSManagedObject *)source sourceValue:(id)sourceValue {
    NSMutableSet *result = [NSMutableSet set];

    RKManagedObjectMapping* objectMapping = (RKManagedObjectMapping *)self.mapping;
    NSObject<RKManagedObjectCaching> *cache = [[objectMapping objectStore] cacheStrategy];
    NSManagedObjectContext *context = [[objectMapping objectStore] managedObjectContextForCurrentThread];

    id values = nil;
    if ([sourceValue conformsToProtocol:@protocol(NSFastEnumeration)]) {
        values = sourceValue;
    } else {
        values = [NSArray arrayWithObject:sourceValue];
    }

    for (id value in values) {
        NSArray *objects = [cache findInstancesOfEntity:objectMapping.entity withPrimaryKeyAttribute:self.destinationKeyPath value:value inManagedObjectContext:context];
        [result addObjectsFromArray:objects];
    }
    return result;
}

- (BOOL)checkMatcher:(NSManagedObject *)source {
    if (!matcher) {
        return YES;
    }
    else {
        return [matcher isMatchForData:source];
    }
}

- (id)findConnected:(NSManagedObject *)source {
    if ([self checkMatcher:source])
    {
        BOOL isToMany = [self isToMany:source];
        id sourceValue = [source valueForKey:self.sourceKeyPath];
        if (isToMany) {
            return [self findAllConnected:source sourceValue:sourceValue];
        }
        else {
            return [self findOneConnected:source sourceValue:sourceValue];
        }
    }
    else {
        return nil;
    }
}
@end