rentgen/sidebar/stolen-data-cluster.tsx

import React from "react";
import { getMemory } from "../memory";
import { RequestCluster } from "../request-cluster";
import {
  MergedStolenDataEntry,
  Sources,
  StolenDataEntry,
} from "../stolen-data-entry";

import { hyphenate, useEmitter } from "../util";

const MAX_STRING_VALUE_LENGTH = 100;

function StolenDataValueTable({
  entry,
  prefixKey = "",
}: {
  entry: MergedStolenDataEntry;
  prefixKey: string;
}) {
  return (
    <table>
      <tbody>
        {Object.keys(entry.getParsedValues(prefixKey)[0]).map((key) => {
          const subkey = `${prefixKey}.${key}`;
          return (
            <tr key={`${prefixKey}.${key}`}>
              <th
                onClick={(e) => {
                  entry.toggleMark(subkey);
                  e.stopPropagation();
                }}
                style={{
                  border: entry.hasMark(subkey)
                    ? "2px solid red"
                    : "2px solid transparent",
                }}
              >
                {hyphenate(key)}
              </th>
              <td>
                <StolenDataValue entry={entry} prefixKey={subkey} />
              </td>
            </tr>
          );
        })}
      </tbody>
    </table>
  );
}

function StolenDataValue({
  entry,
  prefixKey = "",
}: {
  entry: MergedStolenDataEntry;
  prefixKey?: string;
}) {
  const [version] = useEmitter(entry);
  const value = entry.getParsedValues(prefixKey)[0];
  let body = null;
  if (!value) {
    body = <></>;
  } else if (typeof value === "string") {
    const content = entry.getParsedValues(prefixKey)[0] as string;
    body = (
      <div
        style={{
          border: entry.hasMark(prefixKey)
            ? "2px solid red"
            : "2px solid transparent",
        }}
        data-version={version}
      >
        {content.slice(0, MAX_STRING_VALUE_LENGTH)}{" "}
        {content.length > MAX_STRING_VALUE_LENGTH ? "(...)" : ""}
      </div>
    );
  } else {
    body = <StolenDataValueTable entry={entry} prefixKey={prefixKey} />;
  }
  return (
    <div
      onClick={(e) => {
        entry.toggleMark(prefixKey);
        e.stopPropagation();
      }}
      data-marks={entry.getMarkedValues().join(", ")}
    >
      {body}
    </div>
  );
}

const icons: Record<Sources, string> = {
  cookie: "🍪",
  pathname: "🛣",
  queryparams: "🅿",
  header: "H",
};

function StolenDataRow({
  entry,
  cluster,
}: {
  entry: MergedStolenDataEntry;
  cluster: RequestCluster;
}) {
  const [version] = useEmitter(entry);
  return (
    <tr
      key={origin + cluster.id + entry.getUniqueKey()}
      data-key={origin + cluster.id + entry.getUniqueKey()}
      data-version={version}
    >
      <th
        style={{
          width: "100px",
          overflowWrap: "anywhere",
          border: entry.hasMark("") ? "2px solid red" : "2px solid transparent",
        }}
        onClick={() => entry.toggleMark("")}
      >
        {entry.getNames().map(hyphenate).join(", ")}
      </th>
      <td>{entry.getSources().map((source) => icons[source])}</td>
      <td style={{ wordWrap: "anywhere" as any }}>
        <StolenDataValue entry={entry} />
      </td>
    </tr>
  );
}

export default function StolenDataCluster({
  origin,
  shorthost,
  minValueLength,
  cookiesOnly,
  cookiesOrOriginOnly,
}: {
  origin: string;
  shorthost: string;
  refreshToken: number;
  minValueLength: number;
  cookiesOnly: boolean;
  cookiesOrOriginOnly: boolean;
}) {
  const cluster = getMemory().getClustersForOrigin(origin)[shorthost];
  return (
    <div>
      <h2>
        <a href={"https://" + cluster.id}>{cluster.id}</a>{" "}
        {cluster.hasCookies() ? "🍪" : ""} x{cluster.requests.length}{" "}
        <a
          href="#"
          style={{ fontSize: "10px" }}
          onClick={() => getMemory().removeCookiesFor(origin, shorthost)}
        >
          Wyczyść cookiesy
        </a>
      </h2>
      <div>
        {cluster.getFullHosts().map((host) => (
          <a href={`https://${host}`}>{host}, </a>
        ))}
      </div>
      <table>
        <tbody>
          {cluster
            .getStolenData({ minValueLength, cookiesOnly, cookiesOrOriginOnly })
            .map((entry) => (
              <StolenDataRow
                {...{
                  entry,
                  cluster,
                  key: origin + cluster.id + entry.getUniqueKey(),
                }}
              />
            ))}
        </tbody>
      </table>
    </div>
  );
}