nushell/src/format/table.rs

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()]
    }
}