PDF_Office/PDF Office/PDF Master/Class/PDFTools/PageEdit/Controller/KMPDFEditViewController.swift

//
//  KMPDFEditViewController.swift
//  PDF Master
//
//  Created by tangchao on 2023/1/6.
//

class KMPDFEditViewController: KMPDFThumbViewBaseController {
    @IBOutlet weak var topBarBox: NSBox!
    @IBOutlet weak var contentBox: NSBox!
    
    var toolBar = KMPDFEditToolbar(frame: NSZeroRect)
    override var thumbnailView: KMPDFThumbnailView {
        get {
            if (super.thumbnailView is KMPageEditThumbnailView) {
                return super.thumbnailView
            }
            
            super.thumbnailView = KMPageEditThumbnailView()
            return super.thumbnailView
        }
        set {
            super.thumbnailView = newValue
        }
    }
    
    var pageEditThumbnailView: KMPageEditThumbnailView {
        get {
            return self.thumbnailView as! KMPageEditThumbnailView
        }
    }
    
    override var menuItems: [[KMPDFThumbViewMenu.Item]] {
        get {
            return [[.delete], [.rightRotate, .leftRotate],
                    [.copy, .cut, .paste], [.showPageSize], [.print]]
        }
    }
    
    private var pdfDocument: CPDFDocument?
    var listView: CPDFView?
    
    var windowController: NSWindowController?
    
    var pageIsUpdate: Bool = true
    /// 临时存储 document
    private var documentCaches: [CPDFDocument] = []
    var selectedPages: [Int] = []
    
    private let defaultItemSize = NSMakeSize(218 + 12 * 2, 292 + 24)
    var itemSize: NSSize = NSMakeSize(218 + 12 * 2, 292 + 24)
    
    private var lockedFiles: Array<URL> = []
    
    // index=1 双击退出;index=2 打印
    var itemClick: KMCommonClickBlock?
    var documentEditedCallback: KMCommonBlock?
    
    internal var viewFirstAppear = true
    private var _listViewCurrentIndex: Int = 0
    var listViewCurrentIndex: Int {
        get {
            return self._listViewCurrentIndex
        }
    }
    
    required init?(coder: NSCoder) {
        super.init(coder: coder)
    }
    
    init(_ document: CPDFDocument) {
        super.init(nibName: "KMPDFEditViewController", bundle: nil)
        
        self.pdfDocument = document
        self.kmUndoManager = UndoManager()
    }
    
    override func viewDidAppear() {
        super.viewDidAppear()
        
        if (self.pdfDocument?.isLocked == false && self.viewFirstAppear) {
            self.viewFirstAppear = false
            self.thumbnailView.document = self.pdfDocument!
            self.thumbnailView.reloadData()
        }
        
        if let view = self.listView { // 存储阅读视图的当前页面索引
            self._listViewCurrentIndex = view.currentPageIndex
        }
        if (self.selectedPages.isEmpty) {
            var index: Int = 0
            if (self.listView != nil) {
                index = self.listView!.currentPageIndex
            }
            
            self.selectPages(indexs: IndexSet(integer: index))
            self.toolBar.setPageRangeString("\(index+1)")
        } else {
            var indexs: IndexSet = []
            for pageIndex in self.selectedPages {
                if (pageIndex < 0 || pageIndex >= (self.listView?.document.pageCount)!) {
                    continue
                }
                
                indexs.insert(pageIndex)
            }
            
            self.selectPages(indexs: indexs)
            self.toolBar.setPageRangeString(KMPageRangeTools.newParseSelectedIndexs(selectedIndex: indexs.sorted()))
        }
        
        self.view.window?.makeFirstResponder(self)
    }
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        self.thumbnailView.collectionView.backgroundColor(NSColor(hex: "#CED0D4"))
        
        let toolBar = self.toolBar
        self.topBarBox.contentView?.addSubview(toolBar)
        toolBar.frame = self.topBarBox.contentView!.bounds
        toolBar.autoresizingMask = [.width, .height]
        self.topBarBox.fillColor = NSColor(hex: "#F7F8FA")
        toolBar.itemClick = { [unowned self] item, index in
            guard let type = KMPageEditType(rawValue: index) else {
                if (index == KMPageEditType.insert_files) {
                    self.item_insertFile(sender: nil)
                } else if (index == KMPageEditType.insert_blank_page) {
                    self.item_insertBlankPage(sender: nil)
                } else if (index == KMPageEditType.insert_custom_page) {
                    self.item_insertCustomPage(sender: nil)
                }
                return
            }
            
            if (type == .leftRotate) {
                self.leftRotateAction()
            } else if (type == .rightRotate) {
                self.rightRotateAction()
            } else if (type == .zoomOut) {
                self.zoomOutButtonAction(sender: item?.zoomOutButton)
            } else if (type == .zoomIn) {
                self.zoomInButtonAction(sender: item?.zoomInButton)
            } else if (type == .extract) {
                self.extractPageItemAction(menu: nil)
            } else if (type == .replace) {
                self.replacePageItemAction(menu: nil)
            } else if (type == .split) {
                self.splitMenuAction(sender: nil)
            } else if (type == .reverse) {
                self.reverseMenuAction(sender: nil)
            } else if (type == .delete) {
                self.deleteAction()
            } else if (type == .pageRange) {
                self.km_comboBoxSelectionDidChange((item?.pageRangeView)!)
            }
        }
        toolBar.pageRangeValueDidChange = { [unowned self] value, _ in
            self.km_controlTextDidEndEditing(self.toolBar.pageRangeView!)
        }
        
        let view = self.thumbnailView
        self.contentBox.contentView?.addSubview(view)
        view.frame = self.contentBox.contentView!.bounds
        view.autoresizingMask = [.width, .height]
        self.pageEditThumbnailView.delegate = self
        self.pageEditThumbnailView.selectionDidChange = { [weak self] selectedIndexs in
            self?.view.window?.makeFirstResponder(self)
            
            var replaceEnable: Bool = false
            if (selectedIndexs.count ==  1) { /// 替换item 置灰
                replaceEnable = true
            }
            
            var enabled: Bool = false
            if (selectedIndexs.count > 0) {
                enabled = true
            }
            
            if (self?.toolBar.toolBar.items == nil) {
                return
            }
            
            var replaceItem: KMToolBoxItem?
            for i in 0 ..< (self?.toolBar.toolBar.items.count)! {
                let item: KMToolBoxItem = (self?.toolBar.toolBar.items[i])!
                if (item.itemIdentifier == KMToolbarPageEditReplaceItemIdentifier) {
                    replaceItem = item
                }

                if (item.itemIdentifier == KMToolbarPageEditExtractItemIdentifier ||
                    item.itemIdentifier == KMToolbarPageEditReverseItemIdentifier ||
                    item.itemIdentifier == KMToolbarPageEditDeleteItemIdentifier ||
                    item.itemIdentifier == KMToolbarPageEditLeftRotateItemIdentifier ||
                    item.itemIdentifier == KMToolbarPageEditRightRotateItemIdentifier) {
                    item.unEnabled = !enabled
                }
            }
            
            if (replaceItem != nil) {
                replaceItem?.unEnabled = !replaceEnable
            }
            
            if (self!.pageIsUpdate == false) {
                self?.pageIsUpdate = true
                return
            }
            
            var indexs: Array<Int> = []
            for indexPath in selectedIndexs {
                indexs.append(indexPath.item)
            }
            
            indexs.sort(){$0 < $1}
            
            self?.toolBar.setPageRangeString(KMPageRangeTools.newParseSelectedIndexs(selectedIndex: indexs))
        }
    }
    
    // MARK: Private Methods
    
    private func getSelecteIndex() -> Int {
        var index: Int = 0
        for indexpath in self.thumbnailView.selectionIndexPaths {
            index = indexpath.item
        }
        return index
    }
    
    private func getSelectedPage() -> CPDFPage {
        return (self.pdfDocument!.page(at: UInt(getSelecteIndex())))!
    }
    
    private func showPasswordWindow(_ insertIndex: Int) {
        if (self.lockedFiles.count > 0) {
            let fileURL = self.lockedFiles.first
            KMPasswordInputWindow.openWindow(window: self.view.window!, type: .open, url: fileURL!) { result, password in
                if (result == .cancel) { /// 取消
                    self.lockedFiles.removeFirst()
                    if (self.lockedFiles.count > 0) {
                        self.showPasswordWindow(insertIndex)
                    }
                    return
                }
                
                /// 解密成功
                let document = CPDFDocument(url: fileURL)
                document?.unlock(withPassword: password)
                var newInsertIndex: Int = insertIndex
                var pages: [CPDFPage] = []
                var indexs: IndexSet = IndexSet()
                for i in 0 ..< document!.pageCount {
                    let page = document?.page(at: i)
                    pages.append(page!)
                    indexs.insert(newInsertIndex)
                    newInsertIndex += 1
                }
                
                self.insertPagesForDescSort(pages: pages, indexs: indexs)
                
                self.lockedFiles.removeFirst()
                if (self.lockedFiles.count > 0) {
                    self.showPasswordWindow(newInsertIndex)
                }
            }
        }
    }
    
    @objc private func insertPagesForDescSort(pages: Array<CPDFPage>, indexs: IndexSet) {
        self.insertPages(pages: pages, at: indexs)
        
        /// 选中插入的页面
        self.selectPages(indexs: indexs)
    }
    
    private func selectPages(indexs: IndexSet) {
        var indexpaths: Set<IndexPath> = []
        for index in indexs {
            indexpaths.insert(IndexPath(item: index, section: 0))
        }
        self.selectPages(indexpaths: indexpaths)
    }
    
    private func km_becomeFirstResponder() {
        DispatchQueue.main.async {
            self.view.window?.makeFirstResponder(self)
        }
    }
    
    fileprivate func dealPdfDocumentDidEditCallback() {
        // 提醒外面文档已编辑
        guard let callback = self.documentEditedCallback else {
            return
        }
        callback()
    }
    
    // MARK: - 清空选择
    
    private func clearSelectPages() {
        self.selectPages(indexs: IndexSet())
    }
    
    override func refreshUI(indexpaths: Set<IndexPath> = []) {
        super.refreshUI(indexpaths: indexpaths)
        
        self.listView?.layoutDocumentView()
    }
    
    override func cutAction() {
        if (self.thumbnailView.selectionIndexPaths.count == (self.listView?.document.pageCount)!) {
            let _ = CustomAlertView(message: NSLocalizedString("Unable to delete all pages", comment: ""), from: self.thumbnailView, with: .blue)
            return
        }
        super.cutAction()
    }
    
    override func deleteIndexPaths(indexpaths: Set<IndexPath>) {
        guard let pageCount = self.listView?.document.pageCount else {
            KMPrint("pageCount invalid.")
            return
        }
        if (indexpaths.count == pageCount) {
            let _ = CustomAlertView(message: NSLocalizedString("Unable to delete all pages", comment: ""), from: self.thumbnailView, with: .blue)
            return
        }
        
        self.dealPdfDocumentDidEditCallback()
        
        self.clearSelectPages()
        super.deleteIndexPaths(indexpaths: indexpaths)
    }
    
    override func leftRotateIndexpaths(indexpaths: Set<IndexPath>) {
        if (!self.indexpathsIsValid(indexpaths: indexpaths) || !self.canRotate()) {
            return
        }
        super.leftRotateIndexpaths(indexpaths: indexpaths)
        self.dealPdfDocumentDidEditCallback()
    }
    
    override func rightRotateIndexpaths(indexpaths: Set<IndexPath>) {
        if (!self.indexpathsIsValid(indexpaths: indexpaths) || !self.canRotate()) {
            return
        }
        
        super.rightRotateIndexpaths(indexpaths: indexpaths)
        self.dealPdfDocumentDidEditCallback()
    }
    
    override func pasteAction() {
        if (self.canPaste()) {
            super.pasteAction()
            
            self.dealPdfDocumentDidEditCallback()
        }
    }
    
    // MARK: Undo / Redo
    
    @objc private func reversePages(indexs: IndexSet) {
        KMPageEditTools.reverse((self.listView?.document)!, indexs
        ) { result, error in
            if (result) {
                self.refreshUI()
                
                self.kmUndoManager?.registerUndo(withTarget: self, selector: #selector(self.reversePages), object: indexs)
            }
        }
    }
    
    @objc private func dragPages(_ indexs: IndexSet, _ toIndex: Int, reversed: Bool = false) {
        if (indexs.count <= 0) {
            return
        }
        
        self.listViewCurrentPageUpdateForDrag(indexs: indexs, toIndex: toIndex)
        
        // 插入位置偏移量
        var toIndexOffset: Int = 0
        // 需要移动的页面index数组
        var newPages: [CPDFPage] = []
        // 从后面往前删除
        for index in indexs.reversed() {
            let newPage = (self.listView?.document.page(at: UInt(index)))?.copy() as? CPDFPage
            if (newPage == nil) {
                continue
            }
            newPages.insert(newPage!, at: 0)
            self.listView?.document.removePage(at: UInt(index))
            if (index < toIndex) { /// 删除了插入位置前面的页面，需要改变 toIndex
                toIndexOffset += 1
            }
        }
        if (indexs.last! < toIndex) { // 往后拖拽
            toIndexOffset -= 1
        }
        let insertIndex = max(0, toIndex-toIndexOffset)
        var selectIndexs: IndexSet = []
        for (index, page) in newPages.enumerated() {
            self.listView?.document.insertPageObject(page, at: UInt(insertIndex+index))
            selectIndexs.insert(insertIndex+index)
        }
        
        // 新增 undo
        self.kmUndoManager?.registerUndo(withTarget: self) { target in
            target.clearSelectPages()
            target.dragPages_reversed(indexs, toIndex, newPages)
        }
        
        // 刷新UI
        self.refreshUI()
        
        if (reversed == false) {
            self.selectPages(indexs: selectIndexs)
        }
    }
    
    @objc private func dragPages_reversed(_ indexs: IndexSet, _ toIndex: Int, _ pages: [CPDFPage]) {
        if (indexs.count <= 0) {
            return
        }
        
        // 插入位置偏移量
        var toIndexOffset: Int = 0
        for index in indexs {
            if (index < toIndex) { /// 删除了插入位置前面的页面，需要改变 toIndex
                toIndexOffset += 1
            }
        }
        if (indexs.last! < toIndex) { // 往后拖拽
            toIndexOffset -= 1
        }
        let insertIndex = max(0, toIndex-toIndexOffset)
        // 需要移动的页面index数组
        var datas: [Int] = []
        for index in insertIndex ..< (insertIndex + indexs.count) {
            datas.append(index)
        }
        
        // 从后面开始删除
        for index in datas.reversed() {
            self.listView?.document.removePage(at: UInt(index))
        }
        // 再插入到对应的位置
        for (index,value) in indexs.enumerated() {
            var page: CPDFPage?
            if (index < pages.count) {
                page = pages[index]
            }
            if (page != nil) {
                self.listView?.document.insertPageObject(page!, at: UInt(value))
            }
            
        }
        // 新增 undo
        self.kmUndoManager?.registerUndo(withTarget: self) { target in
            target.dragPages(indexs, toIndex, reversed: true)
        }
        // 刷新UI
        self.refreshUI()
    }
    
    // MARK: topBar 事件
    
    @objc func zoomOutButtonAction(sender: NSButton?) {
        if (self.itemSize.width > self.defaultItemSize.width) {
            return
        }
        if (self.itemSize.width < self.defaultItemSize.width) {
            self.itemSize = self.defaultItemSize
        } else if (self.itemSize.width == self.defaultItemSize.width) {
            self.itemSize = NSSize(width: self.defaultItemSize.width*1.5, height: self.defaultItemSize.height*1.5)
        }

        self.toolBar.zoomInButton?.isEnabled = true
        if (self.itemSize.width > self.defaultItemSize.width) {
            sender?.isEnabled = false
        }
        self.pageEditThumbnailView.zoomOut()
    }
    
    @objc func zoomInButtonAction(sender: NSButton?) {
        if (self.itemSize.width < self.defaultItemSize.width) {
            return
        }
        if (self.itemSize.width > self.defaultItemSize.width) {
            self.itemSize = self.defaultItemSize
        } else if (self.itemSize.width == self.defaultItemSize.width) {
            self.itemSize = NSSize(width: self.defaultItemSize.width*0.5, height: self.defaultItemSize.height*0.5)
        }

        self.toolBar.zoomOutButton?.isEnabled = true
        if (self.itemSize.width < self.defaultItemSize.width) {
            sender?.isEnabled = false
        }
        self.pageEditThumbnailView.zoomIn()
    }
    
    // MARK: menu 菜单事件
    
    @objc func item_insertFile(sender: NSMenuItem?) {
        let panel = NSOpenPanel()
        panel.allowedFileTypes = ["pdf"]
        panel.allowsMultipleSelection = true
        panel.beginSheetModal(for: self.view.window!) { [weak self] response in
            if (response == .cancel) {
                return
            }
            
            /// 处理page
            var insertIndex: Int = (self?.getSelecteIndex())!
            self!.lockedFiles.removeAll()
            var pages: Array<CPDFPage> = []
            self?.documentCaches.removeAll()
            var indexs = IndexSet()
            for fileURL in panel.urls {
                let document = CPDFDocument(url: fileURL)
                if ((document?.isLocked)!) {
                    self?.lockedFiles.append(fileURL)
                    continue
                }
                
                self?.documentCaches.append(document!)
                for i in 0 ..< document!.pageCount {
                    let page = document?.page(at: i)
                    pages.append(page!)
                    insertIndex += 1
                    indexs.insert(insertIndex)
                }
            }
            self?.insertPagesForDescSort(pages: pages, indexs: indexs)
            
            if ((self?.lockedFiles.count)! <= 0) {
                self?.refreshUI()
                return
            }
            
            self?.refreshUI()
            self!.showPasswordWindow(insertIndex)
        }
    }
    
    @objc func item_insertBlankPage(sender: NSMenuItem?) {
        let index = self.getSelecteIndex()
        let page = self.listView?.document.page(at: UInt(index))
        
        self.documentCaches.removeAll()
        let document = CPDFDocument()
        document?.insertPage((page?.bounds.size)!, at: 0)
        self.documentCaches.append(document!)
        let newPage: CPDFPage = (document?.page(at: 0))!
        self.insertPagesForDescSort(pages: [newPage], indexs: IndexSet(integer: index+1))
    }
    
    @objc func item_insertCustomPage(sender: NSMenuItem?) {
        let windowController = KMPageEditInsertCustomPageWindowController(windowNibName: "KMPageEditInsertCustomPageWindowController")
        if (self.thumbnailView.selectionIndexPaths.count > 0) {
            windowController.selectedPageSize = self.getSelectedPage().bounds.size
        }
        
        self.windowController = windowController
        self.view.window?.beginSheet(windowController.window!, completionHandler: { response in})
        
        windowController.itemClick = {
            [weak self] (index: Int) in
            
            if (index == 1) { /// 取消
                self?.view.window?.endSheet((self!.windowController?.window)!)
                self?.windowController = nil
                return
            }
            
            /// 插入
            /// 样式
            let windowController_Insert: KMPageEditInsertCustomPageWindowController = self?.windowController as! KMPageEditInsertCustomPageWindowController
            let type = windowController_Insert.typeIndex
            
            /// 页面大小
            let pageSize = windowController_Insert.pageSize
            
            /// 方向
            let direction = windowController_Insert.direction
            
            /// 插入位置
            let insertIndex: Int = (self?.getSelecteIndex())!
            if (type == 1) { /// 空白页
                let document = CPDFDocument()
                document?.insertPage(pageSize, at: 0)
                let page: CPDFPage = (document?.page(at: 0))!
                
                if (direction == 0) { /// 纵向
                    page.rotation = 90
                }

                self?.insertPagesForDescSort(pages: [page], indexs: IndexSet(integer: insertIndex+1))
            } else {
                let document = CPDFDocument()
                var imageName = "plaid"
                if (type == 2) {
                    imageName = "horizontal_line"
                } else if (type == 3) {
                    imageName = "five_line_score"
                }
                let imagePath = Bundle.main.pathForImageResource(imageName)
                document?.insertPage(pageSize, withImage: imagePath, at: 0)
                let page: CPDFPage = (document?.page(at: 0))!
                if (direction == 0) { /// 纵向
                    page.rotation = 90
                }

                self?.insertPagesForDescSort(pages: [page], indexs: IndexSet(integer: insertIndex+1))
            }
            
            self?.view.window?.endSheet((self?.windowController?.window)!)
            self?.windowController = nil
        }
    }
    
//    // MARK: toolBar 菜单事件

    @objc func splitMenuAction(sender: AnyObject?) {
        let model = KMPageEditSplitSettingModel()
        model.documentURL = self.listView?.document.documentURL
        model.fileName = model.documentURL.lastPathComponent
        model.pathExtension = model.fileName.components(separatedBy: ".").last
        
        if (self.thumbnailView.selectionIndexPaths.count > 0) {
            model.type = 2
            model.pageRangeType = 2
            var selectedIndexs: Array<Int> = []
            for indexpath in self.thumbnailView.selectionIndexPaths {
                selectedIndexs.append(indexpath.item)
            }
            
            model.pageRangeString = KMPageRangeTools.newParseSelectedIndexs(selectedIndex: selectedIndexs)
        } else {
            model.type = 0
        }
        
        let windowController = KMPageEditSplitWindowController(model)
        self.view.window?.beginSheet(windowController.window!)
        self.windowController = windowController
        
        windowController.itemClick = { [weak self] index, value in
            if (index == 1) { /// 取消
                self?.view.window?.endSheet((self?.windowController!.window)!)
                self?.windowController = nil
                return
            }
            
            /// 拆分
            let windowController_split: KMPageEditSplitWindowController = self?.windowController as! KMPageEditSplitWindowController
            let outputModel: KMPageEditSplitSettingModel = windowController_split.model! as! KMPageEditSplitSettingModel
            
            self?.view.window?.endSheet((self?.windowController!.window)!)
            self?.windowController = nil
            
            let panel = NSOpenPanel()
            panel.canChooseFiles = false
            panel.canChooseDirectories = true
            panel.canCreateDirectories = true
            let button = NSButton.init(checkboxWithTitle: "保存后打开文档", target: nil, action: nil)
            button.state = .off
            panel.accessoryView = button
            panel.beginSheetModal(for: (self?.view.window)!) { response in
                if (response == .cancel) {
                    return
                }
                KMPageEditTools.split((self?.listView?.document)!, outputModel, panel.url!.path, outputModel.outputFileNameDeletePathExtension) { result, outputFilepaths, error in
                    if (result) {
                        if (button.state == .on) {
                            for filepath in outputFilepaths! {
                                if !filepath.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()
                                } else {
                                    NSDocumentController.shared.openDocument(withContentsOf: URL(fileURLWithPath: filepath), display: true) { document, alreadyOpen, error in
                                        
                                    }
                                }
                            }

                        } else {
                            var urls: [URL] = []
                            for filepath in outputFilepaths! {
                                urls.append(URL(fileURLWithPath: filepath))
                            }
                            NSWorkspace.shared.activateFileViewerSelecting(urls)
                        }
                    }
                }
            }
        }
    }
    
    @objc func reverseMenuAction(sender: AnyObject?) {
        let selectedIndexs = self.thumbnailView.selectionIndexPaths
        if (selectedIndexs.count < 2 ) {
            let _ = CustomAlertView(message: "请先选择至少两个页面进行编辑操作", from: self.thumbnailView, with: .blue)
            return
        }
        
        var indexs: IndexSet = []
        for indexpath in selectedIndexs {
            indexs.insert(indexpath.item)
        }
        
        self.reversePages(indexs: indexs)
    }
}

extension KMPDFEditViewController {
    //MARK: menu Action
    
    @objc func insertPageItemAction(menu:NSMenuItem) {
        if (menu.tag == 0) {
            self.item_insertFile(sender: nil)
        } else if (menu.tag == 1) {
            self.item_insertBlankPage(sender: nil)
        } else if (menu.tag == 2) {
            self.item_insertCustomPage(sender: nil)
        }
    }
    
    @objc func extractPageItemAction(menu:NSMenuItem?) {
        if (self.thumbnailView.selectionIndexPaths.count == 0) {
            let alert = NSAlert()
            alert.messageText = "请选择需要提取的页面"
            alert.runModal()
            return
        }
        
        let windowController = KMPageEditExtractWindowController(windowNibName: "KMPageEditSettingBaseWindowController")
        self.view.window?.beginSheet(windowController.window!)
        self.windowController = windowController
        
        windowController.itemClick = { [weak self] index, value in
            if (index == 1) { /// 取消
                self?.view.window?.endSheet((self?.windowController?.window)!)
                self?.windowController = nil
                return
            }
            
            let windowController_extract: KMPageEditExtractWindowController = self?.windowController as! KMPageEditExtractWindowController
            /// 提取设置项
            let settingView: KMPageEditExtractSettingView = windowController_extract.settingView! as! KMPageEditExtractSettingView
            let oneDocumentPerPage: Bool = settingView.independentPerPageCheck.state != .on
            let extractAfterDelete: Bool = settingView.extractAfterDeletePageCheck.state == .on
            
            self?.view.window?.endSheet((self?.windowController!.window)!)
            self?.windowController = nil
            
            //            /// 提取的页面
            let indexpaths = self?.thumbnailView.selectionIndexPaths
            var indexs: IndexSet = []
            var extractPages: Array<CPDFPage> = []
            for indexpath in indexpaths! {
                let page = self?.listView?.document.page(at: UInt(indexpath.item))
                if (page != nil) {
                    extractPages.append(page!)
                }
                indexs.insert(indexpath.item)
            }
            
            KMPageEditTools.extract((self?.listView?.document)!, indexs, oneDocumentPerPage, (self?.view.window)!) { result, urls, error in
                if (result == false) {
                    return
                }
                
                if (extractAfterDelete) {
                    var indexpaths = Set<IndexPath>()
                    for page in extractPages {
                        let indexPage: Int = Int((self?.listView?.document.index(for: page))!)
                        indexpaths.insert(IndexPath(item: indexPage, section: 0))
                    }
                    self?.deleteIndexPaths(indexpaths: indexpaths)
                }
            }
        }
    }
    
    @objc func replacePageItemAction(menu:NSMenuItem?) {
        let panel = NSOpenPanel()
        panel.allowedFileTypes = ["pdf"]
        panel.allowsMultipleSelection = true
        panel.beginSheetModal(for: self.view.window!) { [weak self] response in
            if (response == .cancel) {
                return
            }
            
            /// 处理page
            var pages: Array<CPDFPage> = []
            let selectedIndex: Int = (self?.getSelecteIndex())!
            self?.listView?.document.removePage(at: UInt(selectedIndex))
            
            var insertIndex: Int = selectedIndex
            self?.lockedFiles.removeAll()
            for fileURL in panel.urls {
                let document = CPDFDocument(url: fileURL)
                if ((document?.isLocked)!) {
                    self?.lockedFiles.append(fileURL)
                    continue
                }
                
                for i in 0 ..< document!.pageCount {
                    let page = document?.page(at: i)
                    pages.append(page!)
                    self!.listView?.document.insertPageObject(page, at:UInt(insertIndex))
                    insertIndex += 1
                }
            }
            
            if ((self?.lockedFiles.count)! <= 0) {
                self?.refreshUI()
                return
            }
            
            self?.refreshUI()
            self!.showPasswordWindow(insertIndex)
        }
    }
    
    @objc func sharePageItemAction(menu:NSMenuItem) {
        let item = menu.parent!
        let index = (item.representedObject as! IndexSet).first ?? -1
        if Int(index) >= 0 {
            let doucument = self.pdfDocument!
            let page = doucument.page(at: UInt(index))
            let filename : String = doucument.documentURL.lastPathComponent
            let folderPath = (NSSearchPathForDirectoriesInDomains(FileManager.SearchPathDirectory.applicationSupportDirectory, FileManager.SearchPathDomainMask.userDomainMask, true).last?.stringByAppendingPathComponent(filename))!
            try? FileManager.default.removeItem(atPath: folderPath)
            let pdfdocument = CPDFDocument()
            let ret = pdfdocument!.importPages((item.representedObject as! IndexSet), from: self.thumbnailView.document, at: 0)
            let url = URL(fileURLWithPath: folderPath)
            if ret {
                let success = pdfdocument!.write(to:url)
                if success {
                    let workspace = NSWorkspace.shared
                    workspace.activateFileViewerSelecting([url])
                }
            }
            let represent : NSSharingService = menu.representedObject as! NSSharingService
            represent.perform(withItems: [url])
            
        }
    }
    
    @objc func showPageSizeItemAction(menu: NSMenuItem) {
        self.thumbnailView.isShowPageSize = !self.thumbnailView.isShowPageSize
        
        self.refreshUIForKeepSelecteds()
    }
    
    @objc func printItemAction(menu:NSMenuItem) {
        var datas: [Int] = []
        for indexpath in self.thumbnailView.selectionIndexPaths {
            datas.append(indexpath.item)
        }
        
        guard let callback = self.itemClick else {
            return
        }
        
        callback(2, datas.sorted())
    }
}

extension KMPDFEditViewController {
    override func menuItemAction_print(_ sender: Any?) {
        self.printItemAction(menu: sender as! NSMenuItem)
    }
    override func menuItemAction_showPageSize(_ sender: Any?) {
        self.showPageSizeItemAction(menu: sender as! NSMenuItem)
    }
}

// MARK: - KMSelectPopButtonDelegate

extension KMPDFEditViewController {
    func km_comboBoxSelectionDidChange(_ obj: KMDesignSelect) {
        self.pageIsUpdate = false
        
        let index: Int = obj.indexOfSelectedItem
        var indexpaths: Set<IndexPath> = []
        var type: KMPageRange?
        if (index <= 0) { /// 全部页面
            type = .all
        } else if (index == 1) { /// 奇数页
            type = .odd
        } else if (index == 2) { /// 偶数页
            type = .even
        } else if (index == 3) { /// 横向页
            type = .horizontal
        } else if (index == 4) { /// 纵向页
            type = .vertical
        } else { /// 自定义
            type = .custom
        }
        
        for i in 0 ..< (self.pdfDocument?.pageCount)! {
            if (type == .all) {
                indexpaths.insert(IndexPath(item: Int(i), section: 0))
            } else if (type == .odd) {
                if (i % 2 == 1) {
                    continue
                }
                indexpaths.insert(IndexPath(item: Int(i), section: 0))
            } else if (type == .even) {
                if (i % 2 == 0) {
                    continue
                }
                indexpaths.insert(IndexPath(item: Int(i), section: 0))
            } else if (type == .horizontal) {
                let page = self.pdfDocument?.page(at: i)
                guard let isVertical = page?.isVertical(), !isVertical else {
                    continue
                }
                indexpaths.insert(IndexPath(item: Int(i), section: 0))
            } else if (type == .vertical) {
                let page = self.pdfDocument?.page(at: i)
                guard let isHorizontal = page?.isHorizontal(), !isHorizontal else {
                    continue
                }
                indexpaths.insert(IndexPath(item: Int(i), section: 0))
            }
        }
        
        obj.editable = type == .custom
        if (type == .custom) {
            self.clearSelectPages()
            DispatchQueue.main.async {
                self.toolBar.pageRangeView?.stringValue = ""
                self.view.window?.makeFirstResponder(self.toolBar.pageRangeView?.textField)
            }
        } else {
            self.pageEditThumbnailView.selectPages(at: indexpaths)
        }
    }
    
    func km_controlTextDidEndEditing(_ obj: KMDesignSelect) {
        self.view.window?.makeFirstResponder(self)
        
        for index in 0 ..< obj.numberOfItems {
            if (obj.items[index] == obj.stringValue) {
                return
            }
        }
        
        let pages = KMPageRangeTools.findSelectPage(pageRangeString: self.toolBar.pageRangeView!.stringValue, pageCount: Int((self.listView?.document.pageCount)!))
        if (pages.isEmpty) {
            let alert = NSAlert()
            alert.messageText = NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: "")
            alert.runModal()
            self.toolBar.setPageRangeString("")
            DispatchQueue.main.async {
                self.view.window?.makeFirstResponder(self.toolBar.pageRangeView?.textField)
            }
            return
        }
        
        var indexpaths: Set<IndexPath> = []
        for page in pages {
            if (page >= 1) {
                indexpaths.insert(IndexPath(item: page-1, section: 0))
            }
        }
        
        self.pageIsUpdate = false
        self.pageEditThumbnailView.selectPages(at: indexpaths)
    }
}

// MARK: -
// MARK: KMThumbnailViewDelegate

extension KMPDFEditViewController: KMThumbnailViewDelegate {
    func thumbnailView(thumbanView: KMThumbnailView, didDragAddFiles files: [URL], indexpath: IndexPath) {
        /// 处理page
//        var insertIndex: Int = indexpath.item
//        self.lockedFiles.removeAll()
//        var pages: Array<CPDFPage> = []
//        self.documentCaches.removeAll()
//        var indexs = IndexSet()
//        for fileURL in files {
//            let document = CPDFDocument(url: fileURL)
//            if ((document?.isLocked)!) {
//                self.lockedFiles.append(fileURL)
//                continue
//            }
//
//            self.documentCaches.append(document!)
//            for i in 0 ..< document!.pageCount {
//                let page = document?.page(at: i)
//                pages.append(page!)
//                indexs.insert(insertIndex)
//                insertIndex += 1
//            }
//        }
//        self.insertPagesForDescSort(pages: pages, indexs: indexs)
//
//        if (self.lockedFiles.count <= 0) {
//            self.refreshUI()
//            return
//        }
//
//        self.refreshUI()
//        self.showPasswordWindow(insertIndex)
    }
    
    func thumbnailView(thumbanView: KMThumbnailView, didDrag dragedIndexPaths: [IndexPath], indexpath: IndexPath) {
        let toIndex = max(0, indexpath.item)
        var indexs = IndexSet()
        for indexpath in dragedIndexPaths {
            indexs.insert(indexpath.item)
        }
        self.dragPages(indexs, toIndex)
        
        if (dragedIndexPaths.count > 0) {
            self.dealPdfDocumentDidEditCallback()
        }
    }
    
    func thumbnailView(thumbanView: KMThumbnailView, insetForSectionAt section: Int) -> NSEdgeInsets {
        return NSEdgeInsets(top: 16, left: 16, bottom: 0, right: 32)
    }
    
    func thumbnailView(thumbanView: KMThumbnailView, sizeForItemAt indexpath: IndexPath) -> NSSize {
        self.thumbnailView.thumbnailSzie = CGSize(width: 120, height: 155)
        if (!self.thumbnailView.isShowPageSize) {
            return NSMakeSize(self.itemSize.width, self.itemSize.height+26)
        }
        return self.itemSize
    }
    
    func thumbnailView(thumbanView: KMThumbnailView, numberOfItemsInSection section: Int) -> Int {
        if let count = self.pdfDocument?.pageCount {
            return Int(count)
        }
        return 0
    }
    
    func thumbnailView(thumbanView: KMThumbnailView, itemForRepresentedObjectAt indexpath: IndexPath) -> NSCollectionViewItem {
        let cellView: KMPageEditThumbnailItem = thumbanView.collectionView.makeItem(withIdentifier: KMPageEditThumbnailItem.km_identifier, for: indexpath) as! KMPageEditThumbnailItem
        cellView.isShowPageSize = self.thumbnailView.isShowPageSize
        cellView.page = (self.pdfDocument?.page(at: UInt(indexpath.item)))!
        cellView.setPage(page: (self.pdfDocument?.page(at: UInt(indexpath.item)))!)
        
        cellView.doubleClickAction = { [unowned self] _ in
            guard let callback = self.itemClick else {
                return
            }
            callback(1, indexpath.item)
        }
        cellView.mouseDownAction = { [unowned self] _ in
            let selectionIndexPaths = self.thumbnailView.selectionIndexPaths
            if (selectionIndexPaths.count == 1 && selectionIndexPaths.first?.item == indexpath.item) {
                return
            }
            
            self.selectPages(indexs: IndexSet.init(integer: indexpath.item))
        }
        
        cellView.rightMouseDownAction = { [unowned self] _ in
            self.selectPages(indexs: IndexSet.init(integer: indexpath.item))
        }
        
        return cellView
    }
}

// MARK: - 系统菜单栏

extension KMPDFEditViewController {
    override func undo(_ sender: Any?) {
        super.undo(sender)
        
        self.km_becomeFirstResponder()
    }
    
    override func redo(_ sender: Any?) {
        super.redo(sender)
        
        self.km_becomeFirstResponder()
    }
}

// MARK: - KMExtensions

extension KMPDFEditViewController {
    override func deleteAfter(indexpaths: Set<IndexPath>) {
        self.km_becomeFirstResponder()
    }
    override func copyAfter(indexpaths: Set<IndexPath>) {
        self.km_becomeFirstResponder()
    }
    override func pasteAfter(at index: Int) {
        self.km_becomeFirstResponder()
    }
    override func cutAfter(indexpaths: Set<IndexPath>) {
        self.km_becomeFirstResponder()
    }
    override func leftRotateAfter(indexpaths: Set<IndexPath>) {
        self.km_becomeFirstResponder()
    }
    override func rightRotateAfter(indexpaths: Set<IndexPath>) {
        self.km_becomeFirstResponder()
    }
    
    func listViewCurrentPageUpdateForDrag(indexs: IndexSet, toIndex: Int) {
        // listView 当前页面 被拖拽
        if (indexs.contains(self._listViewCurrentIndex)) {
//            let idx = indexs.firstIndex(of: self._listViewCurrentIndex)
            var idx: Int = 0
            for index in indexs {
                if (index == self._listViewCurrentIndex) {
                    break
                }
                idx += 1
            }
            // 插入位置偏移量
            var toIndexOffset: Int = 0
            // 从后面往前删除
            for index in indexs.reversed() {
                let newPage = (self.listView?.document.page(at: UInt(index)))?.copy() as? CPDFPage
                if (newPage == nil) {
                    continue
                }

                if (index < toIndex) { /// 删除了插入位置前面的页面，需要改变 toIndex
                    toIndexOffset += 1
                }
            }
            if (indexs.last! < toIndex) { // 往后拖拽
                toIndexOffset -= 1
            }
            let insertIndex = max(0, toIndex-toIndexOffset)
            self._listViewCurrentIndex = insertIndex+idx
        }
    }
}