Supported references

Author: qiuxiang

src/index.ts

@@ -11,6 +11,7 @@ import { onCompletion } from "./completion";
 import { onDefinition } from "./definition";
 import { onFormatting } from "./formatting";
 import { onHover } from "./hover";
+import { onReferences } from "./references";
 import { onRename } from "./rename";
 import { Solidity } from "./solidity";
 
@@ -36,6 +37,7 @@ export function createServer(
   connection.onHover(onHover);
   connection.onCompletion(onCompletion);
   connection.onRenameRequest(onRename);
+  connection.onReferences(onReferences);
 
   connection.onDidChangeConfiguration(({ settings }) => {
     options = settings.solidity;

src/parse.ts

@@ -6,13 +6,16 @@ import {
   EventDefinition,
   Expression,
   FunctionDefinition,
+  Identifier,
   IdentifierPath,
   ImportDirective,
+  MemberAccess,
   ModifierDefinition,
   SourceUnit,
   Statement,
   StructDefinition,
   TypeName,
+  UserDefinedTypeName,
   UserDefinedValueTypeDefinition,
   VariableDeclaration,
 } from "solidity-ast";
@@ -50,10 +53,13 @@ export type Definition =
 
 export type DefinitionNode = Definition & ASTNodeData;
 export type ImportNode = ImportDirective & ASTNodeData;
+export type IdentifierNode = (Identifier | MemberAccess | UserDefinedTypeName) &
+  ASTNodeData;
 
 export function parse(
   node: ASTNode,
   root: ASTNode,
+  identifiers: ASTNode[],
   definitions: ASTNode[],
   scopes: Map<number, DefinitionNode[]>,
   nodes: ASTNode[],
@@ -83,6 +89,10 @@ export function parse(
       } else {
         scopes.set(scopeId, [node]);
       }
+    case "Identifier":
+    case "MemberAccess":
+    case "UserDefinedTypeName":
+      identifiers.push(node);
       break;
   }
   switch (node.nodeType) {
@@ -175,7 +185,7 @@ export function parse(
   for (const child of children) {
     if (!child) continue;
     child.parent = node;
-    parse(child, root, definitions, scopes, nodes, nodeMap);
+    parse(child, root, identifiers, definitions, scopes, nodes, nodeMap);
   }
 }
 

src/references.ts

@@ -0,0 +1,33 @@
+import { Location, Range, ReferenceParams } from "vscode-languageserver";
+import { documents, solidityMap } from ".";
+
+export function onReferences({
+  textDocument: { uri },
+  position,
+}: ReferenceParams): Location[] {
+  const document = documents.get(uri);
+  if (!document) return [];
+
+  const solidity = solidityMap.get(uri);
+  if (!solidity) return [];
+
+  const node = solidity.getCurrentNodes(position)[0];
+  let nodeId = node.id;
+  const ref = Reflect.get(node, "referencedDeclaration");
+  if (ref) {
+    nodeId = ref;
+  }
+  const items = solidity.identifiers.filter(
+    (i) => i.referencedDeclaration == nodeId
+  );
+
+  return items.map((i) => {
+    return Location.create(
+      uri,
+      Range.create(
+        document.positionAt(i.srcStart!),
+        document.positionAt(i.srcEnd!)
+      )
+    );
+  });
+}

src/solidity.ts

@@ -1,9 +1,16 @@
 import { SourceUnit } from "solidity-ast";
 import { Position, TextDocument } from "vscode-languageserver-textdocument";
-import { ASTNode, DefinitionNode, ImportNode, parse } from "./parse";
+import {
+  ASTNode,
+  DefinitionNode,
+  IdentifierNode,
+  ImportNode,
+  parse,
+} from "./parse";
 
 export class Solidity {
   document: TextDocument;
+  identifiers: IdentifierNode[] = [];
   definitions: DefinitionNode[] = [];
   nodes = new Map<string, ASTNode[]>();
   scopes = new Map<number, DefinitionNode[]>();
@@ -21,6 +28,7 @@ export class Solidity {
       parse(
         root,
         root,
+        this.identifiers,
         this.definitions,
         this.scopes,
         this.nodes.get(uri)!,