import SafariServices

class SafariWebExtensionHandler: NSObject, NSExtensionRequestHandling {
    func beginRequest(with context: NSExtensionContext) {
        let item = context.inputItems[0] as? NSExtensionItem
        let message = item?.userInfo?[SFExtensionMessageKey] as? [String: Any]
        // if message received without name, ignore
        guard let name = message?["name"] as? String else {
            err("could not get message name from web extension")
            return
        }
        logText("Got message with name: \(name)")
        // got a valid message, construct response based on message received
        let response = NSExtensionItem()
        // send standard error when there's an issue parsing inbound message
        var inBoundError = false
        // these if/else if statement are formatted so that they can be neatly collapsed in Xcode
        // typically the "else if" would be on the same line as the preceding statements close bracket
        // ie. } else if {
        if name == "OPEN_APP" {
            if let scheme = Bundle.main.infoDictionary?["US_URL_SCHEME"],
               let url = URL(string: "\(scheme):") {
                #if os(macOS)
                    NSWorkspace.shared.open(url)
                #endif
            }
        }
        else if name == "REQ_PLATFORM" {
            let platform = getPlatform()
            response.userInfo = [SFExtensionMessageKey: ["platform": platform]]
        }
        else if name == "REQ_USERSCRIPTS" {
            if let url = message?["url"] as? String, let isTop = message?["isTop"] as? Bool {
                if
                    let matches = getInjectionFilenames(url),
                    let code = getCode(matches, isTop)
                {
                    response.userInfo = [SFExtensionMessageKey: code]
                } else {
                    response.userInfo = [SFExtensionMessageKey: ["error": "REQ_USERSCRIPTS failed"]]
                }
            } else {
                inBoundError = true
            }
        }
        else if name == "REQ_REQUESTS" {
            if let requestScripts = getRequestScripts() {
                response.userInfo = [SFExtensionMessageKey: requestScripts]
            } else {
                response.userInfo = [SFExtensionMessageKey: ["error": "failed to get requestScripts"]]
            }
        }
        else if name == "REQ_CONTEXT_MENU_SCRIPTS" {
            if let contextMenuScripts = getContextMenuScripts() {
                response.userInfo = [SFExtensionMessageKey: contextMenuScripts]
            } else {
                response.userInfo = [SFExtensionMessageKey: ["error": "failed to get contextMenuScripts"]]
            }
        }
        else if name == "POPUP_BADGE_COUNT" {
            if let url = message?["url"] as? String, let frameUrls = message?["frameUrls"] as? [String] {
                if let matches = getPopupBadgeCount(url, frameUrls) {
                    response.userInfo = [SFExtensionMessageKey: ["count": matches]]
                } else {
                    response.userInfo = [SFExtensionMessageKey: ["error": "failed to update badge count"]]
                }
            } else {
                inBoundError = true
            }
        }
        else if name == "POPUP_INIT" {
            if let initData = popupInit() {
                response.userInfo = [SFExtensionMessageKey: ["initData": initData]]
            } else {
                response.userInfo = [SFExtensionMessageKey: ["error": "failed to get init data"]]
            }
        }
        else if name == "POPUP_MATCHES"{
            if let url = message?["url"] as? String, let frameUrls = message?["frameUrls"] as? [String] {
                if let matches = getPopupMatches(url, frameUrls) {
                    response.userInfo = [SFExtensionMessageKey: ["matches": matches]]
                } else {
                    response.userInfo = [SFExtensionMessageKey: ["error": "failed to get matches"]]
                }
            } else {
                inBoundError = true
            }
        }
        else if name == "POPUP_UPDATES" {
            if let updates = checkForRemoteUpdates() {
                response.userInfo = [SFExtensionMessageKey: ["updates": updates]]
            } else {
                response.userInfo = [SFExtensionMessageKey: ["error": "failed to get updates"]]
            }
        }
        else if name == "POPUP_UPDATE_ALL" {
            if popupUpdateAll(), let updates = checkForRemoteUpdates() {
                response.userInfo = [SFExtensionMessageKey: ["updates": updates]]
            } else {
               response.userInfo = [SFExtensionMessageKey: ["error": "failed to run update sequence"]]
            }
        }
        else if name == "POPUP_UPDATE_SINGLE" {
            if
                let filename = message?["filename"] as? String,
                let url = message?["url"] as? String,
                let frameUrls = message?["frameUrls"] as? [String]
            {
                if let matches = popupUpdateSingle(filename, url, frameUrls) {
                    response.userInfo = [SFExtensionMessageKey: ["items": matches]]
                } else {
                    response.userInfo = [SFExtensionMessageKey: ["error": "failed to update file"]]
                }
            } else {
                inBoundError = true
            }
        }
        else if name == "POPUP_CHECK_UPDATES" {
            if let updates = checkForRemoteUpdates() {
                response.userInfo = [SFExtensionMessageKey: ["updates": updates]]
            } else {
                response.userInfo = [SFExtensionMessageKey: ["error": "failed to check for updates"]]
            }
        }
        else if name == "TOGGLE_EXTENSION" {
            if let active = message?["active"] as? String, ["true", "false"].contains(active) {
                var manifest = getManifest()
                manifest.settings["active"] = active
                if updateManifest(with: manifest) {
                    response.userInfo = [SFExtensionMessageKey: ["success": true]]
                } else {
                    response.userInfo = [SFExtensionMessageKey: ["error": "failed to update injection state"]]
                }
            } else {
                response.userInfo = [SFExtensionMessageKey: ["error": "missing or wrong message content"]]
            }
        }
        else if name == "TOGGLE_ITEM" {
            // the current status of the item to toggle comes in as 0 || 1
            if
                let item = message?["item"] as? [String: Any],
                let filename = item["filename"] as? String,
                let current = item["disabled"] as? Int
            {
                let action = current == 0 ? "disable" : "enable"
                if toggleFile(filename, action) {
                    response.userInfo = [SFExtensionMessageKey: ["success": true]]
                } else {
                    response.userInfo = [SFExtensionMessageKey: ["error": "failed to toggle file"]]
                }
            } else {
                inBoundError = true
            }
        }
        else if name == "OPEN_SAVE_LOCATION" {
            #if os(macOS)
                if openSaveLocation() {
                    response.userInfo = [SFExtensionMessageKey: ["success": true]]
                }
            #elseif os(iOS)
                if let files = getAllFiles() {
                    response.userInfo = [SFExtensionMessageKey: ["items": files]]
                } else {
                    response.userInfo = [SFExtensionMessageKey: ["error": "failed to get all files"]]
                }
            #endif
        }
        else if name == "CHANGE_SAVE_LOCATION" {
            #if os(macOS)
                if let scheme = Bundle.main.infoDictionary?["US_URL_SCHEME"],
                   let url = URL(string: "\(scheme)://changesavelocation") {
                    NSWorkspace.shared.open(url)
                }
            #endif
        }
        else if name == "POPUP_INSTALL_CHECK" {
            if let content = message?["content"] as? String {
                response.userInfo = [SFExtensionMessageKey: installCheck(content)]
            } else {
                response.userInfo = [SFExtensionMessageKey: ["error": "failed to get script content"]]
            }
        }
        else if name == "POPUP_INSTALL_SCRIPT" {
            if let content = message?["content"] as? String {
                response.userInfo = [SFExtensionMessageKey: installUserscript(content)]
            } else {
                response.userInfo = [SFExtensionMessageKey: ["error": "failed to get script content (2)"]]
            }
        }
        else if name == "PAGE_INIT_DATA" {
            #if os(macOS)
                if let initData = getInitData(), checkDefaultDirectories() {
                    response.userInfo = [SFExtensionMessageKey: initData]
                } else {
                    response.userInfo = [SFExtensionMessageKey: ["error": "failed to get init data"]]
                }
            #endif
        }
        else if name == "PAGE_LEGACY_IMPORT" {
            #if os(macOS)
                if let settings = getLegacyData() {
                    response.userInfo = [SFExtensionMessageKey: settings]
                } else {
                    response.userInfo = [SFExtensionMessageKey: ["error": "failed to get legacy data"]]
                }
            #endif
        }
        else if name == "PAGE_ALL_FILES" {
            #if os(macOS)
                if let files = getAllFiles() {
                    response.userInfo = [SFExtensionMessageKey: files]
                } else {
                    response.userInfo = [SFExtensionMessageKey: ["error": "failed to get all files"]]
                }
            #endif
        }
        else if name == "PAGE_SAVE" {
            #if os(macOS)
                if
                    let item = message?["item"] as? [String: Any],
                    let content = message?["content"] as? String
                {
                    let saveResponse = saveFile(item, content)
                    response.userInfo = [SFExtensionMessageKey: saveResponse]
                } else {
                    inBoundError = true
                }
            #endif
        }
        else if name == "PAGE_TRASH" {
            #if os(macOS)
                if let item = message?["item"] as? [String: Any] {
                    if trashFile(item) {
                        response.userInfo = [SFExtensionMessageKey: ["success": true]]
                    } else {
                        response.userInfo = [SFExtensionMessageKey: ["error": "failed to trash file"]]
                    }
                } else {
                    inBoundError = true
                }
            #endif
        }
        else if name == "PAGE_UPDATE" {
            #if os(macOS)
                if let content = message?["content"] as? String {
                    let updateResponse = getFileRemoteUpdate(content)
                    response.userInfo = [SFExtensionMessageKey: updateResponse]
                } else {
                    inBoundError = true
                }
            #endif
        }
        else if name == "CANCEL_REQUESTS" {
            URLSession.shared.getAllTasks { tasks in
                for task in tasks {
                    task.cancel()
                }
            }
            response.userInfo = [SFExtensionMessageKey: ["success": true]]
        }
        else if name == "PAGE_NEW_REMOTE" {
            #if os(macOS)
                if let url = message?["url"] as? String {
                    if !validateUrl(url) {
                        response.userInfo = [SFExtensionMessageKey: ["error": "Failed to get remote content, invalid url"]]
                    } else if let content = getRemoteFileContents(url) {
                        response.userInfo = [SFExtensionMessageKey: content]
                    } else {
                        response.userInfo = [SFExtensionMessageKey: ["error": "Failed to get remote content"]]
                    }
                } else {
                    inBoundError = true
                }
            #endif
        }
        else if name == "PAGE_UPDATE_SETTINGS" {
            #if os(macOS)
                if let settings = message?["settings"] as? [String: String] {
                    if updateSettings(settings) {
                        response.userInfo = [SFExtensionMessageKey: ["success": true]]
                    } else {
                        response.userInfo = [SFExtensionMessageKey: ["error": "Failed to save settings to disk"]]
                    }
                } else {
                    inBoundError = true
                }
            #endif
        }
        else if name == "PAGE_UPDATE_BLACKLIST" {
            #if os(macOS)
                if let blacklist = message?["blacklist"] as? [String] {
                    var manifest = getManifest()
                    manifest.blacklist = blacklist
                    if !updateManifest(with: manifest) {
                        response.userInfo = [SFExtensionMessageKey: ["error": "Failed to save blacklist to disk"]]
                    } else {
                        response.userInfo = [SFExtensionMessageKey: ["success": true]]
                    }
                } else {
                    inBoundError = true
                }
            #endif
        }
        // send inBoundError if found
        if inBoundError {
            response.userInfo = [SFExtensionMessageKey: ["error": "Failed to parse inbound message"]]
        }
        context.completeRequest(returningItems: [response], completionHandler: nil)
    }
}