feat: add confirmation prompt when overwriting

pull/12902/head
Nikita Revenco 2025-02-18 12:15:35 +00:00
parent e177c48208
commit 382803c803
1 changed files with 108 additions and 36 deletions

View File

@ -286,6 +286,39 @@ pub fn file_picker(root: PathBuf, config: &helix_view::editor::Config) -> FilePi
type FileExplorer = Picker<(PathBuf, bool), (PathBuf, Style)>; type FileExplorer = Picker<(PathBuf, bool), (PathBuf, Style)>;
fn create_confirmation_prompt(
input: String,
cx: &mut Context,
to_create_str: String,
to_create: PathBuf,
callback: fn(&str, &Path) -> Option<Result<String, String>>,
) {
let callback = Box::pin(async move {
let call: Callback = Callback::EditorCompositor(Box::new(move |_editor, compositor| {
let prompt = Prompt::new(
input.into(),
None,
crate::ui::completers::none,
move |cx, input: &str, event: PromptEvent| {
if event != PromptEvent::Validate || input != "y" {
return;
};
match callback(&to_create_str, &to_create) {
Some(Ok(msg)) => cx.editor.set_status(msg),
Some(Err(msg)) => cx.editor.set_error(msg),
None => (),
};
},
);
compositor.push(Box::new(prompt));
}));
Ok(call)
});
cx.jobs.callback(callback);
}
fn create_file_operation_prompt( fn create_file_operation_prompt(
prompt: &'static str, prompt: &'static str,
cx: &mut Context, cx: &mut Context,
@ -381,42 +414,52 @@ pub fn file_explorer(root: PathBuf, editor: &Editor) -> Result<FileExplorer, std
|cx, (path, _is_dir): &(PathBuf, bool)|, |cx, (path, _is_dir): &(PathBuf, bool)|,
// create // create
alt!('n') => { alt!('n') => {
create_file_operation_prompt("create:", cx, path, |_cx, _path, to_create_str| { create_file_operation_prompt("create:", cx, path, |cx, _path, to_create_str| {
let to_create = helix_stdx::path::expand_tilde(PathBuf::from(to_create_str)); let to_create = helix_stdx::path::expand_tilde(PathBuf::from(to_create_str));
if to_create.exists() { let create = |to_create_str: &str, to_create: &Path| {
// TODO: confirmation prompt if to_create_str.ends_with(std::path::MAIN_SEPARATOR) {
return Some(Err(format!("Path {} already exists", to_create.display()))) if let Err(err) = fs::create_dir_all(to_create).map_err(
|err| format!("Unable to create directory {}: {err}", to_create.display())
) {
return Some(Err(err));
}
Some(Ok(format!("Created directory: {}", to_create.display())))
} else {
if let Err(err) = fs::File::create(to_create).map_err(
|err| format!("Unable to create file {}: {err}", to_create.display())
) {
return Some(Err(err));
};
Some(Ok(format!("Created file: {}", to_create.display())))
}
}; };
if to_create_str.ends_with(std::path::MAIN_SEPARATOR) { if to_create.exists() {
if let Err(err) = fs::create_dir_all(&to_create).map_err( create_confirmation_prompt(
|err| format!("Unable to create directory {}: {err}", to_create.display()) format!(
) { "Path {} already exists. Overwrite? (y/n):", to_create.display()
return Some(Err(err)); ),
} cx,
to_create_str.to_string(),
to_create.to_path_buf(),
create
);
return None;
};
Some(Ok(format!("Created directory: {}", to_create.display()))) create(to_create_str, &to_create)
} else {
if let Err(err) = fs::File::create(&to_create).map_err(
|err| format!("Unable to create file {}: {err}", to_create.display())
) {
return Some(Err(err));
};
Some(Ok(format!("Created file: {}", to_create.display())))
}
}) })
}, },
// move // move
alt!('m') => { alt!('m') => {
create_file_operation_prompt("move:", cx, path, |_, move_from, move_to_str| { create_file_operation_prompt("move:", cx, path, |cx, move_from, move_to_str| {
let move_to = helix_stdx::path::expand_tilde(PathBuf::from(move_to_str)); let move_to = helix_stdx::path::expand_tilde(PathBuf::from(move_to_str));
if move_to.exists() { let move_op = |move_to_str: &str, move_from: &Path| {
// TODO: confirmation prompt let move_to = helix_stdx::path::expand_tilde(PathBuf::from(move_to_str));
Some(Err(format!("Path {} already exists", move_to.display())))
} else {
if let Err(err) = fs::rename(move_from, &move_to).map_err(|err| if let Err(err) = fs::rename(move_from, &move_to).map_err(|err|
format!( format!(
"Unable to move {} {} -> {}: {err}", "Unable to move {} {} -> {}: {err}",
@ -432,7 +475,22 @@ pub fn file_explorer(root: PathBuf, editor: &Editor) -> Result<FileExplorer, std
return Some(Err(err)) return Some(Err(err))
}; };
None None
} };
if move_to.exists() {
create_confirmation_prompt(
format!(
"Path {} already exists. Overwrite? (y/n):", move_to.display()
),
cx,
move_to_str.to_string(),
move_from.to_path_buf(),
move_op
);
return None;
};
move_op(move_to_str, move_from)
}) })
}, },
// delete // delete
@ -472,17 +530,11 @@ pub fn file_explorer(root: PathBuf, editor: &Editor) -> Result<FileExplorer, std
}, },
// copy file / directory // copy file / directory
alt!('c') => { alt!('c') => {
create_file_operation_prompt("copy-to:", cx, path, |_, copy_from, copy_to_str| { create_file_operation_prompt("copy-to:", cx, path, |cx, copy_from, copy_to_str| {
let copy_to = helix_stdx::path::expand_tilde(PathBuf::from(copy_to_str)); let copy_to = helix_stdx::path::expand_tilde(PathBuf::from(copy_to_str));
if copy_from.is_dir() || copy_to_str.ends_with('/') {
// TODO: support copying directories (recursively)?. This isn't built-in to the standard library let copy_op = |copy_to_str: &str, copy_from: &Path| {
Some(Err(format!( let copy_to = helix_stdx::path::expand_tilde(PathBuf::from(copy_to_str));
"Copying directories is not supported: {} is a directory", copy_from.display()
)))
} else if copy_to.exists() {
// TODO: confirmation prompt
Some(Err(format!("Path {} exists", copy_to.display())))
} else {
if let Err(err) = std::fs::copy(copy_from, &copy_to).map_err( if let Err(err) = std::fs::copy(copy_from, &copy_to).map_err(
|err| format!("Unable to copy from file {} to {}: {err}", |err| format!("Unable to copy from file {} to {}: {err}",
copy_from.display(), copy_to.display() copy_from.display(), copy_to.display()
@ -493,6 +545,26 @@ pub fn file_explorer(root: PathBuf, editor: &Editor) -> Result<FileExplorer, std
Some(Ok(format!( Some(Ok(format!(
"Copied contents of file {} to {}", copy_from.display(), copy_to.display() "Copied contents of file {} to {}", copy_from.display(), copy_to.display()
))) )))
};
if copy_from.is_dir() || copy_to_str.ends_with('/') {
// TODO: support copying directories (recursively)?. This isn't built-in to the standard library
Some(Err(format!(
"Copying directories is not supported: {} is a directory", copy_from.display()
)))
} else if copy_to.exists() {
create_confirmation_prompt(
format!(
"Path {} already exists. Overwrite? (y/n):", copy_to.display()
),
cx,
copy_to_str.to_string(),
copy_from.to_path_buf(),
copy_op
);
None
} else {
copy_op(copy_to_str, copy_from)
} }
}) })
}, },