spring-tools/vscode-extensions/vscode-spring-boot/lib/apiManager.ts

import { commands } from "vscode";
import { Emitter, LanguageClient } from "vscode-languageclient/node";
import {Bean, BeansParams, ExtensionAPI} from "./api";
import {
    LiveProcess,
    LiveProcessConnectedNotification,
    LiveProcessDisconnectedNotification,
    LiveProcessUpdatedNotification,
    LiveProcessGcPausesMetricsUpdatedNotification,
    LiveProcessMemoryMetricsUpdatedNotification,
    SpringIndexUpdatedNotification,
} from "./notification";
import {RequestType} from "vscode-languageclient";

export class ApiManager {
    public api: ExtensionAPI;
    private onDidLiveProcessConnectEmitter: Emitter<LiveProcess> = new Emitter<LiveProcess>();
    private onDidLiveProcessDisconnectEmitter: Emitter<LiveProcess> = new Emitter<LiveProcess>();
    private onDidLiveProcessUpdateEmitter: Emitter<LiveProcess> = new Emitter<LiveProcess>();
    private onDidLiveProcessGcPausesMetricsUpdateEmitter: Emitter<LiveProcess> = new Emitter<LiveProcess>();
    private onDidLiveProcessMemoryMetricsUpdateEmitter: Emitter<LiveProcess> = new Emitter<LiveProcess>();
    private onSpringIndexUpdateEmitter: Emitter<void> = new Emitter<void>();

    public constructor(client: LanguageClient) {
        const onDidLiveProcessConnect = this.onDidLiveProcessConnectEmitter.event;
        const onDidLiveProcessDisconnect = this.onDidLiveProcessDisconnectEmitter.event;
        const onDidLiveProcessUpdate = this.onDidLiveProcessUpdateEmitter.event;
        const onDidLiveProcessGcPausesMetricsUpdate = this.onDidLiveProcessGcPausesMetricsUpdateEmitter.event;
        const onDidLiveProcessMemoryMetricsUpdate = this.onDidLiveProcessMemoryMetricsUpdateEmitter.event;
        const onSpringIndexUpdated = this.onSpringIndexUpdateEmitter.event;
        // const onDidLiveProcessLoggersUpdate = this.onDidLiveProcessLoggersUpdateEmitter.event;

        const COMMAND_LIVEDATA_GET = "sts/livedata/get";
        const getLiveProcessData = async (query) => {
            return await commands.executeCommand(COMMAND_LIVEDATA_GET, query);
        }

        const COMMAND_LIVEDATA_REFRESH = "sts/livedata/refresh";
        const refreshLiveProcessData = async (query) => {
            return await commands.executeCommand(COMMAND_LIVEDATA_REFRESH, query);
        }

        const COMMAND_LIVEDATA_LIST_CONNECTED = "sts/livedata/listConnected"
        const listConnectedProcesses = async () : Promise<LiveProcess[]> => {
            return await commands.executeCommand(COMMAND_LIVEDATA_LIST_CONNECTED);
        }

        const COMMAND_LIVEDATA_GET_METRICS = "sts/livedata/get/metrics"
        const getLiveProcessMetricsData = async (query) : Promise<LiveProcess[]> => {
            return await commands.executeCommand(COMMAND_LIVEDATA_GET_METRICS, query);
        }

        const COMMAND_LIVEDATA_REFRESH_METRICS = "sts/livedata/refresh/metrics";
        const refreshLiveProcessMetricsData = async (query) => {
            return await commands.executeCommand(COMMAND_LIVEDATA_REFRESH_METRICS, query);
        }

        client.onNotification(LiveProcessConnectedNotification.type, (process: LiveProcess) => this.onDidLiveProcessConnectEmitter.fire(process));
        client.onNotification(LiveProcessDisconnectedNotification.type, (process: LiveProcess) => this.onDidLiveProcessDisconnectEmitter.fire(process));
        client.onNotification(LiveProcessUpdatedNotification.type, (process: LiveProcess) => this.onDidLiveProcessUpdateEmitter.fire(process));
        client.onNotification(LiveProcessGcPausesMetricsUpdatedNotification.type, (process: LiveProcess) => this.onDidLiveProcessGcPausesMetricsUpdateEmitter.fire(process));
        client.onNotification(LiveProcessMemoryMetricsUpdatedNotification.type, (process: LiveProcess) => this.onDidLiveProcessMemoryMetricsUpdateEmitter.fire(process));

        client.onNotification(SpringIndexUpdatedNotification.type, () => this.onSpringIndexUpdateEmitter.fire());

        const beansRequestType = new RequestType<BeansParams, Bean[], void>('spring/index/beans');
        const beans = (params: BeansParams) => {
            return client.sendRequest(beansRequestType, params);
        }

        const getSpringIndex = () => ({
            onSpringIndexUpdated,
            beans
        })

        this.api = {
            client,
            onDidLiveProcessConnect,
            onDidLiveProcessDisconnect,
            onDidLiveProcessUpdate,
            onDidLiveProcessMemoryMetricsUpdate,
            onDidLiveProcessGcPausesMetricsUpdate,
            getLiveProcessData,
            refreshLiveProcessData,
            getLiveProcessMetricsData,
            refreshLiveProcessMetricsData,
            listConnectedProcesses,
            getSpringIndex
        };
    }
}