Pull Request Code Suggestions with Amazon Q Developer

Do a review of the files in changes.txt for any CRITICAL or HIGH issues.
Put any review comments in seperate json files under the 'amazon-q/review' directory the json files must be in the format

{
  "filepath": <relative path to the file>,
  "comment": <The review comment>,
  "suggested_change": <the suggested code change>,
  "start_line": <the start line of the suggestion>,
  "end_line": <the end line of the suggestion>
}

// Import required Node.js modules
import fs from 'fs';
import { Octokit } from '@octokit/rest';
import path from 'path';
import { fileURLToPath } from 'url';

// Configure logging utility to add timestamps to log messages
const logger = {
  info: (msg) => console.log(`INFO: ${new Date().toISOString()} - ${msg}`)
};

async function addPrLineComments(repoName, prNumber, directoryPath, githubToken) {
  /**
   * Add line-specific review comments to a GitHub PR for files in a directory
   * @param {string} repoName - GitHub repository in format owner/repo
   * @param {number} prNumber - Pull request number
   * @param {string} directoryPath - Path to directory containing review files
   * @param {string} githubToken - GitHub authentication token
   */
  // Initialize GitHub API client with auth token
  logger.info(`Initializing GitHub client for repo ${repoName}`);
  const octokit = new Octokit({ auth: githubToken });

  // Split repository name into owner and repo parts
  logger.info(`Getting repository ${repoName}`);
  const [owner, repo] = repoName.split('/');
  logger.info(`Getting PR #${prNumber}`);

  // Get list of files in the specified directory
  logger.info(`Processing files in directory: ${directoryPath}`);
  const files = fs.readdirSync(directoryPath);

  // Process each file in the directory
  for (const filename of files) {
    const filePath = path.join(directoryPath, filename);
    // Only process regular files, not directories
    if (fs.statSync(filePath).isFile()) {
      try {
        logger.info(`Processing file: ${filename}`);
        // Read and parse the review data JSON file
        const content = fs.readFileSync(filePath, 'utf8');
        const reviewData = JSON.parse(content);

        // Get the latest commit SHA from the pull request
        const { data: pullRequest } = await octokit.pulls.get({
          owner,
          repo,
          pull_number: prNumber
        });
        const commitSha = pullRequest.head.sha;

        // Determine if comment spans multiple lines
        let multiLine = true;
        if (reviewData.start_line == reviewData.end_line)
          multiLine = false;

        // Log the comment being added
        logger.info(`Adding review comment to ${filename} at line ${reviewData.end_line}`);

        // Create GitHub review comment - different API calls for single vs multi-line
        if (multiLine) {
          // Multi-line comment requires start_line and start_side parameters
          await octokit.pulls.createReviewComment({
            owner,
            repo,
            pull_number: prNumber,
            body: `${reviewData.comment}\n\n\`\`\`suggestion\n${reviewData.suggested_change}\n\`\`\``,
            commit_id: commitSha,
            path: reviewData.filepath,
            line: reviewData.end_line,
            start_line: reviewData.start_line,
            start_side: 'RIGHT',
            side: 'RIGHT'
          });
        }
        else {
          // Single line comment has simpler parameters
          await octokit.pulls.createReviewComment({
            owner,
            repo,
            pull_number: prNumber,
            body: `${reviewData.comment}\n\n\`\`\`suggestion\n${reviewData.suggested_change}\n\`\`\``,
            commit_id: commitSha,
            path: reviewData.filepath,
            line: reviewData.end_line,
            side: 'RIGHT'
          });
        }
        logger.info(`Successfully added comment to ${filename}`);
      } catch (error) {
        // Log any errors that occur during processing
        logger.info(`Failed to process review comment for file ${filename}: ${error.message}`);
        console.error(`Error details:`, error);
      }
    }
  }
}

// Get the current file name using ES modules
const __filename = fileURLToPath(import.meta.url);
// Check if this file is being run directly
if (import.meta.url === `file://${__filename}`) {
  // Load configuration from environment variables
  const GITHUB_TOKEN = process.env.GITHUB_TOKEN;
  const REPO_NAME = process.env.GITHUB_REPOSITORY;
  const PR_NUMBER = parseInt(process.env.PR_NUMBER);
  const DIRECTORY_PATH = process.env.DIRECTORY_PATH || "amazon-q/review";

  // Execute the main function
  logger.info("Starting PR comment process");
  addPrLineComments(REPO_NAME, PR_NUMBER, DIRECTORY_PATH, GITHUB_TOKEN)
    .then(() => logger.info("Completed PR comment process"))
    .catch(error => console.error("Error:", error));
}