fix render data

MCP-UI-Org/mcp-ui#106

Author: kentcdodds

exercises/05.advanced/02.problem.render-data/app/utils/mcp.ts

@@ -98,8 +98,8 @@ export { sendMcpMessage }
 
 // 🐨 export a waitForRenderData function that works like sendMcpMessage, but for render data
 // 🐨 it should create a messageId and send it to the parent frame with the type 'ui-request-render-data'
-// 🐨 handleMessage should check the event.data.type is 'ui-message-response' and the messageId is the same as the one sent
+// 🐨 handleMessage should check the event.data.type is 'ui-lifecycle-iframe-render-data' and the messageId is the same as the one sent
 // 🐨 if the event.data.payload.error is present, return reject(error)
-// 🐨 if the event.data.payload.response is present, return resolve(response)
-// 💯 add schema as an optional parameter and parse the response if it is present
+// 🐨 if the event.data.payload.renderData is present, return resolve(renderData)
+// 💯 add schema as an optional parameter and parse the renderData if it is present
 // 🦺 if you'd like to make it more typesafe, make waitForRenderData a generic (withForRenderData<RenderData>), pass the generic type to the schema (z.ZodSchema<RenderData>) and set it as the return type (Promise<RenderData>)

exercises/05.advanced/02.solution.render-data/app/utils/mcp.ts

@@ -108,16 +108,16 @@ export function waitForRenderData<RenderData>(
 		)
 
 		function handleMessage(event: MessageEvent) {
-			if (event.data?.type !== 'ui-message-response') return
+			if (event.data?.type !== 'ui-lifecycle-iframe-render-data') return
 			if (event.data.messageId !== messageId) return
 			window.removeEventListener('message', handleMessage)
 
-			const { response, error } = event.data.payload
+			const { renderData, error } = event.data.payload
 
 			if (error) return reject(error)
-			if (!schema) return resolve(response)
+			if (!schema) return resolve(renderData)
 
-			const parseResult = schema.safeParse(response)
+			const parseResult = schema.safeParse(renderData)
 			if (!parseResult.success) return reject(parseResult.error)
 
 			return resolve(parseResult.data)