chore: 🚧 Keep state

getcodelimit · Dec 5, 2024 · 6bf68d3 · 6bf68d3
1 parent dea62ae
commit 6bf68d3
Show file tree

Hide file tree

Showing 3 changed files with 47 additions and 34 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -23,4 +23,4 @@ jobs:
       run: yarn install
 
     - name: 'Run Code Limit action'
-      uses: getcodelimit/codelimit-action@v1
+      uses: getcodelimit/codelimit-action@main
diff --git a/dist/index.js b/dist/index.js
@@ -49039,14 +49039,13 @@ function generateMarkdownReport(clBinary) {
 }
 function updateReportsBranch(octokit, markdownReport) {
   return __awaiter(this, void 0, void 0, function* () {
-    var _a;
     const owner = (0, github_2.getRepoOwner)(github_1.context);
     const repo = (0, github_2.getRepoName)(github_1.context);
-    if (!owner || !repo) {
-      console.error("Could not determine repository owner or name");
+    const branch = (0, github_2.getSourceBranch)();
+    if (!owner || !repo || !branch) {
+      console.error("Could not determine repository owner, name, or branch");
       process.exit(1);
     }
-    const branch = (0, github_2.getSourceBranch)();
     yield (0, github_2.createBranchIfNotExists)(octokit, owner, repo, "_codelimit_reports");
     const reportContent = (0, codelimit_1.getReportContent)();
     yield (0, github_2.createOrUpdateFile)(octokit, owner, repo, "_codelimit_reports", `${branch}/badge.svg`, (0, codelimit_1.getBadgeContent)(reportContent));
@@ -49055,19 +49054,27 @@ function updateReportsBranch(octokit, markdownReport) {
     }
     yield (0, github_2.createOrUpdateFile)(octokit, owner, repo, "_codelimit_reports", `${branch}/codelimit.md`, markdownReport);
     if ((0, github_2.isPullRequest)()) {
-      const prNumber = (_a = github_1.context.payload.pull_request) === null || _a === void 0 ? void 0 : _a.number;
-      if (prNumber) {
-        const actionStateFile = yield (0, github_2.getFile)(octokit, owner, repo, "_codelimit_reports", `${branch}/action.json`);
-        if (actionStateFile) {
-          const actionState = JSON.parse(actionStateFile.content);
-          const commentId = actionState.commentId;
-          yield (0, github_2.updateComment)(octokit, owner, repo, prNumber, markdownReport, commentId);
-        } else {
-          const commentId = yield (0, github_2.createPRComment)(octokit, owner, repo, prNumber, markdownReport);
-          const actionState = { commentId };
-          const actionStateJson = JSON.stringify(actionState);
-          yield (0, github_2.createOrUpdateFile)(octokit, owner, repo, "_codelimit_reports", `${branch}/action.json`, actionStateJson);
-        }
+      yield updatePullRequestComment(octokit, owner, repo, branch, markdownReport);
+    }
+  });
+}
+function updatePullRequestComment(octokit, owner, repo, branchName, markdownReport) {
+  return __awaiter(this, void 0, void 0, function* () {
+    var _a;
+    const prNumber = (_a = github_1.context.payload.pull_request) === null || _a === void 0 ? void 0 : _a.number;
+    if (prNumber) {
+      const actionStateFile = yield (0, github_2.getFile)(octokit, owner, repo, "_codelimit_reports", `${branchName}/action.json`);
+      if (actionStateFile) {
+        const actionState = JSON.parse(actionStateFile.content);
+        const commentId = actionState.commentId;
+        console.log(`Updating existing comment with ID: ${commentId}`);
+        yield (0, github_2.updateComment)(octokit, owner, repo, prNumber, markdownReport, commentId);
+      } else {
+        console.log("State file not found, creating new comment");
+        const commentId = yield (0, github_2.createPRComment)(octokit, owner, repo, prNumber, markdownReport);
+        const actionState = { commentId };
+        const actionStateJson = JSON.stringify(actionState);
+        yield (0, github_2.createOrUpdateFile)(octokit, owner, repo, "_codelimit_reports", `${branchName}/action.json`, actionStateJson);
       }
     }
   });

diff --git a/src/action.ts b/src/action.ts
@@ -30,11 +30,11 @@ async function generateMarkdownReport(clBinary: string) {
 async function updateReportsBranch(octokit: Octokit, markdownReport: string) {
     const owner = getRepoOwner(context);
     const repo = getRepoName(context);
-    if (!owner || !repo) {
-        console.error('Could not determine repository owner or name');
+    const branch = getSourceBranch();
+    if (!owner || !repo || !branch) {
+        console.error('Could not determine repository owner, name, or branch');
         process.exit(1);
     }
-    const branch = getSourceBranch();
     await createBranchIfNotExists(octokit, owner, repo, '_codelimit_reports');
     const reportContent = getReportContent();
     await createOrUpdateFile(octokit, owner, repo, '_codelimit_reports', `${branch}/badge.svg`, getBadgeContent(reportContent));
@@ -43,19 +43,25 @@ async function updateReportsBranch(octokit: Octokit, markdownReport: string) {
     }
     await createOrUpdateFile(octokit, owner, repo, '_codelimit_reports', `${branch}/codelimit.md`, markdownReport);
     if (isPullRequest()) {
-        const prNumber = context.payload.pull_request?.number;
-        if (prNumber) {
-            const actionStateFile = await getFile(octokit, owner, repo, '_codelimit_reports', `${branch}/action.json`);
-            if (actionStateFile) {
-                const actionState = JSON.parse(actionStateFile.content) as ActionState;
-                const commentId = actionState.commentId;
-                await updateComment(octokit, owner, repo, prNumber, markdownReport, commentId);
-            } else {
-                const commentId = await createPRComment(octokit, owner, repo, prNumber, markdownReport);
-                const actionState: ActionState = {commentId: commentId};
-                const actionStateJson = JSON.stringify(actionState);
-                await createOrUpdateFile(octokit, owner, repo, '_codelimit_reports', `${branch}/action.json`, actionStateJson);
-            }
+        await updatePullRequestComment(octokit, owner, repo, branch, markdownReport);
+    }
+}
+
+async function updatePullRequestComment(octokit: Octokit, owner: string, repo: string, branchName: string, markdownReport: string) {
+    const prNumber = context.payload.pull_request?.number;
+    if (prNumber) {
+        const actionStateFile = await getFile(octokit, owner, repo, '_codelimit_reports', `${branchName}/action.json`);
+        if (actionStateFile) {
+            const actionState = JSON.parse(actionStateFile.content) as ActionState;
+            const commentId = actionState.commentId;
+            console.log(`Updating existing comment with ID: ${commentId}`);
+            await updateComment(octokit, owner, repo, prNumber, markdownReport, commentId);
+        } else {
+            console.log('State file not found, creating new comment');
+            const commentId = await createPRComment(octokit, owner, repo, prNumber, markdownReport);
+            const actionState: ActionState = {commentId: commentId};
+            const actionStateJson = JSON.stringify(actionState);
+            await createOrUpdateFile(octokit, owner, repo, '_codelimit_reports', `${branchName}/action.json`, actionStateJson);
         }
     }
 }