PDF_Office/PDF Office/PDF Master/Class/PDFWindowController/ViewController/KMMainViewController+UI.swift

//
//  KMMainViewController+UI.swift
//  PDF Reader Pro
//
//  Created by wanjun on 2022/12/15.
//

import Foundation
 
extension KMMainViewController {
    //通知
    func preferenceDidChangeNotification(notification:Notification) {
        let info : [AnyHashable : Any] = notification.userInfo ?? [:]
        
        if info.keys.contains(KMPreference.viewZoomScaleTypeKey) {
            self.selectZoom(KMPreferenceManager.shared.viewZoomScaleType)
        }
        
        if notification.name.rawValue == "pdfViewDocumentDidLoaded" || info.keys.contains(KMPreference.viewPageDisplayTypeKey) {

        }
        
        if info.keys.contains(KMPreference.displayBackgroundNormalColorKey) || notification.name.rawValue == "pdfViewDocumentDidLoaded" {
            self.listView.backgroundColor = KMPreferenceManager.shared.displayBackgroundNormalColor
            self.listView.layoutDocumentView()
        }
         
        if (info.keys.contains(KMPreference.generalAuthorNameKey)) { // 作者名称
            CPDFKitConfig.sharedInstance().setAnnotationAuthor((info[KMPreference.generalAuthorNameKey] as! String))
        }
        if (info.keys.contains(KMPreference.highlightLinksKey)) {
            let hlLink = info[KMPreference.highlightLinksKey] as? Bool
            CPDFKitConfig.sharedInstance().setEnableLinkFieldHighlight(hlLink == nil ? false : hlLink!)
            self.listView.setNeedsDisplayForVisiblePages()
        }
        if info.keys.contains(KMPreference.thumbPageSizeKey) {
            self.leftSideViewController.refreshUIOfThumbnailIfNeed(preference: true)
        }
        if info.keys.contains(KMPreference.thumbSnapshotSizeKey) {
            self.leftSideViewController.refreshUIOfSnapshotIfNeed(preference: true)
        }
        if info.keys.contains(KMPreference.outlineFontSizeKey) {
            self.leftSideViewController.updateTableFont()
        }
        if info.keys.contains(KMPreference.greekThresholdKey) {
            let value = KMPreference.shared.greekThreshold.cgFloat
            self.listView.setGreekingThreshold(value)
        }
        if info.keys.contains(KMPreference.antiAliasTextKey) {
            self.listView.setShouldAntiAlias(KMPreference.shared.antiAliasText)
            self.listView.applyDefaultInterpolationQuality()
        }
         
        if info.keys.contains(KMPreference.editPDFPopWindowShowKey) {
            let show = KMPreference.shared.editPDFPopWindowIsShow
             
        }
    }
      
    func convertNotesUsingPDFDocument(_ pdfDocument: CPDFDocument, callback: (()->Void)? = nil) {
        guard let doc = self.listView.document else {
            return
        }
        DispatchQueue.main.async {
            self.beginProgressSheet(withMessage: "", maxValue: 0)
            
            let count = pdfDocument.pageCount
            DispatchQueue.global().async {
                 self.model.addAnnotations.removeAll()
                self.model.removeAnnotations.removeAll()
                for i in 0..<count {
                    let page = pdfDocument.page(at: i)
                     
                    for annotation in page?.annotations ?? [] {
                        var newAnnotation: CPDFAnnotation?
                        if let inkAnnotation = annotation as? CPDFInkAnnotation, inkAnnotation.contents.hasPrefix("<?xml version=\"1.0\" encoding=\"utf-8\"?>") {
                            let table = KMTableAnnotation(KMNoteBounds: inkAnnotation.bounds, document: doc)
                            table.border = inkAnnotation.border
                            table.color = inkAnnotation.color
                            table.createForm(withList: inkAnnotation.contents, andPaths: inkAnnotation.bezierPaths())
                            table.updateAppearanceInk(withIsAdd: false)
                            table.contents = annotation.contents
                            newAnnotation = table
                        }
                        if let newAnnotation = newAnnotation {
                            self.model.addAnnotations.append(newAnnotation)
                            self.model.removeAnnotations.append(annotation)
                        }
                    }
                     
                }
                
                DispatchQueue.main.async {
                    for i in 0..<self.model.addAnnotations.count {
                        let newAnnotation = self.model.addAnnotations[i]
                        let annotation = self.model.removeAnnotations[i]
                        
                        let page = annotation.page
                        // this is only to make sure markup annotations generate the lineRects, for thread safety
                        self.listView.addAnnotation(with: newAnnotation, to: page)
                        self.listView.remove(annotation)
                        if newAnnotation.contents != nil {
                            if newAnnotation.contents.count == 0 {
                                newAnnotation.autoUpdateString()
                            }
                        }
                    }
                    
                    self.dismissProgressSheet()
                    
                    self.listView.undoManager?.removeAllActions()
                    self.undoManager?.removeAllActions()
                    
                    // 清空数据
                    self.model.addAnnotations.removeAll()
                    self.model.removeAnnotations.removeAll()
                    
                    callback?()
                }
            }
        }
    }
    
    // MARK: - KMInterfaceThemeChangedProtocol
    
    override func interfaceThemeDidChanged(_ appearance: NSAppearance.Name) {
        super.interfaceThemeDidChanged(appearance)
        
        self.leftSideViewController.interfaceThemeDidChanged(appearance)
        
        self.pdfEditController?.interfaceThemeDidChanged(appearance)
         
    }
}

// MARK: - KMLeftSideViewControllerDelegate

extension KMMainViewController: KMLeftSideViewControllerDelegate {
    func controlStateChange(_ obj: KMLeftSideViewController, show: Bool) {
        if show {
            self.openLeftPane()
        } else {
            self.closeLeftPane()
        }
    }
    
    func enterEditMode(_ obj: KMLeftSideViewController, _ pages: [Int]) {
     }
    
    func searchAction(searchString: String,isCase:Bool) {
        self.search(searchString: searchString,isCase: isCase)
    }
     
    func controller(controller: KMLeftSideViewController, itemClick item: Any?, itemKey: KMItemKey, params: Any?) {
        if (itemKey == .print) { // 打印
            if params is NSImage {
                KMPrintWindowController.cpdf_printImage(image: params as! NSImage)
                return
            }
            var pageRange: KMPrintPageRange = KMPrintPageRange(type: .allPage, selectPages: [], reversePrintOrder: false)
            if (params != nil) {
                pageRange.type = .custom
                pageRange.selectPages = params as! [Int]
            }
            
            self.showPrintWindow(pageRange: pageRange)
        } else if (itemKey == .cut || itemKey == .paste || itemKey == .delete || itemKey == .leftRotate || itemKey == .rightRotate) {
            self.recordIsPDFDocumentEdited(type: itemKey.toSubscribeWaterMarkType())
        } else if itemKey == .pageEdit {
         } else if itemKey == .demote { // 大纲降级
            guard let currentOutline: CPDFOutline = controller.tocOutlineView.km.clickedItem() else {
                return
            }
            self.listView.demote(outline: currentOutline)
        } else if itemKey == .promote { // 大纲升级
            guard let currentOutline: CPDFOutline = controller.tocOutlineView.km.clickedItem() else {
                return
            }
            self.listView.promote(outline: currentOutline)
        } else if itemKey == .tts {
            FMTrackEventManager.defaultManager.trackEvent(event: "SubTbr_PageDisplay", withProperties: ["SubTbr_Btn": "Btn_SubTbr_PageDisplay_TTS"])
            self.listView.annotationType = .unkown
            self.showTTSWindow()
        }
    }
    
    func controller(controller: KMLeftSideViewController, bookMarkDidChange bookMarks: [KMBookmarkItem]) {
        self.needSave = true
        self.recordIsPDFDocumentEdited()
    }
    
    func controller(controller: KMLeftSideViewController, rotateType: KMRotateType) {
        if rotateType == .clockwise {
            self.menuItemAction_rotateRight((Any).self)
        } else if rotateType == .anticlockwise {
            self.menuItemAction_rotateLeft((Any).self)
        }
    }
    
    func controller(controller: KMLeftSideViewController, listViewSelectionDidChange object: Any?, info: [String : Any]?) {
        if controller.thumbnailTableView.isEqual(to: object) {
            let row = controller.thumbnailTableView.selectedRow
            let curIndex = self.listView.currentPageIndex
            if (row != -1 && row != curIndex) {
                self.listView.go(toPageIndex: row, animated: true)
            }
        }
    }
}

//MARK: - ReadModel
extension KMMainViewController {
      
    @objc func addOutLineItemAction() {
        let labelString = "\(KMLocalizedString("Page")) \((self.listView.currentPageIndex ?? 0) + 1)"
        let label = self.listView.currentSelection?.string() ?? labelString
        let dest =  self.listView.currentDestination
        if let row = self.leftSideViewController.selectedRowIndexes().last {
            if let ol = self.leftSideViewController.tocOutlineView.item(atRow: row) as? CPDFOutline {
                _ = self.listView.addOutline(for: ol.parent, label: label, dest: dest, at: ol.index+1)
            } else {
                let idx = self.listView.document?.outlineRoot()?.numberOfChildren ?? 0
                _ = self.listView.addOutlineForRoot(label: label, dest:dest, at: idx)
            }
        } else {
            let idx = self.listView.document?.outlineRoot()?.numberOfChildren ?? 0
            _ = self.listView.addOutlineForRoot(label: label, dest:dest, at: idx)
        }
    }
     
    @objc func lookUpAction() {
        let label = self.listView.currentSelection?.string() ?? ""
 
    }
    
    @objc func searchBaiduAction() {
        let label = self.listView.currentSelection?.string() ?? ""
        let query = label.addingPercentEncoding(withAllowedCharacters: .urlQueryAllowed) ?? ""
        if let url = URL(string: "https://www.baidu.com/s?wd=\(query)") {
            NSWorkspace.shared.open(url)
        }
    }
    
    @objc func showInfoInFinder() {
        
    }
    
    @objc func NextPageAction() {
        
    }
    
    @objc func PreviousPageAction() {
        
    }
    
    @objc func TranslateItemAction() {
        // 获取选中的文本
        if let selection = self.listView.currentSelection?.string() {
            // 进行翻译
            let escapedText = selection.addingPercentEncoding(withAllowedCharacters: .urlQueryAllowed)!
            let urlStr = "https://translate.google.com/?sl=auto&amp;tl=zh-CN&amp;text=\(escapedText)"
            if let url = URL(string: urlStr) {
                NSWorkspace.shared.open(url)
            }
        } else {
            print("No text selected.")
        }
    }
    @objc func AutoScrollItemAction() {
        //增加判断，如果是正在滚动，就停止，否则就开始滚动
        self.listView.autoFlow()
    }
    
    
    @objc func cutAction(sender: NSMenuItem) {
        
    }
    @objc func deleteAction(sender: NSMenuItem) {
        
    }
    @objc func ColorsItemAction(sender: NSMenuItem) {
        
    }
    @objc func LinesItemAction(sender: NSMenuItem) {
        
    }
    @objc func EditNoteItemAction(sender: NSMenuItem) {
        
    }
    @objc func AITranslateItemAction(sender: NSMenuItem) {
        self.showAITypeChooseView(aiConfigType: .translate)
    }
    @objc func AIProofreadItemAction(sender: NSMenuItem) {
        self.showAITypeChooseView(aiConfigType: .proofreading)
    }
    @objc func AIRewriteItemAction(sender: NSMenuItem) {
        self.showAITypeChooseView(aiConfigType: .reWriting)
    }
     
}
 

// MARK: - KMSnapshotWindowControllerDelegate

extension KMMainViewController: KMSnapshotWindowControllerDelegate {
    func snapshotControllerWillClose(_ controller: KMSnapshotWindowController) {
        self.leftSideViewController.snapshotControllerWillClose(controller)
    }
    
    func snapshotController(_ controller: KMSnapshotWindowController, miniaturizedRect isMiniaturize: Bool) -> NSRect {
        if isMiniaturize && self.interactionMode != .presentation {
            if self.interactionMode != .legacyFullScreen && self.model.leftPanelOpen == false {
                self.toggleLeftPane()
            } else if self.interactionMode == .legacyFullScreen {
             }

            var row = NSNotFound
            for (i,sp) in self.leftSideViewController.snapshots.enumerated() {
                if controller.isEqual(sp.windowC) {
                    row = i
                }
            }
            if (row != NSNotFound) {
                if self.leftSideViewController.type.methodType != .snapshot {
                    self.leftSideViewController.leftView.segmentedControl.selectedSegment = 5
                } else {
                    self.leftSideViewController.refreshUIOfSnapshotIfNeed()
                }
                self.leftSideViewController.snapshotTableView.scrollRowToVisible(row)
            }
        }
        
        return self.leftSideViewController.snapshotController(controller, miniaturizedRect: isMiniaturize)
    }
    
    func snapshotControllerDidFinishSetup(_ controller: KMSnapshotWindowController) {
        self.leftSideViewController.snapshotControllerDidFinishSetup(controller)
    }
}

// MARK: - CPDFDocumentDelegate

extension KMMainViewController: CPDFDocumentDelegate {
    func documentDidBeginDocumentFind(_ document: CPDFDocument!) {
        self.leftSideViewController.documentDidBeginFind()
     }
    
    func documentDidEndDocumentFind(_ document: CPDFDocument!) {
        self.leftSideViewController.documentDidEndFind()
     }
}