Rust: also implement localReferences.ql

Author: aibaars

rust/ql/lib/ide-contextual-queries/Definitions.qll

@@ -0,0 +1,17 @@
+/**
+ * Provides classes and predicates related to jump-to-definition links
+ * in the code viewer.
+ */
+
+private import codeql.rust.elements.Variable
+private import codeql.rust.elements.Locatable
+
+private predicate localVariable(Locatable e, Variable def) { e = def.getAnAccess() }
+
+/**
+ * Gets an element, of kind `kind`, that element `use` uses, if any.
+ */
+cached
+Variable definitionOf(Locatable use, string kind) {
+  localVariable(use, result) and kind = "local variable"
+}

rust/ql/lib/ide-contextual-queries/localDefinitions.ql

@@ -3,21 +3,19 @@
  * @description Generates use-definition pairs that provide the data
  *              for jump-to-definition in the code viewer.
  * @kind definitions
- * @id rus/ide-jump-to-definition
+ * @id rust/ide-jump-to-definition
  * @tags ide-contextual-queries/local-definitions
  */
 
 import codeql.IDEContextual
 import codeql.rust.elements.Variable
 import codeql.rust.elements.Locatable
+import Definitions
 
 external string selectedSourceFile();
 
-predicate localVariable(Locatable e, Variable def) { e = def.getAnAccess() }
-
 from Locatable e, Variable def, string kind
 where
-  e.getLocation().getFile() = getFileBySourceArchiveName(selectedSourceFile()) and
-  localVariable(e, def) and
-  kind = "local variable"
+  def = definitionOf(e, kind) and
+  e.getLocation().getFile() = getFileBySourceArchiveName(selectedSourceFile())
 select e, def, kind

rust/ql/lib/ide-contextual-queries/localReferences.ql

@@ -0,0 +1,21 @@
+/**
+ * @name Find-references links
+ * @description Generates use-definition pairs that provide the data
+ *              for find-references in the code viewer.
+ * @kind definitions
+ * @id rust/ide-find-references
+ * @tags ide-contextual-queries/local-references
+ */
+
+import codeql.IDEContextual
+import codeql.rust.elements.Locatable
+import codeql.rust.elements.Variable
+import Definitions
+
+external string selectedSourceFile();
+
+from Locatable e, Variable def, string kind
+where
+  def = definitionOf(e, kind) and
+  def.getLocation().getFile() = getFileBySourceArchiveName(selectedSourceFile())
+select e, def, kind