use std::io::{Read, Write}; use {IntoRawMode, TerminalError}; #[cfg(feature = "nightly")] use std::io::{Chars, CharsError}; /// A key. #[derive(Clone, Copy, PartialEq, Eq, Hash, Debug)] pub enum Key { /// Backspace. Backspace, /// Left arrow. Left, /// Right arrow. Right, /// Up arrow. Up, /// Down arrow. Down, /// Alt modified character. Alt(char), /// Normal character. Char(char), /// Invalid character code. Invalid, // TODO handle errors better? /// IO error. Error, } /// An iterator over input keys. #[cfg(feature = "nightly")] pub struct Keys { chars: I, } #[cfg(feature = "nightly")] impl>> Iterator for Keys { type Item = Key; fn next(&mut self) -> Option { match self.chars.next() { Some(Ok('\x1B')) => Some(match self.chars.next() { Some(Ok('[')) => match self.chars.next() { Some(Ok('D')) => Key::Left, Some(Ok('C')) => Key::Right, Some(Ok('A')) => Key::Up, Some(Ok('B')) => Key::Down, _ => Key::Invalid, }, Some(Ok(c)) => Key::Alt(c), Some(Err(_)) | None => Key::Invalid, }), Some(Ok('\x7F')) => Some(Key::Backspace), Some(Ok(c)) => Some(Key::Char(c)), None => None, Some(Err(_)) => Some(Key::Error), } } } /// Extension to `Read` trait. pub trait TermRead { /// An iterator over key inputs. #[cfg(feature = "nightly")] fn keys(self) -> Keys> where Self: Sized; /// Read a password. /// /// EOT and ETX will abort the prompt, returning `None`. Newline or carriage return will /// complete the password input. fn read_passwd(&mut self, writer: &mut W) -> Result, TerminalError>; } impl TermRead for R { #[cfg(feature = "nightly")] fn keys(self) -> Keys> { Keys { chars: self.chars(), } } fn read_passwd(&mut self, writer: &mut W) -> Result, TerminalError> { let _raw = try!(writer.into_raw_mode()); let mut passbuf = Vec::with_capacity(30); for c in self.bytes() { match c { Err(_) => return Err(TerminalError::StdinError), Ok(0) | Ok(3) | Ok(4) => return Ok(None), Ok(b'\n') | Ok(b'\r') => break, Ok(c) => passbuf.push(c), } } let passwd = try!(String::from_utf8(passbuf).map_err(|_| TerminalError::UnicodeError)); Ok(Some(passwd)) } } #[cfg(test)] mod test { #[cfg(feature = "nightly")] #[test] fn test_keys() { use {TermRead, Key}; let mut i = b"\x1Bayo\x7F\x1B[D".keys(); assert_eq!(i.next(), Some(Key::Alt('a'))); assert_eq!(i.next(), Some(Key::Char('y'))); assert_eq!(i.next(), Some(Key::Char('o'))); assert_eq!(i.next(), Some(Key::Backspace)); assert_eq!(i.next(), Some(Key::Left)); assert_eq!(i.next(), None); } }