Adding protection mode while writing content to the file

path · Jan 18, 2017 · 8d8604a · 8d8604a
1 parent 2615d27
commit 8d8604a
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 2 deletions.
diff --git a/FastImageCache/FastImageCache/FastImageCache/FICImageFormat.h b/FastImageCache/FastImageCache/FastImageCache/FICImageFormat.h
@@ -147,6 +147,8 @@ NS_ASSUME_NONNULL_BEGIN
  */
 @property (nonatomic, assign, readonly) NSString *protectionModeString;
 
+@property (nonatomic, assign, readonly) NSDataWritingOptions protectionModeOption;
+
 /**
  The dictionary representation of this image format.
  
@@ -181,4 +183,4 @@ NS_ASSUME_NONNULL_BEGIN
 + (instancetype)formatWithName:(NSString *)name family:(NSString *)family imageSize:(CGSize)imageSize style:(FICImageFormatStyle)style maximumCount:(NSInteger)maximumCount devices:(FICImageFormatDevices)devices protectionMode:(FICImageFormatProtectionMode)protectionMode;
 
 @end
-NS_ASSUME_NONNULL_END
+NS_ASSUME_NONNULL_END
diff --git a/FastImageCache/FastImageCache/FastImageCache/FICImageFormat.m b/FastImageCache/FastImageCache/FastImageCache/FICImageFormat.m
@@ -143,6 +143,22 @@ - (NSString *)protectionModeString {
     return protectionModeString;
 }
 
+- (NSDataWritingOptions)protectionModeOption {
+    NSDataWritingOptions protectionModeOption = NSDataWritingFileProtectionNone;
+    switch (_protectionMode) {
+        case FICImageFormatProtectionModeNone:
+            protectionModeOption = NSDataWritingFileProtectionNone;
+            break;
+        case FICImageFormatProtectionModeComplete:
+            protectionModeOption = NSDataWritingFileProtectionComplete;
+            break;
+        case FICImageFormatProtectionModeCompleteUntilFirstUserAuthentication:
+            protectionModeOption = NSDataWritingFileProtectionCompleteUntilFirstUserAuthentication;
+            break;
+    }
+    return protectionModeOption;
+}
+
 #pragma mark - Object Lifecycle
 
 + (instancetype)formatWithName:(NSString *)name family:(NSString *)family imageSize:(CGSize)imageSize style:(FICImageFormatStyle)style maximumCount:(NSInteger)maximumCount devices:(FICImageFormatDevices)devices protectionMode:(FICImageFormatProtectionMode)protectionMode {

diff --git a/FastImageCache/FastImageCache/FastImageCache/FICImageTable.m b/FastImageCache/FastImageCache/FastImageCache/FICImageTable.m
@@ -714,7 +714,7 @@ - (void)saveMetadata {
                     return;
                 }
 
-                BOOL fileWriteResult = [data writeToFile:[self metadataFilePath] atomically:NO];
+                BOOL fileWriteResult = [data writeToFile:[self metadataFilePath] options:[_imageFormat protectionModeOption] error:nil];
                 if (fileWriteResult == NO) {
                     NSString *message = [NSString stringWithFormat:@"*** FIC Error: %s couldn't write metadata for format %@", __PRETTY_FUNCTION__, [_imageFormat name]];
                     [self.imageCache _logMessage:message];