Merge pull request #6 from workingjubilee/deal-with-distant-relatives

Fix build script
Kyllingene · May 2, 2024 · ac408c0 · ac408c0
2 parents 1c34751 + 419e5ea
commit ac408c0
Showing 1 changed file with 23 additions and 11 deletions.
diff --git a/build.rs b/build.rs
@@ -2,17 +2,29 @@ use nasm_rs;
 use std::env;
 
 fn main() {
-    if env::var("CARGO_CFG_TARGET_FAMILY").unwrap() == "unix" {
-		if let Err(e) = nasm_rs::compile_library_args("libunix_main.a", &["src/unix_main.asm"], &["-Psrc/sprites.inc", "-f elf64"]) {
-			// TODO: add better error handling
-			panic!("{}", e);
-		}
-
+    if env::var("CARGO_CFG_UNIX").is_ok() {
+        if let Err(e) = nasm_rs::compile_library_args(
+            "libunix_main.a",
+            &["src/unix_main.asm"],
+            &["-Psrc/sprites.inc", "-f elf64"],
+        ) {
+            // TODO: add better error handling
+            panic!("{}", e);
+        }
+    } else if env::var("CARGO_CFG_WINDOWS").is_ok() {
+        println!("cargo:rustc-link-lib=static=windows_main");
+        if let Err(e) = nasm_rs::compile_library_args(
+            "windows_main.lib",
+            &["src/windows_main.asm"],
+            &["-Psrc/sprites.inc", "-f win64"],
+        ) {
+            // TODO: add better error handling
+            panic!("{}", e);
+        }
     } else {
-		println!("cargo:rustc-link-lib=static=windows_main");
-        if let Err(e) = nasm_rs::compile_library_args("windows_main.lib", &["src/windows_main.asm"], &["-Psrc/sprites.inc", "-f win64"]) {
-			// TODO: add better error handling
-			panic!("{}", e);
-		}
+        panic!(
+            "hit or miss? i guess {} never miss, huh?",
+            env::var("CARGO_CFG_TARGET_OS").unwrap_or("they".into())
+        )
     }
 }