#[macro_use] extern crate log; extern crate stderrlog; extern crate termion; extern crate tui; mod util; use std::io; use std::sync::mpsc; use std::thread; use std::time; use termion::event; use termion::input::TermRead; use tui::backend::MouseBackend; use tui::layout::{Direction, Group, Rect, Size}; use tui::style::{Color, Modifier, Style}; use tui::widgets::canvas::{Canvas, Line, Map, MapResolution}; use tui::widgets::{Axis, BarChart, Block, Borders, Chart, Dataset, Gauge, Item, List, Marker, Paragraph, Row, SelectableList, Sparkline, Table, Tabs, Widget}; use tui::Terminal; use util::*; struct Server<'a> { name: &'a str, location: &'a str, coords: (f64, f64), status: &'a str, } struct App<'a> { size: Rect, items: Vec<&'a str>, events: Vec<(&'a str, &'a str)>, selected: usize, tabs: MyTabs<'a>, show_chart: bool, progress: u16, data: Vec, data2: Vec<(f64, f64)>, data3: Vec<(f64, f64)>, data4: Vec<(&'a str, u64)>, window: [f64; 2], colors: [Color; 2], color_index: usize, servers: Vec>, } enum Event { Input(event::Key), Tick, } fn main() { stderrlog::new() .module(module_path!()) .verbosity(4) .init() .unwrap(); info!("Start"); let mut rand_signal = RandomSignal::new(0, 100); let mut sin_signal = SinSignal::new(0.2, 3.0, 18.0); let mut sin_signal2 = SinSignal::new(0.1, 2.0, 10.0); let mut app = App { size: Rect::default(), items: vec![ "Item1", "Item2", "Item3", "Item4", "Item5", "Item6", "Item7", "Item8", "Item9", "Item10", "Item11", "Item12", "Item13", "Item14", "Item15", "Item16", "Item17", "Item18", "Item19", "Item20", "Item21", "Item22", "Item23", "Item24", ], events: vec![ ("Event1", "INFO"), ("Event2", "INFO"), ("Event3", "CRITICAL"), ("Event4", "ERROR"), ("Event5", "INFO"), ("Event6", "INFO"), ("Event7", "WARNING"), ("Event8", "INFO"), ("Event9", "INFO"), ("Event10", "INFO"), ("Event11", "CRITICAL"), ("Event12", "INFO"), ("Event13", "INFO"), ("Event14", "INFO"), ("Event15", "INFO"), ("Event16", "INFO"), ("Event17", "ERROR"), ("Event18", "ERROR"), ("Event19", "INFO"), ("Event20", "INFO"), ("Event21", "WARNING"), ("Event22", "INFO"), ("Event23", "INFO"), ("Event24", "WARNING"), ("Event25", "INFO"), ("Event26", "INFO"), ], selected: 0, tabs: MyTabs { titles: vec!["Tab0", "Tab1"], selection: 0, }, show_chart: true, progress: 0, data: rand_signal.clone().take(300).collect(), data2: sin_signal.clone().take(100).collect(), data3: sin_signal2.clone().take(200).collect(), data4: vec![ ("B1", 9), ("B2", 12), ("B3", 5), ("B4", 8), ("B5", 2), ("B6", 4), ("B7", 5), ("B8", 9), ("B9", 14), ("B10", 15), ("B11", 1), ("B12", 0), ("B13", 4), ("B14", 6), ("B15", 4), ("B16", 6), ("B17", 4), ("B18", 7), ("B19", 13), ("B20", 8), ("B21", 11), ("B22", 9), ("B23", 3), ("B24", 5), ], window: [0.0, 20.0], colors: [Color::Magenta, Color::Red], color_index: 0, servers: vec![ Server { name: "NorthAmerica-1", location: "New York City", coords: (40.71, -74.00), status: "Up", }, Server { name: "Europe-1", location: "Paris", coords: (48.85, 2.35), status: "Failure", }, Server { name: "SouthAmerica-1", location: "São Paulo", coords: (-23.54, -46.62), status: "Up", }, Server { name: "Asia-1", location: "Singapore", coords: (1.35, 103.86), status: "Up", }, ], }; let (tx, rx) = mpsc::channel(); let input_tx = tx.clone(); for _ in 0..100 { sin_signal.next(); } for _ in 0..200 { sin_signal2.next(); } thread::spawn(move || { let stdin = io::stdin(); for c in stdin.keys() { let evt = c.unwrap(); input_tx.send(Event::Input(evt)).unwrap(); if evt == event::Key::Char('q') { break; } } }); thread::spawn(move || { let tx = tx.clone(); loop { tx.send(Event::Tick).unwrap(); thread::sleep(time::Duration::from_millis(200)); } }); let backend = MouseBackend::new().unwrap(); let mut terminal = Terminal::new(backend).unwrap(); terminal.clear().unwrap(); terminal.hide_cursor().unwrap(); loop { let size = terminal.size().unwrap(); if size != app.size { terminal.resize(size).unwrap(); app.size = size; } draw(&mut terminal, &app).unwrap(); let evt = rx.recv().unwrap(); match evt { Event::Input(input) => match input { event::Key::Char('q') => { break; } event::Key::Up => { if app.selected > 0 { app.selected -= 1 }; } event::Key::Down => if app.selected < app.items.len() - 1 { app.selected += 1; }, event::Key::Left => { app.tabs.previous(); } event::Key::Right => { app.tabs.next(); } event::Key::Char('t') => { app.show_chart = !app.show_chart; } _ => {} }, Event::Tick => { app.progress += 5; if app.progress > 100 { app.progress = 0; } app.data.insert(0, rand_signal.next().unwrap()); app.data.pop(); for _ in 0..5 { app.data2.remove(0); app.data2.push(sin_signal.next().unwrap()); } for _ in 0..10 { app.data3.remove(0); app.data3.push(sin_signal2.next().unwrap()); } let i = app.data4.pop().unwrap(); app.data4.insert(0, i); app.window[0] += 1.0; app.window[1] += 1.0; let i = app.events.pop().unwrap(); app.events.insert(0, i); app.color_index += 1; if app.color_index >= app.colors.len() { app.color_index = 0; } } } } terminal.show_cursor().unwrap(); terminal.clear().unwrap(); } fn draw(t: &mut Terminal, app: &App) -> Result<(), io::Error> { Group::default() .direction(Direction::Vertical) .sizes(&[Size::Fixed(3), Size::Min(0)]) .render(t, &app.size, |t, chunks| { Tabs::default() .block(Block::default().borders(Borders::ALL).title("Tabs")) .titles(&app.tabs.titles) .style(Style::default().fg(Color::Green)) .highlight_style(Style::default().fg(Color::Yellow)) .select(app.tabs.selection) .render(t, &chunks[0]); match app.tabs.selection { 0 => { draw_first_tab(t, app, &chunks[1]); } 1 => { draw_second_tab(t, app, &chunks[1]); } _ => {} }; }); try!(t.draw()); Ok(()) } fn draw_first_tab(t: &mut Terminal, app: &App, area: &Rect) { Group::default() .direction(Direction::Vertical) .sizes(&[Size::Fixed(7), Size::Min(7), Size::Fixed(7)]) .render(t, area, |t, chunks| { draw_gauges(t, app, &chunks[0]); draw_charts(t, app, &chunks[1]); draw_text(t, &chunks[2]); }); } fn draw_gauges(t: &mut Terminal, app: &App, area: &Rect) { Block::default() .borders(Borders::ALL) .title("Graphs") .render(t, area); Group::default() .direction(Direction::Vertical) .margin(1) .sizes(&[Size::Fixed(2), Size::Fixed(3)]) .render(t, area, |t, chunks| { Gauge::default() .block(Block::default().title("Gauge:")) .style( Style::default() .fg(Color::Magenta) .bg(Color::Black) .modifier(Modifier::Italic), ) .label(&format!("{} / 100", app.progress)) .percent(app.progress) .render(t, &chunks[0]); Sparkline::default() .block(Block::default().title("Sparkline:")) .style(Style::default().fg(Color::Green)) .data(&app.data) .render(t, &chunks[1]); }); } fn draw_charts(t: &mut Terminal, app: &App, area: &Rect) { let sizes = if app.show_chart { vec![Size::Percent(50), Size::Percent(50)] } else { vec![Size::Percent(100)] }; Group::default() .direction(Direction::Horizontal) .sizes(&sizes) .render(t, area, |t, chunks| { Group::default() .direction(Direction::Vertical) .sizes(&[Size::Percent(50), Size::Percent(50)]) .render(t, &chunks[0], |t, chunks| { Group::default() .direction(Direction::Horizontal) .sizes(&[Size::Percent(50), Size::Percent(50)]) .render(t, &chunks[0], |t, chunks| { SelectableList::default() .block(Block::default().borders(Borders::ALL).title("List")) .items(&app.items) .select(app.selected) .highlight_style( Style::default().fg(Color::Yellow).modifier(Modifier::Bold), ) .highlight_symbol(">") .render(t, &chunks[0]); let info_style = Style::default().fg(Color::White); let warning_style = Style::default().fg(Color::Yellow); let error_style = Style::default().fg(Color::Magenta); let critical_style = Style::default().fg(Color::Red); let events = app.events.iter().map(|&(evt, level)| { Item::StyledData( format!("{}: {}", level, evt), match level { "ERROR" => &error_style, "CRITICAL" => &critical_style, "WARNING" => &warning_style, _ => &info_style, }, ) }); List::new(events) .block(Block::default().borders(Borders::ALL).title("List")) .render(t, &chunks[1]); }); BarChart::default() .block(Block::default().borders(Borders::ALL).title("Bar chart")) .data(&app.data4) .bar_width(3) .bar_gap(2) .value_style( Style::default() .fg(Color::Black) .bg(Color::Green) .modifier(Modifier::Italic), ) .label_style(Style::default().fg(Color::Yellow)) .style(Style::default().fg(Color::Green)) .render(t, &chunks[1]); }); if app.show_chart { Chart::default() .block( Block::default() .title("Chart") .title_style(Style::default().fg(Color::Cyan).modifier(Modifier::Bold)) .borders(Borders::ALL), ) .x_axis( Axis::default() .title("X Axis") .style(Style::default().fg(Color::Gray)) .labels_style(Style::default().modifier(Modifier::Italic)) .bounds(app.window) .labels(&[ &format!("{}", app.window[0]), &format!("{}", (app.window[0] + app.window[1]) / 2.0), &format!("{}", app.window[1]), ]), ) .y_axis( Axis::default() .title("Y Axis") .style(Style::default().fg(Color::Gray)) .labels_style(Style::default().modifier(Modifier::Italic)) .bounds([-20.0, 20.0]) .labels(&["-20", "0", "20"]), ) .datasets(&[ Dataset::default() .name("data2") .marker(Marker::Dot) .style(Style::default().fg(Color::Cyan)) .data(&app.data2), Dataset::default() .name("data3") .marker(Marker::Braille) .style(Style::default().fg(Color::Yellow)) .data(&app.data3), ]) .render(t, &chunks[1]); } }); } fn draw_text(t: &mut Terminal, area: &Rect) { Paragraph::default() .block( Block::default() .borders(Borders::ALL) .title("Footer") .title_style(Style::default().fg(Color::Magenta).modifier(Modifier::Bold)), ) .wrap(true) .text( "This is a paragraph with several lines.\nYou can change the color.\nUse \ \\{fg=[color];bg=[color];mod=[modifier] [text]} to highlight the text with a color. \ For example, {fg=red u}{fg=green n}{fg=yellow d}{fg=magenta e}{fg=cyan r} \ {fg=gray t}{fg=light_gray h}{fg=light_red e} {fg=light_green r}{fg=light_yellow a} \ {fg=light_magenta i}{fg=light_cyan n}{fg=white b}{fg=red o}{fg=green w}.\n\ Oh, and if you didn't {mod=italic notice} you can {mod=bold automatically} \ {mod=invert wrap} your {mod=underline text} =).\nOne more thing is that \ it should display unicode characters properly: 日本国, ٩(-̮̮̃-̃)۶ ٩(●̮̮̃•̃)۶ ٩(͡๏̯͡๏)۶ \ ٩(-̮̮̃•̃).", ) .render(t, area); } fn draw_second_tab(t: &mut Terminal, app: &App, area: &Rect) { Group::default() .direction(Direction::Horizontal) .sizes(&[Size::Percent(30), Size::Percent(70)]) .render(t, area, |t, chunks| { let up_style = Style::default().fg(Color::Green); let failure_style = Style::default().fg(Color::Red); Table::new( ["Server", "Location", "Status"].into_iter(), app.servers.iter().map(|s| { let style = if s.status == "Up" { &up_style } else { &failure_style }; Row::StyledData(vec![s.name, s.location, s.status].into_iter(), style) }), ).block(Block::default().title("Servers").borders(Borders::ALL)) .header_style(Style::default().fg(Color::Yellow)) .widths(&[15, 15, 10]) .render(t, &chunks[0]); Canvas::default() .block(Block::default().title("World").borders(Borders::ALL)) .paint(|ctx| { ctx.draw(&Map { color: Color::White, resolution: MapResolution::High, }); ctx.layer(); for (i, s1) in app.servers.iter().enumerate() { for s2 in &app.servers[i + 1..] { ctx.draw(&Line { x1: s1.coords.1, y1: s1.coords.0, y2: s2.coords.0, x2: s2.coords.1, color: Color::Yellow, }); } } for server in &app.servers { let color = if server.status == "Up" { Color::Green } else { Color::Red }; ctx.print(server.coords.1, server.coords.0, "X", color); } }) .x_bounds([-180.0, 180.0]) .y_bounds([-90.0, 90.0]) .render(t, &chunks[1]); }) }