it works

Change-Id: I02b74ae07dde27aca0c4c44266b54734bc955d89

Author: paulirish

src/tools/performance.ts

@@ -8,12 +8,7 @@ import {logger} from '../logger.js';
 import {zod} from '../third_party/index.js';
 import type {Page} from '../third_party/index.js';
 import type {InsightName} from '../trace-processing/parse.js';
-import {
-  getInsightOutput,
-  getTraceSummary,
-  parseRawTraceBuffer,
-  traceResultIsSuccess,
-} from '../trace-processing/parse.js';
+import {getInsightOutput, getTraceSummary, parseRawTraceBuffer, traceResultIsSuccess} from '../trace-processing/parse.js';
 
 import {ToolCategory} from './categories.js';
 import type {Context, Response} from './ToolDefinition.js';
@@ -28,21 +23,13 @@ export const startTrace = defineTool({
     readOnlyHint: true,
   },
   schema: {
-    reload: zod
-      .boolean()
-      .describe(
-        'Determines if, once tracing has started, the page should be automatically reloaded.',
-      ),
-    autoStop: zod
-      .boolean()
-      .describe(
-        'Determines if the trace recording should be automatically stopped.',
-      ),
+    reload: zod.boolean().describe('Determines if, once tracing has started, the page should be automatically reloaded.'),
+    autoStop: zod.boolean().describe('Determines if the trace recording should be automatically stopped.'),
   },
   handler: async (request, response, context) => {
     if (context.isRunningPerformanceTrace()) {
       response.appendResponseLine(
-        'Error: a performance trace is already running. Use performance_stop_trace to stop it. Only one trace can be running at any given time.',
+        'Error: a performance trace is already running. Use performance_stop_trace to stop it. Only one trace can be running at any given time.'
       );
       return;
     }
@@ -93,17 +80,14 @@ export const startTrace = defineTool({
       await new Promise(resolve => setTimeout(resolve, 5_000));
       await stopTracingAndAppendOutput(page, response, context);
     } else {
-      response.appendResponseLine(
-        `The performance trace is being recorded. Use performance_stop_trace to stop it.`,
-      );
+      response.appendResponseLine(`The performance trace is being recorded. Use performance_stop_trace to stop it.`);
     }
   },
 });
 
 export const stopTrace = defineTool({
   name: 'performance_stop_trace',
-  description:
-    'Stops the active performance trace recording on the selected page.',
+  description: 'Stops the active performance trace recording on the selected page.',
   annotations: {
     category: ToolCategory.PERFORMANCE,
     readOnlyHint: true,
@@ -120,32 +104,22 @@ export const stopTrace = defineTool({
 
 export const analyzeInsight = defineTool({
   name: 'performance_analyze_insight',
-  description:
-    'Provides more detailed information on a specific Performance Insight that was highlighted in the results of a trace recording.',
+  description: 'Provides more detailed information on a specific Performance Insight that was highlighted in the results of a trace recording.',
   annotations: {
     category: ToolCategory.PERFORMANCE,
     readOnlyHint: true,
   },
   schema: {
-    insightName: zod
-      .string()
-      .describe(
-        'The name of the Insight you want more information on. For example: "DocumentLatency" or "LCPBreakdown"',
-      ),
+    insightName: zod.string().describe('The name of the Insight you want more information on. For example: "DocumentLatency" or "LCPBreakdown"'),
   },
   handler: async (request, response, context) => {
     const lastRecording = context.recordedTraces().at(-1);
     if (!lastRecording) {
-      response.appendResponseLine(
-        'No recorded traces found. Record a performance trace so you have Insights to analyze.',
-      );
+      response.appendResponseLine('No recorded traces found. Record a performance trace so you have Insights to analyze.');
       return;
     }
 
-    const insightOutput = getInsightOutput(
-      lastRecording,
-      request.params.insightName as InsightName,
-    );
+    const insightOutput = getInsightOutput(lastRecording, request.params.insightName as InsightName);
     if ('error' in insightOutput) {
       response.appendResponseLine(insightOutput.error);
       return;
@@ -155,11 +129,7 @@ export const analyzeInsight = defineTool({
   },
 });
 
-async function stopTracingAndAppendOutput(
-  page: Page,
-  response: Response,
-  context: Context,
-): Promise<void> {
+async function stopTracingAndAppendOutput(page: Page, response: Response, context: Context): Promise<void> {
   try {
     const traceEventsBuffer = await page.tracing.stop();
     const result = await parseRawTraceBuffer(traceEventsBuffer);
@@ -169,26 +139,21 @@ async function stopTracingAndAppendOutput(
       const traceSummaryText = getTraceSummary(result);
       response.appendResponseLine(traceSummaryText);
     } else {
-      response.appendResponseLine(
-        'There was an unexpected error parsing the trace:',
-      );
+      response.appendResponseLine('There was an unexpected error parsing the trace:');
       response.appendResponseLine(result.error);
     }
   } catch (e) {
     const errorText = e instanceof Error ? e.message : JSON.stringify(e);
     logger(`Error stopping performance trace: ${errorText}`);
-    response.appendResponseLine(
-      'An error occurred generating the response for this trace:',
-    );
+    response.appendResponseLine('An error occurred generating the response for this trace:');
     response.appendResponseLine(errorText);
   } finally {
     context.setIsRunningPerformanceTrace(false);
   }
 }
 
-// This key is expected to be visible.
-// b/349721878
-const CRUX_API_KEY = 'AIzaSyCCSOx25vrb5z0tbedCB3_JRzzbVW6Uwgw';
+// This key is expected to be visible. b/349721878
+const CRUX_API_KEY = 'AIzaSyBn5gimNjhiEyA_euicSKko6IlD3HdgUfk';
 const CRUX_ENDPOINT = `https://chromeuxreport.googleapis.com/v1/records:queryRecord?key=${CRUX_API_KEY}`;
 
 export const queryChromeUXReport = defineTool({
@@ -200,53 +165,35 @@ export const queryChromeUXReport = defineTool({
     readOnlyHint: true,
   },
   schema: {
-    origin: zod
-      .string()
-      .describe(
-        'The origin to query, e.g., "https://www.google.com". Do not provide this if "url" is specified.',
-      )
-      .optional(),
+    origin: zod.string().describe('The origin to query, e.g., "https://www.google.com". Do not provide this if "url" is specified.').optional(),
     url: zod
       .string()
-      .describe(
-        'The specific page URL to query, e.g., "https://www.google.com/search?q=puppies". Do not provide this if "origin" is specified.',
-      )
+      .describe('The specific page URL to query, e.g., "https://www.google.com/search?q=puppies". Do not provide this if "origin" is specified.')
       .optional(),
     formFactor: zod
       .enum(['DESKTOP', 'PHONE', 'TABLET'])
-      .describe(
-        'The form factor to filter by. If omitted, data for all form factors is aggregated.',
-      )
+      .describe('The form factor to filter by. If omitted, data for all form factors is aggregated.')
       .optional(),
   },
   handler: async (request, response) => {
     const {origin, url, formFactor} = request.params;
 
     if ((!origin && !url) || (origin && url)) {
-      response.appendResponseLine(
-        'Error: you must provide either "origin" or "url", but not both.',
-      );
+      response.appendResponseLine('Error: you must provide either "origin" or "url", but not both.');
       return;
     }
 
     const body = JSON.stringify({
       origin,
       url,
       formFactor,
-      metrics: [
-        'first_contentful_paint',
-        'largest_contentful_paint',
-        'cumulative_layout_shift',
-        'interaction_to_next_paint',
-      ],
+      metrics: ['first_contentful_paint', 'largest_contentful_paint', 'cumulative_layout_shift', 'interaction_to_next_paint'],
     });
 
     try {
       const cruxResponse = await fetch(CRUX_ENDPOINT, {
         method: 'POST',
-        headers: {
-          'Content-Type': 'application/json',
-        },
+        headers: {'Content-Type': 'application/json', referer: 'devtools://mcp'},
         body,
       });