feat: add flag to check when use specify --dir

This commit is contained in:
Talison Fabio 2025-03-31 08:56:25 -03:00 committed by João Marcos
parent 82c551ddef
commit 4f9a786e57
2 changed files with 51 additions and 0 deletions

View File

@ -31,6 +31,7 @@ pub struct DecompressOptions<'a> {
pub formats: Vec<Extension>,
pub output_dir: &'a Path,
pub output_file_path: PathBuf,
pub is_output_dir_explicit: bool,
pub question_policy: QuestionPolicy,
pub quiet: bool,
pub password: Option<&'a [u8]>,
@ -73,6 +74,7 @@ pub fn decompress_file(options: DecompressOptions) -> crate::Result<()> {
options.output_dir,
&options.output_file_path,
options.question_policy,
options.is_output_dir_explicit,
)? {
files
} else {
@ -150,6 +152,7 @@ pub fn decompress_file(options: DecompressOptions) -> crate::Result<()> {
options.output_dir,
&options.output_file_path,
options.question_policy,
options.is_output_dir_explicit,
)? {
files
} else {
@ -183,6 +186,7 @@ pub fn decompress_file(options: DecompressOptions) -> crate::Result<()> {
options.output_dir,
&options.output_file_path,
options.question_policy,
options.is_output_dir_explicit,
)? {
files
} else {
@ -214,6 +218,7 @@ pub fn decompress_file(options: DecompressOptions) -> crate::Result<()> {
options.output_dir,
&options.output_file_path,
options.question_policy,
options.is_output_dir_explicit,
)? {
files
} else {
@ -255,6 +260,7 @@ pub fn decompress_file(options: DecompressOptions) -> crate::Result<()> {
options.output_dir,
&options.output_file_path,
options.question_policy,
options.is_output_dir_explicit,
)? {
files
} else {
@ -284,6 +290,48 @@ pub fn decompress_file(options: DecompressOptions) -> crate::Result<()> {
Ok(())
}
fn execute_decompression(
unpack_fn: impl FnOnce(&Path) -> crate::Result<usize>,
output_dir: &Path,
output_file_path: &Path,
question_policy: QuestionPolicy,
is_output_dir_explicit: bool,
) -> crate::Result<ControlFlow<(), usize>> {
if is_output_dir_explicit {
return unpack(unpack_fn, output_dir, question_policy);
} else {
return smart_unpack(
unpack_fn,
output_dir,
output_file_path,
question_policy,
);
}
}
fn unpack(
unpack_fn: impl FnOnce(&Path) -> crate::Result<usize>,
output_dir: &Path,
question_policy: QuestionPolicy,
) -> crate::Result<ControlFlow<(), usize>> {
let has_files = output_dir.exists() && output_dir.read_dir().map(|dir| dir.count() > 0).unwrap_or(false);
let output_dir_cleaned = if has_files {
let output_file_path = utils::resolve_path_conflict(&output_dir, question_policy)?.unwrap();
output_file_path
} else {
output_dir.to_owned()
};
if !output_dir_cleaned.exists() {
fs::create_dir(&output_dir_cleaned)?;
}
let files = unpack_fn(&output_dir_cleaned)?;
Ok(ControlFlow::Continue(files))
}
/// Unpacks an archive with some heuristics
/// - If the archive contains only one file, it will be extracted to the `output_dir`
/// - If the archive contains multiple files, it will be extracted to a subdirectory of the
@ -295,6 +343,7 @@ fn smart_unpack(
output_dir: &Path,
output_file_path: &Path,
question_policy: QuestionPolicy,
is_output_dir_explicit: bool,
) -> crate::Result<ControlFlow<(), usize>> {
assert!(output_dir.exists());
let temp_dir = tempfile::Builder::new().prefix(".tmp-ouch-").tempdir_in(output_dir)?;

View File

@ -176,6 +176,7 @@ pub fn run(
// The directory that will contain the output files
// We default to the current directory if the user didn't specify an output directory with --dir
let is_output_dir_explicit = output_dir.is_some();
let output_dir = if let Some(dir) = output_dir {
utils::create_dir_if_non_existent(&dir)?;
dir
@ -199,6 +200,7 @@ pub fn run(
formats,
output_dir: &output_dir,
output_file_path,
is_output_dir_explicit,
question_policy,
quiet: args.quiet,
password: args.password.as_deref().map(|str| {