fix: settle puzzle failures and profile tasks

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

@@ -11,20 +11,20 @@ use module_puzzle::{
     PuzzleAgentMessageRole, PuzzleAgentMessageSnapshot, PuzzleAgentSessionCreateInput,
     PuzzleAgentSessionGetInput, PuzzleAgentSessionProcedureResult, PuzzleAgentSessionSnapshot,
     PuzzleAgentStage, PuzzleAnchorPack, PuzzleDraftCompileInput, PuzzleFormDraftSaveInput,
-    PuzzleGeneratedImageCandidate, PuzzleGeneratedImagesSaveInput, PuzzleLeaderboardEntry,
-    PuzzleLeaderboardSubmitInput, PuzzlePublicationStatus, PuzzlePublishInput,
-    PuzzleRecommendedNextWork, PuzzleResultDraft, PuzzleRunDragInput, PuzzleRunGetInput,
-    PuzzleRunNextLevelInput, PuzzleRunPauseInput, PuzzleRunProcedureResult, PuzzleRunPropInput,
-    PuzzleRunSnapshot, PuzzleRunStartInput, PuzzleRunSwapInput, PuzzleRuntimeLevelStatus,
-    PuzzleSelectCoverImageInput, PuzzleUiBackgroundSaveInput, PuzzleWorkDeleteInput,
-    PuzzleWorkGetInput, PuzzleWorkLikeRecordInput as PuzzleWorkLikeInput,
+    PuzzleDraftCompileFailureInput, PuzzleGeneratedImageCandidate, PuzzleGeneratedImagesSaveInput,
+    PuzzleLeaderboardEntry, PuzzleLeaderboardSubmitInput, PuzzlePublicationStatus,
+    PuzzlePublishInput, PuzzleRecommendedNextWork, PuzzleResultDraft, PuzzleRunDragInput,
+    PuzzleRunGetInput, PuzzleRunNextLevelInput, PuzzleRunPauseInput, PuzzleRunProcedureResult,
+    PuzzleRunPropInput, PuzzleRunSnapshot, PuzzleRunStartInput, PuzzleRunSwapInput,
+    PuzzleRuntimeLevelStatus, PuzzleSelectCoverImageInput, PuzzleUiBackgroundSaveInput,
+    PuzzleWorkDeleteInput, PuzzleWorkGetInput, PuzzleWorkLikeRecordInput as PuzzleWorkLikeInput,
     PuzzleWorkPointIncentiveClaimInput, PuzzleWorkProcedureResult, PuzzleWorkProfile,
     PuzzleWorkRemixInput, PuzzleWorkUpsertInput, PuzzleWorksListInput, PuzzleWorksProcedureResult,
     apply_publish_overrides_to_draft, apply_selected_candidate, build_form_draft_from_seed,
     build_result_preview, compile_result_draft_from_seed, create_work_profile, infer_anchor_pack,
-    normalize_puzzle_draft, normalize_puzzle_levels, normalize_theme_tags, publish_work_profile,
-    replace_puzzle_level, select_next_profiles, selected_profile_level_after_runtime_level,
-    selected_puzzle_level, tag_similarity_score,
+    mark_failed_puzzle_result_draft_generation, normalize_puzzle_draft, normalize_puzzle_levels,
+    normalize_theme_tags, publish_work_profile, replace_puzzle_level, select_next_profiles,
+    selected_profile_level_after_runtime_level, selected_puzzle_level, tag_similarity_score,
 };
 use module_runtime::RuntimeProfileWalletLedgerSourceType;
 use module_runtime::visible_runtime_profile_user_tags;
@@ -363,6 +363,25 @@ pub fn compile_puzzle_agent_draft(
     }
 }
 
+#[spacetimedb::procedure]
+pub fn mark_puzzle_draft_generation_failed(
+    ctx: &mut ProcedureContext,
+    input: PuzzleDraftCompileFailureInput,
+) -> PuzzleAgentSessionProcedureResult {
+    match ctx.try_with_tx(|tx| mark_puzzle_draft_generation_failed_tx(tx, input.clone())) {
+        Ok(session) => PuzzleAgentSessionProcedureResult {
+            ok: true,
+            session: Some(session),
+            error_message: None,
+        },
+        Err(message) => PuzzleAgentSessionProcedureResult {
+            ok: false,
+            session: None,
+            error_message: Some(message),
+        },
+    }
+}
+
 /// 保存拼图入口表单草稿。
 /// 中文注释：该 procedure 只更新 session 与创作中心草稿卡，不触发图片生成或发布校验。
 #[spacetimedb::procedure]
@@ -999,6 +1018,60 @@ fn compile_puzzle_agent_draft_tx(
     )
 }
 
+fn mark_puzzle_draft_generation_failed_tx(
+    ctx: &TxContext,
+    input: PuzzleDraftCompileFailureInput,
+) -> Result<PuzzleAgentSessionSnapshot, String> {
+    let row = get_owned_session_row(ctx, &input.session_id, &input.owner_user_id)?;
+    let updated_at = Timestamp::from_micros_since_unix_epoch(input.failed_at_micros);
+    let draft = match deserialize_optional_draft(&row.draft_json)? {
+        Some(draft) => mark_failed_puzzle_result_draft_generation(draft),
+        None => {
+            let anchor_pack = deserialize_anchor_pack(&row.anchor_pack_json)?;
+            let messages = list_session_messages(ctx, &row.session_id);
+            mark_failed_puzzle_result_draft_generation(compile_result_draft_from_seed(
+                &anchor_pack,
+                &messages,
+                Some(&row.seed_text),
+            ))
+        }
+    };
+    upsert_puzzle_draft_work_profile(
+        ctx,
+        &row.session_id,
+        &row.owner_user_id,
+        &draft,
+        input.failed_at_micros,
+    )?;
+
+    replace_puzzle_agent_session(
+        ctx,
+        &row,
+        PuzzleAgentSessionRow {
+            session_id: row.session_id.clone(),
+            owner_user_id: row.owner_user_id.clone(),
+            seed_text: row.seed_text.clone(),
+            current_turn: row.current_turn,
+            progress_percent: row.progress_percent.max(88),
+            stage: row.stage,
+            anchor_pack_json: row.anchor_pack_json.clone(),
+            draft_json: Some(serialize_json(&draft)),
+            last_assistant_reply: Some(input.error_message),
+            published_profile_id: row.published_profile_id.clone(),
+            created_at: row.created_at,
+            updated_at,
+        },
+    );
+
+    get_puzzle_agent_session_tx(
+        ctx,
+        PuzzleAgentSessionGetInput {
+            session_id: input.session_id,
+            owner_user_id: input.owner_user_id,
+        },
+    )
+}
+
 fn save_puzzle_form_draft_tx(
     ctx: &TxContext,
     input: PuzzleFormDraftSaveInput,