PDF_Office/PDF Office/PDF Master/KMClass/KMPDFViewController/HeaderFooter/Views/KMHFTemplateController.swift

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

import Cocoa
import KMComponentLibrary

@objc protocol KMHFTemplateControllerDelegate: AnyObject {
    
    @objc optional func templateControllerDidAddData(_ controller: KMHFTemplateController)
    
    @objc optional func templateControllerDidEditData(_ controller: KMHFTemplateController, _ data: KMHeaderFooterModel)
    
    //选中内容切换
    @objc optional func templateControllerDidSelectedChanged(_ controller: KMHFTemplateController)
    
}

class KMHFTemplateController: NSViewController {
    @IBOutlet var contendView: NSView!
    @IBOutlet var titleLabel: NSTextField!
    @IBOutlet var addButton: ComponentButton!
    
    @IBOutlet var emptyView: ComponentEmpty!
    
    @IBOutlet var scrollView: NSScrollView!
    @IBOutlet var collectionView: NSCollectionView!
    
    weak open var delegate: KMHFTemplateControllerDelegate?
    
    var selectedModel: KMHeaderFooterModel?
    
    override func viewDidLoad() {
        super.viewDidLoad()
        // Do view setup here.
        
        configUI()
        
        setupProperty()
        
    }
    
    func setupProperty() {
        
        addButton.properties = ComponentButtonProperty(type: .text_gray, size: .xxs, onlyIcon: true, icon: NSImage(named: "watermark_template_add"), keepPressState: false)
        addButton.setTarget(self, action: #selector(buttonClicked(_:)))
        
        emptyView.properties = ComponentEmptyProperty(emptyType: .noWatermark, text: KMLocalizedString("No Template"), subText: KMLocalizedString("Here is the description."))
    }
    
    func configUI() {
        view.wantsLayer = true
        view.layer?.backgroundColor = ComponentLibrary.shared.getComponentColorFromKey("colorBg/layout-middle").cgColor
        
        titleLabel.stringValue = KMLocalizedString("Templates")
        titleLabel.textColor = ComponentLibrary.shared.getComponentColorFromKey("colorText/2")
        titleLabel.font = ComponentLibrary.shared.getFontFromKey("mac/body-m-bold")
        
        collectionView.backgroundColors = [NSColor.clear]
        collectionView.wantsLayer = true
        collectionView.layer?.backgroundColor = NSColor.clear.cgColor
        collectionView.delegate = self
        collectionView.dataSource = self
        collectionView.allowsEmptySelection = false
        collectionView.allowsMultipleSelection = false
        
        collectionView.register(KMHeaderTemplateItem.self, forItemWithIdentifier: NSUserInterfaceItemIdentifier(rawValue: "KMHeaderTemplateItem"))
    }
    
    //刷新所有数据
    public func reloadData() {
        if KMHeaderFooterManager.defaultManager.headFooterObjects.count == 0 {
            collectionView.isHidden = true
            emptyView.isHidden = false
        } else {
            emptyView.isHidden = true
            collectionView.isHidden = false
            collectionView.reloadData()
            
            updateSelectedState()
        }
    }
    
    public func updateSelectedState() {
        if let selData = selectedModel {
            if let index = KMHeaderFooterManager.defaultManager.headFooterObjects.firstIndex(of: selData) {
                let indexpath = IndexPath(item: index, section: 0)
                var set = Set<IndexPath>()
                set.insert(indexpath)
                
                collectionView.selectItems(at: set, scrollPosition: .top)
            }
        } else {
            collectionView.deselectAll(nil)
        }
    }
    
    private func collectionViewSelectedChanged() {
        let indexs = collectionView.selectionIndexPaths
        if indexs.count > 0 {
            for index in indexs {
                let data = KMHeaderFooterManager.defaultManager.headFooterObjects[index.item]
                selectedModel = data
            }
        } else {
            selectedModel = nil
        }
        DispatchQueue.main.async {
            self.delegate?.templateControllerDidSelectedChanged?(self)
        }
    }
    
    //MARK: - action
    @objc func buttonClicked(_ sender: ComponentButton) {
        if sender == addButton {
            if selectedModel != nil {
                selectedModel = nil
                delegate?.templateControllerDidSelectedChanged?(self)
            }
            reloadData()
            
            delegate?.templateControllerDidAddData?(self)
        }
    }
}

//MARK: - NSCollectionViewDelegate, NSCollectionViewDataSource
extension KMHFTemplateController: NSCollectionViewDelegate, NSCollectionViewDataSource, NSCollectionViewDelegateFlowLayout {
    
    public func collectionView(_ collectionView: NSCollectionView, numberOfItemsInSection section: Int) -> Int {
        return KMHeaderFooterManager.defaultManager.headFooterObjects.count
    }
    
    public func collectionView(_ collectionView: NSCollectionView, itemForRepresentedObjectAt indexPath: IndexPath) -> NSCollectionViewItem {
        if indexPath.item >= KMHeaderFooterManager.defaultManager.headFooterObjects.count {
            return NSCollectionViewItem()
        }
        
        let item: KMHeaderTemplateItem = collectionView.makeItem(withIdentifier: NSUserInterfaceItemIdentifier(rawValue: "KMHeaderTemplateItem"), for: indexPath) as! KMHeaderTemplateItem
        item.delegate = self
        item.dataModel = KMHeaderFooterManager.defaultManager.headFooterObjects[indexPath.item]
        item.reloadData()
        
        return item
    }
    
    public func collectionView(_ collectionView: NSCollectionView, layout collectionViewLayout: NSCollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> NSSize {
        var itemHeight = 56
        
        let dataModel = KMHeaderFooterManager.defaultManager.headFooterObjects[indexPath.item]
        if dataModel.topLeftString.count > 0 {
            itemHeight += 24
            itemHeight += 8
        }
        if dataModel.topCenterString.count > 0 {
            itemHeight += 24
            itemHeight += 8
        }
        if dataModel.topRightString.count > 0 {
            itemHeight += 24
            itemHeight += 8
        }
        if dataModel.bottomLeftString.count > 0 {
            itemHeight += 24
            itemHeight += 8
        }
        if dataModel.bottomCenterString.count > 0 {
            itemHeight += 24
            itemHeight += 8
        }
        if dataModel.bottomRightString.count > 0 {
            itemHeight += 24
            itemHeight += 8
        }
        itemHeight -= 8
        
        return CGSize(width: 232, height: itemHeight)
    }
    
    func collectionView(_ collectionView: NSCollectionView, layout collectionViewLayout: NSCollectionViewLayout, minimumInteritemSpacingForSectionAt section: Int) -> CGFloat {
        return 8
    }
    
    func collectionView(_ collectionView: NSCollectionView, layout collectionViewLayout: NSCollectionViewLayout, minimumLineSpacingForSectionAt section: Int) -> CGFloat {
        return 8
    }
    
    func collectionView(_ collectionView: NSCollectionView, layout collectionViewLayout: NSCollectionViewLayout, insetForSectionAt section: Int) -> NSEdgeInsets {
        return NSEdgeInsets(top: 8, left: 0, bottom: 8, right: 0)
    }
    
    public func collectionView(_ collectionView: NSCollectionView, didSelectItemsAt indexPaths: Set<IndexPath>) {
        collectionViewSelectedChanged()
    }
    
    public func collectionView(_ collectionView: NSCollectionView, didDeselectItemsAt indexPaths: Set<IndexPath>) {
        collectionViewSelectedChanged()
    }
    
}

//MARK: - kmNWatermarkTemplateItemDelegate
extension KMHFTemplateController: KMHeaderTemplateItemDelegate {
    func kmHeaderTemplateItemDidEdit(_ view: KMHeaderTemplateItem) {
        if let data = view.dataModel {
            delegate?.templateControllerDidEditData?(self, data)
        }
    }
    
    func kmHeaderTemplateItemDidDelete(_ view: KMHeaderTemplateItem) {
        if let data = view.dataModel {
            let alert = NSAlert()
            alert.alertStyle = .critical
            alert.messageText = String(format: KMLocalizedString("Are you sure you want to remove the background?"))
            alert.addButton(withTitle: KMLocalizedString("Delete"))
            alert.addButton(withTitle: KMLocalizedString("Cancel"))
            
            alert.beginSheetModal(for: NSApp.mainWindow!) { (response) in
                if response == .alertFirstButtonReturn {
                    let _ = KMHeaderFooterManager.defaultManager.removeHeaderFooter(data)
                    if data == self.selectedModel {
                        self.selectedModel = nil
                        
                        self.delegate?.templateControllerDidSelectedChanged?(self)
                    }
                    self.reloadData()
                }
            }
        }
    }
}