PDF_Office/PDF Office/PDF Master/Class/PDFTools/Compress/Controller/KMCompressWindowController.swift

//
//  KMCompressWindowController.swift
//  PDF Master
//
//  Created by tangchao on 2022/11/25.
//

import Cocoa

typealias KMCompressWindowControllerItemClick = (Int)->()
typealias KMCompressWindowControllerResultCallback = (_ result: Bool,_ openDocuemt: Bool, _ fileURL: URL, _ error: String)->()
class KMCompressWindowController: NSWindowController {

    @IBOutlet weak var titleLabel: NSTextField!
    
    @IBOutlet weak var largeBox: NSBox!
    @IBOutlet weak var standardBox: NSBox!
    @IBOutlet weak var smallBox: NSBox!
    @IBOutlet weak var minimumBox: NSBox!
    
    @IBOutlet weak var batchButton: NSButton!
    @IBOutlet weak var cancelBox: NSBox!
    @IBOutlet weak var compressBox: NSBox!
    
    var documentURL: URL!
    var password: String!
    private var datas: [String] = []
    var selectIndex = 2
    
    var limit = true
    var oriDocumentUrl: URL?
    
    var itemClick       : KMCompressWindowControllerItemClick!
    var resultCallback  : KMCompressWindowControllerResultCallback!
    var largeVC         : KMDesignButton!
    var standardVC      : KMDesignButton!
    var smallVC         : KMDesignButton!
    var minimumVC       : KMDesignButton!
    var cancelVC        : KMDesignButton!
    var compressVC      : KMDesignButton!

    deinit {
        KMPrint("KMCompressWindowController 已释放")
    }
    
    override func windowDidLoad() {
        super.windowDidLoad()
        self.window?.appearance = NSAppearance(named: .aqua)
        largeVC = KMDesignButton.init(withType: .RadioButton)
        standardVC = KMDesignButton.init(withType: .RadioButton)
        smallVC = KMDesignButton.init(withType: .RadioButton)
        minimumVC = KMDesignButton.init(withType: .RadioButton)
        cancelVC = KMDesignButton.init(withType: .Text)
        compressVC = KMDesignButton.init(withType: .Text)
        largeBox.fillColor = .clear
        largeBox.contentView = largeVC.view
        standardBox.fillColor = .clear
        standardBox.contentView = standardVC.view
        smallBox.fillColor = .clear
        smallBox.contentView = smallVC.view
        minimumBox.fillColor = .clear
        minimumBox.contentView = minimumVC.view
        cancelBox.fillColor = .clear
        cancelBox.contentView = cancelVC.view
        compressBox.fillColor = .clear
        compressBox.contentView = compressVC.view
        
        titleLabel.stringValue = NSLocalizedString("Compress", comment: "")
        
        batchButton.isBordered = false
        batchButton.title = NSLocalizedString("Batch", comment: "")
        batchButton.wantsLayer = true
        batchButton.layer?.borderWidth = 1
        batchButton.layer?.borderColor = NSColor.black.cgColor
        batchButton.layer?.cornerRadius = 4
        batchButton.target = self
        batchButton.action = #selector(batchButtonAction)
        batchButton.isHidden = true
        
        cancelVC.target = self
        cancelVC.action = #selector(cancelButtonAction)
        cancelVC.stringValue = NSLocalizedString("Cancel", comment: "")
        cancelVC.button(type: .Sec, size: .m)
        cancelVC.button.keyEquivalent = KMKeyEquivalent.esc.string()
        
        compressVC.target = self
        compressVC.action = #selector(compressButtonAction)
        compressVC.stringValue = NSLocalizedString("Compress", comment: "")
        compressVC.button(type: .Cta, size: .m)
        compressVC.button.keyEquivalent = KMKeyEquivalent.enter
        
        largeVC.target = self
        largeVC.action = #selector(self.radioButtonAction(_:))
        largeVC.stringValue = NSLocalizedString("Large File Size", comment: "")
        largeVC.checkbox_radio()
        
        standardVC.target = self
        standardVC.action = #selector(self.radioButtonAction(_:))
        standardVC.stringValue = NSLocalizedString("Standard File Size", comment: "")
        standardVC.checkbox_radio()

        smallVC.target = self
        smallVC.action = #selector(self.radioButtonAction(_:))
        smallVC.stringValue = NSLocalizedString("Small File Size", comment: "")
        smallVC.checkbox_radio()

        minimumVC.target = self
        minimumVC.action = #selector(self.radioButtonAction(_:))
        minimumVC.stringValue = NSLocalizedString("Minimum File Size", comment: "")
        minimumVC.checkbox_radio()
        
        self.initUIProperty()
        
        smallVC.state = .Checked
        
        Task { @MainActor in
            if await (KMLightMemberManager.manager.canPayFunction() == false) {
                self.limit = true
            } else {
                self.limit = false
            }
        }
    }
    
    private func initUIProperty() {
        self.titleLabel.textColor = NSColor.titleColor()
        self.titleLabel.font = .SFProTextSemiboldFont(16)
        
        self.batchButton.setTitleColor(color: NSColor.buttonTitleColor())
        self.batchButton.layer?.borderColor = NSColor.buttonBorderColor().cgColor
        self.batchButton.font = .SFProTextRegularFont(14)
    }
    
    @objc func radioButtonAction(_ sender: NSButton) {
        largeVC.state = .Norm
        standardVC.state = .Norm
        smallVC.state = .Norm
        minimumVC.state = .Norm
        if (largeVC.button == sender) {
            largeVC.state = .Checked
            self.selectIndex = 0
        } else if (standardVC.button == sender) {
            standardVC.state = .Checked
            self.selectIndex = 1
        } else if (smallVC.button == sender) {
            smallVC.state = .Checked
            self.selectIndex = 2
        } else if (minimumVC.button == sender) {
            minimumVC.state = .Checked
            self.selectIndex = 3
        }
    }
    
    @objc func batchButtonAction() {
        guard let callback = itemClick else {
            return
        }
        
        callback(1)
    }
    
    @objc func cancelButtonAction() {
        self._clearData()
        
        guard let callback = itemClick else {
            return
        }
        
        callback(2)
    }
    
    @objc func compressButtonAction() {
        if documentURL == nil {
            return
        }
        
        Task { @MainActor in
            if await (KMLightMemberManager.manager.canPayFunction() == false) {
                let _ = KMSubscribeWaterMarkWindowController.show(window: self.window!) { isSubscribeSuccess, isWaterMarkExport, isClose in
                    if (isClose) {
                        return
                    }
                    if (isSubscribeSuccess) {
                        self._compressButtonAction()
                    }
                    if (isWaterMarkExport) {
                        self._compressButtonAction(limit: true)
                    }
                }
                return
            }
            self._compressButtonAction()
        }
    }
    
    func getUniqueFilePath(filePath: String) -> String {
        var i: Int = 0
        var isDirectory: ObjCBool = false
        var uniqueFilePath = filePath
        let fileManager = FileManager.default
        fileManager.fileExists(atPath: uniqueFilePath, isDirectory: &isDirectory)
        if isDirectory.boolValue {
            var path: String = ""
            while fileManager.fileExists(atPath: uniqueFilePath) {
                i += 1
                
                path = filePath
                path.append("(\(i))")
                uniqueFilePath = path
            }
        } else {
            let fileURL = URL(fileURLWithPath: filePath)
            var path: String = ""
            while fileManager.fileExists(atPath: uniqueFilePath) {
                i += 1
                
                path = fileURL.deletingPathExtension().path
                path.append("(\(i))")
                path.append(".")
                path.append(fileURL.pathExtension)
                uniqueFilePath = path
            }
        }
        return uniqueFilePath
    }
    
    private func _compressButtonAction(limit: Bool = false) {
        DispatchQueue.main.async {
            NSPanel.savePanel(self.window!, true) { panel in
                var url: URL = self.documentURL
                if (self.oriDocumentUrl != nil) {
                    url = self.oriDocumentUrl!
                }
                panel.nameFieldStringValue = ""+url.deletingPathExtension().lastPathComponent+"_Compressed"
                panel.allowedFileTypes = ["pdf"]
            } completion: { response, url, isOpen in
                if (response == .cancel) {
                    return
                }
                
                self.beginLoading()
                DispatchQueue.global().async {
                    let docuemt = CPDFDocument.init(url: self.documentURL)
                    if (docuemt?.isLocked)! && self.password != nil {
                        docuemt?.unlock(withPassword: self.password)
                    }
                    
                    var option = 60
                    if self.selectIndex == 0 {
                        option = 120
                    } else if self.selectIndex == 1 {
                        option = 90
                    } else if self.selectIndex == 2 {
                        option = 60
                    } else if self.selectIndex == 3 {
                        option = 30
                    }
                    //
                    
                    var result = false
                    if (limit) {
                        if let _document = docuemt, let _ = KMTools.saveWatermarkDocumentForCompress(document: _document, to: url!, imageQuality: option) {
                            result = true
                        }
                    } else {
                        if let data = docuemt?.writeOptimize(to: url, withOptions: [.imageQualityOption : option]) {
                            result = data
                        }
                    }
                    
                    if (result) {
                        self._clearData()
                    }
                    
                    DispatchQueue.main.async {
                        self.endLoading()
                        guard let callback = self.resultCallback else {
                            if (result) {
                                if isOpen { /// 开启文档
                                    NSDocumentController.shared.km_safe_openDocument(withContentsOf: url!, display: true) { _, _, _ in
                                        
                                    }
                                } else {
                                    NSWorkspace.shared.activateFileViewerSelecting([url!])
                                }
                            }
                            return
                        }
                        
                        callback(result, isOpen, url!, "")
                    }
                }
            }
        }
    }
    
    private func _clearData() {
        if let _ = self.oriDocumentUrl {
            if let data = self.documentURL?.path, FileManager.default.fileExists(atPath: data) {
                try?FileManager.default.removeItem(atPath: data)
            }
        }
    }
}

extension KMCompressWindowController: NSTableViewDataSource {
    func numberOfRows(in tableView: NSTableView) -> Int {
         datas.count
    }
    
}

extension KMCompressWindowController: KMLoadingProtocol {
    func beginLoading() {
        self.window?.contentView?.beginLoading()
    }
    
    func endLoading() {
        self.window?.contentView?.endLoading()
    }
}