mirror of
https://github.com/cranci1/Sora.git
synced 2026-04-18 15:12:09 +00:00
298 lines
16 KiB
Swift
298 lines
16 KiB
Swift
//
|
|
// JSLoader-Streams.swift
|
|
// Sulfur
|
|
//
|
|
// Created by Francesco on 30/03/25.
|
|
//
|
|
|
|
import JavaScriptCore
|
|
|
|
extension JSController {
|
|
|
|
func fetchStreamUrl(episodeUrl: String, softsub: Bool = false, module: ScrapingModule, completion: @escaping ((streams: [String]?, subtitles: [String]?, sources: [[String:Any]]? )) -> Void) {
|
|
guard let url = URL(string: episodeUrl) else {
|
|
completion((nil, nil,nil))
|
|
return
|
|
}
|
|
|
|
URLSession.custom.dataTask(with: url) { [weak self] data, _, error in
|
|
guard let self = self else { return }
|
|
|
|
if let error = error {
|
|
Logger.shared.log("Network error: \(error)", type: "Error")
|
|
DispatchQueue.main.async { completion((nil, nil,nil)) }
|
|
return
|
|
}
|
|
|
|
guard let data = data, let html = String(data: data, encoding: .utf8) else {
|
|
Logger.shared.log("Failed to decode HTML", type: "Error")
|
|
DispatchQueue.main.async { completion((nil, nil, nil)) }
|
|
return
|
|
}
|
|
|
|
Logger.shared.log(html, type: "HTMLStrings")
|
|
if let parseFunction = self.context.objectForKeyedSubscript("extractStreamUrl"),
|
|
let resultString = parseFunction.call(withArguments: [html]).toString() {
|
|
if let data = resultString.data(using: .utf8) {
|
|
do {
|
|
if let json = try? JSONSerialization.jsonObject(with: data, options: []) as? [String: Any] {
|
|
print("JSON DATA IS \(json) 2")
|
|
var streamUrls: [String]? = nil
|
|
var subtitleUrls: [String]? = nil
|
|
var streamUrlsAndHeaders : [[String:Any]]? = nil
|
|
if let streamSources = json["streams"] as? [[String:Any]]
|
|
{
|
|
streamUrlsAndHeaders = streamSources
|
|
Logger.shared.log("Found \(streamSources.count) streams and headers", type: "Stream")
|
|
}
|
|
else if let streamSource = json["stream"] as? [String:Any]
|
|
{
|
|
streamUrlsAndHeaders = [streamSource]
|
|
Logger.shared.log("Found single stream with headers", type: "Stream")
|
|
}
|
|
else if let streamsArray = json["streams"] as? [String] {
|
|
streamUrls = streamsArray
|
|
Logger.shared.log("Found \(streamsArray.count) streams", type: "Stream")
|
|
} else if let streamUrl = json["stream"] as? String {
|
|
streamUrls = [streamUrl]
|
|
Logger.shared.log("Found single stream", type: "Stream")
|
|
}
|
|
|
|
if let subsArray = json["subtitles"] as? [String] {
|
|
subtitleUrls = subsArray
|
|
Logger.shared.log("Found \(subsArray.count) subtitle tracks", type: "Stream")
|
|
} else if let subtitleUrl = json["subtitles"] as? String {
|
|
subtitleUrls = [subtitleUrl]
|
|
Logger.shared.log("Found single subtitle track", type: "Stream")
|
|
}
|
|
|
|
Logger.shared.log("Starting stream with \(streamUrls?.count ?? 0) sources and \(subtitleUrls?.count ?? 0) subtitles", type: "Stream")
|
|
DispatchQueue.main.async {
|
|
completion((streamUrls, subtitleUrls,streamUrlsAndHeaders))
|
|
}
|
|
return
|
|
}
|
|
|
|
if let streamsArray = try? JSONSerialization.jsonObject(with: data, options: []) as? [String] {
|
|
Logger.shared.log("Starting multi-stream with \(streamsArray.count) sources", type: "Stream")
|
|
DispatchQueue.main.async { completion((streamsArray, nil,nil)) }
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
Logger.shared.log("Starting stream from: \(resultString)", type: "Stream")
|
|
DispatchQueue.main.async { completion(([resultString], nil,nil)) }
|
|
} else {
|
|
Logger.shared.log("Failed to extract stream URL", type: "Error")
|
|
DispatchQueue.main.async { completion((nil, nil,nil)) }
|
|
}
|
|
}.resume()
|
|
}
|
|
|
|
func fetchStreamUrlJS(episodeUrl: String, softsub: Bool = false, module: ScrapingModule, completion: @escaping ((streams: [String]?, subtitles: [String]?,sources: [[String:Any]]? )) -> Void) {
|
|
if let exception = context.exception {
|
|
Logger.shared.log("JavaScript exception: \(exception)", type: "Error")
|
|
completion((nil, nil,nil))
|
|
return
|
|
}
|
|
|
|
guard let extractStreamUrlFunction = context.objectForKeyedSubscript("extractStreamUrl") else {
|
|
Logger.shared.log("No JavaScript function extractStreamUrl found", type: "Error")
|
|
completion((nil, nil,nil))
|
|
return
|
|
}
|
|
|
|
let promiseValue = extractStreamUrlFunction.call(withArguments: [episodeUrl])
|
|
guard let promise = promiseValue else {
|
|
Logger.shared.log("extractStreamUrl did not return a Promise", type: "Error")
|
|
completion((nil, nil,nil))
|
|
return
|
|
}
|
|
|
|
let thenBlock: @convention(block) (JSValue) -> Void = { [weak self] result in
|
|
guard self != nil else { return }
|
|
|
|
if let jsonString = result.toString(),
|
|
let data = jsonString.data(using: .utf8) {
|
|
do {
|
|
if let json = try? JSONSerialization.jsonObject(with: data, options: []) as? [String: Any] {
|
|
print("JSON object is \(json) 1")
|
|
var streamUrls: [String]? = nil
|
|
var subtitleUrls: [String]? = nil
|
|
var streamUrlsAndHeaders : [[String:Any]]? = nil
|
|
if let streamSources = json["streams"] as? [[String:Any]]
|
|
{
|
|
streamUrlsAndHeaders = streamSources
|
|
Logger.shared.log("Found \(streamSources.count) streams and headers", type: "Stream")
|
|
}
|
|
else if let streamSource = json["stream"] as? [String:Any]
|
|
{
|
|
streamUrlsAndHeaders = [streamSource]
|
|
Logger.shared.log("Found single stream with headers", type: "Stream")
|
|
}
|
|
else if let streamsArray = json["streams"] as? [String] {
|
|
streamUrls = streamsArray
|
|
Logger.shared.log("Found \(streamsArray.count) streams", type: "Stream")
|
|
} else if let streamUrl = json["stream"] as? String {
|
|
streamUrls = [streamUrl]
|
|
Logger.shared.log("Found single stream", type: "Stream")
|
|
}
|
|
|
|
if let subsArray = json["subtitles"] as? [String] {
|
|
subtitleUrls = subsArray
|
|
Logger.shared.log("Found \(subsArray.count) subtitle tracks", type: "Stream")
|
|
} else if let subtitleUrl = json["subtitles"] as? String {
|
|
subtitleUrls = [subtitleUrl]
|
|
Logger.shared.log("Found single subtitle track", type: "Stream")
|
|
}
|
|
|
|
Logger.shared.log("Starting stream with \(streamUrls?.count ?? 0) sources and \(subtitleUrls?.count ?? 0) subtitles", type: "Stream")
|
|
DispatchQueue.main.async {
|
|
completion((streamUrls, subtitleUrls,streamUrlsAndHeaders))
|
|
}
|
|
return
|
|
}
|
|
|
|
if let streamsArray = try? JSONSerialization.jsonObject(with: data, options: []) as? [String] {
|
|
Logger.shared.log("Starting multi-stream with \(streamsArray.count) sources", type: "Stream")
|
|
DispatchQueue.main.async { completion((streamsArray, nil,nil)) }
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
let streamUrl = result.toString()
|
|
Logger.shared.log("Starting stream from: \(streamUrl ?? "nil")", type: "Stream")
|
|
DispatchQueue.main.async {
|
|
completion((streamUrl != nil ? [streamUrl!] : nil, nil,nil))
|
|
}
|
|
}
|
|
|
|
let catchBlock: @convention(block) (JSValue) -> Void = { error in
|
|
Logger.shared.log("Promise rejected: \(String(describing: error.toString()))", type: "Error")
|
|
DispatchQueue.main.async {
|
|
completion((nil, nil,nil))
|
|
}
|
|
}
|
|
|
|
let thenFunction = JSValue(object: thenBlock, in: context)
|
|
let catchFunction = JSValue(object: catchBlock, in: context)
|
|
|
|
promise.invokeMethod("then", withArguments: [thenFunction as Any])
|
|
promise.invokeMethod("catch", withArguments: [catchFunction as Any])
|
|
}
|
|
|
|
func fetchStreamUrlJSSecond(episodeUrl: String, softsub: Bool = false, module: ScrapingModule, completion: @escaping ((streams: [String]?, subtitles: [String]?,sources: [[String:Any]]? )) -> Void) {
|
|
let url = URL(string: episodeUrl)!
|
|
let task = URLSession.custom.dataTask(with: url) { [weak self] data, response, error in
|
|
guard let self = self else { return }
|
|
|
|
if let error = error {
|
|
Logger.shared.log("URLSession error: \(error.localizedDescription)", type: "Error")
|
|
DispatchQueue.main.async { completion((nil, nil,nil)) }
|
|
return
|
|
}
|
|
|
|
guard let data = data, let htmlString = String(data: data, encoding: .utf8) else {
|
|
Logger.shared.log("Failed to fetch HTML data", type: "Error")
|
|
DispatchQueue.main.async { completion((nil, nil, nil)) }
|
|
return
|
|
}
|
|
|
|
DispatchQueue.main.async {
|
|
if let exception = self.context.exception {
|
|
Logger.shared.log("JavaScript exception: \(exception)", type: "Error")
|
|
completion((nil, nil, nil))
|
|
return
|
|
}
|
|
|
|
guard let extractStreamUrlFunction = self.context.objectForKeyedSubscript("extractStreamUrl") else {
|
|
Logger.shared.log("No JavaScript function extractStreamUrl found", type: "Error")
|
|
completion((nil, nil, nil))
|
|
return
|
|
}
|
|
|
|
let promiseValue = extractStreamUrlFunction.call(withArguments: [htmlString])
|
|
guard let promise = promiseValue else {
|
|
Logger.shared.log("extractStreamUrl did not return a Promise", type: "Error")
|
|
completion((nil, nil, nil))
|
|
return
|
|
}
|
|
|
|
let thenBlock: @convention(block) (JSValue) -> Void = { [weak self] result in
|
|
guard self != nil else { return }
|
|
|
|
if let jsonString = result.toString(),
|
|
let data = jsonString.data(using: .utf8) {
|
|
do {
|
|
if let json = try? JSONSerialization.jsonObject(with: data, options: []) as? [String: Any] {
|
|
print("JSON object is \(json) 3 ")
|
|
var streamUrls: [String]? = nil
|
|
var subtitleUrls: [String]? = nil
|
|
var streamUrlsAndHeaders : [[String:Any]]? = nil
|
|
if let streamSources = json["streams"] as? [[String:Any]]
|
|
{
|
|
streamUrlsAndHeaders = streamSources
|
|
Logger.shared.log("Found \(streamSources.count) streams and headers", type: "Stream")
|
|
}
|
|
else if let streamSource = json["stream"] as? [String:Any]
|
|
{
|
|
streamUrlsAndHeaders = [streamSource]
|
|
Logger.shared.log("Found single stream with headers", type: "Stream")
|
|
}
|
|
else if let streamsArray = json["streams"] as? [String] {
|
|
streamUrls = streamsArray
|
|
Logger.shared.log("Found \(streamsArray.count) streams", type: "Stream")
|
|
} else if let streamUrl = json["stream"] as? String {
|
|
streamUrls = [streamUrl]
|
|
Logger.shared.log("Found single stream", type: "Stream")
|
|
}
|
|
|
|
if let subsArray = json["subtitles"] as? [String] {
|
|
subtitleUrls = subsArray
|
|
Logger.shared.log("Found \(subsArray.count) subtitle tracks", type: "Stream")
|
|
} else if let subtitleUrl = json["subtitles"] as? String {
|
|
subtitleUrls = [subtitleUrl]
|
|
Logger.shared.log("Found single subtitle track", type: "Stream")
|
|
}
|
|
|
|
Logger.shared.log("Starting stream with \(streamUrls?.count ?? 0) sources and \(subtitleUrls?.count ?? 0) subtitles", type: "Stream")
|
|
DispatchQueue.main.async {
|
|
completion((streamUrls, subtitleUrls, streamUrlsAndHeaders))
|
|
}
|
|
return
|
|
}
|
|
|
|
if let streamsArray = try? JSONSerialization.jsonObject(with: data, options: []) as? [String] {
|
|
Logger.shared.log("Starting multi-stream with \(streamsArray.count) sources", type: "Stream")
|
|
DispatchQueue.main.async { completion((streamsArray, nil, nil)) }
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
let streamUrl = result.toString()
|
|
Logger.shared.log("Starting stream from: \(streamUrl ?? "nil")", type: "Stream")
|
|
DispatchQueue.main.async {
|
|
completion((streamUrl != nil ? [streamUrl!] : nil, nil, nil))
|
|
}
|
|
}
|
|
|
|
let catchBlock: @convention(block) (JSValue) -> Void = { error in
|
|
Logger.shared.log("Promise rejected: \(String(describing: error.toString()))", type: "Error")
|
|
DispatchQueue.main.async {
|
|
completion((nil, nil, nil))
|
|
}
|
|
}
|
|
|
|
let thenFunction = JSValue(object: thenBlock, in: self.context)
|
|
let catchFunction = JSValue(object: catchBlock, in: self.context)
|
|
|
|
promise.invokeMethod("then", withArguments: [thenFunction as Any])
|
|
promise.invokeMethod("catch", withArguments: [catchFunction as Any])
|
|
}
|
|
}
|
|
task.resume()
|
|
}
|
|
}
|