playwright-codegen-pro-core 1.0.0

package/lib/utils/isomorphic/trace/snapshotStorage.js

@@ -0,0 +1,89 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var snapshotStorage_exports = {};
+__export(snapshotStorage_exports, {
+  SnapshotStorage: () => SnapshotStorage
+});
+module.exports = __toCommonJS(snapshotStorage_exports);
+var import_snapshotRenderer = require("./snapshotRenderer");
+var import_lruCache = require("../lruCache");
+class SnapshotStorage {
+  constructor() {
+    this._frameSnapshots = /* @__PURE__ */ new Map();
+    this._cache = new import_lruCache.LRUCache(1e8);
+    // 100MB per each trace
+    this._contextToResources = /* @__PURE__ */ new Map();
+    this._resourceUrlsWithOverrides = /* @__PURE__ */ new Set();
+  }
+  addResource(contextId, resource) {
+    resource.request.url = (0, import_snapshotRenderer.rewriteURLForCustomProtocol)(resource.request.url);
+    this._ensureResourcesForContext(contextId).push(resource);
+  }
+  addFrameSnapshot(contextId, snapshot, screencastFrames) {
+    for (const override of snapshot.resourceOverrides)
+      override.url = (0, import_snapshotRenderer.rewriteURLForCustomProtocol)(override.url);
+    let frameSnapshots = this._frameSnapshots.get(snapshot.frameId);
+    if (!frameSnapshots) {
+      frameSnapshots = {
+        raw: [],
+        renderers: []
+      };
+      this._frameSnapshots.set(snapshot.frameId, frameSnapshots);
+      if (snapshot.isMainFrame)
+        this._frameSnapshots.set(snapshot.pageId, frameSnapshots);
+    }
+    frameSnapshots.raw.push(snapshot);
+    const resources = this._ensureResourcesForContext(contextId);
+    const renderer = new import_snapshotRenderer.SnapshotRenderer(this._cache, resources, frameSnapshots.raw, screencastFrames, frameSnapshots.raw.length - 1);
+    frameSnapshots.renderers.push(renderer);
+    return renderer;
+  }
+  snapshotByName(pageOrFrameId, snapshotName) {
+    const snapshot = this._frameSnapshots.get(pageOrFrameId);
+    return snapshot?.renderers.find((r) => r.snapshotName === snapshotName);
+  }
+  snapshotsForTest() {
+    return [...this._frameSnapshots.keys()];
+  }
+  finalize() {
+    for (const resources of this._contextToResources.values())
+      resources.sort((a, b) => (a._monotonicTime || 0) - (b._monotonicTime || 0));
+    for (const frameSnapshots of this._frameSnapshots.values()) {
+      for (const snapshot of frameSnapshots.raw) {
+        for (const override of snapshot.resourceOverrides)
+          this._resourceUrlsWithOverrides.add(override.url);
+      }
+    }
+  }
+  hasResourceOverride(url) {
+    return this._resourceUrlsWithOverrides.has(url);
+  }
+  _ensureResourcesForContext(contextId) {
+    let resources = this._contextToResources.get(contextId);
+    if (!resources) {
+      resources = [];
+      this._contextToResources.set(contextId, resources);
+    }
+    return resources;
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  SnapshotStorage
+});

package/lib/utils/isomorphic/trace/traceLoader.js

@@ -0,0 +1,131 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var traceLoader_exports = {};
+__export(traceLoader_exports, {
+  TraceLoader: () => TraceLoader
+});
+module.exports = __toCommonJS(traceLoader_exports);
+var import_traceUtils = require("./traceUtils");
+var import_snapshotStorage = require("./snapshotStorage");
+var import_traceModernizer = require("./traceModernizer");
+class TraceLoader {
+  constructor() {
+    this.contextEntries = [];
+    this._resourceToContentType = /* @__PURE__ */ new Map();
+  }
+  async load(backend, unzipProgress) {
+    this._backend = backend;
+    const ordinals = [];
+    let hasSource = false;
+    for (const entryName of await this._backend.entryNames()) {
+      const match = entryName.match(/(.+)\.trace$/);
+      if (match)
+        ordinals.push(match[1] || "");
+      if (entryName.includes("src@"))
+        hasSource = true;
+    }
+    if (!ordinals.length)
+      throw new Error("Cannot find .trace file");
+    this._snapshotStorage = new import_snapshotStorage.SnapshotStorage();
+    const total = ordinals.length * 3;
+    let done = 0;
+    for (const ordinal of ordinals) {
+      const contextEntry = createEmptyContext();
+      contextEntry.hasSource = hasSource;
+      const modernizer = new import_traceModernizer.TraceModernizer(contextEntry, this._snapshotStorage);
+      const trace = await this._backend.readText(ordinal + ".trace") || "";
+      modernizer.appendTrace(trace);
+      unzipProgress(++done, total);
+      const network = await this._backend.readText(ordinal + ".network") || "";
+      modernizer.appendTrace(network);
+      unzipProgress(++done, total);
+      contextEntry.actions = modernizer.actions().sort((a1, a2) => a1.startTime - a2.startTime);
+      if (!backend.isLive()) {
+        for (const action of contextEntry.actions.slice().reverse()) {
+          if (!action.endTime && !action.error) {
+            for (const a of contextEntry.actions) {
+              if (a.parentId === action.callId && action.endTime < a.endTime)
+                action.endTime = a.endTime;
+            }
+          }
+        }
+      }
+      const stacks = await this._backend.readText(ordinal + ".stacks");
+      if (stacks) {
+        const callMetadata = (0, import_traceUtils.parseClientSideCallMetadata)(JSON.parse(stacks));
+        for (const action of contextEntry.actions)
+          action.stack = action.stack || callMetadata.get(action.callId);
+      }
+      unzipProgress(++done, total);
+      for (const resource of contextEntry.resources) {
+        if (resource.request.postData?._sha1)
+          this._resourceToContentType.set(resource.request.postData._sha1, stripEncodingFromContentType(resource.request.postData.mimeType));
+        if (resource.response.content?._sha1)
+          this._resourceToContentType.set(resource.response.content._sha1, stripEncodingFromContentType(resource.response.content.mimeType));
+      }
+      this.contextEntries.push(contextEntry);
+    }
+    this._snapshotStorage.finalize();
+  }
+  async hasEntry(filename) {
+    return this._backend.hasEntry(filename);
+  }
+  async resourceForSha1(sha1) {
+    const blob = await this._backend.readBlob("resources/" + sha1);
+    const contentType = this._resourceToContentType.get(sha1);
+    if (!blob || contentType === void 0 || contentType === "x-unknown")
+      return blob;
+    return new Blob([blob], { type: contentType });
+  }
+  storage() {
+    return this._snapshotStorage;
+  }
+}
+function stripEncodingFromContentType(contentType) {
+  const charset = contentType.match(/^(.*);\s*charset=.*$/);
+  if (charset)
+    return charset[1];
+  return contentType;
+}
+function createEmptyContext() {
+  return {
+    origin: "testRunner",
+    startTime: Number.MAX_SAFE_INTEGER,
+    wallTime: Number.MAX_SAFE_INTEGER,
+    endTime: 0,
+    browserName: "",
+    options: {
+      deviceScaleFactor: 1,
+      isMobile: false,
+      viewport: { width: 1280, height: 800 }
+    },
+    pages: [],
+    resources: [],
+    actions: [],
+    events: [],
+    errors: [],
+    stdio: [],
+    hasSource: false,
+    contextId: ""
+  };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  TraceLoader
+});

package/lib/utils/isomorphic/trace/traceModel.js

@@ -0,0 +1,366 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var traceModel_exports = {};
+__export(traceModel_exports, {
+  TraceModel: () => TraceModel,
+  buildActionTree: () => buildActionTree,
+  context: () => context,
+  eventsForAction: () => eventsForAction,
+  nextActionByStartTime: () => nextActionByStartTime,
+  previousActionByEndTime: () => previousActionByEndTime,
+  stats: () => stats
+});
+module.exports = __toCommonJS(traceModel_exports);
+var import_protocolFormatter = require("../protocolFormatter");
+const contextSymbol = Symbol("context");
+const nextInContextSymbol = Symbol("nextInContext");
+const prevByEndTimeSymbol = Symbol("prevByEndTime");
+const nextByStartTimeSymbol = Symbol("nextByStartTime");
+const eventsSymbol = Symbol("events");
+class TraceModel {
+  constructor(traceUri, contexts) {
+    contexts.forEach((contextEntry) => indexModel(contextEntry));
+    const libraryContext = contexts.find((context2) => context2.origin === "library");
+    this.traceUri = traceUri;
+    this.browserName = libraryContext?.browserName || "";
+    this.sdkLanguage = libraryContext?.sdkLanguage;
+    this.channel = libraryContext?.channel;
+    this.testIdAttributeName = libraryContext?.testIdAttributeName;
+    this.platform = libraryContext?.platform || "";
+    this.playwrightVersion = contexts.find((c) => c.playwrightVersion)?.playwrightVersion;
+    this.title = libraryContext?.title || "";
+    this.options = libraryContext?.options || {};
+    this.testTimeout = contexts.find((c) => c.origin === "testRunner")?.testTimeout;
+    this.actions = mergeActionsAndUpdateTiming(contexts);
+    this.pages = [].concat(...contexts.map((c) => c.pages));
+    this.wallTime = contexts.map((c) => c.wallTime).reduce((prev, cur) => Math.min(prev || Number.MAX_VALUE, cur), Number.MAX_VALUE);
+    this.startTime = contexts.map((c) => c.startTime).reduce((prev, cur) => Math.min(prev, cur), Number.MAX_VALUE);
+    this.endTime = contexts.map((c) => c.endTime).reduce((prev, cur) => Math.max(prev, cur), Number.MIN_VALUE);
+    this.events = [].concat(...contexts.map((c) => c.events));
+    this.stdio = [].concat(...contexts.map((c) => c.stdio));
+    this.errors = [].concat(...contexts.map((c) => c.errors));
+    this.hasSource = contexts.some((c) => c.hasSource);
+    this.hasStepData = contexts.some((context2) => context2.origin === "testRunner");
+    this.resources = [...contexts.map((c) => c.resources)].flat().map((entry) => ({ ...entry, id: `${entry.pageref}-${entry.time}-${entry.request.url}` }));
+    this.attachments = this.actions.flatMap((action) => action.attachments?.map((attachment) => ({ ...attachment, callId: action.callId, traceUri })) ?? []);
+    this.visibleAttachments = this.attachments.filter((attachment) => !attachment.name.startsWith("_"));
+    this.events.sort((a1, a2) => a1.time - a2.time);
+    this.resources.sort((a1, a2) => a1._monotonicTime - a2._monotonicTime);
+    this.errorDescriptors = this.hasStepData ? this._errorDescriptorsFromTestRunner() : this._errorDescriptorsFromActions();
+    this.sources = collectSources(this.actions, this.errorDescriptors);
+    this.actionCounters = /* @__PURE__ */ new Map();
+    for (const action of this.actions) {
+      action.group = action.group ?? (0, import_protocolFormatter.getActionGroup)({ type: action.class, method: action.method });
+      if (action.group)
+        this.actionCounters.set(action.group, 1 + (this.actionCounters.get(action.group) || 0));
+    }
+  }
+  createRelativeUrl(path) {
+    const url = new URL("http://localhost/" + path);
+    url.searchParams.set("trace", this.traceUri);
+    return url.toString().substring("http://localhost/".length);
+  }
+  failedAction() {
+    return this.actions.findLast((a) => a.error);
+  }
+  filteredActions(actionsFilter) {
+    const filter = new Set(actionsFilter);
+    return this.actions.filter((action) => !action.group || filter.has(action.group));
+  }
+  renderActionTree(filter) {
+    const actions = this.filteredActions(filter ?? []);
+    const { rootItem } = buildActionTree(actions);
+    const actionTree = [];
+    const visit = (actionItem, indent) => {
+      const title = (0, import_protocolFormatter.renderTitleForCall)({ ...actionItem.action, type: actionItem.action.class });
+      actionTree.push(`${indent}${title || actionItem.id}`);
+      for (const child of actionItem.children)
+        visit(child, indent + "  ");
+    };
+    rootItem.children.forEach((a) => visit(a, ""));
+    return actionTree;
+  }
+  _errorDescriptorsFromActions() {
+    const errors = [];
+    for (const action of this.actions || []) {
+      if (!action.error?.message)
+        continue;
+      errors.push({
+        action,
+        stack: action.stack,
+        message: action.error.message
+      });
+    }
+    return errors;
+  }
+  _errorDescriptorsFromTestRunner() {
+    return this.errors.filter((e) => !!e.message).map((error, i) => ({
+      stack: error.stack,
+      message: error.message
+    }));
+  }
+}
+function indexModel(context2) {
+  for (const page of context2.pages)
+    page[contextSymbol] = context2;
+  for (let i = 0; i < context2.actions.length; ++i) {
+    const action = context2.actions[i];
+    action[contextSymbol] = context2;
+  }
+  let lastNonRouteAction = void 0;
+  for (let i = context2.actions.length - 1; i >= 0; i--) {
+    const action = context2.actions[i];
+    action[nextInContextSymbol] = lastNonRouteAction;
+    if (action.class !== "Route")
+      lastNonRouteAction = action;
+  }
+  for (const event of context2.events)
+    event[contextSymbol] = context2;
+  for (const resource of context2.resources)
+    resource[contextSymbol] = context2;
+}
+function mergeActionsAndUpdateTiming(contexts) {
+  const result = [];
+  const actions = mergeActionsAndUpdateTimingSameTrace(contexts);
+  result.push(...actions);
+  result.sort((a1, a2) => {
+    if (a2.parentId === a1.callId)
+      return 1;
+    if (a1.parentId === a2.callId)
+      return -1;
+    return a1.endTime - a2.endTime;
+  });
+  for (let i = 1; i < result.length; ++i)
+    result[i][prevByEndTimeSymbol] = result[i - 1];
+  result.sort((a1, a2) => {
+    if (a2.parentId === a1.callId)
+      return -1;
+    if (a1.parentId === a2.callId)
+      return 1;
+    return a1.startTime - a2.startTime;
+  });
+  for (let i = 0; i + 1 < result.length; ++i)
+    result[i][nextByStartTimeSymbol] = result[i + 1];
+  return result;
+}
+let lastTmpStepId = 0;
+function mergeActionsAndUpdateTimingSameTrace(contexts) {
+  const map = /* @__PURE__ */ new Map();
+  const libraryContexts = contexts.filter((context2) => context2.origin === "library");
+  const testRunnerContexts = contexts.filter((context2) => context2.origin === "testRunner");
+  if (!testRunnerContexts.length || !libraryContexts.length) {
+    return contexts.map((context2) => {
+      return context2.actions.map((action) => ({ ...action, context: context2 }));
+    }).flat();
+  }
+  for (const context2 of libraryContexts) {
+    for (const action of context2.actions) {
+      map.set(action.stepId || `tmp-step@${++lastTmpStepId}`, { ...action, context: context2 });
+    }
+  }
+  const delta = monotonicTimeDeltaBetweenLibraryAndRunner(testRunnerContexts, map);
+  if (delta)
+    adjustMonotonicTime(libraryContexts, delta);
+  const nonPrimaryIdToPrimaryId = /* @__PURE__ */ new Map();
+  for (const context2 of testRunnerContexts) {
+    for (const action of context2.actions) {
+      const existing = action.stepId && map.get(action.stepId);
+      if (existing) {
+        nonPrimaryIdToPrimaryId.set(action.callId, existing.callId);
+        if (action.error)
+          existing.error = action.error;
+        if (action.attachments)
+          existing.attachments = action.attachments;
+        if (action.annotations)
+          existing.annotations = action.annotations;
+        if (action.parentId)
+          existing.parentId = nonPrimaryIdToPrimaryId.get(action.parentId) ?? action.parentId;
+        if (action.group)
+          existing.group = action.group;
+        existing.startTime = action.startTime;
+        existing.endTime = action.endTime;
+        continue;
+      }
+      if (action.parentId)
+        action.parentId = nonPrimaryIdToPrimaryId.get(action.parentId) ?? action.parentId;
+      map.set(action.stepId || `tmp-step@${++lastTmpStepId}`, { ...action, context: context2 });
+    }
+  }
+  return [...map.values()];
+}
+function adjustMonotonicTime(contexts, monotonicTimeDelta) {
+  for (const context2 of contexts) {
+    context2.startTime += monotonicTimeDelta;
+    context2.endTime += monotonicTimeDelta;
+    for (const action of context2.actions) {
+      if (action.startTime)
+        action.startTime += monotonicTimeDelta;
+      if (action.endTime)
+        action.endTime += monotonicTimeDelta;
+    }
+    for (const event of context2.events)
+      event.time += monotonicTimeDelta;
+    for (const event of context2.stdio)
+      event.timestamp += monotonicTimeDelta;
+    for (const page of context2.pages) {
+      for (const frame of page.screencastFrames)
+        frame.timestamp += monotonicTimeDelta;
+    }
+    for (const resource of context2.resources) {
+      if (resource._monotonicTime)
+        resource._monotonicTime += monotonicTimeDelta;
+    }
+  }
+}
+function monotonicTimeDeltaBetweenLibraryAndRunner(nonPrimaryContexts, libraryActions) {
+  for (const context2 of nonPrimaryContexts) {
+    for (const action of context2.actions) {
+      if (!action.startTime)
+        continue;
+      const libraryAction = action.stepId ? libraryActions.get(action.stepId) : void 0;
+      if (libraryAction)
+        return action.startTime - libraryAction.startTime;
+    }
+  }
+  return 0;
+}
+function buildActionTree(actions) {
+  const itemMap = /* @__PURE__ */ new Map();
+  for (const action of actions) {
+    itemMap.set(action.callId, {
+      id: action.callId,
+      parent: void 0,
+      children: [],
+      action
+    });
+  }
+  const rootItem = { action: { ...kFakeRootAction }, id: "", parent: void 0, children: [] };
+  for (const item of itemMap.values()) {
+    rootItem.action.startTime = Math.min(rootItem.action.startTime, item.action.startTime);
+    rootItem.action.endTime = Math.max(rootItem.action.endTime, item.action.endTime);
+    const parent = item.action.parentId ? itemMap.get(item.action.parentId) || rootItem : rootItem;
+    parent.children.push(item);
+    item.parent = parent;
+  }
+  const inheritStack = (item) => {
+    for (const child of item.children) {
+      child.action.stack = child.action.stack ?? item.action.stack;
+      inheritStack(child);
+    }
+  };
+  inheritStack(rootItem);
+  return { rootItem, itemMap };
+}
+function context(action) {
+  return action[contextSymbol];
+}
+function nextInContext(action) {
+  return action[nextInContextSymbol];
+}
+function previousActionByEndTime(action) {
+  return action[prevByEndTimeSymbol];
+}
+function nextActionByStartTime(action) {
+  return action[nextByStartTimeSymbol];
+}
+function stats(action) {
+  let errors = 0;
+  let warnings = 0;
+  for (const event of eventsForAction(action)) {
+    if (event.type === "console") {
+      const type = event.messageType;
+      if (type === "warning")
+        ++warnings;
+      else if (type === "error")
+        ++errors;
+    }
+    if (event.type === "event" && event.method === "pageError")
+      ++errors;
+  }
+  return { errors, warnings };
+}
+function eventsForAction(action) {
+  let result = action[eventsSymbol];
+  if (result)
+    return result;
+  const nextAction = nextInContext(action);
+  result = context(action).events.filter((event) => {
+    return event.time >= action.startTime && (!nextAction || event.time < nextAction.startTime);
+  });
+  action[eventsSymbol] = result;
+  return result;
+}
+function collectSources(actions, errorDescriptors) {
+  const result = /* @__PURE__ */ new Map();
+  for (const action of actions) {
+    for (const frame of action.stack || []) {
+      let source = result.get(frame.file);
+      if (!source) {
+        source = { errors: [], content: void 0 };
+        result.set(frame.file, source);
+      }
+    }
+  }
+  for (const error of errorDescriptors) {
+    const { action, stack, message } = error;
+    if (!action || !stack)
+      continue;
+    result.get(stack[0].file)?.errors.push({
+      line: stack[0].line || 0,
+      message
+    });
+  }
+  return result;
+}
+const kFakeRootAction = {
+  type: "action",
+  callId: "",
+  startTime: 0,
+  endTime: 0,
+  class: "",
+  method: "",
+  params: {},
+  log: [],
+  context: {
+    origin: "library",
+    startTime: 0,
+    endTime: 0,
+    browserName: "",
+    wallTime: 0,
+    options: {},
+    pages: [],
+    resources: [],
+    actions: [],
+    events: [],
+    stdio: [],
+    errors: [],
+    hasSource: false,
+    contextId: ""
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  TraceModel,
+  buildActionTree,
+  context,
+  eventsForAction,
+  nextActionByStartTime,
+  previousActionByEndTime,
+  stats
+});