PDF_Office/PDF Office/PDF Master/Class/ChromiumTabs/KMBrowserWindowController.swift

//
//  KMBrowserWindowController.swift
//  PDF Master
//
//  Created by wanjun on 2022/12/9.
//

import Cocoa

@objcMembers class KMBrowserWindowController: CTBrowserWindowController {
    var rightSegmentControl: KMSegmentedBox?
    var modeType: KMHomeToolState = .Home
    var filterType: Int = 0
    var mainViewController: NSViewController?
    var isShowingTokenTimeOutAlert: Bool = false
    var currentTimer: Timer?
    private(set) var isMultiTabMode: Bool = false

    override func windowDidLoad() {
        super.windowDidLoad()

//        window?.setFrameAutosaveName("")
        rightTabStripView_.delete = self
        homeRightTabStripView_.delete = self
        rightTabStripView_.updateView()
        fileUploadPanel.delete = self
        homeRightTabStripView_.homeRightSearchField.delegate = self
        isMultiTabMode = true
        addObserverForAppearanceChange()
        
        refreshToolBar(homeToolState: .Home)
        
        NotificationCenter.default.addObserver(self, selector: #selector(closeAllTabs(_:)), name: NSNotification.Name.init(rawValue: "KMTabControllerCloseAllTabs"), object: nil)
        NotificationCenter.default.addObserver(self, selector: #selector(openNewWindow(_:)), name: NSNotification.Name.init(rawValue: "KMTabControllerOpenNewWindow"), object: nil)
        
        if (WelcomeWindowController.welcomeHasShow() == false) {
            DispatchQueue.main.async {
                let welcome = WelcomeWindowController()
                self.window?.beginSheet(welcome.window!)
                
                welcome.itemClick = { [weak self] idx, param in
                    if (idx == 1) { // 关闭
                        self?.window?.endSheet((param as! NSWindowController).window!)
                    } else if (idx == 2) { // 以后提醒
                        self?.window?.endSheet((param as! NSWindowController).window!)
                    } else if (idx == 3) { // 注册
                        self?.window?.endSheet((param as! NSWindowController).window!)
                        
                        let _ = KMLoginWindowController.show(window: (self?.window)!, .Batch, .register)
                    }
                }
            }
        }
    }
    
    override func windowShouldClose(_ sender: NSWindow) -> Bool {
        if self.browser.tabStripModel.count() > 1 {
            self.browser.windowDidBeginToClose()
            return false
        }
        return true
    }
    
    // MARK: Dark&Light
    
    func addObserverForAppearanceChange() -> Void {
        window?.contentView!.addObserver(self, forKeyPath: "effectiveAppearance", options: .new, context: nil)
    }
    
    func removeObserverForAppearanceChange() -> Void {
        window?.contentView!.removeObserver(self, forKeyPath: "effectiveAppearance")
    }
    
    override func observeValue(forKeyPath keyPath: String?, of object: Any?, change: [NSKeyValueChangeKey : Any]?, context: UnsafeMutableRawPointer?) {
        if keyPath == "effectiveAppearance" {
            updateViewColor()
        }
    }
    
    override func updateViewColor() {
        super.updateViewColor()
        
        window?.backgroundColor = KMTabAppearance.tabsViewBackgroundColor()
    }
    
    // MARK: Public Methods
    
    func refreshToolBar(homeToolState toolState: KMHomeToolState) -> Void {
        modeType = toolState
        
        let document = browser.activeTabContents()
        if document != nil {
            if document!.isHome {
                homeRightTabStripView_.isHidden = modeType == .Home
            } else {
                self.homeRightStripView.isHidden = true
            }
        } else {
            self.homeRightStripView.isHidden = true
        }
        
        homeRightTabStripView_.homeRefreshButton.isEnabled = true
        homeRightTabStripView_.homeUploadButton.isEnabled = true
        
        filterType = 0
        rightTabStripView_.sortPopUpButton.removeAllItems()
        rightTabStripView_.filterPopUpButton.removeAllItems()
        rightTabStripView_.updateView()
    }
    
    override var hasToolbar: Bool {
        get {
            return false
        }
    }
    
    // MARK: Private Methods
    
    func openDocumentWindow() -> Void {
//        let panel = NSOpenPanel()
//        panel.allowsMultipleSelection = false
//        panel.allowedFileTypes = ["pdf"]
//        panel.beginSheetModal(for: NSApp.mainWindow!) { response in
//            if response == .OK {
//                let openPath = panel.url?.path
//                let selectDocument = self.selectTabContents(path: openPath!)
//                if selectDocument != nil {
//                    let currentIndex = selectDocument!.browser.tabStripModel.index(of: selectDocument)
//                    self.browser.tabStripModel.selectTabContents(at: Int32(currentIndex), userGesture: true)
//                    return
//                }
//                if !panel.url!.path.isPDFValid() {
//                    let alert = NSAlert()
//                    alert.alertStyle = .critical
//                    alert.messageText = NSLocalizedString("An error occurred while opening this document. The file is damaged and could not be repaired.", comment: "")
//                    alert.runModal()
//                    return
//                }
//                NSDocumentController.shared.openDocument(withContentsOf: panel.url!, display: true) { pdfDocument, documentWasAlreadyOpen, error in
//                    print("openDocumentWindow")
//                }
//            }
//        }
        
        // MARK TODO: + 号开启
        self.browser.addHomeTabContents()
    }
    
    override func openDocument(_ sender: Any!) {
        self.openDocumentWindow()
    }
    
    func selectTabContents(path filePath: String) -> KMMainDocument? {
        let document = NSDocumentController.shared.document(for: URL(fileURLWithPath: filePath))
        var selectDocument: KMMainDocument?
        if document != nil {
            if document!.isKind(of: KMMainDocument.self) {
                selectDocument = document as! KMMainDocument
            }
        }
        return selectDocument
    }
    
    
    // MARK: Getter
    
    override var rightStripView: NSView! {
        get {
            return rightTabStripView_
        }
    }
    
    override var homeRightStripView: NSView! {
        get {
            return homeRightTabStripView_
        }
    }
    
    // MARK: Button Action
    
    func commandDispatch(_ sender: Any) -> Void {
        openDocumentWindow()
    }

    func closeAllTabs(_ sender: Any) -> Void {
        if self.browser != nil {
            self.browser.closeAllTabs()
        }
    }
    
    func openNewWindow(_ sender: Any) -> Void {
        if self.browser != nil {
            let activeInt = self.browser.activeTabIndex()
            let activeTab = self.browser.activeTabContents()
            
            self.browser.closeTab(at: activeInt, makeHistory: false)
            
            let browser: KMBrowser = KMBrowser.init()
            browser.windowController = KMBrowserWindowController.init(browser: browser)
            browser.addHomeTabContents()
            browser.windowController.showWindow(self)
            
//            browser.add(activeTab, inForeground: false)
//            browser.selectTab(at: 1)
            
            let pdfDoc = CPDFDocument.init(url: (activeTab?.fileURL)!)
            let document = NSDocumentController.shared.document(for: (activeTab?.fileURL)!)
            KMMainDocument().tryToUnlockDocument(pdfDoc!)
            
            if ((pdfDoc?.isLocked)! == true) {
                KMPasswordInputWindow.openWindow(window: self.window!, url: (activeTab?.fileURL)!) { result, password in
                    if result == .cancel { /// 关闭
                        return
                    }
                    /// 解密成功
                    var selectDocument: KMMainDocument? = nil
                    if ((document?.isKind(of: KMMainDocument.self)) != nil) {
                        selectDocument = (document as! KMMainDocument)
                    }
                    if selectDocument != nil {
                        let currentIndex = selectDocument?.browser.tabStripModel.index(of: selectDocument)
                        selectDocument?.browser.tabStripModel.selectTabContents(at: Int32(currentIndex!), userGesture: true)
                        if (selectDocument?.browser.window.isVisible)! as Bool {
                            selectDocument?.browser.window.orderFront(nil)
                        } else if (selectDocument?.browser.window.isMiniaturized)! as Bool {
                            selectDocument?.browser.window.orderFront(nil)
                        }
                    } else {
                        if !(activeTab?.fileURL)!.path.isPDFValid() {
                            let alert = NSAlert()
                            alert.alertStyle = .critical
                            alert.messageText = NSLocalizedString("An error occurred while opening this document. The file is damaged and could not be repaired.", comment: "")
                            alert.runModal()
                            return
                        }

                        NSDocumentController.shared.openDocument(withContentsOf: (activeTab?.fileURL)!, display: true) { document, documentWasAlreadyOpen, error in
                            if error != nil {
                                NSApp.presentError(error!)
                                return
                            }

                            (document as! KMMainDocument).mainViewController.password = password
                        }
                    }
                }
            } else {
                var selectDocument: KMMainDocument? = nil
                if ((document?.isKind(of: KMMainDocument.self)) != nil) {
                    selectDocument = (document as! KMMainDocument)
                }
                if selectDocument != nil {
                    let currentIndex = selectDocument?.browser.tabStripModel.index(of: selectDocument)
                    selectDocument?.browser.tabStripModel.selectTabContents(at: Int32(currentIndex!), userGesture: true)
                    if (selectDocument?.browser.window.isVisible)! as Bool {
                        selectDocument?.browser.window.orderFront(nil)
                    } else if (selectDocument?.browser.window.isMiniaturized)! as Bool {
                        selectDocument?.browser.window.orderFront(nil)
                    }
                } else {
                    if !(activeTab?.fileURL)!.path.isPDFValid() {
                        let alert = NSAlert()
                        alert.alertStyle = .critical
                        alert.messageText = NSLocalizedString("An error occurred while opening this document. The file is damaged and could not be repaired.", comment: "")
                        alert.runModal()
                        return
                    }
                    NSDocumentController.shared.openDocument(withContentsOf: (activeTab?.fileURL)!, display: true) { document, documentWasAlreadyOpen, error in
                        if error != nil {
                            NSApp.presentError(error!)
                            return
                        }
                    }
                }
            }
        }
    }
    
    // MARK:  待补充[万军]
    func mergeAllWindow(_ sender: Any) {
        
    }
    
    override func layoutTabContentArea(_ frame: NSRect) {
        super.layoutTabContentArea(frame)
        
        self.rightStripView.autoresizingMask = [.minXMargin, .minYMargin]
    }
}

// MARK: - KMToolbarRightViewDelegate

extension KMBrowserWindowController: KMToolbarRightViewDelegate {
    func pdfRightSegmentedControlAction(_ sender: KMSegmentedBox?) {
        KMComparativeTableViewController.show(window: self.window!)
    }
    
    func userInfoButtonAction(_ sender: NSButton) {
        if KMLightMemberManager.manager.isLogin() {
            Task { @MainActor in
                if await KMLightMemberManager.manager.canUseAdvanced() {
                    KMAccountInfoWindowController.show(window: self.window!)
                } else {
                    KMLoginWindowController.show(window: self.window!)
                }
            }
        } else {
            KMLoginWindowController.show(window: self.window!)
        }
    }
    
    func homeRefreshButtonAction(_ sender: NSButton) {
        
    }
    
    func homeUploadButtonAction(_ sender: NSButton) {
        
    }
    
    func homeMenuSortAction(_ sender: NSPopUpButton) {
        
    }
    
    func homeMenuFilterAction(_ sender: NSPopUpButton) {
        
    }
}

// MARK: - KMUploadFileDelegate

extension KMBrowserWindowController: KMUploadFileDelegate {
//    override func cancelOperation(_ sender: Any?) {
//        
//    }
}

// MARK: - NSSearchFieldDelegate

extension KMBrowserWindowController: NSSearchFieldDelegate {
    func controlTextDidChange(_ obj: Notification) {
        
    }
    
    func controlTextDidEndEditing(_ obj: Notification) {
        
    }
    
    func control(_ control: NSControl, textView: NSTextView, doCommandBy commandSelector: Selector) -> Bool {
        var result = false
        return result
    }
}

// MARK: -
// MARK: Menu Actions

// MARK: -
// MARK: File Menu

extension KMBrowserWindowController {
    @IBAction func menuItemAction_openFile(_ sender: Any) {
        super.openDocument(sender)
    }
    
    @IBAction func importFromFile(_ sender: Any) {
        let document: KMMainDocument = self.browser.activeTabContents() as! KMMainDocument
        document.homeViewController.importFromFile(sender)
    }
    
    @IBAction func openBlankPage(_ sender: Any) {
        let document: KMMainDocument = self.browser.activeTabContents() as! KMMainDocument
        document.homeViewController.openBlankPage(sender)
    }
    
    @IBAction func importFromScanner(_ sender: Any) {
        let document: KMMainDocument = self.browser.activeTabContents() as! KMMainDocument
        document.homeViewController.importFromScanner(sender)
    }
    
    @IBAction func menuItemAction_newTab(_ sender: Any) {
        self.openDocumentWindow()
    }
}