A great re-architecture of the cli

src/cli/output.rs

@@ -0,0 +1,49 @@
+use std::fs::File;
+use std::path::Path;
+use rand::Rng;
+use crate::cli::cli::OutputFormat;
+use crate::system::model::DLASystem;
+use crate::system::{Position, Storage};
+use crate::system::spawner::Spawner;
+use crate::system::sticker::Sticker;
+use crate::system::walker::Walker;
+
+pub fn write<R: Rng, P: Position, S: Storage<P>, W: Walker<P>, Sp: Spawner<P>, St: Sticker<P, S>>(
+    sys: &DLASystem<R, P, S, W, Sp, St>,
+    format: OutputFormat,
+    output: &Path,
+) {
+    match format {
+        OutputFormat::FullDataJson => write_json_full_data(sys, output),
+        OutputFormat::Positions => write_csv_positions(sys, output),
+    }
+}
+
+fn write_csv_positions<R: Rng, P: Position, S: Storage<P>, W: Walker<P>, Sp: Spawner<P>, St: Sticker<P, S>>(sys: &DLASystem<R, P, S, W, Sp, St>, csv_path: &Path) {
+    let mut wtr = csv::Writer::from_path(csv_path)
+        .expect("Failed to open file");
+
+    // CSVs can only store the raw positions
+    let positions: Vec<&P> = sys.history
+        .iter()
+        .map(|line| &line.position)
+        .collect();
+
+    wtr.serialize(positions)
+        .unwrap();
+
+    wtr.flush()
+        .unwrap();
+}
+
+fn write_json_full_data<R: Rng, P: Position, S: Storage<P>, W: Walker<P>, Sp: Spawner<P>, St: Sticker<P, S>>(sys: &DLASystem<R, P, S, W, Sp, St>, output_path: &Path) {
+    let file = File::create(output_path).expect("Failed to open file");
+
+    let vec: Vec<P> = sys.history
+        .iter()
+        .map(|l| l.position.clone())
+        .collect();
+
+    serde_json::to_writer(file, &vec)
+        .expect("Failed to write json");
+}