feat(getArgumentReferencesFromCurrentParameter): add getArgumentReferencesFromCurrentParameter special command

Author: Ilya Golovin

package.json

@@ -62,6 +62,10 @@
             {
                 "command": "replaceGlobalTypescriptWithLocalVersion",
                 "title": "Replace Global Typescript with Local Version"
+            },
+            {
+                "command": "getArgumentReferencesFromCurrentParameter",
+                "title": "Get Argument References from Current Parameter"
             }
         ],
         "keybindings": [

src/specialCommands.ts

@@ -289,6 +289,21 @@ export default () => {
         await vscode.env.clipboard.writeText(text)
     })
 
+    registerExtensionCommand('getArgumentReferencesFromCurrentParameter', async () => {
+        const result = await sendCommand<RequestResponseTypes['getArgumentReferencesFromCurrentParameter']>('getArgumentReferencesFromCurrentParameter')
+        if (!result) return
+        const editor = vscode.window.activeTextEditor!
+        const { uri } = editor.document
+        await vscode.commands.executeCommand(
+            'editor.action.goToLocations',
+            uri,
+            editor.selection.active,
+            result.map(({ filename, line, character }) => new vscode.Location(vscode.Uri.file(filename), new vscode.Position(line, character))),
+            vscode.workspace.getConfiguration('editor').get('gotoLocation.multipleReferences') ?? 'peek',
+            'No references',
+        )
+    })
+
     // registerExtensionCommand('insertImportFlatten', () => {
     //     // got -> default, got
     //     type A = ts.Type

typescript/src/ipcTypes.ts

@@ -15,6 +15,7 @@ export const triggerCharacterCommands = [
     'acceptRenameWithParams',
     'getExtendedCodeActionEdits',
     'getLastResolvedCompletion',
+    'getArgumentReferencesFromCurrentParameter',
 ] as const
 
 export type TriggerCharacterCommand = (typeof triggerCharacterCommands)[number]
@@ -84,6 +85,7 @@ export type RequestResponseTypes = {
     getFullType: {
         text: string
     }
+    getArgumentReferencesFromCurrentParameter: Array<{ line: number; character: number; filename: string }>
 }
 
 // INPUT

typescript/src/specialCommands/handle.ts

@@ -246,6 +246,37 @@ export default (
             text: checker.typeToString(type, undefined, ts.TypeFormatFlags.NoTruncation | ts.TypeFormatFlags.NoTypeReduction),
         }
     }
+    if (specialCommand === 'getArgumentReferencesFromCurrentParameter') {
+        const node = findChildContainingExactPosition(sourceFile, position)
+        if (!node || !ts.isIdentifier(node) || !ts.isParameter(node.parent) || !ts.isFunctionLike(node.parent.parent)) return
+        let functionDecl = node.parent.parent as ts.Node
+        const functionParameters = node.parent.parent.parameters
+        if (ts.isVariableDeclaration(functionDecl.parent)) {
+            functionDecl = functionDecl.parent
+        }
+        const parameterIndex = functionParameters.indexOf(node.parent)
+        const references = languageService.findReferences(fileName, functionDecl.pos + functionDecl.getLeadingTriviaWidth(sourceFile))
+        if (!references) return
+
+        return compact(
+            references.flatMap(({ references }) => {
+                return references.map(reference => {
+                    const sourceFile = languageService.getProgram()!.getSourceFile(reference.fileName)!
+                    const position = reference.textSpan.start
+
+                    const node = findChildContainingExactPosition(sourceFile, position)
+                    if (!node || !ts.isIdentifier(node) || !ts.isCallExpression(node.parent)) return
+
+                    const arg = node.parent.arguments[parameterIndex]
+                    if (!arg) return
+                    return {
+                        filename: reference.fileName,
+                        ...sourceFile.getLineAndCharacterOfPosition(arg.pos + arg.getLeadingTriviaWidth(sourceFile)),
+                    }
+                })
+            }),
+        )
+    }
 
     return null
 }