prettytable-rs 0.10.0

//! CSV impl and reexported types

use csv;

pub use self::csv::{Reader, ReaderBuilder, Result, Writer};
use crate::AsTableSlice;
use std::io::{Read, Write};
use std::path::Path;

impl<'a> super::TableSlice<'a> {
    /// Write the table to the specified writer.
    pub fn to_csv<W: Write>(&self, w: W) -> Result<Writer<W>> {
        self.to_csv_writer(Writer::from_writer(w))
    }

    /// Write the table to the specified writer.
    ///
    /// This allows for format customisation.
    pub fn to_csv_writer<W: Write>(&self, mut writer: Writer<W>) -> Result<Writer<W>> {
        for title in self.titles {
            writer.write_record(title.iter().map(|c| c.get_content()))?;
        }
        for row in self.rows {
            writer.write_record(row.iter().map(|c| c.get_content()))?;
        }

        writer.flush()?;
        Ok(writer)
    }
}

impl super::Table {
    /// Create a table from a CSV string
    ///
    /// For more customisability use `from_csv()`
    pub fn from_csv_string(csv_s: &str) -> Result<Self> {
        Ok(Self::from_csv(
            &mut ReaderBuilder::new()
                .has_headers(false)
                .from_reader(csv_s.as_bytes()),
        ))
    }

    /// Create a table from a CSV file
    ///
    /// For more customisability use `from_csv()`
    pub fn from_csv_file<P: AsRef<Path>>(csv_p: P) -> Result<Self> {
        Ok(Self::from_csv(
            &mut ReaderBuilder::new().has_headers(false).from_path(csv_p)?,
        ))
    }

    /// Create a table from a CSV reader
    pub fn from_csv<R: Read>(reader: &mut Reader<R>) -> Self {
        Self::init(
            reader
                .records()
                .map(|row| {
                    super::Row::new(
                        row.unwrap()
                            .into_iter()
                            .map(super::Cell::new)
                            .collect(),
                    )
                })
                .collect(),
        )
    }

    /// Write the table to the specified writer.
    pub fn to_csv<W: Write>(&self, w: W) -> Result<Writer<W>> {
        self.as_slice().to_csv(w)
    }

    /// Write the table to the specified writer.
    ///
    /// This allows for format customisation.
    pub fn to_csv_writer<W: Write>(&self, writer: Writer<W>) -> Result<Writer<W>> {
        self.as_slice().to_csv_writer(writer)
    }
}

#[cfg(test)]
mod tests {
    use crate::{Cell, Row, Table};

    static CSV_S: &str = "ABC,DEFG,HIJKLMN\n\
                        foobar,bar,foo\n\
                        foobar2,bar2,foo2\n";

    fn test_table() -> Table {
        let mut table = Table::new();
        table.add_row(Row::new(vec![
            Cell::new("ABC"),
            Cell::new("DEFG"),
            Cell::new("HIJKLMN"),
        ]));
        table.add_row(Row::new(vec![
            Cell::new("foobar"),
            Cell::new("bar"),
            Cell::new("foo"),
        ]));
        table.add_row(Row::new(vec![
            Cell::new("foobar2"),
            Cell::new("bar2"),
            Cell::new("foo2"),
        ]));
        table
    }

    #[test]
    fn from() {
        assert_eq!(
            test_table().to_string().replace("\r\n", "\n"),
            Table::from_csv_string(CSV_S)
                .unwrap()
                .to_string()
                .replace("\r\n", "\n")
        );
    }

    #[test]
    fn to() {
        assert_eq!(
            String::from_utf8(
                test_table()
                    .to_csv(Vec::new())
                    .unwrap()
                    .into_inner()
                    .unwrap()
            )
            .unwrap(),
            CSV_S
        );
    }

    #[test]
    fn trans() {
        assert_eq!(
            Table::from_csv_string(
                &String::from_utf8(
                    test_table()
                        .to_csv(Vec::new())
                        .unwrap()
                        .into_inner()
                        .unwrap()
                )
                .unwrap()
            )
            .unwrap()
            .to_string()
            .replace("\r\n", "\n"),
            test_table().to_string().replace("\r\n", "\n")
        );
    }

    #[test]
    fn extend_table() {
        let mut table = Table::new();
        table.add_row(Row::new(vec![
            Cell::new("ABC"),
            Cell::new("DEFG"),
            Cell::new("HIJKLMN"),
        ]));
        table.extend(vec![vec!["A", "B", "C"]]);
        let t2 = table.clone();
        table.extend(t2.rows);
        assert_eq!(
            table.get_row(1).unwrap().get_cell(2).unwrap().get_content(),
            "C"
        );
        assert_eq!(
            table.get_row(2).unwrap().get_cell(1).unwrap().get_content(),
            "DEFG"
        );
    }
}