PDF_Office/PDF Office/PDF Master/KMClass/KMPDFViewController/RightSideController/Views/Line/KMLineController.swift

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

import Cocoa
import KMComponentLibrary

class KMLineController: NSViewController {
    //Color
    @IBOutlet var colorBGView: NSView!
    @IBOutlet var colorLabel: NSTextField!
    @IBOutlet var colorGroup: ComponentCColorGroup!
    @IBOutlet var colorSlider: ComponentSlider!
    @IBOutlet var colorOpacitySelect: ComponentSelect!
    
    //Line
    @IBOutlet var lineBGView: NSView!
    @IBOutlet var lineLabel: NSTextField!
    @IBOutlet var lineTypeSelector: ComponentCSelectorGroup!
    @IBOutlet var lineWidthSlider: ComponentSlider!
    @IBOutlet var lineWidthSelect: ComponentSelect!
    @IBOutlet var linedashInfoView: NSView!
    @IBOutlet var lineDashSlider: ComponentSlider!
    @IBOutlet var lineDashSelect: ComponentSelect!
    
    @IBOutlet var arrowTypeView: NSView!
    @IBOutlet var arrowTypeTopConst: NSLayoutConstraint!
    @IBOutlet var arrow_leftTypeSelect: ComponentSelect!
    @IBOutlet var arrow_RightTypeSelect: ComponentSelect!
     
    private let solidProperty = ComponentCSelectorProperty.init(size: .s, state: .normal, text: "", iconImage: NSImage(named: "lineStyle_solid"))
    private let dashProperty = ComponentCSelectorProperty.init(size: .s, state: .normal, text: "", iconImage: NSImage(named: "lineStyle_dash"))
    
    private var annotations: [CPDFLineAnnotation] = []
    
    var pdfView: CPDFListView?
    var viewManager: KMPDFViewManager?
    
    //MARK: - func
    override func viewDidAppear() {
        super.viewDidAppear()
        
        
        colorSlider.reloadData()
        
        lineWidthSlider.reloadData()
        
        lineDashSlider.reloadData()
    }
    
    override func viewDidLoad() {
        super.viewDidLoad()
        // Do view setup here.
        
        setupProperty()
    }
    
    
    func getValidAnnotations() -> [CPDFLineAnnotation] {
        return annotations
    }
    
    func setupProperty() {
        
        //Color
        colorLabel.stringValue = KMLocalizedString("Color")
        colorLabel.textColor = ComponentLibrary.shared.getComponentColorFromKey("colorText/2")
        colorLabel.font = ComponentLibrary.shared.getFontFromKey("mac/body-s-medium")
        
        colorGroup.delegate = self
        
        colorSlider.properties = ComponentSliderProperty(size: .m, percent: 1)
        colorSlider.delegate = self
        
        colorOpacitySelect.properties = ComponentSelectProperties(size: .s,
                                                                  state: .normal,
                                                                  creatable: true,
                                                                  text: "100%",
                                                                  textUnit: "%",
                                                                  regexString: "0123456789%")
        if true {
            var opacityItems: [ComponentMenuitemProperty] = []
            for string in ["25%", "50%", "75%", "100%"] {
                let item = ComponentMenuitemProperty(type: .normal, text: string)
                opacityItems.append(item)
            }
            colorOpacitySelect.updateMenuItemsArr(opacityItems)
        }
        colorOpacitySelect.delegate = self
        
        //Line
        lineLabel.stringValue = KMLocalizedString("Line")
        lineLabel.textColor = ComponentLibrary.shared.getComponentColorFromKey("colorText/2")
        lineLabel.font = ComponentLibrary.shared.getFontFromKey("mac/body-s-medium")
        
        lineTypeSelector.updateItemProperty([solidProperty, dashProperty])
        lineTypeSelector.delegate = self
        
        lineWidthSlider.properties = ComponentSliderProperty(size: .m, percent: 1)
        lineWidthSlider.delegate = self
        
        lineWidthSelect.properties = ComponentSelectProperties(size: .s,
                                                               state: .normal,
                                                               creatable: true,
                                                               text: "2",
                                                               textUnit: " pt",
                                                               regexString: "0123456789 pt")
        if true {
            var opacityItems: [ComponentMenuitemProperty] = []
            for string in ["1 pt", "3 pt", "6 pt", "9 pt", "12 pt", "15 pt", "18 pt"] {
                let item = ComponentMenuitemProperty(type: .normal, text: string)
                opacityItems.append(item)
            }
            lineWidthSelect.updateMenuItemsArr(opacityItems)
        }
        lineWidthSelect.delegate = self
        
        lineDashSlider.properties = ComponentSliderProperty(size: .m, percent: 1)
        lineDashSlider.delegate = self
        
        lineDashSelect.properties = ComponentSelectProperties(size: .s,
                                                              state: .normal,
                                                              creatable: true,
                                                              text: "2",
                                                              textUnit: " pt",
                                                              regexString: "0123456789 pt")
        if true {
            var opacityItems: [ComponentMenuitemProperty] = []
            for string in ["1 pt", "3 pt", "6 pt", "9 pt", "12 pt", "15 pt", "18 pt"] {
                let item = ComponentMenuitemProperty(type: .normal, text: string)
                opacityItems.append(item)
            }
            lineDashSelect.updateMenuItemsArr(opacityItems)
        }
        lineDashSelect.delegate = self
        
        //ArrowType
        arrow_leftTypeSelect.properties = ComponentSelectProperties(size: .s,
                                                              state: .normal,
                                                              creatable: false)
        if true {
            let stringItems = ["None", "Arrow Open", "Arrow Close", "Square", "Oval", "Diamond", "Butt", "Slash", "R Open Arrow", "R Close Arrow"]
            let imageNames = ["arrowType_line", "arrowType_left_Arrow_open", "arrowType_left_Arrow_close",
                              "arrowType_left_Square", "arrowType_left_Oval", "arrowType_left_Diamond",
                              "arrowType_left_Butt", "arrowType_left_Slash", "arrowType_left_Open_arrow",
                              "arrowType_left_Close_arrow"]
            
            var opacityItems: [ComponentMenuitemProperty] = []
            for i in 0...stringItems.count-1 {
                let string = stringItems[i]
                let imageName = imageNames[i]
                
                let item = ComponentMenuitemProperty(type: .normal, lefticon: NSImage(named: imageName), text: string, identifier: string)
                opacityItems.append(item)
            }
            arrow_leftTypeSelect.updateMenuItemsArr(opacityItems)
        }
        arrow_leftTypeSelect.delegate = self
        
        arrow_RightTypeSelect.properties = ComponentSelectProperties(size: .s,
                                                              state: .normal,
                                                              creatable: false)
        if true {
            let stringItems = ["None", "Arrow Open", "Arrow Close", "Square", "Oval", "Diamond", "Butt", "Slash", "R Open Arrow", "R Close Arrow"]
            let imageNames = ["arrowType_line", "arrowType_right_Arrow_open", "arrowType_right_Arrow_close",
                              "arrowType_right_Square", "arrowType_right_Oval", "arrowType_right_Diamond",
                              "arrowType_right_Butt", "arrowType_right_Slash", "arrowType_right_Open_arrow",
                              "arrowType_right_Close_arrow"]
            
            var opacityItems: [ComponentMenuitemProperty] = []
            for i in 0...stringItems.count-1 {
                let string = stringItems[i]
                let imageName = imageNames[i]
                
                let item = ComponentMenuitemProperty(type: .normal, lefticon: NSImage(named: imageName), text: string, identifier: string)
                opacityItems.append(item)
            }
            arrow_RightTypeSelect.updateMenuItemsArr(opacityItems)
        }
        arrow_RightTypeSelect.delegate = self
        
    }
    
    func reloadData() {
        guard let pdfView = self.pdfView else {
            return
        }
        
        var firstAnnotation: CPDFLineAnnotation?
        
        self.annotations.removeAll()
        let allAnnotations: [CPDFAnnotation] = pdfView.activeAnnotations as? [CPDFAnnotation] ?? []
        for annotation in allAnnotations {
            if annotation is CPDFLineAnnotation {
                annotations.append((annotation as! CPDFLineAnnotation))
            }
        }
        
        if annotations.count > 0 {
            firstAnnotation = annotations.first
        }
        
        //Color
        var colors: [NSColor] = []
        if let annotation = firstAnnotation {
            if annotation.isArrowLine() == true {
                colors = KMAnnotationPropertiesColorManager.manager.arrowColors
            } else {
                colors = KMAnnotationPropertiesColorManager.manager.lineColors
            }
        } else if viewManager?.subToolMode == .Line {
            colors = KMAnnotationPropertiesColorManager.manager.lineColors
        } else if viewManager?.subToolMode == .Arrow {
            colors = KMAnnotationPropertiesColorManager.manager.arrowColors
        }
        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)
        }
         
        var startLineStyle = CPDFLineAnnotation.defaultArrow_StartLineStyle()
        var endLineStyle = CPDFLineAnnotation.defaultArrow_EndLineStyle()
        
        if annotations.count == 0 {
            var annotationType: CAnnotationType = .line
            if viewManager?.subToolMode == .Arrow {
                annotationType = .arrow
            }
            
            colorGroup.currentColor = CPDFLineAnnotation.defaultColor(annotationType)
            colorGroup.refreshUI()
            
            let opacity = CPDFLineAnnotation.defaultOpacity(annotationType)
            colorSlider.properties.percent = opacity
            colorSlider.reloadData()
            
            colorOpacitySelect.properties.text = String(format: "%.0f%@", opacity*100, "%")
            colorOpacitySelect.reloadData()
            
            let border: CPDFBorder = CPDFLineAnnotation.defaultBorder(annotationType)
            
            dashProperty.state = .normal
            solidProperty.state = .normal
            if border.style == .dashed {
                dashProperty.state = .pressed
            } else if border.style == .solid {
                solidProperty.state = .pressed
            }
            lineTypeSelector.reloadData()
            
            let percent = (border.lineWidth - 1)/17
            lineWidthSlider.properties.percent = percent
            lineWidthSlider.reloadData()
            
            lineWidthSelect.properties.text = String(format: "%.0f%@", border.lineWidth, " pt")
            lineWidthSelect.reloadData()
            
            linedashInfoView.isHidden = true
            if border.style == .dashed {
                linedashInfoView.isHidden = false
                
                var dash = 1.0
                for dashPattern in border.dashPattern {
                    if let value = dashPattern as? CGFloat {
                        dash = value
                        break
                    }
                }
                let percent: CGFloat = (CGFloat(dash) - 1)/17
                lineDashSlider.properties.percent = percent
                lineDashSlider.reloadData()
                
                lineDashSelect.properties.text = String(format: "%.0f%@", CGFloat(dash), " pt")
                lineDashSelect.reloadData()
            }
        } else if annotations.count == 1, let annotation = firstAnnotation {
            colorGroup.currentColor = annotation.color
            colorGroup.refreshUI()
            
            let opacity = annotation.opacity
            colorSlider.properties.percent = opacity
            colorSlider.reloadData()
            
            colorOpacitySelect.properties.text = String(format: "%.0f%@", opacity*100, "%")
            colorOpacitySelect.reloadData()
            
            let border: CPDFBorder = annotation.border ?? CPDFBorder()
            
            dashProperty.state = .normal
            solidProperty.state = .normal
            if border.style == .dashed {
                dashProperty.state = .pressed
            } else if border.style == .solid {
                solidProperty.state = .pressed
            }
            lineTypeSelector.reloadData()
            
            let percent = (border.lineWidth - 1)/17
            lineWidthSlider.properties.percent = percent
            lineWidthSlider.reloadData()
            
            lineWidthSelect.properties.text = String(format: "%.0f%@", border.lineWidth, " pt")
            lineWidthSelect.reloadData()
            
            linedashInfoView.isHidden = true
            if border.style == .dashed {
                linedashInfoView.isHidden = false
                
                var dash = 1.0
                for dashPattern in border.dashPattern {
                    if let value = dashPattern as? CGFloat {
                        dash = value
                        break
                    }
                }
                let percent: CGFloat = (CGFloat(dash) - 1)/17
                lineDashSlider.properties.percent = percent
                lineDashSlider.reloadData()
                
                lineDashSelect.properties.text = String(format: "%.0f%@", CGFloat(dash), " pt")
                lineDashSelect.reloadData()
            }
            
            startLineStyle = annotation.startLineStyle
            endLineStyle = annotation.endLineStyle
            
        } else {
            guard let annotation = firstAnnotation else {
                return
            }
            if true {
                let multiColor: Bool = CPDFListView.isAnnotationsContainMultiType(annotations, withType: .color)
                if multiColor == true {
                    colorGroup.currentColor = NSColor.clear
                } else {
                    colorGroup.currentColor = annotation.color
                }
                colorGroup.refreshUI()
            }
            
            if true {
                let multiOpacity: Bool = CPDFListView.isAnnotationsContainMultiType(annotations, withType: .opacity)
                if multiOpacity {
                    colorSlider.properties.percent = 0
                    colorSlider.reloadData()
                    
                    colorOpacitySelect.resetText("-")
                } else {
                    let opacity = annotation.opacity
                    
                    colorSlider.properties.percent = opacity
                    colorSlider.reloadData()
                    
                    colorOpacitySelect.properties.text = String(format: "%.0f%@", opacity*100, "%")
                    colorOpacitySelect.reloadData()
                }
            }
            
            if true {
                let multiStyle: Bool = CPDFListView.isAnnotationsContainMultiType(annotations, withType: .border_Style)
                
                linedashInfoView.isHidden = true
                if multiStyle {
                    dashProperty.state = .normal
                    solidProperty.state = .normal
                    
                    lineTypeSelector.reloadData()
                } else {
                    let style = annotation.border?.style
                    dashProperty.state = .normal
                    solidProperty.state = .normal
                    if style == .dashed {
                        dashProperty.state = .pressed
                    } else if style == .solid {
                        solidProperty.state = .pressed
                    }
                    lineTypeSelector.reloadData()
                    
                    if style == .dashed {
                        linedashInfoView.isHidden = false
                    }
                }
            }
            
            if true {
                let multiLineWidth: Bool = CPDFListView.isAnnotationsContainMultiType(annotations, withType: .line_Width)
                if multiLineWidth {
                    lineWidthSlider.properties.percent = 0
                    lineWidthSlider.reloadData()
                    
                    lineWidthSelect.resetText("-")
                } else {
                    let border: CPDFBorder = annotation.border ?? CPDFBorder()
                    
                    let percent = (border.lineWidth - 1)/17
                    lineWidthSlider.properties.percent = percent
                    lineWidthSlider.reloadData()
                    
                    lineWidthSelect.properties.text = String(format: "%.0f%@", border.lineWidth, " pt")
                    lineWidthSelect.reloadData()
                }
            }
            
            if true {
                let multiLineDash: Bool = CPDFListView.isAnnotationsContainMultiType(annotations, withType: .dash_Pattern)
                if multiLineDash {
                    lineDashSlider.properties.percent = 0
                    lineDashSlider.reloadData()
                    
                    lineDashSelect.resetText("-")
                } else {
                    var dashA = 1.0
                    for dashPattern in annotation.border?.dashPattern ?? [] {
                        if let value = dashPattern as? CGFloat {
                            dashA = value
                            break
                        }
                    }
                    
                    let percent = (dashA - 1)/17
                    lineDashSlider.properties.percent = percent
                    lineDashSlider.reloadData()
                    
                    lineDashSelect.properties.text = String(format: "%.0f%@", dashA, " pt")
                    lineDashSelect.reloadData()
                }
            }
            
            startLineStyle = annotation.startLineStyle
            endLineStyle = annotation.endLineStyle
         }
         
        //ArrowType
        arrowTypeView.isHidden = true
        if let annotation = firstAnnotation, annotation.isArrowLine() == true {
            arrowTypeView.isHidden = false
        } else if viewManager?.subToolMode == .Arrow {
            arrowTypeView.isHidden = false
        }
        
        if linedashInfoView.isHidden {
            arrowTypeTopConst.constant = 8
        } else {
            arrowTypeTopConst.constant = 48
        }
         
        arrow_leftTypeSelect.selectItemAtIndex(startLineStyle.rawValue)
        
        arrow_RightTypeSelect.selectItemAtIndex(endLineStyle.rawValue)
        
    }
    
    //MARK: - Mouse
    override func mouseDown(with event: NSEvent) {
        super.mouseDown(with: event)
        
        view.window?.makeFirstResponder(nil)
    }
    
}

//MARK: - ComponentCColorDelegate
extension KMLineController: ComponentCColorDelegate {
    func componentCColorDidChooseColor(_ view: NSView, _ color: NSColor?) {
        CPDFLineAnnotation.updateColor(annotations, color, withPDFView: pdfView, isArrow: viewManager?.subToolMode == .Arrow)
        
        reloadData()
    }
}

//MARK: - ComponentSliderDelegate
extension KMLineController: ComponentSliderDelegate {
    func componentSliderDidUpdate(_ view: ComponentSlider) {
        if view == colorSlider {
            let percent = view.properties.percent
            CPDFLineAnnotation.updateOpacity(annotations, percent, withPDFView: pdfView, isArrow: viewManager?.subToolMode == .Arrow)
        } else if view == lineWidthSlider {
            let percent = view.properties.percent * 17 + 1
            CPDFLineAnnotation.updateLineWidth(annotations, percent, withPDFView: pdfView, isArrow: viewManager?.subToolMode == .Arrow)
        } else if view == lineDashSlider {
            let percent = view.properties.percent * 17 + 1
            CPDFLineAnnotation.updateDashPattern(annotations, percent, withPDFView: pdfView, isArrow: viewManager?.subToolMode == .Arrow)
        }
        reloadData()
    }
}

//MARK: - ComponentSelectDelegate
extension KMLineController: ComponentSelectDelegate {
    func componentSelectTextDidEndEditing(_ view: ComponentSelect, removeUnit text: String?) {
        if let result = text {
            if view == colorOpacitySelect {
                let opacity = max(0, min(1, result.stringToCGFloat()/100))
                CPDFLineAnnotation.updateOpacity(annotations, opacity, withPDFView: pdfView, isArrow: viewManager?.subToolMode == .Arrow)
            } else if view == lineWidthSelect {
                var value = result.stringToCGFloat()
                if value > 18 {
                    value = 18
                } else if value < 1 {
                    value = 1
                }
                CPDFLineAnnotation.updateLineWidth(annotations, value, withPDFView: pdfView, isArrow: viewManager?.subToolMode == .Arrow)
            } else if view == lineDashSelect {
                var value = result.stringToCGFloat()
                if value > 18 {
                    value = 18
                } else if value < 1 {
                    value = 1
                }
                CPDFLineAnnotation.updateDashPattern(annotations, value, withPDFView: pdfView, isArrow: viewManager?.subToolMode == .Arrow)
            }
            reloadData()
        }
    }
    
    func componentSelectDidSelect(view: ComponentSelect?, menuItemProperty: ComponentMenuitemProperty?) {
        if var result = menuItemProperty?.text {
            if let textUnit = view?.properties.textUnit {
                result = result.stringByDeleteCharString(textUnit)
            }
            if view == colorOpacitySelect {
                let opacity = max(0, min(1, result.stringToCGFloat()/100))
                CPDFLineAnnotation.updateOpacity(annotations, opacity, withPDFView: pdfView, isArrow: viewManager?.subToolMode == .Arrow)
            } else if view == lineWidthSelect {
                var value = result.stringToCGFloat()
                if value > 18 {
                    value = 18
                } else if value < 1 {
                    value = 1
                }
                CPDFLineAnnotation.updateLineWidth(annotations, value, withPDFView: pdfView, isArrow: viewManager?.subToolMode == .Arrow)
            } else if view == lineDashSelect {
                var value = result.stringToCGFloat()
                if value > 18 {
                    value = 18
                } else if value < 1 {
                    value = 1
                }
                CPDFLineAnnotation.updateDashPattern(annotations, value, withPDFView: pdfView, isArrow: viewManager?.subToolMode == .Arrow)
            } else if view == arrow_leftTypeSelect {
                let idx = view?.indexOfSelect() ?? 0
                CPDFLineAnnotation.updateStartLineStyle(annotations, CPDFLineStyle(rawValue: idx) ?? .none, withPDFView: pdfView)
                
            } else if view == arrow_RightTypeSelect {
                let idx = view?.indexOfSelect() ?? 0
                CPDFLineAnnotation.updateEndLineStyle(annotations, CPDFLineStyle(rawValue: idx) ?? .none, withPDFView: pdfView)
                
            }
            reloadData()
        }
    }
}


//MARK: - ComponentCSelectorGroupDelegate
extension KMLineController: ComponentCSelectorGroupDelegate {
    func componentCSelectorGroupDidChoose(_ view: ComponentCSelectorGroup, _ item: ComponentCSelectorItem) {
        if item.properties == solidProperty {
            CPDFLineAnnotation.updateStyle(annotations, .solid, withPDFView: pdfView, isArrow: viewManager?.subToolMode == .Arrow)
        } else if item.properties == dashProperty {
            CPDFLineAnnotation.updateStyle(annotations, .dashed, withPDFView: pdfView, isArrow: viewManager?.subToolMode == .Arrow)
        }
        reloadData()
    }
}