cursive/src/backend/termion.rs

extern crate termion;

use ::backend;
use ::event::{Event, Key};
use self::termion::color;
use self::termion::event::Key as TKey;
use self::termion::input::TermRead;
use self::termion::style;
use self::termion::raw::IntoRawMode;
use std::cell::Cell;
use std::collections::BTreeMap;
use std::io::Write;
use std::fmt;

use ::theme;

pub struct Concrete {
    terminal: termion::raw::RawTerminal<::std::io::Stdout>,
    current_style: Cell<theme::ColorStyle>,
    colors: BTreeMap<i16, (Box<color::Color>, Box<color::Color>)>,
}

trait Effectable {
    fn on(&self);
    fn off(&self);
}

struct ColorRef<'a>(&'a color::Color);

impl <'a> color::Color for ColorRef<'a> {

    #[inline]
    fn write_fg(&self, f: &mut fmt::Formatter) -> fmt::Result {
        self.0.write_fg(f)
    }

    #[inline]
    fn write_bg(&self, f: &mut fmt::Formatter) -> fmt::Result {
        self.0.write_bg(f)
    }

}

impl Effectable for theme::Effect {
    fn on(&self) {
        match *self {
            theme::Effect::Simple => (),
            theme::Effect::Reverse => print!("{}", style::Invert),
        }
    }

    fn off(&self) {
        match *self {
            theme::Effect::Simple => (),
            theme::Effect::Reverse => print!("{}", style::NoInvert),
        }
    }
}

fn apply_colors(fg: &color::Color, bg: &color::Color) {
    print!("{}{}", color::Fg(ColorRef(fg)), color::Bg(ColorRef(bg)));
}

impl Concrete {
    fn apply_colorstyle(&self, color_style: theme::ColorStyle) {
        let (ref fg, ref bg) = self.colors[&color_style.id()];
        apply_colors(&**fg, &**bg);
    }
}

impl backend::Backend for Concrete {
    fn init() -> Self {
        print!("{}", termion::cursor::Hide);

        let backend = Concrete {
            terminal: ::std::io::stdout().into_raw_mode().unwrap(),
            current_style: Cell::new(theme::ColorStyle::Background),
            colors: BTreeMap::new(),
        };

        backend
    }

    fn finish(&mut self) {
        // Maybe we should clear everything?
        print!("{}{}", termion::cursor::Show, termion::cursor::Goto(1, 1));
        self.clear();
    }

    fn init_color_style(&mut self, style: theme::ColorStyle,
                        foreground: &theme::Color, background: &theme::Color) {
        // Step 1: convert foreground and background into proper termion Color
        self.colors.insert(style.id(),
                           (colour_to_termion_colour(foreground),
                            colour_to_termion_colour(background)));
    }

    fn with_color<F: FnOnce()>(&self, color: theme::ColorStyle, f: F) {
        // TODO: actually use colors
        // TODO: careful! need to remember the previous state
        //                and apply it back

        let current_style = self.current_style.get();

        self.apply_colorstyle(color);

        self.current_style.set(color);
        f();
        self.current_style.set(current_style);

        self.apply_colorstyle(current_style);
    }

    fn with_effect<F: FnOnce()>(&self, effect: theme::Effect, f: F) {
        // TODO: actually use effects
        // TODO: careful! need to remember the previous state
        //                and apply it back
        effect.on();
        f();
        effect.off();
    }

    fn has_colors(&self) -> bool {
        // TODO: color support detection?
        true
    }

    fn screen_size(&self) -> (usize, usize) {
        let (x, y) = termion::terminal_size().unwrap_or((1, 1));
        (x as usize, y as usize)
    }

    fn clear(&self) {
        self.apply_colorstyle(theme::ColorStyle::Background);
        print!("{}", termion::clear::All);
    }

    fn refresh(&mut self) {
        self.terminal.flush().unwrap();
    }

    fn print_at(&self, (x, y): (usize, usize), text: &str) {
        // TODO: terminals are 1-based. Should we add 1 here?
        print!("{}{}", termion::cursor::Goto(x as u16, y as u16), text);
    }

    fn set_refresh_rate(&mut self, fps: u32) {
        // TODO: handle async refresh, when no input is entered.
        // Could be done with a timeout on the event polling,
        // if it was supportedd.
    }

    fn poll_event(&self) -> Event {
        // TODO: parse input
        if let Some(key) = ::std::io::stdin().keys().next() {
            match key.unwrap() {
                TKey::Esc => Event::Key(Key::Esc),
                TKey::Backspace => Event::Key(Key::Backspace),
                TKey::Left => Event::Key(Key::Left),
                TKey::Right => Event::Key(Key::Right),
                TKey::Up => Event::Key(Key::Up),
                TKey::Down => Event::Key(Key::Down),
                TKey::Home => Event::Key(Key::Home),
                TKey::End => Event::Key(Key::End),
                TKey::PageUp => Event::Key(Key::PageUp),
                TKey::PageDown => Event::Key(Key::PageDown),
                TKey::Delete => Event::Key(Key::Del),
                TKey::Insert => Event::Key(Key::Ins),
                TKey::F(i) if i < 12 => Event::Key(Key::from_f(i)),
                TKey::F(j) => Event::Unknown(-(j as i32)),
                TKey::Char('\n') => Event::Key(Key::Enter),
                TKey::Char('\t') => Event::Key(Key::Tab),
                TKey::Char(c) => Event::Char(c),
                TKey::Ctrl('c') => panic!("Ctrl-C pressed"),
                TKey::Ctrl(c) => Event::CtrlChar(c),
                TKey::Alt(c) => Event::AltChar(c),
                _ => Event::Unknown(-1),
            }
        } else {
            Event::Refresh
        }
    }
}

fn colour_to_termion_colour(clr: &theme::Color) -> Box<color::Color> {
    match *clr {
        theme::Color::Dark(theme::BaseColor::Black) => Box::new(color::Black),
        theme::Color::Dark(theme::BaseColor::Red) => Box::new(color::Red),
        theme::Color::Dark(theme::BaseColor::Green) => Box::new(color::Green),
        theme::Color::Dark(theme::BaseColor::Yellow) => {
            Box::new(color::Yellow)
        }
        theme::Color::Dark(theme::BaseColor::Blue) => Box::new(color::Blue),
        theme::Color::Dark(theme::BaseColor::Magenta) => {
            Box::new(color::Magenta)
        }
        theme::Color::Dark(theme::BaseColor::Cyan) => Box::new(color::Cyan),
        theme::Color::Dark(theme::BaseColor::White) => Box::new(color::White),

        theme::Color::Light(theme::BaseColor::Black) => {
            Box::new(color::LightBlack)
        }
        theme::Color::Light(theme::BaseColor::Red) => {
            Box::new(color::LightRed)
        }
        theme::Color::Light(theme::BaseColor::Green) => {
            Box::new(color::LightGreen)
        }
        theme::Color::Light(theme::BaseColor::Yellow) => {
            Box::new(color::LightYellow)
        }
        theme::Color::Light(theme::BaseColor::Blue) => {
            Box::new(color::LightBlue)
        }
        theme::Color::Light(theme::BaseColor::Magenta) => {
            Box::new(color::LightMagenta)
        }
        theme::Color::Light(theme::BaseColor::Cyan) => {
            Box::new(color::LightCyan)
        }
        theme::Color::Light(theme::BaseColor::White) => {
            Box::new(color::LightWhite)
        }

        theme::Color::Rgb(r, g, b) => Box::new(color::Rgb(r, g, b)),
        theme::Color::RgbLowRes(r, g, b) => {
            Box::new(color::AnsiValue::rgb(r, g, b))
        }
    }
}