asterisk 0.1.0

Universal language-agnostic AST walking and accurate call stack generation with tree-sitter
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
58
59
60
61
62

use asterisk::config::Config;
/// Example CLI application demonstrating usage of the asterisk library.
///
/// This example:
/// 1. Reads the asterisk configuration from the `asterisk.toml` file
/// 2. Indexes the directory specified by the first command-line argument
/// 3. Generates a JSON output file with the indexed blocks and call stack
/// 4. Generates a Graphviz DOT file representing the call graph
///
/// To run the example, use:
/// ```
/// cargo run --example cli -- /path/to/directory/to/index
/// ```
use asterisk::indexer::index_directory;
use serde::Serialize;
use serde_json::json;
use std::env;
use std::fs;
use std::fs::File;
use std::io::Write;

#[derive(Serialize)]
struct Output {
    blocks: Vec<asterisk::block::Block>,
    call_stack: asterisk::call_stack::CallStack,
}

fn main() {
    let toml_str = fs::read_to_string("asterisk.toml").expect("Unable to read file");
    let config = Config::from_toml(&toml_str).unwrap();

    let args: Vec<String> = env::args().collect();
    let dir_path = &args[1];
    let (blocks, call_stack, call_graph) = index_directory(&config, dir_path);

    let output = Output { blocks, call_stack };

    let json_output = json!({
        "blocks": output.blocks,
        "call_stack": output.call_stack
    });

    let pretty_json = serde_json::to_string_pretty(&json_output).unwrap();

    let project_name = dir_path.split('/').last().unwrap_or("blockoli");
    let output_file_name = format!("{}.json", project_name);

    let mut output_file = File::create(&output_file_name).expect("Failed to create output file");
    write!(output_file, "{}", pretty_json).expect("Failed to write to output file");
    println!("Indexing completed. Output written to {}", output_file_name);

    let graphviz = call_graph.to_graphviz();
    let graphviz_file_name = format!("{}_call_graph.dot", project_name);
    let mut graphviz_file =
        File::create(&graphviz_file_name).expect("Failed to create Graphviz file");
    write!(graphviz_file, "{}", graphviz).expect("Failed to write to Graphviz file");

    println!(
        "Call graph generated. Graphviz file written to {}",
        graphviz_file_name
    );
}