Newer
Older
//! This implementation is based on [Antirez's Linenoise](https://github.com/antirez/linenoise)
//! ```
//! let mut rl = rustyline::Editor::new();
//! let readline = rl.readline(">> ");
//! match readline {
//! Ok(line) => println!("Line: {:?}",line),
//! Err(_) => println!("No input"),
//! }
#![feature(unicode)]
#![cfg_attr(feature="clippy", feature(plugin))]
#![cfg_attr(feature="clippy", plugin(clippy))]
extern crate nix;
extern crate unicode_width;
pub mod completion;
use std::path::Path;
use completion::Completer;
use consts::{KeyPress, char_to_key_press};
use line_buffer::{LineBuffer, WordAction, MAX_LINE};
/// The error type for I/O and Linux Syscalls (Errno)
pub type Result<T> = result::Result<T, error::ReadlineError>;
// Represent the state during line editing.
struct State<'out, 'prompt> {
out: &'out mut Write,
prompt: &'prompt str, // Prompt to display
prompt_size: Position, // Prompt Unicode width and height
cursor: Position, // Cursor position (relative to the start of the prompt for `row`)
cols: usize, // Number of columns in terminal
history_index: usize, // The history index we are currently editing.
snapshot: LineBuffer, // Current edited line before history browsing/completion
#[derive(Copy, Clone, Debug, Default)]
struct Position {
col: usize,
row: usize,
fn new(out: &'out mut Write,
prompt: &'prompt str,
capacity: usize,
cols: usize,
history_index: usize)
-> State<'out, 'prompt> {
let prompt_size = calculate_position(prompt, Default::default(), cols);
/// Rewrite the currently edited line accordingly to the buffer content,
/// cursor position, and number of columns of the terminal.
fn refresh_line(&mut self) -> Result<()> {
let prompt_size = self.prompt_size;
self.refresh(self.prompt, prompt_size)
}
fn refresh_prompt_and_line(&mut self, prompt: &str) -> Result<()> {
let prompt_size = calculate_position(prompt, Default::default(), self.cols);
self.refresh(prompt, prompt_size)
fn refresh(&mut self, prompt: &str, prompt_size: Position) -> Result<()> {
let end_pos = calculate_position(self.line.as_str(), prompt_size, self.cols);
let cursor = calculate_position(&self.line.as_str()[..self.line.pos()],
prompt_size,
self.cols);
let mut ab = String::new();
let cursor_row_movement = self.cursor.row - self.prompt_size.row;
// move the cursor up as required
if cursor_row_movement > 0 {
write!(ab, "\x1b[{}A", cursor_row_movement).unwrap();
// position at the start of the prompt, clear to end of screen
ab.push_str("\r\x1b[J");
// display the prompt
// we have to generate our own newline on line wrap
if end_pos.col == 0 && end_pos.row > 0 {
ab.push_str("\n");
// position the cursor
let cursor_row_movement = end_pos.row - cursor.row;
// move the cursor up as required
if cursor_row_movement > 0 {
write!(ab, "\x1b[{}A", cursor_row_movement).unwrap();
// position the cursor within the line
if cursor.col > 0 {
write_and_flush(self.out, ab.as_bytes())
}
impl<'out, 'prompt> fmt::Debug for State<'out, 'prompt> {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f.debug_struct("State")
.field("cols", &self.cols)
.field("history_index", &self.history_index)
/// Unsupported Terminals that don't support RAW mode
static UNSUPPORTED_TERM: [&'static str; 3] = ["dumb", "cons25", "emacs"];
/// Check to see if `fd` is a TTY
fn is_a_tty(fd: libc::c_int) -> bool {
unsafe { libc::isatty(fd) != 0 }
/// Check to see if the current `TERM` is unsupported
use std::ascii::AsciiExt;
let mut unsupported = false;
for iter in &UNSUPPORTED_TERM {
unsupported = (*iter).eq_ignore_ascii_case(&term)
fn from_errno(errno: Errno) -> error::ReadlineError {
error::ReadlineError::from(nix::Error::from_errno(errno))
}
fn enable_raw_mode() -> Result<termios::Termios> {
use nix::sys::termios::{BRKINT, ICRNL, INPCK, ISTRIP, IXON, OPOST, CS8, ECHO, ICANON, IEXTEN,
ISIG, VMIN, VTIME};
let original_term = try!(termios::tcgetattr(libc::STDIN_FILENO));
let mut raw = original_term;
raw.c_iflag = raw.c_iflag & !(BRKINT | ICRNL | INPCK | ISTRIP | IXON); // disable BREAK interrupt, CR to NL conversion on input, input parity check, strip high bit (bit 8), output flow control
raw.c_oflag = raw.c_oflag & !(OPOST); // disable all output processing
raw.c_cflag = raw.c_cflag | (CS8); // character-size mark (8 bits)
raw.c_lflag = raw.c_lflag & !(ECHO | ICANON | IEXTEN | ISIG); // disable echoing, canonical mode, extended input processing and signals
raw.c_cc[VMIN] = 1; // One character-at-a-time input
raw.c_cc[VTIME] = 0; // with blocking read
try!(termios::tcsetattr(libc::STDIN_FILENO, termios::TCSAFLUSH, &raw));
Ok(original_term)
}
fn disable_raw_mode(original_termios: termios::Termios) -> Result<()> {
try!(termios::tcsetattr(libc::STDIN_FILENO, termios::TCSAFLUSH, &original_termios));
#[cfg(any(target_os = "macos", target_os = "freebsd"))]
const TIOCGWINSZ: libc::c_ulong = 0x40087468;
#[cfg(any(target_os = "linux", target_os = "android"))]
const TIOCGWINSZ: libc::c_ulong = 0x5413;
/// Try to get the number of columns in the current terminal,
/// or assume 80 if it fails.
#[cfg(any(target_os = "linux",
target_os = "android",
target_os = "macos",
target_os = "freebsd"))]
fn get_columns() -> usize {
use std::mem::zeroed;
use libc::c_ushort;
unsafe {
#[repr(C)]
struct winsize {
ws_row: c_ushort,
ws_col: c_ushort,
ws_xpixel: c_ushort,
}
let mut size: winsize = zeroed();
match libc::ioctl(libc::STDOUT_FILENO, TIOCGWINSZ, &mut size) {
0 => size.ws_col as usize, // TODO getCursorPosition
_ => 80,
fn write_and_flush(w: &mut Write, buf: &[u8]) -> Result<()> {
try!(w.write_all(buf));
try!(w.flush());
/// Clear the screen. Used to handle ctrl+l
fn clear_screen(out: &mut Write) -> Result<()> {
}
/// Beep, used for completion when there is nothing to complete or when all
/// the choices were already shown.
write_and_flush(&mut io::stderr(), b"\x07") // TODO bell-style
/// Calculate the number of columns and rows used to display `s` on a `cols` width terminal
/// starting at `orig`.
/// Control characters are treated as having zero width.
/// Characters with 2 column width are correctly handled (not splitted).
#[cfg_attr(feature="clippy", allow(if_same_then_else))]
fn calculate_position(s: &str, orig: Position, cols: usize) -> Position {
let mut pos = orig;
let mut esc_seq = 0;
for c in s.chars() {
let cw = if esc_seq == 1 {
if c == '[' {
// CSI
esc_seq = 2;
// two-character sequence
esc_seq = 0;
}
None
} else if esc_seq == 2 {
if c == ';' || (c >= '0' && c <= '9') {
} else if c == 'm' {
// last
esc_seq = 0;
// not supported
esc_seq = 0;
}
None
} else if c == '\x1b' {
esc_seq = 1;
None
} else if c == '\n' {
pos.col = 0;
pos.row += 1;
None
} else {
unicode_width::UnicodeWidthChar::width(c)
};
if let Some(cw) = cw {
pos.col += cw;
if pos.col > cols {
pos.row += 1;
pos.col = cw;
if pos.col == cols {
pos.col = 0;
pos.row += 1;
}
pos
/// Insert the character `ch` at cursor current position.
fn edit_insert(s: &mut State, ch: char) -> Result<()> {
if let Some(push) = s.line.insert(ch) {
if push {
if s.cursor.col + unicode_width::UnicodeWidthChar::width(ch).unwrap_or(0) < s.cols {
// Avoid a full update of the line in the trivial case.
let bits = ch.encode_utf8();
let bits = bits.as_slice();
write_and_flush(s.out, bits)
}
} else {
Ok(())
}
}
if let Some(_) = s.line.yank(text) {
s.refresh_line()
// Delete previously yanked text and yank/paste `text` at current position.
fn edit_yank_pop(s: &mut State, yank_size: usize, text: &str) -> Result<()> {
fn edit_move_left(s: &mut State) -> Result<()> {
} else {
Ok(())
}
}
/// Move cursor on the right.
fn edit_move_right(s: &mut State) -> Result<()> {
s.refresh_line()
} else {
Ok(())
}
}
/// Move cursor to the start of the line.
fn edit_move_home(s: &mut State) -> Result<()> {
} else {
Ok(())
}
}
/// Move cursor to the end of the line.
fn edit_move_end(s: &mut State) -> Result<()> {
if s.line.move_end() {
s.refresh_line()
} else {
Ok(())
}
}
/// Delete the character at the right of the cursor without altering the cursor
/// position. Basically this is what happens with the "Delete" keyboard key.
} else {
Ok(())
}
}
/// Backspace implementation.
fn edit_backspace(s: &mut State) -> Result<()> {
} else {
Ok(())
}
}
/// Kill the text from point to the end of the line.
}
}
/// Kill backward from point to the beginning of the line.
}
}
/// Exchange the char before cursor with the character at cursor.
fn edit_transpose_chars(s: &mut State) -> Result<()> {
} else {
Ok(())
}
}
s.refresh_line()
} else {
Ok(())
}
}
/// Delete the previous word, maintaining the cursor at the start of the
/// current word.
fn edit_delete_prev_word<F>(s: &mut State, test: F) -> Result<Option<String>>
where F: Fn(char) -> bool
{
if let Some(text) = s.line.delete_prev_word(test) {
s.refresh_line()
} else {
Ok(())
}
}
/// Kill from the cursor to the end of the current word, or, if between words, to the end of the next word.
fn edit_delete_word(s: &mut State) -> Result<Option<String>> {
try!(s.refresh_line());
Ok(Some(text))
} else {
Ok(None)
}
}
/// Substitute the currently edited line with the next or previous history
/// entry.
fn edit_history_next(s: &mut State, history: &History, prev: bool) -> Result<()> {
if history.is_empty() {
return Ok(());
}
if s.history_index == history.len() {
} else if s.history_index == 0 && prev {
return Ok(());
}
if prev {
s.history_index -= 1;
if s.history_index < history.len() {
let buf = history.get(s.history_index).unwrap();
} else {
// Restore current edited line
s.snapshot();
fn complete_line<R: io::Read>(chars: &mut io::Chars<R>,
s: &mut State,
completer: &Completer)
-> Result<Option<char>> {
let (start, candidates) = try!(completer.complete(s.line.as_str(), s.line.pos()));
if candidates.is_empty() {
try!(beep());
Ok(None)
} else {
// Save the current edited line before to overwrite it
s.backup();
let mut ch;
let mut i = 0;
loop {
// Show completion or original buffer
if i < candidates.len() {
completer.update(&mut s.line, start, &candidates[i]);
// Restore current edited line
s.snapshot();
}
ch = try!(chars.next().unwrap());
let key = char_to_key_press(ch);
match key {
KeyPress::TAB => {
if i == candidates.len() {
try!(beep());
}
}
}
}
Ok(Some(ch))
}
}
fn reverse_incremental_search<R: io::Read>(chars: &mut io::Chars<R>,
s: &mut State,
history: &History)
-> Result<Option<KeyPress>> {
// Save the current edited line (and cursor position) before to overwrite it
let mut search_buf = String::new();
let mut history_idx = history.len() - 1;
let mut success = true;
let mut ch;
let mut key;
// Display the reverse-i-search prompt and process chars
loop {
let prompt = if success {
format!("(reverse-i-search)`{}': ", search_buf)
} else {
format!("(failed reverse-i-search)`{}': ", search_buf)
};
try!(s.refresh_prompt_and_line(&prompt));
ch = try!(chars.next().unwrap());
if !ch.is_control() {
search_buf.push(ch);
} else {
key = char_to_key_press(ch);
if key == KeyPress::ESC {
key = try!(escape_sequence(chars));
}
match key {
KeyPress::CTRL_H | KeyPress::BACKSPACE => {
search_buf.pop();
KeyPress::CTRL_R => {
if history_idx > 0 {
history_idx -= 1;
} else {
success = false;
continue;
}
// Restore current edited line (before search)
s.snapshot();
}
}
success = match history.search(&search_buf, history_idx, true) {
Some(idx) => {
history_idx = idx;
let entry = history.get(idx).unwrap();
};
}
Ok(Some(key))
}
fn escape_sequence<R: io::Read>(chars: &mut io::Chars<R>) -> Result<KeyPress> {
// Read the next two bytes representing the escape sequence.
let seq1 = try!(chars.next().unwrap());
let seq2 = try!(chars.next().unwrap());
let seq3 = try!(chars.next().unwrap());
if seq3 == '~' {
match seq2 {
'3' => Ok(KeyPress::ESC_SEQ_DELETE),
// TODO '1' // Home
// TODO '4' // End
_ => Ok(KeyPress::UNKNOWN_ESC_SEQ),
}
} else {
Ok(KeyPress::UNKNOWN_ESC_SEQ)
}
} else {
match seq2 {
'A' => Ok(KeyPress::CTRL_P), // Up
'B' => Ok(KeyPress::CTRL_N), // Down
'C' => Ok(KeyPress::CTRL_F), // Right
'D' => Ok(KeyPress::CTRL_B), // Left
'F' => Ok(KeyPress::CTRL_E), // End
'H' => Ok(KeyPress::CTRL_A), // Home
let seq2 = try!(chars.next().unwrap());
match seq2 {
'F' => Ok(KeyPress::CTRL_E),
'H' => Ok(KeyPress::CTRL_A),
}
} else {
// TODO ESC-N (n): search history forward not interactively
// TODO ESC-P (p): search history backward not interactively
// TODO ESC-R (r): Undo all changes made to this line.
// TODO EST-T (t): transpose words
// TODO ESC-<: move to first entry in history
// TODO ESC->: move to last entry in history
'b' | 'B' => Ok(KeyPress::ESC_B),
'c' | 'C' => Ok(KeyPress::ESC_C),
'f' | 'F' => Ok(KeyPress::ESC_F),
'l' | 'L' => Ok(KeyPress::ESC_L),
'u' | 'U' => Ok(KeyPress::ESC_U),
_ => {
writeln!(io::stderr(), "key: {:?}, seq1, {:?}", KeyPress::ESC, seq1).unwrap();
Ok(KeyPress::UNKNOWN_ESC_SEQ)
}
}
/// Handles reading and editting the readline buffer.
/// It will also handle special inputs in an appropriate fashion
/// (e.g., C-c will exit readline)
kill_ring: &mut KillRing,
original_termios: termios::Termios)
let mut stdout = io::stdout();
try!(write_and_flush(&mut stdout, prompt.as_bytes()));
let mut s = State::new(&mut stdout, prompt, MAX_LINE, get_columns(), history.len());
let stdin = io::stdin();
let mut chars = stdin.lock().chars();
Main
committed
loop {
let c = chars.next().unwrap();
if c.is_err() && SIGWINCH.compare_and_swap(true, false, atomic::Ordering::SeqCst) {
s.cols = get_columns();
try!(s.refresh_line());
continue;
}
let mut ch = try!(c);
try!(edit_insert(&mut s, ch));
continue;
}
let mut key = char_to_key_press(ch);
// autocomplete
if key == KeyPress::TAB && completer.is_some() {
let next = try!(complete_line(&mut chars, &mut s, completer.unwrap()));
if next.is_some() {
if !ch.is_control() {
try!(edit_insert(&mut s, ch));
continue;
}
} else if key == KeyPress::CTRL_R {
// Search history backward
let next = try!(reverse_incremental_search(&mut chars, &mut s, history));
if next.is_some() {
key = try!(escape_sequence(&mut chars));
if key == KeyPress::UNKNOWN_ESC_SEQ {
continue;
}
KeyPress::CTRL_A => {
kill_ring.reset();
// Move to the beginning of line.
try!(edit_move_home(&mut s))
}
KeyPress::CTRL_B => {
kill_ring.reset();
// Move back a character.
try!(edit_move_left(&mut s))
}
KeyPress::CTRL_C => {
kill_ring.reset();
return Err(error::ReadlineError::Interrupted);
}
KeyPress::CTRL_E => {
kill_ring.reset();
// Move to the end of line.
try!(edit_move_end(&mut s))
}
KeyPress::CTRL_F => {
kill_ring.reset();
// Move forward a character.
try!(edit_move_right(&mut s))
}
KeyPress::CTRL_H | KeyPress::BACKSPACE => {
kill_ring.reset();
// Delete one character backward.
try!(edit_backspace(&mut s))
}
KeyPress::CTRL_K => {
// Kill the text from point to the end of the line.
if let Some(text) = try!(edit_kill_line(&mut s)) {
kill_ring.kill(&text, true)
KeyPress::CTRL_L => {
// Clear the screen leaving the current line at the top of the screen.
try!(edit_history_next(&mut s, history, false))
try!(edit_history_next(&mut s, history, true))
KeyPress::CTRL_T => {
kill_ring.reset();
// Exchange the char before cursor with the character at cursor.
try!(edit_transpose_chars(&mut s))
}
KeyPress::CTRL_U => {
// Kill backward from point to the beginning of the line.
if let Some(text) = try!(edit_discard_line(&mut s)) {
kill_ring.kill(&text, false)
// TODO CTRL_V // Quoted insert
KeyPress::CTRL_W => {
// Kill the word behind point, using white space as a word boundary
if let Some(text) = try!(edit_delete_prev_word(&mut s, char::is_whitespace)) {
}
}
KeyPress::CTRL_Y => {
// retrieve (yank) last item killed
if let Some(text) = kill_ring.yank() {
try!(edit_yank(&mut s, text))
KeyPress::CTRL_Z => {
try!(disable_raw_mode(original_termios));
try!(signal::raise(signal::SIGSTOP));
try!(enable_raw_mode()); // TODO original_termios may have changed
try!(s.refresh_line())
}
// TODO CTRL-_ // undo
KeyPress::ENTER | KeyPress::CTRL_J => {
// Accept the line regardless of where the cursor is.
kill_ring.reset();
try!(edit_move_end(&mut s));
break;
}
KeyPress::ESC_BACKSPACE => {
// kill one word backward
// Kill from the cursor the start of the current word, or, if between words, to the start of the previous word.
if let Some(text) = try!(edit_delete_prev_word(&mut s,
|ch| !ch.is_alphanumeric())) {
kill_ring.kill(&text, false)
}
}
KeyPress::ESC_B => {
// move backwards one word
kill_ring.reset();
try!(edit_move_to_prev_word(&mut s))
}
KeyPress::ESC_C => {
// capitalize word after point
kill_ring.reset();
try!(edit_word(&mut s, WordAction::CAPITALIZE))
}
KeyPress::ESC_D => {
// kill one word forward
if let Some(text) = try!(edit_delete_word(&mut s)) {
kill_ring.kill(&text, true)
KeyPress::ESC_F => {
// move forwards one word
kill_ring.reset();
try!(edit_move_to_next_word(&mut s))
}
KeyPress::ESC_L => {
// lowercase word after point
kill_ring.reset();
try!(edit_word(&mut s, WordAction::LOWERCASE))
}
KeyPress::ESC_U => {
// uppercase word after point
kill_ring.reset();
try!(edit_word(&mut s, WordAction::UPPERCASE))
if let Some((yank_size, text)) = kill_ring.yank_pop() {
try!(edit_yank_pop(&mut s, yank_size, text))
}
}
KeyPress::ESC_SEQ_DELETE => {
kill_ring.reset();
try!(edit_delete(&mut s))
}
_ => {
kill_ring.reset();
// Insert the character typed.
try!(edit_insert(&mut s, ch))
}
Main
committed
}
}
struct Guard(termios::Termios);
#[allow(unused_must_use)]
impl Drop for Guard {
fn drop(&mut self) {
let Guard(termios) = *self;
disable_raw_mode(termios);
}
}
/// Readline method that will enable RAW mode, call the `readline_edit()`
let original_termios = try!(enable_raw_mode());
let guard = Guard(original_termios);
let user_input = readline_edit(prompt, history, completer, kill_ring, original_termios);
drop(guard); // try!(disable_raw_mode(original_termios));
}
fn readline_direct() -> Result<String> {
if try!(io::stdin().read_line(&mut line)) > 0 {
Ok(line)
} else {
Err(error::ReadlineError::Eof)
}
}
/// Line editor
pub struct Editor<'completer> {
unsupported_term: bool,
stdin_isatty: bool,
history: History,
completer: Option<&'completer Completer>,
impl<'completer> Editor<'completer> {
pub fn new() -> Editor<'completer> {
// TODO check what is done in rl_initialize()
// if the number of columns is stored here, we need a SIGWINCH handler...
unsupported_term: is_unsupported_term(),
stdin_isatty: is_a_tty(libc::STDIN_FILENO),
stdout_isatty: is_a_tty(libc::STDOUT_FILENO),
};
if !editor.unsupported_term && editor.stdin_isatty && editor.stdout_isatty {
install_sigwinch_handler();
}
/// This method will read a line from STDIN and will display a `prompt`
pub fn readline(&mut self, prompt: &str) -> Result<String> {
// Write prompt and flush it to stdout
let mut stdout = io::stdout();
try!(write_and_flush(&mut stdout, prompt.as_bytes()));
readline_direct()
} else {