claude-code-action/src/github/operations/comments/update-with-branch.ts

#!/usr/bin/env bun

/**
 * Update the initial tracking comment with branch link
 * This happens after the branch is created for issues
 */

import {
  createJobRunLink,
  createBranchLink,
  createCommentBody,
} from "./common";
import { type Octokits } from "../../api/client";
import {
  isPullRequestReviewCommentEvent,
  type ParsedGitHubContext,
} from "../../context";

export async function updateTrackingComment(
  octokit: Octokits,
  context: ParsedGitHubContext,
  commentId: number,
  branch?: string,
) {
  const { owner, repo } = context.repository;

  const jobRunLink = createJobRunLink(owner, repo, context.runId);

  // Add branch link for issues (not PRs)
  let branchLink = "";
  if (branch && !context.isPR) {
    branchLink = createBranchLink(owner, repo, branch);
  }

  const updatedBody = createCommentBody(jobRunLink, branchLink);

  // Update the existing comment with the branch link
  try {
    if (isPullRequestReviewCommentEvent(context)) {
      // For PR review comments (inline comments), use the pulls API
      await octokit.rest.pulls.updateReviewComment({
        owner,
        repo,
        comment_id: commentId,
        body: updatedBody,
      });
      console.log(`✅ Updated PR review comment ${commentId} with branch link`);
    } else {
      // For all other comments, use the issues API
      await octokit.rest.issues.updateComment({
        owner,
        repo,
        comment_id: commentId,
        body: updatedBody,
      });
      console.log(`✅ Updated issue comment ${commentId} with branch link`);
    }
  } catch (error) {
    console.error("Error updating comment with branch link:", error);
    throw error;
  }
}