Rust: Adjust some generated Impl files

Author: hvitved

rust/codegen.conf

@@ -1,4 +1,4 @@
-# configuration file for Swift code generation default options
+# configuration file for Rust code generation default options
 --generate=dbscheme,rusttest,ql,rust
 --dbscheme=ql/lib/rust.dbscheme
 --ql-output=ql/lib/codeql/rust/generated

rust/ql/.generated.list

@@ -1,4 +1,3 @@
-// generated by codegen, remove this comment if you wish to edit this file
 /**
  * This module provides a hand-modifiable wrapper around the generated class `Element`.
  *
@@ -12,5 +11,9 @@ private import codeql.rust.generated.Element
  * be referenced directly.
  */
 module Impl {
-  class Element extends Generated::Element { }
+  class Element extends Generated::Element {
+    override string toString() { result = this.getAPrimaryQlClass() }
+
+    predicate isUnknown() { none() } // compatibility with test generation, to be fixed
+  }
 }

rust/ql/.gitattributes

@@ -1,4 +1,3 @@
-// generated by codegen, remove this comment if you wish to edit this file
 /**
  * This module provides a hand-modifiable wrapper around the generated class `Function`.
  *
@@ -12,6 +11,7 @@ private import codeql.rust.generated.Function
  * be referenced directly.
  */
 module Impl {
+  // the following QLdoc is generated: if you need to edit it, do it in the schema file
   /**
    * A function declaration. For example
    * ```
@@ -24,5 +24,7 @@ module Impl {
    * }
    * ```
    */
-  class Function extends Generated::Function { }
+  class Function extends Generated::Function {
+    override string toString() { result = this.getName() }
+  }
 }

rust/ql/lib/codeql/rust/elements/ElementImpl.qll

@@ -1,16 +1,36 @@
-// generated by codegen, remove this comment if you wish to edit this file
 /**
  * This module provides a hand-modifiable wrapper around the generated class `Locatable`.
  *
  * INTERNAL: Do not use.
  */
 
 private import codeql.rust.generated.Locatable
+import codeql.Locations
+private import codeql.rust.generated.Synth
+private import codeql.rust.generated.Raw
 
 /**
  * INTERNAL: This module contains the customizable definition of `Locatable` and should not
  * be referenced directly.
  */
 module Impl {
-  class Locatable extends Generated::Locatable { }
+  class Locatable extends Generated::Locatable {
+    pragma[nomagic]
+    final Location getLocation() {
+      exists(Raw::Locatable raw |
+        raw = Synth::convertLocatableToRaw(this) and
+        (
+          locatable_locations(raw, result)
+          or
+          not exists(Location loc | locatable_locations(raw, loc)) and
+          result instanceof EmptyLocation
+        )
+      )
+    }
+
+    /**
+     * Gets the primary file where this element occurs.
+     */
+    File getFile() { result = this.getLocation().getFile() }
+  }
 }