feat: add support for /pr command to create PR targeting original PR

- Add comment_body input parameter to action.yml
- Implement createPRForChanges function to create new branch and PR
- Update main function to detect /commit vs /pr command
- For /pr: create new branch, commit changes, and create PR targeting original PR branch
- For /commit: maintain existing behavior of committing directly to PR branch

Author: tulga-bytes

assistant/action.yml

@@ -25,6 +25,9 @@ inputs:
   augment_api_url:
     description: "URL endpoint for Augment API requests. Store as repository variable."
     required: false
+  comment_body:
+    description: "The body of the comment (used to detect /commit vs /pr command)"
+    required: false
 
 outputs:
   success:
@@ -87,6 +90,7 @@ runs:
         INPUT_COMMENT_ID: ${{ inputs.comment_id }}
         INPUT_EVENT_NAME: ${{ inputs.event_name }}
         INPUT_REACTION: ${{ inputs.reaction }}
+        INPUT_COMMENT_BODY: ${{ inputs.comment_body }}
         GITHUB_REPOSITORY: ${{ github.repository }}
         AUGMENT_API_TOKEN: ${{ inputs.augment_api_token }}
         AUGMENT_API_URL: ${{ inputs.augment_api_url }}

assistant/src/index.ts

@@ -250,6 +250,83 @@ Automatically generated by Auggie Assistant`;
   core.info('✅ Changes committed and pushed successfully');
 }
 
+/**
+ * Create a new branch, commit changes, and create a PR targeting the original PR
+ */
+async function createPRForChanges(
+  commentId: number,
+  headBranch: string,
+  githubToken: string,
+  owner: string,
+  repo: string,
+  prNumber: number,
+  octokit: Octokit
+): Promise<{ newBranch: string; newPrNumber: number }> {
+  core.info('📝 Creating new branch for PR...');
+
+  // Stage all changes
+  await exec.exec('git', ['add', '-A']);
+
+  // Check if there are changes to commit using git status
+  let statusOutput = '';
+  await exec.exec('git', ['status', '--porcelain'], {
+    listeners: {
+      stdout: (data: Buffer) => {
+        statusOutput += data.toString();
+      },
+    },
+  });
+
+  const hasChanges = statusOutput.trim().length > 0;
+
+  if (!hasChanges) {
+    core.info('ℹ️ No changes to commit');
+    throw new Error('No changes to create PR with');
+  }
+
+  core.info(`📝 Changes detected:\n${statusOutput}`);
+
+  // Create a new branch name based on the comment ID
+  const newBranch = `auggie/pr-${prNumber}-comment-${commentId}`;
+  core.info(`🌿 Creating new branch: ${newBranch}`);
+
+  // Create and checkout new branch
+  await exec.exec('git', ['checkout', '-b', newBranch]);
+
+  // Commit changes
+  const commitMessage = `feat: implement changes requested in comment #${commentId}
+
+Automatically generated by Auggie Assistant`;
+
+  await exec.exec('git', ['commit', '-m', commitMessage]);
+
+  // Push new branch
+  core.info(`🚀 Pushing to ${newBranch}...`);
+  const remoteUrl = `https://x-access-token:${githubToken}@github.com/${owner}/${repo}.git`;
+  await exec.exec('git', ['push', remoteUrl, `HEAD:${newBranch}`]);
+
+  core.info('✅ Changes committed and pushed to new branch');
+
+  // Create PR targeting the original PR's branch
+  core.info(`📬 Creating PR targeting ${headBranch}...`);
+  const { data: newPr } = await octokit.rest.pulls.create({
+    owner,
+    repo,
+    title: `Changes from comment #${commentId} on PR #${prNumber}`,
+    body: `This PR contains changes requested in [comment #${commentId}](https://github.com/${owner}/${repo}/pull/${prNumber}#issuecomment-${commentId}) on PR #${prNumber}.
+
+Automatically generated by Auggie Assistant.
+
+**Target Branch:** \`${headBranch}\` (from PR #${prNumber})`,
+    head: newBranch,
+    base: headBranch,
+  });
+
+  core.info(`✅ Created PR #${newPr.number}`);
+
+  return { newBranch, newPrNumber: newPr.number };
+}
+
 /**
  * Invoke Auggie to implement changes
  */
@@ -395,6 +472,7 @@ async function main(): Promise<void> {
     const githubToken = getInput('github_token', true);
     const commentIdStr = getInput('comment_id', true);
     const eventName = getInput('event_name', true);
+    const commentBody = getInput('comment_body', false);
 
     // Validate inputs
     const commentId = Number.parseInt(commentIdStr, 10);
@@ -407,10 +485,20 @@ async function main(): Promise<void> {
     // Create Octokit instance
     const octokit = new Octokit({ auth: githubToken });
 
-    // Note: Reaction is added in action.yml for immediate feedback
+    // Detect command type (/commit or /pr)
+    const isPrCommand = commentBody.includes('/pr');
+    const isCommitCommand = commentBody.includes('/commit');
+
+    if (!isPrCommand && !isCommitCommand) {
+      core.info('ℹ️ No /commit or /pr command found in comment');
+      core.setOutput('success', 'true');
+      return;
+    }
+
     core.info(`🎯 Starting PR Assistant for comment ${commentId}`);
     core.info(`📦 Repository: ${owner}/${repo}`);
     core.info(`📝 Event: ${eventName}`);
+    core.info(`🔧 Command: ${isPrCommand ? '/pr' : '/commit'}`);
 
     // Step 2: Get PR number
     core.info('🔍 Finding associated PR...');
@@ -435,29 +523,66 @@ async function main(): Promise<void> {
     try {
       await invokeAuggie(context);
 
-      // Step 6: Commit and push changes
-      await commitAndPush(commentId, context.headBranch, githubToken, owner, repo);
+      // Step 6: Commit and push changes (different behavior for /pr vs /commit)
+      if (isPrCommand) {
+        // For /pr: create a new branch and PR targeting the original PR
+        const { newPrNumber } = await createPRForChanges(
+          commentId,
+          context.headBranch,
+          githubToken,
+          owner,
+          repo,
+          prNumber,
+          octokit
+        );
+
+        // Step 7: Add success comment to original PR
+        const quotedComment = context.commentBody
+          .split('\n')
+          .map(line => `> ${line}`)
+          .join('\n');
+
+        await addPRComment(
+          octokit,
+          owner,
+          repo,
+          prNumber,
+          `${quotedComment}
+
+✅ Successfully implemented the requested changes!
 
-      // Step 7: Add success comment to PR
-      const quotedComment = context.commentBody
-        .split('\n')
-        .map(line => `> ${line}`)
-        .join('\n');
+Created PR #${newPrNumber} with the changes targeting \`${context.headBranch}\`.
 
-      await addPRComment(
-        octokit,
-        owner,
-        repo,
-        prNumber,
-        `${quotedComment}
+👉 [View the new PR](https://github.com/${owner}/${repo}/pull/${newPrNumber})`
+        );
+
+        core.info(`✨ PR Assistant completed successfully - created PR #${newPrNumber}`);
+      } else {
+        // For /commit: commit directly to the PR branch (existing behavior)
+        await commitAndPush(commentId, context.headBranch, githubToken, owner, repo);
+
+        // Step 7: Add success comment to PR
+        const quotedComment = context.commentBody
+          .split('\n')
+          .map(line => `> ${line}`)
+          .join('\n');
+
+        await addPRComment(
+          octokit,
+          owner,
+          repo,
+          prNumber,
+          `${quotedComment}
 
 ✅ Successfully implemented the requested changes!
 
 The changes have been committed to the \`${context.headBranch}\` branch.`
-      );
+        );
+
+        core.info('✨ PR Assistant completed successfully');
+      }
 
       core.setOutput('success', 'true');
-      core.info('✨ PR Assistant completed successfully');
     } catch (error) {
       // Add failure comment to PR
       const quotedComment = context.commentBody