PDF_Office/PDF Office/PDF Master/KMClass/KMPDFViewController/EditTool/Watermark/Model/KMWatermarkModel.swift

//
//  KMWatermarkModel.swift
//  PDF Reader Pro
//
//  Created by tangchao on 2023/11/2.
//

import Cocoa
  
@objcMembers class KMWatermarkModel: NSObject {
    
    var watermarkType: CPDFWatermarkType = .text
    
    var text: String? = nil
    var fontName: String = ""
    var fontStyle: String? = ""
    var fontSize: CGFloat = 48
    var textColor: NSColor = NSColor.clear
    
    var imagePath:String? = nil
    
    var isScale: Bool = false
    var scale: CGFloat = 0.5
    
    var rotation: CGFloat = 0
    var opacity: CGFloat = 1
 
    var verticalPosition: CPDFWatermarkVerticalPosition = .center
    var horizontalPosition: CPDFWatermarkHorizontalPosition = .center
    
    var tx: CGFloat = 0
    var ty: CGFloat = 0
    
    var isFront: Bool = true
    var isTilePage: Bool = false
    
    var verticalSpacing: CGFloat = 60
    var horizontalSpacing: CGFloat = 60
    
    var tag: String = ""
    
    var watermarkName: String = ""
    
    var cPDFWatermark: CPDFWatermark?
    
    func modelPreviewImage() -> NSImage? {
        if let resultPath = KMWatermarkManager.defaultManager.watermarkImageFolder?.stringByAppendingPathComponent((self.tag + ".png")) {
            if let image = NSImage(contentsOfFile: resultPath) {
                return image
            }
        }
        return nil
    }
    
    func image() -> NSImage? {
        if let imagePath = self.imagePath {
            if let image = NSImage(contentsOfFile: imagePath) {
                return image
            }
        }
        return nil
    }
    
    //更新预览图
    func updatePreviewImage() {
        if let resultPath = KMWatermarkManager.defaultManager.watermarkImageFolder?.stringByAppendingPathComponent((self.tag + ".png")) {
            if let image = KMWatermarkManager.defaultManager.drawImageAtpageRect(rect: CGRectMake(0, 0, 136, 181), data: self) {   
                try?image.pngData()?.write(to: URL(fileURLWithPath: resultPath))
            }
        }
    }
    
    func isWatermarkModelChanged() -> Bool {
         
        let defaultModel = KMWatermarkManager.defaultManager.defaultWatermarkData
        
        if self.watermarkType == .text {
            if self.text != defaultModel.text {
                return true
            }
        } else if self.watermarkType == .image {
            if self.imagePath != nil {
                return true
            }
        }
        
        if self.fontName != defaultModel.fontName {
            return true
        }
        
        if self.fontStyle != defaultModel.fontStyle {
            return true
        }
        
        if self.fontSize != defaultModel.fontSize {
            return true
        }
          
        if self.textColor != defaultModel.textColor {
            return true
        }
        
        if self.isScale {
            return true
        }
         
        if self.scale != defaultModel.scale {
            return true
        }
        
        if self.rotation != defaultModel.rotation {
            return true
        }
        
        if self.opacity != defaultModel.opacity {
            return true
        }
         
        if self.verticalPosition != defaultModel.verticalPosition {
            return true
        }
         
        if self.horizontalPosition != defaultModel.horizontalPosition {
            return true
        }
         
        if self.tx != defaultModel.tx {
            return true
        }
         
        if self.ty != defaultModel.ty {
            return true
        }
         
        if self.isFront != defaultModel.isFront {
            return true
        }
        if self.isTilePage != defaultModel.isTilePage {
            return true
        }
        if self.verticalSpacing != defaultModel.verticalSpacing {
            return true
        }
        if self.horizontalSpacing != defaultModel.horizontalSpacing {
            return true
        }
        
        return false
    }
}

//MARK: - Class func
extension KMWatermarkModel {
    class func defaultWDData() -> KMWatermarkModel {
        let data = KMWatermarkModel()
        data.watermarkType = .text
        data.text = KMLocalizedString("PDF Reader Pro")
        
        data.fontName = "Helvetica"
        data.fontStyle = "Regular"
        data.fontSize = 48
        data.textColor = NSColor(red: 205/255, green: 205/255, blue: 205/255, alpha: 1)
        
        data.imagePath = nil
        
        data.isScale = false
        data.scale = 0.5
    
        data.rotation = 0
        data.opacity = 1
     
        data.verticalPosition = .center
        data.horizontalPosition = .center
        
        data.tx = 0
        data.ty = 0
        
        data.isFront = true
        data.isTilePage = false
        
        data.verticalSpacing = 60
        data.horizontalSpacing = 60
        
        data.tag = NSString.tagString()
        
        return data
    }
    
    class func returnWaterMarkWith(_ model: KMWatermarkModel, _ document: CPDFDocument) -> CPDFWatermark {
        var watermark = CPDFWatermark()
        
        if (model.watermarkType == .text) {
            watermark = CPDFWatermark(document: document, type: .text)
            watermark.text = model.text
            watermark.textColor = model.textColor
            watermark.cFont = CPDFFont(familyName: model.fontName, fontStyle: model.fontStyle ?? "Regular")
            if model.isScale {
                watermark.setAutoSizeWithPercentage(model.scale)
            } else {
                watermark.fontSize = model.fontSize
            }
        } else if model.watermarkType == .image {
            watermark = CPDFWatermark(document: document, type: .image)
            watermark.image = model.image()
            
            if model.isScale {
                watermark.setImageSizeWithPercentage(model.scale)
            } else {
                watermark.scale = 1
            }
        }
        
        watermark.opacity = model.opacity
        watermark.rotation = model.rotation
        
        watermark.tx = model.tx
        watermark.ty = model.ty
        
        watermark.isFront = model.isFront
        
        watermark.horizontalPosition = model.horizontalPosition
        watermark.verticalPosition = model.verticalPosition
        
        watermark.isTilePage = model.isTilePage
        watermark.horizontalSpacing = model.horizontalSpacing
        watermark.verticalSpacing = model.verticalSpacing
        
        return watermark
        
    }
    
}