arduino-ide/arduino-ide-extension/src/browser/monitor-manager-proxy-client-impl.ts

import { Emitter, JsonRpcProxy, MessageService } from "@theia/core";
import { inject, injectable } from "@theia/core/shared/inversify";
import { Board, Port } from "../common/protocol";
import { Monitor, MonitorManagerProxy, MonitorManagerProxyClient, MonitorSettings } from "../common/protocol/monitor-service";

@injectable()
export class MonitorManagerProxyClientImpl implements MonitorManagerProxyClient {
    // When pluggable monitor messages are received from the backend
    // this event is triggered.
    // Ideally a frontend component is connected to this event
    // to update the UI.
    protected readonly onMessagesReceivedEmitter = new Emitter<{ messages: string[] }>();
    readonly onMessagesReceived = this.onMessagesReceivedEmitter.event;

    // WebSocket used to handle pluggable monitor communication between
    // frontend and backend.
    private webSocket?: WebSocket;
    private wsPort?: number;

    getWebSocketPort(): number | undefined {
        return this.wsPort;
    }

    constructor(
        @inject(MessageService)
        protected messageService: MessageService,

        // This is necessary to call the backend methods from the frontend
        @inject(MonitorManagerProxy)
        protected server: JsonRpcProxy<MonitorManagerProxy>
    ) {

    }

    /**
     * Connects a localhost WebSocket using the specified port.
     * @param addressPort port of the WebSocket
     */
    connect(addressPort: number): void {
        if (this.webSocket) {
            return;
        }
        try {
            this.webSocket = new WebSocket(`ws://localhost:${addressPort}`);
        } catch {
            this.messageService.error('Unable to connect to websocket');
            return;
        }

        this.webSocket.onmessage = (res) => {
            const messages = JSON.parse(res.data);
            this.onMessagesReceivedEmitter.fire({ messages });
        }
        this.wsPort = addressPort;
    }

    /**
     * Disconnects the WebSocket if connected.
     */
    disconnect(): void {
        try {
            this.webSocket?.close();
            this.webSocket = undefined;
        } catch {
            this.messageService.error('Unable to close websocket');
        }
    }

    async isWSConnected(): Promise<boolean> {
        return !!this.webSocket;
    }

    async startMonitor(board: Board, port: Port, settings?: MonitorSettings): Promise<void> {
        return this.server.startMonitor(board, port, settings);
    }

    getCurrentSettings(board: Board, port: Port): MonitorSettings {
        return this.server.getCurrentSettings(board, port);
    }

    send(message: string): void {
        if (!this.webSocket) {
            return;
        }

        this.webSocket.send(JSON.stringify({
            command: Monitor.Command.SEND_MESSAGE,
            data: message,
        }));
    }

    changeSettings(settings: MonitorSettings): void {
        if (!this.webSocket) {
            return;
        }

        this.webSocket.send(JSON.stringify({
            command: Monitor.Command.CHANGE_SETTINGS,
            // TODO: This might be wrong, verify if it works
            data: settings,
        }));
    }
}