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