darkfi/bin/map/src/main.rs

use darkfi::{
    cli::{
        cli_config::{log_config, spawn_config},
        Config, MapConfig,
    },
    error::{Error, Result},
    rpc::{jsonrpc, jsonrpc::JsonResult},
    util::{async_util, join_config_path},
};

use async_std::sync::Arc;
use easy_parallel::Parallel;
use log::debug;
use serde_json::{json, Value};
use smol::Executor;
use std::{io, io::Read, path::PathBuf};
use termion::{async_stdin, event::Key, input::TermRead, raw::IntoRawMode};
use tui::{
    backend::{Backend, TermionBackend},
    Terminal,
};

use map::{
    model::{Connection, IdList, InfoList, NodeInfo},
    ui,
    view::{IdListView, InfoListView},
    Model, View,
};

const CONFIG_FILE_CONTENTS: &[u8] = include_bytes!("../map_config.toml");

struct Map {
    url: String,
}

impl Map {
    pub fn new(url: String) -> Self {
        Self { url }
    }

    async fn request(&self, r: jsonrpc::JsonRequest) -> Result<Value> {
        let reply: JsonResult = match jsonrpc::send_request(&self.url, json!(r)).await {
            Ok(v) => v,
            Err(e) => return Err(e),
        };

        match reply {
            JsonResult::Resp(r) => {
                debug!(target: "RPC", "<-- {}", serde_json::to_string(&r)?);
                Ok(r.result)
            }

            JsonResult::Err(e) => {
                debug!(target: "RPC", "<-- {}", serde_json::to_string(&e)?);
                Err(Error::JsonRpcError(e.error.message.to_string()))
            }

            JsonResult::Notif(n) => {
                debug!(target: "RPC", "<-- {}", serde_json::to_string(&n)?);
                Err(Error::JsonRpcError("Unexpected reply".to_string()))
            }
        }
    }

    // --> {"jsonrpc": "2.0", "method": "say_hello", "params": [], "id": 42}
    // <-- {"jsonrpc": "2.0", "result": "hello world", "id": 42}
    async fn _say_hello(&self) -> Result<Value> {
        let req = jsonrpc::request(json!("say_hello"), json!([]));
        Ok(self.request(req).await?)
    }

    //--> {"jsonrpc": "2.0", "method": "poll", "params": [], "id": 42}
    // <-- {"jsonrpc": "2.0", "result": {"nodeID": [], "nodeinfo" [], "id": 42}
    async fn get_info(&self) -> Result<Value> {
        let req = jsonrpc::request(json!("get_info"), json!([]));
        Ok(self.request(req).await?)
    }
}

#[async_std::main]
async fn main() -> Result<()> {
    let config_path = join_config_path(&PathBuf::from("map_config.toml"))?;

    spawn_config(&config_path, CONFIG_FILE_CONTENTS)?;

    let config = Config::<MapConfig>::load(config_path)?;

    let stdout = io::stdout().into_raw_mode()?;
    let backend = TermionBackend::new(stdout);
    let mut terminal = Terminal::new(backend)?;

    terminal.clear()?;

    let infos = Vec::new();
    let info_list = InfoList::new(infos.clone());
    let ids = Vec::new();
    let id_list = IdList::new(ids);

    let model = Arc::new(Model::new(id_list, info_list));

    let nthreads = num_cpus::get();
    let (signal, shutdown) = async_channel::unbounded::<()>();

    let ex = Arc::new(Executor::new());
    let ex2 = ex.clone();

    let (_, result) = Parallel::new()
        .each(0..nthreads, |_| smol::future::block_on(ex.run(shutdown.recv())))
        .finish(|| {
            smol::future::block_on(async move {
                run_rpc(&config, ex2.clone(), model.clone()).await?;
                render(&config, &mut terminal, model.clone()).await?;
                drop(signal);
                Ok::<(), darkfi::Error>(())
            })
        });

    result
}

async fn run_rpc(config: &MapConfig, ex: Arc<Executor<'_>>, model: Arc<Model>) -> Result<()> {
    // TODO: listen to multiple nodes
    let client = Map::new(config.nodes[0].node_id.to_string());

    ex.spawn(poll(client, model)).detach();

    Ok(())
}

async fn poll(client: Map, model: Arc<Model>) -> Result<()> {
    loop {
        let reply = client.get_info().await?;

        if reply.as_object().is_some() && !reply.as_object().unwrap().is_empty() {
            let id = reply.as_object().unwrap().get("id").unwrap();

            let connections = reply.as_object().unwrap().get("connections").unwrap();
            let outgoing = connections.get("outgoing").unwrap();
            let incoming = connections.get("incoming").unwrap();

            let mut outconnects = Vec::new();
            let mut inconnects = Vec::new();

            let out0 = Connection::new(
                outgoing[0].get("id").unwrap().as_str().unwrap().to_string(),
                outgoing[0].get("message").unwrap().as_str().unwrap().to_string(),
            );
            let out1 = Connection::new(
                outgoing[1].get("id").unwrap().as_str().unwrap().to_string(),
                outgoing[1].get("message").unwrap().as_str().unwrap().to_string(),
            );

            let in0 = Connection::new(
                incoming[0].get("id").unwrap().as_str().unwrap().to_string(),
                incoming[0].get("message").unwrap().as_str().unwrap().to_string(),
            );
            let in1 = Connection::new(
                incoming[1].get("id").unwrap().as_str().unwrap().to_string(),
                incoming[1].get("message").unwrap().as_str().unwrap().to_string(),
            );

            outconnects.push(out0);
            outconnects.push(out1);

            inconnects.push(in0);
            inconnects.push(in1);

            let infos = vec![NodeInfo {
                // TODO: should never crash
                id: id.as_str().unwrap().to_string(),
                outgoing: outconnects,
                incoming: inconnects,
            }];

            for node in infos {
                // update node info if we don't have it already
                if !model.id_list.node_id.lock().await.contains(&node.clone().id) {
                    model.id_list.node_id.lock().await.push(node.clone().id);
                    // TODO: update new info if we don't have
                    model.info_list.infos.lock().await.push(node.clone());
                }
            }
        } else {
            // TODO: error handling
            println!("Reply is an error");
        }

        async_util::sleep(2).await;
    }
}

async fn render<B: Backend>(
    _config: &MapConfig,
    terminal: &mut Terminal<B>,
    model: Arc<Model>,
) -> io::Result<()> {
    let mut asi = async_stdin();

    terminal.clear()?;

    let id_list = IdListView::new(Vec::new());
    let info_list = InfoListView::new(Vec::new());
    let mut view = View::new(id_list.clone(), info_list.clone());

    view.id_list.state.select(Some(1));
    view.info_list.index = 0;

    loop {
        // on first run, add available nodes
        // every time run the program, simply update nodes
        let mut view = view.clone();
        view.update(
            model.id_list.node_id.lock().await.clone(),
            model.info_list.infos.lock().await.clone(),
        );
        if view.info_list.infos.is_empty() {
            // TODO: make this a loading widget
            println!("Initializing...");
            async_util::sleep(1).await;
            terminal.clear()?;
        } else {
            terminal.draw(|f| {
                ui::ui(f, view.clone());
            })?;
        }
        for k in asi.by_ref().keys() {
            match k.unwrap() {
                Key::Char('q') => {
                    terminal.clear()?;
                    return Ok(());
                }
                Key::Char('j') => {
                    view.id_list.next();
                    view.info_list.next().await;
                }
                Key::Char('k') => {
                    view.id_list.previous();
                    view.info_list.previous().await;
                }
                _ => (),
            }
        }
    }
}