aboutsummaryrefslogtreecommitdiff
path: root/src/format-table/lib.rs
diff options
context:
space:
mode:
authorAlex <alex@adnab.me>2023-05-17 12:33:45 +0000
committerAlex <alex@adnab.me>2023-05-17 12:33:45 +0000
commitc26a4308b4454e2f36e2824280b9f587a6918fa9 (patch)
tree3086149203b8cf9f0280c9b8e784eb893b0d7465 /src/format-table/lib.rs
parentb66f247580dcfb733718949c35240500903a0802 (diff)
parent217d4299379b70d1b081cb88e208ab5f2d1808a7 (diff)
downloadgarage-c26a4308b4454e2f36e2824280b9f587a6918fa9.tar.gz
garage-c26a4308b4454e2f36e2824280b9f587a6918fa9.zip
Merge pull request 'Split format_table into separate crate and reduce k2v-client dependencies' (#572) from split-format-table into main
Reviewed-on: https://git.deuxfleurs.fr/Deuxfleurs/garage/pulls/572
Diffstat (limited to 'src/format-table/lib.rs')
-rw-r--r--src/format-table/lib.rs49
1 files changed, 49 insertions, 0 deletions
diff --git a/src/format-table/lib.rs b/src/format-table/lib.rs
new file mode 100644
index 00000000..55252ba9
--- /dev/null
+++ b/src/format-table/lib.rs
@@ -0,0 +1,49 @@
+//! Format tables with a stupid API.
+//!
+//! Example:
+//!
+//! ```rust
+//! let mut table = vec!["product\tquantity\tprice".to_string()];
+//! for (p, q, r) in [("tomato", 12, 15), ("potato", 10, 20), ("rice", 5, 12)] {
+//! table.push(format!("{}\t{}\t{}", p, q, r));
+//! }
+//! format_table::format_table(table);
+//! ```
+//!
+//! A table to be formatted is a `Vec<String>`, containing one string per line.
+//! Table columns in each line are separated by a `\t` character.
+
+/// Format a table and return the result as a string.
+pub fn format_table_to_string(data: Vec<String>) -> String {
+ let data = data
+ .iter()
+ .map(|s| s.split('\t').collect::<Vec<_>>())
+ .collect::<Vec<_>>();
+
+ let columns = data.iter().map(|row| row.len()).fold(0, std::cmp::max);
+ let mut column_size = vec![0; columns];
+
+ let mut out = String::new();
+
+ for row in data.iter() {
+ for (i, col) in row.iter().enumerate() {
+ column_size[i] = std::cmp::max(column_size[i], col.chars().count());
+ }
+ }
+
+ for row in data.iter() {
+ for (col, col_len) in row[..row.len() - 1].iter().zip(column_size.iter()) {
+ out.push_str(col);
+ (0..col_len - col.chars().count() + 2).for_each(|_| out.push(' '));
+ }
+ out.push_str(row[row.len() - 1]);
+ out.push('\n');
+ }
+
+ out
+}
+
+/// Format a table and print the result to stdout.
+pub fn format_table(data: Vec<String>) {
+ print!("{}", format_table_to_string(data));
+}