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
44
45
46
47
48
49
50
51
52
53

// -*- 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 once_cell::sync::Lazy;
use pretty_assertions::assert_eq;
use stack_graphs::graph::StackGraph;
use std::path::PathBuf;
use tree_sitter::Language;
use tree_sitter_stack_graphs::loader::FileAnalyzers;
use tree_sitter_stack_graphs::loader::LanguageConfiguration;
use tree_sitter_stack_graphs::loader::Loader;
use tree_sitter_stack_graphs::NoCancellation;
use tree_sitter_stack_graphs::StackGraphLanguage;

static PATH: Lazy<PathBuf> = Lazy::new(|| PathBuf::from("test.py"));
static TSG: Lazy<String> = Lazy::new(|| {
    r#"
      (module) {}
    "#
    .to_string()
});

#[test]
fn can_load_from_provided_language_configuration() {
    let language: Language = tree_sitter_python::LANGUAGE.into();
    let sgl = StackGraphLanguage::from_str(language.clone(), &TSG).unwrap();
    let lc = LanguageConfiguration {
        language: language.clone(),
        scope: Some("source.py".into()),
        content_regex: None,
        file_types: vec!["py".into()],
        sgl,
        builtins: StackGraph::new(),
        special_files: FileAnalyzers::new(),
        no_similar_paths_in_file: false,
    };
    let mut loader =
        Loader::from_language_configurations(vec![lc], None).expect("Expected loader to succeed");

    let tsl = loader
        .load_tree_sitter_language_for_file(&PATH, &mut None)
        .expect("Expected loading tree-sitter language to succeed");
    assert_eq!(tsl, Some(&language));

    let lc = loader
        .load_for_file(&PATH, &mut None, &NoCancellation)
        .expect("Expected loading stack graph language to succeed");
    assert_eq!(lc.primary.map(|lc| &lc.language), Some(&language));
}