PDF_Office/PDF Office/PDF Master/Class/PDFWindowController/Side/RightSide/EditPDF/KMEditImagePropertyViewController.swift

//
//  KMEditImagePropertyViewController.swift
//  PDF Master
//
//  Created by lxy on 2022/12/28.
//

import Cocoa

public enum KMOperationState {
    case none
    case began
    case changed
    case ended
}

class KMEditImagePropertyViewController: NSViewController {

    @IBOutlet weak var titleLabel: NSTextField!
    @IBOutlet weak var editImageView: NSImageView!
    
    @IBOutlet weak var headerBox: NSBox!
    @IBOutlet weak var headerBoxHeight: NSLayoutConstraint!
    @IBOutlet weak var headerBoxMaginWidthConstraint: NSLayoutConstraint!
    
    @IBOutlet weak var opacityTitleLabel: NSTextField!
    @IBOutlet weak var opacitySlider: NSSlider!
    @IBOutlet weak var opacityBox: NSBox!
    @IBOutlet weak var opacityBoxBottomMaginConstraint: NSLayoutConstraint!
    
    @IBOutlet weak var cropBox: NSBox!
    @IBOutlet weak var cancelBox: NSBox!
    @IBOutlet weak var replaceBox: NSBox!
    @IBOutlet weak var exportBox: NSBox!
    @IBOutlet weak var cropBoxWidthConstraint: NSLayoutConstraint!
    
//    @IBOutlet weak var cutButton: NSButton!
//    @IBOutlet weak var replaceButton: NSButton!
//    @IBOutlet weak var exportButton: NSButton!
//    @IBOutlet weak var cancelCutButton: NSButton!
//    @IBOutlet weak var cutRightConstraint: NSLayoutConstraint!

    @IBOutlet weak var imageBox: NSBox!
    @IBOutlet weak var imageBoxHeight: NSLayoutConstraint!
    @IBOutlet weak var opacityBoxHeight: NSLayoutConstraint!
    @IBOutlet weak var buttonBoxHeight: NSLayoutConstraint!
    @IBOutlet weak var buttonBox: NSBox!
    @IBOutlet weak var rotateLeftBox: NSBox!
    @IBOutlet weak var rotateRightBox: NSBox!
    @IBOutlet weak var flipHorizontalBox: NSBox!
    @IBOutlet weak var flipVerticalBox: NSBox!
    
    var rotateLeftVC: KMDesignPropertySelector?
    var rotateRightVC: KMDesignPropertySelector?
    var flipHorizontalVC: KMDesignPropertySelector?
    var flipVerticalVC: KMDesignPropertySelector?
    var opacityVC: KMDesignSelect?
    
    var cropVC: KMDesignButton?
    var cancelVC: KMDesignButton?
    var replaceVC: KMDesignButton?
    var exportVC: KMDesignButton?
    
    @IBOutlet weak var alignmentView: KMEditPropertyAlignmentView!
    var imagesAreas : [CPDFEditImageArea] = []
    var listView : CPDFListView!
    
    var editingAreas: [Any] {
        get {
            return self.listView.editingAreas() ?? []
        }
    }
    
    override func viewDidLoad() {
        super.viewDidLoad()
    
        self.setup()
        self.initData()
        self.updateLanguage()
        self.updateButtonState(hidden: true)
        self.reloadData()
    }
    
    func initData() {
        self.updateButtonState(hidden: true)
    }
    
    func setup() {
        self.titleLabel.font = NSFont.SFProTextSemibold(14.0)
        self.titleLabel.textColor = NSColor(hex: "#252629")
        
        self.opacityTitleLabel.font = NSFont.SFProTextSemibold(12.0)
        self.opacityTitleLabel.textColor = NSColor(hex: "#616469")
        
        self.editImageView.wantsLayer = true
        self.editImageView.layer?.backgroundColor = NSColor.white.cgColor
        
        self.alignmentView.didChange = {[unowned self] view, areasArray, boundsArray in
            self.changeAreasAlign(areas: areasArray, newBounds: boundsArray)
        }
        
        imageBox.backgroundColor(NSColor(hex: "#FFFFFF"))
        
        opacityVC = KMDesignSelect.init(withType: .PopButton)
        opacityBox.contentView = opacityVC?.view
        opacityBox.fillColor = NSColor.clear
        opacityVC?.addItems(withObjectValues: ["25%","50%","75%","100%"])
        opacityVC?.selectItem(at: 0)
        opacityVC?.delete = self
        
        //alignment
        rotateLeftVC = KMDesignPropertySelector.init(withType: .Icon_Btn)
        rotateLeftBox.contentView = rotateLeftVC?.view
        rotateLeftBox.fillColor = NSColor.clear
        rotateLeftVC?.target = self
        rotateLeftVC?.action = #selector(leftRotationImageAction)
        rotateLeftVC?.image = NSImage(named: "KMImageNameEditPDFRotationLeft")!
        rotateLeftVC?._image_disabled = NSImage(named: "KMImageNameEditPDFRotationLeft")!
        rotateLeftVC?.updateUI()
        
        rotateRightVC = KMDesignPropertySelector.init(withType: .Icon_Btn)
        rotateRightBox.contentView = rotateRightVC?.view
        rotateRightBox.fillColor = NSColor.clear
        rotateRightVC?.target = self
        rotateRightVC?.action = #selector(rightRotationImageAction)
        rotateRightVC?.image = NSImage(named: "KMImageNameEditPDFRotationRight")!
        rotateRightVC?.updateUI()
        
        flipHorizontalVC = KMDesignPropertySelector.init(withType: .Icon_Btn)
        flipHorizontalBox.contentView = flipHorizontalVC?.view
        flipHorizontalBox.fillColor = NSColor.clear
        flipHorizontalVC?.target = self
        flipHorizontalVC?.action = #selector(flipHorizontalImageAction)
        flipHorizontalVC?.image = NSImage(named: "KMImageNameEditPDFFlipHorizontal")!
        flipHorizontalVC?.updateUI()
        
        flipVerticalVC = KMDesignPropertySelector.init(withType: .Icon_Btn)
        flipVerticalBox.contentView = flipVerticalVC?.view
        flipVerticalBox.fillColor = NSColor.clear
        flipVerticalVC?.target = self
        flipVerticalVC?.action = #selector(flipVerticalImageAction)
        flipVerticalVC?.image = NSImage(named: "KMImageNameEditPDFFlipVertical")!
        flipVerticalVC?.updateUI()
        
        // 设置默认状态下的滑块图像
        let customCell = CustomSliderCell()
        // 设置滑块图像
        customCell.knobImage = NSImage(named: "KMImageNameEditPDFSlider")
        // 应用自定义的NSSliderCell到NSSlider
        opacitySlider.cell = customCell
        // 添加滑动事件处理程序
        opacitySlider.target = self
        opacitySlider.action = #selector(sliderValueChanged(_:))
        
        cropVC = KMDesignButton.init(withType: .Text)
        cropBox.contentView = cropVC?.view
        cropBox.fillColor = NSColor.clear
        cropVC?.target = self
        cropVC?.action = #selector(cutImageAction)
        cropVC?.button(type: .Sec, size: .m)
        cropVC?.stringValue = NSLocalizedString("Crop", comment: "")
        cropVC?.updateUI()
        
        cancelVC = KMDesignButton.init(withType: .Text)
        cancelBox.contentView = cancelVC?.view
        cancelBox.fillColor = NSColor.clear
        cancelVC?.target = self
        cancelVC?.action = #selector(cancelCutImageAction)
        cancelVC?.button(type: .Sec, size: .m)
        cancelVC?.stringValue = NSLocalizedString("Cancel", comment: "")
        cancelVC?.updateUI()
        
        replaceVC = KMDesignButton.init(withType: .Text)
        replaceBox.contentView = replaceVC?.view
        replaceBox.fillColor = NSColor.clear
        replaceVC?.target = self
        replaceVC?.action = #selector(replaceImageAction)
        replaceVC?.button(type: .Sec, size: .m)
        replaceVC?.stringValue = NSLocalizedString("Replace", comment: "")
        replaceVC?.updateUI()
        
        exportVC = KMDesignButton.init(withType: .Text)
        exportBox.contentView = exportVC?.view
        exportBox.fillColor = NSColor.clear
        exportVC?.target = self
        exportVC?.action = #selector(exportImageAction)
        exportVC?.button(type: .Sec, size: .m)
        exportVC?.stringValue = NSLocalizedString("Extract", comment: "")
        exportVC?.updateUI()
    }
    
    func updateLanguage(){
        self.titleLabel.stringValue = NSLocalizedString("Image", comment: "")
        self.opacityTitleLabel.stringValue = NSLocalizedString("Opacity", comment: "")
        self.cropVC?.stringValue = NSLocalizedString("Crop", comment: "")
        self.cancelVC?.stringValue = NSLocalizedString("Cancel", comment: "")
        self.replaceVC?.stringValue = NSLocalizedString("Replace", comment: "")
        self.exportVC?.stringValue = NSLocalizedString("Extract", comment: "")
    }
    
    func reloadData() {
        imagesAreas = []
        if self.editingAreas.count > 0 {
            let areas = self.editingAreas
            self.alignmentView.editingAreas = areas
            
            for i in 0 ... areas.count-1 {
                if areas[i] is CPDFEditImageArea {
                    imagesAreas.append(areas[i] as! CPDFEditImageArea)
                }
            }
            if imagesAreas.count == 1 && imagesAreas.count == areas.count{  //单个图片
                self.listView.selectImageAreas = imagesAreas.first
                self.headerBox.isHidden = false
                self.headerBoxHeight.constant = 176
                self.headerBoxMaginWidthConstraint.constant = 8
                self.imageBox.isHidden = false
                self.imageBoxHeight.constant = 88
                self.opacityBox.isHidden = false
                self.opacityBoxHeight.constant = 56
                self.opacityBoxBottomMaginConstraint.constant = 16
                self.buttonBox.isHidden = false
                self.cropBox.isHidden = false
                self.cancelBox.isHidden = false
                self.replaceBox.isHidden = false
                self.buttonBoxHeight.constant = 112
                self.editImageView.image = self.listView.selectImageAreas.thumbnailImage

                let opacity: CGFloat = self.listView.opacity(for: imagesAreas.first)
                self.updateImageAreasOpacity(opacity: opacity, state: .ended, needListView: false)
            } else if imagesAreas.count > 1 && imagesAreas.count == areas.count { //多选图片
                self.headerBox.isHidden = false
                self.headerBoxHeight.constant = 82
                self.headerBoxMaginWidthConstraint.constant = 0
                self.imageBox.isHidden = true
                self.imageBoxHeight.constant = 0
                self.opacityBox.isHidden = false
                self.opacityBoxHeight.constant = 56
                self.opacityBoxBottomMaginConstraint.constant = 0
                self.buttonBox.isHidden = false
                self.cropBox.isHidden = true
                self.cancelBox.isHidden = true
                self.replaceBox.isHidden = true
                self.buttonBoxHeight.constant = 0
                var opacity = self.listView.opacity(for: imagesAreas.first)
                for area in imagesAreas {
                    let newOpacity = self.listView.opacity(for: area)
                    if (opacity < newOpacity) {
                        opacity = newOpacity
                    }
                }
                self.updateImageAreasOpacity(opacity: opacity, state: .ended, needListView: false)
            } else if imagesAreas.count > 0 && imagesAreas.count != areas.count { //多选图片跟文字
                self.headerBox.isHidden = true
                self.headerBoxHeight.constant = 0
                self.imageBox.isHidden = true
                self.imageBoxHeight.constant = 0
                self.opacityBox.isHidden = true
                self.opacityBoxHeight.constant = 0
                self.buttonBox.isHidden = true
                self.buttonBoxHeight.constant = 0
            }
        }
    }
    
    func updateButtonState(hidden:Bool) {
        if hidden {
            self.cancelBox.isHidden = true
            self.cropBoxWidthConstraint.constant = 0
            self.cropBox.title = NSLocalizedString("Crop", comment: "")
            self.replaceVC?.state = .Norm
            self.exportVC?.state = .Norm
            self.opacitySlider.isEnabled = true
            self.opacityVC?.state = .Norm
            self.rotateLeftVC?.state = .Norm
            self.rotateRightVC?.state = .Norm
            self.flipVerticalVC?.state = .Norm
            self.flipHorizontalVC?.state = .Norm
        } else {
            self.cancelBox.isHidden = false
            self.cropBoxWidthConstraint.constant = 120
            self.cropBox.title = NSLocalizedString("Confirm Cut", comment: "")
            self.replaceVC?.state = .Disabled
            self.exportVC?.state = .Disabled
            self.opacitySlider.isEnabled = false
            self.opacityVC?.state = .Disabled
            self.rotateLeftVC?.state = .Disabled
            self.rotateRightVC?.state = .Disabled
            self.flipVerticalVC?.state = .Disabled
            self.flipHorizontalVC?.state = .Disabled
        }
    }
    
    func updateImageAreasOpacity(opacity: CGFloat, state: KMOperationState, needListView: Bool = true) {
        if self.editingAreas.count >= 1 {
            if needListView {
                for area in imagesAreas {
                    self.listView.setOpacityEditArea(area, opacity: opacity)
                }
            }
            
//            self.listView.editingAreas()!.count == 1 &&
            if (self.listView.editingAreas()?.first is CPDFEditImageArea) {
                self.listView.selectImageAreas = self.listView.editingAreas()!.first as? CPDFEditImageArea
                self.editImageView.image = self.listView.selectImageAreas.thumbnailImage
                self.opacitySlider.objectValue = self.listView.opacity(for: self.listView.selectImageAreas)
                self.opacityVC?.stringValue = Int(opacity * 100).description + "%"
                self.editImageView.alphaValue = opacity
            }
        }
    }
    
    private func changeAreasAlign(areas:[Any],newBounds:[String]) {
        var oldBounds : [String] = []
        for i in 0 ... areas.count-1 {
            let area : CPDFEditArea = areas[i] as! CPDFEditArea
            oldBounds.append(NSStringFromRect(area.bounds))
            self.listView.setBoundsEditArea(area, withBounds: NSRectFromString(newBounds[i]))
        }
        self.listView.setNeedsDisplayForVisiblePages()
    }
}

extension KMEditImagePropertyViewController: KMSelectPopButtonDelegate {
    func km_comboBoxSelectionDidChange(_ obj: KMDesignSelect) {
        if obj == opacityVC {
            print("km_comboBoxSelectionDidChange")
            let index = obj.indexOfSelectedItem
            var string = obj.items[index]
            string = string.replacingOccurrences(of: "%", with: "")
            let value = CGFloat(Float(string)! * 0.01)
            self.updateImageAreasOpacity(opacity: value, state: .changed)
        }
    }
    
    func km_controlTextDidEndEditing(_ obj: KMDesignSelect) {
        if obj == opacityVC {
            print("km_comboBoxSelectionDidChange")
            let index = obj.indexOfSelectedItem
            var string = obj.items[index]
            string = string.replacingOccurrences(of: "%", with: "")
            let value = CGFloat(Float(string)! * 0.01)
            self.updateImageAreasOpacity(opacity: value, state: .ended)
        }
    }
}


//MARK: - Action
extension KMEditImagePropertyViewController {
    @objc func sliderValueChanged(_ sender: NSSlider) {
        let sliderValue = sender.floatValue
        self.updateImageAreasOpacity(opacity:CGFloat(sliderValue), state: .changed)
    }
    
    @IBAction func rightRotationImageAction(_ sender: Any) {
        if self.listView.editingAreas()?.count ?? 0 < 1 {
            return
        }
        let areas = self.listView.editingAreas()
        if areas!.count == 1 && (areas!.first is CPDFEditImageArea) {
            let imageArea = areas!.first as! CPDFEditImageArea
            self.listView.rotate(with: imageArea, rotate: 90)
            self.listView.selectImageAreas = imageArea
            self.editImageView.image = self.listView.selectImageAreas.thumbnailImage
        } else if areas!.count > 1 {
            for imagesArea in imagesAreas {
                self.listView.rotate(with: imagesArea, rotate: 90)
            }
        }
    }
    
    @IBAction func leftRotationImageAction(_ sender: Any) {
        if self.listView.editingAreas()?.count ?? 0 < 1 {
            return
        }
        let areas = self.listView.editingAreas()
        if areas!.count == 1 && (areas!.first is CPDFEditImageArea) {
            self.listView.rotate(with: self.listView.selectImageAreas, rotate: -90)
            if self.listView.editingAreas()!.count == 1 && (self.listView.editingAreas()!.first is CPDFEditImageArea) {
                self.listView.selectImageAreas = self.listView.editingAreas()!.first as? CPDFEditImageArea
            }
            self.editImageView.image = self.listView.selectImageAreas.thumbnailImage
        } else {
            for imagesArea in imagesAreas {
                self.listView.rotate(with: imagesArea, rotate: -90)
            }
        }
    }
    
    @IBAction func flipHorizontalImageAction(_ sender: Any) {
        if self.listView.editingAreas()?.count ?? 0 < 1 {
            return
        }
        let areas = self.listView.editingAreas()
        if areas != nil {
            for item in areas! {
                if item is CPDFEditImageArea {
                    self.listView.horizontalMirror(with: item as? CPDFEditImageArea)
                    if areas?.count == 1 {
                        self.editImageView.image = (item as AnyObject).thumbnailImage
                    }
                } else {
                    for imagesArea in imagesAreas {
                        self.listView.horizontalMirror(with: imagesArea)
                    }
                }
            }
        }
    }
    
    @IBAction func flipVerticalImageAction(_ sender: Any) {
        if self.listView.editingAreas()?.count ?? 0 < 1 {
            return
        }
        let areas = self.listView.editingAreas()
        if areas != nil {
            for item in areas! {
                if item is CPDFEditImageArea {
                    self.listView.verticalMirror(with: item as? CPDFEditImageArea)
                    self.editImageView.image = self.listView.selectImageAreas.thumbnailImage
                } else {
                    for imagesArea in imagesAreas {
                        self.listView.verticalMirror(with: imagesArea)
                    }
                }
            }
        }
    }
    
    @IBAction func cutImageAction(_ sender: Any) {
        if self.listView.selectImageAreas == nil {
            return
        }
        if self.listView.cropAreas != nil && self.listView.selectImageAreas != nil{
            self.listView.cropEditImageArea(self.listView.selectImageAreas, withBounds: self.listView.cropAreas.cropRect)
        } else if self.listView.selectImageAreas != nil {
            self.listView.isEditImage = true
            self.listView.enterCrop(with: self.listView.selectImageAreas)
            self.updateButtonState(hidden: false)
        }
    }
    
    @IBAction func cancelCutImageAction(_ sender: Any) {
        if self.listView.selectImageAreas == nil {
            return
        }
        self.listView.exitCrop(with: self.listView.selectImageAreas)
        self.listView.cropAreas = nil
        self.listView.isEditImage = false
        self.updateButtonState(hidden: true)
    }
    
    @IBAction func replaceImageAction(_ sender: NSButton) {
        if self.listView.selectImageAreas == nil {
            return
        }
        let panel = NSOpenPanel()
        panel.allowsMultipleSelection = false
        panel.allowedFileTypes = ["png","jpg"]
        panel.beginSheetModal(for: NSApp.mainWindow!) { response in
            if response == .OK {
                let openPath = panel.url?.path
//                let s = self.listView.extractImage(with: self.selectImageAreas, toImagePath: openPath!)
                let s = self.listView.replace(self.listView.selectImageAreas, imagePath: openPath!)
                if s {
                    
                }
            }
        }
    }
    
//    @available(macOS 13.0, *)
    @IBAction func exportImageAction(_ sender: Any) {
        if self.listView.selectImageAreas == nil {
            return
        }
        if imagesAreas.count == 1 {
            let panel = NSSavePanel()
            panel.nameFieldStringValue = "\(NSLocalizedString("Untitled", comment: "")).png"
            let button = NSButton.init(checkboxWithTitle: NSLocalizedString("Open the document after saving", comment: ""), target: nil, action: nil)
            button.state = .on
            panel.accessoryView = button
            panel.isExtensionHidden = true
            let response = panel.runModal()
            if response == .OK {
                let url = panel.url
                let result = self.listView.extractImage(with: self.listView.selectImageAreas, toImagePath: url!.path)
                if result {
                    if button.state == .on { /// 开启文档
                        NSWorkspace.shared.openFile(url!.path)
                    } else {
                        
                    }
                }
            }
        } else if imagesAreas.count > 1 {
            let panel = NSOpenPanel()
            panel.canChooseFiles = false
            panel.canChooseDirectories = true
            panel.canCreateDirectories = true
            panel.allowsMultipleSelection = false
            panel.beginSheetModal(for: NSApp.mainWindow!) { response in
                if response == .OK {
                    let outputURL = panel.url
                    let s = self.listView.document.documentURL.lastPathComponent
                    let folderPath = self.listView.document.documentURL.deletingPathExtension().lastPathComponent + "_extract"
                    var filePath = outputURL?.path.stringByAppendingPathComponent(folderPath)
                    
                    var i = 1
                    let testFilePath = filePath
                    while FileManager.default.fileExists(atPath: filePath!) {
                        filePath = testFilePath! + "\(i)"
                        i = i + 1
                    }
                    try? FileManager.default.createDirectory(atPath: filePath!, withIntermediateDirectories: false, attributes: nil)
                    var saveURLs : [URL] = []
                    for j in 0 ... self.imagesAreas.count-1 {
                        let documentFileName = self.listView.document.documentURL.deletingPathExtension().lastPathComponent
                        var outPath = filePath!
                        outPath = outPath.stringByAppendingPathComponent(documentFileName)
                        outPath = outPath + "page \(j+1)"
                        outPath = outPath.stringByAppendingPathExtension("png")
                        let result = self.listView.extractImage(with: self.imagesAreas[j], toImagePath: outPath)
                        if result {
                            saveURLs.append(URL(fileURLWithPath: outPath))
                        }
                    }
                    NSWorkspace.shared.activateFileViewerSelecting(saveURLs)
                }
            }
        }
      
    }
}

class CustomSliderCell: NSSliderCell {
    var knobImage: NSImage?
    
    override func drawKnob(_ knobRect: NSRect) {
        if let image = knobImage {
            let imageSize = NSSize(width: knobRect.size.height, height: knobRect.size.height)
            let imageRect = NSRect(origin: knobRect.origin, size: imageSize)
            image.draw(in: imageRect)
        } else {
            super.drawKnob(knobRect)
        }
    }
}