PDF_Office/PDF Office/PDF Master/KMClass/KMPDFViewController/RightSideViews/Watermark/New/KMWatermarkController.swift

//
//  KMWatermarkController.swift
//  PDF Reader Pro
//
//  Created by Niehaoyu on 2024/11/1.
//

import Cocoa
import KMComponentLibrary

//工具类型
@objc public enum KMWaterMarkModeType: Int, CaseIterable {
    case Template = 0
    case AddWatermark
    case EditWatermark
}

@objc protocol KMWatermarkControllerDelegate: AnyObject {
    
    @objc optional func kmWatermarkControllerDidUpdateMode(_ view: KMWatermarkController)
    
}

class KMWatermarkController: NSViewController {
    
    @IBOutlet var contendView: NSView!
    
    @IBOutlet var preInfoView: NSView!
    @IBOutlet var propertyInfoView: NSView!
    
    var documentPreview: KMEditDocumentController = KMEditDocumentController.init()
    var propertyController: KMNWatermarkPropertyController = KMNWatermarkPropertyController.init()
    var templatesController: KMNWatermarkTemplateController = KMNWatermarkTemplateController()
    
    var saveTemplateMessage: ComponentMessage = ComponentMessage()
    
    var pdfDocument: CPDFDocument?
    
    var currentWatermark: CPDFWatermark?
    
    weak open var delegate: KMWatermarkControllerDelegate?
    
    public var watermarkMode: KMWaterMarkModeType = .AddWatermark {
        didSet {
            resetUI()
        }
    }
    
    //MARK: - func
    override func viewDidLoad() {
        super.viewDidLoad()
        // Do view setup here.
        
        documentPreview.view.frame = preInfoView.bounds
        documentPreview.view.autoresizingMask = [.width, .height]
        preInfoView.addSubview(documentPreview.view)
        
        propertyController.view.frame = propertyInfoView.bounds
        propertyController.view.autoresizingMask = [.width, .height]
        propertyController.delegate = self
        propertyController.view.isHidden = true
        propertyInfoView.addSubview(propertyController.view)
        
        templatesController.view.frame = propertyInfoView.bounds
        templatesController.view.autoresizingMask = [.width, .height]
        templatesController.delegate = self
        templatesController.view.isHidden = true
        propertyInfoView.addSubview(templatesController.view)
        
        setupProperty()
    }
    
    func setupProperty() {
        saveTemplateMessage.properties = ComponentMessageProperty(messageType: .success, title: KMLocalizedString("Saved Template"))
        
        
    }
    
    func resetUI() {
        propertyController.view.isHidden = true
        templatesController.view.isHidden = true
        
        if watermarkMode == .Template {
            templatesController.view.isHidden = false
            
        } else if watermarkMode == .AddWatermark {
            propertyController.view.isHidden = false
            propertyController.watermarkMode = watermarkMode
            
        } else if watermarkMode == .EditWatermark {
            propertyController.view.isHidden = false
            propertyController.watermarkMode = watermarkMode
        }
    }
    
    
    func reloadData() {
        
        documentPreview.pdfDocument = self.pdfDocument
        documentPreview.reloadData()
        
    }
    
    func removeWatermark(_ watermark: CPDFWatermark?) {
        if let tempData = watermark {
            let document = documentPreview.pdfDocument
            document?.removeWatermark(tempData)
            documentPreview.refreshUI()
        }
        
    }
    
    func waterMarkDataDidUpdated(_ model: KMPDFWatermarkData) {
        let document = documentPreview.pdfDocument
        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.scale = 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
        
        if let modelMark = model.watermark {
            document?.removeWatermark(modelMark)
        }
        
        document?.addWatermark(watermark)
        documentPreview.refreshUI()
        
        model.watermark = watermark
        
        currentWatermark = watermark
        
    }
    
}

//MARK: - KMNWatermarkPropertyControllerDelegate
extension KMWatermarkController: KMNWatermarkPropertyControllerDelegate {
    
    func watermarkPropertyControllerDidUpdate(_ controller: KMNWatermarkPropertyController) {
        removeWatermark(currentWatermark)
        
        waterMarkDataDidUpdated(controller.watermarkData)
    }
    
    func watermarkPropertyControllerDidChangetoTemplate(_ controller: KMNWatermarkPropertyController) {
        watermarkMode = .Template
        
        removeWatermark(currentWatermark)
        
    }
    
    func watermarkPropertyControllerSaveTemplateSuccess(_ controller: KMNWatermarkPropertyController, _ data: KMPDFWatermarkData) {
        saveTemplateMessage.frame = CGRectMake((CGRectGetWidth(self.view.frame) - saveTemplateMessage.properties.propertyInfo.viewWidth)/2,
                                               CGRectGetHeight(self.view.frame) - saveTemplateMessage.properties.propertyInfo.viewHeight - 8,
                                               saveTemplateMessage.properties.propertyInfo.viewWidth,
                                               saveTemplateMessage.properties.propertyInfo.viewHeight)
        saveTemplateMessage.reloadData()
        saveTemplateMessage.show(inView: self.view, autoHideSeconde: 2)
        
        watermarkMode = .Template
        delegate?.kmWatermarkControllerDidUpdateMode?(self)
        
        templatesController.selectedWatermark = data
        templatesController.reloadData()
    }
    
    func watermarkPropertyControllerCancelTemplateEdit(_ controller: KMNWatermarkPropertyController) {
        watermarkMode = .Template
        
        templatesController.reloadData()
        
    }
    
}

extension KMWatermarkController: KMNWatermarkTemplateControllerDelegate {
    func templateControllerDidAddWatermarkData(_ controller: KMNWatermarkTemplateController) {
        watermarkMode = .AddWatermark

        removeWatermark(currentWatermark)

        waterMarkDataDidUpdated(propertyController.watermarkData)
    }
    
    func templateControllerDidEditWatermarkData(_ controller: KMNWatermarkTemplateController, _ watermark: KMPDFWatermarkData) {
        watermarkMode = .EditWatermark
        
        propertyController.watermarkData = watermark
        propertyController.reloadData()
        waterMarkDataDidUpdated(watermark)
        
    }
    
    func templateControllerDidSelectedChanged(_ controller: KMNWatermarkTemplateController) {
        removeWatermark(currentWatermark)
        
        if let data = controller.selectedWatermark {
            waterMarkDataDidUpdated(data)
        } else {
            currentWatermark = nil
        }
    }
}