neovim-gtk/src/ui.rs

use std::cell::{RefCell, Ref, RefMut};
use std::{env, thread};
use std::rc::Rc;
use std::sync::Arc;

use gtk;
use gtk_sys;
use gtk::prelude::*;
use gtk::{ApplicationWindow, HeaderBar, ToolButton, Image, AboutDialog};
use gio::prelude::*;
use gio::{Menu, MenuExt, MenuItem, MenuItemExt, SimpleAction};

use nvim_config::NvimConfig;
use settings::Settings;
use shell::{Shell, ShellOptions};
use shell_dlg;
use project::Projects;
use plug_manager;

pub struct Ui {
    initialized: bool,
    comps: Arc<UiMutex<Components>>,
    settings: Rc<RefCell<Settings>>,
    shell: Rc<RefCell<Shell>>,
    projects: Rc<RefCell<Projects>>,
    plug_manager: Arc<UiMutex<plug_manager::Manager>>,
}

pub struct Components {
    window: Option<ApplicationWindow>,
    open_btn: ToolButton,
}

impl Components {
    fn new() -> Components {
        let save_image =
            Image::new_from_icon_name("document-open", gtk_sys::GTK_ICON_SIZE_SMALL_TOOLBAR as i32);

        Components {
            open_btn: ToolButton::new(Some(&save_image), "Open"),
            window: None,
        }
    }

    pub fn close_window(&self) {
        self.window.as_ref().unwrap().destroy();
    }

    pub fn window(&self) -> &ApplicationWindow {
        self.window.as_ref().unwrap()
    }
}

impl Ui {
    pub fn new(options: ShellOptions) -> Ui {
        let mut plug_manager = plug_manager::Manager::new();
        let plug_config = plug_manager.load_config();

        let nvim_config = NvimConfig::new(plug_config);
        let plug_manager = Arc::new(UiMutex::new(plug_manager));
        let comps = Arc::new(UiMutex::new(Components::new()));
        let settings = Rc::new(RefCell::new(Settings::new()));
        let shell = Rc::new(RefCell::new(Shell::new(settings.clone(), options, nvim_config)));
        settings.borrow_mut().set_shell(Rc::downgrade(&shell));

        let projects = Projects::new(&comps.borrow().open_btn, shell.clone());

        Ui {
            initialized: false,
            comps,
            shell,
            settings,
            projects,
            plug_manager,
        }
    }

    pub fn init(&mut self, app: &gtk::Application) {
        if self.initialized {
            return;
        }
        self.initialized = true;

        if app.prefers_app_menu() {
            self.create_main_menu(app);
        }

        let mut settings = self.settings.borrow_mut();
        settings.init();

        let mut comps = self.comps.borrow_mut();

        self.shell.borrow_mut().init();

        comps.window = Some(ApplicationWindow::new(app));
        let window = comps.window.as_ref().unwrap();

        // Client side decorations including the toolbar are disabled via NVIM_GTK_NO_HEADERBAR=1
        let use_header_bar = env::var("NVIM_GTK_NO_HEADERBAR")
            .map(|opt| opt.trim() != "1")
            .unwrap_or(true);

        if use_header_bar {
            let header_bar = HeaderBar::new();

            let projects = self.projects.clone();
            header_bar.pack_start(&comps.open_btn);
            comps.open_btn.connect_clicked(
                move |_| projects.borrow_mut().show(),
            );

            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), "Save");

            let shell = self.shell.clone();
            save_btn.connect_clicked(move |_| shell.borrow_mut().edit_save_all());
            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), "Paste");
            let shell = self.shell.clone();
            paste_btn.connect_clicked(move |_| shell.borrow_mut().edit_paste());
            header_bar.pack_start(&paste_btn);

            let plug_image = Image::new_from_icon_name(
                "system-software-install",
                gtk_sys::GTK_ICON_SIZE_SMALL_TOOLBAR as i32,
            );
            let plug_btn = ToolButton::new(Some(&plug_image), "Plug");

            let comps_ref = self.comps.clone();
            let plug_manager_ref = self.plug_manager.clone();
            plug_btn.connect_clicked(move |_| {
                plug_manager::Ui::new(&plug_manager_ref).show(
                    comps_ref
                        .borrow()
                        .window
                        .as_ref()
                        .unwrap(),
                )
            });
            header_bar.pack_start(&plug_btn);

            header_bar.set_show_close_button(true);

            window.set_titlebar(Some(&header_bar));
        }

        window.set_default_size(800, 600);

        let shell = self.shell.borrow();
        window.add(&**shell);

        window.show_all();
        window.set_title("NeovimGtk");

        let comps_ref = self.comps.clone();
        let shell_ref = self.shell.clone();
        window.connect_delete_event(move |_, _| gtk_delete(&*comps_ref, &*shell_ref));

        shell.grab_focus();

        let comps_ref = self.comps.clone();
        shell.set_detach_cb(Some(move || {
            let comps_ref = comps_ref.clone();
            gtk::idle_add(move || {
                comps_ref.borrow().close_window();
                Continue(false)
            });
        }));

        let state_ref = self.shell.borrow().state.clone();
        let plug_manager_ref = self.plug_manager.clone();
        shell.set_nvim_started_cb(Some(move || {
            plug_manager_ref.borrow_mut().init_nvim_client(
                state_ref.borrow().nvim_clone(),
            );
        }));
    }

    fn create_main_menu(&self, app: &gtk::Application) {
        let menu = Menu::new();

        let about = MenuItem::new("About", None);
        about.set_detailed_action("app.HelpAbout");
        menu.append_item(&about);

        app.set_app_menu(Some(&menu));

        let about_action = SimpleAction::new("HelpAbout", None);
        let comps = self.comps.clone();
        about_action.connect_activate(move |_, _| on_help_about(&*comps.borrow()));
        about_action.set_enabled(true);
        app.add_action(&about_action);
    }
}

fn on_help_about(comps: &Components) {
    let about = AboutDialog::new();
    about.set_transient_for(comps.window.as_ref());
    about.set_program_name("NeovimGtk");
    about.set_version(env!("CARGO_PKG_VERSION"));
    about.set_logo_icon_name("org.daa.NeovimGtk");
    about.set_authors(&[env!("CARGO_PKG_AUTHORS")]);

    about.connect_response(|about, _| about.destroy());
    about.show();
}

fn gtk_delete(comps: &UiMutex<Components>, shell: &RefCell<Shell>) -> Inhibit {
    if !shell.borrow().is_nvim_initialized() {
        return Inhibit(false);
    }

    Inhibit(if shell_dlg::can_close_window(comps, shell) {
        let comps = comps.borrow();
        comps.close_window();
        shell.borrow_mut().detach_ui();
        false
    } else {
        true
    })
}

pub struct UiMutex<T: ?Sized> {
    thread: thread::ThreadId,
    data: RefCell<T>,
}

unsafe impl<T: ?Sized> Send for UiMutex<T> {}
unsafe impl<T: ?Sized> Sync for UiMutex<T> {}

impl<T> UiMutex<T> {
    pub fn new(t: T) -> UiMutex<T> {
        UiMutex {
            thread: thread::current().id(),
            data: RefCell::new(t),
        }
    }
}

impl<T: ?Sized> UiMutex<T> {
    pub fn borrow(&self) -> Ref<T> {
        self.assert_ui_thread();
        self.data.borrow()
    }

    pub fn borrow_mut(&self) -> RefMut<T> {
        self.assert_ui_thread();
        self.data.borrow_mut()
    }

    #[inline]
    fn assert_ui_thread(&self) {
        if thread::current().id() != self.thread {
            panic!("Can access to UI only from main thread");
        }
    }
}