be-ytb/js/main.swift

import JavaScriptCore
import Alamofire

func downloadJSFile(url: URL, completion: @escaping (Result<String, Error>) -> Void) {
    AF.download(url).responseData { (response) in
        switch response.result {
        case .success(let data):
            if let jsString = String(data: data, encoding: .utf8) {
                completion(.success(jsString))
            } else {
                completion(.failure(NSError(domain: "Download Error", code: -1, userInfo: nil)))
            }
        case .failure(let error):
            completion(.failure(error))
        }
    }
}

func createJSContext() -> JSContext {
    let ctx = JSContext()
    
    // 注入 console.log
    ctx?.evaluateScript("var console = { log: function(message) { _consoleLog(message) } }")
    let consoleLog: @convention(block) (String) -> Void = { message in
        print("JavaScript Console.log: " + message)
    }
    ctx?.setObject(unsafeBitCast(consoleLog, to: AnyObject.self), forKeyedSubscript: "_consoleLog" as (NSCopying & NSObjectProtocol)?)
    
    // 注入AF
    ctx?.evaluateScript("var AF = { request: function(url, method, data, headers, callback) { _request(url, method, data, headers, callback) } }")
    let af: @convention(block) (String, String, String?, [String: String]?, JSValue?) -> Void = { url, method, data, headers, callback in
        var request = URLRequest(url: URL(string: url)!)
        request.httpMethod = method
        
        if method == "POST" {
            if let data = data {
                request.setValue("application/json", forHTTPHeaderField: "Content-Type")
                request.httpBody = data.data(using: .utf8)
            }
        }
        
        if let headers = headers {
            request.headers = HTTPHeaders(headers)
        }
        
        AF.request(request).response { response in
            if let data = response.data {
                callback?.call(withArguments: [String(data: data, encoding: .utf8), nil])
            }
            if let error = response.error {
                debugPrint(response)
                callback?.call(withArguments: [nil, error.localizedDescription])
            }
        }
    }
    ctx?.setObject(unsafeBitCast(af, to: AnyObject.self), forKeyedSubscript: "_request" as (NSCopying & NSObjectProtocol)?)
    
    // 捕捉JS运行异常
    ctx?.exceptionHandler = { context, exception in
        if let exception = exception {
            print("JavaScript Exception: \(exception)")
        }
    }
    
    return ctx!
}

func testSearch(keyword: String, ctx: JSContext) -> Void {
    if let jsFunction = ctx.objectForKeyedSubscript("search") {
        let promise = jsFunction.call(withArguments: ["周杰伦", nil])
        // 检查返回值是否为 Promise
        // 获取 Promise 的 then 函数
        if let thenFunction = promise?.objectForKeyedSubscript("then") {
            // 调用 then 函数，并传入回调函数
            thenFunction.call(withArguments: [ { (resolvedValue: JSValue?) in
                if let value = resolvedValue {
                    print("搜索成功", value)
                }
            }, { (rejectedReason: JSValue?) in
                if let reason = rejectedReason {
                    print("搜索失败", reason)
                }
            }])
        }
    }
}

let ctx = createJSContext()

if let url = URL(string: "http://hubgit.cn/ben/be-ytb/raw/master/js/info.js") {
    downloadJSFile(url: url) { result in
        switch result {
        case .success(let jsString):
            print("下载远程JS成功")
            ctx.evaluateScript(jsString)
            testSearch(keyword: "周杰伦", ctx: ctx)
        case .failure(let error):
            print("Download Error: \(error)")
        }
    }
}

RunLoop.main.run()