treeedb 0.1.0

Generate Datalog facts from tree-sitter parse trees
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57

use std::fs::File;
use std::io;
use std::path::PathBuf;

use csv::Writer;
use tree_sitter::Node;

use super::consumer::FactConsumer;

pub struct WideCsvConsumer {
    node: csv::Writer<File>,
    field: csv::Writer<File>,
}

impl WideCsvConsumer {
    pub fn new(node_file_path: PathBuf, field_file_path: PathBuf) -> Result<Self, io::Error> {
        Ok(WideCsvConsumer {
            node: Writer::from_writer(File::create(node_file_path)?),
            field: Writer::from_writer(File::create(field_file_path)?),
        })
    }
}

impl FactConsumer for WideCsvConsumer {
    type Err = csv::Error;

    fn field(&mut self, parent: &Node, name: &'static str, child: &Node) -> Result<(), Self::Err> {
        self.field
            .write_record(&[&parent.id().to_string(), name, &child.id().to_string()])?;
        Ok(())
    }

    fn node(&mut self, node: &Node, source: &[u8]) -> Result<(), Self::Err> {
        let start = node.start_position();
        let end = node.end_position();
        self.node.write_record(&[
            &node.id().to_string(),
            node.kind(),
            &node.is_named().to_string(),
            &node.is_extra().to_string(),
            &node.is_error().to_string(),
            &node.is_missing().to_string(),
            &node.start_byte().to_string(),
            &node.end_byte().to_string(),
            &start.row.to_string(),
            &start.column.to_string(),
            &end.row.to_string(),
            &end.column.to_string(),
            &node
                .utf8_text(source)
                .expect("Source file was not UTF-8")
                .to_string()
                .replace('\n', "\\n"),
        ])?;
        Ok(())
    }
}