PDF_Office/PDF Office/PDF Master/KMClass/KMPDFViewController/KMPDFViewController.swift

//
//  KMPDFViewController.swift
//  PDF Reader Pro
//
//  Created by Niehaoyu on 2024/10/18.
//

import Cocoa

class KMPDFViewController: NSViewController {

    @IBOutlet var contendBox: NSBox!
    
    @IBOutlet var toolbarBox: NSBox! //工具栏Box
    
    @IBOutlet var bottomContendBox: NSBox! //
    
    @IBOutlet var sidebarBox: NSBox! //左侧边栏Box
        
    @IBOutlet var infoContendSplitView: KMPDFSplitView!
    @IBOutlet var infoSplitLeftView: NSView!
    @IBOutlet var infoSplitRightView: NSView!
    @IBOutlet var infoSplitCenterView: NSView!
    
    @IBOutlet var pdfSplitView: NSSplitView!
    @IBOutlet var pdfSplitTopView: NSView!
    @IBOutlet var pdfSplitBottomView: NSView!
    
    @IBOutlet var toolbarBoxHeightConst: NSLayoutConstraint!
    @IBOutlet var infoSplitViewLeftConst: NSLayoutConstraint!
    
    
    
    var viewManager: KMPDFViewManager = KMPDFViewManager.init()
    var toolbarManager: KMPDFToolbarManager = KMPDFToolbarManager.init()
    
    var pdfView: CPDFListView!
    var document: CPDFDocument?
    var myDocument: NSDocument?
    
    var insertDocuments: Set<CPDFDocument> = [] //插入新文档时,SDK会出现崩溃,临时记录

    //工具栏
    private var pdfToolbarController: KMPDFToolbarController?
    
    //BOTA SideBar
    private var sideBarController: KMPDFSideBarController?
    
    //页面编辑
    private var pageEditViewController: KMNPageEditViewController?
    
    //DisplaySetting
    private var displaySettingController: KMNDisplayViewController?
    
    private var splitPDFController: KMSplitPDFViewController?

    override func viewDidLoad() {
        super.viewDidLoad()
        // Do view setup here.
        
        toolbarManager.pdfViewManager = viewManager
    }
    
    override func viewDidAppear() {
        super.viewDidAppear()
        
        setupUI()
        
    }
    
    func setupUI() {
        
        initPDFView()
        
        initToolbar()
        
        initSideBar()
        
        setUpSplitView()
        
    }
    
    //MARK: - PDFView
    func initPDFView() {
        if pdfView == nil {
            pdfView = CPDFListView.init(frame: CGRectMake(0, 0, 100, 100))
            pdfView?.autoresizingMask = [.width, .height]
        }
        pdfView.document = self.document
        
        pdfView.delegate = self
        pdfView.pdfListViewDelegate = self
        
        reloadPDFSplitInfo()
    }
    
    //MARK: - SplitView
    func setUpSplitView() {
        infoContendSplitView.wantsLayer = true
        infoContendSplitView.layer?.backgroundColor = NSColor.clear.cgColor
        infoContendSplitView.delegate = self
        infoContendSplitView.layer?.masksToBounds = true
        
        infoContendSplitView.setPosition(-10, ofDividerAt: 0)
        infoContendSplitView.setPosition(CGRectGetWidth(infoContendSplitView.frame), ofDividerAt: 1)
        
    }
    
    func setupSplitPDFController() {
        if splitPDFController == nil {
            splitPDFController = KMSplitPDFViewController.init()
        }
        splitPDFController?.view.frame = pdfSplitBottomView.bounds
        splitPDFController?.view.autoresizingMask = [.height, .width]
        splitPDFController?.viewManager = self.viewManager
        splitPDFController?.delegate = self
        
        pdfSplitBottomView.addSubview(splitPDFController!.view)
    }
    
    //MARK: - 工具栏
    func initToolbar() {
        toolbarBox.borderWidth = 0
        
        if pdfToolbarController == nil {
            pdfToolbarController = KMPDFToolbarController.init()
        }
        pdfToolbarController?.view.frame = toolbarBox.bounds
        pdfToolbarController?.view.autoresizingMask = [.width, .height]
        pdfToolbarController?.delegate = self
        toolbarBox.contentView = pdfToolbarController?.view
        
        pdfToolbarController?.viewManager = viewManager
        pdfToolbarController?.toolbarManager = toolbarManager
        pdfToolbarController?.pdfView = pdfView
        pdfToolbarController?.setUpData()
        
        refreshToolbarView()
    }
    
    func refreshToolbarView() {
        
        let _manager = viewManager
        
        if _manager.toolMode == .Markup ||
            _manager.toolMode == .Edit ||
            _manager.toolMode == .Form ||
            _manager.toolMode == .Fill ||
            _manager.toolMode == .Convert ||
            _manager.toolMode == .Protect ||
            _manager.toolMode == .Tools {
            
            toolbarBoxHeightConst.constant = 80
        } else {
            if _manager.isPageEditMode {
                toolbarBoxHeightConst.constant = 80
            } else {
                toolbarBoxHeightConst.constant = 40
            }
        }
        
    }
    
    //MARK: - 侧边栏
    func initSideBar() {
        sidebarBox.borderWidth = 0
        
        if sideBarController == nil {
            sideBarController = KMPDFSideBarController.init()
        }
        sideBarController?.view.frame = sidebarBox.bounds
        sideBarController?.view.autoresizingMask = [.width, .height]
        sidebarBox.contentView = sideBarController?.view
        
        sideBarController?.pdfView = pdfView
        sideBarController?.delegate = self
        sideBarController?.pdfViewManager = viewManager
        
        sideBarController?.reloadData()
    }
    
    //MARK: - PDFDisplayView
    func updatePDFDisplaySettingView() {
        
        if viewManager.showDisplayView {
            infoSplitViewLeftConst.constant = 264
        } else {
            infoSplitViewLeftConst.constant = 44
        }
        
        if viewManager.showDisplayView {
            if displaySettingController == nil {
                displaySettingController = KMNDisplayViewController.init()
            }
            displaySettingController?.view.frame = CGRectMake(0, 0, 264, CGRectGetHeight(bottomContendBox.frame))
            displaySettingController?.view.autoresizingMask = [.height, .maxXMargin]
            bottomContendBox.addSubview(displaySettingController!.view)
            displaySettingController?.pdfView = self.pdfView
            displaySettingController?.viewManager = self.viewManager
            displaySettingController?.delegate = self
            
            displaySettingController?.reloadData()
            
        } else {
            displaySettingController?.view.removeFromSuperview()
            displaySettingController = nil
        }
    }
    
    
    //MARK: - 页面编辑
    func enterPageEditMode() {
        pageEditViewController = KMNPageEditViewController(self.document)
        if(pageEditViewController != nil) {
            bottomContendBox.addSubview(pageEditViewController!.view)
            pageEditViewController?.view.frame = bottomContendBox.bounds
            pageEditViewController?.thumbnailBaseViewDelegate = self
            pageEditViewController?.view.autoresizingMask = [.width,.height]
        }
    }
    
    func exitPageEditMode() {
        if pageEditViewController != nil {
            pageEditViewController?.view.removeFromSuperview()
            pageEditViewController = nil
        }
    }
    
    //MARK: - 阅读模式
    func openPDFReadMode() {
        if viewManager.showDisplayView {
            viewManager.showDisplayView = false
            pdfToolbarController?.reloadLeftIconView()
            
            updatePDFDisplaySettingView()
        }
        infoSplitViewLeftConst.constant = 0
        toolbarBoxHeightConst.constant = 0
        
        
    }
    
    func exitPDFReadMode() {
        viewManager.isPDFReadMode = false
        
        infoSplitViewLeftConst.constant = 44
        
        refreshToolbarView()

    }
    
    //MARK: - PPT
    
    
    //MARK: - PDF分割
    func reloadPDFSplitInfo() {
        if pdfView?.viewSplitMode == .disable {
            pdfSplitView.isHidden = true
            pdfView.frame = infoSplitCenterView.bounds
            infoSplitCenterView.addSubview(pdfView)
            
            if splitPDFController != nil {
                splitPDFController = nil
            }
        } else {
            pdfSplitView.isHidden = false
            pdfView.frame = pdfSplitTopView.bounds
            pdfSplitTopView.addSubview(pdfView)
            
            setupSplitPDFController()
            
            if pdfView.viewSplitMode == .horizontal {
                pdfSplitView.isVertical = true
                
            } else if pdfView.viewSplitMode == .vertical {
                pdfSplitView.isVertical = false
                 
            }
        }
        
        
    }
    
    
    
}

//MARK: - NSSplitViewDelegate
extension KMPDFViewController: NSSplitViewDelegate {
    
    func splitView(_ splitView: NSSplitView, shouldHideDividerAt dividerIndex: Int) -> Bool {
        return true
        
    }
    
    func splitView(_ splitView: NSSplitView, canCollapseSubview subview: NSView) -> Bool {
        return true
    }
    
    func splitView(_ splitView: NSSplitView, constrainSplitPosition proposedPosition: CGFloat, ofSubviewAt dividerIndex: Int) -> CGFloat {
        if dividerIndex == 0 {
            if proposedPosition > CGRectGetWidth(infoContendSplitView.frame)/2 - 10 {
                print(CGRectGetWidth(infoContendSplitView.frame)/2 - 10, NSDate())
                return CGRectGetWidth(infoContendSplitView.frame)/2 - 10
            }
        } else if dividerIndex == 1 {
            print(CGRectGetWidth(infoContendSplitView.frame)/2 - 10, NSDate())

        }
        return proposedPosition
    }
    
    func splitViewDidResizeSubviews(_ notification: Notification) {
    
        NSObject.cancelPreviousPerformRequests(withTarget: self, selector: #selector(splitViewResizeFinish), object: nil)
        self.perform(#selector(splitViewResizeFinish), with: nil, afterDelay: 0.15)

    }
     
    @objc func splitViewResizeFinish() {
        if infoContendSplitView.isSubviewCollapsed(infoSplitLeftView) {
            if viewManager.pdfSideBarType != .none {
                viewManager.pdfSideBarType = .none
                
                sideBarController?.reloadBOTAData()
            }
        }
    }
    
}

//MARK: - KMPDFSideBarControllerDelegate 左侧Sidebar代理
extension KMPDFViewController: KMPDFSideBarControllerDelegate {
    func kmPDFSideBarControllerDidSidebarTypeUpdated(_ view: KMPDFSideBarController) {
        if viewManager.pdfSideBarType == .none {
            infoContendSplitView.setPosition(-20, ofDividerAt: 0)
        } else if viewManager.pdfSideBarType == .search {
            infoContendSplitView.setPosition(264, ofDividerAt: 0)
        } else if viewManager.pdfSideBarType == .thumbnail {
            infoContendSplitView.setPosition(264, ofDividerAt: 0)
        } else if viewManager.pdfSideBarType == .outline {
            infoContendSplitView.setPosition(264, ofDividerAt: 0)
        } else if viewManager.pdfSideBarType == .bookmark {
            infoContendSplitView.setPosition(264, ofDividerAt: 0)
        } else if viewManager.pdfSideBarType == .annotation {
            infoContendSplitView.setPosition(264, ofDividerAt: 0)
        }
    }
    
    func kmPDFSideBarControllerDidGotoPage(_ view: KMPDFSideBarController, _ pageIndex: Int) {
        pdfView.go(toPageIndex: pageIndex, animated: true)
        
    }
    
}

//MARK: - KMPDFToolbarControllerDelegate 工具栏代理
extension KMPDFViewController: KMPDFToolbarControllerDelegate {
    func kmPDFToolbarControllerDidToolbarItemClicked(_ controller: KMPDFToolbarController, _ itemIdentifier: String) {
        
        print("toolbar点击", itemIdentifier)
        if itemIdentifier == KMPDFToolbar_ViewDisplay_Identifier {
            updatePDFDisplaySettingView()
        } else if itemIdentifier == KMPDFToolbar_PageEdit_Identifier {
            if viewManager.isPageEditMode == true {
                enterPageEditMode()
            } else {
                exitPageEditMode()
            }
        } else if itemIdentifier == KMPDFToolbar_convert_word_Identifier {
            let winC = KMConvertWordWindowController()
            let model = KMDocumentModel(url: self.pdfView.document.documentURL)
            winC.documentModel = model
            winC.own_beginSheetModal(for: self.view.window, completionHandler: nil)
        } else if itemIdentifier == KMPDFToolbar_convert_ppt_Identifier {
            let winC = KMConvertPPTsWindowController()
            winC.subType = 1
            let model = KMDocumentModel(url: self.pdfView.document.documentURL)
            winC.documentModel = model
            winC.own_beginSheetModal(for: self.view.window, completionHandler: nil)
        } else if itemIdentifier == KMPDFToolbar_convert_RTF_Identifier {
            let winC = KMConvertPPTsWindowController()
            winC.subType = 2
            let model = KMDocumentModel(url: self.pdfView.document.documentURL)
            winC.documentModel = model
            winC.own_beginSheetModal(for: self.view.window, completionHandler: nil)
        } else if itemIdentifier == KMPDFToolbar_convert_Text_Identifier {
            let winC = KMConvertPPTsWindowController()
            winC.subType = 4
            let model = KMDocumentModel(url: self.pdfView.document.documentURL)
            winC.documentModel = model
            winC.own_beginSheetModal(for: self.view.window, completionHandler: nil)
        } else if itemIdentifier == KMPDFToolbar_convert_CSV_Identifier {
            let winC = KMConvertPPTsWindowController()
            winC.subType = 5
            let model = KMDocumentModel(url: self.pdfView.document.documentURL)
            winC.documentModel = model
            winC.own_beginSheetModal(for: self.view.window, completionHandler: nil)
        } else if itemIdentifier == KMPDFToolbar_convert_excel_Identifier {
            let winC = KMConvertExcelWindowController()
            let model = KMDocumentModel(url: self.pdfView.document.documentURL)
            winC.documentModel = model
            winC.own_beginSheetModal(for: self.view.window, completionHandler: nil)
        } else if itemIdentifier == KMPDFToolbar_convert_image_Identifier {
            let winC = KMConvertImageWindowController()
            let model = KMDocumentModel(url: self.pdfView.document.documentURL)
            winC.documentModel = model
            winC.own_beginSheetModal(for: self.view.window, completionHandler: nil)
        } else if itemIdentifier == KMPDFToolbar_convert_HTML_Identifier {
            let winC = KMConvertHtmlWindowController()
            let model = KMDocumentModel(url: self.pdfView.document.documentURL)
            winC.documentModel = model
            winC.own_beginSheetModal(for: self.view.window, completionHandler: nil)
        } else if itemIdentifier == KMPDFToolbar_convert_Json_Identifier {
            let winC = KMConvertJsonWindowController()
            let model = KMDocumentModel(url: self.pdfView.document.documentURL)
            winC.documentModel = model
            winC.own_beginSheetModal(for: self.view.window, completionHandler: nil)
        } else if itemIdentifier == KMPDFToolbar_PageEdit_InsertFile_Identifier {
            pageEditViewController?.insertFromPDFAction()
        } else if itemIdentifier == KMPDFToolbar_PageEdit_InsertBlank_Identifier {
            pageEditViewController?.insertFromBlankAction()
        } else if itemIdentifier == KMPDFToolbar_PageEdit_InsertClip_Identifier {
            pageEditViewController?.insertFromClipboardAction()
        } else if itemIdentifier == KMPDFToolbar_PageEdit_InsertScanner_Identifier {
            pageEditViewController?.insertFromScannerAction()
        } else if itemIdentifier == KMPDFToolbar_PageEdit_Extract_Identifier {
            pageEditViewController?.extractPDFAction()
        } else if itemIdentifier == KMPDFToolbar_PageEdit_Replace_Identifier {
            pageEditViewController?.replacePDFAction()
        } else if itemIdentifier == KMPDFToolbar_PageEdit_Split_Identifier {
            pageEditViewController?.splitPDFAction()
        } else if itemIdentifier == KMPDFToolbar_PageEdit_Reverse_Identifier {
            pageEditViewController?.reversePDFAction()
        } else if itemIdentifier == KMPDFToolbar_PageEdit_LeftRotate_Identifier {
            pageEditViewController?.rotatePageLeftAction()
        } else if itemIdentifier == KMPDFToolbar_PageEdit_RightRotate_Identifier{
            pageEditViewController?.rotatePageRightAction()
        } else if itemIdentifier == KMPDFToolbar_PageEdit_Delete_Identifier {
            pageEditViewController?.deletePageAction()
        } else if itemIdentifier == KMPDFToolbar_PageEdit_Reduce_Identifier {
            pageEditViewController?.zoomOutPageAction()
        } else if itemIdentifier == KMPDFToolbar_PageEdit_Increase_Identifier {
            pageEditViewController?.zoomInPageAction()
        }
        
        refreshToolbarView()
    }
}


//MARK: - KMNDisplayViewControllerDelegate代理
extension KMPDFViewController: KMNDisplayViewControllerDelegate {
    //Display Mode
    func displayViewControllerDidDisplayModeChanged(_ controller: KMNDisplayViewController) {
        pdfView.layoutDocumentView()
        
    }
    
    //阅读模式
    func displayViewControllerDidReadModeUpdated(_ controller: KMNDisplayViewController) {
        if viewManager.isPDFReadMode {
            openPDFReadMode()
        } else {
            exitPDFReadMode()
        }
    }
    
    //PPT
    func displayViewControllerDidGotoSlideShow(_ controller: KMNDisplayViewController) {
        
    }

    //SplitView
    func displayViewControllerDidSplitModeChanged(_ controller: KMNDisplayViewController) {
        reloadPDFSplitInfo()
        
    }
    
    func displayViewControllerDidSplitFileChanged(_ controller: KMNDisplayViewController) {
        splitPDFController?.reloadData()
        
    }
}

//MARK: - KMSplitPDFViewControllerDelegate SplitPDFView
extension KMPDFViewController: KMSplitPDFViewControllerDelegate {
    func splitPDFViewControllerDidUpdateFilePath(_ controller: KMSplitPDFViewController) {
        displaySettingController?.reloadData()
        
    }
}

//MARK: - CPDFViewDelegate PDFView代理
extension KMPDFViewController: CPDFViewDelegate {
    func pdfViewDocumentDidLoaded(_ pdfView: CPDFView!) {
        sideBarController?.reloadPageTurnerData()

    }
    
    func pdfViewCurrentPageDidChanged(_ pdfView: CPDFView!) {
        sideBarController?.reloadPageTurnerData()
        
    }
    
    func pdfViewScaleDidChanged(_ pdfView: CPDFView!) {
        pdfToolbarController?.reloadSelectZoomView()
        
    }
    
}

extension KMPDFViewController: CPDFListViewDelegate {
    
    func pdfListViewKeyDownIsContinue(_ pdfListView: CPDFListView!, theEvent: NSEvent!) -> Bool {
        let command = theEvent.modifierFlags.contains(.command)
        let control = theEvent.modifierFlags.contains(.control)

        KMPrint(theEvent.keyCode)
        if pdfView.isEditing() {
            if control && theEvent.keyCode == 11 { // ctr + b
                
                return false
            } else if control && theEvent.keyCode == 34 { // ctr +i
           
                return false
            } else if theEvent.keyCode == 36 { // enter
                if pdfView.isCropMode {
                    
                    return false
                }
            }
        }
        
        if (theEvent.keyCode == 11 && command) { // command + B [添加书签]
           
            return false
        } else if (command && control && theEvent.keyCode == 14) { // command + control + E [注释 橡皮擦]
            return false
        } else if (theEvent.keyCode == 123) { // 向左
            if(self.pdfView.isEditing() && !self.pdfView.isSelecteditAreaNotEdit()) {
                return false
            } else {
                
            }
        } else if (theEvent.keyCode == 126) { // 向上
            if(self.pdfView.isEditing() && !self.pdfView.isSelecteditAreaNotEdit()) {
                return false
            } else {
                if (self.pdfView.isContinousScroll()) {
                    return true
                }
            }
        } else if (theEvent.keyCode == 124) { // 向右
            if(self.pdfView.isEditing() && !self.pdfView.isSelecteditAreaNotEdit()) {
                return false
            } else {
                 
            }
        } else if (theEvent.keyCode == 125) { // 向下
            if(self.pdfView.isEditing() && !self.pdfView.isSelecteditAreaNotEdit()) {
                return false
            } else {
                if (self.pdfView.isContinousScroll()) {
                    return true
                }
            }
        } else if (theEvent.keyCode == 36) {
            if pdfView.annotationType == .addImage || pdfView.annotationType == .addText {
                if self.pdfView.isEditImage {

                }
            }
        }
    
        if theEvent.keyCode == 53 {
            //ESC
            if view.window?.styleMask.contains(.fullScreen) == true {
                view.window?.toggleFullScreen(nil)
            }
            
            //阅读模式
            if viewManager.isPDFReadMode {
                exitPDFReadMode()
                viewManager.isPDFReadMode = false
                
                return true
            }
            
        }
        return true
    }
}
    

//MARK: - KMNThumbnailBaseViewDelegate
extension KMPDFViewController: KMNThumbnailBaseViewDelegate {
    func clickThumbnailViewControlle(pageEditVC:KMNThumbnailBaseViewController?,currentIndex:Int)  {
        exitPageEditMode()
        viewManager.isPageEditMode = false
        pdfToolbarController?.reloadPageEditView()
    }
    
    func insertPDFThumbnailViewControlle(pageEditVC: KMNThumbnailBaseViewController?, pdfDocment: CPDFDocument?) {
        if(pdfDocment != nil) {
            insertDocuments.insert(pdfDocment!)
        }
    }
}