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

// import custom type for use in specifying particular vector
use crate::Record;

// import things needed for dealing with files, so saving can work.
use std::{fs::File, io::{ Write, Error } };

// import serde_json, to save contents as JSON
use serde_json;

/// enables the ability to save data when implemented.
pub trait Save {
    fn save(&self, path: &str) -> Result<(), Error>;
    fn save_tsv(&self, path: &str) -> Result<(), Error>;
}

// add implementation of Save trait to Vector of Records.
impl Save for Vec<Record> {
    fn save(&self, path: &str) -> Result<(), Error> {
        let mut output = File::create(path)?;
        let json_string = serde_json::to_string_pretty(self)?;

        match write!(output, "{}", format!("{}", json_string)) {
            Ok(()) => Ok(()),
            Err(error) => Err(error)
        }  
    }

    fn save_tsv(&self, path: &str) -> Result<(), Error> {
        let mut output = File::create(path)?;
        let tsv_string: String = if cfg!(target_os = "unix") {
            self.iter().map(|record| record.to_string() + "\n").collect()
        } else {
            self.iter().map(|record| record.to_string() + "\r\n").collect()
        };

        match write!(output, "{}", format!("{}", tsv_string)) {
            Ok(()) => Ok(()),
            Err(error) => Err(error)
        } 
    }
}