PDF_Office/PDF Office/PDF Master/KMClass/NewBatch/View/KMBatchProcessingView/KMBatchProcessingView.swift

//
//  KMBatchProcessingView.swift
//  PDF Master
//
//  Created by lizhe on 2022/11/16.
//

import Cocoa
import KMComponentLibrary

class KMBatchProcessingView: BaseXibView {
    @IBOutlet weak var tableView: KMBatchProcessingTableView!
    @IBOutlet weak var selectedFilesView: KMBatchSelectedFilesView!
//    @IBOutlet weak var addFilesButton: ComponentDropdownTool!
    @IBOutlet weak var deleteButton: ComponentButton!
    @IBOutlet weak var addFilesButton: ComponentButton!
    @IBOutlet weak var addFilesMoreButton: ComponentButton!
    @IBOutlet var addFilesSepline: ComponentDivider!
    @IBOutlet weak var lineView: NSView!
    
    var groupView: ComponentGroup?
    
    weak var delegate: KMBatchProcessingViewDelegate?
    var inputType: KMBatchCollectionViewType = .convertPDF {
        didSet {
            self.selectedFilesView.inputType = inputType
            self.tableView.inputType = inputType
        }
    }
    
    /**
      @abstract 外部传入数据
      @param inputData 文件路劲
      */
    var inputData: [URL]! {
        didSet {
            self.tableView.inputData = inputData
            self.reloadData()
        }
    }
    
    var inputDataModel: [KMBatchProcessingTableViewModel]! {
        didSet {
            self.tableView.presenter.addDataModel(datas: inputDataModel)
            self.reloadData()
        }
    }
    
    var isDisable: Bool = false {
        didSet {
            self.tableView.isDisable = self.isDisable
            self.deleteButton.properties.isDisabled = self.isDisable
            self.deleteButton.reloadData()
        }
    }
    
    override func draw(_ dirtyRect: NSRect) {
        super.draw(dirtyRect)

        // Drawing code here.
    }
    
    deinit {
        NotificationCenter.default.removeObserver(self)
    }
    
    override func setup() {
        
        self.backgroundColor(ComponentLibrary.shared.getComponentColorFromKey("colorBg/layout-low"))
        self.border(ComponentLibrary.shared.getComponentColorFromKey("colorBorder/divider"), 0.5, 0)
        self.lineView.backgroundColor(ComponentLibrary.shared.getComponentColorFromKey("colorBorder/divider"))
        
        addFilesSepline.properties = ComponentDividerProperty(type: .vertical)
        
        let properties = ComponentButtonProperty()
        properties.type = .gray
        properties.size = .s
        properties.buttonText = KMLocalizedString("Add Files")
        addFilesButton.properties = properties
        properties.propertyInfo.cornerRadius_topRight = 0
        properties.propertyInfo.cornerRadius_bottomRight = 0
        addFilesButton.reloadData()
        addFilesButton.setTarget(self, action: #selector(addFileAction))
        
        let properties2 = ComponentButtonProperty()
        properties2.type = .gray
        properties2.size = .s
        properties2.onlyIcon = true
        properties2.icon = NSImage(named: "KMImageNameImageToPDFDropdown")
        addFilesMoreButton.properties = properties2
        properties2.propertyInfo.cornerRadius_topLeft = 0
        properties2.propertyInfo.cornerRadius_bottomLeft = 0
        addFilesMoreButton.reloadData()
        addFilesMoreButton.setTarget(self, action: #selector(addFileAction))
        
        
        
        self.deleteButton.properties = ComponentButtonProperty(type: .default_tertiary,
                                                               size: .s,
                                                               state: .normal,
                                                               onlyIcon: true,
                                                               icon:NSImage(named: "icon_btn_clear_norm_false"),
                                                               keepPressState: false)
        self.deleteButton.setTarget(self, action: #selector(cleanAll(_:)))
        self.deleteButton.reloadData()
        
        
        self.selectedFilesView.dragEnd = {[unowned self] (data) in
            self.inputData = data
        }
        
        self.selectedFilesView.addAction = {[unowned self] (sender) in
            self.addFiles()
        }
        
        self.tableView.delegate = self
        self.reloadData()
        
        NotificationCenter.default.addObserver(self, selector: #selector(bacthFilesProcessingNotification), name: NSNotification.Name(kBacthFilesProcessNotification), object: nil)
    }
    
    func reloadData() {
        if self.tableView.data.count != 0 {
            self.deleteButton.properties.isDisabled = false
            self.selectedFilesView.isHidden = true
            self.tableView.isHidden = false
            self.lineView.isHidden = false
        } else {
            self.selectedFilesView.isHidden = false
            self.tableView.isHidden = true
            self.lineView.isHidden = true
            self.deleteButton.properties.isDisabled = true
        }
        self.deleteButton.reloadData()
        
        self.tableView.reloadData()
    }
    
    func showGroupView() {
        
        var viewHeight: CGFloat = 8
        var menuItemArr: [ComponentMenuitemProperty] = []
        
        var items: [String] = []
        if self.inputType == .imageToPDF {
            items = ["Add Files", "Add Folder", "Import From Clipboard", "Import From Scanner"]
        } else {
            items = ["Add Files", "Add Folder"]
        }
        
        for i in items {
            let properties_Menuitem: ComponentMenuitemProperty = ComponentMenuitemProperty(multipleSelect: false,
                                                                                               itemSelected: false,
                                                                                               isDisabled: false,
                                                                                               keyEquivalent: nil,
                                                                                               text: KMLocalizedString(i))
            menuItemArr.append(properties_Menuitem)
            viewHeight += 36
        }
        
        if groupView == nil {
            groupView = ComponentGroup.createFromNib(in: ComponentLibrary.shared.componentBundle())
        }
        groupView?.groupDelegate = self
        groupView?.frame = CGRectMake(310, 0, 200, viewHeight)
        groupView?.updateGroupInfo(menuItemArr)
        
        var point = addFilesButton.convert(addFilesButton.frame.origin, to: nil)
        point.y -= viewHeight
        groupView?.showWithPoint(point, relativeTo: addFilesButton)
        
//        addFileButton_?.properties.state = .pressed
//        addFileButton_?.reloadData()
    }
}


protocol KMBatchProcessingViewAction{}
extension KMBatchProcessingView: KMBatchProcessingViewAction {
    
    @IBAction func cleanAll(_ sender: Any) {
        self.fileDeleteAction()
    }
    
    func _cleanAll() {
        if self.tableView.selectModels.count != 0 {
            self.tableView.presenter.deleteData(models: self.tableView.selectModels)
            self.tableView.selectModels.removeAll()
            
        } else {
            self.tableView.presenter.cleanAll()
            self.tableView.selectModels.removeAll()
        }
        
        self.reloadData()
    }
    
    func fileDeleteAction() {
        if UserDefaults.standard.object(forKey: "kKMBatchProcessingView") != nil {
            self._cleanAll()
            return
        }
        let alert = NSAlert()
        alert.alertStyle = .informational
        if HistoryFilesManager.manager.selectFiles.count > 0 {
            alert.informativeText = KMLocalizedString("Remove Selected Files from your Recent Files?")
        } else {
            alert.informativeText = KMLocalizedString("Remove Files from your Recent Files?")
        }
        alert.messageText = KMLocalizedString("The file will disappear from the recent list.")
        alert.addButton(withTitle: KMLocalizedString("Delete"))
        alert.addButton(withTitle: KMLocalizedString("Cancel"))
        alert.showsSuppressionButton = true
        let response = alert.runModal()
        if response.rawValue == 1000 {
            self._cleanAll()
            
            if alert.suppressionButton?.state == .on {
                UserDefaults.standard.set("YES", forKey: "kKMBatchProcessingView")
                UserDefaults.standard.synchronize()
            }
        }
    }
    
    func addFileAction() {
        self.showGroupView()
    }
    
    func addFiles() {
        KMBatchProcessingView.openfiles(window: window!) { [unowned self] openPanel in
            if self.inputType == .OCR {
                openPanel.allowedFileTypes = KMOCRModel.supportedTypes()
            } else if self.inputType == .imageToPDF {
                openPanel.allowedFileTypes = KMBatchProcessingTableViewModel.supportedImageTypes()
            } else {
                openPanel.allowedFileTypes = ["pdf","PDF"]
            }
            
            openPanel.prompt = "Open"
            openPanel.canChooseDirectories = false //是否允许选择目录
            openPanel.canChooseFiles = true //是否可以选择文件
            openPanel.allowsMultipleSelection = true //是否允许多选
            openPanel.message = KMLocalizedString("Press the keyboard ⌘/⇧ to select multiple files", comment: "")
        } completion: { [unowned self] (panel ,data) in
            if data.count != 0 {
                self.tableView.presenter.addData(data: data)
                self.reloadData()
            }
        }
    }
    
    func addFolder() {
        KMPrint("addFolder")
        KMBatchProcessingView.openfiles(window: window!) { [unowned self] openPanel in
            if self.inputType == .OCR {
                openPanel.allowedFileTypes = KMOCRModel.supportedTypes()
            } else if self.inputType == .imageToPDF {
                openPanel.allowedFileTypes = KMBatchProcessingTableViewModel.supportedImageTypes()
            } else {
                openPanel.allowedFileTypes = ["pdf","PDF"]
            }
            
            openPanel.prompt = "Open"
            openPanel.canChooseDirectories = true //是否允许选择目录
            openPanel.canChooseFiles = false //是否可以选择文件
            openPanel.allowsMultipleSelection = false //是否允许多选
        } completion: { [unowned self] (panel ,data) in
            if data.count != 0 {
                var filePaths: [URL] = []
                
                var types: [String] = []
                if self.inputType == .OCR {
                    types = KMOCRModel.supportedTypes()
                } else if self.inputType == .imageToPDF {
                    types = KMBatchProcessingTableViewModel.supportedImageTypes()
                } else {
                    types = ["pdf","PDF"]
                }
                KMMergeTools.findAllFiles(folder: data.first!, result: &filePaths, types: types)
                self.tableView.presenter.addData(data: filePaths)
                self.reloadData()
            }
        }
    }
    
    func addOpenFiles() {
//        let array = ImageToPDFTools.fetchOpenDocumentFiles()
//        self.tableView.presenter.addData(data: array)
//        self.reloadData()
    }
    
    func canAddOpenFile() -> Bool {
        var result: Bool = false
//        let array = ImageToPDFTools.fetchOpenDocumentFiles()
//        if array.count != 0 {
//            result = true
//        }
        return result
    }
    
    func importFromClipboard() {
        let path = self.saveImageFromClipboardToTempDirectory()
        self.tableView.presenter.addData(data: [URL(fileURLWithPath: path ?? "")])
    }
    
    func saveImageFromClipboardToTempDirectory() -> String? {
        // 1. 优先获取 PNG 格式数据
        guard let imageData = getImageDataFromClipboard() else {
            print("剪切板中没有可识别的图片数据")
            return nil
        }
        
        // 2. 创建临时文件路径
        let tempDir = URL(fileURLWithPath: NSTemporaryDirectory(), isDirectory: true)
        let fileName = "clipboard_image_\(UUID().uuidString).png"
        let fileURL = tempDir.appendingPathComponent(fileName)
        
        do {
            // 3. 直接写入原始数据（避免二次编码）
            try imageData.write(to: fileURL)
            print("图片已保存到: \(fileURL.path)")
            
            // 4. 验证文件完整性
            guard validateImageFile(at: fileURL) else {
                print("文件完整性验证失败")
                try? FileManager.default.removeItem(at: fileURL)
                return nil
            }
            
            return fileURL.path
        } catch {
            print("保存失败: \(error.localizedDescription)")
            return nil
        }
    }

    // MARK: - 剪切板数据处理
    private func getImageDataFromClipboard() -> Data? {
        let pasteboard = NSPasteboard.general
        
        // 优先级检查顺序：PNG > TIFF > FileURL
        if let pngData = pasteboard.data(forType: .png) {
            print("检测到PNG格式数据")
            return pngData
        }
        
        if let tiffData = pasteboard.data(forType: .tiff),
           let bitmapRep = NSBitmapImageRep(data: tiffData),
           let pngData = bitmapRep.representation(using: .png, properties: [:]) {
            print("转换TIFF到PNG格式")
            return pngData
        }
        
        if let fileURL = pasteboard.readObjects(forClasses: [NSURL.self])?.first as? URL,
           fileURL.pathExtension.lowercased() == "png",
           let data = try? Data(contentsOf: fileURL) {
            print("从文件URL读取PNG数据")
            return data
        }
        
        return nil
    }

    // MARK: - 文件验证
    private func validateImageFile(at url: URL) -> Bool {
        // 验证文件大小
        guard let attrs = try? FileManager.default.attributesOfItem(atPath: url.path),
              let fileSize = attrs[.size] as? Int64,
              fileSize > 256 else { // 有效PNG文件至少256字节
            return false
        }
        
        // 验证文件头签名
        guard let fileHandle = try? FileHandle(forReadingFrom: url),
              let data = try? fileHandle.read(upToCount: 8) else {
            return false
        }
        
        // PNG文件头签名: 89 50 4E 47 0D 0A 1A 0A
        let pngSignature: [UInt8] = [0x89, 0x50, 0x4E, 0x47, 0x0D, 0x0A, 0x1A, 0x0A]
        return data.elementsEqual(pngSignature)
    }
    
    func importFromScanner() {
        let vc = KMDeviceBrowserWindowController.shared
        vc.type = .scanner
        vc.importScannerFileCallback = { [weak self] (url: NSURL) -> Void in
            let data: [URL] = [url as URL]
            self?.tableView.presenter.addData(data: data)
            self?.reloadData()
        }
        vc.showWindowInCenter(nil)
    }
}

protocol KMBatchProcessingViewOpenPanel {}
extension KMBatchProcessingView: KMBatchProcessingViewOpenPanel {
    /**
      @abstract 打开文件
      @param window 显示window
      @param title 标题
      */
    static func openfiles(window:NSWindow, openPanel:((_ openPanel: NSOpenPanel) -> Void)?, completion:@escaping/**逃逸闭包*/ (_ panel: NSOpenPanel ,_ data:[URL]) -> Void) {
        let panel = NSOpenPanel()
        if openPanel != nil {
            openPanel!(panel)
        } else {
            panel.prompt = ""
            panel.canChooseDirectories = false //是否允许选择目录
            panel.canChooseFiles = true //是否可以选择文件
            panel.allowsMultipleSelection = true //是否允许多选
            panel.allowedFileTypes = KMBatchProcessingTableViewModel.supportedImageTypes() //支持类型
            
//            let button = NSButton.init(checkboxWithTitle: "保存后打开文档", target: nil, action: nil)
//            button.state = .off
//            panel.accessoryView = button
        }
        
        panel.beginSheetModal(for: window, completionHandler: { result in
            if result != .OK {
                KMPrint("取消")
                completion(panel, [])
            } else {
                completion(panel, panel.urls)
            }
        })
    }
}

extension KMBatchProcessingView: KMBatchProcessingTableViewDelegate {
    func reloadData(data: [KMBatchProcessingTableViewModel]) {
        self.reloadData()
        if (self.delegate != nil) {
            self.delegate?.reloadData(data: data)
        }
    }
}

protocol KMBatchProcessingViewDelegate: NSObject {
    /**
     数据刷新
     */
    func reloadData(data: [KMBatchProcessingTableViewModel])
}

extension KMBatchProcessingView: ComponentGroupDelegate {
    func componentGroupDidSelect(group: ComponentGroup?, menuItemProperty: ComponentMenuitemProperty?) {
        let familyString = menuItemProperty?.text
        
        if familyString == KMLocalizedString("Add Files", comment: "") {
            self.addFiles()
        } else if familyString == KMLocalizedString("Add Folder", comment: "") {
            self.addFolder()
        } else if familyString == KMLocalizedString("Add Open Files", comment: "") {
            self.addOpenFiles()
        } else if familyString == KMLocalizedString("Import From Clipboard", comment: "") {
            self.importFromClipboard()
        } else if familyString == KMLocalizedString("Import From Scanner", comment: "") {
            self.importFromScanner()
        }
    }
    
    func componentGroupDidDismiss(group: ComponentGroup?) {
        self.addFilesButton.properties.state = .normal
        self.addFilesMoreButton.properties.state = .normal
        
        self.addFilesButton.reloadData()
        self.addFilesMoreButton.reloadData()
    }
}

extension KMBatchProcessingView {
    @objc func bacthFilesProcessingNotification() {
        self.tableView.reloadData()
    }
}