import Foundation
import UserNotifications
import WebKit

class Notification: NSObject, WKScriptMessageHandler, UNUserNotificationCenterDelegate {
    let center = UNUserNotificationCenter.current()
    let baseDir = FileManager.default.urls(for: .applicationSupportDirectory, in: .userDomainMask).first!
    var webView: WKWebView!
    
    func userContentController(_ controller: WKUserContentController, didReceive message: WKScriptMessage) {
        center.delegate = self
        self.webView = message.webView
        let data = message.body as! [String: AnyObject]
        let key = data["_id"] as! String
        let callback = { (result: String?) in
            let output = result == nil ? "undefined" : "'\(result!)'";
            DispatchQueue.main.async {
                message.webView!.evaluateJavaScript("nativeMessage('\(key)',\(output))")
            }
        }
        switch(data["_type"] as! String) {
        case "notify":
            notify(data["title"] as! String, data["text"] as! String, data["icon"] as! String, data["data"] as! String, callback)
        case "requestPermission":
            requestPermission(callback)
        default:
            message.webView!.evaluateJavaScript("nativeError('\(key)',new Error('Unknown message type'))")
            return
        }
    }
    
    func userNotificationCenter(_ center: UNUserNotificationCenter, didReceive response: UNNotificationResponse, withCompletionHandler completionHandler: @escaping () -> Void) {
        if(response.actionIdentifier == UNNotificationDefaultActionIdentifier) {
            webView.evaluateJavaScript("document.dispatchEvent(new CustomEvent('notification-clicked',{detail:{data:'\(response.notification.request.content.userInfo["data"]!)'}}))")
        }
        completionHandler()
    }
    
    func notify(_ title: String, _ text: String, _ icon: String, _ data: String, _ cb: (String?) -> Void) {
        let content = UNMutableNotificationContent()
        content.title = title
        content.body = text
        content.userInfo["data"] = data
        center.add(UNNotificationRequest(identifier: "1", content: content, trigger: UNTimeIntervalNotificationTrigger.init(timeInterval: 1, repeats: false)))
        cb("1");
    }
    
    func requestPermission(_ cb: @escaping (String?) -> Void) {
        center.requestAuthorization(options: [.alert, .sound]) { (_, _) in
            cb(nil)
        }
    }
}