Add cairo.showMemoryUsage command (#165)

Author: Draggu

package.json

@@ -85,6 +85,11 @@
         "command": "cairo.viewSyntaxTree",
         "title": "View syntax tree of the current file content",
         "category": "Cairo"
+      },
+      {
+        "command": "cairo.showMemoryUsage",
+        "title": "Show current memory usage of Cairo Language Server",
+        "category": "Cairo"
       }
     ],
     "menus": {

src/cairols.ts

@@ -8,6 +8,7 @@ import {
   registerMacroExpandProvider,
   registerVfsProvider,
   registerViewAnalyzedCratesProvider,
+  registeShowMemoryUsageProvider,
 } from "./textDocumentProviders";
 
 import { executablesEqual, getLSExecutables, LSExecutable } from "./lsExecutable";
@@ -71,6 +72,7 @@ export async function setupLanguageServer(ctx: Context): Promise<SetupResult | u
 
   registerVfsProvider(client, ctx);
   registerMacroExpandProvider(client, ctx);
+  registeShowMemoryUsageProvider(client, ctx);
   registerViewAnalyzedCratesProvider(client, ctx);
 
   ctx.extension.subscriptions.push(

src/lspRequests.ts

@@ -9,6 +9,8 @@ export const expandMacro = new lc.RequestType<ExpandMacroRequest, ExpandMacroRes
   "cairo/expandMacro",
 );
 
+export const showMemoryUsage = new lc.RequestType0<unknown, void>("cairo/showMemoryUsage");
+
 export interface ProvideVirtualFileRequest {
   uri: string;
 }

src/textDocumentProviders.ts

@@ -1,7 +1,13 @@
 import * as lc from "vscode-languageclient/node";
 import * as vscode from "vscode";
 import { Context } from "./context";
-import { expandMacro, viewSyntaxTree, vfsProvide, viewAnalyzedCrates } from "./lspRequests";
+import {
+  expandMacro,
+  viewSyntaxTree,
+  vfsProvide,
+  viewAnalyzedCrates,
+  showMemoryUsage,
+} from "./lspRequests";
 import { AnsiDecorationProvider } from "./ansi";
 
 export const registerVfsProvider = (client: lc.LanguageClient, ctx: Context) => {
@@ -56,6 +62,37 @@ export const registerMacroExpandProvider = (client: lc.LanguageClient, ctx: Cont
   );
 };
 
+export const registeShowMemoryUsageProvider = (client: lc.LanguageClient, ctx: Context) => {
+  const uri = vscode.Uri.parse("show-memory-usage://memory-usage");
+  const eventEmitter = new vscode.EventEmitter<vscode.Uri>();
+
+  const tdcp: vscode.TextDocumentContentProvider = {
+    async provideTextDocumentContent(): Promise<string> {
+      const expanded = await client.sendRequest(showMemoryUsage);
+
+      return (
+        JSON.stringify(expanded, null, 4) ??
+        "Fetching memory usage failed. This may be caused by LS version that does not support this"
+      );
+    },
+    onDidChange: eventEmitter.event,
+  };
+
+  ctx.extension.subscriptions.push(
+    vscode.workspace.registerTextDocumentContentProvider("show-memory-usage", tdcp),
+  );
+
+  ctx.extension.subscriptions.push(
+    vscode.commands.registerCommand("cairo.showMemoryUsage", async () => {
+      const document = await vscode.workspace.openTextDocument(uri);
+
+      eventEmitter.fire(uri);
+
+      return vscode.window.showTextDocument(document, vscode.ViewColumn.Two, true);
+    }),
+  );
+};
+
 export const registerViewAnalyzedCratesProvider = (client: lc.LanguageClient, ctx: Context) => {
   const uri = vscode.Uri.parse(
     "cairo-view-analyzed-crates://viewAnalyzedCrates/[ANALYZED_CRATES].md",