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_var("DISPATCH_QUEUE_PRIORITY_DEFAULT")
23 .allowlist_var("DISPATCH_TIME_NOW")
24 .allowlist_function("dispatch_get_global_queue")
25 .allowlist_function("dispatch_async_f")
26 .allowlist_function("dispatch_after_f")
27 .allowlist_function("dispatch_time")
28 .parse_callbacks(Box::new(bindgen::CargoCallbacks))
29 .layout_tests(false)
30 .generate()
31 .expect("unable to generate bindings");
32
33 let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
34 bindings
35 .write_to_file(out_path.join("dispatch_sys.rs"))
36 .expect("couldn't write dispatch bindings");
37}
38
39fn generate_shader_bindings() -> PathBuf {
40 let output_path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("scene.h");
41 let crate_dir = PathBuf::from(env::var("CARGO_MANIFEST_DIR").unwrap());
42 let mut config = Config::default();
43 config.include_guard = Some("SCENE_H".into());
44 config.language = cbindgen::Language::C;
45 config.export.include.extend([
46 "Bounds".into(),
47 "Corners".into(),
48 "Edges".into(),
49 "Size".into(),
50 "Pixels".into(),
51 "PointF".into(),
52 "Hsla".into(),
53 "ContentMask".into(),
54 "Uniforms".into(),
55 "AtlasTile".into(),
56 "PathRasterizationInputIndex".into(),
57 "PathVertex_ScaledPixels".into(),
58 "ShadowInputIndex".into(),
59 "Shadow".into(),
60 "QuadInputIndex".into(),
61 "Underline".into(),
62 "UnderlineInputIndex".into(),
63 "Quad".into(),
64 "SpriteInputIndex".into(),
65 "MonochromeSprite".into(),
66 "PolychromeSprite".into(),
67 "PathSprite".into(),
68 "SurfaceInputIndex".into(),
69 "SurfaceBounds".into(),
70 ]);
71 config.no_includes = true;
72 config.enumeration.prefix_with_name = true;
73
74 let mut builder = cbindgen::Builder::new();
75
76 let src_paths = [
77 crate_dir.join("src/scene.rs"),
78 crate_dir.join("src/geometry.rs"),
79 crate_dir.join("src/color.rs"),
80 crate_dir.join("src/window.rs"),
81 crate_dir.join("src/platform.rs"),
82 crate_dir.join("src/platform/mac/metal_renderer.rs"),
83 ];
84 for src_path in src_paths {
85 println!("cargo:rerun-if-changed={}", src_path.display());
86 builder = builder.with_src(src_path);
87 }
88
89 builder
90 .with_config(config)
91 .generate()
92 .expect("Unable to generate bindings")
93 .write_to_file(&output_path);
94
95 output_path
96}
97
98fn compile_metal_shaders(header_path: &Path) {
99 let shader_path = "./src/platform/mac/shaders.metal";
100 let air_output_path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("shaders.air");
101 let metallib_output_path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("shaders.metallib");
102
103 println!("cargo:rerun-if-changed={}", header_path.display());
104 println!("cargo:rerun-if-changed={}", shader_path);
105
106 let output = Command::new("xcrun")
107 .args([
108 "-sdk",
109 "macosx",
110 "metal",
111 "-gline-tables-only",
112 "-mmacosx-version-min=10.15.7",
113 "-MO",
114 "-c",
115 shader_path,
116 "-include",
117 &header_path.to_str().unwrap(),
118 "-o",
119 ])
120 .arg(&air_output_path)
121 .output()
122 .unwrap();
123
124 if !output.status.success() {
125 eprintln!(
126 "metal shader compilation failed:\n{}",
127 String::from_utf8_lossy(&output.stderr)
128 );
129 process::exit(1);
130 }
131
132 let output = Command::new("xcrun")
133 .args(["-sdk", "macosx", "metallib"])
134 .arg(air_output_path)
135 .arg("-o")
136 .arg(metallib_output_path)
137 .output()
138 .unwrap();
139
140 if !output.status.success() {
141 eprintln!(
142 "metallib compilation failed:\n{}",
143 String::from_utf8_lossy(&output.stderr)
144 );
145 process::exit(1);
146 }
147}