PDF_Office/PDF Office/PDF Master/KMClass/KMPDFViewController/RightSideController/Views/Note/KMNoteController.swift

//
//  KMNoteController.swift
//  PDF Reader Pro
//
//  Created by Niehaoyu on 2024/11/26.
//

import Cocoa
import KMComponentLibrary

class KMNoteController: KMNBaseViewController {
    @IBOutlet var colorBGView: NSView!
    @IBOutlet var colorLabel: NSTextField!
    @IBOutlet var colorGroup: ComponentCColorGroup!
    
    @IBOutlet var typeBGView: NSView!
    @IBOutlet var typeLabel: NSTextField!
    @IBOutlet var typeItemA: ComponentCSelector!
    @IBOutlet var typeItemB: ComponentCSelector!
    @IBOutlet var typeItemC: ComponentCSelector!
    @IBOutlet var typeItemD: ComponentCSelector!
    @IBOutlet var typeItemE: ComponentCSelector!
    @IBOutlet var typeItemF: ComponentCSelector!
    @IBOutlet var typeItemG: ComponentCSelector!
    
    private var annotations: [CPDFTextAnnotation] = []
    
    var pdfView: CPDFListView?
    
    override func viewDidLoad() {
        super.viewDidLoad()
        // Do view setup here.
        setupProperty()
    }
    
    override func updateUILanguage() {
        super.updateUILanguage()
        
        reloadUI()
        
    }
    
    override func updateUIThemeColor() {
        super.updateUIThemeColor()
        
        reloadUI()
        
    }
    
    func setupProperty() {
        colorGroup.delegate = self
        
        typeItemA.properties = ComponentCSelectorProperty(size: .m, state: .normal, iconImage: NSImage(named: "note_1"), identifier: "")
        typeItemB.properties = ComponentCSelectorProperty(size: .m, state: .normal, iconImage: NSImage(named: "note_2"), identifier: "")
        typeItemC.properties = ComponentCSelectorProperty(size: .m, state: .normal, iconImage: NSImage(named: "note_3"), identifier: "")
        typeItemD.properties = ComponentCSelectorProperty(size: .m, state: .normal, iconImage: NSImage(named: "note_4"), identifier: "")
        typeItemE.properties = ComponentCSelectorProperty(size: .m, state: .normal, iconImage: NSImage(named: "note_5"), identifier: "")
        typeItemF.properties = ComponentCSelectorProperty(size: .m, state: .normal, iconImage: NSImage(named: "note_6"), identifier: "")
        typeItemG.properties = ComponentCSelectorProperty(size: .m, state: .normal, iconImage: NSImage(named: "note_7"), identifier: "")
        
        for item in [typeItemA, typeItemB, typeItemC, typeItemD, typeItemE, typeItemF, typeItemG] {
            item!.setTarget(self, action: #selector(selectItemClick(_:)))
        }
        
    }
    
    func reloadUI() {
        colorLabel.stringValue = KMLocalizedString("Color")
        colorLabel.textColor = ComponentLibrary.shared.getComponentColorFromKey("colorText/2")
        colorLabel.font = ComponentLibrary.shared.getFontFromKey("mac/body-s-medium")
        
        typeLabel.stringValue = KMLocalizedString("Color")
        typeLabel.textColor = ComponentLibrary.shared.getComponentColorFromKey("colorText/2")
        typeLabel.font = ComponentLibrary.shared.getFontFromKey("mac/body-s-medium")
        
    }
    
    func reloadData() {
        guard let pdfView = self.pdfView else {
            return
        }
        
        self.annotations.removeAll()
        let allAnnotations: [CPDFAnnotation] = pdfView.activeAnnotations as? [CPDFAnnotation] ?? []
        for annotation in allAnnotations {
            if annotation is CPDFTextAnnotation {
                annotations.append((annotation as! CPDFTextAnnotation))
            }
        }
        var firstAnnotation: CPDFTextAnnotation?
        if annotations.count > 0 {
            firstAnnotation = annotations.first
        }
        
        for item in [typeItemA, typeItemB, typeItemC, typeItemD, typeItemE, typeItemF, typeItemG] {
            item!.properties.state = .normal
        }
        
        if true {
            let colors: [NSColor] = KMAnnotationPropertiesColorManager.manager.noteColors
            if colors.count > 4 {
                let colorAProperty = ComponentCColorProperty(colorType: .color, state: .normal, isCustom: false, color: colors[0])
                let colorBProperty = ComponentCColorProperty(colorType: .color, state: .normal, isCustom: false, color: colors[1])
                let colorCProperty = ComponentCColorProperty(colorType: .color, state: .normal, isCustom: false, color: colors[2])
                let colorDProperty = ComponentCColorProperty(colorType: .color, state: .normal, isCustom: false, color: colors[3])
                let colorEProperty = ComponentCColorProperty(colorType: .color, state: .normal, isCustom: true, color: colors[4])
                colorGroup.setUpWithColorPropertys([colorAProperty, colorBProperty, colorCProperty, colorDProperty], customItemProperty: colorEProperty)
            }
        }
        
        if annotations.count == 0 {
            colorGroup.currentColor = CPDFTextAnnotation.defaultColor()
            colorGroup.refreshUI()
            
            let anchoredIconType: CPDFTextAnnotationIconType = CPDFTextAnnotation.defaultType()
            if anchoredIconType == .comment {
                typeItemA.properties.state = .pressed
            } else if anchoredIconType == .key {
                typeItemC.properties.state = .pressed
            } else if anchoredIconType == .note {
                typeItemB.properties.state = .pressed
            } else if anchoredIconType == .help {
                typeItemD.properties.state = .pressed
            } else if anchoredIconType == .newParagraph {
                typeItemG.properties.state = .pressed
            } else if anchoredIconType == .paragraph {
                typeItemE.properties.state = .pressed
            } else if anchoredIconType == .insert {
                typeItemF.properties.state = .pressed
            }
            
        } else if annotations.count == 1, let annotation = firstAnnotation {
            colorGroup.currentColor = annotation.color
            colorGroup.refreshUI()
            
            let anchoredIconType: CPDFTextAnnotationIconType = annotation.iconType()
            if anchoredIconType == .comment {
                typeItemA.properties.state = .pressed
            } else if anchoredIconType == .key {
                typeItemC.properties.state = .pressed
            } else if anchoredIconType == .note {
                typeItemB.properties.state = .pressed
            } else if anchoredIconType == .help {
                typeItemD.properties.state = .pressed
            } else if anchoredIconType == .newParagraph {
                typeItemG.properties.state = .pressed
            } else if anchoredIconType == .paragraph {
                typeItemE.properties.state = .pressed
            } else if anchoredIconType == .insert {
                typeItemF.properties.state = .pressed
            }
        } else {
            let multiColor: Bool = CPDFListView.isAnnotationsContainMultiType(annotations, withType: .color)
            if multiColor == true {
                colorGroup.currentColor = nil
            } else {
                if let annotation = firstAnnotation {
                    colorGroup.currentColor = annotation.color
                }
            }
            colorGroup.refreshUI()
            
            let multiType = CPDFListView.isAnnotationsContainMultiType(annotations, withType: .note_Type)
            if multiType == true {
                
            } else if let annotation = firstAnnotation {
                let anchoredIconType: CPDFTextAnnotationIconType = annotation.iconType()
                if anchoredIconType == .comment {
                    typeItemA.properties.state = .pressed
                } else if anchoredIconType == .key {
                    typeItemC.properties.state = .pressed
                } else if anchoredIconType == .note {
                    typeItemB.properties.state = .pressed
                } else if anchoredIconType == .help {
                    typeItemD.properties.state = .pressed
                } else if anchoredIconType == .newParagraph {
                    typeItemG.properties.state = .pressed
                } else if anchoredIconType == .paragraph {
                    typeItemE.properties.state = .pressed
                } else if anchoredIconType == .insert {
                    typeItemF.properties.state = .pressed
                }
            }
        }
        for item in [typeItemA, typeItemB, typeItemC, typeItemD, typeItemE, typeItemF, typeItemG] {
            item!.reloadData()
        }
    }
    
    @objc func selectItemClick(_ sender: ComponentCSelector) {
        var anchoredIconType: CPDFTextAnnotationIconType = .comment
        if sender == typeItemA {
            anchoredIconType = .comment
        } else if sender == typeItemB {
            anchoredIconType = .note
        } else if sender == typeItemC {
            anchoredIconType = .key
        } else if sender == typeItemD {
            anchoredIconType = .help
        } else if sender == typeItemE {
            anchoredIconType = .paragraph
        } else if sender == typeItemF {
            anchoredIconType = .insert
        } else if sender == typeItemG {
            anchoredIconType = .newParagraph
        }
        
        CPDFAnnotation.updateAnnotations(annotations, newIconType: anchoredIconType, withPDFView: pdfView)
        CPDFTextAnnotation.updateDefaultIconType(anchoredIconType)
        
        reloadData()
    }
}

extension KMNoteController: ComponentCColorDelegate {
    func componentCColorDidChooseColor(_ view: NSView, _ color: NSColor?) {
        self.componentCColorDidChooseColor(view, color, true)
    }
    
    func componentCColorDidChooseColor(_ view: NSView, _ color: NSColor?, _ refresh: Bool) {
        
        CPDFAnnotation.updateAnnotations(annotations, newColor: color, withPDFView: pdfView)
        CPDFTextAnnotation.updateDefaultColor(color)
        
        NotificationCenter.default.post(name: toolbarImageColorChangedNotificationName, object: nil)
        
        if refresh == true {
            reloadData()
        }
    }
    
    func componentCColorGroupColorsUpdates(_ view: NSView, _ colors: [NSColor]) {
        KMAnnotationPropertiesColorManager.manager.updateDefaultColors(colors, forKey: KMNoteColorsKey)
        
    }
    
    func componentCColorDidRightMouseUpWithStrings(_ view: NSView) -> [String] {
        return [KMLocalizedString("Change Color"), KMLocalizedString("Restore default color")]
    }
    
    func componentCColorDidRightMenuItemClicked(_ view: NSView, menuItemProperty: ComponentMenuitemProperty?) {
        if menuItemProperty?.text == KMLocalizedString("Change Color") {
            if NSColorPanel.sharedColorPanelExists {
                NSColorPanel.shared.setTarget(nil)
                NSColorPanel.shared.setAction(nil)
            }
            NSColorPanel.shared.setTarget(self)
            NSColorPanel.shared.setAction(#selector(colorChoose(_:)))
            if let colorItem = view as? ComponentCColorItem, let color = colorItem.properties?.color {
                NSColorPanel.shared.color = color
            }
            
            let viewRect = view.superview?.convert(view.frame, to: nil) ?? CGRectZero
            var rect = NSColorPanel.shared.frame
            rect.origin.x = viewRect.minX
            rect.origin.y = viewRect.minY - 30
            NSColorPanel.shared.setFrame(rect, display: true)
            
            NSColorPanel.shared.orderFront(nil)
        } else if menuItemProperty?.text == KMLocalizedString("Restore default color") {
            if let colorItem = colorGroup.rightClickedItem {
                let colors: [NSColor] = KMAnnotationPropertiesColorManager.manager.defaultColors(key: KMNoteColorsKey)
                if colors.count > 4, colorItem.itemIndex >= 0, colorItem.itemIndex < colors.count {
                    colorItem.properties?.color = colors[colorItem.itemIndex]
                    colorItem.reloadData()
                    colorGroup.componentCColorDidChoosePanelColor(colorGroup, nil)
                    
                    self.componentCColorDidChooseColor(colorGroup, colorItem.properties?.color, false)
                }
            }
        }
    }
    
    @objc func colorChoose(_ sender: Any) {
        if let colorItem = colorGroup.rightClickedItem {
            let color = NSColorPanel.shared.color
            colorItem.properties?.color = color
            
            colorItem.reloadData()
            
            colorGroup.componentCColorDidChoosePanelColor(colorGroup, nil)
            
            self.componentCColorDidChooseColor(colorGroup, color, false)
        }
    }
}