refactor: implement pr feedback - create directory if it doesn't exist

Co-authored-by: Mitchell Mosure <[email protected]>
mosure · Jan 8, 2024 · 114f995 · 114f995
1 parent caacf84
commit 114f995
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/examples/headless.rs b/examples/headless.rs
@@ -331,11 +331,13 @@ mod frame_capture {
                             Ok(img) => img.to_rgba8(),
                             Err(e) => panic!("Failed to create image buffer {e:?}"),
                         };
-    
-                        let images_path =
+
+                        let images_dir =
                             PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("test_images");
+                        std::fs::create_dir_all(&images_dir).unwrap();
+
                         let uuid = bevy::utils::Uuid::new_v4();
-                        let image_path = images_path.join(format!("{uuid}.png"));
+                        let image_path = images_dir.join(format!("{uuid}.png"));
                         if let Err(e) = img.save(image_path){
                             panic!("Failed to save image: {}", e);
                         };