GeezerVison is showing github.com — RustPython/src/shell.rs at main · RustPython/RustPython

Expand file tree

shell.rs

Copy path

More file actions

Latest commit

History

41 lines (39 loc) · 1.33 KB

shell.rs

Copy path

File metadata and controls

41 lines (39 loc) · 1.33 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

mod helper;

use rustpython_vm::{

PyResult, VirtualMachine,

readline::{Readline, ReadlineResult},

scope::Scope,

vm::InteractiveLine,

};

/// Enter a repl loop

///

/// rustyline takes the role of `PyOS_Readline` with the readline module,

/// adding line editing, completion and history.

pub fn run_shell(vm: &VirtualMachine, scope: Scope) -> PyResult<()> {

let mut repl = Readline::new(helper::ShellHelper::new(vm, scope.globals));

let history = dirs::config_dir().map_or_else(

|| ".repl_history.txt".into(),

|mut path| {

path.push("rustpython");

path.push("repl_history.txt");

path

);

let _ = repl.load_history(&history);

let result = vm.run_interactive_loop(|prompt| match repl.readline(prompt) {

ReadlineResult::Line(line) => {

let _ = repl.add_history_entry(line.trim_end());

Ok(Some(InteractiveLine::Text(line)))

}

ReadlineResult::Eof => Ok(None),

ReadlineResult::Interrupt => {

Err(vm.new_exception_empty(vm.ctx.exceptions.keyboard_interrupt.to_owned()))

}

ReadlineResult::Io(_) => Ok(None),

#[cfg(unix)]

ReadlineResult::OsError(_) => Ok(None),

ReadlineResult::Other(_) => Ok(None),

});

let _ = repl.save_history(&history);

result

}