Beans model extension API for VSCode

vscode-extensions/vscode-spring-boot/lib/api.d.ts

@@ -1,6 +1,7 @@
 import { Event } from "vscode";
 import { LanguageClient } from "vscode-languageclient/node";
 import { LiveProcess } from "./notification";
+import {Location} from "vscode-languageclient";
 
 export interface ExtensionAPI {
     readonly client: LanguageClient;
@@ -55,7 +56,16 @@ export interface ExtensionAPI {
      * 
      * Returns a list of processKeys.
      */
-    readonly listConnectedProcesses: () => Promise<LiveProcess[]>
+    readonly listConnectedProcesses: () => Promise<LiveProcess[]>;
+
+    /**
+     * Spring Model capable of computing beans and other static spring related information
+     *
+     * Returns Spring Model object.
+     */
+    readonly getSpringModel: () => SpringModel;
+
+    readonly onSpringModelUpdated: Event<void>;
 }
 
 interface LiveProcessDataQuery {
@@ -82,4 +92,26 @@ interface MetricsQuery extends LiveProcessDataQuery {
     endpoint: "metrics";
     metricName: string;
     tags?: string;
+}
+
+interface Bean {
+    readonly name: string;
+    readonly type: string;
+    readonly location: Location;
+    readonly injectionPoints: InjectionPoint[];
+    readonly supertypes: string[];
+}
+
+interface InjectionPoint {
+    readonly name: string;
+    readonly type: string;
+    readonly location: Location;
+}
+
+interface SpringModel {
+   readonly beans: (params: BeansParams) =>  Promise<Bean[]>;
+}
+
+interface BeansParams {
+    projectName: string;
 }

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

@@ -1,7 +1,17 @@
 import { commands, Uri } from "vscode";
 import { Emitter, LanguageClient } from "vscode-languageclient/node";
-import { ExtensionAPI } from "./api";
-import { LiveProcess, LiveProcessConnectedNotification, LiveProcessDisconnectedNotification, LiveProcessUpdatedNotification, LiveProcessGcPausesMetricsUpdatedNotification, LiveProcessMemoryMetricsUpdatedNotification } from "./notification";
+import {Bean, BeansParams, ExtensionAPI, SpringModel} from "./api";
+import {
+    LiveProcess,
+    LiveProcessConnectedNotification,
+    LiveProcessDisconnectedNotification,
+    LiveProcessUpdatedNotification,
+    LiveProcessGcPausesMetricsUpdatedNotification,
+    LiveProcessMemoryMetricsUpdatedNotification,
+    SpringModelUpdatedNotification
+} from "./notification";
+import VSCode from "vscode";
+import {RequestType} from "vscode-languageclient";
 
 export class ApiManager {
     public api: ExtensionAPI;
@@ -10,6 +20,7 @@ export class ApiManager {
     private onDidLiveProcessUpdateEmitter: Emitter<LiveProcess> = new Emitter<LiveProcess>();
     private onDidLiveProcessGcPausesMetricsUpdateEmitter: Emitter<LiveProcess> = new Emitter<LiveProcess>();
     private onDidLiveProcessMemoryMetricsUpdateEmitter: Emitter<LiveProcess> = new Emitter<LiveProcess>();
+    private onSpringModelUpdateEmitter: Emitter<void> = new Emitter<void>();
 
     public constructor(client: LanguageClient) {
         const onDidLiveProcessConnect = this.onDidLiveProcessConnectEmitter.event;
@@ -17,6 +28,7 @@ export class ApiManager {
         const onDidLiveProcessUpdate = this.onDidLiveProcessUpdateEmitter.event;
         const onDidLiveProcessGcPausesMetricsUpdate = this.onDidLiveProcessGcPausesMetricsUpdateEmitter.event;
         const onDidLiveProcessMemoryMetricsUpdate = this.onDidLiveProcessMemoryMetricsUpdateEmitter.event;
+        const onSpringModelUpdated = this.onSpringModelUpdateEmitter.event;
 
         const COMMAND_LIVEDATA_GET = "sts/livedata/get";
         const getLiveProcessData = async (query) => {
@@ -49,6 +61,17 @@ export class ApiManager {
         client.onNotification(LiveProcessGcPausesMetricsUpdatedNotification.type, (process: LiveProcess) => this.onDidLiveProcessGcPausesMetricsUpdateEmitter.fire(process));
         client.onNotification(LiveProcessMemoryMetricsUpdatedNotification.type, (process: LiveProcess) => this.onDidLiveProcessMemoryMetricsUpdateEmitter.fire(process));
 
+        client.onNotification(SpringModelUpdatedNotification.type, () => this.onSpringModelUpdateEmitter.fire());
+
+        const beansRequestType = new RequestType<BeansParams, Bean[], void>('spring/beans');
+        const beans = (params: BeansParams) => {
+            return client.sendRequest(beansRequestType, params);
+        }
+
+        const getSpringModel = () => ({
+            beans
+        })
+
         this.api = {
             client,
             onDidLiveProcessConnect,
@@ -56,11 +79,13 @@ export class ApiManager {
             onDidLiveProcessUpdate,
             onDidLiveProcessMemoryMetricsUpdate,
             onDidLiveProcessGcPausesMetricsUpdate,
+            onSpringModelUpdated,
             getLiveProcessData,
             refreshLiveProcessData,
             getLiveProcessMetricsData,
             refreshLiveProcessMetricsData,
             listConnectedProcesses,
+            getSpringModel
         };
     }
 }

vscode-extensions/vscode-spring-boot/lib/notification.ts

@@ -36,4 +36,8 @@ export namespace LiveProcessGcPausesMetricsUpdatedNotification {
 
 export namespace LiveProcessMemoryMetricsUpdatedNotification {
 	export const type = new NotificationType<LiveProcess>('sts/liveprocess/memory/metrics/updated');
+}
+
+export namespace SpringModelUpdatedNotification {
+	export const type = new NotificationType<void>('spring/model/updated');
 }