UTM/Platform/macOS/Display/VMDisplayAppleTerminalWindowController.swift

//
// Copyright © 2022 osy. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//

import Foundation
import SwiftTerm

class VMDisplayAppleTerminalWindowController: VMDisplayAppleWindowController, VMDisplayTerminal {
    var terminalView: TerminalView! {
        mainView as? TerminalView
    }
    
    var serialConfig: UTMAppleConfigurationSerial! {
        appleConfig.serials[index]
    }
    
    var serialPort: UTMSerialPort! {
        serialConfig.interface
    }
    
    override var defaultTitle: String {
        if isSecondary {
            return String.localizedStringWithFormat(NSLocalizedString("%@ (Terminal %lld)", comment: "VMDisplayAppleTerminalWindowController"), super.defaultTitle, index + 1)
        } else {
            return super.defaultTitle
        }
    }
    
    private(set) var isPrimary: Bool = true
    private(set) var index: Int = 0
    
    private var isSizeChangeIgnored: Bool = true
    @Setting("OptionAsMetaKey") var isOptionAsMetaKey: Bool = false
    
    convenience init(primaryForIndex index: Int, vm: UTMAppleVirtualMachine, onClose: ((Notification) -> Void)?) {
        self.init(vm: vm, onClose: onClose)
        self.index = index
    }
    
    convenience init(secondaryForIndex index: Int, vm: UTMAppleVirtualMachine) {
        self.init(vm: vm, onClose: nil)
        self.index = index
    }
    
    override func windowDidLoad() {
        mainView = TerminalView()
        terminalView!.terminalDelegate = self
        terminalView.allowMouseReporting = false
        super.windowDidLoad()
    }
    
    override func updateWindowFrame() {
        isSizeChangeIgnored = true
        setupTerminal(terminalView, using: serialConfig.terminal!, id: index, for: window!)
        isSizeChangeIgnored = false
        super.updateWindowFrame()
    }
    
    override func resizeConsoleButtonPressed(_ sender: Any) {
        let cmd = resizeCommand(for: terminalView, using: serialConfig!.terminal!)
        serialPort.write(data: cmd.data(using: .ascii)!)
        
    }
    
    override func captureMouseButtonPressed(_ sender: Any) {
        terminalView.allowMouseReporting = !terminalView.allowMouseReporting
        captureMouseToolbarButton.state = terminalView.allowMouseReporting ? .on : .off
    }
    
    override func enterLive() {
        serialPort.delegate = self
        super.enterLive()
        resizeConsoleToolbarItem.isEnabled = true
    }
    
    func sendString(_ string: String) {
        if let serialPort = serialPort, let data = string.data(using: .nonLossyASCII) {
            serialPort.write(data: data)
        } else {
            logger.error("failed to send: \(string)")
        }
    }
}

// MARK: - Terminal view delegate
extension VMDisplayAppleTerminalWindowController: TerminalViewDelegate, UTMSerialPortDelegate {
    func sizeChanged(source: TerminalView, newCols: Int, newRows: Int) {
        if !isSizeChangeIgnored {
            sizeChanged(id: index, newCols: newCols, newRows: newRows)
        }
    }
    
    func setTerminalTitle(source: TerminalView, title: String) {
        window!.subtitle = title
    }
    
    func hostCurrentDirectoryUpdate(source: TerminalView, directory: String?) {
    }
    
    func send(source: TerminalView, data: ArraySlice<UInt8>) {
        serialPort.write(data: Data(data))
    }
    
    func scrolled(source: TerminalView, position: Double) {
    }
    
    func serialPort(_ serialPort: UTMSerialPort, didRecieveData data: Data) {
        if let terminalView = terminalView {
            let arr = [UInt8](data)[...]
            DispatchQueue.main.async {
                terminalView.feed(byteArray: arr)
            }
        }
    }
    
    func clipboardCopy(source: SwiftTerm.TerminalView, content: Data) {
        if let str = String(bytes: content, encoding: .utf8) {
            let pasteBoard = NSPasteboard.general
            pasteBoard.clearContents()
            pasteBoard.writeObjects([str as NSString])
        }
    }
    
    func rangeChanged(source: SwiftTerm.TerminalView, startY: Int, endY: Int) {
    }
}