diff --git a/frontend/src/utils/previewBridge.ts b/frontend/src/utils/previewBridge.ts index e90d1988..501badc0 100644 --- a/frontend/src/utils/previewBridge.ts +++ b/frontend/src/utils/previewBridge.ts @@ -40,10 +40,16 @@ export interface OpenInEditorPayload { export interface ClickToComponentMessage { source: 'click-to-component'; version: number; - type: 'ready' | 'open-in-editor'; + type: 'ready' | 'open-in-editor' | 'enable-button'; payload?: OpenInEditorPayload; } +export interface ClickToComponentEnableMessage { + source: 'click-to-component'; + version: 1; + type: 'enable-button'; +} + export interface EventHandlers { onReady?: () => void; onOpenInEditor?: (payload: OpenInEditorPayload) => void; @@ -76,6 +82,14 @@ export class ClickToComponentListener { switch (data.type) { case 'ready': + if (event.source) { + const enableMsg: ClickToComponentEnableMessage = { + source: 'click-to-component', + version: 1, + type: 'enable-button', + }; + (event.source as Window).postMessage(enableMsg, '*'); + } this.handlers.onReady?.(); break;