PDF_Office/PDF Office/KMComponentLibrary/KMComponentLibrary/View/GroupItem/ComponentGroup.swift

//
//  ComponentGroup.swift
//  KMComponentLibrary
//
//  Created by Niehaoyu on 2024/9/3.
//

import Cocoa
import AppKit

@objc public protocol ComponentGroupDelegate: AnyObject {
    
    @objc optional func componentGroupDidSelect(group: ComponentGroup?, menuItemProperty: ComponentMenuitemProperty?)
    
    @objc optional func componentGroupDidDismiss(group: ComponentGroup?)
    
    @objc optional func componentGroupMouseEvent(group: ComponentGroup?, menuItemProperty: ComponentMenuitemProperty, _ mouseState: ComponentMouseState, _ event: NSEvent)
    
}

public class ComponentGroup: NSView, NibLoadable {
    
    @IBOutlet var contendBox: NSBox!
    @IBOutlet var scrollView: NSScrollView!
    @IBOutlet var collectionView: NSCollectionView!
     
    // MARK: Private Property
    private var localMonitor: Any?
    private var globalMonitor: Any?
    
    private var _shadowInfo : ComponentShadowInfo?
    
    public var menuItemArr: [ComponentMenuitemProperty] = []
    
    public var clickedAutoHide = true //点击Item后，是否自动隐藏
    
    var subGroupView: ComponentGroup!
    
    var choosedIndex: Int = -1
    
    weak open var groupDelegate: ComponentGroupDelegate?
    
    var windowVC = ComponentGroupWindowController.init(windowNibName: "ComponentGroupWindowController")
    var subWindowVC = ComponentSubGroupWindowController.init(windowNibName: "ComponentSubGroupWindowController")
    
    deinit {
        NotificationCenter.default.removeObserver(self)
        
    }
    
    public override func draw(_ dirtyRect: NSRect) {
        super.draw(dirtyRect)
        
        // Drawing code here.
        
        if shadowInfo != nil {
            let shadow = NSShadow()
            shadow.shadowColor = shadowInfo?.shadowColor
            shadow.shadowOffset = shadowInfo?.shadowOffset ?? CGSizeZero
            shadow.shadowBlurRadius = shadowInfo?.shadowBlurRadius ?? 0
            shadow.set()
            
            let cornerRadius: CGFloat = shadowInfo?.cornerRadius ?? 0
            let roundedRect = NSBezierPath(roundedRect: bounds, xRadius: cornerRadius, yRadius: cornerRadius)
            
            if let color = shadowInfo?.color {
                color.setFill()
                roundedRect.fill()
            }
        }
        
    }
    
    public override func removeFromSuperview() {
        super.removeFromSuperview()
        
        clearMonitor()
    }
    
    public required init?(coder decoder: NSCoder) {
        super.init(coder: decoder)
        
    }
    
    override init(frame frameRect: NSRect) {
        super.init(frame: frameRect)
    }
    
    public override func awakeFromNib() {
        super.awakeFromNib()
        
        shadowInfo = ComponentLibrary.shared.configShadowInfoWithKey("")
        
        if let value = ComponentLibrary.shared.getComponentValueFromKey("colorBg/popup") {
            let currentValue = value as! [String : Any]
            contendBox.fillColor = ComponentLibrary.shared.getColor(rgbaDict: currentValue)
        } else {
            contendBox.fillColor = NSColor.clear
        }
        contendBox.borderColor = NSColor.clear
        contendBox.borderWidth = 0
        
        scrollView.scrollerStyle = .overlay
        
        collectionView.delegate = self
        collectionView.dataSource = self
        collectionView.backgroundColors = [NSColor.clear]
        collectionView.wantsLayer = true
        collectionView.layer?.backgroundColor = NSColor.clear.cgColor
        collectionView.register(ComponentGroupItem.self, forItemWithIdentifier: NSUserInterfaceItemIdentifier(rawValue: "componentGroupItemID"))
        
    }
    
    //MARK: - Setter and Getter
    public var shadowInfo : ComponentShadowInfo? {
        get {
            return _shadowInfo
        }
        set {
            _shadowInfo = newValue
            
            display()
        }
    }
    
    //MARK: - Public Method
    public func updateGroupInfo(_ arr: [ComponentMenuitemProperty]) {
        menuItemArr.removeAll()
        
        for item in arr {
            menuItemArr.append(item)
        }
        
        reloadData()
    }
    
    public func reloadData() {
        collectionView.reloadData()
        
        if choosedIndex > 0 {
            DispatchQueue.main.asyncAfter(deadline: DispatchTime.now() + 0.15) {
                let indexpath = IndexPath(item: self.choosedIndex, section: 0)
                var set = Set<IndexPath>()
                set.insert(indexpath)
                
                self.collectionView.scrollToItems(at: set, scrollPosition: .centeredVertically)
            }
        }
    }
    
    //MARK: - Show
    //Point是相对于当前WIndow的point，
    public func showWithPoint(_ point: CGPoint, relativeTo positioningView: NSView?) {
        
        var screenRect = NSScreen.main!.frame
        if let keyWindow = NSApplication.shared.keyWindow, let screen = keyWindow.screen {
            screenRect = screen.visibleFrame
        }
        if self.windowVC.window?.isVisible == true {
            self.windowVC.close()
        }
        let windowVC = ComponentGroupWindowController.init(windowNibName: "ComponentGroupWindowController")
        windowVC.window?.setFrame(screenRect, display: true)
        if let window = positioningView?.window {
            windowVC.window?.level = window.level + 1
        }
        windowVC.window?.backgroundColor = NSColor.clear
        windowVC.showWindow(nil)
        
        self.windowVC = windowVC
        
        var rect = self.frame
        if let subView = positioningView {
            
            let subviewRect = subView.bounds // NSView的坐标
            let subviewWindow = subView.window
            let subviewWindowRect = subView.convert(subviewRect, to: nil) // 转换到窗口坐标
            let subviewscreenRect = subviewWindow?.convertToScreen(subviewWindowRect) ?? CGRectZero // 转换到屏幕坐标 ?
            
            rect.origin.x = subviewscreenRect.origin.x - screenRect.origin.x
            rect.origin.y = subviewscreenRect.origin.y - screenRect.origin.y
            rect.origin.y -= rect.size.height
            
        } else {
            var editPoint = point
            if let keyWindow = NSApplication.shared.keyWindow {
                editPoint.x += keyWindow.frame.origin.x - screenRect.origin.x
                editPoint.y += keyWindow.frame.origin.y - screenRect.origin.y
            }
            rect.origin.x = editPoint.x
            rect.origin.y = editPoint.y
        }
        var viewWidth: CGFloat = CGRectGetWidth(self.frame)
        for item in menuItemArr {
            ComponentLibrary.shared.configMenuItemComponent(properties: item)
            viewWidth = max(viewWidth, item.propertyInfo.viewWidth)
        }
        rect.size.width = viewWidth
        if(rect.origin.y < 0) {
            rect.origin.y = 0
        }
        self.frame = rect
        
        if let window = windowVC.window {
            window.contentView?.addSubview(self)
        }
        
        clearMonitor()
        
        addMonitor()
        
    }
    
    public func removeGroupView() {
        clearMonitor()
        
        if let tempWindow = self.window {
            tempWindow.orderOut(nil)
            tempWindow.close()
        }
        
        if let window = windowVC.window {
            window.close()
        }
        
        removeFromSuperview()
        
    }
    
    //MARK: - SubGroupView
    func removeSubGroupView() {
        if subGroupView != nil {
            subGroupView.removeGroupView()
            subGroupView = nil
        }
        
        if let window = subGroupView?.window {
            window.close()
        }
        
    }
    
    func showSubGroupWithPoint(_ point: CGPoint) {
        
        var screenRect = NSScreen.main!.frame
        if let keyWindow = NSApplication.shared.keyWindow, let screen = keyWindow.screen {
            screenRect = screen.visibleFrame
        }
        
        if self.subWindowVC.window?.isVisible == true {
            self.subWindowVC.close()
        }
        let windowVC = ComponentSubGroupWindowController.init(windowNibName: "ComponentSubGroupWindowController")
        windowVC.window?.setFrame(screenRect, display: true)
        windowVC.window?.level = .popUpMenu
        windowVC.window?.backgroundColor = NSColor.clear
        windowVC.showWindow(nil)
        self.subWindowVC = windowVC
        
        var rect = self.frame
        rect.origin.x = point.x
        rect.origin.y = point.y
        
        var viewWidth: CGFloat = CGRectGetWidth(self.frame)
        for item in menuItemArr {
            ComponentLibrary.shared.configMenuItemComponent(properties: item)
            viewWidth = max(viewWidth, item.propertyInfo.viewWidth)
        }
        rect.size.width = viewWidth
        if(rect.origin.y < 0) {
            rect.origin.y = 0
        }
        self.frame = rect
        
        if let window = windowVC.window {
            window.contentView?.addSubview(self)
        }
        
        clearMonitor()
        
        addMonitor()
        
    }
    
    func showSubGroupView(itemView: ComponentGroupItem) {
        if subGroupView == nil {
            subGroupView = ComponentGroup.createFromNib(in: ComponentLibrary.shared.componentBundle())
        }
        var viewHeight: CGFloat = 8
        for property in itemView.properties.subPropertys {
            if property.type == .divider {
                viewHeight += 8
            } else if property.type == .normal {
                viewHeight += 36
            } else if property.type == .header {
                viewHeight += 36
            }
        }
        subGroupView.groupDelegate = self.groupDelegate
        subGroupView?.frame = CGRectMake(0, 0, 160, viewHeight)
        subGroupView.updateGroupInfo(itemView.properties.subPropertys)
        
        var point = itemView.contendBox.viewFrameInScreen().origin
        point.x += CGRectGetWidth(itemView.contendBox.frame)
        point.y -= viewHeight
        point.y += CGRectGetHeight(itemView.contendBox.frame)
         
        guard let _ = itemView.view.window?.contentView else {
            return
        }
        subGroupView.showSubGroupWithPoint(point)
    }
    
    //MARK: - Private Method
    func addMonitor() {
        if globalMonitor == nil {
            globalMonitor = NSEvent.addGlobalMonitorForEvents(matching: [.leftMouseDown], handler: { (event) in
                if event.type == .leftMouseDown {
                    self.leftMouseDownAction()
                }
            })
        }
        
        if localMonitor == nil {
            localMonitor = NSEvent.addLocalMonitorForEvents(matching: [.leftMouseDown, .scrollWheel, .rightMouseDown, .otherMouseDown], handler: { (event) in
                if event.type == .leftMouseDown {
                    let point = self.convert(event.locationInWindow, from: nil)
                    if CGRectContainsPoint(self.bounds, point) {
                        
                    } else {
                        self.leftMouseDownAction()
                    }
                } else if event.type == .scrollWheel {
                    let point = self.convert(event.locationInWindow, from: nil)

                    var subPoint = CGPointZero
                    if let subview = self.subGroupView {
                        subPoint = subview.convert(event.locationInWindow, from: nil)
                    }
                    if CGRectContainsPoint(self.bounds, point) {
                        
                    } else if let subView = self.subGroupView, CGRectContainsPoint(subView.bounds, subPoint) {
                        
                    } else {
                       
                    }
                } else if event.type == .rightMouseDown {
                    let point = self.convert(event.locationInWindow, from: nil)
                    if CGRectContainsPoint(self.bounds, point) {
                        
                    } else {
                        self.leftMouseDownAction()
                    }
                } else if event.type == .otherMouseDown {
                    self.leftMouseDownAction()
                }
                return event
            })
        }
    }
    
    func clearMonitor() {
        if localMonitor != nil {
            NSEvent.removeMonitor(self.localMonitor as Any)
            localMonitor = nil
        }
        
        if globalMonitor != nil {
            NSEvent.removeMonitor(self.globalMonitor as Any)
            globalMonitor = nil
        }
    }
    
    @objc func leftMouseDownAction() {
        groupDelegate?.componentGroupDidDismiss?(group: self)
        
        removeGroupView()
        
        removeSubGroupView()
    }
}

//MARK: - NSCollectionViewDelegate
extension ComponentGroup: NSCollectionViewDelegate {
    
}

extension ComponentGroup: NSCollectionViewDataSource {
    public func numberOfSections(in collectionView: NSCollectionView) -> Int {
        return 1
    }
    
    public func collectionView(_ collectionView: NSCollectionView, numberOfItemsInSection section: Int) -> Int {
        return menuItemArr.count
    }
    
    public func collectionView(_ collectionView: NSCollectionView, itemForRepresentedObjectAt indexPath: IndexPath) -> NSCollectionViewItem {
        
        let itemView = collectionView.makeItem(withIdentifier: NSUserInterfaceItemIdentifier(rawValue: "componentGroupItemID"), for: indexPath) as! ComponentGroupItem
        if indexPath.item < menuItemArr.count {
            let properties = menuItemArr[indexPath.item]
            itemView.setUpWithProperties(properties)
        }
        itemView.clickHandle = {[weak self] view in
            DispatchQueue.main.async {
                let property = view.properties
                if self?.clickedAutoHide == true {
                    self?.groupDelegate?.componentGroupDidDismiss?(group: self)
                    self?.removeGroupView()
                }
                self?.groupDelegate?.componentGroupDidSelect?(group: self, menuItemProperty: property)
            }
        }
        
        itemView.mouseHandle = {[weak self] view, mouseState, event in
            guard let weakSelf = self else { return }
            if view.properties.subPropertys.isEmpty == false {
                if mouseState == .enter {
                    self?.showSubGroupView(itemView: view)
                } else if mouseState == .move {
                    
                } else if mouseState == .exit {
                    
                }
            } else {
                self?.removeSubGroupView()
            }
            self?.groupDelegate?.componentGroupMouseEvent?(group: self, menuItemProperty: view.properties, mouseState, event)
        }
        return itemView
    }
}

extension ComponentGroup: NSCollectionViewDelegateFlowLayout {
    
    public func collectionView(_ collectionView: NSCollectionView, layout collectionViewLayout: NSCollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> NSSize {
        
        var viewHeight: CGFloat = 36
        if indexPath.item < menuItemArr.count {
            let properties = menuItemArr[indexPath.item]
            if properties.type == .divider {
                viewHeight = 8
            }
        }
        return NSSize(width: NSWidth(self.collectionView.frame), height: viewHeight)
    }
    
    public func collectionView(_ collectionView: NSCollectionView, layout collectionViewLayout: NSCollectionViewLayout, referenceSizeForHeaderInSection section: Int) -> NSSize {
        return NSSize(width: 0, height: 0)
    }
    
    public func collectionView(_ collectionView: NSCollectionView, layout collectionViewLayout: NSCollectionViewLayout, referenceSizeForFooterInSection section: Int) -> NSSize {
        return NSSize(width: 0, height: 0)
    }
    
    public func collectionView(_ collectionView: NSCollectionView, layout collectionViewLayout: NSCollectionViewLayout, minimumLineSpacingForSectionAt section: Int) -> CGFloat {
        return 0
    }
    
    public func collectionView(_ collectionView: NSCollectionView, layout collectionViewLayout: NSCollectionViewLayout, insetForSectionAt section: Int) -> NSEdgeInsets {
        return NSEdgeInsetsMake(0, 0, 0, 0)
    }
    
    public func collectionView(_ collectionView: NSCollectionView, didSelectItemsAt indexPaths: Set<IndexPath>) {
        
    }
}