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 [[.insert, .extract, .replace, .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)
    
    // 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 = toolBar
        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.insertFileAction()
                } else if (index == KMPageEditType.insert_blank_page) {
                    self.insertBlankPageAction()
                } 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.extractAction()
            } else if (type == .replace) {
                self.replaceAction()
            } 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 = { [unowned self] selectedIndexs in
            self.view.window?.makeFirstResponder(self)
            
            var enabled = (selectedIndexs.count > 0)
            for i in 0 ..< self.toolBar.toolBar.items.count {
                let item: KMToolBoxItem = self.toolBar.toolBar.items[i]
                if (item.itemIdentifier == KMToolbarPageEditExtractItemIdentifier ||
                    item.itemIdentifier == KMToolbarPageEditReverseItemIdentifier ||
                    item.itemIdentifier == KMToolbarPageEditDeleteItemIdentifier ||
                    item.itemIdentifier == KMToolbarPageEditLeftRotateItemIdentifier ||
                    item.itemIdentifier == KMToolbarPageEditRightRotateItemIdentifier ||
                    item.itemIdentifier == KMToolbarPageEditReplaceItemIdentifier) {
                    item.unEnabled = !enabled
                }
            }
            
            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 {
        let index = self.getPasteIndex()
        if (index != NSNotFound) {
            return index
        }
        return 0
    }
    
    private func getSelectedPage() -> CPDFPage {
        return (self.pdfDocument!.page(at: UInt(getSelecteIndex())))!
    }
    
    @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 insertPages(pages: Array<CPDFPage>, at indexs: IndexSet) {
        super.insertPages(pages: pages, at: indexs)
        
        self.selectPages(indexs: indexs)
        self.dealPdfDocumentDidEditCallback()
    }
    
    override func pasteAction() {
        if (self.canPaste()) {
            super.pasteAction()
            
            self.dealPdfDocumentDidEditCallback()
        }
    }
    
    override func insertBlankPage(size: NSSize, at index: Int) {
        super.insertBlankPage(size: size, at: index)
        
        self.selectPages(indexs: IndexSet(integer: index))
        self.dealPdfDocumentDidEditCallback()
    }
    
    override func getInsertIndex() -> Int {
        var index = 0
        if let pageCount = self.thumbnailView.document?.pageCount, pageCount > 0 {
            // 默认为最后一页
            index = Int(pageCount-1)
        }
        
        // 在当前选中item之后插入
        // 多选时，则取最后一页
        for indexpath in self.thumbnailView.selectionIndexPaths {
            index = indexpath.item
        }
        return index
    }
    
    // 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_insertCustomPage(sender: NSMenuItem?) {
        let windowController = KMPageEditInsertCustomPageWindowController()
        if (self.thumbnailView.selectionIndexPaths.count > 0) {
            windowController.selectedPageSize = self.getSelectedPage().bounds.size
        }
        
        self.km_beginSheet(windowC: windowController)
        windowController.itemClick = { [weak self] index in
            if (index == 1) { /// 取消
                self?.km_endSheet()
                return
            }
            
            /// 插入
            guard let windowC = self?.currentWindowC as? KMPageEditInsertCustomPageWindowController else {
                self?.km_endSheet()
                return
            }
            guard let _insertIndex = self?.getInsertIndex() else {
                self?.km_endSheet()
                return
            }
            
            /// 样式
//            let type = windowC.typeIndex
            let pageSize = windowC.pageSize
            let direction = windowC.direction
            
            /// 插入位置
//            if (type == 1) { /// 空白页
                let document = CPDFDocument()
                document?.insertPage(pageSize, at: 0)
                if let page: CPDFPage = (document?.page(at: 0)) {
                    if (direction == 0) { /// 纵向
                        page.rotation = 90
                    }

                    self?.insertPages(pages: [page], at: 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?.km_endSheet()
        }
    }
    
//    // 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.km_beginSheet(windowC: windowController)
        
        windowController.itemClick = { [weak self] index, value in
            if (index == 1) { /// 取消
                self?.km_endSheet()
                return
            }
            guard let _windowC = self?.currentWindowC as? KMPageEditSplitWindowController else {
                self?.km_endSheet()
                return
            }
            
            /// 拆分
            let outputModel = _windowC.model! as! KMPageEditSplitSettingModel
            self?.km_endSheet()
            
            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 alertView = CustomAlertView(message: NSLocalizedString("Please select at least two pages for editing first", comment: ""), from: self.thumbnailView, with: .blue)
            alertView?.layer?.backgroundColor = NSColor(hex: "#36383B").cgColor
            alertView?.layer?.cornerRadius = 4
            return
        }
        
        var indexs: IndexSet = []
        for indexpath in selectedIndexs {
            indexs.insert(indexpath.item)
        }
        
        self.reversePages(indexs: indexs)
    }
    
    override func addMenuItem(menu: NSMenu, item: KMPDFThumbViewMenu.Item) -> NSMenuItem? {
        let _ = super.addMenuItem(menu: menu, item: item)
        
        var item_: NSMenuItem?
        if (item.sel == KMPDFThumbViewMenu.insertSelector) {
            if (self.thumbnailView.selectionIndexPaths.count == 1) {
                item_ = menu.addItem(withTitle: NSLocalizedString("Insert", comment: ""), action: nil, target: self)
                
                let subMenu = NSMenu()
                subMenu.addItem(withTitle: NSLocalizedString("Insert File", comment: ""), action: #selector(insertPageItemAction), target: self, tag:0)
                subMenu.addItem(withTitle: NSLocalizedString("Insert a blank page", comment: ""), action: #selector(insertPageItemAction), target: self, tag:1)
                subMenu.addItem(withTitle: NSLocalizedString("Insert Custom Page", comment: ""), action: #selector(insertPageItemAction), target: self, tag:2)
                item_?.submenu = subMenu
            }
        }
        return item_
    }
}

extension KMPDFEditViewController {
    //MARK: menu Action
    
    @objc func insertPageItemAction(menu:NSMenuItem) {
        if (menu.tag == 0) {
            self.insertFileAction()
        } else if (menu.tag == 1) {
            self.insertBlankPageAction()
        } else if (menu.tag == 2) {
            self.item_insertCustomPage(sender: nil)
        }
    }
    
    @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) {
        self.km_add_pdf_multi(fileUrls: files) { [unowned self] documents in
            var insertIndex: Int = indexpath.item
            var pages: Array<CPDFPage> = []
            self.documentCaches.removeAll()
            var indexs = IndexSet()
            for document in documents {
                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)
        }
    }
    
    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>) {
        NotificationCenter.default.post(name: NSNotification.Name(rawValue: "KMPDFViewRotatePage"), object: self.thumbnailView.document)
        
        self.km_becomeFirstResponder()
    }
    override func rightRotateAfter(indexpaths: Set<IndexPath>) {
        NotificationCenter.default.post(name: NSNotification.Name(rawValue: "KMPDFViewRotatePage"), object: self.thumbnailView.document)
        
        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
        }
    }
}