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::chopper::chopper::HeaderSink;
use crate::error::{CliResult, Error};
use crate::source::csv_configs::CSVOutputConfig;
use crate::write::csv_sink;
use crate::write::dc_sink;

pub fn new_header_sink(
    output: Option<&str>,
    csv_output_config: Option<CSVOutputConfig>,
) -> CliResult<Box<dyn HeaderSink + 'static>> {
    let csv_output_config = match csv_output_config {
        Some(c) => c,
        None => CSVOutputConfig::new_default(),
    };
    let writer: Box<dyn HeaderSink + 'static>;
    match output {
        Some(p) => {
            let p = p.to_string();
            if p.ends_with("csv") {
                writer = Box::new(csv_sink::CSVSink::new(&Some(p), csv_output_config)?);
            } else if p.ends_with("dc") {
                writer = Box::new(dc_sink::DCSink::new(&Some(p))?);
            } else {
                return Err(Error::from(format!("file type -- {} is not supported", p)));
            }
        }
        None => {
            // if none use csv sink
            writer = Box::new(csv_sink::CSVSink::new(&None, csv_output_config)?);
        }
    }
    Ok(writer)
}