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?
    
    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()
        
    }
    
    
}

//MARK: - KMNWatermarkPropertyControllerDelegate
extension KMWatermarkController: KMNWatermarkPropertyControllerDelegate {
    
    func watermarkPropertyControllerDidUpdate(_ controller: KMNWatermarkPropertyController) {
        let model = controller.watermarkData
        
        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")
            watermark.fontSize = model.fontSize
            
        } else if model.watermarkType == .image {
            watermark = CPDFWatermark(document: document, type: .image)
            watermark.image = model.image()
        }
        
        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
        
    }
    
    func watermarkPropertyControllerDidChangetoTemplate(_ controller: KMNWatermarkPropertyController) {
        watermarkMode = .Template
    }
    
    func watermarkPropertyControllerSaveTemplateSuccess(_ controller: KMNWatermarkPropertyController) {
        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.reloadData()
    }
    
    func watermarkPropertyControllerCancelTemplateEdit(_ controller: KMNWatermarkPropertyController) {
        watermarkMode = .Template
        
        templatesController.reloadData()

    }
    
}

extension KMWatermarkController: KMNWatermarkTemplateControllerDelegate {
    func templateControllerDidAddWatermarkData(_ controller: KMNWatermarkTemplateController) {
        watermarkMode = .AddWatermark
    }
    
    func templateControllerDidEditWatermarkData(_ controller: KMNWatermarkTemplateController, _ watermark: KMPDFWatermarkData) {
        watermarkMode = .EditWatermark
        
    }
}