Reflect/Shared/Persisence/Persistence.swift

//
//  Persistence.swift
//  Shared
//
//  Created by Trey Tartt on 1/5/22.
//

import CoreData
import SwiftUI

class PersistenceController {
    @AppStorage(UserDefaultsStore.Keys.useCloudKit.rawValue, store: GroupUserDefaults.groupDefaults)
    
    private var useCloudKit = false
    
    static let shared = PersistenceController.persistenceController
        
    private static var persistenceController: PersistenceController {
        return PersistenceController(inMemory: true)
    }
    
    public var viewContext: NSManagedObjectContext {
        return PersistenceController.shared.container.viewContext
    }
    
    public lazy var childContext: NSManagedObjectContext = {
        NSManagedObjectContext(concurrencyType: .mainQueueConcurrencyType)
    }()
    
    public var switchContainerListeners = [(() -> Void)]()
    
    private var editedDataClosure = [() -> Void]()
    
    public var earliestEntry: MoodEntry? {
        let fetchRequest = NSFetchRequest<MoodEntry>(entityName: "MoodEntry")
        fetchRequest.sortDescriptors = [NSSortDescriptor(key: "forDate", ascending: true)]
        let first = try! viewContext.fetch(fetchRequest).first
        return first ?? nil
    }
    
    public var latestEntry: MoodEntry? {
        let fetchRequest = NSFetchRequest<MoodEntry>(entityName: "MoodEntry")
        fetchRequest.sortDescriptors = [NSSortDescriptor(key: "forDate", ascending: true)]
        let last = try! viewContext.fetch(fetchRequest).last
        return last ?? nil
    }

    lazy var container: NSPersistentContainer = {
        setupContainer()
    }()
    
    func switchContainer() {
        try? viewContext.save()
        container = setupContainer()
        for item in switchContainerListeners {
            item()
        }
    }
    
    public func addNewDataListener(closure: @escaping (() -> Void)) {
        editedDataClosure.append(closure)
    }
    
    public func saveAndRunDataListerners() {
        do {
            try viewContext.save()
            
            for closure in editedDataClosure {
                closure()
            }
        } catch {
            print(error)
        }
    }
    
    private func setupContainer() -> NSPersistentContainer {
        if useCloudKit {
            container = NSPersistentCloudKitContainer(name: "Feels")
        } else {
            container = NSCustomPersistentContainer(name: "Feels")
        }
            
        for description in container.persistentStoreDescriptions {
            description.setOption(true as NSNumber, forKey: NSPersistentStoreRemoteChangeNotificationPostOptionKey)
            description.setOption(true as NSNumber, forKey: NSPersistentHistoryTrackingKey)
            description.setOption(true as NSNumber, forKey: NSMigratePersistentStoresAutomaticallyOption)
            description.setOption(true as NSNumber, forKey: NSInferMappingModelAutomaticallyOption)
        }
        
        container.loadPersistentStores(completionHandler: { (storeDescription, error) in
            self.container.viewContext.automaticallyMergesChangesFromParent = true

            if let error = error as NSError? {
                fatalError("Unresolved error \(error), \(error.userInfo)")
            }
        })
        
        return container
    }
        
    init(inMemory: Bool = false) {
        container = setupContainer()
    }
}

extension NSManagedObjectContext {
    /// Executes the given `NSBatchDeleteRequest` and directly merges the changes to bring the given managed object context up to date.
    ///
    /// - Parameter batchDeleteRequest: The `NSBatchDeleteRequest` to execute.
    /// - Throws: An error if anything went wrong executing the batch deletion.
    public func executeAndMergeChanges(using batchDeleteRequest: NSBatchDeleteRequest) throws {
        batchDeleteRequest.resultType = .resultTypeObjectIDs
        let result = try execute(batchDeleteRequest) as? NSBatchDeleteResult
        let changes: [AnyHashable: Any] = [NSDeletedObjectsKey: result?.result as? [NSManagedObjectID] ?? []]
        NSManagedObjectContext.mergeChanges(fromRemoteContextSave: changes, into: [self])
    }
}

class NSCustomPersistentContainer: NSPersistentContainer {
    override open class func defaultDirectoryURL() -> URL {
#if DEBUG
        if let storeURLDebug = FileManager.default.containerURL(forSecurityApplicationGroupIdentifier: Constants.groupShareIdDebug) {
            return storeURLDebug.appendingPathComponent("Feels-Debug.sqlite")
        }
        // Fallback to default location if App Group not available
        print("⚠️ App Group not available, using default Core Data location")
        return super.defaultDirectoryURL()
#else
        if let storeURL = FileManager.default.containerURL(forSecurityApplicationGroupIdentifier: Constants.groupShareId) {
            return storeURL.appendingPathComponent("Feels.sqlite")
        }
        // Fallback to default location if App Group not available
        print("⚠️ App Group not available, using default Core Data location")
        return super.defaultDirectoryURL()
#endif
    }
}