import UIKit
import WebKit
// 自动布局组件
import SnapKit
import PYH5Bridge

class PYH5BridgeTest: UIViewController, WKUIDelegate, WKNavigationDelegate, UIWebViewDelegate {
    public var useWKWebView = true
    public var serverURL: String?
    var webView: UIView?
    var webViewHelper: PYCWebViewHelper?
    
    override func viewDidLoad() {
        super.viewDidLoad()
        self.customBackItem()
        
        self.title = self.useWKWebView ? "WKWebView" : "UIWebView"
        self.initWebView()
    }
    
    override func didReceiveMemoryWarning() {
        super.didReceiveMemoryWarning()
        
    }
    
    func initWebView() -> Void {
        if self.useWKWebView {
            let webConfig = WKWebViewConfiguration()
            let wkWebView = WKWebView(frame: .zero, configuration: webConfig)
            
            //WKWebView必须实现navigationDelegate和UIDelegate
            wkWebView.uiDelegate = self
            wkWebView.navigationDelegate = self
            
            self.webView = wkWebView
        } else {
            let uiWebView = UIWebView(frame: .zero)
            //UIWebView必须实现UIWebViewDelegate
            uiWebView.delegate = self
            
            self.webView = uiWebView;
        }
        self.view.addSubview(webView!)
        
        webView!.snp.makeConstraints { (make) -> Void in
            make.width.equalToSuperview()
            if #available(iOS 11.0, *) {
                make.top.equalTo(self.view.safeAreaLayoutGuide.snp.top)
            }
            else {
                make.top.equalToSuperview()
            }
            make.bottom.equalTo(self.view.safeAreaLayoutGuide.snp.bottom)
        }
        
        if let urlString = self.serverURL  {
            let urlRequest = URLRequest(url: URL(string: urlString)!)
            
            if self.useWKWebView {
                (self.webView as? WKWebView)?.load(urlRequest)
            }
            else {
                (self.webView as? UIWebView)?.loadRequest(urlRequest)
            }
            
            // 无广告的初始化方法，与有广告的初始化方法二者选其一
            self.webViewHelper = PYCWebViewHelper(url: nil, webViewHelperBlock: nil)
            // 有广告的初始化方法，URL为图片链接（建议使用https协议链接应对苹果后续的ATS政策），Block为用户点击广告时的回调方法
            self.webViewHelper = PYCWebViewHelper(url: "https://www.xxx.com/xxx.png", webViewHelperBlock: { (urlString: String?) -> () in
                //可以自由跳转WebView或App内部模块
                })
            self.webViewHelper?.addScriptMessageHandler(toWebView: self.webView, webViewDelegate: self)
        }
    }
    
    // MARK: WKUIDelegate
    public func webView(_ webView: WKWebView, createWebViewWith configuration: WKWebViewConfiguration, for navigationAction: WKNavigationAction, windowFeatures: WKWindowFeatures) -> WKWebView?
    {
        return self.webViewHelper?.pyWebView(webView, createWebViewWith: configuration, for: navigationAction, windowFeatures: windowFeatures)
    }
    
    //MARK: WKNavigationDelegate
    public func webView(_ webView: WKWebView, decidePolicyFor navigationAction: WKNavigationAction, decisionHandler: @escaping (WKNavigationActionPolicy) -> Swift.Void)
    {
        self.webViewHelper?.pycWebView(webView, decidePolicyFor: navigationAction, decisionHandler: decisionHandler)
    }
    
    public func webViewWebContentProcessDidTerminate(_ webView: WKWebView)
    {
        self.webViewHelper?.pyWebViewWebContentProcessDidTerminate(webView)
    }
    
    //MARK: UIWebViewDelegate
    public func webView(_ webView: UIWebView, shouldStartLoadWith request: URLRequest, navigationType: UIWebViewNavigationType) -> Bool
    {
        return (self.webViewHelper?.pycWebView(webView, shouldStartLoadWith: request, navigationType: navigationType))!
    }
    
    public func webViewDidFinishLoad(_ webView: UIWebView)
    {
        self.webViewHelper?.pyWebViewDidFinishLoad(webView)
    }
}

func customBackItem() -> Void {
    let backItem = UIBarButtonItem(title: "返回", style: .plain, target: self, action: #selector(backBtnClicked))
    self.navigationItem.leftBarButtonItem = backItem
}

@objc func backBtnClicked(sender: AnyObject?) -> Void {
    var canGoBack = false
    
    if self.useWKWebView {
        if let wkWebView = self.webView as? WKWebView {
            canGoBack = wkWebView.canGoBack
            if canGoBack {
                wkWebView.goBack()
            }
        }
        
    } else {
        if let uiwebView = self.webView as? UIWebView {
            canGoBack = uiwebView.canGoBack
            if canGoBack {
                uiwebView.goBack()
            }
        }
    }
    
    if canGoBack {
        self.closeAndBackItem()
    } else {
        self.navigationController?.popViewController(animated: true)
    }
}    

// 无广告的初始化方法，与有广告的初始化方法二者选其一
self.webViewHelper = PYCWebViewHelper(url: nil, webViewHelperBlock: nil)
// 有广告的初始化方法，URL为图片链接（建议使用https协议链接应对苹果后续的ATS政策），Block为用户点击广告时的回调方法
self.webViewHelper = PYCWebViewHelper(url: "https://www.xxx.com/xxx.png", webViewHelperBlock: { (urlString: String?) -> () in
    //可以自由跳转WebView或App内部模块
    })

// UIViewController中添加
override var shouldAutorotate: Bool {
    return true
}
    
override var supportedInterfaceOrientations: UIInterfaceOrientationMask {
    return [.portrait, .portraitUpsideDown]
}

//若使用了UINavigationController，则需扩展UINavigationController
extension UINavigationController {
    override open var shouldAutorotate: Bool {
        return visibleViewController?.shouldAutorotate ?? true  // 默认为true
    }
    
    override open var supportedInterfaceOrientations: UIInterfaceOrientationMask {
        return visibleViewController?.supportedInterfaceOrientations ?? [.all]  // 默认支持所有方向
    }
}

func closeAndBackItem() -> Void {
    let backItem = UIBarButtonItem(title: "返回", style: .plain, target: self, action: #selector(backBtnClicked))
    let closeItem = UIBarButtonItem(title: "关闭", style: .plain, target: self, action: #selector(closeBtnClicked))
    self.navigationItem.leftBarButtonItems = [backItem, closeItem]
}
    
@objc func backBtnClicked(sender: AnyObject?) -> Void {
    var canGoBack = false
    
    if self.useWKWebView {
        if let wkWebView = self.webView as? WKWebView {
            canGoBack = wkWebView.canGoBack
            if canGoBack {
                wkWebView.goBack()
            }
        }
        
    } else {
        if let uiwebView = self.webView as? UIWebView {
            canGoBack = uiwebView.canGoBack
            if canGoBack {
                uiwebView.goBack()
            }
        }
    }
    
    if canGoBack {
        self.closeAndBackItem()
    } else {
        self.navigationController?.popViewController(animated: true)
    }
}

@objc func closeBtnClicked(sender: Any?) -> Void {
    self.navigationController?.popViewController(animated: true)
}