You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.8 KiB
Rust
62 lines
1.8 KiB
Rust
use clap::{Parser, Subcommand};
|
|
use extract_archive::format::{parse_format, FileFormat};
|
|
use std::fs;
|
|
use std::path::PathBuf;
|
|
use std::str::FromStr;
|
|
use tracing_subscriber::fmt::format::FmtSpan;
|
|
use tracing_subscriber::EnvFilter;
|
|
|
|
#[derive(Debug, Clone, Parser)]
|
|
#[clap(name="extract-archive", version=env!("CARGO_PKG_VERSION"), about=env!("CARGO_PKG_DESCRIPTION"))]
|
|
struct Args {
|
|
/// The operation to perform
|
|
#[clap(subcommand)]
|
|
pub operation: Operation,
|
|
}
|
|
|
|
#[derive(Debug, Clone, Subcommand)]
|
|
pub enum Operation {
|
|
/// Extracts a given file
|
|
Extract {
|
|
/// The file to extract
|
|
file: PathBuf,
|
|
|
|
/// The output folder for the given file
|
|
output: PathBuf,
|
|
},
|
|
}
|
|
|
|
fn main() {
|
|
color_eyre::install().unwrap();
|
|
init_logger();
|
|
let args: Args = Args::parse();
|
|
match args.operation {
|
|
Operation::Extract { output, file } => {
|
|
if !output.exists() {
|
|
fs::create_dir_all(&output).expect("Failed to create output directory");
|
|
}
|
|
if output.is_file() {
|
|
panic!("The given output path is a file");
|
|
}
|
|
let format = parse_format(&file).expect("Failed to parse file format");
|
|
format
|
|
.extract(&file, &output)
|
|
.expect("Failed to extract file");
|
|
}
|
|
}
|
|
}
|
|
|
|
fn init_logger() {
|
|
const DEFAULT_ENV_FILTER: &str = "info";
|
|
let filter_string =
|
|
std::env::var("RUST_LOG").unwrap_or_else(|_| DEFAULT_ENV_FILTER.to_string());
|
|
let env_filter =
|
|
EnvFilter::from_str(&*filter_string).expect("failed to parse env filter string");
|
|
tracing_subscriber::fmt::SubscriberBuilder::default()
|
|
.with_env_filter(env_filter)
|
|
.with_writer(std::io::stdout)
|
|
.with_span_events(FmtSpan::NEW | FmtSpan::CLOSE)
|
|
.compact()
|
|
.init();
|
|
}
|