@rozenite/network-activity-plugin 1.7.0 → 1.8.1

package/dist/react-native/chunks/useNetworkActivityDevTools.require.cjs

@@ -4,6 +4,7 @@ const react = require("react");
 const pluginBridge = require("@rozenite/plugin-bridge");
 const bootRecording = require("./boot-recording.cjs");
 const agentBridge = require("@rozenite/agent-bridge");
+const agentShared = require("@rozenite/agent-shared");
 const DEFAULT_PAGE_LIMIT = 20;
 const MAX_PAGE_LIMIT = 100;
 const HTTP_BUFFER_CAPACITY = 500;
@@ -623,121 +624,132 @@ const getNetworkActivityAgentState = /* @__PURE__ */ (() => {
     return instance;
   };
 })();
-const startRecordingTool = {
-  name: "startRecording",
-  description: "Start recording network activity in the fallback network activity plugin.",
-  inputSchema: {
-    type: "object",
-    properties: {}
-  }
-};
-const stopRecordingTool = {
-  name: "stopRecording",
-  description: "Stop recording network activity without clearing the captured plugin buffer.",
-  inputSchema: {
-    type: "object",
-    properties: {}
-  }
-};
-const getRecordingStatusTool = {
-  name: "getRecordingStatus",
-  description: "Return network activity plugin recording state and buffer metadata.",
-  inputSchema: {
-    type: "object",
-    properties: {}
-  }
-};
-const listRequestsTool = {
-  name: "listRequests",
-  description: "List captured HTTP request summaries with cursor pagination from the fallback plugin.",
-  inputSchema: {
-    type: "object",
-    properties: {
-      limit: {
-        type: "number",
-        description: "Maximum number of requests to return. Defaults to 20."
-      },
-      cursor: {
-        type: "string",
-        description: "Opaque pagination cursor from a previous listRequests call."
-      }
+const NETWORK_ACTIVITY_AGENT_PLUGIN_ID = "@rozenite/network-activity-plugin";
+const networkActivityToolDefinitions = {
+  startRecording: agentShared.defineAgentToolContract({
+    name: "startRecording",
+    description: "Start recording network activity in the fallback network activity plugin.",
+    inputSchema: {
+      type: "object",
+      properties: {}
     }
-  }
-};
-const getRequestDetailsTool = {
-  name: "getRequestDetails",
-  description: "Return detailed metadata for a captured HTTP request without fetching response body.",
-  inputSchema: {
-    type: "object",
-    properties: {
-      requestId: {
-        type: "string",
-        description: "Captured plugin request ID to inspect."
-      }
-    },
-    required: ["requestId"]
-  }
-};
-const getRequestBodyTool = {
-  name: "getRequestBody",
-  description: "Return the captured request body for a plugin-recorded HTTP request when available.",
-  inputSchema: {
-    type: "object",
-    properties: {
-      requestId: {
-        type: "string",
-        description: "Captured plugin request ID to inspect."
-      }
-    },
-    required: ["requestId"]
-  }
-};
-const getResponseBodyTool = {
-  name: "getResponseBody",
-  description: "Return the captured response body for a plugin-recorded HTTP request when available.",
-  inputSchema: {
-    type: "object",
-    properties: {
-      requestId: {
-        type: "string",
-        description: "Captured plugin request ID to inspect."
+  }),
+  stopRecording: agentShared.defineAgentToolContract({
+    name: "stopRecording",
+    description: "Stop recording network activity without clearing the captured plugin buffer.",
+    inputSchema: {
+      type: "object",
+      properties: {}
+    }
+  }),
+  getRecordingStatus: agentShared.defineAgentToolContract({
+    name: "getRecordingStatus",
+    description: "Return network activity plugin recording state and buffer metadata.",
+    inputSchema: {
+      type: "object",
+      properties: {}
+    }
+  }),
+  listRequests: agentShared.defineAgentToolContract({
+    name: "listRequests",
+    description: "List captured HTTP request summaries with cursor pagination from the fallback plugin.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        limit: {
+          type: "number",
+          description: "Maximum number of requests to return. Defaults to 20."
+        },
+        cursor: {
+          type: "string",
+          description: "Opaque pagination cursor from a previous listRequests call."
+        }
       }
-    },
-    required: ["requestId"]
-  }
-};
-const listRealtimeConnectionsTool = {
-  name: "listRealtimeConnections",
-  description: "List captured WebSocket and SSE connections with cursor pagination.",
-  inputSchema: {
-    type: "object",
-    properties: {
-      limit: {
-        type: "number",
-        description: "Maximum number of realtime connections to return. Defaults to 20."
+    }
+  }),
+  getRequestDetails: agentShared.defineAgentToolContract({
+    name: "getRequestDetails",
+    description: "Return detailed metadata for a captured HTTP request without fetching response body.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        requestId: {
+          type: "string",
+          description: "Captured plugin request ID to inspect."
+        }
       },
-      cursor: {
-        type: "string",
-        description: "Opaque pagination cursor from a previous listRealtimeConnections call."
-      }
+      required: ["requestId"]
     }
-  }
-};
-const getRealtimeConnectionDetailsTool = {
-  name: "getRealtimeConnectionDetails",
-  description: "Return details for a captured WebSocket or SSE connection, including recent messages.",
-  inputSchema: {
-    type: "object",
-    properties: {
-      requestId: {
-        type: "string",
-        description: "Captured realtime request ID to inspect."
+  }),
+  getRequestBody: agentShared.defineAgentToolContract({
+    name: "getRequestBody",
+    description: "Return the captured request body for a plugin-recorded HTTP request when available.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        requestId: {
+          type: "string",
+          description: "Captured plugin request ID to inspect."
+        }
+      },
+      required: ["requestId"]
+    }
+  }),
+  getResponseBody: agentShared.defineAgentToolContract({
+    name: "getResponseBody",
+    description: "Return the captured response body for a plugin-recorded HTTP request when available.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        requestId: {
+          type: "string",
+          description: "Captured plugin request ID to inspect."
+        }
+      },
+      required: ["requestId"]
+    }
+  }),
+  listRealtimeConnections: agentShared.defineAgentToolContract({
+    name: "listRealtimeConnections",
+    description: "List captured WebSocket and SSE connections with cursor pagination.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        limit: {
+          type: "number",
+          description: "Maximum number of realtime connections to return. Defaults to 20."
+        },
+        cursor: {
+          type: "string",
+          description: "Opaque pagination cursor from a previous listRealtimeConnections call."
+        }
       }
-    },
-    required: ["requestId"]
-  }
+    }
+  }),
+  getRealtimeConnectionDetails: agentShared.defineAgentToolContract({
+    name: "getRealtimeConnectionDetails",
+    description: "Return details for a captured WebSocket or SSE connection, including recent messages.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        requestId: {
+          type: "string",
+          description: "Captured realtime request ID to inspect."
+        }
+      },
+      required: ["requestId"]
+    }
+  })
 };
-const pluginId = "@rozenite/network-activity-plugin";
+const startRecordingTool = networkActivityToolDefinitions.startRecording;
+const stopRecordingTool = networkActivityToolDefinitions.stopRecording;
+const getRecordingStatusTool = networkActivityToolDefinitions.getRecordingStatus;
+const listRequestsTool = networkActivityToolDefinitions.listRequests;
+const getRequestDetailsTool = networkActivityToolDefinitions.getRequestDetails;
+const getRequestBodyTool = networkActivityToolDefinitions.getRequestBody;
+const getResponseBodyTool = networkActivityToolDefinitions.getResponseBody;
+const listRealtimeConnectionsTool = networkActivityToolDefinitions.listRealtimeConnections;
+const getRealtimeConnectionDetailsTool = networkActivityToolDefinitions.getRealtimeConnectionDetails;
 const useNetworkActivityAgentTools = ({
   client,
   networkInspector,
@@ -766,6 +778,22 @@ const useNetworkActivityAgentTools = ({
         "request-failed",
         (event) => state.onRequestFailed(event)
       ),
+      networkInspector.nitro.on(
+        "request-sent",
+        (event) => state.onRequestSent(event)
+      ),
+      networkInspector.nitro.on(
+        "response-received",
+        (event) => state.onResponseReceived(event)
+      ),
+      networkInspector.nitro.on(
+        "request-completed",
+        (event) => state.onRequestCompleted(event)
+      ),
+      networkInspector.nitro.on(
+        "request-failed",
+        (event) => state.onRequestFailed(event)
+      ),
       networkInspector.websocket.on(
         "websocket-connect",
         (event) => state.onWebSocketConnect(event)
@@ -794,8 +822,35 @@ const useNetworkActivityAgentTools = ({
         "websocket-connection-status-changed",
         (event) => state.onWebSocketConnectionStatusChanged(event)
       ),
+      networkInspector.nitro.on(
+        "websocket-connect",
+        (event) => state.onWebSocketConnect(event)
+      ),
+      networkInspector.nitro.on(
+        "websocket-open",
+        (event) => state.onWebSocketOpen(event)
+      ),
+      networkInspector.nitro.on(
+        "websocket-close",
+        (event) => state.onWebSocketClose(event)
+      ),
+      networkInspector.nitro.on(
+        "websocket-message-sent",
+        (event) => state.onWebSocketMessageSent(event)
+      ),
+      networkInspector.nitro.on(
+        "websocket-message-received",
+        (event) => state.onWebSocketMessageReceived(event)
+      ),
+      networkInspector.nitro.on(
+        "websocket-error",
+        (event) => state.onWebSocketError(event)
+      ),
       networkInspector.sse.on("sse-open", (event) => state.onSSEOpen(event)),
-      networkInspector.sse.on("sse-message", (event) => state.onSSEMessage(event)),
+      networkInspector.sse.on(
+        "sse-message",
+        (event) => state.onSSEMessage(event)
+      ),
       networkInspector.sse.on("sse-error", (event) => state.onSSEError(event)),
       networkInspector.sse.on("sse-close", (event) => state.onSSEClose(event))
     ];
@@ -822,7 +877,7 @@ const useNetworkActivityAgentTools = ({
     };
   }, [client, enabledInspectors, state]);
   agentBridge.useRozenitePluginAgentTool({
-    pluginId,
+    pluginId: NETWORK_ACTIVITY_AGENT_PLUGIN_ID,
     tool: startRecordingTool,
     handler: () => {
       networkInspector.http.getNetworkRequestsRegistry().clear();
@@ -835,7 +890,7 @@ const useNetworkActivityAgentTools = ({
     }
   });
   agentBridge.useRozenitePluginAgentTool({
-    pluginId,
+    pluginId: NETWORK_ACTIVITY_AGENT_PLUGIN_ID,
     tool: stopRecordingTool,
     handler: () => {
       const result = state.stopRecording();
@@ -847,27 +902,27 @@ const useNetworkActivityAgentTools = ({
     }
   });
   agentBridge.useRozenitePluginAgentTool({
-    pluginId,
+    pluginId: NETWORK_ACTIVITY_AGENT_PLUGIN_ID,
     tool: getRecordingStatusTool,
     handler: () => state.getStatus()
   });
   agentBridge.useRozenitePluginAgentTool({
-    pluginId,
+    pluginId: NETWORK_ACTIVITY_AGENT_PLUGIN_ID,
     tool: listRequestsTool,
     handler: (input = {}) => state.listRequests(input)
   });
   agentBridge.useRozenitePluginAgentTool({
-    pluginId,
+    pluginId: NETWORK_ACTIVITY_AGENT_PLUGIN_ID,
     tool: getRequestDetailsTool,
     handler: ({ requestId }) => state.getRequestDetails(requestId)
   });
   agentBridge.useRozenitePluginAgentTool({
-    pluginId,
+    pluginId: NETWORK_ACTIVITY_AGENT_PLUGIN_ID,
     tool: getRequestBodyTool,
     handler: ({ requestId }) => state.getRequestBody(requestId)
   });
   agentBridge.useRozenitePluginAgentTool({
-    pluginId,
+    pluginId: NETWORK_ACTIVITY_AGENT_PLUGIN_ID,
     tool: getResponseBodyTool,
     handler: async ({ requestId }) => {
       const record = state.getHttpRecord(requestId);
@@ -888,15 +943,7 @@ const useNetworkActivityAgentTools = ({
           reason: "Response body is unavailable until the request finishes loading."
         };
       }
-      const request = networkInspector.http.getNetworkRequestsRegistry().getEntry(requestId);
-      if (!request) {
-        return {
-          requestId,
-          available: false,
-          reason: "Response body is unavailable because the request object is no longer in the plugin registry."
-        };
-      }
-      const body = await bootRecording.getResponseBody(request);
+      const body = await networkInspector.getResponseBody(requestId);
       if (body === null) {
         return {
           requestId,
@@ -915,39 +962,34 @@ const useNetworkActivityAgentTools = ({
     }
   });
   agentBridge.useRozenitePluginAgentTool({
-    pluginId,
+    pluginId: NETWORK_ACTIVITY_AGENT_PLUGIN_ID,
     tool: listRealtimeConnectionsTool,
     handler: (input = {}) => state.listRealtimeConnections(input)
   });
   agentBridge.useRozenitePluginAgentTool({
-    pluginId,
+    pluginId: NETWORK_ACTIVITY_AGENT_PLUGIN_ID,
     tool: getRealtimeConnectionDetailsTool,
     handler: ({ requestId }) => state.getRealtimeConnectionDetails(requestId)
   });
 };
 const overridesRegistry = bootRecording.getOverridesRegistry();
-const useHttpInspector = (client, httpInspector, isEnabled, isRecordingEnabled) => {
+const useHttpInspector = (client, networkInspector, isEnabled, isRecordingEnabled) => {
   react.useEffect(() => {
     if (!client || !isEnabled) {
       return;
     }
-    const networkRequestsRegistry = httpInspector.getNetworkRequestsRegistry();
     const subscriptions = [
       client.onMessage("network-enable", () => {
-        httpInspector.enable();
+        networkInspector.http.enable();
       }),
       client.onMessage("network-disable", () => {
-        httpInspector.disable();
+        networkInspector.http.disable();
       }),
       client.onMessage("set-overrides", (data) => {
         overridesRegistry.setOverrides(data.overrides);
       }),
       client.onMessage("get-response-body", async ({ requestId }) => {
-        const request = networkRequestsRegistry.getEntry(requestId);
-        if (!request) {
-          return;
-        }
-        const body = await bootRecording.getResponseBody(request);
+        const body = await networkInspector.getResponseBody(requestId);
         client.send("response-body", {
           requestId,
           body
@@ -955,13 +997,13 @@ const useHttpInspector = (client, httpInspector, isEnabled, isRecordingEnabled)
       })
     ];
     if (isRecordingEnabled) {
-      httpInspector.enable();
+      networkInspector.http.enable();
     }
     return () => {
       subscriptions.forEach((subscription) => subscription.remove());
-      httpInspector.dispose();
+      networkInspector.http.dispose();
     };
-  }, [client, httpInspector, isEnabled, isRecordingEnabled]);
+  }, [client, networkInspector, isEnabled, isRecordingEnabled]);
 };
 const useWebSocketInspector = (client, websocketInspector, isEnabled, isRecordingEnabled) => {
   react.useEffect(() => {
@@ -1047,6 +1089,11 @@ const useNetworkActivityDevTools = (config = bootRecording.DEFAULT_CONFIG) => {
     const subscriptions = [
       client.onMessage("network-enable", () => {
         isRecordingEnabledRef.current = true;
+        networkInspector.enable({
+          http: isHttpInspectorEnabled,
+          websocket: isWebSocketInspectorEnabled,
+          sse: isSSEInspectorEnabled
+        });
         eventsListener.connect(client.send, (message) => {
           const type = message.type;
           if (bootRecording.isHttpEvent(type)) {
@@ -1063,11 +1110,15 @@ const useNetworkActivityDevTools = (config = bootRecording.DEFAULT_CONFIG) => {
       }),
       client.onMessage("network-disable", () => {
         isRecordingEnabledRef.current = false;
+        networkInspector.disable();
       }),
       client.onMessage("get-client-ui-settings", () => {
         sendClientUISettings();
       })
     ];
+    client.send("recording-state", {
+      isRecording: isRecordingEnabledRef.current
+    });
     sendClientUISettings();
     return () => {
       subscriptions.forEach((subscription) => subscription.remove());
@@ -1081,7 +1132,7 @@ const useNetworkActivityDevTools = (config = bootRecording.DEFAULT_CONFIG) => {
   ]);
   useHttpInspector(
     client,
-    networkInspector.http,
+    networkInspector,
     isHttpInspectorEnabled,
     isRecordingEnabledRef.current
   );