veilid/veilid-cli/src/cursive_ui.rs

use crate::command_processor::*;
use crate::peers_table_view::*;
use crate::settings::Settings;
use crate::tools::*;
use crate::ui::*;
use crossbeam_channel::Sender;
use cursive::align::*;
use cursive::event::*;
use cursive::theme::*;
use cursive::traits::*;
use cursive::utils::markup::ansi;
use cursive::utils::markup::StyledString;
use cursive::view::SizeConstraint;
use cursive::views::*;
use cursive::Cursive;
use cursive::CursiveRunnable;
use flexi_logger::writers::LogWriter;
use flexi_logger::DeferredNow;
// use cursive_multiplex::*;
use crate::cached_text_view::*;
use chrono::{Datelike, Timelike};
use std::collections::{HashMap, VecDeque};
use std::io::Write;
use std::path::PathBuf;
use std::sync::atomic::{AtomicU64, Ordering};
use std::time::{SystemTime, UNIX_EPOCH};
use thiserror::Error;
//////////////////////////////////////////////////////////////
///
struct Dirty<T>
where
    T: PartialEq,
{
    value: T,
    dirty: bool,
}

impl<T> Dirty<T>
where
    T: PartialEq,
{
    pub fn new(value: T) -> Self {
        Self { value, dirty: true }
    }
    pub fn set(&mut self, value: T) {
        self.dirty = self.value != value;
        self.value = value;
    }
    pub fn get(&self) -> &T {
        &self.value
    }
    // pub fn get_mut(&mut self) -> &mut T {
    //     &mut self.value
    // }
    pub fn take_dirty(&mut self) -> bool {
        let is_dirty = self.dirty;
        self.dirty = false;
        is_dirty
    }
}

pub type NodeEventsPanel = Panel<NamedView<ScrollView<NamedView<CachedTextView>>>>;

static START_TIME: AtomicU64 = AtomicU64::new(0);

pub type CursiveUICallback = Box<dyn Fn(&mut Cursive) + Send>;

struct UIState {
    attachment_state: Dirty<String>,
    public_internet_ready: Dirty<bool>,
    local_network_ready: Dirty<bool>,
    network_started: Dirty<bool>,
    network_down_up: Dirty<(f32, f32)>,
    connection_state: Dirty<ConnectionState>,
    peers_state: Dirty<Vec<json::JsonValue>>,
    node_id: Dirty<String>,
}

impl UIState {
    pub fn new() -> Self {
        Self {
            attachment_state: Dirty::new("Detached".to_owned()),
            public_internet_ready: Dirty::new(false),
            local_network_ready: Dirty::new(false),
            network_started: Dirty::new(false),
            network_down_up: Dirty::new((0.0, 0.0)),
            connection_state: Dirty::new(ConnectionState::Disconnected),
            peers_state: Dirty::new(Vec::new()),
            node_id: Dirty::new("".to_owned()),
        }
    }
}

//#[derive(Error, Debug)]
//#[error("???")]
//struct UIError;

pub struct CursiveUIInner {
    ui_state: UIState,
    log_colors: HashMap<Level, cursive::theme::Color>,
    cmdproc: Option<CommandProcessor>,
    cmd_history: VecDeque<String>,
    cmd_history_position: usize,
    cmd_history_max_size: usize,
    connection_dialog_state: Option<ConnectionState>,
}

pub struct CursiveUI {
    siv: CursiveRunnable,
    inner: Arc<Mutex<CursiveUIInner>>,
}

#[derive(Error, Debug)]
pub enum DumbError {
    // #[error("{0}")]
    // Message(String),
}

impl CursiveUI {
    /////////////////////////////////////////////////////////////////////////////////////
    // Private functions
    fn command_processor(s: &mut Cursive) -> CommandProcessor {
        let inner = Self::inner(s);
        inner.cmdproc.as_ref().unwrap().clone()
    }

    fn inner(s: &mut Cursive) -> MutexGuard<'_, CursiveUIInner> {
        s.user_data::<Arc<Mutex<CursiveUIInner>>>().unwrap().lock()
    }
    fn inner_mut(s: &mut Cursive) -> MutexGuard<'_, CursiveUIInner> {
        s.user_data::<Arc<Mutex<CursiveUIInner>>>().unwrap().lock()
    }

    fn setup_colors(siv: &mut CursiveRunnable, inner: &mut CursiveUIInner, settings: &Settings) {
        // Make colors
        let mut theme = cursive::theme::load_default();
        theme.shadow = settings.interface.theme.shadow;
        theme.borders = BorderStyle::from(&settings.interface.theme.borders);
        theme.palette.set_color(
            "background",
            Color::parse(settings.interface.theme.colors.background.as_str()).unwrap(),
        );
        theme.palette.set_color(
            "shadow",
            Color::parse(settings.interface.theme.colors.shadow.as_str()).unwrap(),
        );
        theme.palette.set_color(
            "view",
            Color::parse(settings.interface.theme.colors.view.as_str()).unwrap(),
        );
        theme.palette.set_color(
            "primary",
            Color::parse(settings.interface.theme.colors.primary.as_str()).unwrap(),
        );
        theme.palette.set_color(
            "secondary",
            Color::parse(settings.interface.theme.colors.secondary.as_str()).unwrap(),
        );
        theme.palette.set_color(
            "tertiary",
            Color::parse(settings.interface.theme.colors.tertiary.as_str()).unwrap(),
        );
        theme.palette.set_color(
            "title_primary",
            Color::parse(settings.interface.theme.colors.title_primary.as_str()).unwrap(),
        );
        theme.palette.set_color(
            "title_secondary",
            Color::parse(settings.interface.theme.colors.title_secondary.as_str()).unwrap(),
        );
        theme.palette.set_color(
            "highlight",
            Color::parse(settings.interface.theme.colors.highlight.as_str()).unwrap(),
        );
        theme.palette.set_color(
            "highlight_inactive",
            Color::parse(settings.interface.theme.colors.highlight_inactive.as_str()).unwrap(),
        );
        theme.palette.set_color(
            "highlight_text",
            Color::parse(settings.interface.theme.colors.highlight_text.as_str()).unwrap(),
        );
        siv.set_theme(theme);

        // Make log colors
        let mut colors = HashMap::<Level, cursive::theme::Color>::new();
        colors.insert(
            Level::Trace,
            Color::parse(settings.interface.theme.log_colors.trace.as_str()).unwrap(),
        );
        colors.insert(
            Level::Debug,
            Color::parse(settings.interface.theme.log_colors.debug.as_str()).unwrap(),
        );
        colors.insert(
            Level::Info,
            Color::parse(settings.interface.theme.log_colors.info.as_str()).unwrap(),
        );
        colors.insert(
            Level::Warn,
            Color::parse(settings.interface.theme.log_colors.warn.as_str()).unwrap(),
        );
        colors.insert(
            Level::Error,
            Color::parse(settings.interface.theme.log_colors.error.as_str()).unwrap(),
        );
        inner.log_colors = colors;
    }
    fn setup_quit_handler(siv: &mut Cursive) {
        siv.clear_global_callbacks(cursive::event::Event::CtrlChar('c'));

        siv.set_on_pre_event(
            cursive::event::Event::CtrlChar('c'),
            CursiveUI::quit_handler,
        );
        siv.set_global_callback(
            cursive::event::Event::Key(Key::Esc),
            CursiveUI::quit_handler,
        );
    }

    fn setup_clear_handler(siv: &mut Cursive) {
        siv.clear_global_callbacks(cursive::event::Event::CtrlChar('k'));
        siv.set_on_pre_event(
            cursive::event::Event::CtrlChar('k'),
            CursiveUI::clear_handler,
        );
    }

    fn quit_handler(siv: &mut Cursive) {
        siv.add_layer(
            Dialog::text("Do you want to exit?")
                .button("Yes", |s| s.quit())
                .button("No", |s| {
                    s.pop_layer();
                    CursiveUI::setup_quit_handler(s);
                }),
        );
        siv.set_on_pre_event(cursive::event::Event::CtrlChar('c'), |s| {
            s.quit();
        });
        siv.set_global_callback(cursive::event::Event::Key(Key::Esc), |s| {
            s.pop_layer();
            CursiveUI::setup_quit_handler(s);
        });
    }
    fn clear_handler(siv: &mut Cursive) {
        Self::node_events_view(siv).set_content([]);
        CursiveUI::update_cb(siv);
    }

    ////////////////////////////////////////////////////////////////////////////////////////////////
    // Selectors

    // fn main_layout(s: &mut Cursive) -> ViewRef<LinearLayout> {
    //     s.find_name("main-layout").unwrap()
    // }
    fn node_events_panel(s: &mut Cursive) -> ViewRef<NodeEventsPanel> {
        s.find_name("node-events-panel").unwrap()
    }
    fn node_events_view(s: &mut Cursive) -> ViewRef<CachedTextView> {
        s.find_name("node-events-view").unwrap()
    }
    fn node_events_scroll_view(s: &mut Cursive) -> ViewRef<ScrollView<NamedView<CachedTextView>>> {
        s.find_name("node-events-scroll-view").unwrap()
    }
    fn command_line(s: &mut Cursive) -> ViewRef<EditView> {
        s.find_name("command-line").unwrap()
    }
    fn button_attach(s: &mut Cursive) -> ViewRef<Button> {
        s.find_name("button-attach").unwrap()
    }
    fn status_bar(s: &mut Cursive) -> ViewRef<TextView> {
        s.find_name("status-bar").unwrap()
    }
    fn peers(s: &mut Cursive) -> ViewRef<PeersTableView> {
        s.find_name("peers").unwrap()
    }
    fn ipc_path(s: &mut Cursive) -> ViewRef<EditView> {
        s.find_name("ipc-path").unwrap()
    }
    fn ipc_path_radio(s: &mut Cursive) -> ViewRef<RadioButton<u32>> {
        s.find_name("ipc-path-radio").unwrap()
    }
    fn connecting_text(s: &mut Cursive) -> ViewRef<TextView> {
        s.find_name("connecting-text").unwrap()
    }
    fn network_address(s: &mut Cursive) -> ViewRef<EditView> {
        s.find_name("network-address").unwrap()
    }
    fn network_address_radio(s: &mut Cursive) -> ViewRef<RadioButton<u32>> {
        s.find_name("network-address-radio").unwrap()
    }
    fn connection_dialog(s: &mut Cursive) -> ViewRef<Dialog> {
        s.find_name("connection-dialog").unwrap()
    }
    ////////////////////////////////////////////////////////////////////////////////////////////////

    fn push_styled_line(s: &mut Cursive, styled_string: StyledString) {
        let mut ctv = CursiveUI::node_events_view(s);
        ctv.append_line(styled_string)
    }

    fn push_ansi_lines(s: &mut Cursive, mut starting_style: Style, lines: String) {
        let mut ctv = CursiveUI::node_events_view(s);

        let mut sslines: Vec<StyledString> = vec![];
        for line in lines.lines() {
            let (spanned_string, end_style) = ansi::parse_with_starting_style(starting_style, line);
            sslines.push(spanned_string);
            starting_style = end_style;
        }

        ctv.append_lines(sslines.into_iter());
    }

    ////////////////////////////////////////////////////////////////////////////////////////////////

    fn render_attachment_state(inner: &mut CursiveUIInner) -> String {
        let att = match inner.ui_state.attachment_state.get().as_str() {
            "Detached" => "[----]",
            "Attaching" => "[/   ]",
            "AttachedWeak" => "[|   ]",
            "AttachedGood" => "[||  ]",
            "AttachedStrong" => "[||| ]",
            "FullyAttached" => "[||||]",
            "OverAttached" => "[++++]",
            "Detaching" => "[////]",
            _ => "[????]",
        };
        let pi = if *inner.ui_state.public_internet_ready.get() {
            "+P"
        } else {
            "-p"
        };
        let ln = if *inner.ui_state.local_network_ready.get() {
            "+L"
        } else {
            "-l"
        };
        format!("{}{}{}", att, pi, ln)
    }
    fn render_network_status(inner: &mut CursiveUIInner) -> String {
        match inner.ui_state.network_started.get() {
            false => "Down: ----KB/s Up: ----KB/s".to_owned(),
            true => {
                let (d, u) = inner.ui_state.network_down_up.get();
                format!("Down: {:.2}KB/s Up: {:.2}KB/s", d, u)
            }
        }
    }
    fn render_button_attach<'a>(inner: &mut CursiveUIInner) -> (&'a str, bool) {
        if let ConnectionState::ConnectedTCP(_, _) = inner.ui_state.connection_state.get() {
            match inner.ui_state.attachment_state.get().as_str() {
                "Detached" => ("Attach", true),
                "Attaching" => ("Detach", true),
                "AttachedWeak" => ("Detach", true),
                "AttachedGood" => ("Detach", true),
                "AttachedStrong" => ("Detach", true),
                "FullyAttached" => ("Detach", true),
                "OverAttached" => ("Detach", true),
                "Detaching" => ("Detach", false),
                _ => ("???", false),
            }
        } else {
            (" ---- ", false)
        }
    }

    fn on_command_line_edit(s: &mut Cursive, text: &str, _pos: usize) {
        let mut inner = Self::inner_mut(s);

        // save edited command to newest history slot
        let hlen = inner.cmd_history.len();
        inner.cmd_history_position = hlen - 1;
        inner.cmd_history[hlen - 1] = text.to_owned();
    }

    fn enable_command_ui(s: &mut Cursive, enabled: bool) {
        Self::command_line(s).set_enabled(enabled);
        Self::button_attach(s).set_enabled(enabled);
    }

    fn display_string_dialog_cb(
        s: &mut Cursive,
        title: String,
        contents: String,
        close_cb: Option<CursiveUICallback>,
    ) {
        // Creates a dialog around some text with a single button
        let close_cb = Rc::new(close_cb);
        let close_cb2 = close_cb.clone();
        s.add_layer(
            Dialog::around(TextView::new(contents).scrollable())
                .title(title)
                .button("Close", move |s| {
                    s.pop_layer();
                    if let Some(close_cb) = close_cb.as_ref() {
                        close_cb(s);
                    }
                }),
        );
        s.set_global_callback(cursive::event::Event::Key(Key::Esc), move |s| {
            s.set_global_callback(
                cursive::event::Event::Key(Key::Esc),
                CursiveUI::quit_handler,
            );
            s.pop_layer();
            if let Some(close_cb2) = close_cb2.as_ref() {
                close_cb2(s);
            }
        });
    }

    fn run_command(s: &mut Cursive, text: &str) -> Result<(), String> {
        // disable ui
        Self::enable_command_ui(s, false);

        // run command
        s.set_global_callback(cursive::event::Event::Key(Key::Esc), |s| {
            let cmdproc = Self::command_processor(s);
            cmdproc.cancel_command();
        });

        let cmdproc = Self::command_processor(s);
        cmdproc.run_command(
            text,
            UICallback::Cursive(Box::new(|s: &mut Cursive| {
                s.set_global_callback(
                    cursive::event::Event::Key(Key::Esc),
                    CursiveUI::quit_handler,
                );
                Self::enable_command_ui(s, true);
            })),
        )
    }

    fn on_command_line_entered(s: &mut Cursive, text: &str) {
        if text.trim().is_empty() {
            return;
        }
        // run command
        CursiveUI::push_ansi_lines(
            s,
            ColorStyle::primary().into(),
            format!("{}> {}\n", CursiveUI::cli_ts(Self::get_start_time()), text),
        );
        match Self::run_command(s, text) {
            Ok(_) => {}
            Err(e) => {
                let color = *Self::inner_mut(s).log_colors.get(&Level::Error).unwrap();
                CursiveUI::push_ansi_lines(
                    s,
                    color.into(),
                    format!(
                        " {} Error: {}",
                        CursiveUI::cli_ts(Self::get_start_time()),
                        e
                    ),
                );
            }
        }
        // save to history unless it's a duplicate
        {
            let mut inner = Self::inner_mut(s);

            let hlen = inner.cmd_history.len();
            inner.cmd_history[hlen - 1] = text.to_owned();

            if hlen >= 2 && inner.cmd_history[hlen - 1] == inner.cmd_history[hlen - 2] {
                inner.cmd_history[hlen - 1] = "".to_string();
            } else {
                if hlen == inner.cmd_history_max_size {
                    inner.cmd_history.pop_front();
                }
                inner.cmd_history.push_back("".to_string());
            }
            let hlen = inner.cmd_history.len();
            inner.cmd_history_position = hlen - 1;
        }

        // Clear the edit field
        let mut cmdline = Self::command_line(s);
        cmdline.set_content("");
    }

    fn on_command_line_history(s: &mut Cursive, dir: bool) {
        let mut cmdline = Self::command_line(s);
        let mut inner = Self::inner_mut(s);
        // if at top of buffer or end of buffer, ignore
        if (!dir && inner.cmd_history_position == 0)
            || (dir && inner.cmd_history_position == (inner.cmd_history.len() - 1))
        {
            return;
        }

        // move the history position
        if dir {
            inner.cmd_history_position += 1;
        } else {
            inner.cmd_history_position -= 1;
        }

        // replace text with current line
        let hlen = inner.cmd_history_position;
        cmdline.set_content(inner.cmd_history[hlen].as_str());
    }

    fn on_button_attach_pressed(s: &mut Cursive) {
        let action: Option<bool> = match Self::inner_mut(s).ui_state.attachment_state.get().as_str()
        {
            "Detached" => Some(true),
            "Attaching" => Some(false),
            "AttachedWeak" => Some(false),
            "AttachedGood" => Some(false),
            "AttachedStrong" => Some(false),
            "FullyAttached" => Some(false),
            "OverAttached" => Some(false),
            "Detaching" => None,
            _ => None,
        };
        let cmdproc = Self::command_processor(s);
        if let Some(a) = action {
            if a {
                cmdproc.attach();
            } else {
                cmdproc.detach();
            }
        }
    }

    fn refresh_button_attach(s: &mut Cursive) {
        let mut button_attach = CursiveUI::button_attach(s);
        let mut inner = Self::inner_mut(s);

        let (button_text, button_enable) = CursiveUI::render_button_attach(&mut inner);

        button_attach.set_label(button_text);
        button_attach.set_enabled(button_enable);
    }

    fn submit_ipc_path(s: &mut Cursive) {
        let edit = Self::ipc_path(s);
        let addr = (*edit.get_content()).clone();
        let ipc_path = match addr.parse::<PathBuf>() {
            Ok(sa) => sa,
            Err(_) => {
                s.add_layer(Dialog::text("Invalid IPC path").button("Close", |s| {
                    s.pop_layer();
                }));
                return;
            }
        };
        Self::command_processor(s).set_ipc_path(Some(ipc_path));
        Self::command_processor(s).set_network_address(None);
        Self::command_processor(s).start_connection();
    }

    fn submit_network_address(s: &mut Cursive) {
        let edit = Self::network_address(s);
        let addr = (*edit.get_content()).clone();
        let sa = match addr.parse::<std::net::SocketAddr>() {
            Ok(sa) => sa,
            Err(_) => {
                s.add_layer(
                    Dialog::text("Invalid network address").button("Close", |s| {
                        s.pop_layer();
                    }),
                );
                return;
            }
        };
        Self::command_processor(s).set_ipc_path(None);
        Self::command_processor(s).set_network_address(Some(sa));
        Self::command_processor(s).start_connection();
    }

    fn copy_to_clipboard_osc52<S: AsRef<str>>(s: &mut Cursive, text: S) {
        // OSC52 clipboard copy for terminals
        if std::io::stdout()
            .write_all(
                format!(
                    "\x1B]52;c;{}\x07",
                    data_encoding::BASE64.encode(text.as_ref().as_bytes()),
                )
                .as_bytes(),
            )
            .is_ok()
            && std::io::stdout().flush().is_ok()
        {
            let color = *Self::inner_mut(s).log_colors.get(&Level::Info).unwrap();
            CursiveUI::push_ansi_lines(
                s,
                color.into(),
                format!(
                    ">> {} Copied: {}",
                    CursiveUI::cli_ts(Self::get_start_time()),
                    text.as_ref()
                ),
            );
        }
    }

    fn copy_to_clipboard<S: AsRef<str>>(s: &mut Cursive, text: S) {
        if let Ok(mut clipboard) = arboard::Clipboard::new() {
            // X11/Wayland/other system copy
            if clipboard.set_text(text.as_ref()).is_ok() {
                let color = *Self::inner_mut(s).log_colors.get(&Level::Info).unwrap();
                CursiveUI::push_ansi_lines(
                    s,
                    color.into(),
                    format!(
                        ">> {} Copied: {}",
                        CursiveUI::cli_ts(Self::get_start_time()),
                        text.as_ref()
                    ),
                );
            } else {
                let color = *Self::inner_mut(s).log_colors.get(&Level::Warn).unwrap();
                CursiveUI::push_ansi_lines(
                    s,
                    color.into(),
                    format!(
                        ">> {} Could not copy to clipboard",
                        CursiveUI::cli_ts(Self::get_start_time())
                    ),
                );
            }
        } else {
            Self::copy_to_clipboard_osc52(s, text)
        }
    }

    fn on_submit_peers_table_view(s: &mut Cursive, _row: usize, index: usize) {
        let peers_table_view = CursiveUI::peers(s);
        let node_id = peers_table_view
            .borrow_item(index)
            .map(|j| j["node_ids"][0].to_string());
        if let Some(node_id) = node_id {
            Self::copy_to_clipboard(s, node_id);
        }
    }

    fn on_focus_peers_table_view<T>(ptv: &mut ResizedView<T>) -> EventResult {
        ptv.set_height(SizeConstraint::Full);
        EventResult::Ignored
    }

    fn on_focus_lost_peers_table_view<T>(ptv: &mut ResizedView<T>) -> EventResult {
        ptv.set_height(SizeConstraint::AtLeast(8));
        EventResult::Ignored
    }

    fn draw_connection_dialog(s: &mut Cursive, state: ConnectionState) -> bool {
        let is_ipc = Self::command_processor(s).get_ipc_path().is_some();
        let mut inner = Self::inner_mut(s);

        let mut connection_type_group: RadioGroup<u32> = RadioGroup::new().on_change(|s, v| {
            if *v == 0 {
                Self::ipc_path(s).enable();
                Self::network_address(s).disable();
            } else if *v == 1 {
                Self::ipc_path(s).disable();
                Self::network_address(s).enable();
            }
        });

        let mut show: bool = false;
        let mut hide: bool = false;
        let mut connecting: bool = false;
        match state {
            ConnectionState::Disconnected => {
                if inner.connection_dialog_state.is_none()
                    || inner
                        .connection_dialog_state
                        .as_ref()
                        .unwrap()
                        .is_connected()
                {
                    show = true;
                } else if inner
                    .connection_dialog_state
                    .as_ref()
                    .unwrap()
                    .is_retrying()
                {
                    hide = true;
                    show = true
                }
            }
            ConnectionState::ConnectedTCP(_, _) | ConnectionState::ConnectedIPC(_, _) => {
                if inner.connection_dialog_state.is_some()
                    && !inner
                        .connection_dialog_state
                        .as_ref()
                        .unwrap()
                        .is_connected()
                {
                    hide = true;
                }
            }
            ConnectionState::RetryingTCP(_, _) | ConnectionState::RetryingIPC(_, _) => {
                if inner.connection_dialog_state.is_none()
                    || inner
                        .connection_dialog_state
                        .as_ref()
                        .unwrap()
                        .is_connected()
                {
                    show = true;
                } else if inner
                    .connection_dialog_state
                    .as_ref()
                    .unwrap()
                    .is_disconnected()
                {
                    hide = true;
                    show = true;
                }
                connecting = true;
            }
        }
        inner.connection_dialog_state = Some(state);
        drop(inner);
        if hide {
            s.pop_layer();
            s.pop_layer();
        }
        if show {
            s.add_fullscreen_layer(Layer::with_color(
                ResizedView::with_full_screen(DummyView {}),
                ColorStyle::new(PaletteColor::Background, PaletteColor::Background),
            ));

            s.add_layer(
                Dialog::around(if connecting {
                    LinearLayout::vertical()
                        .child(TextView::new(" "))
                        .child(
                            TextView::new(if is_ipc {
                                "Connecting to IPC:"
                            } else {
                                "Connecting to TCP:"
                            })
                            .min_width(40),
                        )
                        .child(TextView::new("").with_name("connecting-text"))
                } else {
                    LinearLayout::vertical()
                        .child(TextView::new(" "))
                        .child(
                            if is_ipc {
                                connection_type_group.button(0, "IPC Path").selected()
                            } else {
                                connection_type_group.button(0, "IPC Path")
                            }
                            .with_name("ipc-path-radio"),
                        )
                        .child(
                            EditView::new()
                                .with_enabled(is_ipc)
                                .on_submit(|s, _| Self::submit_ipc_path(s))
                                .with_name("ipc-path")
                                .fixed_height(1)
                                .min_width(40),
                        )
                        .child(TextView::new(" "))
                        .child(
                            if is_ipc {
                                connection_type_group.button(1, "Network Address")
                            } else {
                                connection_type_group
                                    .button(1, "Network Address")
                                    .selected()
                            }
                            .with_name("network-address-radio"),
                        )
                        .child(
                            EditView::new()
                                .with_enabled(!is_ipc)
                                .on_submit(|s, _| Self::submit_network_address(s))
                                .with_name("network-address")
                                .fixed_height(1)
                                .min_width(40),
                        )
                        .child(TextView::new(" "))
                })
                .title(if connecting {
                    "Connecting to server..."
                } else {
                    "Connect to server"
                })
                .with_name("connection-dialog"),
            );

            return true;
        }

        false
    }

    fn refresh_connection_dialog(s: &mut Cursive) {
        let new_state = Self::inner(s).ui_state.connection_state.get().clone();

        if !Self::draw_connection_dialog(s, new_state.clone()) {
            return;
        }

        match new_state {
            ConnectionState::Disconnected => {
                Self::ipc_path_radio(s).set_enabled(true);
                Self::network_address_radio(s).set_enabled(true);

                let (network_address, network_address_enabled) =
                    match Self::command_processor(s).get_network_address() {
                        None => ("".to_owned(), false),
                        Some(addr) => (addr.to_string(), true),
                    };
                let mut edit = Self::network_address(s);
                edit.set_content(network_address);
                edit.set_enabled(network_address_enabled);

                let (ipc_path, ipc_path_enabled) = match Self::command_processor(s).get_ipc_path() {
                    None => ("".to_owned(), false),
                    Some(ipc_path) => (ipc_path.to_string_lossy().to_string(), true),
                };
                let mut edit = Self::ipc_path(s);
                edit.set_content(ipc_path);
                edit.set_enabled(ipc_path_enabled);

                let mut dlg = Self::connection_dialog(s);
                dlg.add_button("Connect", |s| {
                    if Self::ipc_path_radio(s).is_selected() {
                        Self::submit_ipc_path(s);
                    } else {
                        Self::submit_network_address(s);
                    }
                });
            }
            ConnectionState::ConnectedTCP(_, _) | ConnectionState::ConnectedIPC(_, _) => {}
            ConnectionState::RetryingTCP(addr, _) => {
                let mut text = Self::connecting_text(s);
                text.set_content(addr.to_string());

                let mut dlg = Self::connection_dialog(s);
                dlg.add_button("Cancel", |s| {
                    Self::command_processor(s).cancel_reconnect();
                });
            }
            ConnectionState::RetryingIPC(ipc_path, _) => {
                let mut text = Self::connecting_text(s);
                text.set_content(ipc_path.to_string_lossy().to_string());

                let mut dlg = Self::connection_dialog(s);
                dlg.add_button("Cancel", |s| {
                    Self::command_processor(s).cancel_reconnect();
                });
            }
        }
    }

    fn refresh_main_titlebar(s: &mut Cursive) {
        let mut main_window = CursiveUI::node_events_panel(s);
        let inner = Self::inner_mut(s);
        main_window.set_title(format!("Node: {}", inner.ui_state.node_id.get()));
    }

    fn refresh_statusbar(s: &mut Cursive) {
        let mut statusbar = CursiveUI::status_bar(s);

        let mut inner = Self::inner_mut(s);

        let mut status = StyledString::new();

        let mut enable_status_fields = false;

        match inner.ui_state.connection_state.get() {
            ConnectionState::Disconnected => {
                status.append_styled(
                    "Disconnected ".to_string(),
                    ColorStyle::highlight_inactive(),
                );
                status.append_styled("|", ColorStyle::highlight_inactive());
            }
            ConnectionState::RetryingTCP(addr, _) => {
                status.append_styled(
                    format!("Reconnecting to {} ", addr),
                    ColorStyle::highlight_inactive(),
                );
                status.append_styled("|", ColorStyle::highlight_inactive());
            }
            ConnectionState::RetryingIPC(path, _) => {
                status.append_styled(
                    format!(
                        "Reconnecting to IPC#{} ",
                        path.file_name()
                            .unwrap_or_default()
                            .to_string_lossy()
                            .into_owned()
                    ),
                    ColorStyle::highlight_inactive(),
                );
                status.append_styled("|", ColorStyle::highlight_inactive());
            }
            ConnectionState::ConnectedTCP(addr, _) => {
                status.append_styled(
                    format!("Connected to {} ", addr),
                    ColorStyle::highlight_inactive(),
                );
                enable_status_fields = true;
            }
            ConnectionState::ConnectedIPC(path, _) => {
                status.append_styled(
                    format!(
                        "Connected to IPC#{} ",
                        path.file_name()
                            .unwrap_or_default()
                            .to_string_lossy()
                            .into_owned()
                    ),
                    ColorStyle::highlight_inactive(),
                );
                enable_status_fields = true;
            }
        }
        if enable_status_fields {
            status.append_styled("|", ColorStyle::highlight_inactive());
            // Add attachment state
            status.append_styled(
                format!(" {} ", CursiveUI::render_attachment_state(&mut inner)),
                ColorStyle::highlight_inactive(),
            );
            status.append_styled("|", ColorStyle::highlight_inactive());
            // Add bandwidth status
            status.append_styled(
                format!(" {} ", CursiveUI::render_network_status(&mut inner)),
                ColorStyle::highlight_inactive(),
            );
            status.append_styled("|", ColorStyle::highlight_inactive());
            // Add tunnel status
            status.append_styled(" No Tunnels ", ColorStyle::highlight_inactive());
            status.append_styled("|", ColorStyle::highlight_inactive());
        };

        statusbar.set_content(status);
    }

    fn refresh_peers(s: &mut Cursive) {
        let mut peers = CursiveUI::peers(s);
        let inner = Self::inner_mut(s);
        let sel_item = peers.item();
        let sel_item_text = peers
            .item()
            .map(|x| peers.borrow_items()[x]["node_ids"][0].clone());

        peers.set_items_stable(inner.ui_state.peers_state.get().clone());

        let mut selected = false;
        if let Some(sel_item_text) = sel_item_text {
            // First select by name
            for n in 0..peers.borrow_items().len() {
                if peers.borrow_items()[n]["node_ids"][0] == sel_item_text {
                    peers.set_selected_item(n);
                    selected = true;
                }
            }
        }
        if !selected {
            if let Some(sel_item) = sel_item {
                peers.set_selected_item(sel_item);
            }
        }
    }

    fn update_cb(s: &mut Cursive) {
        let mut inner = Self::inner_mut(s);

        let mut refresh_statusbar = false;
        let mut refresh_button_attach = false;
        let mut refresh_connection_dialog = false;
        let mut refresh_peers = false;
        let mut refresh_main_titlebar = false;
        if inner.ui_state.attachment_state.take_dirty() {
            refresh_statusbar = true;
            refresh_button_attach = true;
            refresh_peers = true;
        }
        if inner.ui_state.network_started.take_dirty() {
            refresh_statusbar = true;
        }
        if inner.ui_state.network_down_up.take_dirty() {
            refresh_statusbar = true;
        }
        if inner.ui_state.connection_state.take_dirty() {
            refresh_statusbar = true;
            refresh_button_attach = true;
            refresh_connection_dialog = true;
            refresh_peers = true;
        }
        if inner.ui_state.peers_state.take_dirty() {
            refresh_peers = true;
        }
        if inner.ui_state.node_id.take_dirty() {
            refresh_main_titlebar = true;
        }

        drop(inner);

        if refresh_statusbar {
            Self::refresh_statusbar(s);
        }
        if refresh_button_attach {
            Self::refresh_button_attach(s);
        }
        if refresh_connection_dialog {
            Self::refresh_connection_dialog(s);
        }
        if refresh_peers {
            Self::refresh_peers(s);
        }
        if refresh_main_titlebar {
            Self::refresh_main_titlebar(s);
        }
    }

    ////////////////////////////////////////////////////////////////////////////
    // Public functions

    #[allow(clippy::format_in_format_args)]
    pub fn cli_ts(ts: u64) -> String {
        let now = chrono::DateTime::<chrono::Utc>::from(SystemTime::now());
        let date = chrono::DateTime::<chrono::Utc>::from(UNIX_EPOCH + Duration::from_micros(ts));

        let show_year = now.year() != date.year();
        let show_month = show_year || now.month() != date.month();
        let show_date = show_month || now.day() != date.day();

        if show_year || show_month || show_date {
            CursiveUI::set_start_time();
        }
        format!(
            "{}{}",
            if show_year || show_month || show_date {
                format!(
                    "Day changed: {:04}/{:02}/{:02} \n",
                    now.year(),
                    now.month(),
                    now.day()
                )
            } else {
                "".to_owned()
            },
            format!("{:02}:{:02}:{:02}", now.hour(), now.minute(), now.second())
        )
    }

    pub fn set_start_time() {
        START_TIME.store(get_timestamp(), Ordering::Relaxed)
    }

    pub fn get_start_time() -> u64 {
        START_TIME.load(Ordering::Relaxed)
    }

    fn make_node_events_panel(
        node_log_scrollback: usize,
    ) -> ResizedView<NamedView<NodeEventsPanel>> {
        Panel::new(
            CachedTextView::new([], node_log_scrollback, Some(node_log_scrollback))
                .with_name("node-events-view")
                .scrollable()
                .scroll_strategy(cursive::view::ScrollStrategy::StickToBottom)
                .on_scroll(|s, _r| {
                    let mut sv = CursiveUI::node_events_scroll_view(s);
                    if sv.is_at_bottom() {
                        sv.set_scroll_strategy(cursive::view::ScrollStrategy::StickToBottom);
                    }
                })
                .with_name("node-events-scroll-view"),
        )
        .title_position(HAlign::Left)
        .title("Node Events")
        .with_name("node-events-panel")
        .full_screen()
    }

    pub fn new(settings: &Settings) -> (Self, CursiveUISender) {
        CursiveUI::set_start_time();
        // Instantiate the cursive runnable
        let runnable = CursiveRunnable::new(
            || -> Result<Box<dyn cursive::backend::Backend>, Box<DumbError>> {
                let backend = cursive::backends::crossterm::Backend::init().unwrap();
                let buffered_backend = cursive_buffered_backend::BufferedBackend::new(backend);
                Ok(Box::new(buffered_backend))
            },
        );

        // Make the callback mechanism easily reachable
        let cb_sink = runnable.cb_sink().clone();

        // Create the UI object
        let mut this = Self {
            siv: runnable,
            inner: Arc::new(Mutex::new(CursiveUIInner {
                ui_state: UIState::new(),
                log_colors: Default::default(),
                cmdproc: None,
                cmd_history: {
                    let mut vd = VecDeque::new();
                    vd.push_back("".to_string());
                    vd
                },
                cmd_history_position: 0,
                cmd_history_max_size: settings.interface.command_line.history_size,
                connection_dialog_state: None,
            })),
        };

        let ui_sender = CursiveUISender {
            inner: this.inner.clone(),
            cb_sink,
            colors: default_log_colors(),
        };

        let mut inner = this.inner.lock();

        // Make the inner object accessible in callbacks easily
        this.siv.set_user_data(this.inner.clone());

        // Create layouts
        let node_events_view = Self::make_node_events_panel(settings.interface.node_log.scrollback);
        let mut peers_table_view = PeersTableView::new()
            .column(PeerTableColumn::NodeId, "Node Id", |c| c.width(48))
            .column(PeerTableColumn::Address, "Address", |c| c)
            .column(PeerTableColumn::LatencyAvg, "Ping", |c| c.width(8))
            .column(PeerTableColumn::TransferDownAvg, "Down", |c| c.width(8))
            .column(PeerTableColumn::TransferUpAvg, "Up", |c| c.width(8));
        peers_table_view.set_on_submit(CursiveUI::on_submit_peers_table_view);
        let peers_table_view = FocusTracker::new(ResizedView::new(
            SizeConstraint::Full,
            SizeConstraint::AtLeast(8),
            peers_table_view.with_name("peers"),
        ))
        .on_focus(CursiveUI::on_focus_peers_table_view)
        .on_focus_lost(CursiveUI::on_focus_lost_peers_table_view);

        // attempt at using Mux. Mux has bugs, like resizing problems.
        // let mut mux = Mux::new();
        // let node_node_events_view = mux
        //     .add_below(node_events_view, mux.root().build().unwrap())
        //     .unwrap();
        // let node_peers_table_view = mux
        //     .add_below(peers_table_view, node_node_events_view)
        //     .unwrap();
        // mux.set_container_split_ratio(node_peers_table_view, 0.75)
        //     .unwrap();
        // let mut mainlayout = LinearLayout::vertical();
        // mainlayout.add_child(mux);

        // Back to fixed layout
        let mut mainlayout = LinearLayout::vertical();
        mainlayout.add_child(node_events_view);
        mainlayout.add_child(peers_table_view);
        // ^^^ fixed layout

        let mut command = StyledString::new();
        command.append_styled("Command> ", ColorStyle::title_primary());
        //
        mainlayout.add_child(
            LinearLayout::horizontal()
                .child(TextView::new(command))
                .child(
                    EditView::new()
                        .on_submit(CursiveUI::on_command_line_entered)
                        .on_edit(CursiveUI::on_command_line_edit)
                        .on_up_down(CursiveUI::on_command_line_history)
                        .style(ColorStyle::new(
                            PaletteColor::Background,
                            PaletteColor::Secondary,
                        ))
                        .with_name("command-line")
                        .full_screen()
                        .fixed_height(1),
                )
                .child(
                    Button::new("Attach", |s| {
                        CursiveUI::on_button_attach_pressed(s);
                    })
                    .with_name("button-attach"),
                ),
        );
        let mut version = StyledString::new();
        version.append_styled(
            concat!(" | veilid-cli v", env!("CARGO_PKG_VERSION")),
            ColorStyle::highlight_inactive(),
        );

        mainlayout.add_child(
            LinearLayout::horizontal()
                .color(Some(ColorStyle::highlight_inactive()))
                .child(
                    TextView::new("")
                        .with_name("status-bar")
                        .full_screen()
                        .fixed_height(1),
                )
                .child(TextView::new(version)),
        );

        this.siv
            .add_fullscreen_layer(mainlayout.with_name("main-layout"));

        CursiveUI::setup_colors(&mut this.siv, &mut inner, settings);
        CursiveUI::setup_quit_handler(&mut this.siv);
        CursiveUI::setup_clear_handler(&mut this.siv);

        drop(inner);

        (this, ui_sender)
    }
}

impl UI for CursiveUI {
    fn set_command_processor(&mut self, cmdproc: CommandProcessor) {
        let mut inner = self.inner.lock();
        inner.cmdproc = Some(cmdproc);
    }
    // Note: Cursive is not re-entrant, can't borrow_mut self.siv again after this
    fn run_async<'a>(&'a mut self) -> Pin<Box<dyn core::future::Future<Output = ()> + 'a>> {
        Box::pin(async {
            self.siv.run_async().await;
        })
    }
}

type CallbackSink = Box<dyn FnOnce(&mut Cursive) + 'static + Send>;

/// Default log colors
fn default_log_colors() -> HashMap<Level, Color> {
    let mut colors = HashMap::<Level, Color>::new();
    colors.insert(Level::Trace, Color::Dark(BaseColor::Green));
    colors.insert(Level::Debug, Color::Dark(BaseColor::Cyan));
    colors.insert(Level::Info, Color::Dark(BaseColor::Blue));
    colors.insert(Level::Warn, Color::Dark(BaseColor::Yellow));
    colors.insert(Level::Error, Color::Dark(BaseColor::Red));
    colors
}

#[derive(Clone)]
pub struct CursiveUISender {
    inner: Arc<Mutex<CursiveUIInner>>,
    cb_sink: Sender<CallbackSink>,
    colors: HashMap<Level, Color>,
}
impl UISender for CursiveUISender {
    fn clone_uisender(&self) -> Box<dyn UISender> {
        Box::new(self.clone())
    }
    fn as_logwriter(&self) -> Option<Box<dyn LogWriter>> {
        Some(Box::new(self.clone()) as Box<dyn LogWriter>)
    }

    fn display_string_dialog(&self, title: &str, text: &str, close_cb: UICallback) {
        let title = title.to_string();
        let text = text.to_string();
        let _ = self.cb_sink.send(Box::new(move |s| {
            CursiveUI::display_string_dialog_cb(
                s,
                title,
                text,
                if let UICallback::Cursive(close_cb) = close_cb {
                    Some(close_cb)
                } else {
                    None
                },
            )
        }));
    }

    fn quit(&self) {
        let _ = self.cb_sink.send(Box::new(|s| {
            s.quit();
        }));
    }

    fn send_callback(&self, callback: UICallback) {
        if let UICallback::Cursive(callback) = callback {
            let _ = self.cb_sink.send(Box::new(move |s| callback(s)));
        }
    }
    fn set_attachment_state(
        &mut self,
        state: &str,
        public_internet_ready: bool,
        local_network_ready: bool,
    ) {
        {
            let mut inner = self.inner.lock();
            inner.ui_state.attachment_state.set(state.to_string());
            inner
                .ui_state
                .public_internet_ready
                .set(public_internet_ready);
            inner.ui_state.local_network_ready.set(local_network_ready);
        }

        let _ = self.cb_sink.send(Box::new(CursiveUI::update_cb));
    }
    fn set_network_status(
        &mut self,
        started: bool,
        bps_down: u64,
        bps_up: u64,
        mut peers: Vec<json::JsonValue>,
    ) {
        {
            let mut inner = self.inner.lock();
            inner.ui_state.network_started.set(started);
            inner.ui_state.network_down_up.set((
                ((bps_down as f64) / 1000.0f64) as f32,
                ((bps_up as f64) / 1000.0f64) as f32,
            ));
            peers.sort_by(|a, b| {
                a["node_ids"][0]
                    .to_string()
                    .cmp(&b["node_ids"][0].to_string())
            });
            inner.ui_state.peers_state.set(peers);
        }
        let _ = self.cb_sink.send(Box::new(CursiveUI::update_cb));
    }
    fn set_config(&mut self, config: &json::JsonValue) {
        let mut inner = self.inner.lock();

        let node_ids = &config["network"]["routing_table"]["node_id"];

        let mut node_id_str = String::new();
        for l in 0..node_ids.len() {
            let nid = &node_ids[l];
            if !node_id_str.is_empty() {
                node_id_str.push(' ');
            }
            node_id_str.push_str(nid.to_string().as_ref());
        }

        inner.ui_state.node_id.set(node_id_str);
    }
    fn set_connection_state(&mut self, state: ConnectionState) {
        {
            let mut inner = self.inner.lock();
            inner.ui_state.connection_state.set(state);
        }
        let _ = self.cb_sink.send(Box::new(CursiveUI::update_cb));
    }

    fn add_node_event(&self, log_color: Level, event: &str) {
        let color = {
            let inner = self.inner.lock();
            *inner.log_colors.get(&log_color).unwrap()
        };

        let _ = self.push_styled_lines(
            color.into(),
            format!(
                "{}: {}\n",
                CursiveUI::cli_ts(CursiveUI::get_start_time()),
                event
            ),
        );
    }
}
impl CursiveUISender {
    pub fn push_styled(&self, styled_string: StyledString) -> std::io::Result<()> {
        let res = self.cb_sink.send(Box::new(move |s| {
            CursiveUI::push_styled_line(s, styled_string);
        }));
        if res.is_err() {
            return Err(std::io::Error::from(std::io::ErrorKind::BrokenPipe));
        }
        Ok(())
    }

    pub fn push_styled_lines(&self, starting_style: Style, lines: String) -> std::io::Result<()> {
        let res = self.cb_sink.send(Box::new(move |s| {
            CursiveUI::push_ansi_lines(s, starting_style, lines);
        }));
        if res.is_err() {
            return Err(std::io::Error::from(std::io::ErrorKind::BrokenPipe));
        }
        Ok(())
    }
}
impl LogWriter for CursiveUISender {
    fn write(&self, _now: &mut DeferredNow, record: &Record) -> std::io::Result<()> {
        let color = *self.colors.get(&record.level()).unwrap();

        let args = format!("{}", &record.args());

        let mut line = StyledString::new();
        let mut indent = 0;
        let levstr = format!("{}: ", record.level());
        indent += levstr.len();
        line.append_styled(levstr, color);
        let filestr = format!(
            "{}:{} ",
            record.file().unwrap_or("(unnamed)"),
            record.line().unwrap_or(0),
        );
        indent += filestr.len();
        line.append_plain(filestr);

        for argline in args.lines() {
            line.append_styled(argline, color);
            line.append_plain("\n");
            self.push_styled(line)?;
            line = StyledString::new();
            line.append_plain(" ".repeat(indent));
        }
        Ok(())
    }

    fn flush(&self) -> std::io::Result<()> {
        // we are not buffering
        Ok(())
    }

    fn max_log_level(&self) -> log::LevelFilter {
        log::LevelFilter::max()
    }
}