osp-cli 1.5.1

CLI and REPL for querying and managing OSP infrastructure data
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96

use unicode_width::UnicodeWidthStr;

use crate::ui::doc::TableBlock;

#[derive(Debug, Clone, PartialEq, Eq)]
pub(super) struct PreparedTable {
    pub headers: Vec<PreparedCell>,
    pub rows: Vec<Vec<PreparedCell>>,
    pub widths: Vec<usize>,
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub(super) struct PreparedCell {
    pub raw: String,
    pub markdown: String,
    pub width: usize,
}

impl PreparedTable {
    pub(super) fn for_terminal(block: &TableBlock) -> Self {
        Self::prepare(block, 0)
    }

    pub(super) fn for_markdown(block: &TableBlock) -> Self {
        Self::prepare(block, 3)
    }

    fn prepare(block: &TableBlock, min_cell_width: usize) -> Self {
        let headers = prepare_cells(&block.headers);
        let rows = block
            .rows
            .iter()
            .map(|row| prepare_cells(row))
            .collect::<Vec<_>>();
        let mut widths = headers
            .iter()
            .map(|cell| cell.width.max(min_cell_width))
            .collect::<Vec<_>>();

        for row in &rows {
            for (index, cell) in row.iter().enumerate() {
                if let Some(width) = widths.get_mut(index) {
                    *width = (*width).max(cell.width.max(min_cell_width));
                }
            }
        }

        Self {
            headers,
            rows,
            widths,
        }
    }
}

fn prepare_cells(cells: &[String]) -> Vec<PreparedCell> {
    cells.iter().map(|cell| PreparedCell::new(cell)).collect()
}

impl PreparedCell {
    fn new(raw: &str) -> Self {
        Self {
            raw: raw.to_string(),
            markdown: raw.replace('|', "\\|"),
            width: UnicodeWidthStr::width(raw),
        }
    }
}

#[cfg(test)]
mod tests {
    use super::PreparedTable;
    use crate::ui::doc::{KeyValueRow, TableBlock};

    #[test]
    fn prepared_table_captures_shared_width_and_markdown_shape_unit() {
        let table = TableBlock {
            summary: vec![KeyValueRow {
                key: "count".to_string(),
                value: "1".to_string(),
                indent: None,
                gap: None,
            }],
            headers: vec!["na|me".to_string(), "id".to_string()],
            rows: vec![vec!["ali|ce".to_string(), "42".to_string()]],
        };

        let prepared = PreparedTable::for_markdown(&table);

        assert_eq!(prepared.widths, vec![6, 3]);
        assert_eq!(prepared.headers[0].raw, "na|me");
        assert_eq!(prepared.headers[0].markdown, "na\\|me");
        assert_eq!(prepared.rows[0][0].raw, "ali|ce");
        assert_eq!(prepared.rows[0][0].markdown, "ali\\|ce");
    }
}