tree-sitter-lean4 0.0.2

Tree-sitter grammar for Lean 4 (generates parser at build time)
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

use std::path::{Path, PathBuf};
use std::process::Command;
use std::{fs, io};

fn copy_dir(src: &Path, dst: &Path) -> io::Result<()> {
    fs::create_dir_all(dst)?;
    for entry in fs::read_dir(src)? {
        let entry = entry?;
        if entry.path().is_file() {
            // Use read/write instead of copy (nix store compatibility)
            let content = fs::read(entry.path())?;
            fs::write(dst.join(entry.file_name()), content)?;
        }
    }
    Ok(())
}

fn generate_parser(out_dir: &Path) -> PathBuf {
    fs::create_dir_all(out_dir).expect("create OUT_DIR");

    // Copy files using read+write (nix store has special permissions that break fs::copy)
    let grammar = fs::read("grammar.js").expect("read grammar.js");
    fs::write(out_dir.join("grammar.js"), grammar).expect("write grammar.js");
    copy_dir(Path::new("grammar"), &out_dir.join("grammar"))
        .unwrap_or_else(|e| panic!("copy grammar/: {}", e));
    copy_dir(Path::new("src"), &out_dir.join("src")).ok();

    println!("cargo:warning=Generating parser.c in {}...", out_dir.display());
    let status = Command::new("tree-sitter")
        .arg("generate")
        .current_dir(out_dir)
        .status()
        .expect("tree-sitter CLI not found");
    assert!(status.success(), "tree-sitter generate failed");

    out_dir.join("src")
}

fn main() {
    println!("cargo:rerun-if-changed=grammar.js");
    println!("cargo:rerun-if-changed=src/scanner.c");
    println!("cargo:rerun-if-changed=src/parser.c");

    let src_dir = if Path::new("src/parser.c").exists() {
        PathBuf::from("src")
    } else {
        let out_dir = PathBuf::from(std::env::var("OUT_DIR").unwrap());
        generate_parser(&out_dir)
    };

    cc::Build::new()
        .include(&src_dir)
        .flag_if_supported("-Wno-unused-parameter")
        .flag_if_supported("-Wno-unused-but-set-variable")
        .flag_if_supported("-Wno-trigraphs")
        .file(src_dir.join("parser.c"))
        .file(src_dir.join("scanner.c"))
        .compile("parser");
}