stack-graphs/tree-sitter-stack-graphs/tests/it/main.rs at c7f9fd9118b0642ae800d4c42237be577afa7d41 · nohehf/stack-graphs

Repository navigation

// -*- coding: utf-8 -*-

// ------------------------------------------------------------------------------------------------

// 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(), 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))

}