import CloudKit

func fetchAllRecords(from recordType: String, completion: @escaping ([CKRecord]?, Error?) -> Void) {
    let container = CKContainer.default()
    let publicDatabase = container.publicCloudDatabase
    
    let query = CKQuery(recordType: recordType, predicate: NSPredicate(value: true))
    
    publicDatabase.perform(query, inZoneWith: nil) { (records, error) in
        DispatchQueue.main.async {
            if let error = error {
                completion(nil, error)
                return
            }
            completion(records, nil)
        }
    }
}

func fetchAllRecordsPaginated(from recordType: String, completion: @escaping ([CKRecord]?, Error?) -> Void) {
    let container = CKContainer.default()
    let publicDatabase = container.publicCloudDatabase
    
    var allRecords = [CKRecord]()
    
    let query = CKQuery(recordType: recordType, predicate: NSPredicate(value: true))
    let operation = CKQueryOperation(query: query)
    operation.resultsLimit = 100 // 每批获取100条记录
    
    operation.recordFetchedBlock = { record in
        allRecords.append(record)
    }
    
    operation.queryCompletionBlock = { (cursor, error) in
        if let error = error {
            DispatchQueue.main.async {
                completion(nil, error)
            }
            return
        }
        
        if let cursor = cursor {
            // 还有更多记录，继续查询
            let newOperation = CKQueryOperation(cursor: cursor)
            newOperation.resultsLimit = operation.resultsLimit
            newOperation.recordFetchedBlock = operation.recordFetchedBlock
            newOperation.queryCompletionBlock = operation.queryCompletionBlock
            publicDatabase.add(newOperation)
        } else {
            // 所有记录已获取完毕
            DispatchQueue.main.async {
                completion(allRecords, nil)
            }
        }
    }
    
    publicDatabase.add(operation)
}