Add command line parameters to specify output files

src/main.rs

@@ -3,13 +3,22 @@ use clap::Parser;
 #[derive(Parser)]
 #[command(author, version, about, long_about = None)]
 struct Cli {
+    /// Input OpenStreetMap in the .pbf format
     source_pbf: String,
+    /// Output path of the csv file that will contain the nodes
+    #[arg(short, long, default_value = "nodes.csv")]
+    nodes_file: String,
+    /// Output path of the csv file that will contain the edges
+    #[arg(short, long, default_value = "nodes.csv")]
+    edges_file: String,
 }
 fn main() {
     let cli = Cli::parse();
 
     match osm4routing::read(&cli.source_pbf) {
-        Ok((nodes, edges)) => osm4routing::writers::csv(nodes, edges),
+        Ok((nodes, edges)) => {
+            osm4routing::writers::csv(nodes, edges, &cli.nodes_file, &cli.edges_file)
+        }
         Err(error) => println!("Error: {}", error),
     }
 }

src/osm4routing/writers.rs

@@ -1,7 +1,7 @@
 use super::models::*;
 
-pub fn csv(nodes: Vec<Node>, edges: Vec<Edge>) {
-    let edges_path = std::path::Path::new("edges.csv");
+pub fn csv(nodes: Vec<Node>, edges: Vec<Edge>, nodes_file: &str, edges_file: &str) {
+    let edges_path = std::path::Path::new(edges_file);
     let mut edges_csv = csv::Writer::from_path(edges_path).unwrap();
     edges_csv
         .serialize(vec![
@@ -38,7 +38,7 @@ pub fn csv(nodes: Vec<Node>, edges: Vec<Edge>) {
             .expect("CSV: unable to write edge");
     }
 
-    let nodes_path = std::path::Path::new("nodes.csv");
+    let nodes_path = std::path::Path::new(nodes_file);
     let mut nodes_csv = csv::Writer::from_path(nodes_path).unwrap();
     nodes_csv
         .serialize(vec!["id", "lon", "lat"])