hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2026-02-19 08:23:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

WIP: populate numlines table

Browse Source
This commit is contained in:
Alex Ford
2021-04-19 18:06:35 +01:00
parent 3590a2c2ac
commit 62bf58b289
1 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
extractor/src/extractor.rs
View File

@@ -81,6 +81,25 @@ impl TrapWriter {
file_label
}
fn populate_numlines(&mut self, file_label: Label, tree: &Tree) {
let root_node = tree.root_node();
let start_line = root_node.start_position().row;
let end_line = root_node.end_position().row;
let total_line_count = end_line - start_line + 1;
// TODO: the code and comment counts are completely wrong
self.add_tuple(
"numlines",
vec![
Arg::Label(file_label),
Arg::Int(total_line_count),
Arg::Int(total_line_count),
Arg::Int(total_line_count),
],
);
}
fn populate_parent_folders(&mut self, child_label: Label, path: Option<&Path>) {
let mut path = path;
let mut child_label = child_label;
@@ -173,6 +192,7 @@ pub fn extract(
let mut trap_writer = new_trap_writer();
trap_writer.comment(format!("Auto-generated TRAP file for {}", path.display()));
let file_label = &trap_writer.populate_file(path);
&trap_writer.populate_numlines(*file_label, &tree);
let mut visitor = Visitor {
source: &source,
trap_writer: trap_writer,