From 563bc16b6fa054a9975c9736c0ec56db617b73ab Mon Sep 17 00:00:00 2001
From: Andrea Brancaleoni <abc@pompel.me>
Date: Tue, 17 Dec 2024 02:50:09 -0800
Subject: [PATCH] action.cjs: include filtered diff in the watermark

---
 action.cjs                | 3 +++
 src/submitReview.js       | 3 ++-
 src/subtleSubmitReview.js | 3 ++-
 3 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/action.cjs b/action.cjs
index 4d88768..a7d7c91 100644
--- a/action.cjs
+++ b/action.cjs
@@ -106,11 +106,14 @@ module.exports = async ({ github, context, inputs, actionPath }) => {
           : `openai debug - ${watermark}`
     }
 
+    const header = '<details><summary>Diff</summary>\n\n```diff' + filteredPatch + '\n\n```\n\n</details>'
+
     await submitReview({
       owner: options.owner,
       repo: options.repo,
       prnum: options.prnum,
       watermark,
+      header,
       explainPatch: explainPatchCb,
       debounceTime: options.debounce_time,
       debug,
diff --git a/src/submitReview.js b/src/submitReview.js
index 169bcbf..da46407 100644
--- a/src/submitReview.js
+++ b/src/submitReview.js
@@ -16,6 +16,7 @@ export default async function submitReview ({
   owner, repo, prnum, watermark, debounceTime,
   explainPatch = async () => 'no explanation provided<!-- Generated by STUB -->',
   githubToken = null,
+  header = '',
   github = null,
   debug = false
 }) {
@@ -81,7 +82,7 @@ export default async function submitReview ({
       owner,
       repo,
       issue_number: prnum,
-      body: watermark + '\n\n' + patchExplained
+      body: watermark + '\n\n' + header + '\n\n' + patchExplained
     })
   }
 }
diff --git a/src/subtleSubmitReview.js b/src/subtleSubmitReview.js
index 4bd0bc5..2912323 100644
--- a/src/subtleSubmitReview.js
+++ b/src/subtleSubmitReview.js
@@ -21,6 +21,7 @@ export default async function subtleSubmitReview ({
   explainPatch = async () => 'no explanation provided<!-- Generated by STUB -->',
   githubToken = null,
   github = null,
+  header = '',
   debug = false
 }) {
   if (!github && githubToken) {
@@ -63,7 +64,7 @@ export default async function subtleSubmitReview ({
   const end = escapeRegex('<!-- Generated by ') + '\\w+' + escapeRegex(' --></details>')
   const re = new RegExp(`${begin}.*${end}`, 's')
 
-  const newExplaination = '<details><summary>AI Review</summary>' + '\n\n' + watermark + '\n\n' + await explainPatch() + '</details>'
+  const newExplaination = '<details><summary>AI Review</summary>' + '\n\n' + watermark + '\n\n' + header + '\n\n' + await explainPatch() + '</details>'
 
   // fetch the message twice, to decrease the chances of race condition, but still debounce
   const msg = (await github.graphql(query, variables)).repository.pullRequest