@rozenite/network-activity-plugin 1.0.0-alpha.8 → 1.0.0-alpha.9

package/dist/useNetworkActivityDevTools.cjs

@@ -2,6 +2,12 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
 const react = require("react");
 const pluginBridge = require("@rozenite/plugin-bridge");
+const nanoevents = require("nanoevents");
+const reactNative = require("react-native");
+const WebSocketInterceptor = require("react-native/Libraries/WebSocket/WebSocketInterceptor");
+const eventSource = require("./event-source.cjs");
+const _interopDefault = (e) => e && e.__esModule ? e : { default: e };
+const WebSocketInterceptor__default = /* @__PURE__ */ _interopDefault(WebSocketInterceptor);
 function getHttpHeaderValue(headers, name) {
   const lowerName = name.toLowerCase();
   for (const key in headers) {
@@ -11,6 +17,26 @@ function getHttpHeaderValue(headers, name) {
   }
   return void 0;
 }
+const getContentType = (request) => {
+  const responseHeaders = request.responseHeaders;
+  const responseType = request.responseType;
+  const contentType = getHttpHeaderValue(responseHeaders || {}, "content-type");
+  if (contentType) {
+    return contentType.split(";")[0].trim();
+  }
+  switch (responseType) {
+    case "arraybuffer":
+    case "blob":
+      return "application/octet-stream";
+    case "text":
+    case "":
+      return "text/plain";
+    case "json":
+      return "application/json";
+    case "document":
+      return "text/html";
+  }
+};
 const REQUEST_TTL = 1e3 * 60 * 5;
 const getNetworkRequestsRegistry = () => {
   const registry = /* @__PURE__ */ new Map();
@@ -52,7 +78,7 @@ let sendCallback;
 let requestHeaderCallback;
 let headerReceivedCallback;
 let responseCallback;
-let isInterceptorEnabled = false;
+let isInterceptorEnabled$1 = false;
 const XHRInterceptor = {
   /**
    * Invoked before XMLHttpRequest.open(...) is called.
@@ -85,10 +111,10 @@ const XHRInterceptor = {
     requestHeaderCallback = callback;
   },
   isInterceptorEnabled() {
-    return isInterceptorEnabled;
+    return isInterceptorEnabled$1;
   },
   enableInterception() {
-    if (isInterceptorEnabled) {
+    if (isInterceptorEnabled$1) {
       return;
     }
     XMLHttpRequest.prototype.open = function(method, url) {
@@ -111,7 +137,7 @@ const XHRInterceptor = {
         this.addEventListener(
           "readystatechange",
           () => {
-            if (!isInterceptorEnabled) {
+            if (!isInterceptorEnabled$1) {
               return;
             }
             if (this.readyState === this.HEADERS_RECEIVED) {
@@ -151,14 +177,14 @@ const XHRInterceptor = {
       }
       originalXHRSend.apply(this, arguments);
     };
-    isInterceptorEnabled = true;
+    isInterceptorEnabled$1 = true;
   },
   // Unpatch XMLHttpRequest methods and remove the callbacks.
   disableInterception() {
-    if (!isInterceptorEnabled) {
+    if (!isInterceptorEnabled$1) {
       return;
     }
-    isInterceptorEnabled = false;
+    isInterceptorEnabled$1 = false;
     XMLHttpRequest.prototype.send = originalXHRSend;
     XMLHttpRequest.prototype.open = originalXHROpen;
     XMLHttpRequest.prototype.setRequestHeader = originalXHRSetRequestHeader;
@@ -170,26 +196,6 @@ const XHRInterceptor = {
   }
 };
 const networkRequestsRegistry = getNetworkRequestsRegistry();
-const getContentType = (request) => {
-  const responseHeaders = request.responseHeaders;
-  const responseType = request.responseType;
-  const contentType = getHttpHeaderValue(responseHeaders || {}, "content-type");
-  if (contentType) {
-    return contentType.split(";")[0].trim();
-  }
-  switch (responseType) {
-    case "arraybuffer":
-    case "blob":
-      return "application/octet-stream";
-    case "text":
-    case "":
-      return "text/plain";
-    case "json":
-      return "application/json";
-    case "document":
-      return "text/html";
-  }
-};
 const getResponseSize = (request) => {
   if (typeof request.response === "object") {
     return request.response.size;
@@ -198,7 +204,7 @@ const getResponseSize = (request) => {
 };
 const getResponseBody = async (request) => {
   const responseType = request.responseType;
-  if (responseType === "text") {
+  if (responseType === "" || responseType === "text") {
     return request.responseText;
   }
   if (responseType === "blob") {
@@ -243,12 +249,13 @@ const getNetworkInspector = (pluginClient) => {
   const handleRequestSend = (data, request) => {
     const sendTime = Date.now();
     const requestId = generateRequestId();
+    request._rozeniteRequestId = requestId;
     const initiator = getInitiatorFromStack();
     networkRequestsRegistry.addEntry(requestId, request);
     let ttfb = 0;
     pluginClient.send("request-sent", {
       requestId,
-      timestamp: sendTime / 1e3,
+      timestamp: sendTime,
       request: {
         url: request._url,
         method: request._method,
@@ -266,7 +273,7 @@ const getNetworkInspector = (pluginClient) => {
     request.addEventListener("load", () => {
       pluginClient.send("response-received", {
         requestId,
-        timestamp: Date.now() / 1e3,
+        timestamp: Date.now(),
         type: "XHR",
         response: {
           url: request._url,
@@ -275,14 +282,14 @@ const getNetworkInspector = (pluginClient) => {
           headers: request.responseHeaders || {},
           contentType: getContentType(request),
           size: getResponseSize(request),
-          responseTime: Date.now() / 1e3
+          responseTime: Date.now()
         }
       });
     });
     request.addEventListener("loadend", () => {
       pluginClient.send("request-completed", {
         requestId,
-        timestamp: Date.now() / 1e3,
+        timestamp: Date.now(),
         duration: Date.now() - sendTime,
         size: getResponseSize(request),
         ttfb
@@ -291,7 +298,7 @@ const getNetworkInspector = (pluginClient) => {
     request.addEventListener("error", () => {
       pluginClient.send("request-failed", {
         requestId,
-        timestamp: Date.now() / 1e3,
+        timestamp: Date.now(),
         type: "XHR",
         error: "Failed",
         canceled: false
@@ -300,7 +307,7 @@ const getNetworkInspector = (pluginClient) => {
     request.addEventListener("abort", () => {
       pluginClient.send("request-failed", {
         requestId,
-        timestamp: Date.now() / 1e3,
+        timestamp: Date.now(),
         type: "XHR",
         error: "Aborted",
         canceled: true
@@ -352,6 +359,333 @@ const getNetworkInspector = (pluginClient) => {
     dispose
   };
 };
+const getWebSocketInterceptor = () => {
+  if (reactNative.Platform.constants.reactNativeVersion.minor >= 79) {
+    return WebSocketInterceptor__default.default;
+  } else {
+    const WebSocketInterceptorPreRN079 = WebSocketInterceptor__default.default;
+    return {
+      ...WebSocketInterceptorPreRN079,
+      setOnMessageCallback: (callback) => {
+        WebSocketInterceptorPreRN079.setOnMessageCallback((socketId, data) => {
+          callback(data, socketId);
+        });
+      },
+      setOnCloseCallback: (callback) => {
+        WebSocketInterceptorPreRN079.setOnCloseCallback((error, socketId) => {
+          callback(socketId, error);
+        });
+      },
+      setOnErrorCallback: (callback) => {
+        WebSocketInterceptorPreRN079.setOnErrorCallback((error, socketId) => {
+          callback(socketId, error);
+        });
+      }
+    };
+  }
+};
+const getWebSocketInspector = () => {
+  const eventEmitter = nanoevents.createNanoEvents();
+  const socketUrlMap = /* @__PURE__ */ new Map();
+  const webSocketInterceptor = getWebSocketInterceptor();
+  return {
+    enable: () => {
+      webSocketInterceptor.setConnectCallback(
+        (url, protocols, options, socketId) => {
+          socketUrlMap.set(socketId, url);
+          const event = {
+            type: "websocket-connect",
+            url,
+            socketId,
+            timestamp: Date.now(),
+            protocols,
+            options
+          };
+          eventEmitter.emit("websocket-connect", event);
+        }
+      );
+      webSocketInterceptor.setCloseCallback(
+        (code, reason, socketId) => {
+          const url = socketUrlMap.get(socketId);
+          if (!url) {
+            return;
+          }
+          const event = {
+            type: "websocket-close",
+            url,
+            socketId,
+            timestamp: Date.now(),
+            code: code || 0,
+            reason: reason || void 0
+          };
+          eventEmitter.emit("websocket-close", event);
+          socketUrlMap.delete(socketId);
+        }
+      );
+      webSocketInterceptor.setOnMessageCallback(
+        (data, socketId) => {
+          const url = socketUrlMap.get(socketId);
+          if (!url) {
+            return;
+          }
+          const event = {
+            type: "websocket-message-received",
+            url,
+            socketId,
+            timestamp: Date.now(),
+            data,
+            messageType: typeof data === "string" ? "text" : "binary"
+          };
+          eventEmitter.emit("websocket-message-received", event);
+        }
+      );
+      webSocketInterceptor.setOnErrorCallback(
+        (error, socketId) => {
+          const url = socketUrlMap.get(socketId);
+          if (!url) {
+            return;
+          }
+          const event = {
+            type: "websocket-error",
+            url,
+            socketId,
+            timestamp: Date.now(),
+            error
+          };
+          eventEmitter.emit("websocket-error", event);
+        }
+      );
+      webSocketInterceptor.setSendCallback((data, socketId) => {
+        const url = socketUrlMap.get(socketId);
+        if (!url) {
+          return;
+        }
+        const event = {
+          type: "websocket-message-sent",
+          url,
+          socketId,
+          timestamp: Date.now(),
+          data,
+          messageType: typeof data === "string" ? "text" : "binary"
+        };
+        eventEmitter.emit("websocket-message-sent", event);
+      });
+      webSocketInterceptor.setOnOpenCallback((socketId) => {
+        const url = socketUrlMap.get(socketId);
+        if (!url) {
+          return;
+        }
+        const event = {
+          type: "websocket-open",
+          url,
+          socketId,
+          timestamp: Date.now()
+        };
+        eventEmitter.emit("websocket-open", event);
+      });
+      webSocketInterceptor.setOnCloseCallback(
+        (error, socketId) => {
+          const url = socketUrlMap.get(socketId);
+          if (!url) {
+            return;
+          }
+          const event = {
+            type: "websocket-close",
+            url,
+            socketId,
+            timestamp: Date.now(),
+            code: error.code,
+            reason: error.reason
+          };
+          eventEmitter.emit("websocket-close", event);
+          socketUrlMap.delete(socketId);
+        }
+      );
+      webSocketInterceptor.enableInterception();
+    },
+    disable: () => {
+      webSocketInterceptor.disableInterception();
+    },
+    isEnabled: () => webSocketInterceptor.isInterceptorEnabled(),
+    dispose: () => {
+      eventEmitter.events = {};
+      socketUrlMap.clear();
+    },
+    on: (event, callback) => eventEmitter.on(event, callback)
+  };
+};
+let connectCallback;
+let messageCallback;
+let errorCallback;
+let openEventCallback;
+let closeCallback;
+let isInterceptorEnabled = false;
+const eventSourceClass = eventSource.getEventSource();
+const originalOpen = eventSourceClass.prototype.open;
+const SSEInterceptor = {
+  /**
+   * Invoked when EventSource.open() is called (connection attempt starting).
+   */
+  setConnectCallback(callback) {
+    connectCallback = callback;
+  },
+  /**
+   * Invoked when a message event is received.
+   */
+  setMessageCallback(callback) {
+    messageCallback = callback;
+  },
+  /**
+   * Invoked when an error event occurs.
+   */
+  setErrorCallback(callback) {
+    errorCallback = callback;
+  },
+  /**
+   * Invoked when the connection is successfully opened (open event fired).
+   */
+  setOpenEventCallback(callback) {
+    openEventCallback = callback;
+  },
+  /**
+   * Invoked when the connection is closed.
+   */
+  setCloseCallback(callback) {
+    closeCallback = callback;
+  },
+  isInterceptorEnabled() {
+    return isInterceptorEnabled;
+  },
+  enableInterception() {
+    if (isInterceptorEnabled) {
+      return;
+    }
+    eventSourceClass.prototype.open = function() {
+      if (connectCallback) {
+        connectCallback(this.url, this);
+      }
+      this.addEventListener("open", (event) => {
+        if (openEventCallback) {
+          openEventCallback(event, this);
+        }
+      });
+      this.addEventListener("message", (event) => {
+        if (messageCallback) {
+          messageCallback(event, this);
+        }
+      });
+      this.addEventListener(
+        "error",
+        (event) => {
+          if (errorCallback) {
+            errorCallback(event, this);
+          }
+        }
+      );
+      this.addEventListener("close", (event) => {
+        if (closeCallback) {
+          closeCallback(event, this);
+        }
+      });
+      return originalOpen.call(this);
+    };
+    isInterceptorEnabled = true;
+  },
+  // Unpatch EventSource open method and remove the callbacks.
+  disableInterception() {
+    if (!isInterceptorEnabled) {
+      return;
+    }
+    isInterceptorEnabled = false;
+    eventSourceClass.prototype.open = originalOpen;
+    connectCallback = null;
+    messageCallback = null;
+    errorCallback = null;
+    openEventCallback = null;
+    closeCallback = null;
+  }
+};
+const getSSEInspector = () => {
+  const eventEmitter = nanoevents.createNanoEvents();
+  const getRequestId = (eventSource2) => {
+    var _a;
+    const requestId = (_a = eventSource2._xhr) == null ? void 0 : _a._rozeniteRequestId;
+    if (!requestId) {
+      throw new Error(
+        "No request ID found for EventSource. This should never happen!"
+      );
+    }
+    return requestId;
+  };
+  return {
+    enable: () => {
+      SSEInterceptor.setOpenEventCallback((_, eventSource2) => {
+        const sseEventSource = eventSource2;
+        const requestId = getRequestId(sseEventSource);
+        const sseXhr = sseEventSource._xhr;
+        const event = {
+          type: "sse-open",
+          requestId,
+          timestamp: Date.now(),
+          response: {
+            url: sseXhr._url,
+            status: sseXhr.status,
+            statusText: sseXhr.statusText,
+            headers: sseXhr.responseHeaders || {},
+            contentType: getContentType(sseXhr),
+            size: 0,
+            responseTime: Date.now()
+          }
+        };
+        eventEmitter.emit("sse-open", event);
+      });
+      SSEInterceptor.setMessageCallback((messageEvent, eventSource2) => {
+        const sseEventSource = eventSource2;
+        const requestId = getRequestId(sseEventSource);
+        const event = {
+          type: "sse-message",
+          requestId,
+          timestamp: Date.now(),
+          data: messageEvent.data || ""
+        };
+        eventEmitter.emit("sse-message", event);
+      });
+      SSEInterceptor.setErrorCallback((errorEvent, eventSource2) => {
+        const sseEventSource = eventSource2;
+        const requestId = getRequestId(sseEventSource);
+        const event = {
+          type: "sse-error",
+          requestId,
+          timestamp: Date.now(),
+          error: {
+            type: errorEvent.type,
+            message: errorEvent.type === "timeout" ? "Timeout" : errorEvent.message
+          }
+        };
+        eventEmitter.emit("sse-error", event);
+      });
+      SSEInterceptor.setCloseCallback((_, eventSource2) => {
+        const sseEventSource = eventSource2;
+        const requestId = getRequestId(sseEventSource);
+        const event = {
+          type: "sse-close",
+          requestId,
+          timestamp: Date.now()
+        };
+        eventEmitter.emit("sse-close", event);
+      });
+      SSEInterceptor.enableInterception();
+    },
+    disable: () => {
+      SSEInterceptor.disableInterception();
+    },
+    isEnabled: () => SSEInterceptor.isInterceptorEnabled(),
+    dispose: () => {
+      eventEmitter.events = {};
+    },
+    on: (event, callback) => eventEmitter.on(event, callback)
+  };
+};
 const useNetworkActivityDevTools = () => {
   const client = pluginBridge.useRozeniteDevToolsClient({
     pluginId: "@rozenite/network-activity-plugin"
@@ -365,6 +699,61 @@ const useNetworkActivityDevTools = () => {
       networkInspector.dispose();
     };
   }, [client]);
+  react.useEffect(() => {
+    if (!client) {
+      return;
+    }
+    const eventsToForward = [
+      "websocket-connect",
+      "websocket-open",
+      "websocket-close",
+      "websocket-message-sent",
+      "websocket-message-received",
+      "websocket-error",
+      "websocket-connection-status-changed"
+    ];
+    const websocketInspector = getWebSocketInspector();
+    eventsToForward.forEach((event) => {
+      websocketInspector.on(event, (event2) => {
+        client.send(event2.type, event2);
+      });
+    });
+    client.onMessage("network-enable", () => {
+      websocketInspector.enable();
+    });
+    client.onMessage("network-disable", () => {
+      websocketInspector.disable();
+    });
+    return () => {
+      websocketInspector.dispose();
+    };
+  }, [client]);
+  react.useEffect(() => {
+    if (!client) {
+      return;
+    }
+    const eventsToForward = [
+      "sse-open",
+      "sse-message",
+      "sse-error",
+      "sse-close"
+    ];
+    const sseInspector = getSSEInspector();
+    eventsToForward.forEach((event) => {
+      sseInspector.on(event, (event2) => {
+        client.send(event2.type, event2);
+      });
+    });
+    client.onMessage("network-enable", () => {
+      sseInspector.enable();
+    });
+    client.onMessage("network-disable", () => {
+      sseInspector.disable();
+    });
+    return () => {
+      sseInspector.dispose();
+    };
+  }, [client]);
   return client;
 };
 exports.useNetworkActivityDevTools = useNetworkActivityDevTools;