Add cloud schema initialization

infinitetoken · Nov 15, 2019 · ee9be9f · ee9be9f
1 parent b7ea8fa
commit ee9be9f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Sources/Cabinet.swift b/Sources/Cabinet.swift
@@ -38,6 +38,9 @@ public class Cabinet: NSObject {
         container.viewContext.automaticallyMergesChangesFromParent = true
 
         do {
+            if let container = container as? NSPersistentCloudKitContainer, self.initializeSchema {
+                try container.initializeCloudKitSchema(options: self.initializeSchemaOptions)
+            }
             try container.viewContext.setQueryGenerationFrom(.current)
         } catch {
             fatalError("Failed to pin viewContext to the current generation: \(error)")
@@ -48,6 +51,8 @@ public class Cabinet: NSObject {
 
     public var containerName: String = "Database"
     public var usesCloudKit: Bool = false
+    public var initializeSchema: Bool = false
+    public var initializeSchemaOptions: NSPersistentCloudKitContainerSchemaInitializationOptions = []
 
     // MARK: - Lifecycle