From 81c706c2629ff25bd5eca415596de269da07637d Mon Sep 17 00:00:00 2001
From: gentlementlegen <fernand.veyrier@epitech.eu>
Date: Tue, 31 Dec 2024 11:53:44 +0900
Subject: [PATCH] feat!: comments can be reused on posting

---
 src/comment.ts | 54 +++++++++++++++++++++++++++++++++++++++-----------
 1 file changed, 42 insertions(+), 12 deletions(-)

diff --git a/src/comment.ts b/src/comment.ts
index a8eb109..3d1b92b 100644
--- a/src/comment.ts
+++ b/src/comment.ts
@@ -5,10 +5,30 @@ import { sanitizeMetadata } from "./util";
 
 const HEADER_NAME = "UbiquityOS";
 
+export interface CommentOptions {
+  /*
+   * Should the comment be posted as send within the log, without adding any sort of formatting.
+   */
+  raw?: boolean;
+  /*
+   * Should the previously posted comment be reused instead of posting a new comment.
+   */
+  updateComment?: boolean;
+}
+
+export type PostComment = {
+  (context: Context, message: LogReturn | Error, options?: CommentOptions): Promise<void>;
+  lastCommentId?: number;
+};
+
 /**
  * Posts a comment on a GitHub issue if the issue exists in the context payload, embedding structured metadata to it.
  */
-export async function postComment(context: Context, message: LogReturn | Error, raw = false) {
+export const postComment: PostComment = async function (
+  context: Context,
+  message: LogReturn | Error,
+  options: CommentOptions = { updateComment: true, raw: false }
+) {
   let issueNumber;
 
   if ("issue" in context.payload) {
@@ -23,19 +43,29 @@ export async function postComment(context: Context, message: LogReturn | Error,
   }
 
   if ("repository" in context.payload && context.payload.repository?.owner?.login) {
-    const body = await createStructuredMetadataWithMessage(context, message, raw);
-    await context.octokit.rest.issues.createComment({
-      owner: context.payload.repository.owner.login,
-      repo: context.payload.repository.name,
-      issue_number: issueNumber,
-      body: body,
-    });
+    const body = await createStructuredMetadataWithMessage(context, message, options);
+    if (options.updateComment && postComment.lastCommentId) {
+      await context.octokit.rest.issues.updateComment({
+        owner: context.payload.repository.owner.login,
+        repo: context.payload.repository.name,
+        comment_id: postComment.lastCommentId,
+        body: body,
+      });
+    } else {
+      const commentData = await context.octokit.rest.issues.createComment({
+        owner: context.payload.repository.owner.login,
+        repo: context.payload.repository.name,
+        issue_number: issueNumber,
+        body: body,
+      });
+      postComment.lastCommentId = commentData.data.id;
+    }
   } else {
     context.logger.info("Cannot post comment because repository is not found in the payload.", { payload: context.payload });
   }
-}
+};
 
-async function createStructuredMetadataWithMessage(context: Context, message: LogReturn | Error, raw = false) {
+async function createStructuredMetadataWithMessage(context: Context, message: LogReturn | Error, options: CommentOptions) {
   let logMessage;
   let callingFnName;
   let instigatorName;
@@ -86,9 +116,9 @@ async function createStructuredMetadataWithMessage(context: Context, message: Lo
 
   if (message instanceof Error) {
     const content = context.logger.error(message.message).logMessage;
-    return `${raw ? content.raw : content.diff}\n\n${metadataSerialized}\n`;
+    return `${options.raw ? content.raw : content.diff}\n\n${metadataSerialized}\n`;
   }
 
   // Add carriage returns to avoid any formatting issue
-  return `${raw ? logMessage?.raw : logMessage?.diff}\n\n${metadataSerialized}\n`;
+  return `${options.raw ? logMessage?.raw : logMessage?.diff}\n\n${metadataSerialized}\n`;
 }