From a8fa4d7e0dd3b874af8cb04ebe69878bcf2d5d92 Mon Sep 17 00:00:00 2001
From: Simon Knott <info@simonknott.de>
Date: Thu, 6 Feb 2025 12:31:45 +0100
Subject: [PATCH] add prompt

---
 packages/html-reporter/src/reportView.tsx    |  4 +--
 packages/html-reporter/src/testErrorView.tsx | 36 +++++++++++++++++++-
 2 files changed, 37 insertions(+), 3 deletions(-)

diff --git a/packages/html-reporter/src/reportView.tsx b/packages/html-reporter/src/reportView.tsx
index ec8966a68c..4531756747 100644
--- a/packages/html-reporter/src/reportView.tsx
+++ b/packages/html-reporter/src/reportView.tsx
@@ -78,7 +78,7 @@ export const ReportView: React.FC<{
     return result;
   }, [report, filter]);
 
-  return <MetadataContext.Provider value={gitCommitInfo}><div className='htmlreport vbox px-4 pb-4'>
+  return <GitCommitInfoContext.Provider value={gitCommitInfo}><div className='htmlreport vbox px-4 pb-4'>
     <main>
       {report?.json() && <HeaderView stats={report.json().stats} filterText={filterText} setFilterText={setFilterText}></HeaderView>}
       <Route predicate={testFilesRoutePredicate}>
@@ -94,7 +94,7 @@ export const ReportView: React.FC<{
         {!!report && <TestCaseViewLoader report={report} tests={filteredTests.tests} testIdToFileIdMap={testIdToFileIdMap} />}
       </Route>
     </main>
-  </div></MetadataContext.Provider>;
+  </div></GitCommitInfoContext.Provider>;
 };
 
 const TestCaseViewLoader: React.FC<{
diff --git a/packages/html-reporter/src/testErrorView.tsx b/packages/html-reporter/src/testErrorView.tsx
index ea402106a8..97db1b5c42 100644
--- a/packages/html-reporter/src/testErrorView.tsx
+++ b/packages/html-reporter/src/testErrorView.tsx
@@ -19,13 +19,47 @@ import * as React from 'react';
 import './testErrorView.css';
 import type { ImageDiff } from '@web/shared/imageDiffView';
 import { ImageDiffView } from '@web/shared/imageDiffView';
+import { GitCommitInfoContext } from './reportView';
 
 export const TestErrorView: React.FC<{
   error: string;
   testId?: string;
 }> = ({ error, testId }) => {
   const html = React.useMemo(() => ansiErrorToHtml(error), [error]);
-  return <div className='test-error-view test-error-text' data-testid={testId} dangerouslySetInnerHTML={{ __html: html || '' }}></div>;
+  return (
+    <div className='test-error-view test-error-text' data-testid={testId}>
+      <PromptButton error={error} />
+      <div dangerouslySetInnerHTML={{ __html: html || '' }}></div>
+    </div>
+  );
+};
+
+const ansiRegex = new RegExp('([\\u001B\\u009B][[\\]()#;?]*(?:(?:(?:[a-zA-Z\\d]*(?:;[-a-zA-Z\\d\\/#&.:=?%@~_]*)*)?\\u0007)|(?:(?:\\d{1,4}(?:;\\d{0,4})*)?[\\dA-PR-TZcf-ntqry=><~])))', 'g');
+export function stripAnsiEscapes(str: string): string {
+  return str.replace(ansiRegex, '');
+}
+
+const PromptButton: React.FC<{
+  error: string;
+}> = ({ error }) => {
+  const gitCommitInfo = React.useContext(GitCommitInfoContext);
+  if (!gitCommitInfo)
+    return undefined;
+
+  const diff = gitCommitInfo['pull.diff'] ?? gitCommitInfo['revision.diff'];
+  if (!diff)
+    return undefined;
+
+  return (
+    <button onClick={() => {
+      navigator.clipboard.writeText([
+        'You are a helpful assistant. Help me understand the error cause. Here is the error:',
+        stripAnsiEscapes(error),
+        'And this is the code diff:',
+        diff
+      ].join('\n'));
+    }}>Copy AI Prompt</button>
+  );
 };
 
 export const TestScreenshotErrorView: React.FC<{