O2OA/o2ios/O2Platform/common/BaseWebViewUIViewController.swift

//
//  BaseWebViewUIViewController.swift
//  O2Platform
//
//  Created by 刘振兴 on 2016/10/11.
//  Copyright © 2016年 zoneland. All rights reserved.
//

import UIKit
import WebKit
import CocoaLumberjack


protocol BaseWebViewUIViewControllerJSDelegate {
    func closeUIViewWindow()
    func actionBarLoaded(show: Bool)
}

protocol O2WKScriptMessageHandlerImplement {
    func userController(_ userContentController: WKUserContentController, didReceive message: WKScriptMessage)
}

open class BaseWebViewUIViewController: UIViewController {
    
    var webView:WKWebView!
    var delegate: BaseWebViewUIViewControllerJSDelegate?
    var o2WKScriptHandlers: [String : O2WKScriptMessageHandlerImplement] = [:]
    
    
    //继承的子类如果要添加MessageHandler必须在theWebView方法前使用
    func addScriptMessageHandler(key: String, handler: O2WKScriptMessageHandlerImplement)  {
        o2WKScriptHandlers[key] = handler
    }
    
    
    open func theWebView(){
        let baseJsHandler = O2BaseJsMessageHandler(viewController: self)
        // bbs 回复
        addScriptMessageHandler(key: "ReplyAction", handler: baseJsHandler)
        // protal已经存在ActionBar
        addScriptMessageHandler(key: "actionBarLoaded", handler: baseJsHandler)
        // 打开工作 {"work":"", "workCompleted":"", "title":""}
        addScriptMessageHandler(key: "openO2Work", handler: baseJsHandler)
        // 4个分类 task taskCompleted read readCompleted
        addScriptMessageHandler(key: "openO2WorkSpace", handler: baseJsHandler)
        // 打开cms appId
        addScriptMessageHandler(key: "openO2CmsApplication", handler: baseJsHandler)
        // 打开cms docId docTitle
        addScriptMessageHandler(key: "openO2CmsDocument", handler: baseJsHandler)
        // 打开meeting
        addScriptMessageHandler(key: "openO2Meeting", handler: baseJsHandler)
        // 打开 calendar
        addScriptMessageHandler(key: "openO2Calendar", handler: baseJsHandler)
        // 打开扫一扫
        addScriptMessageHandler(key: "openScan", handler: baseJsHandler)
        
        addScriptMessageHandler(key: "openO2Alert", handler: baseJsHandler)
        // 打开钉钉
        addScriptMessageHandler(key: "openDingtalk", handler: baseJsHandler)
        // 关闭当前UIViewController
        addScriptMessageHandler(key:"closeNativeWindow", handler: baseJsHandler)
        // 上传图片到云盘，放入对应的业务区域
        addScriptMessageHandler(key: "uploadImage2FileStorage", handler: baseJsHandler)
        // 打印日志
        addScriptMessageHandler(key: "o2mLog", handler: baseJsHandler)
        
        //o2m.notification
        let o2Notification = O2JsApiNotification(viewController: self)
        addScriptMessageHandler(key: "o2mNotification", handler: o2Notification)
        //o2m.util
        let o2Util = O2JsApiUtil(viewController: self)
        addScriptMessageHandler(key: "o2mUtil", handler: o2Util)
        // o2m.biz
        let biz = O2JsApiBizUtil(viewController: self)
        addScriptMessageHandler(key: "o2mBiz", handler: biz)
        
        setupWebView()
    }
    
    
    public func setupWebView() {
        let userContentController = WKUserContentController()
        //cookie脚本
        if let cookies = HTTPCookieStorage.shared.cookies {
            let script = getJSCookiesString(cookies: cookies)
            let cookieScript = WKUserScript(source: script, injectionTime: WKUserScriptInjectionTime.atDocumentStart, forMainFrameOnly: false)
            userContentController.addUserScript(cookieScript)
        }
        // console.log 打印到ios log debug用
        let consoleLog = "console.log = (function(oriLogFunc){return function(str){window.webkit.messageHandlers.o2mLog.postMessage(str);oriLogFunc.call(console,str);} })(console.log);"
        let logScript = WKUserScript(source: consoleLog, injectionTime: WKUserScriptInjectionTime.atDocumentStart, forMainFrameOnly: false)
        userContentController.addUserScript(logScript)
        let webViewConfig = WKWebViewConfiguration()
        webViewConfig.userContentController = userContentController
         //加入js-app message handler
        for item in o2WKScriptHandlers {
            userContentController.add(self, name: item.key)
        }
        self.webView = WKWebView(frame: self.view.frame, configuration: webViewConfig)
    }
    
    ///Generates script to create given cookies
    public func getJSCookiesString(cookies: [HTTPCookie]) -> String {
        var result = ""
        let dateFormatter = DateFormatter()
        dateFormatter.timeZone = NSTimeZone(abbreviation: "UTC") as TimeZone?
        dateFormatter.dateFormat = "EEE, d MMM yyyy HH:mm:ss zzz"
        
        for cookie in cookies {
            result += "document.cookie='\(cookie.name)=\(cookie.value); domain=\(cookie.domain); path=\(cookie.path); "
            if let date = cookie.expiresDate {
                result += "expires=\(dateFormatter.string(from: date)); "
            }
            if (cookie.isSecure) {
                result += "secure; "
            }
            result += "'; "
        }
        return result
    }
    
    
}

extension BaseWebViewUIViewController: WKScriptMessageHandler {
    
    public func userContentController(_ userContentController: WKUserContentController, didReceive message: WKScriptMessage) {
        let name = message.name
        if let handler = o2WKScriptHandlers[name] {
            handler.userController(userContentController, didReceive: message)
        }else {
            let err = "没有找到对应的key，key:\(name)"
            self.showError(title: err)
        }
    }
}