mq-conv 0.1.4

A CLI tool for converting various file formats to Markdown
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

use std::io::Write;

use crate::converter::Converter;
use crate::error::{Error, Result};

pub struct CsvConverter;

impl Converter for CsvConverter {
    fn format_name(&self) -> &'static str {
        "csv"
    }

    fn convert(&self, input: &[u8], writer: &mut dyn Write) -> Result<()> {
        let mut reader = csv::ReaderBuilder::new()
            .flexible(true)
            .from_reader(input);

        let headers = reader.headers().map_err(|e| Error::Conversion {
            format: "csv",
            message: e.to_string(),
        })?;

        let col_count = headers.len();
        if col_count == 0 {
            writeln!(writer, "*Empty CSV*")?;
            return Ok(());
        }

        // Header row
        write!(writer, "|")?;
        for field in headers.iter() {
            write!(writer, " {} |", escape_pipe(field))?;
        }
        writeln!(writer)?;

        // Separator
        write!(writer, "|")?;
        for _ in 0..col_count {
            write!(writer, "---|")?;
        }
        writeln!(writer)?;

        // Data rows
        for result in reader.records() {
            let record = result.map_err(|e| Error::Conversion {
                format: "csv",
                message: e.to_string(),
            })?;
            write!(writer, "|")?;
            for i in 0..col_count {
                let cell = record.get(i).unwrap_or("");
                write!(writer, " {} |", escape_pipe(cell))?;
            }
            writeln!(writer)?;
        }

        Ok(())
    }
}

fn escape_pipe(s: &str) -> String {
    s.replace('|', "\\|")
}