Forward renderer responses to injected window handler (#1631)

react-cosmos · Mar 11, 2024 · db623aa · db623aa
1 parent 0d0e717
commit db623aa
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/packages/react-cosmos-renderer/src/rendererConnect/createWebSocketsConnect.ts b/packages/react-cosmos-renderer/src/rendererConnect/createWebSocketsConnect.ts
@@ -1,10 +1,17 @@
 import {
   RendererConnect,
   RendererRequest,
+  RendererResponse,
   SocketMessage,
   rendererSocketMessage,
 } from 'react-cosmos-core';
 
+declare global {
+  interface Window {
+    cosmosRendererResponse?: (msg: RendererResponse) => void;
+  }
+}
+
 export function createWebSocketsConnect(url: string): RendererConnect {
   let pendingMessages: SocketMessage[] = [];
 
@@ -24,6 +31,11 @@ export function createWebSocketsConnect(url: string): RendererConnect {
       } else {
         pendingMessages.push(socketMessage);
       }
+
+      // Allow headless browsers to capture renderer responses
+      if (window.cosmosRendererResponse) {
+        window.cosmosRendererResponse(rendererResponse);
+      }
     },
 
     onMessage(onMessage) {