tree-sitter-stack-graphs 0.10.0

Create stack graphs using tree-sitter parsers
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

// -*- coding: utf-8 -*-
// ------------------------------------------------------------------------------------------------
// Copyright © 2021, stack-graphs authors.
// Licensed under either of Apache License, Version 2.0, or MIT license, at your option.
// Please see the LICENSE-APACHE or LICENSE-MIT files in this distribution for license details.
// ------------------------------------------------------------------------------------------------

use std::path::Path;

use stack_graphs::arena::Handle;
use stack_graphs::graph::File;
use stack_graphs::graph::StackGraph;
use tree_sitter_graph::Variables;
use tree_sitter_stack_graphs::BuildError;
use tree_sitter_stack_graphs::NoCancellation;
use tree_sitter_stack_graphs::StackGraphLanguage;
use tree_sitter_stack_graphs::FILE_PATH_VAR;

mod builder;
mod edges;
mod loader;
mod nodes;
mod test;

pub(self) fn build_stack_graph(
    python_source: &str,
    tsg_source: &str,
) -> Result<(StackGraph, Handle<File>), BuildError> {
    let file_name = "test.py";
    let language =
        StackGraphLanguage::from_str(tree_sitter_python::LANGUAGE.into(), tsg_source).unwrap();
    let mut graph = StackGraph::new();
    let file = graph.get_or_create_file(file_name);
    let mut globals = Variables::new();
    let source_path = Path::new(file_name);

    globals
        .add(FILE_PATH_VAR.into(), source_path.to_str().unwrap().into())
        .expect("failed to add file path variable");

    language.build_stack_graph_into(&mut graph, file, python_source, &globals, &NoCancellation)?;
    Ok((graph, file))
}