path: root/src/main.rs

#[cfg(test)]
mod tests;

mod boundaries;
mod cli;
mod command_is_executing;
mod errors;
mod input;
mod layout;
mod os_input_output;
mod panes;
mod pty_bus;
mod screen;
mod tab;
mod utils;

mod wasm_vm;

use std::cell::RefCell;
use std::collections::HashMap;
use std::io::Write;
use std::os::unix::net::UnixStream;
use std::path::PathBuf;
use std::sync::mpsc::{channel, sync_channel, Receiver, SendError, Sender, SyncSender};
use std::thread;

use panes::PaneId;
use serde::{Deserialize, Serialize};
use structopt::StructOpt;
use termion::input::TermRead;
use wasm_vm::PluginEnv;
use wasmer::{ChainableNamedResolver, Instance, Module, Store, Value};
use wasmer_wasi::{Pipe, WasiState};

use crate::cli::CliArgs;
use crate::command_is_executing::CommandIsExecuting;
use crate::errors::{
    AppContext, ContextType, ErrorContext, PluginContext, PtyContext, ScreenContext,
};
use crate::input::input_loop;
use crate::layout::Layout;
use crate::os_input_output::{get_os_input, OsApi};
use crate::pty_bus::{PtyBus, PtyInstruction, VteEvent};
use crate::screen::{Screen, ScreenInstruction};
use crate::utils::{
    consts::{MOSAIC_IPC_PIPE, MOSAIC_TMP_DIR, MOSAIC_TMP_LOG_DIR},
    logging::*,
};
use crate::wasm_vm::{mosaic_imports, wasi_stdout, wasi_write_string, PluginInstruction};

thread_local!(static OPENCALLS: RefCell<ErrorContext> = RefCell::default());

#[derive(Serialize, Deserialize, Debug)]
enum ApiCommand {
    OpenFile(PathBuf),
    SplitHorizontally,
    SplitVertically,
    MoveFocus,
}

pub type ChannelWithContext<T> = (Sender<(T, ErrorContext)>, Receiver<(T, ErrorContext)>);
pub type SyncChannelWithContext<T> = (SyncSender<(T, ErrorContext)>, Receiver<(T, ErrorContext)>);

#[derive(Clone)]
enum SenderType<T: Clone> {
    Sender(Sender<(T, ErrorContext)>),
    SyncSender(SyncSender<(T, ErrorContext)>),
}

#[derive(Clone)]
pub struct SenderWithContext<T: Clone> {
    err_ctx: ErrorContext,
    sender: SenderType<T>,
}

impl<T: Clone> SenderWithContext<T> {
    fn new(err_ctx: ErrorContext, sender: SenderType<T>) -> Self {
        Self { err_ctx, sender }
    }

    pub fn send(&self, event: T) -> Result<(), SendError<(T, ErrorContext)>> {
        match self.sender {
            SenderType::Sender(ref s) => s.send((event, self.err_ctx)),
            SenderType::SyncSender(ref s) => s.send((event, self.err_ctx)),
        }
    }

    pub fn update(&mut self, new_ctx: ErrorContext) {
        self.err_ctx = new_ctx;
    }
}

unsafe impl<T: Clone> Send for SenderWithContext<T> {}
unsafe impl<T: Clone> Sync for SenderWithContext<T> {}

pub fn main() {
    let opts = CliArgs::from_args();
    if let Some(split_dir) = opts.split {
        match split_dir {
            'h' => {
                let mut stream = UnixStream::connect(MOSAIC_IPC_PIPE).unwrap();
                let api_command = bincode::serialize(&ApiCommand::SplitHorizontally).unwrap();
                stream.write_all(&api_command).unwrap();
            }
            'v' => {
                let mut stream = UnixStream::connect(MOSAIC_IPC_PIPE).unwrap();
                let api_command = bincode::serialize(&ApiCommand::SplitVertically).unwrap();
                stream.write_all(&api_command).unwrap();
            }
            _ => {}
        };
    } else if opts.move_focus {
        let mut stream = UnixStream::connect(MOSAIC_IPC_PIPE).unwrap();
        let api_command = bincode::serialize(&ApiCommand::MoveFocus).unwrap();
        stream.write_all(&api_command).unwrap();
    } else if let Some(file_to_open) = opts.open_file {
        let mut stream = UnixStream::connect(MOSAIC_IPC_PIPE).unwrap();
        let api_command = bincode::serialize(&ApiCommand::OpenFile(file_to_open)).unwrap();
        stream.write_all(&api_command).unwrap();
    } else {
        let os_input = get_os_input();
        atomic_create_dir(MOSAIC_TMP_DIR).unwrap();
        atomic_create_dir(MOSAIC_TMP_LOG_DIR).unwrap();
        start(Box::new(os_input), opts);
    }
}

#[derive(Clone)]
pub enum AppInstruction {
    Exit,
    Error(String),
}

pub fn start(mut os_input: Box<dyn OsApi>, opts: CliArgs) {
    let mut active_threads = vec![];

    let command_is_executing = CommandIsExecuting::new();

    let full_screen_ws = os_input.get_terminal_size_using_fd(0);
    os_input.set_raw_mode(0);
    let (send_screen_instructions, receive_screen_instructions): ChannelWithContext<
        ScreenInstruction,
    > = channel();
    let err_ctx = OPENCALLS.with(|ctx| *ctx.borrow());
    let mut send_screen_instructions =
        SenderWithContext::new(err_ctx, SenderType::Sender(send_screen_instructions));

    let (send_pty_instructions, receive_pty_instructions): ChannelWithContext<PtyInstruction> =
        channel();
    let mut send_pty_instructions =
        SenderWithContext::new(err_ctx, SenderType::Sender(send_pty_instructions));

    let (send_plugin_instructions, receive_plugin_instructions): ChannelWithContext<
        PluginInstruction,
    > = channel();
    let send_plugin_instructions =
        SenderWithContext::new(err_ctx, SenderType::Sender(send_plugin_instructions));

    let (send_app_instructions, receive_app_instructions): SyncChannelWithContext<AppInstruction> =
        sync_channel(0);
    let send_app_instructions =
        SenderWithContext::new(err_ctx, SenderType::SyncSender(send_app_instructions));

    let mut pty_bus = PtyBus::new(
        receive_pty_instructions,
        send_screen_instructions.clone(),
        send_plugin_instructions.clone(),
        os_input.clone(),
        opts.debug,
    );
    let maybe_layout = opts.layout.map(Layout::new);

    #[cfg(not(test))]
    std::panic::set_hook({
        use crate::errors::handle_panic;
        let send_app_instructions = send_app_instructions.clone();
        Box::new(move |info| {
            handle_panic(info, &send_app_instructions);
        })
    });

    active_threads.push(
        thread::Builder::new()
            .name("pty".to_string())
            .spawn({
                let mut command_is_executing = command_is_executing.clone();
                move || {
                    if let Some(layout) = maybe_layout {
                        pty_bus.spawn_terminals_for_layout(layout);
                    } else {
                        let pid = pty_bus.spawn_terminal(None);
                        pty_bus
                            .send_screen_instructions
                            .send(ScreenInstruction::NewTab(pid))
                            .unwrap();
                    }

                    loop {
                        let (event, mut err_ctx) = pty_bus
                            .receive_pty_instructions
                            .recv()
                            .expect("failed to receive event on channel");
                        err_ctx.add_call(ContextType::Pty(PtyContext::from(&event)));
                        pty_bus.send_screen_instructions.update(err_ctx);
                        match event {
                            PtyInstruction::SpawnTerminal(file_to_open) => {
                                let pid = pty_bus.spawn_terminal(file_to_open);
                                pty_bus
                                    .send_screen_instructions
                                    .send(ScreenInstruction::NewPane(PaneId::Terminal(pid)))
                                    .unwrap();
                            }
                            PtyInstruction::SpawnTerminalVertically(file_to_open) => {
                                let pid = pty_bus.spawn_terminal(file_to_open);
                                pty_bus
                                    .send_screen_instructions
                                    .send(ScreenInstruction::VerticalSplit(PaneId::Terminal(pid)))
                                    .unwrap();
                            }
                            PtyInstruction::SpawnTerminalHorizontally(file_to_open) => {
                                let pid = pty_bus.spawn_terminal(file_to_open);
                                pty_bus
                                    .send_screen_instructions
                                    .send(ScreenInstruction::HorizontalSplit(PaneId::Terminal(pid)))
                                    .unwrap();
                            }
                            PtyInstruction::NewTab => {
                                let pid = pty_bus.spawn_terminal(None);
                                pty_bus
                                    .send_screen_instructions
                                    .send(ScreenInstruction::NewTab(pid))
                                    .unwrap();
                            }
                            PtyInstruction::ClosePane(id) => {
                                pty_bus.close_pane(id);
                                command_is_executing.done_closing_pane();
                            }
                            PtyInstruction::CloseTab(ids) => {
                                pty_bus.close_tab(ids);
                                command_is_executing.done_closing_pane();
                            }
                            PtyInstruction::Quit => {
                                break;
                            }
                        }
                    }
                }
            })
            .unwrap(),
    );

    active_threads.push(
        thread::Builder::new()
            .name("screen".to_string())
            .spawn({
                let mut command_is_executing = command_is_executing.clone();
                let os_input = os_input.clone();
                let send_pty_instructions = send_pty_instructions.clone();
                let send_plugin_instructions = send_plugin_instructions.clone();
                let send_app_instructions = send_app_instructions.clone();
                let max_panes = opts.max_panes;

                move || {
                    let mut screen = Screen::new(
                        receive_screen_instructions,
                        send_pty_instructions,
                        send_plugin_instructions,
                        send_app_instructions,
                        &full_screen_ws,
                        os_input,
                        max_panes,
                    );
                    loop {
                        let (event, mut err_ctx) = screen
                            .receiver
                            .recv()
                            .expect("failed to receive event on channel");
                        err_ctx.add_call(ContextType::Screen(ScreenContext::from(&event)));
                        screen.send_app_instructions.update(err_ctx);
                        screen.send_pty_instructions.update(err_ctx);
                        match event {
                            ScreenInstruction::Pty(pid, vte_event) => {
                                screen
                                    .get_active_tab_mut()
                                    .unwrap()
                                    .handle_pty_event(pid, vte_event);
                            }
                            ScreenInstruction::Render => {
                                screen.render();
                            }
                            ScreenInstruction::NewPane(pid) => {
                                screen.get_active_tab_mut().unwrap().new_pane(pid);
                                command_is_executing.done_opening_new_pane();
                            }
                            ScreenInstruction::HorizontalSplit(pid) => {
                                screen.get_active_tab_mut().unwrap().horizontal_split(pid);
                                command_is_executing.done_opening_new_pane();
                            }
                            ScreenInstruction::VerticalSplit(pid) => {
                                screen.get_active_tab_mut().unwrap().vertical_split(pid);
                                command_is_executing.done_opening_new_pane();
                            }
                            ScreenInstruction::WriteCharacter(bytes) => {
                                screen
                                    .get_active_tab_mut()
                                    .unwrap()
                                    .write_to_active_terminal(bytes);
                            }
                            ScreenInstruction::ResizeLeft => {
                                screen.get_active_tab_mut().unwrap().resize_left();
                            }
                            ScreenInstruction::ResizeRight => {
                                screen.get_active_tab_mut().unwrap().resize_right();
                            }
                            ScreenInstruction::ResizeDown => {
                                screen.get_active_tab_mut().unwrap().resize_down();
                            }
                            ScreenInstruction::ResizeUp => {
                                screen.get_active_tab_mut().unwrap().resize_up();
                            }
                            ScreenInstruction::MoveFocus => {
                                screen.get_active_tab_mut().unwrap().move_focus();
                            }
                            ScreenInstruction::MoveFocusLeft => {
                                screen.get_active_tab_mut().unwrap().move_focus_left();
                            }
                            ScreenInstruction::MoveFocusDown => {
                                screen.get_active_tab_mut().unwrap().move_focus_down();
                            }
                            ScreenInstruction::MoveFocusRight => {
                                screen.get_active_tab_mut().unwrap().move_focus_right();
                            }
                            ScreenInstruction::MoveFocusUp => {
                                screen.get_active_tab_mut().unwrap().move_focus_up();
                            }
                            ScreenInstruction::ScrollUp => {
                                screen
                                    .get_active_tab_mut()
                                    .unwrap()
                                    .scroll_active_terminal_up();
                            }
                            ScreenInstruction::ScrollDown => {
                                screen
                                    .get_active_tab_mut()
                                    .unwrap()
                                    .scroll_active_terminal_down();
                            }
                            ScreenInstruction::ClearScroll => {
                                screen
                                    .get_active_tab_mut()
                                    .unwrap()
                                    .clear_active_terminal_scroll();
                            }
                            ScreenInstruction::CloseFocusedPane => {
                                screen.get_active_tab_mut().unwrap().close_focused_pane();
                                screen.render();
                            }
                            ScreenInstruction::ClosePane(id) => {
                                screen.get_active_tab_mut().unwrap().close_pane(id);
                                screen.render();
                            }
                            ScreenInstruction::ToggleActiveTerminalFullscreen => {
                                screen
                                    .get_active_tab_mut()
                                    .unwrap()
                                    .toggle_active_pane_fullscreen();
                            }
                            ScreenInstruction::NewTab(pane_id) => {
                                screen.new_tab(pane_id);
                                command_is_executing.done_opening_new_pane();
                            }
                            ScreenInstruction::SwitchTabNext => screen.switch_tab_next(),
                            ScreenInstruction::SwitchTabPrev => screen.switch_tab_prev(),
                            ScreenInstruction::CloseTab => screen.close_tab(),
                            ScreenInstruction::ApplyLayout((layout, new_pane_pids)) => {
                                screen.apply_layout(layout, new_pane_pids)
                            }
                            ScreenInstruction::Quit => {
                                break;
                            }
                        }
                    }
                }
            })
            .unwrap(),
    );

    active_threads.push(
        thread::Builder::new()
            .name("wasm".to_string())
            .spawn({
                let mut send_pty_instructions = send_pty_instructions.clone();
                let mut send_screen_instructions = send_screen_instructions.clone();

                move || {
                    let store = Store::default();

                    let mut plugin_id = 0;
                    let mut plugin_map = HashMap::new();

                    loop {
                        let (event, mut err_ctx) = receive_plugin_instructions
                            .recv()
                            .expect("failed to receive event on channel");
                        err_ctx.add_call(ContextType::Plugin(PluginContext::from(&event)));
                        send_screen_instructions.update(err_ctx);
                        send_pty_instructions.update(err_ctx);
                        match event {
                            PluginInstruction::Load(pid_tx, path) => {
                                // FIXME: Cache this compiled module on disk. I could use `(de)serialize_to_file()` for that
                                let module = Module::from_file(&store, &path).unwrap();

                                let output = Pipe::new();
                                let input = Pipe::new();
                                let mut wasi_env = WasiState::new("mosaic")
                                    .env("CLICOLOR_FORCE", "1")
                                    .preopen(|p| {
                                        p.directory(".") // FIXME: Change this to a more meaningful dir
                                            .alias(".")
                                            .read(true)
                                            .write(true)
                                            .create(true)
                                    })
                                    .unwrap()
                                    .stdin(Box::new(input))
                                    .stdout(Box::new(output))
                                    .finalize()
                                    .unwrap();

                                let wasi = wasi_env.import_object(&module).unwrap();

                                let plugin_env = PluginEnv {
                                    send_pty_instructions: send_pty_instructions.clone(),
                                    wasi_env,
                                };

                                let mosaic = mosaic_imports(&store, &plugin_env);
                                let instance =
                                    Instance::new(&module, &mosaic.chain_back(wasi)).unwrap();

                                let start = instance.exports.get_function("_start").unwrap();

                                // This eventually calls the `.init()` method
                                start.call(&[]).unwrap();

                                plugin_map.insert(plugin_id, (instance, plugin_env));
                                pid_tx.send(plugin_id).unwrap();
                                plugin_id += 1;
                            }
                            PluginInstruction::Draw(buf_tx, pid, rows, cols) => {
                                let (instance, plugin_env) = plugin_map.get(&pid).unwrap();

                                let draw = instance.exports.get_function("draw").unwrap();

                                draw.call(&[Value::I32(rows as i32), Value::I32(cols as i32)])
                                    .unwrap();

                                buf_tx.send(wasi_stdout(&plugin_env.wasi_env)).unwrap();
                            }
                            PluginInstruction::Input(pid, input_bytes) => {
                                // FIXME: Set up an event subscription system, and timed callbacks
                                for (&id, (instance, plugin_env)) in &plugin_map {
                                    let handler = if PaneId::Plugin(id) == pid {
                                        "handle_key"
                                    } else {
                                        "handle_global_key"
                                    };
                                    let handler = instance.exports.get_function(handler).unwrap();
                                    for key in input_bytes.keys() {
                                        if let Ok(key) = key {
                                            wasi_write_string(
                                                &plugin_env.wasi_env,
                                                &serde_json::to_string(&key).unwrap(),
                                            );
                                            handler.call(&[]).unwrap();
                                        }
                                    }
                                }

                                send_screen_instructions
                                    .send(ScreenInstruction::Render)
                                    .unwrap();
                            }
                            PluginInstruction::Unload(pid) => drop(plugin_map.remove(&pid)),
                            PluginInstruction::Quit => break,
                        }
                    }
                }
            })
            .unwrap(),
    );

    // TODO: currently we don't push this into active_threads
    // because otherwise the app will hang. Need to fix this so it both
    // listens to the ipc-bus and is able to quit cleanly
    #[cfg(not(test))]
    let _ipc_thread = thread::Builder::new()
        .name("ipc_server".to_string())
        .spawn({
            use std::io::Read;
            let mut send_pty_instructions = send_pty_instructions.clone();
            let mut send_screen_instructions = send_screen_instructions.clone();
            move || {
                std::fs::remove_file(MOSAIC_IPC_PIPE).ok();
                let listener = std::os::unix::net::UnixListener::bind(MOSAIC_IPC_PIPE)
                    .expect("could not listen on ipc socket");
                let mut err_ctx = OPENCALLS.with(|ctx| *ctx.borrow());
                err_ctx.add_call(ContextType::IPCServer);
                send_pty_instructions.update(err_ctx);
                send_screen_instructions.update(err_ctx);

                for stream in listener.incoming() {
                    match stream {
                        Ok(mut stream) => {
                            let mut buffer = [0; 65535]; // TODO: more accurate
                            let _ = stream
                                .read(&mut buffer)
                                .expect("failed to parse ipc message");
                            let decoded: ApiCommand = bincode::deserialize(&buffer)
                                .expect("failed to deserialize ipc message");
                            match &decoded {
                                ApiCommand::OpenFile(file_name) => {
                                    let path = PathBuf::from(file_name);
                                    send_pty_instructions
                                        .send(PtyInstruction::SpawnTerminal(Some(path)))
                                        .unwrap();
                                }
                                ApiCommand::SplitHorizontally => {
                                    send_pty_instructions
                                        .send(PtyInstruction::SpawnTerminalHorizontally(None))
                                        .unwrap();
                                }
                                ApiCommand::SplitVertically => {
                                    send_pty_instructions
                                        .send(PtyInstruction::SpawnTerminalVertically(None))
                                        .unwrap();
                                }
                                ApiCommand::MoveFocus => {
                                    send_screen_instructions
                                        .send(ScreenInstruction::MoveFocus)
                                        .unwrap();
                                }
                            }
                        }
                        Err(err) => {
                            panic!("err {:?}", err);
                        }
                    }
                }
            }
        })
        .unwrap();

    let _stdin_thread = thread::Builder::new()
        .name("stdin_handler".to_string())
        .spawn({
            let send_screen_instructions = send_screen_instructions.clone();
            let send_pty_instructions = send_pty_instructions.clone();
            let os_input = os_input.clone();
            move || {
                input_loop(
                    os_input,
                    command_is_executing,
                    send_screen_instructions,
                    send_pty_instructions,
                    send_app_instructions,
                )
            }
        });

    #[warn(clippy::never_loop)]
    loop {
        let (app_instruction, mut err_ctx) = receive_app_instructions
            .recv()
            .expect("failed to receive app instruction on channel");

        err_ctx.add_call(ContextType::App(AppContext::from(&app_instruction)));
        send_screen_instructions.update(err_ctx);
        send_pty_instructions.update(err_ctx);
        match app_instruction {
            AppInstruction::Exit => {
                let _ = send_screen_instructions.send(ScreenInstruction::Quit);
                let _ = send_pty_instructions.send(PtyInstruction::Quit);

                let _ = send_plugin_instructions.send(PluginInstruction::Quit);
                break;
            }
            AppInstruction::Error(backtrace) => {
                let _ = send_screen_instructions.send(ScreenInstruction::Quit);
                let _ = send_pty_instructions.send(PtyInstruction::Quit);

                let _ = send_plugin_instructions.send(PluginInstruction::Quit);
                os_input.unset_raw_mode(0);
                let goto_start_of_last_line = format!("\u{1b}[{};{}H", full_screen_ws.rows, 1);
                let error = format!("{}\n{}", goto_start_of_last_line, backtrace);
                let _ = os_input
                    .get_stdout_writer()
                    .write(error.as_bytes())
                    .unwrap();
                for thread_handler in active_threads {
                    let _ = thread_handler.join();
                }
                std::process::exit(1);
            }
        }
    }

    for thread_handler in active_threads {
        thread_handler.join().unwrap();
    }
    // cleanup();
    let reset_style = "\u{1b}[m";
    let show_cursor = "\u{1b}[?25h";
    let goto_start_of_last_line = format!("\u{1b}[{};{}H", full_screen_ws.rows, 1);
    let goodbye_message = format!(
        "{}\n{}{}Bye from Mosaic!",
        goto_start_of_last_line, reset_style, show_cursor
    );

    os_input.unset_raw_mode(0);
    let _ = os_input
        .get_stdout_writer()
        .write(goodbye_message.as_bytes())
        .unwrap();
    os_input.get_stdout_writer().flush().unwrap();
}