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

use crate::chopper::chopper::Source;
use crate::error::{CliResult, Error};
use crate::source::csv_configs::CSVOutputConfig;

pub fn parse_into_delimiter(str: &str) -> CliResult<u8> {
    /* Code in this function was adapted from public domain xsv project. */
    match &*str {
        r"\t" => Ok(b'\t'),
        s => {
            if s.len() != 1 {
                return Err(Error::from(
                    format!("Error: specified delimiter '{}' is not a single ASCII character.", s)))
            }
            let c = s.chars().next().unwrap();
            if c.is_ascii() {
                Ok(c as u8)
            } else {
                Err(Error::from(
                    format!("Error: specified delimiter '{}' is not an ASCII character.", c)))
            }
        }
    }
}

pub fn create_csv_output_config_from_source(sources: &mut Vec<Box<dyn Source>>, delimiter: &str) -> CSVOutputConfig {
    let mut all_sources_have_native_timestamps = true;
    for source in sources {
        if !source.has_native_timestamp_column() {
            all_sources_have_native_timestamps = false;
            break;
        }
    }
    CSVOutputConfig::new(delimiter, all_sources_have_native_timestamps)
}