Filename for HKDataStore

Author: gstf

Common/Classes/HKDataStore.h

@@ -51,6 +51,7 @@ typedef void (^HKDataStoreSaveHandler)( NSManagedObjectContext *context, NSSet *
     NSMutableSet                    *_chandlers;
     NSMutableSet                    *_shandlers;
     NSMutableSet                    *_detached;
+    NSString                        *_filename;
     HKDataStoreLocation              _location;
     BOOL                             _setup;
     BOOL                             _change;
@@ -65,6 +66,8 @@ typedef void (^HKDataStoreSaveHandler)( NSManagedObjectContext *context, NSSet *
 
 + (HKDataStore *)defaultStore;
 
+- (void)setFilename:(NSString *)filename;
+
 - (void)addModelBundle:(NSBundle *)bundle;
 - (void)setDataStoreLocation:(HKDataStoreLocation)location;
 - (void)setUsesLightweightMigration:(BOOL)flag;

Common/Classes/HKDataStore.m

@@ -83,6 +83,7 @@ - (id)init
 {
     if ( self = [super init] )
     {
+        [self setFilename:@"Data.db"];
     }
 
     return self;
@@ -117,6 +118,17 @@ + (HKDataStore *)defaultStore
     return gHKDataStore;
 }
 
+- (void)setFilename:(NSString *)filename
+{
+    if ( !_setup )
+    {
+        if ( filename != _filename )
+        {
+            _filename = [filename copy];
+        }
+    }
+}
+
 - (void)addModelBundle:(NSBundle *)bundle
 {
     @synchronized (self)
@@ -470,7 +482,7 @@ - (void)setup
                 break;
         }
 
-        url = [NSURL fileURLWithPath:[dir stringByAppendingPathComponent:@"Data.db"]];
+        url = [NSURL fileURLWithPath:[dir stringByAppendingPathComponent:_filename]];
 
         if ( ![fm fileExistsAtPath:dir isDirectory:NULL] )
         {