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

//删除数据时调用
let kHeaderFooterListDeleteAllNotiName = NSNotification.Name("kHeaderFooterListDeleteAllNotiName")
let kHeaderFooterListDeleteNotiName    = NSNotification.Name("kHeaderFooterListDeleteNotiName")


@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 groupView: ComponentGroup?
    var editCell: KMHeaderTemplateItem?
    
    var selectedModel: KMHeaderFooterModel?
    
    deinit {
        NotificationCenter.default.removeObserver(self)
    }
    
    override func viewDidLoad() {
        super.viewDidLoad()
        // Do view setup here.
        
        configUI()
        
        setupProperty()
        
        NotificationCenter.default.addObserver(self, selector: #selector(deleteItemNoti), name: kHeaderFooterListDeleteNotiName, object: nil)
        NotificationCenter.default.addObserver(self, selector: #selector(deleteAllItemsNoti), name: kHeaderFooterListDeleteAllNotiName, object: nil)
        NotificationCenter.default.addObserver(self, selector: #selector(dataUpdated), name: KMHeaderFooterDatasUpdateNotiName , object: nil)

    }
    
    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)
        }
    }
    
    override func rightMouseUp(with event: NSEvent) {
        super.rightMouseUp(with: event)
        
        var items: [String] = []
        var point = self.view.convert(event.locationInWindow, from: nil)
        let collectionPoint = self.view.convert(point, to: collectionView)
        
        self.editCell = nil
        
        if let indexPath = self.collectionView.indexPathForItem(at: collectionPoint) {
            let cellView = collectionView.item(at: indexPath) as? KMHeaderTemplateItem
            self.editCell = cellView
            items.append(KMLocalizedString("Rename"))
            items.append(KMLocalizedString("Edit"))
            items.append(KMLocalizedString("Delete"))
            items.append(KMLocalizedString("Remove All"))
            
            self.showGroupView(items, event.locationInWindow)
        }
    }
    
    func showRenameWindow() {
        let saveWindow: KMWatermarkSaveWindow = KMWatermarkSaveWindow(windowNibName: "KMWatermarkSaveWindow")
        if let item = self.editCell, let name = item.dataModel?.name {
            saveWindow.nameValue = name
        } else {
            saveWindow.nameValue = ""
        }
        saveWindow.saveHandler = {[weak self] string in
            guard let weakSelf = self else { return }
            DispatchQueue.main.async {
                weakSelf.editCell?.dataModel?.name = string ?? ""
                if let model = weakSelf.editCell?.dataModel, KMHeaderFooterManager.defaultManager.updateHeaderFooter(theModel: model) {
                    weakSelf.reloadData()
                }
            }
        }
        saveWindow.own_beginSheetModal(for: view.window) { string in
            
        }
        saveWindow.titleLabel.stringValue = KMLocalizedString("Rename")
    }
    
    func showGroupView(_ items: [String], _ point: CGPoint) {
        var viewHeight: CGFloat = 8
        var menuItemArr: [ComponentMenuitemProperty] = []
        for i in items {
            let properties_Menuitem: ComponentMenuitemProperty = ComponentMenuitemProperty(multipleSelect: false,
                                                                                           itemSelected: false,
                                                                                           isDisabled: false,
                                                                                           keyEquivalent: nil,
                                                                                           text: KMLocalizedString(i),
                                                                                           identifier: i)
            menuItemArr.append(properties_Menuitem)
            viewHeight += 36
        }
        
        if groupView == nil {
            groupView = ComponentGroup.createFromNib(in: ComponentLibrary.shared.componentBundle())
        }
        groupView?.groupDelegate = self
        var viewPoint = point
        viewPoint.y -= viewHeight

        groupView?.frame = CGRectMake(310, 0, 200, viewHeight)
        groupView?.updateGroupInfo(menuItemArr)
        groupView?.showWithPoint(viewPoint, relativeTo: nil)
         
    }
     
    //MARK: - Noti
    @objc func deleteItemNoti(_ noti: Notification) {
        guard let notiTag = noti.object as? String else {
            return
        }
        if notiTag == self.selectedModel?.tag {
            self.selectedModel = nil
            
            self.delegate?.templateControllerDidSelectedChanged?(self)
        }
        self.reloadData()
    }
    
    @objc func deleteAllItemsNoti() {
        if let _ = self.selectedModel {
            self.selectedModel = nil
            
            self.delegate?.templateControllerDidSelectedChanged?(self)
        }
        self.reloadData()
    }
    
    @objc func dataUpdated() {
        self.reloadData()
    }
    
}

//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 Header&Footer?"))
            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()
                    
                    NotificationCenter.default.post(name: kHeaderFooterListDeleteNotiName, object: data.tag)

                }
            }
        }
    }
    
    func kmHeaderTemplateItemDidDeleteAll() {
        let alert = NSAlert()
        alert.alertStyle = .critical
        alert.messageText = String(format: KMLocalizedString("Are you sure you want to remove all the Header&Footer?"))
        alert.addButton(withTitle: KMLocalizedString("Delete"))
        alert.addButton(withTitle: KMLocalizedString("Cancel"))
        
        alert.beginSheetModal(for: NSApp.mainWindow!) { (response) in
            if response == .alertFirstButtonReturn {
                let _ = KMHeaderFooterManager.defaultManager.removeAllHeaderFooters()
                if let _ = self.selectedModel {
                    self.selectedModel = nil
                    
                    self.delegate?.templateControllerDidSelectedChanged?(self)
                }
                self.reloadData()
                
               NotificationCenter.default.post(name: kHeaderFooterListDeleteAllNotiName, object: nil)

            }
        }
    }
}

//MARK: - ComponentGroupDelegate
extension KMHFTemplateController: ComponentGroupDelegate {
    func componentGroupDidSelect(group: ComponentGroup?, menuItemProperty: ComponentMenuitemProperty?) {
        if menuItemProperty?.identifier == KMLocalizedString("Rename") {
            self.showRenameWindow()
        } else if menuItemProperty?.identifier == KMLocalizedString("Edit") {
            if let cell = self.editCell {
                self.kmHeaderTemplateItemDidEdit(cell)
            }
        } else if menuItemProperty?.identifier == KMLocalizedString("Delete") {
            if let cell = self.editCell {
                self.kmHeaderTemplateItemDidDelete(cell)
            }
        } else if menuItemProperty?.identifier == KMLocalizedString("Remove All") {
            self.kmHeaderTemplateItemDidDeleteAll()
            
        }
    }
    
}