PDF_Office/PDF Office/PDF Master/Class/PDFTools/Compress/Manager/KMCompressManager.swift

//
//  KMCompressManager.swift
//  PDF Reader Pro
//
//  Created by lizhe on 2024/12/17.
//

import Cocoa
import ComPDFKit

class KMCompressManager: NSObject {
    static let shared = KMCompressManager()
    
    func compress(documentURL: URL,
                  fileURL: URL,
                  password: String = "",
                  model: KMCompressSettingModel,
                  tProgressHandler: (@escaping (Float, Float) -> Void),
                  tCancelHandler: (@escaping () -> Bool),
                  tCompletionHandler: (@escaping (Bool) -> Void)) {
        DispatchQueue.global().async {
            let docuemt = CPDFDocument.init(url: documentURL)
            if (docuemt?.isLocked)! && password.count != 0 {
                docuemt?.unlock(withPassword: password)
            }
            
            self.compress(document: docuemt!, 
                          fileURL: fileURL,
                          model: model,
                          tProgressHandler: tProgressHandler,
                          tCancelHandler: tCancelHandler,
                          tCompletionHandler: tCompletionHandler)
        }
    }
    
    func compress(document: CPDFDocument,
                  fileURL: URL,
                  model: KMCompressSettingModel,
                  tProgressHandler: (@escaping (Float, Float) -> Void),
                  tCancelHandler: (@escaping () -> Bool),
                  tCompletionHandler: (@escaping (Bool) -> Void)) {
        let url = fileURL
        // 创建 imageOption 对象
        var imageOption = CPDFOptimizeImageOption()
        imageOption.uperPpi = Int32(model.ppi);
        imageOption.targetPpi = Int32(model.maxPpi);
        imageOption.compAlg = .CPDFCOMP_ALG_JPEG2000
        imageOption.blockSize = 256
        
        if model.imageQualityType == .hight {
            imageOption.quality = 100;
        } else if model.imageQualityType == .medium {
            imageOption.quality = 60;
        } else if model.imageQualityType == .low {
            imageOption.quality = 20;
        }
        
        // 创建 optimizeFlag
        var optimizeFlag: Int = 0
        model.objectOptions.contains(.formAndAction) ? optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMFORMCOMMITIMPORTRESETACTION.rawValue) : nil
        model.objectOptions.contains(.javaScript) ? optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMJSACTION.rawValue) : nil
        model.objectOptions.contains(.thumbnails) ? optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMPAGETHUMBNAIL.rawValue) : nil
        model.objectOptions.contains(.documentTags) ? optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMLABEL.rawValue) : nil
        model.objectOptions.contains(.bookmarks) ? optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMBK.rawValue) : nil
        
        if model.userDataOptions.contains(.commentsFormAndMultimedia) {
            optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMMULMEDIA.rawValue)
            optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMFORM.rawValue)
            optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMANNOT.rawValue)
        }
        
        if model.userDataOptions.contains(.documentInfomationAndMetadata) {
            optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMDOCINFO.rawValue)
            optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMMEDTADATA.rawValue)
        }
        
        model.userDataOptions.contains(.allObject) ? optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMOBJDATA.rawValue) : nil
        model.userDataOptions.contains(.fileAttachments) ? optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMFILEATTACHMENT.rawValue) : nil
        if model.userDataOptions.contains(.hiddenLayerContentAndFlattenVisibleLayers) {
            optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMHIDERLAYER.rawValue)
            optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_MERGEVISIBLELAYER.rawValue)
        }
        
        model.otherDataOptions.contains(.invalidBookmarks) ? optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMINVABK.rawValue) : nil
        model.otherDataOptions.contains(.invalidLinks) ? optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMINVALINK.rawValue) : nil
        model.otherDataOptions.contains(.unrefrencedNamedDestinations) ? optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMBK.rawValue) : nil
        model.otherDataOptions.contains(.pageContent) ? optimizeFlag |= Int(CPDF_OPTIMIZE_FLAG.CPDFOPTIMIZE_FLAG_RMUNUSEDTARGET.rawValue) : nil
        
        //数据转换
        var imageOptionValue: NSValue?
        // 使用 `withUnsafeBytes` 获取 UnsafeRawPointer
        withUnsafeBytes(of: imageOption) { rawBufferPointer in
            guard let baseAddress = rawBufferPointer.baseAddress else {
                return
            }
            // 获取结构体的 Objective-C 类型编码
            let objCType = KMCompressTool.encodeForStruct()
//            let objCType = "{CPDFOptimizeImageOption=iiiii}"
            imageOptionValue = NSValue(bytes: baseAddress, objCType: objCType)
        }
        
        // 创建字典
        let options: [String: Any] = [
            CPDFOptimizeCOLImageOptionKey: imageOptionValue as Any,
            CPDFOptimizeFlagKey: NSNumber(value: optimizeFlag)
        ]
        
        // 调用方法
        document.writeOptimize(to: url, withOptimizeOptions: options,
                               progressHandler: { (currentPage, totalPages) in
            // 进度处理
            print("当前页：\(currentPage), 总页数：\(totalPages)")
            tProgressHandler(currentPage, totalPages)
        },
                               cancelHandler: {
            // 返回是否取消
            let shouldCancel = false
            return shouldCancel
//            return tCancelHandler()
        },
                               completionHandler: { (finished) in
            // 完成处理
            if finished {
                print("优化完成")
            } else {
                print("优化未完成")
            }
            
            tCompletionHandler(finished)
        })
    }
}