refactor(labs): remove TS Memory Treemap feature

Author: johnsoncodehk

extensions/labs/package.json

@@ -56,8 +56,7 @@
 		"prepack": "npm run build -- --minify",
 		"pack": "npm run prepack && vsce package",
 		"release": "npm run prepack && vsce publish",
-		"release:next": "npm run prepack && vsce publish --pre-release",
-		"size": "npm run prepack -- --metafile && esbuild-visualizer --metadata ./meta.json && open ./stats.html"
+		"release:next": "npm run prepack && vsce publish --pre-release"
 	},
 	"devDependencies": {
 		"@types/node": "latest",
@@ -66,8 +65,6 @@
 		"@volar/source-map": "2.4.23",
 		"@volar/vscode": "2.4.23",
 		"@vscode/vsce": "latest",
-		"esbuild": "latest",
-		"esbuild-plugin-copy": "latest",
-		"esbuild-visualizer": "latest"
+		"esbuild": "latest"
 	}
 }

extensions/labs/scripts/build.js

@@ -1,5 +1,4 @@
 // @ts-check
-
 require('esbuild').context({
 	entryPoints: {
 		extension: './src/extension.ts',
@@ -16,30 +15,6 @@ require('esbuild').context({
 	tsconfig: './tsconfig.json',
 	define: { 'process.env.NODE_ENV': '"production"' },
 	minify: process.argv.includes('--minify'),
-	plugins: [
-		require('esbuild-plugin-copy').copy({
-			resolveFrom: 'cwd',
-			assets: {
-				from: ['./node_modules/esbuild-visualizer/dist/lib/**/*'],
-				to: ['./lib'],
-			},
-			// @ts-expect-error
-			keepStructure: true,
-		}),
-		{
-			name: 'meta',
-			setup(build) {
-				build.onEnd((result) => {
-					if (result.metafile && result.errors.length === 0) {
-						require('fs').writeFileSync(
-							require('path').resolve(__dirname, '../meta.json'),
-							JSON.stringify(result.metafile),
-						);
-					}
-				});
-			},
-		},
-	],
 }).then(async ctx => {
 	console.log('building...');
 	if (process.argv.includes('--watch')) {

extensions/labs/src/common/quickPick.ts

@@ -1,10 +1,6 @@
 import type { LabsInfo } from '@volar/vscode';
 import * as lsp from '@volar/vscode';
-import * as fs from 'fs';
-import * as os from 'os';
-import * as path from 'path';
 import * as vscode from 'vscode';
-import { quickPick } from '../common/quickPick';
 import { getIconPath, useVolarExtensions } from '../common/shared';
 
 interface LanguageClientItem {
@@ -17,7 +13,7 @@ interface InvalidLanguageClientItem {
 }
 
 interface LanguageClientFieldItem extends LanguageClientItem {
-	field: 'start' | 'stop' | 'restart' | 'initializationOptions' | 'initializeResult' | 'memory';
+	field: 'start' | 'stop' | 'restart' | 'initializationOptions' | 'initializeResult';
 }
 
 export function activate(context: vscode.ExtensionContext) {
@@ -56,7 +52,6 @@ export function activate(context: vscode.ExtensionContext) {
 					stats.push({ ...element, field: 'restart' });
 					stats.push({ ...element, field: 'initializationOptions' });
 					stats.push({ ...element, field: 'initializeResult' });
-					stats.push({ ...element, field: 'memory' });
 				}
 				else if (element.client.state === lsp.State.Starting) {
 					stats.push({ ...element, field: 'stop' });
@@ -106,17 +101,6 @@ export function activate(context: vscode.ExtensionContext) {
 						},
 					};
 				}
-				else if (element.field === 'memory') {
-					return {
-						label: 'TS Memory Treemap',
-						collapsibleState: vscode.TreeItemCollapsibleState.None,
-						command: {
-							command: '_volar.action.tsMemoryTreemap',
-							title: '',
-							arguments: [element.client, element.extension.exports.volarLabs.languageServerProtocol],
-						},
-					};
-				}
 				else if (element.field === 'initializationOptions') {
 					return {
 						iconPath: new vscode.ThemeIcon('file'),
@@ -178,70 +162,6 @@ export function activate(context: vscode.ExtensionContext) {
 			await client.stop();
 			await client.start();
 		}),
-		vscode.commands.registerCommand('_volar.action.tsMemoryTreemap', async (client: lsp.BaseLanguageClient, protocol: typeof import('@volar/vscode/protocol')) => {
-
-			const select = await quickPick([
-				{
-					openInBrowser: {
-						label: 'Open in Browser',
-						description: 'Open the TypeScript Memory Treemap in your browser',
-					},
-					showInVSCode: {
-						label: 'Show in VS Code',
-						description: 'Show the html file in VS Code',
-					},
-					saveFile: {
-						label: 'Save File',
-						description: 'Pick a location to save the html file',
-					},
-				}
-			]);
-
-			if (select === undefined) {
-				return; // cancel
-			}
-
-			vscode.window.withProgress({
-				location: vscode.ProgressLocation.Window,
-				cancellable: false,
-				title: 'Loading Memory Data'
-			}, async progress => {
-
-				progress.report({ increment: 0 });
-
-				const meta = await client.sendRequest(protocol.LoadedTSFilesMetaRequest.type);
-				const { visualizer } = await import('esbuild-visualizer/dist/plugin/index.js');
-				const fileContent = await visualizer(meta as any);
-
-				if (select === 'openInBrowser') {
-					const tmpPath = path.join(os.tmpdir(), 'memory-report.html');
-					fs.writeFileSync(tmpPath, fileContent);
-					await vscode.env.openExternal(vscode.Uri.file(tmpPath));
-				}
-				else if (select === 'showInVSCode') {
-					const doc = await vscode.workspace.openTextDocument({ content: fileContent, language: 'html' });
-					vscode.window.showTextDocument(doc);
-				}
-				else if (select === 'saveFile') {
-					const workspaces = vscode.workspace.workspaceFolders;
-					if (!workspaces?.length) {
-						return;
-					}
-
-					const defaultUri = vscode.Uri.joinPath(workspaces[0].uri, 'stats.html');
-					const pickedUri = await vscode.window.showSaveDialog({ defaultUri });
-
-					if (!pickedUri) {
-						return;
-					}
-
-					await vscode.workspace.fs.writeFile(pickedUri, Buffer.from(fileContent));
-					await vscode.window.showTextDocument(pickedUri);
-				}
-
-				progress.report({ increment: 100 });
-			});
-		}),
 		vscode.commands.registerCommand('volar.action.serverStat.initializationOptions', async (client: lsp.BaseLanguageClient) => {
 			const doc = await vscode.workspace.openTextDocument({ content: JSON.stringify(client.clientOptions.initializationOptions, undefined, '\t'), language: 'json' });
 			vscode.window.showTextDocument(doc);

extensions/labs/src/views/serversView.ts

@@ -7,7 +7,6 @@ import {
 	GetServicePluginsRequest,
 	GetVirtualCodeRequest,
 	GetVirtualFileRequest,
-	LoadedTSFilesMetaRequest,
 	UpdateServicePluginStateNotification,
 	UpdateVirtualCodeStateNotification,
 	WriteVirtualFilesNotification
@@ -112,69 +111,6 @@ export function register(server: LanguageServerState) {
 				}
 			}
 		});
-		server.connection.onRequest(LoadedTSFilesMetaRequest.type, async () => {
-
-			const sourceFilesData = new Map<ts.SourceFile, {
-				projectNames: string[];
-				size: number;
-			}>();
-
-			for (const languageService of await project.getExistingLanguageServices()) {
-				const tsLanguageService: ts.LanguageService | undefined = languageService.context.inject<any>('typescript/languageService');
-				const program = tsLanguageService?.getProgram();
-				const tsProject = languageService.context.project.typescript;
-				if (program && tsProject) {
-					const { languageServiceHost, configFileName } = tsProject;
-					const projectName = configFileName ?? (languageServiceHost.getCurrentDirectory() + '(inferred)');
-					const sourceFiles = program.getSourceFiles() ?? [];
-					for (const sourceFile of sourceFiles) {
-						if (!sourceFilesData.has(sourceFile)) {
-							let nodes = 0;
-							sourceFile.forEachChild(function walk(node) {
-								nodes++;
-								node.forEachChild(walk);
-							});
-							sourceFilesData.set(sourceFile, {
-								projectNames: [],
-								size: nodes * 128,
-							});
-						}
-						sourceFilesData.get(sourceFile)!.projectNames.push(projectName);
-					};
-				}
-			}
-
-			const result: {
-				inputs: {};
-				outputs: Record<string, {
-					imports: string[];
-					exports: string[];
-					entryPoint: string;
-					inputs: Record<string, { bytesInOutput: number; }>;
-					bytes: number;
-				}>;
-			} = {
-				inputs: {},
-				outputs: {},
-			};
-
-			for (const [sourceFile, fileData] of sourceFilesData) {
-				let key = fileData.projectNames.sort().join(', ');
-				if (fileData.projectNames.length >= 2) {
-					key = `Shared in ${fileData.projectNames.length} projects (${key})`;
-				}
-				result.outputs[key] ??= {
-					imports: [],
-					exports: [],
-					entryPoint: '',
-					inputs: {},
-					bytes: 0,
-				};
-				result.outputs[key].inputs[sourceFile.fileName] = { bytesInOutput: fileData.size };
-			}
-
-			return result;
-		});
 		server.connection.onNotification(UpdateVirtualCodeStateNotification.type, async params => {
 			const uri = URI.parse(params.fileUri);
 			const languageService = await project.getLanguageService(uri);

packages/language-server/lib/features/editorFeatures.ts

@@ -146,7 +146,3 @@ export namespace GetVirtualCodeRequest {
 	export type ErrorType = never;
 	export const type = new protocol.RequestType<ParamsType, ResponseType, ErrorType>('volar/client/virtualFile');
 }
-
-export namespace LoadedTSFilesMetaRequest {
-	export const type = new protocol.RequestType0('volar/client/loadedTsFiles');
-}

packages/language-server/protocol.ts

@@ -145,7 +145,3 @@ export namespace GetVirtualCodeRequest {
 	export type ErrorType = never;
 	export const type = new protocol.RequestType<ParamsType, ResponseType, ErrorType>('volar/client/virtualFile');
 }
-
-export namespace LoadedTSFilesMetaRequest {
-	export const type = new protocol.RequestType0('volar/client/loadedTsFiles');
-}