Add dump-art command (#9)

Author: adamthedash

Cargo.toml

@@ -25,7 +25,8 @@ glob = "0.3.2"
 serde = {version = "1.0.217", features = ["derive"]}
 serde_json = "1.0.137"
 polars = {version = "0.45.1", features=["csv", "lazy", "list_eval", "dtype-full"]}
-rayon = "1.10.0"
+rayon = "1.10.0"
+image = { git = "https://github.com/RunDevelopment/image", branch = "new-dds-decoder" }
 
 [dev-dependencies]
 criterion = { version = "0.3", features = ["html_reports"] }

src/bin/poe_files.rs

@@ -5,7 +5,8 @@ use poe_tools::{
     bundle_fs::{from_cdn, from_steam},
     bundle_loader::cdn_base_url,
     commands::{
-        cat::cat_file, dump_tables::dump_tables, extract::extract_files, list::list_files, Patch,
+        cat::cat_file, dump_art::extract_art, dump_tables::dump_tables, extract::extract_files,
+        list::list_files, Patch,
     },
 };
 use std::path::PathBuf;
@@ -42,6 +43,13 @@ enum Command {
         /// Path to write out the parsed tables to - Only supports CSV for now
         output_folder: PathBuf,
     },
+    DumpArt {
+        /// Path to the folder to output the extracted files
+        output_folder: PathBuf,
+        /// Glob pattern to filter the list of files
+        #[clap(default_value = "*")]
+        glob: Pattern,
+    },
 }
 
 /// A simple CLI tool that extracts the virtual filenames from PoE data files.
@@ -143,6 +151,10 @@ fn main() -> Result<()> {
             output_folder,
         } => dump_tables(&datc64_root, &schema_path, &output_folder, &args.patch)
             .context("Dump Tables command failed")?,
+        Command::DumpArt {
+            output_folder,
+            glob,
+        } => extract_art(&mut fs, &glob, &output_folder).context("Dump Art command failed")?,
     }
 
     Ok(())

src/commands/dump_art.rs

@@ -0,0 +1,38 @@
+use std::{fs, path::Path};
+
+use glob::Pattern;
+
+use crate::bundle_fs::FS;
+use anyhow::{ensure, Context, Result};
+
+/// Extract files to disk matching a glob pattern
+pub fn extract_art(fs: &mut FS, pattern: &Pattern, output_folder: &Path) -> Result<()> {
+    ensure!(
+        pattern.as_str().ends_with(".dds"),
+        "Only .dds art export is supported."
+    );
+
+    fs.list()
+        .iter()
+        .filter(|filename| pattern.matches(filename))
+        .map(|filename| -> Result<_, anyhow::Error> {
+            // Dump it to disk
+            let contents = fs.read(filename).context("Failed to read file")?;
+
+            let img = image::load_from_memory(&contents).context("Failed to pares DDS image")?;
+
+            let out_filename = output_folder.join(filename).with_extension("png");
+            fs::create_dir_all(out_filename.parent().unwrap())
+                .context("Failed to create folder")?;
+
+            img.save(out_filename).context("Failed to write file")?;
+
+            Ok(filename)
+        })
+        .for_each(|result| match result {
+            Ok(filename) => eprintln!("Extracted file: {}", filename),
+            Err(e) => eprintln!("Failed to extract file: {:?}", e),
+        });
+
+    Ok(())
+}

src/commands/mod.rs

@@ -1,4 +1,5 @@
 pub mod cat;
+pub mod dump_art;
 pub mod dump_tables;
 pub mod extract;
 pub mod list;