melior-macro 0.13.2

Internal macros for Melior
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

use std::{env, error::Error, path::Path, process::Command, str};

const LLVM_MAJOR_VERSION: usize = 19;

fn main() -> Result<(), Box<dyn Error>> {
    let version_variable = format!("MLIR_SYS_{}0_PREFIX", LLVM_MAJOR_VERSION);

    println!("cargo:rerun-if-env-changed={version_variable}");
    println!(
        "cargo:rustc-env=LLVM_INCLUDE_DIRECTORY={}",
        // spell-checker: disable-next-line
        llvm_config("--includedir", &version_variable)?
    );

    Ok(())
}

fn llvm_config(
    argument: &str,
    version_variable: &str,
) -> Result<String, Box<dyn std::error::Error>> {
    let prefix = env::var(version_variable)
        .map(|path| Path::new(&path).join("bin"))
        .unwrap_or_default();
    let call = format!(
        "{} --link-static {}",
        prefix.join("llvm-config").display(),
        argument
    );

    Ok(str::from_utf8(
        &if cfg!(target_os = "windows") {
            Command::new("cmd").args(["/C", &call]).output()?
        } else {
            Command::new("sh").arg("-c").arg(&call).output()?
        }
        .stdout,
    )?
    .trim()
    .to_string())
}