Add WAD support to VFS & wad commands

src/cmd/mod.rs

@@ -14,5 +14,6 @@ pub mod rso;
 pub mod shasum;
 pub mod u8_arc;
 pub mod vfs;
+pub mod wad;
 pub mod yay0;
 pub mod yaz0;

src/cmd/wad.rs

@@ -0,0 +1,108 @@
+use anyhow::Result;
+use argp::FromArgs;
+use size::Size;
+use typed_path::Utf8NativePathBuf;
+
+use crate::{
+    cmd::vfs,
+    util::{
+        path::native_path,
+        wad::{process_wad, verify_wad},
+    },
+    vfs::open_file,
+};
+
+#[derive(FromArgs, PartialEq, Debug)]
+/// Commands for processing Wii WAD files.
+#[argp(subcommand, name = "wad")]
+pub struct Args {
+    #[argp(subcommand)]
+    command: SubCommand,
+}
+
+#[derive(FromArgs, PartialEq, Debug)]
+#[argp(subcommand)]
+enum SubCommand {
+    Extract(ExtractArgs),
+    Info(InfoArgs),
+    Verify(VerifyArgs),
+}
+
+#[derive(FromArgs, PartialEq, Eq, Debug)]
+/// Extracts WAD file contents.
+#[argp(subcommand, name = "extract")]
+pub struct ExtractArgs {
+    #[argp(positional, from_str_fn(native_path))]
+    /// WAD file
+    file: Utf8NativePathBuf,
+    #[argp(option, short = 'o', from_str_fn(native_path))]
+    /// output directory
+    output: Option<Utf8NativePathBuf>,
+    #[argp(switch)]
+    /// Do not decompress files when copying.
+    no_decompress: bool,
+    #[argp(switch, short = 'q')]
+    /// Quiet output. Don't print anything except errors.
+    quiet: bool,
+}
+
+#[derive(FromArgs, PartialEq, Eq, Debug)]
+/// Views WAD file information.
+#[argp(subcommand, name = "info")]
+pub struct InfoArgs {
+    #[argp(positional, from_str_fn(native_path))]
+    /// WAD file
+    file: Utf8NativePathBuf,
+}
+
+#[derive(FromArgs, PartialEq, Eq, Debug)]
+/// Verifies WAD file integrity.
+#[argp(subcommand, name = "verify")]
+pub struct VerifyArgs {
+    #[argp(positional, from_str_fn(native_path))]
+    /// WAD file
+    file: Utf8NativePathBuf,
+}
+
+pub fn run(args: Args) -> Result<()> {
+    match args.command {
+        SubCommand::Info(c_args) => info(c_args),
+        SubCommand::Verify(c_args) => verify(c_args),
+        SubCommand::Extract(c_args) => extract(c_args),
+    }
+}
+
+fn info(args: InfoArgs) -> Result<()> {
+    let mut file = open_file(&args.file, true)?;
+    let wad = process_wad(file.as_mut())?;
+    println!("Title ID: {}", hex::encode(wad.ticket().title_id));
+    println!("Title key: {}", hex::encode(wad.title_key));
+    println!("Fake signed: {}", wad.fake_signed);
+    for content in wad.contents() {
+        println!(
+            "Content {:08x}: Offset {:#X}, size {}",
+            content.content_index.get(),
+            wad.content_offset(content.content_index.get()),
+            Size::from_bytes(content.size.get())
+        );
+    }
+    Ok(())
+}
+
+fn verify(args: VerifyArgs) -> Result<()> {
+    let mut file = open_file(&args.file, true)?;
+    let wad = process_wad(file.as_mut())?;
+    verify_wad(&wad, file.as_mut())?;
+    println!("Verification successful");
+    Ok(())
+}
+
+fn extract(args: ExtractArgs) -> Result<()> {
+    let path = Utf8NativePathBuf::from(format!("{}:", args.file));
+    let output = args.output.unwrap_or_else(|| Utf8NativePathBuf::from("."));
+    vfs::cp(vfs::CpArgs {
+        paths: vec![path, output],
+        no_decompress: args.no_decompress,
+        quiet: args.quiet,
+    })
+}