repak/examples/subcommands/mod.rs
2023-01-20 13:07:28 -06:00

28 lines
811 B
Rust

mod list;
mod unpack;
mod version;
pub use {list::list, unpack::unpack, version::version};
fn load_pak(
path: String,
key: Option<String>,
) -> Result<unpak::PakReader<std::io::BufReader<std::fs::File>>, unpak::Error> {
use aes::cipher::KeyInit;
use base64::{engine::general_purpose, Engine as _};
let key = key
.map(|k| {
general_purpose::STANDARD
.decode(k)
.as_ref()
.map_err(|_| unpak::Error::Base64)
.and_then(|bytes| {
aes::Aes256Dec::new_from_slice(bytes).map_err(|_| unpak::Error::Aes)
})
})
.transpose()?;
unpak::PakReader::new_any(
std::io::BufReader::new(std::fs::OpenOptions::new().read(true).open(&path)?),
key,
)
}