PDF_Office/PDF Office/PDF Master/Class/PDFWindowController/Side/LeftSide/KMLeftSideViewController+Snapshot.swift

//
//  KMLeftSideViewController+Snapshot.swift
//  PDF Master
//
//  Created by tangchao on 2024/1/12.
//

import Foundation

extension KMLeftSideViewController.Key {
    static let snapshotSizeScaling = "KMSnapshotSizeScalingKey"
}

// MARK: - Actions

extension KMLeftSideViewController {
    @objc func menuItemClick_ExportPNG(_ sender: AnyObject?) {
        guard let snapshot = (sender as? NSMenuItem)?.representedObject as? KMSnapshotWindowController else {
            return
        }

        let image = snapshot.thumbnailWithSize(0)
        NSPanel.savePanel_data_success(self.view.window!, imageData: image?.pngData(), allowedTypes: ["png"]) { url in
            NSWorkspace.shared.selectFile(url.path, inFileViewerRootedAtPath: "")
        }
    }
    
    @objc func menuItemClick_ExportJPG(_ sender: AnyObject?) {
        guard let snapshot = (sender as? NSMenuItem)?.representedObject as? KMSnapshotWindowController else {
            return
        }

        let image = snapshot.thumbnailWithSize(0)
        NSPanel.savePanel_data_success(self.view.window!, imageData: image?.jpgData(), allowedTypes: ["jpg"]) { url in
            NSWorkspace.shared.selectFile(url.path, inFileViewerRootedAtPath: "")
        }
    }
    
    @objc func menuItemClick_ExportPDF(_ sender: AnyObject?) {
        guard let snapshot = (sender as? NSMenuItem)?.representedObject as? KMSnapshotWindowController else {
            return
        }

        if let image = snapshot.thumbnailWithSize(0) {
            let document = CPDFDocument()
            _ = document?.km_insert(image: image, at: 0)
            NSPanel.savePanel_pdf_success(self.view.window!, document: document) { url in
                NSWorkspace.shared.selectFile(url.path, inFileViewerRootedAtPath: "")
            }
        }
    }
    
    @objc func menuItemClick_Print(_ sender: AnyObject?) {
        guard let snapshot = (sender as? NSMenuItem)?.representedObject as? KMSnapshotWindowController else {
            return
        }
        
        if let image = snapshot.thumbnailWithSize(0) {
            self.delegate?.controller?(controller: self, itemClick: nil, itemKey: .print, params: image)
        }
    }
    
    @objc func menuItemClick_SelectAll(_ sender: AnyObject?) {
        let selected = self.snapshotListIsAllSelected()
        for model in self.snapshots {
            model.isSelected = !selected
        }
        Task { @MainActor in
            self.snapshotTableView.reloadData()
        }
    }
    
    @objc func deleteAllSnapshot(_ sender: AnyObject?) {
        for model in self.snapshots {
            model.windowC?.close()
        }
    }
    
    @objc func hideSnapshot(_ sender: AnyObject?) {
        guard let snapshot = (sender as? NSMenuItem)?.representedObject as? KMSnapshotWindowController else {
            return
        }
        if let data = snapshot.window?.isVisible, data {
            snapshot.miniaturize()
        }
    }
    
    @objc func showSnapshot(_ sender: AnyObject?) {
        guard let snapshot = (sender as? NSMenuItem)?.representedObject as? KMSnapshotWindowController else {
            return
        }
        if let data = snapshot.window?.isVisible, data {
            snapshot.window?.orderFront(nil)
        } else {
            snapshot.deminiaturize()
        }
    }
    
    @objc func deleteSnapshot(_ sender: AnyObject?) {
        guard let snapshot = (sender as? NSMenuItem)?.representedObject as? KMSnapshotWindowController else {
            return
        }
        snapshot.close()
    }
    
    @objc func menuItemClick_Copy(_ sender: AnyObject?) {
        guard let snapshot = (sender as? NSMenuItem)?.representedObject as? KMSnapshotWindowController else {
            return
        }
        let image = snapshot.thumbnailWithSize(0)
        if let tiffData = image?.tiffRepresentation {
            let pasteboardItem = NSPasteboardItem()
            pasteboardItem.setData(tiffData, forType: .tiff)
            let pboard = NSPasteboard.general
            pboard.clearContents()
            pboard.writeObjects([pasteboardItem])
        }
    }
    
    func snapshot_initSubViews() {
        self.snapshotController?.snapshotNormalZoomInButton.target = self
        self.snapshotController?.snapshotNormalZoomInButton.tag = 2
        self.snapshotController?.snapshotNormalZoomInButton.action = #selector(thumbnailSizeScaling)
        self.snapshotController?.snapshotNormalZoomOutButton.target = self
        self.snapshotController?.snapshotNormalZoomOutButton.tag = 3
        self.snapshotController?.snapshotNormalZoomOutButton.action = #selector(thumbnailSizeScaling)
  
        self.snapshotController?.snapshotNormalMoreButton.target = self
        self.snapshotController?.snapshotNormalMoreButton.tag = 300
        self.snapshotController?.snapshotNormalMoreButton.action = #selector(leftSideViewMoreButtonAction)
 
        self.snapshotController?.snapshotDoneButton.target = self
        self.snapshotController?.snapshotDoneButton.tag = 312
        self.snapshotController?.snapshotDoneButton.action = #selector(leftSideViewDoneButtonAction)
        
        self.snapshotTableView.delegate = self
        self.snapshotTableView.dataSource = self
        self.snapshotTableView.menu = NSMenu()
        self.snapshotTableView.menu?.delegate = self
        self.snapshotTableView.doubleAction = #selector(toggleSelectedSnapshots)
        self.snapshotTableView.target = self
        self.snapshotTableView.botaDelegate = self
        
        self.snapshotTableView.setDraggingSourceOperationMask(.every, forLocal: false)
        
        self.snapshotController?.snapshotSearchField.changeCallBack = { [weak self] changeContent in
            var editContent = ""
            if changeContent != nil {
                editContent = changeContent!
            }
            self?.searchFieldChangeAction(editContent)
        }
    }
    
    func snapshot_initDefalutValue() {
        
    }
    
    func updataLeftSideSnapView() {
        if (snapshots.count < 1) {
            self.snapshotController?.snapshotNormalSearchButton.isEnabled = false
            self.snapshotController?.snapshotNormalZoomOutButton.isEnabled = false
            self.snapshotController?.snapshotNormalZoomInButton.isEnabled = false
            self.snapshotController?.snapshotNormalMoreButton.isEnabled = false
            
            self.leftSideEmptyVC.deleteSnapBtn.isEnabled = false
            self.leftSideEmptyVC.exportSnapBtn.isEnabled = false
            self.leftSideEmptyVC.printSnapBtn.isEnabled = false
            self.snapshotTableView.usesAlternatingRowBackgroundColors = false

            self.snapshotController?.emptySnapView.isHidden = false
            
        } else {
            self.snapshotController?.snapshotNormalSearchButton.isEnabled = true
            self.snapshotController?.snapshotNormalZoomOutButton.isEnabled = true
            self.snapshotController?.snapshotNormalZoomInButton.isEnabled = true
            self.snapshotController?.snapshotNormalMoreButton.isEnabled = true

            self.snapshotController?.emptySnapView.isHidden = true
            
            self.leftSideEmptyVC.deleteSnapBtn.isEnabled = true
            self.leftSideEmptyVC.exportSnapBtn.isEnabled = true
            self.leftSideEmptyVC.printSnapBtn.isEnabled = true
        }
    }
    
    @IBAction func snapshotNormalSearchButtonAction(_ sender: NSButton) {
        self.snapshotController?.snapshotSearchField.isHidden = false
        self.snapshotController?.snapshotNormalSearchButton.isHidden = true
        self.snapshotController?.snapshotDoneButton.isHidden = false
        self.snapshotController?.snapshotLabel.isHidden = true
        self.snapshotController?.snapshotNormalZoomOutButton.isHidden = true
        self.snapshotController?.snapshotNormalZoomInButton.isHidden = true
        
        self.snapshotController?.snapshotSearchField.becomeFirstResponder()
    }
}

extension KMLeftSideViewController {
    public func refreshUIOfSnapshotIfNeed(preference: Bool = false) {
        if self.type.methodType != .snapshot {
            return
        }
        if preference {
            self.reloadSnapshotSize()
        }
        
        Task { @MainActor in
            self.snapshotTableView.reloadData()
        }
    }
    
    public func reloadSnapshotDataIfNeed() {
        if self.type.methodType != .snapshot {
            return
        }
        Task { @MainActor in
            self.snapshotTableView.reloadData()
        }
    }
    
    public func reloadSnapshotSize() {
        let defaultSize = roundf(KMPreference.shared.thumbSnapshotSize)
        var thumbnailSize = Self.kTinySize
        if defaultSize < Self.kTinySize + Self.kFudgeSize {
            
        } else {
            if defaultSize < Self.kSmallSize + Self.kFudgeSize {
                thumbnailSize = Self.kSmallSize
            } else {
                if defaultSize < Self.kLargeSize + Self.kFudgeSize {
                    thumbnailSize = Self.kLargeSize
                } else {
                    thumbnailSize = Self.kHugeSize
                }
            }
        }
        
        if (abs(thumbnailSize - Float(self.snapshotCacheSize)) > Self.kFudgeSize) {
            self.snapshotCacheSize = thumbnailSize.cgFloat
        }
        
        Task { @MainActor in
            self.snapshotTableView.reloadData()
        }
    }
    
    func updateSnapshotsIfNeed(for page: CPDFPage?) {
        if self.type.methodType != .snapshot {
            return
        }
        
        if page != nil {
            for model in self.snapshots {
                if let wc = model.windowC, wc.isPageVisible(page) {
                    self.snapshotNeedsUpdate(wc)
                    wc.redisplay()
                }
            }
        } else {
            for model in self.snapshots {
                model.windowC?.redisplay()
            }
            self.allSnapshotsNeedUpdate()
        }
    }
    
    func snapshotNeedsUpdate(_ dirtySnapshot: KMSnapshotWindowController) {
        if self.dirtySnapshots.contains(dirtySnapshot) == false {
            self.dirtySnapshots.append(dirtySnapshot)
            self.updateSnapshotsIfNeeded()
        }
    }
    
    func updateSnapshotsIfNeeded() {
//        if ([rightSideController.snapshotTableView window] != nil && [dirtySnapshots count] > 0 && snapshotTimer == nil)
//            snapshotTimer = [[NSTimer scheduledTimerWithTimeInterval:0.03 target:self selector:@selector(updateSnapshot:) userInfo:NULL repeats:YES] retain];
    }
    
    func allSnapshotsNeedUpdate() {
        self.dirtySnapshots.removeAll()
        for model in self.snapshots {
            if let winC = model.windowC {
                self.dirtySnapshots.append(winC)
            }
        }
        self.updateSnapshotsIfNeeded()
    }
}

extension KMLeftSideViewController: KMSnapshotWindowControllerDelegate {
    func snapshotControllerWillClose(_ controller: KMSnapshotWindowController) {
        var flag: KMSnapshotModel?
        for model in self.snapshots {
            if controller.isEqual(model.windowC) {
                flag = model
            }
        }
        if flag != nil {
            self.snapshots.removeObject(flag!)
            Task { @MainActor in
                self.snapshotTableView.reloadData()
                self.updataLeftSideSnapView()
            }
        }
    }
    
    func snapshotControllerDidFinishSetup(_ controller: KMSnapshotWindowController) {
        let image = controller.thumbnailWithSize(self.snapshotCacheSize)
        controller.thumbnail = image
        
        let model = KMSnapshotModel()
        model.windowC = controller
        self.snapshots.append(model)
        Task { @MainActor in
            self.snapshotTableView.ks_reloadData()
        }
    }
    
    func snapshotController(_ controller: KMSnapshotWindowController, miniaturizedRect isMiniaturize: Bool) -> NSRect {
        var row = NSNotFound
        for (i,sp) in self.snapshots.enumerated() {
            if controller.isEqual(sp.windowC) {
                row = i
            }
        }
        var rect: NSRect = .zero
        if (row != NSNotFound) {
            rect = self.snapshotTableView.frameOfCell(atColumn: 0, row: row)
        } else {
            rect.origin = KMBottomLeftPoint(self.snapshotTableView.visibleRect)
            rect.size.width = 1.0
            rect.size.height = 1.0
        }
        rect = self.snapshotTableView.convertRectToScreen(rect)
        return rect
    }
}