yattee/Model/Accounts/AccountValidator.swift

256 lines
7.7 KiB
Swift
Raw Normal View History

2022-12-04 10:39:32 +00:00
import Alamofire
2021-09-25 08:18:22 +00:00
import Foundation
import Siesta
import SwiftUI
final class AccountValidator: Service {
let app: Binding<VideosApp?>
2021-09-25 08:18:22 +00:00
let url: String
let account: Account!
2021-09-25 08:18:22 +00:00
var formObjectID: Binding<String>
var isValid: Binding<Bool>
var isValidated: Binding<Bool>
var isValidating: Binding<Bool>
2021-09-25 08:18:22 +00:00
var error: Binding<String?>?
private var appsToValidateInstance = VideosApp.allCases
2021-09-25 08:18:22 +00:00
init(
app: Binding<VideosApp?>,
2021-09-25 08:18:22 +00:00
url: String,
2021-10-20 22:21:50 +00:00
account: Account? = nil,
id: Binding<String>,
isValid: Binding<Bool>,
isValidated: Binding<Bool>,
isValidating: Binding<Bool>,
2021-09-25 08:18:22 +00:00
error: Binding<String?>? = nil
) {
2021-10-16 22:48:58 +00:00
self.app = app
2021-09-25 08:18:22 +00:00
self.url = url
self.account = account
formObjectID = id
self.isValid = isValid
self.isValidated = isValidated
self.isValidating = isValidating
2021-09-25 08:18:22 +00:00
self.error = error
super.init(baseURL: url)
configure()
}
func configure() {
2021-10-16 22:48:58 +00:00
configure {
$0.pipeline[.parsing].add(SwiftyJSONTransformer, contentTypes: ["*/json"])
}
configure("/login", requestMethods: [.post]) {
$0.headers["Content-Type"] = "application/json"
2021-09-25 08:18:22 +00:00
}
}
func instanceValidationResource(_ app: VideosApp) -> Resource {
switch app {
case .invidious:
return resource("/api/v1/videos/dQw4w9WgXcQ")
case .piped:
return resource("/streams/dQw4w9WgXcQ")
2022-12-09 00:15:19 +00:00
case .peerTube:
// TODO: fixme
return resource("")
case .local:
return resource("")
}
}
2021-09-25 08:18:22 +00:00
func validateInstance() {
reset()
guard let app = appsToValidateInstance.popLast() else { return }
tryValidatingUsing(app)
}
func tryValidatingUsing(_ app: VideosApp) {
instanceValidationResource(app)
2021-09-25 08:18:22 +00:00
.load()
2021-10-16 22:48:58 +00:00
.onSuccess { response in
2021-09-25 08:18:22 +00:00
guard self.url == self.formObjectID.wrappedValue else {
return
}
guard !response.json.isEmpty else {
guard let app = self.appsToValidateInstance.popLast() else {
self.isValid.wrappedValue = false
self.isValidated.wrappedValue = true
self.isValidating.wrappedValue = false
return
}
self.tryValidatingUsing(app)
return
}
2021-11-07 21:39:28 +00:00
let json = response.json.dictionaryValue
let author = app == .invidious ? json["author"] : json["uploader"]
2021-11-07 21:39:28 +00:00
if author == "Rick Astley" {
self.app.wrappedValue = app
2021-10-16 22:48:58 +00:00
self.isValid.wrappedValue = true
self.error?.wrappedValue = nil
} else {
self.isValid.wrappedValue = false
}
self.isValidated.wrappedValue = true
self.isValidating.wrappedValue = false
2021-09-25 08:18:22 +00:00
}
.onFailure { error in
guard self.url == self.formObjectID.wrappedValue else {
return
}
if self.appsToValidateInstance.isEmpty {
self.isValidating.wrappedValue = false
self.isValidated.wrappedValue = true
self.isValid.wrappedValue = false
self.error?.wrappedValue = error.userMessage
} else {
guard let app = self.appsToValidateInstance.popLast() else { return }
self.tryValidatingUsing(app)
}
2021-09-25 08:18:22 +00:00
}
}
func validateAccount() {
2021-09-25 08:18:22 +00:00
reset()
2022-12-04 10:39:32 +00:00
switch app.wrappedValue {
case .invidious:
invidiousValidation()
case .piped:
pipedValidation()
default:
setValidationResult(false)
}
}
func invidiousValidation() {
guard let username = account?.username,
let password = account?.password
else {
setValidationResult(false)
return
}
AF
.request(login.url, method: .post, parameters: ["email": username, "password": password], encoding: URLEncoding.default)
.redirect(using: .doNotFollow)
.response { response in
guard let headers = response.response?.headers,
let cookies = headers["Set-Cookie"]
else {
self.setValidationResult(false)
return
}
let sidRegex = #"SID=(?<sid>[^;]*);"#
guard let sidRegex = try? NSRegularExpression(pattern: sidRegex),
let match = sidRegex.matches(in: cookies, range: NSRange(cookies.startIndex..., in: cookies)).first
else {
self.setValidationResult(false)
return
}
let matchRange = match.range(withName: "sid")
if let substringRange = Range(matchRange, in: cookies) {
let sid = String(cookies[substringRange])
if !sid.isEmpty {
self.setValidationResult(true)
}
} else {
self.setValidationResult(false)
}
}
}
func pipedValidation() {
guard let request = accountRequest else {
2022-12-04 10:39:32 +00:00
setValidationResult(false)
2021-09-25 08:18:22 +00:00
return
}
request.onSuccess { response in
guard self.account!.username == self.formObjectID.wrappedValue else {
return
2021-09-25 08:18:22 +00:00
}
switch self.app.wrappedValue {
case .invidious:
self.isValid.wrappedValue = true
case .piped:
let error = response.json.dictionaryValue["error"]?.string
let token = response.json.dictionaryValue["token"]?.string
self.isValid.wrappedValue = error?.isEmpty ?? !(token?.isEmpty ?? true)
self.error!.wrappedValue = error
default:
return
}
}
.onFailure { _ in
guard self.account!.username == self.formObjectID.wrappedValue else {
return
2021-09-25 08:18:22 +00:00
}
self.isValid.wrappedValue = false
}
.onCompletion { _ in
self.isValidated.wrappedValue = true
self.isValidating.wrappedValue = false
}
2021-09-25 08:18:22 +00:00
}
2022-12-04 10:39:32 +00:00
func setValidationResult(_ result: Bool) {
isValid.wrappedValue = result
isValidated.wrappedValue = true
isValidating.wrappedValue = false
}
var accountRequest: Siesta.Request? {
switch app.wrappedValue {
case .invidious:
guard let password = account.password else { return nil }
return login.request(.post, urlEncoded: ["email": account.username, "password": password])
case .piped:
return login.request(.post, json: ["username": account.username, "password": account.password])
default:
return nil
}
}
2021-09-25 08:18:22 +00:00
func reset() {
appsToValidateInstance = VideosApp.allCases
app.wrappedValue = nil
isValid.wrappedValue = false
isValidated.wrappedValue = false
isValidating.wrappedValue = false
2021-09-25 08:18:22 +00:00
error?.wrappedValue = nil
}
var login: Resource {
resource("/login")
2021-09-25 08:18:22 +00:00
}
2021-11-07 21:39:28 +00:00
var videoResourceBasePath: String {
app.wrappedValue == .invidious ? "/api/v1/videos" : "/streams"
}
var neverGonnaGiveYouUp: Resource {
resource("\(videoResourceBasePath)/dQw4w9WgXcQ")
}
2021-09-25 08:18:22 +00:00
}