Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
  • Loading branch information
ivktac committed Nov 2, 2022
1 parent 1bed2f3 commit 876e05f
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 41 deletions.
32 changes: 0 additions & 32 deletions helix-core/src/shellwords.rs
Original file line number Diff line number Diff line change
@@ -1,22 +1,5 @@
use std::borrow::Cow;

/// Auto escape for shellwords usage.
pub fn escape(input: &str) -> Cow<'_, str> {
if !input.chars().any(|x| x.is_ascii_whitespace()) {
Cow::Borrowed(input)
} else if cfg!(unix) {
Cow::Owned(input.chars().fold(String::new(), |mut buf, c| {
if c.is_ascii_whitespace() {
buf.push('\\');
}
buf.push(c);
buf
}))
} else {
Cow::Owned(format!("\"{}\"", input))
}
}

/// Get the vec of escaped / quoted / doublequoted filenames from the input str
pub fn shellwords(input: &str) -> Vec<Cow<'_, str>> {
enum State {
Expand Down Expand Up @@ -243,19 +226,4 @@ mod test {
];
assert_eq!(expected, result);
}

#[test]
#[cfg(unix)]
fn test_escaping_unix() {
assert_eq!(escape("foobar"), Cow::Borrowed("foobar"));
assert_eq!(escape("foo bar"), Cow::Borrowed("foo\\ bar"));
assert_eq!(escape("foo\tbar"), Cow::Borrowed("foo\\\tbar"));
}

#[test]
#[cfg(windows)]
fn test_escaping_windows() {
assert_eq!(escape("foobar"), Cow::Borrowed("foobar"));
assert_eq!(escape("foo bar"), Cow::Borrowed("\"foo bar\""));
}
}
9 changes: 5 additions & 4 deletions helix-term/src/commands/typed.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2183,10 +2183,12 @@ pub(super) fn command_mode(cx: &mut Context) {
static FUZZY_MATCHER: Lazy<fuzzy_matcher::skim::SkimMatcherV2> =
Lazy::new(fuzzy_matcher::skim::SkimMatcherV2::default);

// we use .this over split_whitespace() because we care about empty segments
let parts = input.split(' ').collect::<Vec<&str>>();

// simple heuristic: if there's no just one part, complete command name.
// if there's a space, per command completion kicks in.
// we use .this over split_whitespace() because we care about empty segments
if input.split(' ').count() <= 1 {
if parts.len() <= 1 {
let mut matches: Vec<_> = typed::TYPABLE_COMMAND_LIST
.iter()
.filter_map(|command| {
Expand All @@ -2202,13 +2204,12 @@ pub(super) fn command_mode(cx: &mut Context) {
.map(|(name, _)| (0.., name.into()))
.collect()
} else {
let parts = shellwords::shellwords(input);
let part = parts.last().unwrap();

if let Some(typed::TypableCommand {
completer: Some(completer),
..
}) = typed::TYPABLE_COMMAND_MAP.get(&parts[0] as &str)
}) = typed::TYPABLE_COMMAND_MAP.get(parts[0])
{
completer(editor, part)
.into_iter()
Expand Down
6 changes: 1 addition & 5 deletions helix-term/src/ui/prompt.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
use crate::compositor::{Component, Compositor, Context, Event, EventResult};
use crate::{alt, ctrl, key, shift, ui};
use helix_core::shellwords;
use helix_view::input::KeyEvent;
use helix_view::keyboard::KeyCode;
use std::{borrow::Cow, ops::RangeFrom};
Expand Down Expand Up @@ -336,10 +335,7 @@ impl Prompt {

let (range, item) = &self.completion[index];

// since we are using shellwords to parse arguments, make sure
// that whitespace in files is properly escaped.
let item = shellwords::escape(item);
self.line.replace_range(range.clone(), &item);
self.line.replace_range(range.clone(), item);

self.move_end();
}
Expand Down

0 comments on commit 876e05f

Please sign in to comment.