#![forbid(unsafe_code)]
mod pages;
use pages::*;
mod tabs;
use tabs::Tabs;
use snarkos_node::Node;
use snarkvm::prelude::Network;
use anyhow::Result;
use crossterm::{
event::{self, DisableMouseCapture, EnableMouseCapture, Event, KeyCode},
execute,
terminal::{EnterAlternateScreen, LeaveAlternateScreen, disable_raw_mode, enable_raw_mode},
};
use ratatui::{
Frame,
Terminal,
backend::{Backend, CrosstermBackend},
layout::{Constraint, Direction, Layout},
style::{Color, Modifier, Style},
text::{Line, Span},
widgets::{Block, Borders, Tabs as TabsTui},
};
use std::{
io,
thread,
time::{Duration, Instant},
};
use tokio::sync::mpsc::Receiver;
pub struct Display<N: Network> {
node: Node<N>,
tick_rate: Duration,
tabs: Tabs,
logs: Logs,
}
impl<N: Network> Display<N> {
pub fn start(node: Node<N>, log_receiver: Receiver<Vec<u8>>) -> Result<()> {
enable_raw_mode()?;
let mut stdout = io::stdout();
execute!(stdout, EnterAlternateScreen, EnableMouseCapture)?;
let backend = CrosstermBackend::new(stdout);
let mut terminal = Terminal::new(backend)?;
let mut display = Self {
node,
tick_rate: Duration::from_secs(1),
tabs: Tabs::new(PAGES.to_vec()),
logs: Logs::new(log_receiver),
};
let res = display.render(&mut terminal);
disable_raw_mode()?;
execute!(terminal.backend_mut(), LeaveAlternateScreen, DisableMouseCapture)?;
terminal.show_cursor()?;
if let Err(err) = res {
println!("{err:?}")
}
Ok(())
}
}
impl<N: Network> Display<N> {
fn render<B: Backend>(&mut self, terminal: &mut Terminal<B>) -> io::Result<()> {
let mut last_tick = Instant::now();
loop {
terminal.draw(|f| self.draw(f))?;
let timeout = self.tick_rate.checked_sub(last_tick.elapsed()).unwrap_or_else(|| Duration::from_secs(0));
if event::poll(timeout)? {
if let Event::Key(key) = event::read()? {
match key.code {
KeyCode::Esc => {
return Ok(());
}
KeyCode::Left => self.tabs.previous(),
KeyCode::Right => self.tabs.next(),
_ => {}
}
}
}
if last_tick.elapsed() >= self.tick_rate {
thread::sleep(Duration::from_millis(50));
last_tick = Instant::now();
}
}
}
fn draw(&mut self, f: &mut Frame) {
let chunks = Layout::default()
.margin(1)
.direction(Direction::Vertical)
.constraints([Constraint::Length(3), Constraint::Min(0)].as_ref())
.split(f.size());
let block = Block::default().style(Style::default().bg(Color::Black).fg(Color::White));
f.render_widget(block, f.size());
let titles = self
.tabs
.titles
.iter()
.map(|t| {
let (first, rest) = t.split_at(1);
Line::from(vec![
Span::styled(first, Style::default().fg(Color::Yellow)),
Span::styled(rest, Style::default().fg(Color::Green)),
])
})
.collect();
let tabs = TabsTui::new(titles)
.block(
Block::default()
.borders(Borders::ALL)
.title("Welcome to Aleo.")
.style(Style::default().add_modifier(Modifier::BOLD)),
)
.select(self.tabs.index)
.style(Style::default().fg(Color::Cyan))
.highlight_style(Style::default().add_modifier(Modifier::BOLD).bg(Color::White));
f.render_widget(tabs, chunks[0]);
match self.tabs.index {
0 => Overview.draw(f, chunks[1], &self.node),
1 => self.logs.draw(f, chunks[1]),
_ => unreachable!(),
};
}
}