PDF_Office/PDF Office/PDF Master/Class/PDFTools/TTS/WindowController/KMTTSWindowController.swift

//
//  KMTTSWindowController.swift
//  PDF Reader Pro
//
//  Created by liujiajie on 2023/12/12.
//

import Cocoa
import KMComponentLibrary

typealias TTSCloseWindowCallback = (_ isCloseWindow: Bool) -> Void

let  minSpeed: Float = 0.6
let  maxSpeed: Float = 5.0
let  standardSpeed: Float = 175.0

class KMTTSWindowController: NSWindowController, NSWindowDelegate{
    
    @IBOutlet var titleLabel: NSTextField!
    @IBOutlet var previousButton: ComponentButton!
    @IBOutlet var playButton: ComponentButton!
    @IBOutlet var nextButton: ComponentButton!
    @IBOutlet var divider: ComponentDivider!
    @IBOutlet var languageLabel: NSTextField!
    @IBOutlet var languageSelect: ComponentSelect!
    @IBOutlet var speedLabel: NSTextField!
    @IBOutlet var speedSlider: ComponentSlider!
    @IBOutlet var speedInput: ComponentInputNumber!
    @IBOutlet var continueCheckbox: ComponentCheckBox!
    
    
    
    
    //    @IBOutlet var sontinuouButton: NSButton!
    //    @IBOutlet var speedLabel: NSTextField!
    //    @IBOutlet var languageLabel: NSTextField!
    //
    //    @IBOutlet var languageComboBox: NSPopUpButton!
    //    @IBOutlet var speedTextField: NSTextField!
    //    @IBOutlet var speedSlider: NSSlider!
    //
    //    @IBOutlet var speedStepper: NSStepper!
    //    @IBOutlet var nextButton: KMToolbarItem!
    //    @IBOutlet var forwardButton: KMToolbarItem!
    //
    //    @IBOutlet var playButton: KMToolbarItem!
    //
    //    @IBOutlet var speedBox: NSBox!
    //
    //    @IBOutlet var sppendCountLabel: NSTextField!
    
    var pdfView: CPDFView!
    var closeWindowCallback: TTSCloseWindowCallback?
    
    var pdfSelection: CPDFSelection?
    
    var currentPageIndex: Int = 0
    
    var isChangePage = false
    
    var voiceArrays = NSMutableArray()
    
    static let share = KMTTSWindowController()
    
    convenience init() {
        self.init(windowNibName: "KMTTSWindowController")
    }
    
    deinit {
        DistributedNotificationCenter.default().removeObserver(self)
        NotificationCenter.default.removeObserver(self)
    }
    
    override func windowDidLoad() {
        super.windowDidLoad()
        
        setUpproperty()
    }
     
    func setUpproperty() {
        
        titleLabel.stringValue = KMLocalizedString("TTS")
        titleLabel.textColor = ComponentLibrary.shared.getComponentColorFromKey("colorText/1")
        titleLabel.font = ComponentLibrary.shared.getFontFromKey("mac/body-s-medium")
        
        previousButton.properties = ComponentButtonProperty(type: .text_gray, size: .s, onlyIcon: true, icon: NSImage(named: "previous_fill"))
        playButton.properties = ComponentButtonProperty(type: .text_gray, size: .s, onlyIcon: true, icon: NSImage(named: "pause_fill"))
        nextButton.properties = ComponentButtonProperty(type: .text_gray, size: .s, onlyIcon: true, icon: NSImage(named: "next_fill"))
        
        previousButton.setTarget(self, action: #selector(buttonItemClick_Forward(_:)))
        playButton.setTarget(self, action: #selector(buttonItemClick_Play(_:)))
        nextButton.setTarget(self, action: #selector(buttonItemClick_Next(_:)))
        
        languageLabel.stringValue = KMLocalizedString("Language:")
        languageLabel.textColor = ComponentLibrary.shared.getComponentColorFromKey("comp-form/colorText-label")
        languageLabel.font = ComponentLibrary.shared.getFontFromKey("mac/body-s-regular")
        
        languageSelect.properties = ComponentSelectProperties(size: .s,
                                                              state: .normal,
                                                              creatable: false,
                                                              text: "English (Albert)",
                                                              textUnit: "")
        languageSelect.delegate = self
        
        speedLabel.stringValue = KMLocalizedString("Speed:")
        speedLabel.textColor = ComponentLibrary.shared.getComponentColorFromKey("comp-form/colorText-label")
        speedLabel.font = ComponentLibrary.shared.getFontFromKey("mac/body-s-regular")
        
        speedSlider.properties = ComponentSliderProperty(size: .m, percent: 1)
        speedSlider.delegate = self
         
        speedInput.properties = ComponentInputNumberProperty(alignment: .left,
                                                             size: .s,
                                                             state: .normal,
                                                             isError: false,
                                                             showErrorInfo: false,
                                                             isDisabled: false,
                                                             showPrefix: false,
                                                             showSuffix: false,
                                                             minSize: 1,
                                                             maxSize: 11,
                                                             text:"1",
                                                             valueType: .floatType)
        speedInput.delegate = self
 
        continueCheckbox.properties = ComponentCheckBoxProperty(size: .s,
                                                                state: .normal,
                                                                isDisabled: false,
                                                                showhelp: false,
                                                                text: KMLocalizedString("Continuous Reading"),
                                                                checkboxType: .normal)
        
        
        let array = KMTTSManager.defalutManager.availableVoices()
        let currentVoicName = KMTTSManager.defalutManager.voice()
        var currentIndex = 0
        let menu = NSMenu()
        
        for voiceType: NSSpeechSynthesizer.VoiceName in array {
            let voiceDic = NSMutableDictionary()
            let voiceLocaleIdentifier = ( NSSpeechSynthesizer.attributes( forVoice: voiceType )[ NSSpeechSynthesizer.VoiceAttributeKey.localeIdentifier ] as! String )
            
            let name = self.switchLanguage(withCode: voiceLocaleIdentifier)
            voiceDic["voiceType"] = voiceType
            voiceDic["voiceName"] = name
            voiceArrays.add(voiceDic)
        }
        
        var opacityItems: [ComponentMenuitemProperty] = []
        voiceArrays = self.sortArray(voiceArrays)
        for i in 0..<voiceArrays.count {
            let voiceDic = voiceArrays[i] as! NSDictionary
            let voiceType = voiceDic.object(forKey: "voiceType") as! NSSpeechSynthesizer.VoiceName
            if currentVoicName == voiceType {
                currentIndex = i
            }
            
            let dic = KMTTSManager.defalutManager.attributesForVoice(voiceType) as NSDictionary
            let code = ( NSSpeechSynthesizer.attributes( forVoice: voiceType )[ NSSpeechSynthesizer.VoiceAttributeKey.localeIdentifier ] as! String )
            var languageStaring: String? = nil
            let voiceName = ( NSSpeechSynthesizer.attributes( forVoice: voiceType )[ NSSpeechSynthesizer.VoiceAttributeKey.name ] as! String )
            let name = self.switchLanguage(withCode: code)
            if voiceName.count > 0 {
                languageStaring = "\(name)(\(voiceName))"
            }
            if let value = languageStaring, value.count > 0 {
                let item = ComponentMenuitemProperty(type: .normal, text: value)
                opacityItems.append(item)
            }
            languageSelect.updateMenuItemsArr(opacityItems)
        }
        languageSelect.selectItemAtIndex(currentIndex)
        
        KMTTSManager.defalutManager.delegate = self
        
        self.updateViewColor()
        
        if KMTTSManager.defalutManager.isContinue {
            continueCheckbox.properties.checkboxType = .selected
        } else {
            continueCheckbox.properties.checkboxType = .normal
        }
        continueCheckbox.setTarget(self, action: #selector(buttonItemClick_Continue(_:)))
        continueCheckbox.reloadData()
        
        DistributedNotificationCenter.default().addObserver(self, selector: #selector(themeChanged(notification:)), name: NSNotification.Name("AppleInterfaceThemeChangedNotification"), object: nil)
        
        NotificationCenter.default.addObserver(self, selector: #selector(handlePageChangedNotification), name: NSNotification.Name.CPDFViewPageChanged, object: self.pdfView)
    }
    
    func updateViewColor() {
        
    }
    
    func sortArray(_ array: NSArray) -> NSMutableArray {
        let sortDesc = [NSSortDescriptor(key: "voiceName", ascending: true)]
        let sortedArr = array.sortedArray(using: sortDesc) as NSArray
        let tArray = NSMutableArray(array: sortedArr)
        return tArray
    }
    
    func windowShouldClose(_ sender: NSWindow) -> Bool {
        self.stopSpeaking()
        if let callBlack = closeWindowCallback {
            callBlack(true)
        }
        return true
    }
    
    override func showWindow(_ sender: Any?) {
        super.showWindow(sender)
        
        self.isChangePage = false
        if self.pdfView.canGoToNextPage() {
            nextButton.properties.isDisabled = false
        } else{
            nextButton.properties.isDisabled = true
        }
        nextButton.reloadData()
        
        if self.pdfView.canGoToPreviousPage() {
            previousButton.properties.isDisabled = false
        } else {
            previousButton.properties.isDisabled = true
        }
        previousButton.reloadData()
        
        speedSlider.reloadData()
        
    }
    
    func stopSpeaking() {
        if KMTTSManager.defalutManager.isSpeaking() || KMTTSManager.defalutManager.isPaused {
            self.isChangePage = true
            KMTTSManager.defalutManager.stopSpeaking()
            
            speedSlider.properties.isDisabled = false
            speedSlider.reloadData()
            
            speedInput.properties.isDisabled = false
            speedInput.reloadData()
            
            playButton.properties.icon = NSImage(named: "pause_fill")
            playButton.reloadData()
            
            self.pdfView.setHighlightedSelections([])
        }
        self.pdfSelection = nil
        self.pdfView = nil
    }
    
    func quikeStartSpeakingPDFPage(_ page: CPDFPage) {
        if KMTTSManager.defalutManager.isSpeaking() || KMTTSManager.defalutManager.isPaused {
            KMTTSManager.defalutManager.stopSpeaking()
            self.isChangePage = true
        }
        self.currentPageIndex = Int(self.pdfView.document?.index(for: page) ?? 0)
        self.startSpeakingPDFPage(page)
    }
    
    func startSpeakingPDFPage(_ page: CPDFPage) {
        
        let dex = self.languageSelect.indexOfSelect()
        if dex >= 0 && dex < self.voiceArrays.count {
            let dic: NSMutableDictionary = self.voiceArrays[dex] as! NSMutableDictionary
            if let voiceName = dic["voiceType"] {
                KMTTSManager.defalutManager.setVoice(voice: voiceName as! NSSpeechSynthesizer.VoiceName)
            }
        }
        KMTTSManager.defalutManager.rate = Float(self.speedSlider.properties.percent) * Float(standardSpeed)
        let isSuccess = KMTTSManager.defalutManager.startSpeakingPDFPage(page)
        if isSuccess {
            playButton.properties.icon = NSImage(named: "play_fill")
            playButton.toolTip = NSLocalizedString("Pause", comment: "")
            speedSlider.properties.isDisabled = true
            speedInput.properties.isDisabled = true
            
        } else {
            playButton.properties.icon = NSImage(named: "pause_fill")
            playButton.toolTip = NSLocalizedString("Play", comment: "")
            speedSlider.properties.isDisabled = false
            speedInput.properties.isDisabled = false
            
        }
        playButton.reloadData()
        speedSlider.reloadData()
        speedInput.reloadData()
    }
    
    func startSpeakingPDFSelection(_ selection: CPDFSelection) {
        if KMTTSManager.defalutManager.isSpeaking() || KMTTSManager.defalutManager.isPaused {
            KMTTSManager.defalutManager.stopSpeaking()
            self.isChangePage = true
        }
        self.pdfSelection = selection
        self.pdfView?.setCurrentSelection(nil, animate: false)
        let dex = self.languageSelect.indexOfSelect()
        if dex >= 0 && dex < self.voiceArrays.count {
            let dic: NSMutableDictionary = self.voiceArrays[dex] as! NSMutableDictionary
            if let voiceName = dic["voiceType"] as? NSSpeechSynthesizer.VoiceName {
                KMTTSManager.defalutManager.setVoice(voice: voiceName)
            }
        }
        KMTTSManager.defalutManager.rate = Float(self.speedSlider.properties.percent) * Float(standardSpeed)
        let isSuccess = KMTTSManager.defalutManager.startSpeakingPDFSelection(selection)
        if isSuccess {
            playButton.properties.icon = NSImage(named: "play_fill")
            playButton.toolTip = NSLocalizedString("Pause", comment: "")
            speedSlider.properties.isDisabled = true
            speedInput.properties.isDisabled = true
            
        } else {
            playButton.properties.icon = NSImage(named: "pause_fill")
            playButton.toolTip = NSLocalizedString("Play", comment: "")
            speedSlider.properties.isDisabled = false
            speedInput.properties.isDisabled = false
            
        }
    }
    
    //MARK: - Noti
    @objc func themeChanged(notification: NSNotification) {
        DispatchQueue.main.asyncAfter(deadline: .now() + 0.3) {
            self.updateViewColor()
        }
    }
    
    @objc func handlePageChangedNotification(notification: NSNotification) {
        guard let PDFView = notification.object as? CPDFView else { return }
        if PDFView == self.pdfView {
            
            if self.pdfView.canGoToNextPage() {
                nextButton.properties.isDisabled = false
            } else{
                nextButton.properties.isDisabled = true
            }
            nextButton.reloadData()
            
            if self.pdfView.canGoToPreviousPage() {
                previousButton.properties.isDisabled = false
            } else {
                previousButton.properties.isDisabled = true
            }
            previousButton.reloadData()
            
        }
    }
    
    //MARK: - IBAction
    @objc func buttonItemClick_Next(_ sender: NSView) {
        if (KMTTSManager.defalutManager.isSpeaking()) {
            var currentPageNum = self.currentPageIndex;
            var page: CPDFPage?
            if (currentPageNum + 1 < self.pdfView.document.pageCount) {
                page = self.pdfView.document.page(at: UInt(currentPageNum+1))
                self.currentPageIndex  = currentPageNum + 1;
                
            } else {
                page = self.pdfView.document.page(at: self.pdfView.document.pageCount - 1)
                self.currentPageIndex = Int(self.pdfView.document.pageCount - 1)
            }
            self.pdfView.go(to: page)
            self.isChangePage = true
            self.startSpeakingPDFPage(page!)
        }
    }
    
    @objc func buttonItemClick_Forward(_ sender: NSView) {
        if (KMTTSManager.defalutManager.isSpeaking()) {
            var currentPageNum = self.currentPageIndex;
            var page: CPDFPage?
            if (currentPageNum - 1 < 0) {
                page = self.pdfView.document.page(at: 0)
                self.currentPageIndex  = 0;
                
            } else {
                page = self.pdfView.document.page(at: UInt(currentPageNum - 1))
                self.currentPageIndex = currentPageNum - 1
            }
            self.pdfView.go(to: page)
            self.isChangePage = true
            self.startSpeakingPDFPage(page!)
        }
    }
    
    @objc func buttonItemClick_Play(_ sender: NSView) {
        if (KMTTSManager.defalutManager.isPaused) {
            self.playButton.properties.icon = NSImage(named: "play_fill")
            self.playButton.toolTip = KMLocalizedString("Pause")
            self.speedSlider.properties.isDisabled = true
            self.speedInput.properties.isDisabled = true
            
            KMTTSManager.defalutManager.continueSpeaking()
        } else if (KMTTSManager.defalutManager.isSpeaking()) {
            self.isChangePage = true
            self.playButton.properties.icon = NSImage(named: "pause_fill")
            self.playButton.toolTip = KMLocalizedString("Play")
            self.speedSlider.properties.isDisabled = false
            self.speedInput.properties.isDisabled = false
            
            self.pdfView.setHighlightedSelection(nil, animated: false)
            KMTTSManager.defalutManager.pauseSpeaking()
        } else {
            if ((self.pdfSelection) != nil) {
                self.startSpeakingPDFSelection(self.pdfSelection!)
            } else {
                let page = self.pdfView.currentPage()
                self.currentPageIndex = self.pdfView.currentPageIndex
                if page != nil {
                    self.startSpeakingPDFPage(page!)
                }
            }
        }
    }
    
    @objc func buttonItemClick_Continue(_ sender: NSView) {
        KMTTSManager.defalutManager.isContinue = !(KMTTSManager.defalutManager.isContinue)
    }
    
    func stepperItemClick_Speed(_ sender: NSButton) {
        //        var rate = self.speedStepper.floatValue
        //        let str = String(format: "%.1f", rate)
        //        self.speedStepper.stringValue = str
        //        self.speedTextField.stringValue = str
        //        self.speedSlider.stringValue = str
        //        rate = Float(standardSpeed) * rate
        //        KMTTSManager.defalutManager.rate = rate
    }
    
    func sliderItemClick_Speed(_ sender: NSButton) {
        //        var rate: CGFloat = CGFloat(self.speedSlider.floatValue)
        //        let str = String(format: "%.1f", rate)
        //        self.speedStepper.stringValue = str
        //        self.speedTextField.stringValue = str
        //        self.speedSlider.stringValue = str
        //        rate = CGFloat(standardSpeed) * rate
        //        KMTTSManager.defalutManager.rate = Float(rate)
    }
    
    func buttonItemClick_Language() {
        if KMTTSManager.defalutManager.isSpeaking() {
            self.isChangePage = true
            if let pdfSelection = self.pdfSelection {
                self.startSpeakingPDFSelection(pdfSelection)
            } else {
                let currentPageInd = self.currentPageIndex
                if currentPageInd < (self.pdfView.document.pageCount - 1) {
                    let page = self.pdfView.document.page(at: UInt(currentPageInd))
                    self.startSpeakingPDFPage(page!)
                }
            }
        }
    }
    
    //MARK: - MouseEvent
    override func mouseDown(with event: NSEvent) {
        super.mouseDown(with: event)
        self.window?.makeFirstResponder(self)
    }
    
    func switchLanguage(withCode code: String) -> String {
        var language = code
        if code == "ar_SA" {
            language = "Arabic (Saudi Arabia)"
        } else if code == "cs_CZ" {
            language = "Czech (Czech republic)"
        } else if code == "da_DK" {
            language = "Danish (Denmark)"
        } else if code == "de_DE" {
            language = "German (Germany)"
        } else if code == "el_GR" {
            language = "Greek (Greece)"
        } else if code == "en_AU" {
            language = "English (Australia)"
        } else if code == "en_GB" {
            language = "English (UK)"
        } else if code == "en_IE" {
            language = "English (Ireland)"
        } else if code == "en_US" {
            language = "English"
        } else if code == "en_ZA" {
            language = "English (South Africa)"
        } else if code == "es_ES" {
            language = "Spanish (Spain)"
        } else if code == "es_MX" {
            language = "Spanish (Mexico)"
        } else if code == "fi_FI" {
            language = "Finnish (Finland)"
        } else if code == "fr_CA" {
            language = "French (Canada)"
        } else if code == "fr_FR" {
            language = "French (France)"
        } else if code == "he_IL" {
            language = "Hebrew"
        } else if code == "hi_IN" {
            language = "Hindi (India)"
        } else if code == "hu_HU" {
            language = "Hungarian (Hungary)"
        } else if code == "id_ID" {
            language = "Indonesian (Indonesia)"
        } else if code == "it_IT" {
            language = "Italian (Italy)"
        } else if code == "ja_JP" {
            language = "日本語"
        } else if code == "ko_KR" {
            language = "Korean (South Korea)"
        } else if code == "nl_BE" {
            language = "Dutch (Belgium)"
        } else if code == "nl_NL" {
            language = "Dutch (Holland)"
        } else if code == "nb_NO" {
            language = "Norwegian (Norway)"
        } else if code == "pl_PL" {
            language = "Polish (Poland)"
        } else if code == "pt_BR" {
            language = "Portuguese (Brazil)"
        } else if code == "pt_PT" {
            language = "Portuguese (Portugal)"
        } else if code == "ro_RO" {
            language = "Romanian (Romania)"
        } else if code == "ru_RU" {
            language = "Russian (Russia)"
        } else if code == "sk_SK" {
            language = "Slovakia (Slovakia)"
        } else if code == "sv_SE" {
            language = "Swe (Sweden)"
        } else if code == "th_TH" {
            language = "Thai (Thailand)"
        } else if code == "tr_TR" {
            language = "Turkish (Turkey)"
        } else if code == "zh_CN" {
            language = "简体中文"
        } else if code == "zh_HK" {
            language = "繁體中文 (香港)"
        } else if code == "zh_TW" {
            language = "繁體中文 (台灣)"
        } else if code == "en_IN" {
            language = "English"
        }
        return language
    }
}


//MARK: KMTTSManagerDelegate
extension KMTTSWindowController: KMTTSManagerDelegate {
    func ttsMananger(_ tts: KMTTSManager, willSpeak selection: CPDFSelection) {
        self.pdfView?.setHighlightedSelection(selection, animated: false)
    }
    
    func ttsManangerDidFinishSpeech(_ tts: KMTTSManager) -> Bool {
        var isFinish = true
        if self.pdfSelection != nil {
            self.pdfSelection = nil
            self.pdfView.setHighlightedSelections(nil)
            self.playButton.properties.icon = NSImage(named: "pause_fill")
            self.speedSlider.properties.isDisabled = false
            self.speedInput.properties.isDisabled = false
            self.playButton.toolTip = KMLocalizedString("Play")
        } else {
            if KMTTSManager.defalutManager.isContinue {
                if !isChangePage {
                    if let pdfSelection = self.pdfSelection {
                        self.pdfView.setHighlightedSelections(nil)
                        self.playButton.properties.icon = NSImage(named: "pause_fill")
                        
                        self.speedSlider.properties.isDisabled = false
                        self.speedInput.properties.isDisabled = false
                        self.playButton.toolTip = KMLocalizedString("Play")
                    } else {
                        let currentPageIndex = self.currentPageIndex
                        if currentPageIndex + 1 < self.pdfView.document.pageCount {
                            let page = self.pdfView.document.page(at: UInt(currentPageIndex+1))
                            self.currentPageIndex = currentPageIndex+1
                            self.pdfView.go(to: page)
                            self.startSpeakingPDFPage(page!)
                            isFinish = false
                        } else {
                            self.pdfView.setHighlightedSelections(nil)
                            self.playButton.properties.icon = NSImage(named: "pause_fill")
                            self.speedSlider.properties.isDisabled = false
                            self.speedInput.properties.isDisabled = false
                            self.playButton.toolTip = KMLocalizedString("Play")
                        }
                    }
                }
            } else {
                if !isChangePage {
                    self.speedSlider.properties.isDisabled = false
                    self.speedInput.properties.isDisabled = false
                    self.pdfView.setHighlightedSelections(nil)
                    self.playButton.properties.icon = NSImage(named: "pause_fill")
                    self.playButton.toolTip = KMLocalizedString("Play")
                }
            }
            if isChangePage {
                isChangePage = false
            }
        }
        return isFinish
    }
    func ttsManangerdidErrorSpeech(_ tts: KMTTSManager, message: String) {
        
    }
    
}

//MARK: NSTextFieldDelegate
extension KMTTSWindowController: NSTextFieldDelegate {
    func controlTextDidEndEditing(_ obj: Notification) {
        //        if let textField = obj.object as? NSTextField {
        //            if textField == self.speedTextField {
        //                var rate: Float = Float(textField.stringValue) ?? 0
        //                if maxSpeed - rate < 0 {
        //                    rate = maxSpeed
        //                } else if rate - minSpeed < 0 {
        //                    rate = minSpeed
        //                }
        //                self.speedStepper.stringValue = String(format: "%.1f", rate)
        //                self.speedTextField.stringValue = String(format: "%.1f", rate)
        //                self.speedSlider.stringValue = String(format: "%.1f", rate)
        //                rate = standardSpeed * rate
        //                KMTTSManager.defalutManager.rate = rate
        //            }
        //        }
    }
    
}

//MARK: ComponentSelectDelegate
extension KMTTSWindowController: ComponentSelectDelegate {
    func componentSelectDidSelect(view: ComponentSelect?, menuItemProperty: ComponentMenuitemProperty?) {
        buttonItemClick_Language()
    }
}

//MARK: ComponentSliderDelegate
extension KMTTSWindowController: ComponentSliderDelegate {
    func componentSliderDidUpdate(_ view: ComponentSlider) {
        let percent = view.properties.percent
        
        
    }
}

//MARK: ComponentInputNumberDelegate
extension KMTTSWindowController: ComponentInputNumberDelegate {
    func componentInputNumberDidValueChanged(inputNumber: ComponentInputNumber?) {
        
    }
}