feat: customizeEnabledFeatures to allow disable some TS language service features per file or globally (e.g. completions or diagnostics)

feat: print warning in logs if some operation tooks too much time

Author: zardoy

buildTsPlugin.mjs

@@ -30,6 +30,7 @@ const result = await buildTsPlugin('typescript', undefined, undefined, {
         js: 'let ts, tsFull;',
         // js: 'const log = (...args) => console.log(...args.map(a => JSON.stringify(a)))',
     },
+    external: ['perf_hooks'],
     plugins: [
         {
             name: 'watch-notifier',

src/configurationType.ts

@@ -1,4 +1,4 @@
-import { ScriptElementKind, ScriptKind } from 'typescript/lib/tsserverlibrary'
+import { ScriptElementKind, ScriptKind, LanguageService } from 'typescript/lib/tsserverlibrary'
 
 type ReplaceRule = {
     /**
@@ -660,6 +660,21 @@ export type Configuration = {
               typeAlias: string
               interface: string
           }
+    customizeEnabledFeatures: {
+        [path: string]:
+            | 'disable-auto-invoked'
+            // | 'disable-heavy-features'
+            | {
+                  /** @default true */
+                  [feature in keyof LanguageService]: boolean
+              }
+    }
+    // bigFilesLimitFeatures: 'do-not-limit' | 'limit-auto-invoking' | 'force-limit-all-features'
+    /**
+     * in kb default is 1.5mb
+     * @default 100000
+     */
+    // bigFilesThreshold: number
     /** @default false */
     enableHooksFile: boolean
 }

typescript/src/decorateProxy.ts

@@ -139,6 +139,49 @@ export const decorateLanguageService = (
         }
     }
 
+    const readonlyModeDisableFeatures: Array<keyof ts.LanguageService> = [
+        'getOutliningSpans',
+        'getSyntacticDiagnostics',
+        'getSemanticDiagnostics',
+        'getSuggestionDiagnostics',
+        'provideInlayHints',
+        'getLinkedEditingRangeAtPosition',
+        'getApplicableRefactors',
+        'getCompletionsAtPosition',
+        'getDefinitionAndBoundSpan',
+        'getFormattingEditsAfterKeystroke',
+        'getDocumentHighlights',
+    ]
+    for (const feature of readonlyModeDisableFeatures) {
+        const orig = proxy[feature]
+        proxy[feature] = (...args) => {
+            const enabledFeaturesSetting = c('customizeEnabledFeatures') ?? {}
+            const toDisableRaw =
+                Object.entries(enabledFeaturesSetting).find(([path]) => {
+                    if (typeof args[0] !== 'string') return false
+                    return args[0].includes(path)
+                })?.[1] ??
+                enabledFeaturesSetting['*'] ??
+                {}
+            const toDisable: string[] =
+                toDisableRaw === 'disable-auto-invoked'
+                    ? // todo
+                      readonlyModeDisableFeatures
+                    : Object.entries(toDisableRaw)
+                          .filter(([, v]) => v === false)
+                          .map(([k]) => k)
+            if (toDisable.includes(feature)) return undefined
+            // eslint-disable-next-line @typescript-eslint/no-require-imports
+            const performance = globalThis.performance ?? require('perf_hooks').performance
+            const start = performance.now()
+            //@ts-expect-error
+            const result = orig(...args)
+            const time = performance.now() - start
+            if (time > 100) console.log(`[typescript-vscode-plugin perf warning] ${feature} took ${time}ms: ${args[0]} ${args[1]}`)
+            return result
+        }
+    }
+
     languageService[thisPluginMarker] = true
 
     if (!__WEB__ && c('enableHooksFile')) {