mirror of
https://github.com/kean/Pulse.git
synced 2026-05-30 21:07:33 +00:00
kean
126 lines
3.9 KiB
Swift
126 lines
3.9 KiB
Swift
// The MIT License (MIT)
|
||
//
|
||
// Copyright (c) 2020–2022 Alexander Grebenyuk (github.com/kean).
|
||
|
||
import CoreData
|
||
import Pulse
|
||
import Combine
|
||
import SwiftUI
|
||
|
||
#if os(iOS) || os(macOS) || os(tvOS)
|
||
|
||
final class NetworkViewModel: NSObject, NSFetchedResultsControllerDelegate, ObservableObject {
|
||
#if os(iOS) || os(macOS)
|
||
let table: ConsoleTableViewModel
|
||
#endif
|
||
@Published private(set) var entities: [NetworkTaskEntity] = []
|
||
|
||
let details: ConsoleDetailsRouterViewModel
|
||
|
||
// Search criteria
|
||
let searchCriteria: NetworkSearchCriteriaViewModel
|
||
@Published var isOnlyErrors: Bool = false
|
||
@Published var filterTerm: String = ""
|
||
|
||
var onDismiss: (() -> Void)?
|
||
|
||
private let store: LoggerStore
|
||
private let controller: NSFetchedResultsController<NetworkTaskEntity>
|
||
private var isActive = false
|
||
private var latestSessionId: UUID?
|
||
private var cancellables = [AnyCancellable]()
|
||
|
||
init(store: LoggerStore) {
|
||
self.store = store
|
||
self.details = ConsoleDetailsRouterViewModel()
|
||
|
||
let request = NSFetchRequest<NetworkTaskEntity>(entityName: "\(NetworkTaskEntity.self)")
|
||
request.fetchBatchSize = 100
|
||
request.sortDescriptors = [NSSortDescriptor(keyPath: \NetworkTaskEntity.createdAt, ascending: false)]
|
||
|
||
self.controller = NSFetchedResultsController<NetworkTaskEntity>(fetchRequest: request, managedObjectContext: store.viewContext, sectionNameKeyPath: nil, cacheName: nil)
|
||
|
||
self.searchCriteria = NetworkSearchCriteriaViewModel(store: store)
|
||
#if os(iOS) || os(macOS)
|
||
self.table = ConsoleTableViewModel(searchCriteriaViewModel: nil)
|
||
#endif
|
||
|
||
super.init()
|
||
|
||
controller.delegate = self
|
||
|
||
$filterTerm.dropFirst().sink { [weak self] filterTerm in
|
||
self?.refresh(filterTerm: filterTerm)
|
||
}.store(in: &cancellables)
|
||
|
||
searchCriteria.dataNeedsReload.throttle(for: 0.5, scheduler: DispatchQueue.main, latest: true).sink { [weak self] in
|
||
self?.refreshNow()
|
||
}.store(in: &cancellables)
|
||
|
||
$isOnlyErrors.receive(on: DispatchQueue.main).dropFirst().sink { [weak self] _ in
|
||
self?.refreshNow()
|
||
}.store(in: &cancellables)
|
||
|
||
refreshNow()
|
||
}
|
||
|
||
// MARK: Appearance
|
||
|
||
func onAppear() {
|
||
isActive = true
|
||
didRefreshEntities()
|
||
}
|
||
|
||
func onDisappear() {
|
||
isActive = false
|
||
}
|
||
|
||
// MARK: Refresh
|
||
|
||
private func refreshNow() {
|
||
refresh(filterTerm: filterTerm)
|
||
}
|
||
|
||
private func refresh(filterTerm: String) {
|
||
// Get sessionId
|
||
if latestSessionId == nil {
|
||
latestSessionId = entities.first?.session
|
||
}
|
||
let sessionId = store === LoggerStore.shared ? LoggerStore.Session.current.id : latestSessionId
|
||
|
||
// Search messages
|
||
NetworkSearchCriteria.update(request: controller.fetchRequest, filterTerm: filterTerm, criteria: searchCriteria.criteria, filters: searchCriteria.filters, isOnlyErrors: isOnlyErrors, sessionId: sessionId)
|
||
try? controller.performFetch()
|
||
|
||
self.didRefreshEntities()
|
||
}
|
||
|
||
// MARK: - NSFetchedResultsControllerDelegate
|
||
|
||
func controller(_ controller: NSFetchedResultsController<NSFetchRequestResult>, didChangeContentWith diff: CollectionDifference<NSManagedObjectID>) {
|
||
guard isActive else { return }
|
||
|
||
#if os(iOS) || os(macOS)
|
||
self.table.diff = diff
|
||
#endif
|
||
withAnimation {
|
||
self.didRefreshEntities()
|
||
}
|
||
}
|
||
|
||
private func didRefreshEntities() {
|
||
// Apply filters that couldn't be done programmatically
|
||
if let filters = searchCriteria.programmaticFilters {
|
||
let objects = controller.fetchedObjects ?? []
|
||
self.entities = objects.filter { evaluateProgrammaticFilters(filters, entity: $0, store: store) }
|
||
} else {
|
||
self.entities = controller.fetchedObjects ?? []
|
||
}
|
||
#if os(iOS) || os(macOS)
|
||
self.table.entities = self.entities
|
||
#endif
|
||
}
|
||
}
|
||
|
||
#endif
|