PDF_Office/PDF Office/PDF Master/KMClass/KMPDFViewController/HeaderFooter/KMHeaderFooterController.swift

//
//  KMHeaderFooterController.swift
//  PDF Reader Pro
//
//  Created by Niehaoyu on 2024/11/8.
//

import Cocoa
import KMComponentLibrary

@objc protocol KMHeaderFooterControllerDelegate: AnyObject {
    
    @objc optional func kmHeaderFooterControllerDidUpdateModeType(_ view: KMHeaderFooterController)
    
    //数据发生变化时处理
    @objc optional func kmHeaderFooterControllerDidModelDataUpdated(_ view: KMHeaderFooterController)
    
    @objc optional func kmHeaderFooterControllerDidChooseApply(_ view: KMHeaderFooterController)
     
    @objc optional func kmHeaderFooterControllerDidChooseExit(_ view: KMHeaderFooterController)
}

class KMHeaderFooterController: NSViewController {
    
    @IBOutlet var contendView: NSView!
    
    @IBOutlet var preInfoView: NSView!
    @IBOutlet var propertyInfoView: NSView!
    
    var documentPreview: KMEditDocumentController = KMEditDocumentController.init()
    var propertyController: KMHeaderPropertyController = KMHeaderPropertyController.init()
    var templatesController: KMHFTemplateController = KMHFTemplateController()
    
    var saveTemplateMessage: ComponentMessage = ComponentMessage()
    
    var totalPDFCount: Int = 1
    var pdfDocument: CPDFDocument?
    
    var pdfViewCurrentIndex: Int = 0
    var validPageString: String?
     
    var headerFooterModel: KMHeaderFooterModel? {
        get {
            if editSubType == .template {
                return templatesController.selectedModel
            } else if editSubType == .add ||
                        editSubType == .edit {
                return propertyController.headerFooterModel
            }
            return nil
        }
    }
    
    weak open var delegate: KMHeaderFooterControllerDelegate?
    
    var editSubType: KMPDFEditSubModeType = .none
    
    //MARK: - func
    override func viewDidLoad() {
        super.viewDidLoad()
        // Do view setup here.
        
        documentPreview.view.frame = preInfoView.bounds
        documentPreview.view.autoresizingMask = [.width, .height]
        documentPreview.delegate = self
        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 applyEnabled() -> Bool {
        var applyEnable = true
        if let model = self.headerFooterModel {
            if model.topLeftString.count == 0 && model.topCenterString.count == 0 && model.topRightString.count == 0 &&
                model.bottomLeftString.count == 0 && model.bottomCenterString.count == 0 && model.bottomRightString.count == 0 {
                applyEnable = false
            }
        } else {
            applyEnable = false
        }
        return applyEnable
    }
    
    func setupProperty() {
        saveTemplateMessage.properties = ComponentMessageProperty(messageType: .success, title: KMLocalizedString("Saved Template"))
        
    }
    
    func resetUI() {
        propertyController.view.isHidden = true
        templatesController.view.isHidden = true
        
        if editSubType == .template {
            templatesController.view.isHidden = false
        } else if editSubType == .add {
            propertyController.view.isHidden = false
            propertyController.editSubType = editSubType
            propertyController.scrollToTop()
        } else if editSubType == .edit {
            propertyController.view.isHidden = false
            propertyController.editSubType = editSubType
            propertyController.scrollToTop()
        }
    }
    
    func reloadData() {
        documentPreview.pdfDocument = self.pdfDocument
        documentPreview.reloadData()
        
        propertyController.totalPDFCount = totalPDFCount
        
        if editSubType == .template {
            templatesController.reloadData()
            
        } else if editSubType == .add {
            propertyController.editSubType = editSubType
            propertyController.headerFooterModel = KMHeaderFooterManager.defaultManager.defaultHeaderFooter
            propertyController.reloadData()
            
        } else if editSubType == .edit {
            propertyController.reloadData()
        }
        
        updatePDFDocumentHeaderFooter()
    }
    
    func updatePDFDocumentHeaderFooter() {
        if let data = headerFooterModel {
            if let validPageString = self.validPageString {
                let validPageIndexs = validPageString.components(separatedBy: ",")
                if validPageIndexs.contains(String(format: "%ld", pdfViewCurrentIndex)) == false {
                    pdfDocument?.headerFooter().clear()
                    documentPreview.refreshUI()
                     
                    return
                }
            }
            
            let cPDFHeaderFooter = documentPreview.pdfDocument?.headerFooter()
            cPDFHeaderFooter?.pageString = "0"
            
            let fontSize = data.fontsize
            let fontName: String = data.fontName
            
            let font = NSFont.boldSystemFont(ofSize:fontSize)
            let style = NSMutableParagraphStyle()
            style.alignment = .center
            style.lineBreakMode = .byCharWrapping
            let size: NSSize = "text".boundingRect(with: NSSize(width: 1000, height: 1000), options: NSString.DrawingOptions(rawValue: 3), attributes: [NSAttributedString.Key.font : font, NSAttributedString.Key.paragraphStyle : style]).size
            
            cPDFHeaderFooter?.margin = NSEdgeInsetsMake(max(CGFloat(data.topMargin)-size.height, 0), CGFloat(data.leftMargin), max(CGFloat(data.bottomMargin)-size.height, 0), CGFloat(data.rightMargin))
            
            let strings = KMHeaderFooterModel.parseModel(model: data, pageCount: totalPDFCount)
            var count: Int = 0
            let color = data.color
            
            for text in strings {
                cPDFHeaderFooter?.setText(text, at: UInt(count))
                cPDFHeaderFooter?.setTextColor(color, at: UInt(count))
                cPDFHeaderFooter?.setFontSize(fontSize, at: UInt(count))
                cPDFHeaderFooter?.setFontName(fontName, at: UInt(count))
                count += 1
            }
            
            cPDFHeaderFooter?.update()
            
            documentPreview.refreshUI()
        } else {
            pdfDocument?.headerFooter().clear()
            documentPreview.refreshUI()
        }
    }
}

//MARK: - KMHeaderPropertyControllerDelegate
extension KMHeaderFooterController: KMHeaderPropertyControllerDelegate {
    func headerFooterPropertyControllerDidUpdate(_ controller: KMHeaderPropertyController) {
        updatePDFDocumentHeaderFooter()
        
        delegate?.kmHeaderFooterControllerDidModelDataUpdated?(self)
    }
    
    func headerFooterPropertyControllerDidChangetoTemplate(_ controller: KMHeaderPropertyController) {
        editSubType = .template
        resetUI()
        delegate?.kmHeaderFooterControllerDidUpdateModeType?(self)
        
        templatesController.selectedModel = nil
        templatesController.reloadData()
        
        updatePDFDocumentHeaderFooter()
        
        delegate?.kmHeaderFooterControllerDidModelDataUpdated?(self)
    }
    
    func headerFooterPropertyControllerSaveTemplateSuccess(_ controller: KMHeaderPropertyController, _ data: KMHeaderFooterModel) {
        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)
        
        editSubType = .template
        resetUI()
        delegate?.kmHeaderFooterControllerDidUpdateModeType?(self)
        
        templatesController.selectedModel = data
        templatesController.reloadData()
        
        updatePDFDocumentHeaderFooter()
        
        delegate?.kmHeaderFooterControllerDidModelDataUpdated?(self)
        
        KMHeaderFooterManager.defaultManager.defaultHeaderFooter = KMHeaderFooterModel()
    }
    
    func headerFooterPropertyControllerCancelTemplateEdit(_ controller: KMHeaderPropertyController) {
        editSubType = .template
        resetUI()
        delegate?.kmHeaderFooterControllerDidUpdateModeType?(self)
        
        templatesController.selectedModel = controller.headerFooterModel
        templatesController.reloadData()
        
        updatePDFDocumentHeaderFooter()
        
        delegate?.kmHeaderFooterControllerDidModelDataUpdated?(self)
        
    }
    
    func headerFooterPropertyControllerFinishTemplateEdit(_ controller: KMHeaderPropertyController) {
        editSubType = .template
        resetUI()
        delegate?.kmHeaderFooterControllerDidUpdateModeType?(self)
        
        templatesController.selectedModel = controller.headerFooterModel
        templatesController.reloadData()
        
        updatePDFDocumentHeaderFooter()
        
        delegate?.kmHeaderFooterControllerDidModelDataUpdated?(self)
    }
}


//MARK: - KMHFTemplateControllerDelegate
extension KMHeaderFooterController: KMHFTemplateControllerDelegate {
    func templateControllerDidSelectedChanged(_ controller: KMHFTemplateController) {
        
        updatePDFDocumentHeaderFooter()
        
        delegate?.kmHeaderFooterControllerDidModelDataUpdated?(self)
    }
    
    func templateControllerDidAddData(_ controller: KMHFTemplateController) {
        editSubType = .add
        resetUI()
        delegate?.kmHeaderFooterControllerDidUpdateModeType?(self)
        
        propertyController.clearData()
        propertyController.headerFooterModel = KMHeaderFooterManager.defaultManager.defaultHeaderFooter
        propertyController.totalPDFCount = self.totalPDFCount
        propertyController.reloadData()
        
        updatePDFDocumentHeaderFooter()
        
        delegate?.kmHeaderFooterControllerDidModelDataUpdated?(self)
        
        templatesController.selectedModel = nil
        templatesController.reloadData()
    }
    
    func templateControllerDidEditData(_ controller: KMHFTemplateController, _ data: KMHeaderFooterModel) {
        editSubType = .edit
        resetUI()
        delegate?.kmHeaderFooterControllerDidUpdateModeType?(self)
        
        propertyController.originalDataDict = KMHeaderFooterManager.defaultManager.parseModel(model: data)
        propertyController.headerFooterModel = data
        propertyController.reloadData()
        
        updatePDFDocumentHeaderFooter()
        
        delegate?.kmHeaderFooterControllerDidModelDataUpdated?(self)
         
        templatesController.selectedModel = nil
        templatesController.reloadData()
    }
    
}

//MARK: - KMEditDocumentControllerDelegate
extension KMHeaderFooterController: KMEditDocumentControllerDelegate {
    func editControllerDidExitEnable(_ controller: KMEditDocumentController) -> Bool {
        return true
    }
    
    func editControllerDidApplyEnable(_ controller: KMEditDocumentController) -> Bool {
        return self.applyEnabled()
    }
    
    func editControllerDidExitClicked(_ controller: KMEditDocumentController) {
        delegate?.kmHeaderFooterControllerDidChooseExit?(self)
    }
    
    func editControllerDidApplyClicked(_ controller: KMEditDocumentController) {
        delegate?.kmHeaderFooterControllerDidChooseApply?(self)
    }
    
}