build.rs

  1use std::{
  2    env,
  3    path::PathBuf,
  4    process::{self, Command},
  5};
  6
  7fn main() {
  8    generate_dispatch_bindings();
  9    compile_context_predicate_parser();
 10    compile_metal_shaders();
 11    generate_shader_bindings();
 12}
 13
 14fn generate_dispatch_bindings() {
 15    println!("cargo:rustc-link-lib=framework=System");
 16    println!("cargo:rerun-if-changed=src/platform/mac/dispatch.h");
 17
 18    let bindings = bindgen::Builder::default()
 19        .header("src/platform/mac/dispatch.h")
 20        .whitelist_var("_dispatch_main_q")
 21        .whitelist_function("dispatch_async_f")
 22        .parse_callbacks(Box::new(bindgen::CargoCallbacks))
 23        .generate()
 24        .expect("unable to generate bindings");
 25
 26    let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
 27    bindings
 28        .write_to_file(out_path.join("dispatch_sys.rs"))
 29        .expect("couldn't write dispatch bindings");
 30}
 31
 32fn compile_context_predicate_parser() {
 33    let dir = PathBuf::from("./grammars/context-predicate/src");
 34    let parser_c = dir.join("parser.c");
 35
 36    println!("cargo:rerun-if-changed={}", &parser_c.to_str().unwrap());
 37    cc::Build::new()
 38        .include(&dir)
 39        .file(parser_c)
 40        .compile("tree_sitter_context_predicate");
 41}
 42
 43const SHADER_HEADER_PATH: &'static str = "./src/platform/mac/shaders/shaders.h";
 44
 45fn compile_metal_shaders() {
 46    let shader_path = "./src/platform/mac/shaders/shaders.metal";
 47    let air_output_path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("shaders.air");
 48    let metallib_output_path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("shaders.metallib");
 49
 50    println!("cargo:rerun-if-changed={}", SHADER_HEADER_PATH);
 51    println!("cargo:rerun-if-changed={}", shader_path);
 52
 53    let output = Command::new("xcrun")
 54        .args(&[
 55            "-sdk",
 56            "macosx",
 57            "metal",
 58            "-gline-tables-only",
 59            "-mmacosx-version-min=10.14",
 60            "-MO",
 61            "-c",
 62            shader_path,
 63            "-o",
 64        ])
 65        .arg(&air_output_path)
 66        .output()
 67        .unwrap();
 68
 69    if !output.status.success() {
 70        eprintln!(
 71            "metal shader compilation failed:\n{}",
 72            String::from_utf8_lossy(&output.stderr)
 73        );
 74        process::exit(1);
 75    }
 76
 77    let output = Command::new("xcrun")
 78        .args(&["-sdk", "macosx", "metallib"])
 79        .arg(air_output_path)
 80        .arg("-o")
 81        .arg(metallib_output_path)
 82        .output()
 83        .unwrap();
 84
 85    if !output.status.success() {
 86        eprintln!(
 87            "metallib compilation failed:\n{}",
 88            String::from_utf8_lossy(&output.stderr)
 89        );
 90        process::exit(1);
 91    }
 92}
 93
 94fn generate_shader_bindings() {
 95    let bindings = bindgen::Builder::default()
 96        .header(SHADER_HEADER_PATH)
 97        .whitelist_type("GPUI.*")
 98        .parse_callbacks(Box::new(bindgen::CargoCallbacks))
 99        .generate()
100        .expect("unable to generate bindings");
101
102    let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
103    bindings
104        .write_to_file(out_path.join("shaders.rs"))
105        .expect("couldn't write shader bindings");
106}