feat: Speed up a bit JSX linked editing by simply caching previous response

Author: zardoy

src/configurationType.ts

@@ -582,6 +582,12 @@ export type Configuration = {
      * @default false
      */
     'experiments.enableInsertNameOfSuggestionFix': boolean
+    /**
+     * Speed up JSX linked editing by not using actual tsserver command when possible, which in theory may introduce some inconsistencies.
+     * Note that currently it doesn't really help if you have `"typescript.tsserver.useSyntaxServer": "auto"` in the settings.
+     * @default true
+     */
+    'experiments.speedLinkedEditing': boolean
     /**
      * Map *symbol - array of modules* to change sorting of imports - first available takes precedence in auto import code fixes (+ import all action)
      *

typescript/src/decorateLinkedEditing.ts

@@ -0,0 +1,37 @@
+import { GetConfig } from './types'
+
+export default (proxy: ts.LanguageService, languageService: ts.LanguageService, languageServiceHost: ts.LanguageServiceHost, c: GetConfig) => {
+    // patch JSX tag linked editing to improve performance (needed for great user experience)
+
+    let lastLinkedEditingRangeRequest:
+        | {
+              pos: number
+              fileName: string
+              result: ts.LinkedEditingInfo
+          }
+        | undefined
+    proxy.getLinkedEditingRangeAtPosition = (fileName, position) => {
+        if (
+            c('experiments.speedLinkedEditing') &&
+            lastLinkedEditingRangeRequest &&
+            lastLinkedEditingRangeRequest.pos === position - 1 &&
+            lastLinkedEditingRangeRequest.fileName === fileName
+        ) {
+            lastLinkedEditingRangeRequest.pos = position
+            lastLinkedEditingRangeRequest.result.ranges[0]!.length++
+            lastLinkedEditingRangeRequest.result.ranges[1]!.start++
+            lastLinkedEditingRangeRequest.result.ranges[1]!.length++
+            return lastLinkedEditingRangeRequest.result
+        }
+        lastLinkedEditingRangeRequest = undefined
+
+        const prior = languageService.getLinkedEditingRangeAtPosition(fileName, position)
+        if (!prior) return
+        lastLinkedEditingRangeRequest = {
+            pos: position,
+            fileName,
+            result: globalThis.structuredClone(prior),
+        }
+        return prior
+    }
+}

typescript/src/decorateProxy.ts

@@ -19,6 +19,7 @@ import decorateSignatureHelp from './decorateSignatureHelp'
 import decorateFindRenameLocations from './decorateFindRenameLocations'
 import decorateQuickInfoAtPosition from './decorateQuickInfoAtPosition'
 import decorateEditsForFileRename from './decorateEditsForFileRename'
+import decorateLinkedEditing from './decorateLinkedEditing'
 
 /** @internal */
 export const thisPluginMarker = '__essentialPluginsMarker__'
@@ -102,6 +103,7 @@ export const decorateLanguageService = (
     decorateSignatureHelp(proxy, languageService, languageServiceHost, c)
     decorateFindRenameLocations(proxy, languageService, c)
     decorateQuickInfoAtPosition(proxy, languageService, languageServiceHost, c)
+    decorateLinkedEditing(proxy, languageService, languageServiceHost, c)
 
     libDomPatching(languageServiceHost, c)