Find references code lens, room for optimization

1 files changed, 7 insertions, 1 deletions

diff --git a/lib/extension.ts b/lib/extension.ts
index dc5c2fd..a735493 100644
--- a/lib/extension.ts
+++ b/lib/extension.ts
@@ -103,7 +103,8 @@ export function activate(context: ExtensionContext) {
     context.subscriptions.push(disposable);
 
     // Register test commands
-	commands.registerCommand('java.command.test.run', runTest);
+    commands.registerCommand('java.command.test.run', runTest);
+    commands.registerCommand('java.command.findReferences', runFindReferences);
 
 	// When the language client activates, register a progress-listener
 	client.onReady().then(() => createProgressListeners(client));
@@ -177,6 +178,11 @@ function findJavaExecutableInJavaHome(javaHome: string, binname: string) {
 export function deactivate() {
 }
 
+function runFindReferences(uri: string, lineNumber: number, column: number) {
+    // LSP is 0-based but VSCode is 1-based
+    return commands.executeCommand('editor.action.findReferences', Uri.parse(uri), {lineNumber: lineNumber+1, column: column+1});
+}
+
 interface JavaTestTask extends TaskDefinition {
     enclosingClass: string
     method: string