From ea4887a41c1655e613d13cf4be01f2659ca1a126 Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Thu, 29 Feb 2024 23:06:56 -0500
Subject: [PATCH] name file based on type

---
 .../bvh_to_workflow/process_job.rs            | 20 +++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

diff --git a/crates/service/job/inference_job/src/job/job_types/bevy_to_workflow/bvh_to_workflow/process_job.rs b/crates/service/job/inference_job/src/job/job_types/bevy_to_workflow/bvh_to_workflow/process_job.rs
index 0296b55fa2..7a56aba4a8 100644
--- a/crates/service/job/inference_job/src/job/job_types/bevy_to_workflow/bvh_to_workflow/process_job.rs
+++ b/crates/service/job/inference_job/src/job/job_types/bevy_to_workflow/bvh_to_workflow/process_job.rs
@@ -69,8 +69,11 @@ pub async fn process_job(args: BvhToWorkflowJobArgs<'_>) -> Result<JobSuccessRes
 
   info!("Download media for conversion...");
 
+  let asset_filename = get_asset_filename(&media_file)
+      .ok_or(ProcessSingleJobError::Other(anyhow!("media_file has the wrong file type")))?;
+
   let original_media_upload_fs_path = {
-    let original_media_file_fs_path = work_temp_dir.path().join("original.bvh");
+    let original_media_file_fs_path = work_temp_dir.path().join(asset_filename);
 
     let media_file_bucket_path = MediaFileBucketPath::from_object_hash(
       &media_file.public_bucket_directory_hash,
@@ -79,7 +82,9 @@ pub async fn process_job(args: BvhToWorkflowJobArgs<'_>) -> Result<JobSuccessRes
 
     let bucket_object_path = media_file_bucket_path.to_full_object_pathbuf();
 
-    info!("Downloading media to bucket path: {:?}", &bucket_object_path);
+    info!("Downloading media to bucket path: {:?} to filesystem path: {:?}",
+      &bucket_object_path,
+      &original_media_file_fs_path);
 
     maybe_download_file_from_bucket(MaybeDownloadArgs {
       name_or_description_of_file: "media upload (original file)",
@@ -282,6 +287,17 @@ pub async fn process_job(args: BvhToWorkflowJobArgs<'_>) -> Result<JobSuccessRes
   })
 }
 
+/// Storyteller Engine uses the file extension to determine engine behavior,
+/// so this is essential to map correctly.
+fn get_asset_filename(media_file: &&MediaFileForInference) -> Option<&'static str> {
+  match media_file.media_type {
+    MediaFileType::Bvh => Some("original.bvh"),
+    MediaFileType::Glb => Some("original.scn.gltf"), // TODO(bt): Is the extension "gltf" for "glb" ??
+    MediaFileType::SceneRon => Some("original.scn.ron"),
+    _ => None,
+  }
+}
+
 #[derive(Deserialize, Default)]
 struct FileMetadata {
   pub duration_millis: Option<u64>,