1

server-rs/crates/spacetime-client/src/mapper.rs

@@ -2917,6 +2917,7 @@ pub(crate) fn map_puzzle_draft_level(snapshot: DomainPuzzleDraftLevel) -> Puzzle
         level_name: snapshot.level_name,
         picture_description: snapshot.picture_description,
         picture_reference: snapshot.picture_reference,
+        background_music: snapshot.background_music.map(map_puzzle_audio_asset),
         candidates: snapshot
             .candidates
             .into_iter()
@@ -2929,6 +2930,21 @@ pub(crate) fn map_puzzle_draft_level(snapshot: DomainPuzzleDraftLevel) -> Puzzle
     }
 }
 
+pub(crate) fn map_puzzle_audio_asset(
+    asset: module_puzzle::PuzzleAudioAsset,
+) -> PuzzleAudioAssetRecord {
+    PuzzleAudioAssetRecord {
+        task_id: asset.task_id,
+        provider: asset.provider,
+        asset_object_id: asset.asset_object_id,
+        asset_kind: asset.asset_kind,
+        audio_src: asset.audio_src,
+        prompt: asset.prompt,
+        title: asset.title,
+        updated_at: asset.updated_at,
+    }
+}
+
 pub(crate) fn map_puzzle_creator_intent(
     snapshot: DomainPuzzleCreatorIntent,
 ) -> PuzzleCreatorIntentRecord {
@@ -7269,6 +7285,7 @@ pub struct PuzzleDraftLevelRecord {
     pub level_name: String,
     pub picture_description: String,
     pub picture_reference: Option<String>,
+    pub background_music: Option<PuzzleAudioAssetRecord>,
     pub candidates: Vec<PuzzleGeneratedImageCandidateRecord>,
     pub selected_candidate_id: Option<String>,
     pub cover_image_src: Option<String>,
@@ -7276,6 +7293,18 @@ pub struct PuzzleDraftLevelRecord {
     pub generation_status: String,
 }
 
+#[derive(Clone, Debug, PartialEq, Eq)]
+pub struct PuzzleAudioAssetRecord {
+    pub task_id: String,
+    pub provider: String,
+    pub asset_object_id: Option<String>,
+    pub asset_kind: Option<String>,
+    pub audio_src: String,
+    pub prompt: Option<String>,
+    pub title: Option<String>,
+    pub updated_at: Option<String>,
+}
+
 #[derive(Clone, Debug, PartialEq, Eq)]
 pub struct PuzzleAgentMessageRecord {
     pub message_id: String,