yattee/Model/SearchModel.swift

103 lines
2.8 KiB
Swift
Raw Normal View History

2021-07-29 22:34:13 +00:00
import Defaults
import Siesta
import SwiftUI
2021-09-25 08:18:22 +00:00
final class SearchModel: ObservableObject {
2021-09-13 20:41:16 +00:00
@Published var store = Store<[Video]>()
@Published var api = InvidiousAPI()
2021-09-25 08:18:22 +00:00
@Published var query = SearchQuery()
2021-09-19 12:42:47 +00:00
@Published var queryText = ""
2021-09-13 20:41:16 +00:00
@Published var querySuggestions = Store<[String]>()
2021-07-29 22:34:13 +00:00
private var previousResource: Resource?
private var resource: Resource!
var isLoading: Bool {
2021-09-25 12:17:58 +00:00
resource?.isLoading ?? false
2021-09-13 20:41:16 +00:00
}
2021-07-29 22:34:13 +00:00
func changeQuery(_ changeHandler: @escaping (SearchQuery) -> Void = { _ in }) {
changeHandler(query)
2021-09-25 08:18:22 +00:00
let newResource = api.search(query)
2021-07-29 22:34:13 +00:00
guard newResource != previousResource else {
return
}
previousResource?.removeObservers(ownedBy: store)
previousResource = newResource
2021-09-19 11:06:54 +00:00
resource = newResource
resource.addObserver(store)
if !query.isEmpty {
loadResourceIfNeededAndReplaceStore()
}
2021-09-19 11:06:54 +00:00
}
func resetQuery(_ query: SearchQuery = SearchQuery()) {
2021-09-19 11:06:54 +00:00
self.query = query
2021-09-25 08:18:22 +00:00
let newResource = api.search(query)
2021-09-19 11:06:54 +00:00
guard newResource != previousResource else {
return
}
store.replace([])
previousResource?.removeObservers(ownedBy: store)
previousResource = newResource
2021-07-29 22:34:13 +00:00
resource = newResource
resource.addObserver(store)
if !query.isEmpty {
loadResourceIfNeededAndReplaceStore()
}
2021-07-29 22:34:13 +00:00
}
func loadResourceIfNeededAndReplaceStore() {
let currentResource = resource!
if let request = resource.loadIfNeeded() {
request.onSuccess { response in
if let videos: [Video] = response.typedContent() {
self.replace(videos, for: currentResource)
}
}
} else {
replace(store.collection, for: currentResource)
}
}
func replace(_ videos: [Video], for resource: Resource) {
if self.resource == resource {
store = Store<[Video]>(videos)
}
}
2021-09-25 12:17:58 +00:00
private var suggestionsDebounceTimer: Timer?
func loadSuggestions(_ query: String) {
suggestionsDebounceTimer?.invalidate()
suggestionsDebounceTimer = Timer.scheduledTimer(withTimeInterval: 1, repeats: false) { _ in
let resource = self.api.searchSuggestions(query: query)
resource.addObserver(self.querySuggestions)
resource.loadIfNeeded()
if let request = resource.loadIfNeeded() {
request.onSuccess { response in
if let suggestions: [String] = response.typedContent() {
self.querySuggestions = Store<[String]>(suggestions)
}
}
} else {
self.querySuggestions = Store<[String]>(self.querySuggestions.collection)
}
}
}
2021-07-29 22:34:13 +00:00
}