Class JSCodeGen

packages/trace-viewer/src/ui/codegen.ts:24–100

}

class JSCodeGen implements APIRequestCodegen {
  generatePlaywrightRequestCall(request: har.Request, body: string | undefined): string {
    let method = request.method.toLowerCase();
    const url = new URL(request.url);
    const urlParam = `${url.origin}${url.pathname}`;
    const options: any = {};
    if (!['delete', 'get', 'head', 'post', 'put', 'patch'].includes(method)) {
      options.method = method;
      method = 'fetch';
    }
    if (url.searchParams.size)
      options.params = Object.fromEntries(url.searchParams.entries());
    if (body)
      options.data = body;
    if (request.headers.length)
      options.headers = Object.fromEntries(request.headers.map(header => [header.name, header.value]));

    const params = [`'${urlParam}'`];
    const hasOptions = Object.keys(options).length > 0;
    if (hasOptions)
      params.push(this.prettyPrintObject(options));
    return `await page.request.${method}(${params.join(', ')});`;
  }

  private prettyPrintObject(obj: any, indent = 2, level = 0): string {
    // Handle null and undefined
    if (obj === null)
      return 'null';
    if (obj === undefined)
      return 'undefined';

    // Handle primitive types
    if (typeof obj !== 'object') {
      if (typeof obj === 'string')
        return this.stringLiteral(obj);
      return String(obj);
    }

    // Handle arrays
    if (Array.isArray(obj)) {
      if (obj.length === 0)
        return '[]';
      const spaces = ' '.repeat(level * indent);
      const nextSpaces = ' '.repeat((level + 1) * indent);

      const items = obj.map(item =>
        `${nextSpaces}${this.prettyPrintObject(item, indent, level + 1)}`
      ).join(',\n');

      return `[\n${items}\n${spaces}]`;
    }

    // Handle regular objects
    if (Object.keys(obj).length === 0)
      return '{}';
    const spaces = ' '.repeat(level * indent);
    const nextSpaces = ' '.repeat((level + 1) * indent);