feat: trigger CodeRabbit plan after issue assignment (#2577)

Signed-off-by: ParasSalonia <parassalonia22@gmail.com>
Signed-off-by: Paras Salonia <parassalonia22@gmail.com>

Author: ParasSalonia

.github/scripts/__tests__/jest/issue-assign.test.js

@@ -13,6 +13,11 @@ jest.mock('../../shared/api/github-api', () => ({
   assignIssue: jest.fn(),
 }));
 
+jest.mock('../../shared/helpers/coderabbit-plan.js', () => ({
+  triggerCodeRabbitPlan: jest.fn(),
+  hasExistingCodeRabbitPlan: jest.fn(),
+}));
+
 jest.mock('../../shared/helpers/comment', () => ({
   buildAlreadyAssignedComment: jest.fn(() => 'already assigned'),
   buildGuardComment: jest.fn(() => 'guard comment'),
@@ -35,6 +40,11 @@ const { runAssignmentFlow } = require('../../shared/core/issue-assign');
 const githubApi = require('../../shared/api/github-api');
 const spam = require('../../shared/helpers/spam');
 
+const {
+  triggerCodeRabbitPlan,
+  hasExistingCodeRabbitPlan,
+} = require('../../shared/helpers/coderabbit-plan.js');
+
 function createContext(overrides = {}) {
   return {
     payload: {
@@ -86,6 +96,9 @@ beforeEach(() => {
   spam.isSpamBlockedLevel.mockReturnValue(false);
   spam.isSpamLimited.mockReturnValue(false);
   spam.getAssignmentLimit.mockReturnValue(5);
+
+  hasExistingCodeRabbitPlan.mockResolvedValue(false);
+  triggerCodeRabbitPlan.mockResolvedValue(true);
 });
 
 describe('runAssignmentFlow - validation', () => {
@@ -597,8 +610,9 @@ describe('runAssignmentFlow - assignment', () => {
     expect(githubApi.assignIssue).not.toHaveBeenCalled();
   });
 
-  test('assigns issue when all checks pass', async () => {
+  test('assigns issue and triggers CodeRabbit when all checks pass', async () => {
     githubApi.getOpenAssignments.mockResolvedValue(0);
+    hasExistingCodeRabbitPlan.mockResolvedValue(false);
 
     const github = createGithub();
     const context = createContext();
@@ -613,8 +627,54 @@ describe('runAssignmentFlow - assignment', () => {
       username: 'parv',
     });
 
+    expect(hasExistingCodeRabbitPlan).toHaveBeenCalledWith(
+      github,
+      'hiero-ledger',
+      'hiero-sdk-python',
+      10
+    );
+
+    expect(triggerCodeRabbitPlan).toHaveBeenCalledWith(
+      github,
+      'hiero-ledger',
+      'hiero-sdk-python',
+      context.payload.issue
+    );
+
     expect(githubApi.postIssueComment).not.toHaveBeenCalled();
   });
+
+  test('does not trigger CodeRabbit when a plan already exists', async () => {
+    githubApi.getOpenAssignments.mockResolvedValue(0);
+    hasExistingCodeRabbitPlan.mockResolvedValue(true);
+
+    const github = createGithub();
+    const context = createContext();
+
+    await runAssignmentFlow({ github, context });
+
+    expect(githubApi.assignIssue).toHaveBeenCalled();
+    expect(hasExistingCodeRabbitPlan).toHaveBeenCalled();
+    expect(triggerCodeRabbitPlan).not.toHaveBeenCalled();
+  });
+
+  test('does not fail assignment when CodeRabbit trigger fails', async () => {
+    githubApi.getOpenAssignments.mockResolvedValue(0);
+    hasExistingCodeRabbitPlan.mockResolvedValue(false);
+    triggerCodeRabbitPlan.mockRejectedValue(
+      new Error('CodeRabbit API failed')
+    );
+
+    const github = createGithub();
+    const context = createContext();
+
+    await expect(
+      runAssignmentFlow({ github, context })
+    ).resolves.not.toThrow();
+
+    expect(githubApi.assignIssue).toHaveBeenCalled();
+    expect(triggerCodeRabbitPlan).toHaveBeenCalled();
+  });
 });
 
 describe('runAssignmentFlow - error handling', () => {
@@ -676,31 +736,21 @@ describe('runAssignmentFlow - error handling', () => {
     expect(githubApi.assignIssue).toHaveBeenCalled();
   });
 
-  test('propagates post comment errors', async () => {
-    githubApi.postIssueComment.mockRejectedValue(
-      new Error('Comment failed')
+  test('does not trigger CodeRabbit when assignment fails', async () => {
+    githubApi.assignIssue.mockRejectedValue(
+      new Error('Assignment failed')
     );
 
-    githubApi.countCompletedIssuesWithLabel.mockResolvedValue(0);
-
     const github = createGithub();
+    const context = createContext();
 
-    const context = createContext({
-      issue: {
-        number: 42,
-        assignees: [],
-        labels: [
-          {
-            name: 'skill: intermediate',
-          },
-        ],
-      },
+    await runAssignmentFlow({
+      github,
+      context,
     });
 
-    await expect(
-      runAssignmentFlow({ github, context })
-    ).rejects.toThrow('Comment failed');
-
-    expect(githubApi.assignIssue).not.toHaveBeenCalled();
+    expect(githubApi.assignIssue).toHaveBeenCalled();
+    expect(hasExistingCodeRabbitPlan).not.toHaveBeenCalled();
+    expect(triggerCodeRabbitPlan).not.toHaveBeenCalled();
   });
 });

.github/scripts/shared/core/issue-assign.js

@@ -52,6 +52,11 @@ const {
   getAssignmentLimit,
 } = require('../helpers/spam.js');
 
+const {
+  triggerCodeRabbitPlan,
+  hasExistingCodeRabbitPlan,
+} = require('../helpers/coderabbit-plan.js');
+
 /**
  * Returns true if a comment contains the `/assign` command.
  *
@@ -281,14 +286,46 @@ async function runAssignmentFlow({ github, context }) {
       username: commenter,
     });
   } catch (error) {
-    console.error("[assign-bot] Failed to assign issue:", {
+    console.error('[assign-bot] Failed to assign issue:', {
       message: error.message,
     });
+    return;
   }
-  return;
 
-}
+  // Trigger CodeRabbit after successful assignment
+  try {
+    const planExists = await hasExistingCodeRabbitPlan(
+      github,
+      owner,
+      repoName,
+      issueNumber
+    );
 
+    if (planExists === false) {
+      await triggerCodeRabbitPlan(
+      github,
+      owner,
+      repoName,
+      issue
+    );
+    } else if (planExists === true) {
+        console.log(
+        `[assign-bot] CodeRabbit plan already exists for #${issueNumber}`
+      );
+    } else {
+      console.log(
+      `[assign-bot] Unable to determine whether a CodeRabbit plan exists for #${issueNumber}. Skipping trigger.`
+      );
+    }
+  } catch (error) {
+    console.error('[assign-bot] CodeRabbit plan trigger failed:', {
+      message: error.message,
+      issueNumber,
+    });
+  }
+
+  return;
+}
 module.exports = {
   runAssignmentFlow,
 };

…ithub/scripts/coderabbit_plan_trigger.js …cripts/shared/helpers/coderabbit-plan.js.github/scripts/coderabbit_plan_trigger.js renamed to .github/scripts/shared/helpers/coderabbit-plan.js .github/scripts/coderabbit_plan_trigger.js renamed to .github/scripts/shared/helpers/coderabbit-plan.js

@@ -1,7 +1,7 @@
 // Script to trigger CodeRabbit plan for all difficulty-labeled issues (including GFI)
 
 const CODERABBIT_MARKER = '<!-- CodeRabbit Plan Trigger -->';
-const { DIFFICULTY_LABELS } = require('./shared/labels.js');
+const { DIFFICULTY_LABELS } = require('../labels.js');
 
 async function triggerCodeRabbitPlan(github, owner, repo, issue, marker = CODERABBIT_MARKER) {
   const comment = `${marker} @coderabbitai plan`;
@@ -73,7 +73,7 @@ async function hasExistingCodeRabbitPlan(github, owner, repo, issueNumber) {
       issueNumber,
     });
     // Return false to allow plan trigger attempt (fail-open for better UX)
-    return false;
+    return null;
   }
 }
 
@@ -110,10 +110,18 @@ function logSummary(owner, repo, issue) {
 }
 
 // Main workflow handler (default export for workflow usage)
-async function main({ github, context }) {
+async function triggerCodeRabbitPlanForIssue({ github, context }) {
   try {
     const { owner, repo } = context.repo;
-    const { issue: eventIssue, label } = context.payload;
+
+    const payload = context?.payload;
+
+    if (!payload) {
+      console.log('No event payload');
+      return;
+    }
+
+    const { issue: eventIssue, label } = payload;
 
     // Validations
     if (!eventIssue?.number) return console.log('No issue in payload');
@@ -149,10 +157,9 @@ async function main({ github, context }) {
   }
 }
 
-// Default export for workflow usage: await script({ github, context })
-module.exports = main;
-
-// Named exports for reuse by other scripts (e.g., GFI assignment bot)
-module.exports.triggerCodeRabbitPlan = triggerCodeRabbitPlan;
-module.exports.hasExistingCodeRabbitPlan = hasExistingCodeRabbitPlan;
-module.exports.CODERABBIT_MARKER = CODERABBIT_MARKER;
+module.exports = {
+  triggerCodeRabbitPlan,
+  hasExistingCodeRabbitPlan,
+  CODERABBIT_MARKER,
+  triggerCodeRabbitPlanForIssue,
+};

.github/workflows/approved-issues.yml

@@ -63,5 +63,5 @@ jobs:
         uses: actions/github-script@3a2844b7e9c422d3c10d287c895573f7108da1b3 # v9.0.0
         with:
           script: |
-            const script = require('./.github/scripts/coderabbit_plan_trigger.js');
-            await script({ github, context });
+            const { triggerCodeRabbitPlanForIssue } = require('./.github/scripts/shared/helpers/coderabbit-plan.js');
+            await triggerCodeRabbitPlanForIssue({ github, context });