summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorBlaž Hrastnik2021-08-31 09:24:24 +0000
committerBlaž Hrastnik2021-08-31 09:24:24 +0000
commit9b96bb5ac802189f567b101f7c8da4b370a2fdd1 (patch)
tree5a86e2c49554009baa39e2969e3ceb6ac6ebc635
parent4a76ea8f8834a880d63246c8d43587693ac7cc81 (diff)
Refactor a bit further
-rw-r--r--helix-term/src/commands.rs127
1 files changed, 62 insertions, 65 deletions
diff --git a/helix-term/src/commands.rs b/helix-term/src/commands.rs
index 42495b7e..e75d46dc 100644
--- a/helix-term/src/commands.rs
+++ b/helix-term/src/commands.rs
@@ -4342,80 +4342,77 @@ fn shell(cx: &mut Context, prompt: &str, behavior: ShellBehavior) {
Some('|'),
|_input: &str| Vec::new(),
move |cx: &mut compositor::Context, input: &str, event: PromptEvent| {
+ if event != PromptEvent::Validate {
+ return;
+ }
let shell = &cx.editor.config.shell;
- if event == PromptEvent::Validate {
- let (view, doc) = current!(cx.editor);
- let selection = doc.selection(view.id);
+ let (view, doc) = current!(cx.editor);
+ let selection = doc.selection(view.id);
- let mut changes = Vec::with_capacity(selection.len());
-
- for range in selection.ranges() {
- let mut process;
- match Command::new(&shell[0])
- .args(&shell[1..])
- .arg(input)
- .stdin(Stdio::piped())
- .stdout(Stdio::piped())
- .stderr(Stdio::piped())
- .spawn()
- {
- Ok(p) => process = p,
- Err(e) => {
- log::error!("Failed to start shell: {}", e);
- cx.editor.set_error("Failed to start shell".to_owned());
- return;
- }
- }
- if pipe {
- let stdin = process.stdin.as_mut().unwrap();
- let fragment = range.fragment(doc.text().slice(..));
- stdin.write_all(fragment.as_bytes()).unwrap();
+ let mut changes = Vec::with_capacity(selection.len());
+
+ for range in selection.ranges() {
+ let mut process = match Command::new(&shell[0])
+ .args(&shell[1..])
+ .arg(input)
+ .stdin(Stdio::piped())
+ .stdout(Stdio::piped())
+ .stderr(Stdio::piped())
+ .spawn()
+ {
+ Ok(process) => process,
+ Err(e) => {
+ log::error!("Failed to start shell: {}", e);
+ cx.editor.set_error("Failed to start shell".to_owned());
+ return;
}
+ };
+ if pipe {
+ let stdin = process.stdin.as_mut().unwrap();
+ let fragment = range.fragment(doc.text().slice(..));
+ stdin.write_all(fragment.as_bytes()).unwrap();
+ }
+ let output = process.wait_with_output().unwrap();
- let output = process.wait_with_output().unwrap();
- if behavior != ShellBehavior::Filter {
- if !output.status.success() {
- let stderr = output.stderr;
- if !stderr.is_empty() {
- log::error!("Shell error: {}", String::from_utf8_lossy(&stderr));
- }
- cx.editor.set_error("Command failed".to_owned());
- return;
+ if behavior != ShellBehavior::Filter {
+ if !output.status.success() {
+ if !output.stderr.is_empty() {
+ log::error!("Shell error: {}", String::from_utf8_lossy(&output.stderr));
}
- let stdout = output.stdout;
- let tendril;
- match Tendril::try_from_byte_slice(&stdout) {
- Ok(t) => tendril = t,
- Err(_) => {
- cx.editor
- .set_error("Process did not output valid UTF-8".to_owned());
- return;
- }
- }
- let (from, to) = match behavior {
- ShellBehavior::Replace => (range.from(), range.to()),
- ShellBehavior::Insert => (range.from(), range.from()),
- ShellBehavior::Append => (range.to(), range.to()),
- _ => (range.from(), range.from()),
- };
- changes.push((from, to, Some(tendril)));
- } else {
- // if the process exits successfully, keep the selection, otherwise delete it.
- let keep = output.status.success();
- changes.push((
- range.from(),
- if keep { range.from() } else { range.to() },
- None,
- ));
+ cx.editor.set_error("Command failed".to_owned());
+ return;
}
+ let tendril = match Tendril::try_from_byte_slice(&output.stdout) {
+ Ok(tendril) => tendril,
+ Err(_) => {
+ cx.editor
+ .set_error("Process did not output valid UTF-8".to_owned());
+ return;
+ }
+ };
+ let (from, to) = match behavior {
+ ShellBehavior::Replace => (range.from(), range.to()),
+ ShellBehavior::Insert => (range.from(), range.from()),
+ ShellBehavior::Append => (range.to(), range.to()),
+ _ => (range.from(), range.from()),
+ };
+ changes.push((from, to, Some(tendril)));
+ } else {
+ // if the process exits successfully, keep the selection, otherwise delete it.
+ let keep = output.status.success();
+ changes.push((
+ range.from(),
+ if keep { range.from() } else { range.to() },
+ None,
+ ));
}
+ }
- let transaction = Transaction::change(doc.text(), changes.into_iter());
+ let transaction = Transaction::change(doc.text(), changes.into_iter());
- if behavior != ShellBehavior::Ignore {
- doc.apply(&transaction, view.id);
- doc.append_changes_to_history(view.id);
- }
+ if behavior != ShellBehavior::Ignore {
+ doc.apply(&transaction, view.id);
+ doc.append_changes_to_history(view.id);
}
},
);