forked from extern/nushell
66 lines
1.6 KiB
Rust
66 lines
1.6 KiB
Rust
use crate::format::RenderView;
|
|
use crate::object::Value;
|
|
use crate::Host;
|
|
use derive_new::new;
|
|
use prettytable::{Cell, Row, Table};
|
|
|
|
// An entries list is printed like this:
|
|
//
|
|
// name : ...
|
|
// name2 : ...
|
|
// another_name : ...
|
|
#[derive(new)]
|
|
pub struct TableView {
|
|
headers: Vec<String>,
|
|
entries: Vec<Vec<String>>,
|
|
}
|
|
|
|
impl TableView {
|
|
pub fn from_list(values: &[Value]) -> Option<TableView> {
|
|
if values.len() == 0 {
|
|
return None;
|
|
}
|
|
|
|
let item = &values[0];
|
|
let descs = item.data_descriptors();
|
|
|
|
let headers: Vec<String> = descs.iter().map(|d| d.name.clone()).collect();
|
|
|
|
let mut entries = vec![];
|
|
|
|
for value in values {
|
|
let row = descs
|
|
.iter()
|
|
.enumerate()
|
|
.map(|(i, d)| value.get_data(d).borrow().format_leaf(Some(&headers[i])))
|
|
.collect();
|
|
|
|
entries.push(row);
|
|
}
|
|
|
|
Some(TableView { headers, entries })
|
|
}
|
|
}
|
|
|
|
impl RenderView for TableView {
|
|
fn render_view(&self, host: &dyn Host) -> Vec<String> {
|
|
if self.entries.len() == 0 {
|
|
return vec![];
|
|
}
|
|
|
|
let mut table = Table::new();
|
|
let header: Vec<Cell> = self.headers.iter().map(|h| Cell::new(h)).collect();
|
|
|
|
table.add_row(Row::new(header));
|
|
|
|
for row in &self.entries {
|
|
table.add_row(Row::new(row.iter().map(|h| Cell::new(h)).collect()));
|
|
}
|
|
|
|
let mut out = vec![];
|
|
table.print(&mut out).unwrap();
|
|
|
|
vec![String::from_utf8_lossy(&out).to_string()]
|
|
}
|
|
}
|