let mut words = vec![String::new()];

let mut startpos = 0;

for (i, c) in line.chars().rev().enumerate() {

match c {

'.' => {

// check for a double dot

if i != 0 && words.last().map_or(false, |s| s.is_empty()) {

return None;

}

reverse_string(words.last_mut().unwrap());

if words.len() == 1 {

startpos = line.len() - i;

}

words.push(String::new());

}

c if c.is_alphanumeric() || c == '_' => words.last_mut().unwrap().push(c),

_ => {

if words.len() == 1 {

if words.last().unwrap().is_empty() {

return None;

}

startpos = line.len() - i;

}

break;

}

}

}

if words == [String::new()] {

return None;

}

reverse_string(words.last_mut().unwrap());

words.reverse();

Some((startpos, words))

pub fn new(vm: &'vm VirtualMachine, globals: PyDictRef) -> Self {

ShellHelper { vm, globals }

}

fn get_available_completions<'w>(

&self,

words: &'w [String],

) -> Option<(&'w str, impl Iterator<Item = PyResult<PyStrRef>> + 'vm)> {

// the very first word and then all the ones after the dot

let (first, rest) = words.split_first().unwrap();

let str_iter_method = |obj, name| {

let iter = self.vm.call_special_method(obj, name, ())?;

ArgIterable::<PyStrRef>::try_from_object(self.vm, iter)?.iter(self.vm)

};

let (word_start, iter1, iter2) = if let Some((last, parents)) = rest.split_last() {

// we need to get an attribute based off of the dir() of an object

// last: the last word, could be empty if it ends with a dot

// parents: the words before the dot

let mut current = self.globals.get_item_opt(first.as_str(), self.vm).ok()??;

for attr in parents {

current = current.get_attr(attr.as_str(), self.vm).ok()?;

}

let current_iter = str_iter_method(current, identifier!(self.vm, __dir__)).ok()?;

(last, current_iter, None)

} else {

// we need to get a variable based off of globals/builtins

let globals =

str_iter_method(self.globals.clone().into(), identifier!(self.vm, keys)).ok()?;

let builtins = str_iter_method(

self.vm.builtins.clone().into(),

identifier!(self.vm, __dir__),

)

.ok()?;

(first, globals, Some(builtins))

};

Some((word_start, iter1.chain(iter2.into_iter().flatten())))

}

fn complete_opt(&self, line: &str) -> Option<(usize, Vec<String>)> {

let (startpos, words) = split_idents_on_dot(line)?;

let (word_start, iter) = self.get_available_completions(&words)?;

let all_completions = iter

.filter(|res| {

res.as_ref()

.ok()

.map_or(true, |s| s.as_str().starts_with(word_start))

})

.collect::<Result<Vec<_>, _>>()

.ok()?;

let mut completions = if word_start.starts_with('_') {

// if they're already looking for something starting with a '_', just give

// them all the completions

all_completions

} else {

// only the completions that don't start with a '_'

let no_underscore = all_completions

.iter()

.cloned()

.filter(|s| !s.as_str().starts_with('_'))

.collect::<Vec<_>>();

// if there are only completions that start with a '_', give them all of the

// completions, otherwise only the ones that don't start with '_'

if no_underscore.is_empty() {

all_completions

} else {

no_underscore

}

};

// sort the completions alphabetically

completions.sort_by(|a, b| std::cmp::Ord::cmp(a.as_str(), b.as_str()));

Some((

startpos,

completions

.into_iter()

.map(|s| s.as_str().to_owned())

.collect(),

))

}