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(&["-sdk", "macosx", "metal", "-c", shader_path, "-o"])
55 .arg(&air_output_path)
56 .output()
57 .unwrap();
58
59 if !output.status.success() {
60 eprintln!(
61 "metal shader compilation failed:\n{}",
62 String::from_utf8_lossy(&output.stderr)
63 );
64 process::exit(1);
65 }
66
67 let output = Command::new("xcrun")
68 .args(&["-sdk", "macosx", "metallib"])
69 .arg(air_output_path)
70 .arg("-o")
71 .arg(metallib_output_path)
72 .output()
73 .unwrap();
74
75 if !output.status.success() {
76 eprintln!(
77 "metallib compilation failed:\n{}",
78 String::from_utf8_lossy(&output.stderr)
79 );
80 process::exit(1);
81 }
82}
83
84fn generate_shader_bindings() {
85 let bindings = bindgen::Builder::default()
86 .header(SHADER_HEADER_PATH)
87 .whitelist_type("GPUIQuadInputIndex")
88 .whitelist_type("GPUIQuad")
89 .whitelist_type("GPUIQuadUniforms")
90 .parse_callbacks(Box::new(bindgen::CargoCallbacks))
91 .generate()
92 .expect("unable to generate bindings");
93
94 let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
95 bindings
96 .write_to_file(out_path.join("shaders.rs"))
97 .expect("couldn't write shader bindings");
98}