From 2360dc6b5358d612d063d234cc32e53537b14273 Mon Sep 17 00:00:00 2001
From: Jakob <jakob@rowanbrook.net>
Date: Mon, 23 Mar 2026 13:43:27 +0100
Subject: [PATCH] Get batch number from file contents instead of filename

---
 src/main.rs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 12533da..981897e 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -178,8 +178,12 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
         batch.filename
     );
 
-    let batch_number = batch.filename.trim_end_matches(".csv");
-    let output_dir = base_output_dir.join(batch_number);
+    let batch_number = batch
+        .transactions
+        .first()
+        .map(|t| t.batch_number.clone())
+        .unwrap_or_else(|| "unknown".to_string());
+    let output_dir = base_output_dir.join(&batch_number);
 
     fs::create_dir_all(&output_dir)?;