feat(special-command): Print Performance & Memory Info

Author: zardoy

package.json

@@ -66,6 +66,10 @@
             {
                 "command": "getArgumentReferencesFromCurrentParameter",
                 "title": "Get Argument References from Current Parameter"
+            },
+            {
+                "command": "printPerformanceMemoryInfo",
+                "title": "Print Performance & Memory Info"
             }
         ],
         "keybindings": [

src/onCompletionAccepted.ts

@@ -116,7 +116,10 @@ export default (tsApi: { onCompletionAccepted }) => {
                         resolve(true)
                     }
                 })
-                if (accepted) void vscode.window.showInformationMessage('Completion accepted, see console for details')
+                if (accepted) {
+                    void vscode.window.showInformationMessage('Completion accepted, see console for details')
+                    console.show(true)
+                }
             },
         )
     })

src/specialCommands.ts

@@ -302,6 +302,13 @@ export default () => {
         )
     })
 
+    registerExtensionCommand('printPerformanceMemoryInfo', async () => {
+        const info = await sendCommand('performanceInfo', {})
+        if (!info) return
+        console.dir(info, { depth: 10 })
+        console.show(true)
+    })
+
     // registerExtensionCommand('insertImportFlatten', () => {
     //     // got -> default, got
     //     type A = ts.Type

typescript/src/ipcTypes.ts

@@ -16,6 +16,7 @@ export const triggerCharacterCommands = [
     'getExtendedCodeActionEdits',
     'getLastResolvedCompletion',
     'getArgumentReferencesFromCurrentParameter',
+    'performanceInfo',
 ] as const
 
 export type TriggerCharacterCommand = (typeof triggerCharacterCommands)[number]

typescript/src/specialCommands/handle.ts

@@ -277,6 +277,34 @@ export default (
             }),
         )
     }
+    if (specialCommand === 'performanceInfo') {
+        const toMb = (bytes: number) => Math.floor(bytes / 1024 / 1024)
+
+        const sourceFilesContents = Object.fromEntries(
+            languageService
+                .getProgram()!
+                .getSourceFiles()
+                .map(x => [x.fileName, x.getFullText().length]),
+        )
+        return {
+            sourceFiles: {
+                totalFilesNumber: Object.keys(sourceFilesContents).length,
+                total: toMb(Object.entries(sourceFilesContents).reduce((a, [, length]) => a + length, 0)),
+                json: toMb(
+                    Object.entries(sourceFilesContents)
+                        .filter(([fileName]) => fileName.endsWith('.json'))
+                        .map(x => x[1])
+                        .reduce((a, b) => a + b, 0),
+                ),
+                top10: Object.entries(sourceFilesContents)
+                    .sort(([, a], [, b]) => b - a)
+                    .slice(0, 10)
+                    .map(([fileName, size]) => ({ fileName, size: toMb(size), raw: size })),
+            },
+            // approx
+            memoryUsedMb: toMb(process.memoryUsage().heapUsed),
+        }
+    }
 
     return null
 }