ustar-tools 0.1.4

Command-line tools for STAR format parsing and processing
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

use clap::Parser;
use pest_railroad::generate_diagram;
use std::fs;
use std::path::PathBuf;

/// Generate railroad diagrams for USTAR grammar files
#[derive(Parser)]
#[command(name = "ustar-grammar-railroad")]
#[command(about = "Generate railroad diagrams for USTAR grammar files")]
struct Cli {
    /// Grammar file to process
    grammar_file: PathBuf,

    /// Output SVG file (defaults to input filename with .svg extension)
    #[arg(short, long)]
    output: Option<PathBuf>,

    /// Show verbose output
    #[arg(short, long)]
    verbose: bool,
}

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let cli = Cli::parse();

    // Determine output file path
    let output_file = if let Some(output) = cli.output {
        output
    } else {
        // Default: replace input extension with .svg
        cli.grammar_file.with_extension("svg")
    };

    if cli.verbose {
        println!("Input grammar file: {}", cli.grammar_file.display());
        println!("Output SVG file: {}", output_file.display());
    }

    // Read grammar file
    let grammar_content = fs::read_to_string(&cli.grammar_file).map_err(|e| {
        format!(
            "Failed to read grammar file {}: {}",
            cli.grammar_file.display(),
            e
        )
    })?;

    if cli.verbose {
        println!("Grammar file loaded, {} bytes", grammar_content.len());
    }

    // Generate railroad diagram
    match generate_diagram(&grammar_content) {
        Ok((diagram, warnings)) => {
            // Print any warnings
            if !warnings.is_empty() {
                if cli.verbose {
                    println!("Warnings:");
                    for warning in &warnings {
                        println!("  {}", warning);
                    }
                } else {
                    println!(
                        "Note: {} warnings about unsupported grammar features",
                        warnings.len()
                    );
                }
            }

            // Convert diagram to SVG
            let svg = format!("{}", diagram);
            fs::write(&output_file, &svg)?;

            println!("Generated railroad diagram: {}", output_file.display());
        }
        Err(e) => {
            eprintln!("Failed to generate railroad diagram: {}", e);
            std::process::exit(1);
        }
    }

    Ok(())
}