RestKit/Code/RKObjectSeeder.m

//
//  RKObjectSeeder.m
//  RestKit
//
//  Created by Blake Watters on 3/4/10.
//  Copyright 2010 Two Toasters. All rights reserved.
//

#import "RKObjectSeeder.h"

@implementation RKObjectSeeder

- (id)initWithResourceManager:(RKResourceManager*)manager {
	if (self = [self init]) {
		_manager = [manager retain];
	}
	
	return self;
}

- (void)dealloc {
	[_manager release];
	[super dealloc];
}

- (NSArray*)seedDatabaseWithBundledFile:(NSString*)fileName ofType:(NSString*)type {
	NSError* error = nil;
	NSString* filePath = [[NSBundle mainBundle] pathForResource:fileName ofType:type];
	NSString* payload = [NSString stringWithContentsOfFile:filePath encoding:NSUTF8StringEncoding error:&error];
	if (nil == error) {
		return [[_manager mapper] mapFromString:payload];
	}
	
	return nil;
}

- (void)seedDatabaseWithBundledFiles:(NSArray*)fileNames ofType:(NSString*)type {
	NSLog(@"[RestKit] RKModelSeeder: Seeding database with contents of %d %@ files...", [fileNames count], [type uppercaseString]);
	for (NSString* fileName in fileNames) {
		NSArray* objects = [self seedDatabaseWithBundledFile:fileName ofType:type];
		NSLog(@"[RestKit] RKModelSeeder: Seeded %d objects from %@...", [objects count], [NSString stringWithFormat:@"%@.%@", fileName, type]);
	}
	
	NSError* error = [[_manager objectStore] save];
	if (error != nil) {
		NSLog(@"[RestKit] RKModelSeeder: Error saving object context: %@", [error localizedDescription]);
	}
	
	NSArray* paths = NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES);
	NSString* basePath = ([paths count] > 0) ? [paths objectAtIndex:0] : nil;
	NSString* storeFileName = [[_manager objectStore] storeFilename];
	NSString* destinationPath = [basePath stringByAppendingPathComponent:storeFileName];
	NSLog(@"[RestKit] RKModelSeeder: A Pre-loaded database has been generated at %@. Please copy into Resources/", destinationPath);
	
	exit(1);
}

@end