neovim-gtk/src/shell.rs

use std::cell::{RefMut, RefCell};
use std::rc::Rc;
use std::sync::Arc;
use std::ops::Deref;
use std::thread;

use cairo;
use pangocairo::CairoContextExt;
use pango;
use pango::FontDescription;
use gdk::{ModifierType, EventConfigure, EventButton, EventMotion, EventType, EventScroll};
use gdk_sys;
use glib;
use gtk;
use gtk::prelude::*;

use neovim_lib::{Neovim, NeovimApi, Value};
use neovim_lib::neovim_api::Tabpage;

use settings::{Settings, FontSource};
use ui_model::{UiModel, Cell, Attrs, Color, ModelRect, COLOR_BLACK, COLOR_WHITE, COLOR_RED};
use nvim;
use nvim::{RedrawEvents, GuiApi, RepaintMode, ErrorReport, NeovimClient};
use input;
use input::keyval_to_input_string;
use cursor::Cursor;
use ui::UiMutex;
use popup_menu::PopupMenu;
use tabline::Tabline;
use error;
use mode;

const DEFAULT_FONT_NAME: &str = "DejaVu Sans Mono 12";
pub const MINIMUM_SUPPORTED_NVIM_VERSION: &str = "0.2";

macro_rules! idle_cb_call {
    ($state:ident.$cb:ident($( $x:expr ),*)) => (
            glib::idle_add(move || {
                               if let Some(ref cb) = $state.borrow().$cb {
                                   (&mut *cb.borrow_mut())($($x),*);
                               }

                               glib::Continue(false)
                           });
    )
}

pub struct State {
    pub model: UiModel,
    bg_color: Color,
    fg_color: Color,
    sp_color: Color,
    cur_attrs: Option<Attrs>,
    mouse_enabled: bool,
    nvim: Rc<RefCell<NeovimClient>>,
    font_desc: FontDescription,
    cursor: Option<Cursor>,
    popup_menu: RefCell<PopupMenu>,
    settings: Rc<RefCell<Settings>>,

    pub mode: mode::Mode,

    stack: gtk::Stack,
    drawing_area: gtk::DrawingArea,
    tabs: Tabline,
    error_area: error::ErrorArea,

    line_height: Option<f64>,
    char_width: Option<f64>,
    request_resize: bool,
    resize_timer: Option<glib::SourceId>,

    options: ShellOptions,

    detach_cb: Option<Box<RefCell<FnMut() + Send + 'static>>>,
}

impl State {
    pub fn new(settings: Rc<RefCell<Settings>>, options: ShellOptions) -> State {
        let drawing_area = gtk::DrawingArea::new();
        let popup_menu = RefCell::new(PopupMenu::new(&drawing_area));

        State {
            model: UiModel::new(1, 1),
            nvim: Rc::new(RefCell::new(NeovimClient::new())),
            cur_attrs: None,
            bg_color: COLOR_BLACK,
            fg_color: COLOR_WHITE,
            sp_color: COLOR_RED,
            mouse_enabled: true,
            font_desc: FontDescription::from_string(DEFAULT_FONT_NAME),
            cursor: None,
            popup_menu,
            settings: settings,

            mode: mode::Mode::new(),

            // UI
            stack: gtk::Stack::new(),
            drawing_area,
            tabs: Tabline::new(),
            error_area: error::ErrorArea::new(),

            line_height: None,
            char_width: None,
            resize_timer: None,
            request_resize: false,

            options,

            detach_cb: None,
        }
    }

    pub fn get_foreground(&self) -> &Color {
        &self.fg_color
    }

    pub fn get_background(&self) -> &Color {
        &self.bg_color
    }

    pub fn nvim(&self) -> RefMut<Neovim> {
        RefMut::map(self.nvim.borrow_mut(), |n| n.nvim_mut())
    }

    pub fn nvim_clone(&self) -> Rc<RefCell<NeovimClient>> {
        self.nvim.clone()
    }

    pub fn set_detach_cb<F>(&mut self, cb: Option<F>)
        where F: FnMut() + Send + 'static
    {
        if cb.is_some() {
            self.detach_cb = Some(Box::new(RefCell::new(cb.unwrap())));
        } else {
            self.detach_cb = None;
        }
    }

    fn create_pango_font(&self) -> FontDescription {
        self.font_desc.clone()
    }

    fn colors<'a>(&'a self, cell: &'a Cell) -> (&'a Color, &'a Color) {
        let bg = if let Some(ref bg) = cell.attrs.background {
            bg
        } else {
            &self.bg_color
        };
        let fg = if let Some(ref fg) = cell.attrs.foreground {
            fg
        } else {
            &self.fg_color
        };

        if cell.attrs.reverse {
            (fg, bg)
        } else {
            (bg, fg)
        }
    }

    pub fn get_font_desc(&self) -> &FontDescription {
        &self.font_desc
    }

    pub fn set_font_desc(&mut self, desc: &str) {
        self.font_desc = FontDescription::from_string(desc);
        self.line_height = None;
        self.char_width = None;
    }

    pub fn open_file(&self, path: &str) {
        let mut nvim = self.nvim();
        nvim.command(&format!("e {}", path)).report_err(&mut *nvim);
    }

    pub fn cd(&self, path: &str) {
        let mut nvim = self.nvim();
        nvim.command(&format!("cd {}", path)).report_err(&mut *nvim);
    }

    fn request_resize(&mut self) {
        self.request_resize = true;
    }

    fn close_popup_menu(&self) {
        if self.popup_menu.borrow().is_open() {
            let mut nvim = self.nvim();
            nvim.input("<Esc>").report_err(&mut *nvim);
        }
    }

    fn queue_draw_area<M: AsRef<ModelRect>>(&self, rect_list: &Vec<M>) {
        match (&self.line_height, &self.char_width) {
            (&Some(line_height), &Some(char_width)) => {
                for rect in rect_list {
                    let mut rect = rect.as_ref().clone();
                    // this need to repain also line under curren line
                    // in case underscore or 'g' symbol is go here
                    // right one for italic symbol
                    rect.extend(0, 1, 0, 1);
                    let (x, y, width, height) = rect.to_area(line_height, char_width);
                    self.drawing_area.queue_draw_area(x, y, width, height);
                }
            }
            _ => self.drawing_area.queue_draw(),
        }
    }

    fn calc_char_bounds(&self, ctx: &cairo::Context) -> (i32, i32) {
        let layout = ctx.create_pango_layout();

        let desc = self.create_pango_font();
        layout.set_font_description(Some(&desc));
        layout.set_text("A", -1);

        layout.get_pixel_size()
    }

    fn calc_line_metrics(&mut self, ctx: &cairo::Context) {
        if self.line_height.is_none() {
            let (width, height) = self.calc_char_bounds(ctx);
            self.line_height = Some(height as f64);
            self.char_width = Some(width as f64);
        }
    }

    fn calc_nvim_size(&self) -> Option<(usize, usize)> {
        if let Some(line_height) = self.line_height {
            if let Some(char_width) = self.char_width {
                let alloc = self.drawing_area.get_allocation();
                return Some(((alloc.width as f64 / char_width).trunc() as usize,
                             (alloc.height as f64 / line_height).trunc() as usize));
            }
        }

        None
    }

    fn show_error_area(&self) {
        let stack = self.stack.clone();
        gtk::idle_add(move || {
                          stack.set_visible_child_name("Error");
                          Continue(false)
                      });
    }
}

pub struct UiState {
    mouse_pressed: bool,
}

impl UiState {
    pub fn new() -> UiState {
        UiState { mouse_pressed: false }
    }
}

#[derive(Clone)]
pub struct ShellOptions {
    nvim_bin_path: Option<String>,
    open_path: Option<String>,
}

impl ShellOptions {
    pub fn new(nvim_bin_path: Option<String>, open_path: Option<String>) -> Self {
        ShellOptions {
            nvim_bin_path,
            open_path,
        }
    }
}

pub struct Shell {
    pub state: Arc<UiMutex<State>>,
    ui_state: Rc<RefCell<UiState>>,

    widget: gtk::Box,
}

impl Shell {
    pub fn new(settings: Rc<RefCell<Settings>>, options: ShellOptions) -> Shell {
        let shell = Shell {
            state: Arc::new(UiMutex::new(State::new(settings, options))),
            ui_state: Rc::new(RefCell::new(UiState::new())),

            widget: gtk::Box::new(gtk::Orientation::Vertical, 0),
        };

        let shell_ref = Arc::downgrade(&shell.state);
        shell.state.borrow_mut().cursor = Some(Cursor::new(shell_ref));

        shell
    }

    pub fn is_nvim_initialized(&self) -> bool {
        let state = self.state.borrow();
        let nvim = state.nvim.borrow();
        nvim.is_initialized()
    }

    pub fn init(&mut self) {
        let mut state = self.state.borrow_mut();
        state.drawing_area.set_hexpand(true);
        state.drawing_area.set_vexpand(true);
        state.drawing_area.set_can_focus(true);

        let nvim_box = gtk::Box::new(gtk::Orientation::Vertical, 0);

        nvim_box.pack_start(&*state.tabs, false, true, 0);
        nvim_box.pack_start(&state.drawing_area, true, true, 0);

        state.stack.add_named(&nvim_box, "Nvim");
        state.stack.add_named(&*state.error_area, "Error");

        self.widget.pack_start(&state.stack, true, true, 0);

        state
            .drawing_area
            .set_events((gdk_sys::GDK_BUTTON_RELEASE_MASK | gdk_sys::GDK_BUTTON_PRESS_MASK |
                         gdk_sys::GDK_BUTTON_MOTION_MASK |
                         gdk_sys::GDK_SCROLL_MASK)
                                .bits() as i32);

        let ref_state = self.state.clone();
        let ref_ui_state = self.ui_state.clone();
        state
            .drawing_area
            .connect_button_press_event(move |_, ev| {
                                            gtk_button_press(&mut *ref_state.borrow_mut(),
                                                             &mut *ref_ui_state.borrow_mut(),
                                                             ev)
                                        });

        let ref_ui_state = self.ui_state.clone();
        state
            .drawing_area
            .connect_button_release_event(move |_, _| {
                                              gtk_button_release(&mut *ref_ui_state.borrow_mut())
                                          });


        let ref_state = self.state.clone();
        let ref_ui_state = self.ui_state.clone();
        state
            .drawing_area
            .connect_motion_notify_event(move |_, ev| {
                                             gtk_motion_notify(&mut *ref_state.borrow_mut(),
                                                               &mut *ref_ui_state.borrow_mut(),
                                                               ev)
                                         });

        let ref_state = self.state.clone();
        state
            .drawing_area
            .connect_draw(move |_, ctx| gtk_draw(&ref_state, ctx));

        let ref_state = self.state.clone();
        state
            .drawing_area
            .connect_key_press_event(move |_, ev| {
                                         let mut shell = ref_state.borrow_mut();
                                         shell.cursor.as_mut().unwrap().reset_state();
                                         let mut nvim = shell.nvim();
                                         input::gtk_key_press(&mut *nvim, ev)
                                     });

        let ref_state = self.state.clone();
        state
            .drawing_area
            .connect_scroll_event(move |_, ev| gtk_scroll_event(&mut *ref_state.borrow_mut(), ev));

        let ref_state = self.state.clone();
        state
            .drawing_area
            .connect_focus_in_event(move |_, _| gtk_focus_in(&mut *ref_state.borrow_mut()));

        let ref_state = self.state.clone();
        state
            .drawing_area
            .connect_focus_out_event(move |_, _| gtk_focus_out(&mut *ref_state.borrow_mut()));

        let ref_state = self.state.clone();
        state
            .drawing_area
            .connect_configure_event(move |_, ev| gtk_configure_event(&ref_state, ev));

        state.cursor.as_mut().unwrap().start();
    }

    #[cfg(unix)]
    pub fn redraw(&self, mode: &RepaintMode) {
        self.state.borrow_mut().on_redraw(mode);
    }

    #[cfg(unix)]
    pub fn set_font_desc(&self, font_name: &str) {
        self.state.borrow_mut().set_font_desc(font_name);
    }

    pub fn grab_focus(&self) {
        self.state.borrow().drawing_area.grab_focus();
    }

    pub fn open_file(&self, path: &str) {
        self.state.borrow().open_file(path);
    }

    pub fn cd(&self, path: &str) {
        self.state.borrow().cd(path);
    }

    pub fn detach_ui(&mut self) {
        let state = self.state.borrow();
        state.nvim().ui_detach().expect("Error in ui_detach");
    }

    pub fn edit_paste(&self) {
        let state = self.state.borrow();
        let paste_command = if state.mode.is(&mode::NvimMode::Normal) {
            "\"*p"
        } else {
            "<Esc>\"*pa"
        };

        let mut nvim = state.nvim();
        nvim.input(paste_command).report_err(&mut *nvim);
    }

    pub fn edit_save_all(&self) {
        let state = self.state.borrow();
        let mut nvim = &mut *state.nvim();
        nvim.command(":wa").report_err(nvim);
    }

    pub fn set_detach_cb<F>(&self, cb: Option<F>)
        where F: FnMut() + Send + 'static
    {
        let mut state = self.state.borrow_mut();
        state.set_detach_cb(cb);
    }
}

impl Deref for Shell {
    type Target = gtk::Box;

    fn deref(&self) -> &gtk::Box {
        &self.widget
    }
}

fn gtk_focus_in(state: &mut State) -> Inhibit {
    state.cursor.as_mut().unwrap().enter_focus();
    let point = state.model.cur_point();
    state.on_redraw(&RepaintMode::Area(point));
    Inhibit(false)
}

fn gtk_focus_out(state: &mut State) -> Inhibit {
    state.cursor.as_mut().unwrap().leave_focus();
    let point = state.model.cur_point();
    state.on_redraw(&RepaintMode::Area(point));

    Inhibit(false)
}

fn gtk_scroll_event(state: &mut State, ev: &EventScroll) -> Inhibit {
    if !state.mouse_enabled {
        return Inhibit(false);
    }

    state.close_popup_menu();

    match ev.as_ref().direction {
        gdk_sys::GdkScrollDirection::Right => {
            mouse_input(state, "ScrollWheelRight", ev.get_state(), ev.get_position())
        }
        gdk_sys::GdkScrollDirection::Left => {
            mouse_input(state, "ScrollWheelLeft", ev.get_state(), ev.get_position())
        }
        gdk_sys::GdkScrollDirection::Up => {
            mouse_input(state, "ScrollWheelUp", ev.get_state(), ev.get_position())
        }
        gdk_sys::GdkScrollDirection::Down => {
            mouse_input(state, "ScrollWheelDown", ev.get_state(), ev.get_position())
        }
        _ => (),
    }
    Inhibit(false)
}

fn gtk_button_press(shell: &mut State, ui_state: &mut UiState, ev: &EventButton) -> Inhibit {
    if ev.get_event_type() != EventType::ButtonPress {
        return Inhibit(false);
    }

    if shell.mouse_enabled {
        ui_state.mouse_pressed = true;

        mouse_input(shell, "LeftMouse", ev.get_state(), ev.get_position());
    }
    Inhibit(false)
}

fn mouse_input(shell: &mut State, input: &str, state: ModifierType, position: (f64, f64)) {
    if let Some(line_height) = shell.line_height {
        if let Some(char_width) = shell.char_width {

            let mut nvim = shell.nvim();
            let (x, y) = position;
            let col = (x / char_width).trunc() as u64;
            let row = (y / line_height).trunc() as u64;
            let input_str = format!("{}<{},{}>", keyval_to_input_string(input, state), col, row);
            nvim.input(&input_str)
                .expect("Can't send mouse input event");
        }
    }
}

fn gtk_button_release(ui_state: &mut UiState) -> Inhibit {
    ui_state.mouse_pressed = false;
    Inhibit(false)
}

fn gtk_motion_notify(shell: &mut State, ui_state: &mut UiState, ev: &EventMotion) -> Inhibit {
    if shell.mouse_enabled && ui_state.mouse_pressed {
        mouse_input(shell, "LeftDrag", ev.get_state(), ev.get_position());
    }
    Inhibit(false)
}

#[inline]
fn update_line_metrics(state_arc: &Arc<UiMutex<State>>, ctx: &cairo::Context) {
    let mut state = state_arc.borrow_mut();

    if state.line_height.is_none() {
        state.calc_line_metrics(ctx);
    }
}

fn gtk_draw(state_arc: &Arc<UiMutex<State>>, ctx: &cairo::Context) -> Inhibit {
    update_line_metrics(state_arc, ctx);
    init_nvim(state_arc);

    let mut state = state_arc.borrow_mut();
    // in case nvim not initialized
    if !state.nvim.borrow().is_error() {
        draw(&*state, ctx);
        request_window_resize(&mut *state);
    }

    Inhibit(false)
}

fn show_nvim_start_error(err: nvim::NvimInitError, state_arc: Arc<UiMutex<State>>) {
    let source = err.source();
    let cmd = err.cmd().unwrap().to_owned();

    glib::idle_add(move || {
        let state = state_arc.borrow();
        state.nvim.borrow_mut().set_error();
        state.error_area.show_nvim_start_error(&source, &cmd);
        state.show_error_area();

        Continue(false)
    });
}

fn show_nvim_init_error(err: nvim::NvimInitError, state_arc: Arc<UiMutex<State>>) {
    let source = err.source();

    glib::idle_add(move || {
        let state = state_arc.borrow();
        state.nvim.borrow_mut().set_error();
        state.error_area.show_nvim_init_error(&source);
        state.show_error_area();

        Continue(false)
    });
}

fn init_nvim_async(state_arc: Arc<UiMutex<State>>,
                   options: ShellOptions,
                   cols: usize,
                   rows: usize) {
    // execute nvim
    let mut nvim = match nvim::start(state_arc.clone(), options.nvim_bin_path.as_ref()) {
        Ok(nvim) => nvim,
        Err(err) => {
            show_nvim_start_error(err, state_arc);
            return;
        }
    };

    // add callback on session end
    let guard = nvim.session.take_dispatch_guard();
    let state_ref = state_arc.clone();
    thread::spawn(move || {
                      guard.join().expect("Can't join dispatch thread");

                      idle_cb_call!(state_ref.detach_cb());
                  });

    // attach ui
    let mut nvim = Some(nvim);
    glib::idle_add(move || {
        let mut nvim = nvim.take().unwrap();
        if let Err(err) = nvim::post_start_init(&mut nvim,
                                                options.open_path.as_ref(),
                                                cols as u64,
                                                rows as u64) {
            show_nvim_init_error(err, state_arc.clone());
        } else {
            let state = state_arc.borrow_mut();
            state.nvim.borrow_mut().set_initialized(nvim);
        }

        Continue(false)
    });
}

fn init_nvim(state_arc: &Arc<UiMutex<State>>) {
    let state = state_arc.borrow();

    let mut nvim = state.nvim.borrow_mut();
    if nvim.is_uninitialized() {
        nvim.set_in_progress();

        let (cols, rows) = state.calc_nvim_size().unwrap();

        let state_arc = state_arc.clone();
        let options = state.options.clone();
        thread::spawn(move || { init_nvim_async(state_arc, options, cols, rows); });
    }
}

#[inline]
fn get_model_clip(state: &State,
                  line_height: f64,
                  char_width: f64,
                  clip: (f64, f64, f64, f64))
                  -> ModelRect {
    let mut model_clip =
        ModelRect::from_area(line_height, char_width, clip.0, clip.1, clip.2, clip.3);
    // in some cases symbols from previous row affect next row
    // for example underscore symbol or 'g'
    // also for italic text it is possible that symbol can affect next one
    // see deference between logical rect and ink rect
    model_clip.extend(1, 0, 1, 0);
    state.model.limit_to_model(&mut model_clip);

    model_clip
}

#[inline]
fn draw_backgound(state: &State,
                  draw_bitmap: &ModelBitamp,
                  ctx: &cairo::Context,
                  line_height: f64,
                  char_width: f64,
                  model_clip: &ModelRect) {
    let line_x = model_clip.left as f64 * char_width;
    let mut line_y: f64 = model_clip.top as f64 * line_height;

    for (line_idx, line) in state.model.clip_model(model_clip) {
        ctx.move_to(line_x, line_y);

        for (col_idx, cell) in line.iter() {
            let current_point = ctx.get_current_point();

            if !draw_bitmap.get(col_idx, line_idx) {
                let (bg, _) = state.colors(cell);

                if &state.bg_color != bg {
                    ctx.set_source_rgb(bg.0, bg.1, bg.2);
                    ctx.rectangle(current_point.0, current_point.1, char_width, line_height);
                    ctx.fill();
                }

            }

            ctx.move_to(current_point.0 + char_width, current_point.1);
        }
        line_y += line_height;
    }
}

fn draw(state: &State, ctx: &cairo::Context) {
    let layout = ctx.create_pango_layout();
    let mut desc = state.create_pango_font();
    let mut buf = String::with_capacity(4);

    let (row, col) = state.model.get_cursor();

    let line_height = state.line_height.unwrap();
    let char_width = state.char_width.unwrap();
    let mut draw_bitmap = ModelBitamp::new(state.model.columns, state.model.rows);

    ctx.set_source_rgb(state.bg_color.0, state.bg_color.1, state.bg_color.2);
    ctx.paint();

    let clip_rects = &ctx.copy_clip_rectangle_list().rectangles;
    for clip_idx in 0..clip_rects.len() {
        let clip = clip_rects.get(clip_idx).unwrap();

        let model_clip =
            get_model_clip(state,
                           line_height,
                           char_width,
                           (clip.x, clip.y, clip.x + clip.width, clip.y + clip.height));

        let line_x = model_clip.left as f64 * char_width;
        let mut line_y: f64 = model_clip.top as f64 * line_height;

        draw_backgound(state,
                       &draw_bitmap,
                       ctx,
                       line_height,
                       char_width,
                       &model_clip);

        for (line_idx, line) in state.model.clip_model(&model_clip) {

            ctx.move_to(line_x, line_y);

            for (col_idx, cell) in line.iter() {
                let current_point = ctx.get_current_point();

                if !draw_bitmap.get(col_idx, line_idx) {
                    let double_width = line.is_double_width(col_idx);

                    let (bg, fg) = state.colors(cell);

                    if row == line_idx && col == col_idx {
                        state
                            .cursor
                            .as_ref()
                            .unwrap()
                            .draw(ctx,
                                  state,
                                  char_width,
                                  line_height,
                                  line_y,
                                  double_width,
                                  bg);

                        ctx.move_to(current_point.0, current_point.1);
                    }


                    if !cell.ch.is_whitespace() {
                        update_font_description(&mut desc, &cell.attrs);

                        layout.set_font_description(Some(&desc));
                        buf.clear();
                        buf.push(cell.ch);
                        layout.set_text(&buf, -1);

                        // correct layout for double_width chars
                        if double_width {
                            let (dw_width, dw_height) = layout.get_pixel_size();
                            let x_offset = (char_width * 2.0 - dw_width as f64) / 2.0;
                            let y_offset = (line_height - dw_height as f64) / 2.0;
                            ctx.rel_move_to(x_offset, y_offset);
                        }

                        ctx.set_source_rgb(fg.0, fg.1, fg.2);
                        ctx.update_pango_layout(&layout);
                        ctx.show_pango_layout(&layout);
                    }

                    if cell.attrs.underline || cell.attrs.undercurl {
                        // [TODO]: Current gtk-rs bindings does not provide fontmetrics access
                        // so it is not possible to find right position for underline or undercurl position
                        // > update_font_description(&mut desc, &cell.attrs);
                        // > layout.get_context().unwrap().get_metrics();
                        let top_offset = line_height * 0.9;

                        let sp = if let Some(ref sp) = cell.attrs.special {
                            sp
                        } else {
                            &state.sp_color
                        };

                        ctx.set_source_rgba(sp.0, sp.1, sp.2, 0.7);
                        if cell.attrs.undercurl {
                            ctx.set_dash(&[4.0, 2.0], 0.0);
                            ctx.set_line_width(2.0);
                            ctx.move_to(current_point.0, line_y + top_offset);
                            ctx.line_to(current_point.0 + char_width, line_y + top_offset);
                            ctx.stroke();
                            ctx.set_dash(&[], 0.0);
                        } else if cell.attrs.underline {
                            ctx.set_line_width(1.0);
                            ctx.move_to(current_point.0, line_y + top_offset);
                            ctx.line_to(current_point.0 + char_width, line_y + top_offset);
                            ctx.stroke();
                        }
                    }
                }

                ctx.move_to(current_point.0 + char_width, current_point.1);
            }

            line_y += line_height;
        }

        draw_bitmap.fill_from_model(&model_clip);
    }
}

#[inline]
fn update_font_description(desc: &mut FontDescription, attrs: &Attrs) {
    desc.unset_fields(pango::FONT_MASK_STYLE | pango::FONT_MASK_WEIGHT);
    if attrs.italic {
        desc.set_style(pango::Style::Italic);
    }
    if attrs.bold {
        desc.set_weight(pango::Weight::Bold);
    }
}

fn request_window_resize(state: &mut State) {
    if !state.request_resize {
        return;
    }
    if state.resize_timer.is_some() {
        return;
    }

    state.request_resize = false;

    let width = state.drawing_area.get_allocated_width();
    let height = state.drawing_area.get_allocated_height();
    let request_height = (state.model.rows as f64 * state.line_height.unwrap()) as i32;
    let request_width = (state.model.columns as f64 * state.char_width.unwrap()) as i32;

    if width != request_width || height != request_height {
        let window: gtk::Window = state
            .drawing_area
            .get_toplevel()
            .unwrap()
            .downcast()
            .unwrap();
        let (win_width, win_height) = window.get_size();
        let h_border = win_width - width;
        let v_border = win_height - height;
        window.resize(request_width + h_border, request_height + v_border);
    }
}

fn split_color(indexed_color: u64) -> Color {
    let r = ((indexed_color >> 16) & 0xff) as f64;
    let g = ((indexed_color >> 8) & 0xff) as f64;
    let b = (indexed_color & 0xff) as f64;
    Color(r / 255.0, g / 255.0, b / 255.0)
}

fn gtk_configure_event(state: &Arc<UiMutex<State>>, _: &EventConfigure) -> bool {
    let mut state_ref = state.borrow_mut();

    if let Some(timer) = state_ref.resize_timer {
        glib::source_remove(timer);
    }

    if !state_ref.nvim.borrow().is_initialized() {
        return false;
    }

    if let Some((columns, rows)) = state_ref.calc_nvim_size() {
        let state = state.clone();
        state_ref.resize_timer = Some(glib::timeout_add(250, move || {
            let mut state_ref = state.borrow_mut();

            state_ref.resize_timer = None;

            if state_ref.model.rows != rows || state_ref.model.columns != columns {
                if let Err(err) = state_ref.nvim().ui_try_resize(columns as u64, rows as u64) {
                    println!("Error trying resize nvim {}", err);
                }
            }
            Continue(false)
        }));
    }
    false
}

impl RedrawEvents for State {
    fn on_cursor_goto(&mut self, row: u64, col: u64) -> RepaintMode {
        RepaintMode::AreaList(self.model.set_cursor(row as usize, col as usize))
    }

    fn on_put(&mut self, text: &str) -> RepaintMode {
        RepaintMode::Area(self.model.put(text, self.cur_attrs.as_ref()))
    }

    fn on_clear(&mut self) -> RepaintMode {
        self.model.clear();
        RepaintMode::All
    }

    fn on_eol_clear(&mut self) -> RepaintMode {
        RepaintMode::Area(self.model.eol_clear())
    }

    fn on_resize(&mut self, columns: u64, rows: u64) -> RepaintMode {
        self.model = UiModel::new(rows, columns);
        self.request_resize();
        RepaintMode::All
    }

    fn on_redraw(&self, mode: &RepaintMode) {
        match mode {
            &RepaintMode::All => self.drawing_area.queue_draw(),
            &RepaintMode::Area(ref rect) => self.queue_draw_area(&vec![rect]),
            &RepaintMode::AreaList(ref list) => self.queue_draw_area(&list.list),
            &RepaintMode::Nothing => (),
        }
    }

    fn on_set_scroll_region(&mut self, top: u64, bot: u64, left: u64, right: u64) -> RepaintMode {
        self.model.set_scroll_region(top, bot, left, right);
        RepaintMode::Nothing
    }

    fn on_scroll(&mut self, count: i64) -> RepaintMode {
        RepaintMode::Area(self.model.scroll(count))
    }

    fn on_highlight_set(&mut self, attrs: &Vec<(Value, Value)>) -> RepaintMode {
        let mut model_attrs = Attrs::new();

        for &(ref key_val, ref val) in attrs {
            if let Some(key) = key_val.as_str() {
                match key {
                    "foreground" => {
                        if let Some(fg) = val.as_u64() {
                            model_attrs.foreground = Some(split_color(fg));
                        }
                    }
                    "background" => {
                        if let Some(bg) = val.as_u64() {
                            model_attrs.background = Some(split_color(bg));
                        }
                    }
                    "special" => {
                        if let Some(bg) = val.as_u64() {
                            model_attrs.special = Some(split_color(bg));
                        }
                    }
                    "reverse" => model_attrs.reverse = true,
                    "bold" => model_attrs.bold = true,
                    "italic" => model_attrs.italic = true,
                    "underline" => model_attrs.underline = true,
                    "undercurl" => model_attrs.undercurl = true,
                    attr_key => println!("unknown attribute {}", attr_key),
                };
            } else {
                panic!("attr key must be string");
            }
        }

        self.cur_attrs = Some(model_attrs);
        RepaintMode::Nothing
    }

    fn on_update_bg(&mut self, bg: i64) -> RepaintMode {
        if bg >= 0 {
            self.bg_color = split_color(bg as u64);
        } else {
            self.bg_color = COLOR_BLACK;
        }
        RepaintMode::Nothing
    }

    fn on_update_fg(&mut self, fg: i64) -> RepaintMode {
        if fg >= 0 {
            self.fg_color = split_color(fg as u64);
        } else {
            self.fg_color = COLOR_WHITE;
        }
        RepaintMode::Nothing
    }

    fn on_update_sp(&mut self, sp: i64) -> RepaintMode {
        if sp >= 0 {
            self.sp_color = split_color(sp as u64);
        } else {
            self.sp_color = COLOR_RED;
        }
        RepaintMode::Nothing
    }

    fn on_mode_change(&mut self, mode: &str, idx: u64) -> RepaintMode {
        self.mode.update(mode, idx as usize);
        RepaintMode::Area(self.model.cur_point())
    }

    fn on_mouse(&mut self, on: bool) -> RepaintMode {
        self.mouse_enabled = on;
        RepaintMode::Nothing
    }

    fn on_busy(&mut self, busy: bool) -> RepaintMode {
        if busy {
            self.cursor.as_mut().unwrap().busy_on();
        } else {
            self.cursor.as_mut().unwrap().busy_off();
        }
        RepaintMode::Area(self.model.cur_point())
    }

    fn popupmenu_show(&mut self,
                      menu: &Vec<Vec<&str>>,
                      selected: i64,
                      row: u64,
                      col: u64)
                      -> RepaintMode {
        match (&self.line_height, &self.char_width) {
            (&Some(line_height), &Some(char_width)) => {
                let point = ModelRect::point(col as usize, row as usize);
                let (x, y, width, height) = point.to_area(line_height, char_width);

                self.popup_menu
                    .borrow_mut()
                    .show(&self, menu, selected, x, y, width, height);
            }
            _ => (),
        };

        RepaintMode::Nothing
    }

    fn popupmenu_hide(&mut self) -> RepaintMode {
        self.popup_menu.borrow_mut().hide();
        RepaintMode::Nothing
    }

    fn popupmenu_select(&mut self, selected: i64) -> RepaintMode {
        self.popup_menu.borrow().select(selected);
        RepaintMode::Nothing
    }


    fn tabline_update(&mut self,
                      selected: Tabpage,
                      tabs: Vec<(Tabpage, Option<String>)>)
                      -> RepaintMode {
        self.tabs.update_tabs(&self.nvim, &selected, &tabs);

        RepaintMode::Nothing
    }


    fn mode_info_set(&mut self,
                     cursor_style_enabled: bool,
                     mode_info: Vec<nvim::ModeInfo>)
                     -> RepaintMode {
        self.mode.set_info(cursor_style_enabled, mode_info);
        RepaintMode::Nothing
    }
}

impl GuiApi for State {
    fn set_font(&mut self, font_desc: &str) {
        self.set_font_desc(font_desc);
        self.request_resize();

        let mut settings = self.settings.borrow_mut();
        settings.set_font_source(FontSource::Rpc);
    }
}

pub struct ModelBitamp {
    words_for_cols: usize,
    model: Vec<u64>,
}

impl ModelBitamp {
    pub fn new(cols: usize, rows: usize) -> ModelBitamp {
        let words_for_cols = cols / 64 + 1;

        ModelBitamp {
            words_for_cols: words_for_cols,
            model: vec![0; rows * words_for_cols],
        }
    }

    fn fill_from_model(&mut self, rect: &ModelRect) {
        for row in rect.top..rect.bot + 1 {
            let row_pos = self.words_for_cols * row;
            for col in rect.left..rect.right + 1 {
                let col_pos = col / 64;
                let col_offset = col % 64;
                self.model[row_pos + col_pos] |= 1 << col_offset;
            }
        }
    }

    #[inline]
    fn get(&self, col: usize, row: usize) -> bool {
        let row_pos = self.words_for_cols * row;
        let col_pos = col / 64;
        let col_offset = col % 64;
        self.model[row_pos + col_pos] & (1 << col_offset) != 0
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_bitmap() {
        let mut bitmap = ModelBitamp::new(80, 24);
        bitmap.fill_from_model(&ModelRect::new(22, 22, 63, 68));

        assert_eq!(true, bitmap.get(63, 22));
        assert_eq!(true, bitmap.get(68, 22));
        assert_eq!(false, bitmap.get(62, 22));
    }
}