1use std::{
2 env,
3 path::{Path, PathBuf},
4 process::{self, Command},
5};
6
7use cbindgen::Config;
8
9fn main() {
10 generate_dispatch_bindings();
11 let header_path = generate_shader_bindings();
12 compile_metal_shaders(&header_path);
13}
14
15fn generate_dispatch_bindings() {
16 println!("cargo:rustc-link-lib=framework=System");
17 println!("cargo:rerun-if-changed=src/platform/mac/dispatch.h");
18
19 let bindings = bindgen::Builder::default()
20 .header("src/platform/mac/dispatch.h")
21 .allowlist_var("_dispatch_main_q")
22 .allowlist_function("dispatch_async_f")
23 .parse_callbacks(Box::new(bindgen::CargoCallbacks))
24 .layout_tests(false)
25 .generate()
26 .expect("unable to generate bindings");
27
28 let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
29 bindings
30 .write_to_file(out_path.join("dispatch_sys.rs"))
31 .expect("couldn't write dispatch bindings");
32}
33
34fn generate_shader_bindings() -> PathBuf {
35 let output_path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("scene.h");
36 let crate_dir = PathBuf::from(env::var("CARGO_MANIFEST_DIR").unwrap());
37 let mut config = Config::default();
38 config.include_guard = Some("SCENE_H".into());
39 config.language = cbindgen::Language::C;
40 config.export.include.extend([
41 "Bounds".into(),
42 "Corners".into(),
43 "Edges".into(),
44 "Size".into(),
45 "Pixels".into(),
46 "PointF".into(),
47 "Hsla".into(),
48 "Quad".into(),
49 "QuadInputIndex".into(),
50 "QuadUniforms".into(),
51 ]);
52 config.no_includes = true;
53 config.enumeration.prefix_with_name = true;
54 cbindgen::Builder::new()
55 .with_src(crate_dir.join("src/scene.rs"))
56 .with_src(crate_dir.join("src/geometry.rs"))
57 .with_src(crate_dir.join("src/color.rs"))
58 .with_src(crate_dir.join("src/platform/mac/metal_renderer.rs"))
59 .with_config(config)
60 .generate()
61 .expect("Unable to generate bindings")
62 .write_to_file(&output_path);
63 output_path
64}
65
66fn compile_metal_shaders(header_path: &Path) {
67 let shader_path = "./src/platform/mac/shaders.metal";
68 let air_output_path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("shaders.air");
69 let metallib_output_path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("shaders.metallib");
70
71 println!("cargo:rerun-if-changed={}", header_path.display());
72 println!("cargo:rerun-if-changed={}", shader_path);
73
74 let output = Command::new("xcrun")
75 .args([
76 "-sdk",
77 "macosx",
78 "metal",
79 "-gline-tables-only",
80 "-mmacosx-version-min=10.15.7",
81 "-MO",
82 "-c",
83 shader_path,
84 "-include",
85 &header_path.to_str().unwrap(),
86 "-o",
87 ])
88 .arg(&air_output_path)
89 .output()
90 .unwrap();
91
92 if !output.status.success() {
93 eprintln!(
94 "metal shader compilation failed:\n{}",
95 String::from_utf8_lossy(&output.stderr)
96 );
97 process::exit(1);
98 }
99
100 let output = Command::new("xcrun")
101 .args(["-sdk", "macosx", "metallib"])
102 .arg(air_output_path)
103 .arg("-o")
104 .arg(metallib_output_path)
105 .output()
106 .unwrap();
107
108 if !output.status.success() {
109 eprintln!(
110 "metallib compilation failed:\n{}",
111 String::from_utf8_lossy(&output.stderr)
112 );
113 process::exit(1);
114 }
115}