Initial commit

src/entrypoints/prepare.ts

@@ -0,0 +1,102 @@
+#!/usr/bin/env bun
+
+/**
+ * Prepare the Claude action by checking trigger conditions, verifying human actor,
+ * and creating the initial tracking comment
+ */
+
+import * as core from "@actions/core";
+import { setupGitHubToken } from "../github/token";
+import { checkTriggerAction } from "../github/validation/trigger";
+import { checkHumanActor } from "../github/validation/actor";
+import { checkWritePermissions } from "../github/validation/permissions";
+import { createInitialComment } from "../github/operations/comments/create-initial";
+import { setupBranch } from "../github/operations/branch";
+import { updateTrackingComment } from "../github/operations/comments/update-with-branch";
+import { prepareMcpConfig } from "../mcp/install-mcp-server";
+import { createPrompt } from "../create-prompt";
+import { createOctokit } from "../github/api/client";
+import { fetchGitHubData } from "../github/data/fetcher";
+import { parseGitHubContext } from "../github/context";
+
+async function run() {
+  try {
+    // Step 1: Setup GitHub token
+    const githubToken = await setupGitHubToken();
+    const octokit = createOctokit(githubToken);
+
+    // Step 2: Parse GitHub context (once for all operations)
+    const context = parseGitHubContext();
+
+    // Step 3: Check write permissions
+    const hasWritePermissions = await checkWritePermissions(
+      octokit.rest,
+      context,
+    );
+    if (!hasWritePermissions) {
+      throw new Error(
+        "Actor does not have write permissions to the repository",
+      );
+    }
+
+    // Step 4: Check trigger conditions
+    const containsTrigger = await checkTriggerAction(context);
+
+    if (!containsTrigger) {
+      console.log("No trigger found, skipping remaining steps");
+      return;
+    }
+
+    // Step 5: Check if actor is human
+    await checkHumanActor(octokit.rest, context);
+
+    // Step 6: Create initial tracking comment
+    const commentId = await createInitialComment(octokit.rest, context);
+
+    // Step 7: Fetch GitHub data (once for both branch setup and prompt creation)
+    const githubData = await fetchGitHubData({
+      octokits: octokit,
+      repository: `${context.repository.owner}/${context.repository.repo}`,
+      prNumber: context.entityNumber.toString(),
+      isPR: context.isPR,
+    });
+
+    // Step 8: Setup branch
+    const branchInfo = await setupBranch(octokit, githubData, context);
+
+    // Step 9: Update initial comment with branch link (only for issues that created a new branch)
+    if (branchInfo.claudeBranch) {
+      await updateTrackingComment(
+        octokit,
+        context,
+        commentId,
+        branchInfo.claudeBranch,
+      );
+    }
+
+    // Step 10: Create prompt file
+    await createPrompt(
+      commentId,
+      branchInfo.defaultBranch,
+      branchInfo.claudeBranch,
+      githubData,
+      context,
+    );
+
+    // Step 11: Get MCP configuration
+    const mcpConfig = await prepareMcpConfig(
+      githubToken,
+      context.repository.owner,
+      context.repository.repo,
+      branchInfo.currentBranch,
+    );
+    core.setOutput("mcp_config", mcpConfig);
+  } catch (error) {
+    core.setFailed(`Prepare step failed with error: ${error}`);
+    process.exit(1);
+  }
+}
+
+if (import.meta.main) {
+  run();
+}

src/entrypoints/update-comment-link.ts

@@ -0,0 +1,231 @@
+#!/usr/bin/env bun
+
+import { createOctokit } from "../github/api/client";
+import * as fs from "fs/promises";
+import {
+  updateCommentBody,
+  type CommentUpdateInput,
+} from "../github/operations/comment-logic";
+import {
+  parseGitHubContext,
+  isPullRequestReviewCommentEvent,
+} from "../github/context";
+import { GITHUB_SERVER_URL } from "../github/api/config";
+import { checkAndDeleteEmptyBranch } from "../github/operations/branch-cleanup";
+
+async function run() {
+  try {
+    const commentId = parseInt(process.env.CLAUDE_COMMENT_ID!);
+    const githubToken = process.env.GITHUB_TOKEN!;
+    const claudeBranch = process.env.CLAUDE_BRANCH;
+    const defaultBranch = process.env.DEFAULT_BRANCH || "main";
+    const triggerUsername = process.env.TRIGGER_USERNAME;
+
+    const context = parseGitHubContext();
+    const { owner, repo } = context.repository;
+    const octokit = createOctokit(githubToken);
+
+    const serverUrl = GITHUB_SERVER_URL;
+    const jobUrl = `${serverUrl}/${owner}/${repo}/actions/runs/${process.env.GITHUB_RUN_ID}`;
+
+    let comment;
+    let isPRReviewComment = false;
+
+    try {
+      // GitHub has separate ID namespaces for review comments and issue comments
+      // We need to use the correct API based on the event type
+      if (isPullRequestReviewCommentEvent(context)) {
+        // For PR review comments, use the pulls API
+        console.log(`Fetching PR review comment ${commentId}`);
+        const { data: prComment } = await octokit.rest.pulls.getReviewComment({
+          owner,
+          repo,
+          comment_id: commentId,
+        });
+        comment = prComment;
+        isPRReviewComment = true;
+        console.log("Successfully fetched as PR review comment");
+      }
+
+      // For all other event types, use the issues API
+      if (!comment) {
+        console.log(`Fetching issue comment ${commentId}`);
+        const { data: issueComment } = await octokit.rest.issues.getComment({
+          owner,
+          repo,
+          comment_id: commentId,
+        });
+        comment = issueComment;
+        isPRReviewComment = false;
+        console.log("Successfully fetched as issue comment");
+      }
+    } catch (finalError) {
+      // If all attempts fail, try to determine more information about the comment
+      console.error("Failed to fetch comment. Debug info:");
+      console.error(`Comment ID: ${commentId}`);
+      console.error(`Event name: ${context.eventName}`);
+      console.error(`Entity number: ${context.entityNumber}`);
+      console.error(`Repository: ${context.repository.full_name}`);
+
+      // Try to get the PR info to understand the comment structure
+      try {
+        const { data: pr } = await octokit.rest.pulls.get({
+          owner,
+          repo,
+          pull_number: context.entityNumber,
+        });
+        console.log(`PR state: ${pr.state}`);
+        console.log(`PR comments count: ${pr.comments}`);
+        console.log(`PR review comments count: ${pr.review_comments}`);
+      } catch {
+        console.error("Could not fetch PR info for debugging");
+      }
+
+      throw finalError;
+    }
+
+    const currentBody = comment.body ?? "";
+
+    // Check if we need to add branch link for new branches
+    const { shouldDeleteBranch, branchLink } = await checkAndDeleteEmptyBranch(
+      octokit,
+      owner,
+      repo,
+      claudeBranch,
+      defaultBranch,
+    );
+
+    // Check if we need to add PR URL when we have a new branch
+    let prLink = "";
+    // If claudeBranch is set, it means we created a new branch (for issues or closed/merged PRs)
+    if (claudeBranch && !shouldDeleteBranch) {
+      // Check if comment already contains a PR URL
+      const serverUrlPattern = serverUrl.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+      const prUrlPattern = new RegExp(
+        `${serverUrlPattern}\\/.+\\/compare\\/${defaultBranch.replace(/[.*+?^${}()|[\]\\]/g, "\\$&")}\\.\\.\\.`,
+      );
+      const containsPRUrl = currentBody.match(prUrlPattern);
+
+      if (!containsPRUrl) {
+        // Check if there are changes to the branch compared to the default branch
+        try {
+          const { data: comparison } =
+            await octokit.rest.repos.compareCommitsWithBasehead({
+              owner,
+              repo,
+              basehead: `${defaultBranch}...${claudeBranch}`,
+            });
+
+          // If there are changes (commits or file changes), add the PR URL
+          if (
+            comparison.total_commits > 0 ||
+            (comparison.files && comparison.files.length > 0)
+          ) {
+            const entityType = context.isPR ? "PR" : "Issue";
+            const prTitle = encodeURIComponent(
+              `${entityType} #${context.entityNumber}: Changes from Claude`,
+            );
+            const prBody = encodeURIComponent(
+              `This PR addresses ${entityType.toLowerCase()} #${context.entityNumber}\n\nGenerated with [Claude Code](https://claude.ai/code)`,
+            );
+            const prUrl = `${serverUrl}/${owner}/${repo}/compare/${defaultBranch}...${claudeBranch}?quick_pull=1&title=${prTitle}&body=${prBody}`;
+            prLink = `\n[Create a PR](${prUrl})`;
+          }
+        } catch (error) {
+          console.error("Error checking for changes in branch:", error);
+          // Don't fail the entire update if we can't check for changes
+        }
+      }
+    }
+
+    // Check if action failed and read output file for execution details
+    let executionDetails: {
+      cost_usd?: number;
+      duration_ms?: number;
+      duration_api_ms?: number;
+    } | null = null;
+    let actionFailed = false;
+
+    // Check for existence of output file and parse it if available
+    try {
+      const outputFile = process.env.OUTPUT_FILE;
+      if (outputFile) {
+        const fileContent = await fs.readFile(outputFile, "utf8");
+        const outputData = JSON.parse(fileContent);
+
+        // Output file is an array, get the last element which contains execution details
+        if (Array.isArray(outputData) && outputData.length > 0) {
+          const lastElement = outputData[outputData.length - 1];
+          if (
+            lastElement.role === "system" &&
+            "cost_usd" in lastElement &&
+            "duration_ms" in lastElement
+          ) {
+            executionDetails = {
+              cost_usd: lastElement.cost_usd,
+              duration_ms: lastElement.duration_ms,
+              duration_api_ms: lastElement.duration_api_ms,
+            };
+          }
+        }
+      }
+
+      // Check if the action failed by looking at the exit code or error marker
+      const claudeSuccess = process.env.CLAUDE_SUCCESS !== "false";
+      actionFailed = !claudeSuccess;
+    } catch (error) {
+      console.error("Error reading output file:", error);
+      // If we can't read the file, check for any failure markers
+      actionFailed = process.env.CLAUDE_SUCCESS === "false";
+    }
+
+    // Prepare input for updateCommentBody function
+    const commentInput: CommentUpdateInput = {
+      currentBody,
+      actionFailed,
+      executionDetails,
+      jobUrl,
+      branchLink,
+      prLink,
+      branchName: shouldDeleteBranch ? undefined : claudeBranch,
+      triggerUsername,
+    };
+
+    const updatedBody = updateCommentBody(commentInput);
+
+    // Update the comment using the appropriate API
+    try {
+      if (isPRReviewComment) {
+        await octokit.rest.pulls.updateReviewComment({
+          owner,
+          repo,
+          comment_id: commentId,
+          body: updatedBody,
+        });
+      } else {
+        await octokit.rest.issues.updateComment({
+          owner,
+          repo,
+          comment_id: commentId,
+          body: updatedBody,
+        });
+      }
+      console.log(
+        `✅ Updated ${isPRReviewComment ? "PR review" : "issue"} comment ${commentId} with job link`,
+      );
+    } catch (updateError) {
+      console.error(
+        `Failed to update ${isPRReviewComment ? "PR review" : "issue"} comment:`,
+        updateError,
+      );
+      throw updateError;
+    }
+
+    process.exit(0);
+  } catch (error) {
+    console.error("Error updating comment with job link:", error);
+    process.exit(1);
+  }
+}
+
+run();