Rust: lines of code, exclude 'SourceFile' node

Author: aibaars

rust/ql/lib/codeql/files/FileSystem.qll

@@ -2,6 +2,8 @@
 
 private import codeql.Locations
 private import codeql.util.FileSystem
+private import codeql.rust.elements.SourceFile
+private import codeql.rust.elements.AstNode
 
 private module Input implements InputSig {
   abstract class ContainerBase extends @container {
@@ -41,8 +43,10 @@ class File extends Container, Impl::File {
   int getNumberOfLinesOfCode() {
     result =
       count(int line |
-        exists(Location loc |
-          loc.getFile() = this and
+        exists(AstNode node, Location loc |
+          not node instanceof SourceFile and loc = node.getLocation()
+        |
+          node.getFile() = this and
           line = [loc.getStartLine(), loc.getEndLine()] and
           not loc instanceof EmptyLocation
         )

rust/ql/test/query-tests/diagnostics/LinesOfCode.expected

@@ -1 +1 @@
-| 49 |
+| 48 |

rust/ql/test/query-tests/diagnostics/LinesOfUserCode.expected

@@ -1 +1 @@
-| 49 |
+| 48 |

rust/ql/test/query-tests/diagnostics/LinesOfUserCodeInFiles.expected

@@ -1,6 +1,6 @@
 | my_struct.rs:0:0:0:0 | my_struct.rs | 21 |
 | my_macro.rs:0:0:0:0 | my_macro.rs | 8 |
-| lib.rs:0:0:0:0 | lib.rs | 7 |
 | main.rs:0:0:0:0 | main.rs | 7 |
+| lib.rs:0:0:0:0 | lib.rs | 6 |
 | does_not_compile.rs:0:0:0:0 | does_not_compile.rs | 3 |
 | error.rs:0:0:0:0 | error.rs | 3 |

rust/ql/test/query-tests/diagnostics/SummaryStats.expected

@@ -1,5 +1,5 @@
 | Elements extracted | 204 |
 | Elements unextracted | 0 |
 | Files extracted | 6 |
-| Lines of code extracted | 49 |
-| Lines of user code extracted | 49 |
+| Lines of code extracted | 48 |
+| Lines of user code extracted | 48 |