Add clap cli

DCNick3 · DCNick3 · commit c0dcb2d6b328 · 2023-03-04T03:40:57.000+03:00
diff --git a/cli/Cargo.toml b/cli/Cargo.toml
@@ -7,6 +7,6 @@ edition = "2021"
 
 [dependencies]
 shader-compiler-bridge = { path = "../bridge" }
-#naga = { version = "0.11.0", features = ["spv-in", "glsl-out"] }
 spirv_cross = { version = "0.23.1", features = ["glsl"] }
+clap = { version = "4.1.8", features = ["derive"] }
 byte-slice-cast = "1.2.2"
diff --git a/cli/src/main.rs b/cli/src/main.rs
@@ -35,18 +35,30 @@
 */
 
 use byte_slice_cast::AsByteSlice;
+use clap::Parser;
+use std::path::PathBuf;
+
+#[derive(clap::Parser)]
+struct Args {
+    /// Path to the input binary maxwell shader
+    ///
+    /// Expects a 0x30-byte header before the SPH
+    input: PathBuf,
+    /// Path to the output GLSL shader
+    output: PathBuf,
+}
 
 fn main() {
-    let shader_path = std::env::args().nth(1).unwrap();
+    let args = Args::parse();
 
-    let shader = std::fs::read(shader_path).unwrap();
+    let shader = std::fs::read(args.input).unwrap();
     // skip the 0x30-bytes header before the SPH
     let shader = &shader[0x30..];
 
     let spirv = shader_compiler_bridge::translate_shader(shader.to_vec());
     let spirv_bytes = spirv.as_byte_slice();
 
-    std::fs::write("shader.spv", spirv_bytes).unwrap();
+    // std::fs::write("shader.spv", spirv_bytes).unwrap();
 
     let module = spirv_cross::spirv::Module::from_words(&spirv);
     let mut ast = spirv_cross::spirv::Ast::<spirv_cross::glsl::Target>::parse(&module)
@@ -61,5 +73,5 @@ fn main() {
 
     let glsl = ast.compile().expect("Failed to compile GLSL");
 
-    println!("{}", glsl);
+    std::fs::write(args.output, glsl).unwrap();
 }
