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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125

pub struct Table<'a> {
    str_arr: &'a Vec<Vec<String>>,
    num_columns: usize,
    num_rows: usize,
    column_width: Vec<usize>,
    num_chars: usize
}

impl<'a> Table<'a> {
    pub fn new(str_arr: &'a Vec<Vec<String>>) -> Table<'a> {
        let num_rows: usize = str_arr.len();
        let num_columns: usize = str_arr[0].len();
        let mut column_width: Vec<usize> = vec![0; num_columns];
        let mut table_width: usize = 0;
        let padding_len: usize = 2;
        let num_lines: usize = 3;

        for col in 0..num_columns {
            // column width is longest string in that column
            for row in 0..num_rows {
                let str_len: usize = str_arr[row][col].len();
                column_width[col] = std::cmp::max(column_width[col], str_len);
            }
            table_width += column_width[col];
        }
        // account for table formatting characters
        table_width += num_columns + 1; // borders
        table_width += padding_len * num_columns; // padding
        table_width += 1; // newlines
        let num_chars: usize = table_width * (num_rows + num_lines);

        Table {
            str_arr,
            num_rows,
            num_columns,
            column_width,
            num_chars
        }
    }

    pub fn to_string(&mut self) -> String {
        let padding_len: usize = 3;
        let border: &str = "|";
        let edge: &str = "+";
        let line: &str = "-";
        let space: &str = " ";
        let newline: &str = "\n";
        let mut table_str: String = String::with_capacity(self.num_chars);


        // top border
        table_str += edge;
        for col in 0..self.num_columns {
            for _ in 0..self.column_width[col] + padding_len - 1 {
                table_str += line;
            }
            table_str += edge;
        }
        table_str += newline;

        // column headers
        table_str += border;
        table_str += space;
        for col in 0..self.num_columns {
            table_str += &self.str_arr[0][col];
            let str_len: usize = self.str_arr[0][col].len();
            for _ in str_len..self.column_width[col] {
                table_str += space;
            }
            // add separators
            if col != self.num_columns - 1 {
                table_str += space;
                table_str += border;
                table_str += space;
            }
        }
        table_str += space;
        table_str += border;
        table_str += newline;

        // middle line
        table_str += edge;
        for col in 0..self.num_columns {
            for _ in 0..self.column_width[col] + padding_len - 1 {
                table_str += line;
            }
            table_str += edge;
        }

        // values
        table_str += newline;
        for row in 1..self.num_rows {
            table_str += border;
            table_str += space;
            for col in 0..self.num_columns {
                table_str += &self.str_arr[row][col];
                // add padding
                let str_len: usize = self.str_arr[row][col].len();
                for _ in str_len..self.column_width[col] {
                    table_str += space;
                }
                // add seperator
                if col != self.num_columns - 1 {
                    table_str += space;
                    table_str += border;
                    table_str += space;
                }
            }
            table_str += space;
            table_str += border;
            table_str += newline;
        }

        // bottom border
        table_str += edge;
        for col in 0..self.num_columns {
            for _ in 0..self.column_width[col] + padding_len - 1 {
                table_str += line;
            }
            table_str += edge;
        }

        table_str
    }
}