125 lines
4.7 KiB
Swift
125 lines
4.7 KiB
Swift
//
|
|
/*
|
|
* SwiftFin is subject to the terms of the Mozilla Public
|
|
* License, v2.0. If a copy of the MPL was not distributed with this
|
|
* file, you can obtain one at https://mozilla.org/MPL/2.0/.
|
|
*
|
|
* Copyright 2021 Aiden Vigue & Jellyfin Contributors
|
|
*/
|
|
|
|
import Combine
|
|
import Foundation
|
|
import JellyfinAPI
|
|
import Stinsen
|
|
|
|
struct AddServerURIPayload: Identifiable {
|
|
|
|
let server: SwiftfinStore.State.Server
|
|
let uri: String
|
|
|
|
var id: String {
|
|
return server.id.appending(uri)
|
|
}
|
|
}
|
|
|
|
final class ConnectToServerViewModel: ViewModel {
|
|
|
|
@RouterObject var router: ConnectToServerCoodinator.Router?
|
|
@Published var discoveredServers: Set<ServerDiscovery.ServerLookupResponse> = []
|
|
@Published var searching = false
|
|
@Published var addServerURIPayload: AddServerURIPayload?
|
|
var backAddServerURIPayload: AddServerURIPayload?
|
|
|
|
private let discovery = ServerDiscovery()
|
|
|
|
var alertTitle: String {
|
|
var message: String = ""
|
|
if errorMessage?.code != ErrorMessage.noShowErrorCode {
|
|
message.append(contentsOf: "\(errorMessage?.code ?? ErrorMessage.noShowErrorCode)\n")
|
|
}
|
|
message.append(contentsOf: "\(errorMessage?.title ?? "Unkown Error")")
|
|
return message
|
|
}
|
|
|
|
func connectToServer(uri: String) {
|
|
#if targetEnvironment(simulator)
|
|
var uri = uri
|
|
if uri == "localhost" {
|
|
uri = "http://localhost:8096"
|
|
}
|
|
#endif
|
|
|
|
LogManager.shared.log.debug("Attempting to connect to server at \"\(uri)\"", tag: "connectToServer")
|
|
SessionManager.main.connectToServer(with: uri)
|
|
.trackActivity(loading)
|
|
.sink(receiveCompletion: { completion in
|
|
// This is disgusting. ViewModel Error handling overall needs to be refactored
|
|
switch completion {
|
|
case .finished: ()
|
|
case .failure(let error):
|
|
switch error {
|
|
case is SwiftfinStore.Errors:
|
|
let swiftfinError = error as! SwiftfinStore.Errors
|
|
switch swiftfinError {
|
|
case .existingServer(let server):
|
|
self.addServerURIPayload = AddServerURIPayload(server: server, uri: uri)
|
|
self.backAddServerURIPayload = AddServerURIPayload(server: server, uri: uri)
|
|
default:
|
|
self.handleAPIRequestError(displayMessage: "Unable to connect to server.", logLevel: .critical, tag: "connectToServer",
|
|
completion: completion)
|
|
}
|
|
default:
|
|
self.handleAPIRequestError(displayMessage: "Unable to connect to server.", logLevel: .critical, tag: "connectToServer",
|
|
completion: completion)
|
|
}
|
|
}
|
|
}, receiveValue: { server in
|
|
LogManager.shared.log.debug("Connected to server at \"\(uri)\"", tag: "connectToServer")
|
|
self.router?.route(to: \.userSignIn, server)
|
|
})
|
|
.store(in: &cancellables)
|
|
}
|
|
|
|
func discoverServers() {
|
|
discoveredServers.removeAll()
|
|
searching = true
|
|
|
|
// Timeout after 3 seconds
|
|
DispatchQueue.main.asyncAfter(deadline: .now() + 3) {
|
|
self.searching = false
|
|
}
|
|
|
|
discovery.locateServer { [self] server in
|
|
if let server = server {
|
|
discoveredServers.insert(server)
|
|
}
|
|
}
|
|
}
|
|
|
|
func addURIToServer(addServerURIPayload: AddServerURIPayload) {
|
|
SessionManager.main.addURIToServer(server: addServerURIPayload.server, uri: addServerURIPayload.uri)
|
|
.sink { completion in
|
|
self.handleAPIRequestError(displayMessage: "Unable to connect to server.", logLevel: .critical, tag: "connectToServer",
|
|
completion: completion)
|
|
} receiveValue: { server in
|
|
SessionManager.main.setServerCurrentURI(server: server, uri: addServerURIPayload.uri)
|
|
.sink { completion in
|
|
self.handleAPIRequestError(displayMessage: "Unable to connect to server.", logLevel: .critical, tag: "connectToServer",
|
|
completion: completion)
|
|
} receiveValue: { server in
|
|
self.router?.dismissCoordinator()
|
|
}
|
|
.store(in: &self.cancellables)
|
|
}
|
|
.store(in: &cancellables)
|
|
}
|
|
|
|
func cancelConnection() {
|
|
for cancellable in cancellables {
|
|
cancellable.cancel()
|
|
}
|
|
|
|
self.isLoading = false
|
|
}
|
|
}
|