1

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

@@ -4,6 +4,7 @@ use crate::module_bindings::claim_puzzle_work_point_incentive_procedure::claim_p
 use crate::module_bindings::delete_puzzle_work_procedure::delete_puzzle_work;
 use crate::module_bindings::record_puzzle_work_like_procedure::record_puzzle_work_like;
 use crate::module_bindings::remix_puzzle_work_procedure::remix_puzzle_work;
+use crate::module_bindings::save_puzzle_ui_background_procedure::save_puzzle_ui_background;
 
 impl SpacetimeClient {
     pub async fn create_puzzle_agent_session(
@@ -190,6 +191,35 @@ impl SpacetimeClient {
         .await
     }
 
+    pub async fn save_puzzle_ui_background(
+        &self,
+        input: PuzzleUiBackgroundSaveRecordInput,
+    ) -> Result<PuzzleAgentSessionRecord, SpacetimeClientError> {
+        let procedure_input = PuzzleUiBackgroundSaveInput {
+            session_id: input.session_id,
+            owner_user_id: input.owner_user_id,
+            level_id: input.level_id,
+            levels_json: input.levels_json,
+            prompt: input.prompt,
+            image_src: input.image_src,
+            image_object_key: input.image_object_key,
+            saved_at_micros: input.saved_at_micros,
+        };
+
+        self.call_after_connect(move |connection, sender| {
+            connection.procedures().save_puzzle_ui_background_then(
+                procedure_input,
+                move |_, result| {
+                    let mapped = result
+                        .map_err(SpacetimeClientError::from_sdk_error)
+                        .and_then(map_puzzle_agent_session_procedure_result);
+                    send_once(&sender, mapped);
+                },
+            );
+        })
+        .await
+    }
+
     pub async fn select_puzzle_cover_image(
         &self,
         input: PuzzleSelectCoverImageRecordInput,