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

use crate::extension::odbc::{OdbcColumn, OdbcColumnItem};
use odbc_common::print_table::Print;
use odbc_common::{StyledString, Table, TableTheme, TextStyle};

#[derive(Debug, Default)]
pub struct QueryResult {
    // table columns header
    pub columns: Vec<OdbcColumn>,
    // table columns data
    pub data: Vec<Vec<OdbcColumnItem>>,
}

impl Print for QueryResult {
    fn convert_table(self) -> anyhow::Result<Table> {
        let headers: Vec<StyledString> = self
            .columns
            .iter()
            .map(|x| StyledString::new(x.name.to_string(), TextStyle::default_header()))
            .collect();

        let rows = self
            .data
            .iter()
            .map(|x| {
                x.iter()
                    .map(|y| y.to_string())
                    .map(|y| StyledString::new(y, TextStyle::basic_left()))
                    .collect::<Vec<_>>()
            })
            .collect();
        Ok(Table::new(headers, rows, TableTheme::rounded()))
    }
}