yattee/Shared/UserAgentManager.swift

import Logging
#if !os(tvOS)
    import WebKit
#endif

final class UserAgentManager {
    static let shared = UserAgentManager()

    private(set) var userAgent: String
    #if !os(tvOS)
        private var webView: WKWebView?
    #endif

    private init() {
        /*
         In case an error occurs while retrieving the actual User-Agent, and on tvOS,
         we set a default User-Agent value that represents a commonly used User-Agent.
         */

        userAgent = "Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Trident/5.0)"
        #if !os(tvOS)
            webView = WKWebView()
            webView?.evaluateJavaScript("navigator.userAgent") { [weak self] result, _ in
                if let userAgent = result as? String {
                    DispatchQueue.main.async {
                        self?.userAgent = userAgent
                        Logger(label: "stream.yattee.userAgentManager").info("User-Agent: \(userAgent)")
                    }
                } else {
                    Logger(label: "stream.yattee.userAgentManager").warning("Failed to update User-Agent.")
                }
            }
        #else
            Logger(label: "stream.yattee.userAgentManager.tvOS").info("User-Agent: \(userAgent)")
        #endif
    }
}