feat: 收口 tree-first graph 主链与前端测试修复

2026-04-18 05:43:49 +08:00
parent d8de820d93
commit d3876e56eb
33 changed files with 2421 additions and 345 deletions
@@ -488,6 +488,19 @@ struct DocumentSaveCommandPayload {
    conflict_detection_key: Option<String>,
 }

+#[derive(Debug, Deserialize)]
+#[serde(rename_all = "camelCase")]
+struct DocumentEmbedCommandPayload {
+    document_id: String,
+    workspace_id: Option<String>,
+    revision: Option<u64>,
+    content: Value,
+    conflict_detection_key: Option<String>,
+    source_document_id: String,
+    target_document_id: String,
+    anchor_block_id: Option<String>,
+}
+
 #[derive(Debug, Serialize, Deserialize)]
 #[serde(rename_all = "camelCase")]
 struct KernelCreateNodeCommandPayload {
@@ -4149,6 +4162,53 @@ fn execute_command(
                }),
            }))
        }
+        "documents.embed" => {
+            let payload: DocumentEmbedCommandPayload = parse_payload(command_wire.payload.clone())?;
+            let command = CommandEnvelope {
+                name: "documents.embed".into(),
+                command_id: command_wire.command_id.clone(),
+                idempotency_key: command_wire.idempotency_key.clone(),
+                actor: to_actor_payload(&command_wire.actor),
+                source: to_source_payload(&command_wire.source),
+                target: to_target_ref(command_wire.target.as_ref()),
+                payload: core_protocol::SavePageContent {
+                    page_id: payload.document_id.clone(),
+                    workspace_id: payload.workspace_id.clone(),
+                    revision: payload.revision,
+                    content_json: serde_json::to_string(&payload.content).map_err(|error| {
+                        BridgeError::validation(format!(
+                            "documents.embed content 序列化失败: {error}"
+                        ))
+                    })?,
+                    conflict_detection_key: payload.conflict_detection_key.clone(),
+                },
+                reason: command_wire.reason,
+                refs: command_wire.refs,
+                dry_run: command_wire.dry_run,
+                validate_only: command_wire.validate_only,
+            };
+            let request = build_write_request(&context, &command)?;
+            Ok(RuntimeExecutionPlan::Command(RuntimeCommandExecutionPlan {
+                command_name: command.name,
+                command_id: command.command_id,
+                function_name: request.function_name,
+                workspace_id: request.workspace_id,
+                request_id: request.request_id,
+                trace_id: request.trace_id,
+                actor_id: request.actor_id,
+                idempotency_key: request.idempotency_key,
+                payload_json: request.payload_json,
+                args_json: json!({
+                    "id": payload.document_id,
+                    "content": payload.content,
+                    "expectedRevision": payload.revision,
+                    "conflictDetectionKey": payload.conflict_detection_key,
+                    "sourceDocumentId": payload.source_document_id,
+                    "targetDocumentId": payload.target_document_id,
+                    "anchorBlockId": payload.anchor_block_id,
+                }),
+            }))
+        }
        "mindmaps.put" => {
            let payload: MindmapPutCommandPayload = parse_payload(command_wire.payload.clone())?;
            let command = CommandEnvelope {
@@ -5509,6 +5569,68 @@ mod tests {
        }
    }

+    #[test]
+    fn documents_embed_command_plan_maps_to_documents_update_content() {
+        let plan = execute_runtime_input(RuntimeInput::Command {
+            context: demo_context(),
+            command: RuntimeCommandEnvelopeWire {
+                name: "documents.embed".into(),
+                command_id: "cmd_doc_embed_1".into(),
+                idempotency_key: Some("idem_doc_embed".into()),
+                actor: RuntimeActorWire {
+                    actor_type: "user".into(),
+                    actor_id: "user_1".into(),
+                    session_id: Some("sess_1".into()),
+                },
+                source: RuntimeSourceWire {
+                    channel: "next-route".into(),
+                    client: "wolai-frontend".into(),
+                },
+                target: Some(RuntimeTargetWire {
+                    workspace_id: Some("ws_1".into()),
+                    page_id: Some("doc_2".into()),
+                    block_id: None,
+                }),
+                payload: json!({
+                    "documentId": "doc_2",
+                    "workspaceId": "ws_1",
+                    "revision": 5,
+                    "content": [{ "id": "block_1", "type": "pageReference" }],
+                    "conflictDetectionKey": "conflict_5",
+                    "sourceDocumentId": "doc_1",
+                    "targetDocumentId": "doc_2",
+                    "anchorBlockId": "anchor_1",
+                }),
+                reason: Some("嵌入页面".into()),
+                refs: vec![],
+                dry_run: false,
+                validate_only: false,
+            },
+        })
+        .expect("command plan should build");
+
+        match plan {
+            RuntimeExecutionPlan::Command(plan) => {
+                assert_eq!(plan.function_name, "documents:updateContent");
+                assert_eq!(plan.command_name, "documents.embed");
+                assert_eq!(
+                    plan.args_json,
+                    json!({
+                        "id": "doc_2",
+                        "content": [{ "id": "block_1", "type": "pageReference" }],
+                        "expectedRevision": 5,
+                        "conflictDetectionKey": "conflict_5",
+                        "sourceDocumentId": "doc_1",
+                        "targetDocumentId": "doc_2",
+                        "anchorBlockId": "anchor_1",
+                    })
+                );
+            }
+            RuntimeExecutionPlan::Query(_) => panic!("expected command plan"),
+            RuntimeExecutionPlan::Tool(_) => panic!("expected command plan"),
+        }
+    }
+
    #[test]
    fn mindmap_get_tool_plan_uses_mindmaps_get_query() {
        let plan = execute_runtime_input(RuntimeInput::Tool {