@rozenite/network-activity-plugin 1.0.0-alpha.4 → 1.0.0-alpha.6

package/dist/useNetworkActivityDevTools.cjs

@@ -0,0 +1,360 @@
+"use strict";
+Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+const react = require("react");
+const pluginBridge = require("@rozenite/plugin-bridge");
+const REQUEST_TTL = 1e3 * 60 * 5;
+const getNetworkRequestsRegistry = () => {
+  const registry = /* @__PURE__ */ new Map();
+  const trimRegistry = () => {
+    const now = Date.now();
+    registry.forEach((entry) => {
+      if (now - entry.sentAt < REQUEST_TTL) {
+        return;
+      }
+      registry.delete(entry.id);
+    });
+  };
+  const addEntry = (id, request) => {
+    trimRegistry();
+    registry.set(id, {
+      id,
+      request,
+      sentAt: Date.now()
+    });
+  };
+  const getEntry = (id) => {
+    var _a;
+    return ((_a = registry.get(id)) == null ? void 0 : _a.request) ?? null;
+  };
+  const clear = () => {
+    registry.clear();
+  };
+  return {
+    addEntry,
+    getEntry,
+    clear
+  };
+};
+const originalXHROpen = XMLHttpRequest.prototype.open;
+const originalXHRSend = XMLHttpRequest.prototype.send;
+const originalXHRSetRequestHeader = XMLHttpRequest.prototype.setRequestHeader;
+let openCallback;
+let sendCallback;
+let requestHeaderCallback;
+let headerReceivedCallback;
+let responseCallback;
+let isInterceptorEnabled = false;
+const XHRInterceptor = {
+  /**
+   * Invoked before XMLHttpRequest.open(...) is called.
+   */
+  setOpenCallback(callback) {
+    openCallback = callback;
+  },
+  /**
+   * Invoked before XMLHttpRequest.send(...) is called.
+   */
+  setSendCallback(callback) {
+    sendCallback = callback;
+  },
+  /**
+   * Invoked after xhr's readyState becomes xhr.HEADERS_RECEIVED.
+   */
+  setHeaderReceivedCallback(callback) {
+    headerReceivedCallback = callback;
+  },
+  /**
+   * Invoked after xhr's readyState becomes xhr.DONE.
+   */
+  setResponseCallback(callback) {
+    responseCallback = callback;
+  },
+  /**
+   * Invoked before XMLHttpRequest.setRequestHeader(...) is called.
+   */
+  setRequestHeaderCallback(callback) {
+    requestHeaderCallback = callback;
+  },
+  isInterceptorEnabled() {
+    return isInterceptorEnabled;
+  },
+  enableInterception() {
+    if (isInterceptorEnabled) {
+      return;
+    }
+    XMLHttpRequest.prototype.open = function(method, url) {
+      if (openCallback) {
+        openCallback(method, url, this);
+      }
+      originalXHROpen.apply(this, arguments);
+    };
+    XMLHttpRequest.prototype.setRequestHeader = function(header, value) {
+      if (requestHeaderCallback) {
+        requestHeaderCallback(header, value, this);
+      }
+      originalXHRSetRequestHeader.apply(this, arguments);
+    };
+    XMLHttpRequest.prototype.send = function(data) {
+      if (sendCallback) {
+        sendCallback(data, this);
+      }
+      if (this.addEventListener) {
+        this.addEventListener(
+          "readystatechange",
+          () => {
+            if (!isInterceptorEnabled) {
+              return;
+            }
+            if (this.readyState === this.HEADERS_RECEIVED) {
+              const contentTypeString = this.getResponseHeader("Content-Type");
+              const contentLengthString = this.getResponseHeader("Content-Length");
+              let responseContentType, responseSize;
+              if (contentTypeString) {
+                responseContentType = contentTypeString.split(";")[0];
+              }
+              if (contentLengthString) {
+                responseSize = parseInt(contentLengthString, 10);
+              }
+              if (headerReceivedCallback) {
+                headerReceivedCallback(
+                  responseContentType,
+                  responseSize,
+                  this.getAllResponseHeaders(),
+                  this
+                );
+              }
+            }
+            if (this.readyState === this.DONE) {
+              if (responseCallback) {
+                responseCallback(
+                  this.status,
+                  this.timeout,
+                  this.response,
+                  this.responseURL,
+                  this.responseType,
+                  this
+                );
+              }
+            }
+          },
+          false
+        );
+      }
+      originalXHRSend.apply(this, arguments);
+    };
+    isInterceptorEnabled = true;
+  },
+  // Unpatch XMLHttpRequest methods and remove the callbacks.
+  disableInterception() {
+    if (!isInterceptorEnabled) {
+      return;
+    }
+    isInterceptorEnabled = false;
+    XMLHttpRequest.prototype.send = originalXHRSend;
+    XMLHttpRequest.prototype.open = originalXHROpen;
+    XMLHttpRequest.prototype.setRequestHeader = originalXHRSetRequestHeader;
+    responseCallback = null;
+    openCallback = null;
+    sendCallback = null;
+    headerReceivedCallback = null;
+    requestHeaderCallback = null;
+  }
+};
+const networkRequestsRegistry = getNetworkRequestsRegistry();
+const getContentType = (request) => {
+  const responseHeaders = request.responseHeaders;
+  const responseType = request.responseType;
+  if (responseHeaders == null ? void 0 : responseHeaders["content-type"]) {
+    return responseHeaders["content-type"].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;
+  }
+  return request.response.length || 0;
+};
+const getResponseBody = async (request) => {
+  const responseType = request.responseType;
+  if (responseType === "text") {
+    return request.responseText;
+  }
+  if (responseType === "blob") {
+    const contentType = request.getResponseHeader("Content-Type") || "";
+    if (contentType.startsWith("text/") || contentType.startsWith("application/json")) {
+      return new Promise((resolve) => {
+        const reader = new FileReader();
+        reader.onload = () => {
+          resolve(reader.result);
+        };
+        reader.readAsText(request.response);
+      });
+    }
+  }
+  return null;
+};
+const getInitiatorFromStack = () => {
+  try {
+    const stack = new Error().stack;
+    if (!stack) {
+      return { type: "other" };
+    }
+    const line = stack.split("\n")[9];
+    const match = line.match(/at\s+(.+?)\s+\((.+?):(\d+):(\d+)\)/);
+    if (match) {
+      return {
+        type: "script",
+        url: match[2],
+        lineNumber: parseInt(match[3]),
+        columnNumber: parseInt(match[4])
+      };
+    }
+  } catch {
+  }
+  return { type: "other" };
+};
+const READY_STATE_HEADERS_RECEIVED = 2;
+const getNetworkInspector = (pluginClient) => {
+  const generateRequestId = () => {
+    return `req_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
+  };
+  const handleRequestSend = (data, request) => {
+    const sendTime = Date.now();
+    const requestId = generateRequestId();
+    const initiator = getInitiatorFromStack();
+    networkRequestsRegistry.addEntry(requestId, request);
+    let ttfb = 0;
+    pluginClient.send("request-sent", {
+      requestId,
+      timestamp: sendTime / 1e3,
+      request: {
+        url: request._url,
+        method: request._method,
+        headers: request._headers,
+        postData: data
+      },
+      type: "XHR",
+      initiator
+    });
+    request.addEventListener("readystatechange", () => {
+      if (request.readyState === READY_STATE_HEADERS_RECEIVED) {
+        ttfb = Date.now() - sendTime;
+      }
+    });
+    request.addEventListener("load", () => {
+      pluginClient.send("response-received", {
+        requestId,
+        timestamp: Date.now() / 1e3,
+        type: "XHR",
+        response: {
+          url: request._url,
+          status: request.status,
+          statusText: request.statusText,
+          headers: request.responseHeaders,
+          contentType: getContentType(request),
+          size: getResponseSize(request),
+          responseTime: Date.now() / 1e3
+        }
+      });
+    });
+    request.addEventListener("loadend", () => {
+      pluginClient.send("request-completed", {
+        requestId,
+        timestamp: Date.now() / 1e3,
+        duration: Date.now() - sendTime,
+        size: getResponseSize(request),
+        ttfb
+      });
+    });
+    request.addEventListener("error", () => {
+      pluginClient.send("request-failed", {
+        requestId,
+        timestamp: Date.now() / 1e3,
+        type: "XHR",
+        error: "Failed",
+        canceled: false
+      });
+    });
+    request.addEventListener("abort", () => {
+      pluginClient.send("request-failed", {
+        requestId,
+        timestamp: Date.now() / 1e3,
+        type: "XHR",
+        error: "Aborted",
+        canceled: true
+      });
+    });
+  };
+  const enable = () => {
+    XHRInterceptor.disableInterception();
+    XHRInterceptor.setSendCallback(handleRequestSend);
+    XHRInterceptor.enableInterception();
+  };
+  const disable = () => {
+    XHRInterceptor.disableInterception();
+    networkRequestsRegistry.clear();
+  };
+  const isEnabled = () => {
+    return XHRInterceptor.isInterceptorEnabled();
+  };
+  const enableSubscription = pluginClient.onMessage("network-enable", () => {
+    enable();
+  });
+  const disableSubscription = pluginClient.onMessage("network-disable", () => {
+    disable();
+  });
+  const handleBodySubscription = pluginClient.onMessage(
+    "get-response-body",
+    async ({ requestId }) => {
+      const request = networkRequestsRegistry.getEntry(requestId);
+      if (!request) {
+        return;
+      }
+      const body = await getResponseBody(request);
+      pluginClient.send("response-body", {
+        requestId,
+        body
+      });
+    }
+  );
+  const dispose = () => {
+    disable();
+    enableSubscription.remove();
+    disableSubscription.remove();
+    handleBodySubscription.remove();
+  };
+  return {
+    enable,
+    disable,
+    isEnabled,
+    dispose
+  };
+};
+const useNetworkActivityDevTools = () => {
+  const client = pluginBridge.useRozeniteDevToolsClient({
+    pluginId: "@rozenite/network-activity-plugin"
+  });
+  react.useEffect(() => {
+    if (!client) {
+      return;
+    }
+    const networkInspector = getNetworkInspector(client);
+    return () => {
+      networkInspector.dispose();
+    };
+  }, [client]);
+  return client;
+};
+exports.useNetworkActivityDevTools = useNetworkActivityDevTools;