use std::cell::{RefCell, Ref, RefMut}; use std::thread; use gtk; use gtk_sys; use gtk::prelude::*; use gtk::{ApplicationWindow, HeaderBar, ToolButton, Image}; use gdk::Event; use neovim_lib::NeovimApi; use settings; use shell::{Shell, NvimMode}; use nvim::ErrorReport; macro_rules! ui_thread_var { ($id:ident, $ty:ty, $expr:expr) => (thread_local!(pub static $id: RefCell<$ty> = { assert_ui_thread(); RefCell::new($expr) });) } ui_thread_var!(UI, Ui, Ui::new()); ui_thread_var!(SH, Shell, Shell::new()); ui_thread_var!(SET, settings::Settings, settings::Settings::new()); #[macro_export] macro_rules! SHELL { (&$id:ident = $expr:expr) => ( SH.with(|shell_cell| { let $id = &shell_cell.borrow(); $expr }); ); ($id:ident = $expr:expr) => ( SH.with(|shell_cell| { let mut $id = &mut shell_cell.borrow_mut(); $expr }); ); } pub struct Ui { pub initialized: bool, pub window: Option, header_bar: HeaderBar, } impl Ui { pub fn new() -> Ui { Ui { window: None, header_bar: HeaderBar::new(), initialized: false, } } pub fn close_window(&self) { self.window.as_ref().unwrap().destroy(); } pub fn destroy(&mut self) { self.close_window(); SHELL!(shell = { shell.nvim().ui_detach().expect("Error in ui_detach"); }); } pub fn init(&mut self, app: >k::Application) { if self.initialized { return; } self.initialized = true; SHELL!(shell = { SET.with(|settings| { let mut settings = settings.borrow_mut(); settings.init(&mut shell); }); self.header_bar.set_show_close_button(true); let save_image = Image::new_from_icon_name("document-save", gtk_sys::GTK_ICON_SIZE_SMALL_TOOLBAR as i32); let save_btn = ToolButton::new(Some(&save_image), None); save_btn.connect_clicked(|_| edit_save_all()); self.header_bar.pack_start(&save_btn); let paste_image = Image::new_from_icon_name("edit-paste", gtk_sys::GTK_ICON_SIZE_SMALL_TOOLBAR as i32); let paste_btn = ToolButton::new(Some(&paste_image), None); paste_btn.connect_clicked(|_| edit_paste()); self.header_bar.pack_start(&paste_btn); shell.init(); self.window = Some(ApplicationWindow::new(app)); let window = self.window.as_ref().unwrap(); window.set_titlebar(Some(&self.header_bar)); window.add(&shell.drawing_area); window.show_all(); window.connect_delete_event(gtk_delete); window.set_title("Neovim-gtk"); shell.add_configure_event(); }); } } fn edit_paste() { SHELL!(shell = { let paste_command = if shell.mode == NvimMode::Normal { "\"*p" } else { "\"*pa" }; let mut nvim = shell.nvim(); nvim.input(paste_command).report_err(nvim); }); } fn edit_save_all() { SHELL!(shell = { let mut nvim = shell.nvim(); nvim.command(":wa").report_err(nvim); }); } fn quit() { UI.with(|ui_cell| { let mut ui = ui_cell.borrow_mut(); ui.destroy(); }); } fn gtk_delete(_: &ApplicationWindow, _: &Event) -> Inhibit { quit(); Inhibit(false) } pub struct UiMutex { data: RefCell, } unsafe impl Send for UiMutex {} unsafe impl Sync for UiMutex {} impl UiMutex { pub fn new(t: T) -> UiMutex { UiMutex { data: RefCell::new(t) } } } impl UiMutex { pub fn borrow(&self) -> Ref { assert_ui_thread(); self.data.borrow() } pub fn borrow_mut(&self) -> RefMut { assert_ui_thread(); self.data.borrow_mut() } } #[inline] fn assert_ui_thread() { match thread::current().name() { Some("main") => (), Some(ref name) => { panic!("Can create UI only from main thread, {}", name); } None => panic!("Can create UI only from main thread, current thiread has no name"), } }