darkfi/bin/map/src/main.rs

// select each connection and show log of traffic
// use rpc to get some info from the ircd network
// ircd::logger keeps track of network info
// map rpc polls logger for info about nodes, etc
use darkfi::{
    error::{Error, Result},
    rpc::{jsonrpc, jsonrpc::JsonResult},
    util::async_util,
};

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

use map::{node_info::NodeInfo, ui, App};

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 stdout = io::stdout().into_raw_mode()?;
    let backend = TermionBackend::new(stdout);
    let mut terminal = Terminal::new(backend)?;

    terminal.clear()?;

    // let current_state = get_current_state.await;
    // let mut app = app.lock();
    // app.current_state = current_state;
    let app = Arc::new(Mutex::new(App::new()));
    //let app = App::new();

    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())))
        // Run the main future on the current thread.
        .finish(|| {
            smol::future::block_on(async move {
                start(ex2.clone(), app.lock().await.clone()).await?;
                run_app(&mut terminal, app.lock().await.clone()).await?;
                drop(signal);
                Ok::<(), darkfi::Error>(())
            })
        });

    result
}

async fn start(ex: Arc<Executor<'_>>, app: App) -> Result<()> {
    let client = Map::new("tcp://127.0.0.1:8000".to_string());

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

    Ok(())
}

async fn poll(client: Map, app: App) -> Result<()> {
    loop {
        let reply = client.get_info().await?;
        update(app.clone(), reply).await?;
        async_util::sleep(1).await;
    }
}

async fn update(app: App, reply: Value) -> Result<()> {
    if reply.as_object().is_some() && !reply.as_object().unwrap().is_empty() {
        //let args = params.as_array();
        let nodes = reply.as_object().unwrap().get("nodes").unwrap();

        let node1 = &nodes[0];
        let node2 = &nodes[1];
        let node3 = &nodes[2];

        let infos = vec![
            NodeInfo {
                id: node1["id"].to_string(),
                connections: node1["connections"].as_u64().unwrap() as usize,
                is_active: node2["is_active"].as_bool().unwrap(),
                last_message: node3["message"].to_string(),
            },
            //NodeInfo {
            //    id: node2["id"].to_string(),
            //    connections: node2["connections"].as_u64().unwrap() as usize,
            //    is_active: node2["is_active"].as_bool().unwrap(),
            //    last_message: node2["message"].to_string(),
            //},
            //NodeInfo {
            //    id: node3["id"].to_string(),
            //    connections: node3["connections"].as_u64().unwrap() as usize,
            //    is_active: node3["is_active"].as_bool().unwrap(),
            //    last_message: node3["message"].to_string(),
            //},
        ];

        //app.node_info(
        //let node_info = NodeInfoView::new(infos.clone());

        //let ids = vec![node1["id"].to_string(), node2["id"].to_string(), node3["id"].to_string()];

        // mutex
        app.update(infos).await;
        //let node_list = NodeIdList::new(ids);
        //println!("{}", test);
        // do something
    } else {
        // TODO: error handling
        println!("Reply is an error");
    }

    Ok(())
}

async fn run_app<B: Backend>(terminal: &mut Terminal<B>, mut app: App) -> io::Result<()> {
    let mut asi = async_stdin();

    terminal.clear()?;

    app.node_list.state.select(Some(0));

    app.node_info.index = 0;

    // acquire the mutex
    // let mut app = app.lock();

    loop {
        terminal.draw(|f| ui::ui(f, &mut app))?;

        for k in asi.by_ref().keys() {
            match k.unwrap() {
                Key::Char('q') => {
                    terminal.clear()?;
                    return Ok(())
                }
                Key::Char('j') => {
                    app.node_list.next();
                    app.node_info.next();
                }
                Key::Char('k') => {
                    app.node_list.previous();
                    app.node_info.previous();
                }
                _ => (),
            }
        }
    }
}