Bubberstation/tools/test_merge_bot/main.js

import fetch from "node-fetch";
import { createComment } from "./comment.js";

const TEST_MERGE_COMMENT_HEADER = "<!-- test_merge_bot:";

const { GET_TEST_MERGES_URL } = process.env;

if (!GET_TEST_MERGES_URL) {
  console.error("GET_TEST_MERGES_URL was not set.");
  process.exit(1);
}

export async function processTestMerges({ github, context }) {
  const rounds = await fetch(GET_TEST_MERGES_URL)
    .then(async (response) => {
      if (response.status !== 200) {
        return Promise.reject(
          `Failed to fetch test merges: ${
            response.status
          } ${await response.text()}`,
        );
      }

      return response;
    })
    .then((response) => response.json())
    .catch((error) => {
      console.error(error);
      process.exit(1);
    });

  // PR # -> server name -> test merge struct
  const testMergesPerPr = {};

  for (const round of rounds) {
    const { server, test_merges } = round;

    for (const testMerge of test_merges) {
      if (!testMergesPerPr[testMerge]) {
        testMergesPerPr[testMerge] = {};
      }

      if (!testMergesPerPr[testMerge][server]) {
        testMergesPerPr[testMerge][server] = [];
      }

      testMergesPerPr[testMerge][server].push(round);
    }
  }

  for (const [prNumber, servers] of Object.entries(testMergesPerPr)) {
    const comments = await github.graphql(
      `
		query($owner:String!, $repo:String!, $prNumber:Int!) {
			repository(owner: $owner, name: $repo) {
				pullRequest(number: $prNumber) {
					comments(last: 100) {
						nodes {
							body
							databaseId
							author {
								login
							}
						}
					}
				}
			}
		}`,
      {
        owner: context.repo.owner,
        repo: context.repo.repo,
        prNumber: parseInt(prNumber, 10),
      },
    );

    const existingComment = comments.repository.pullRequest.comments.nodes.find(
      (comment) =>
        comment.author?.login === "github-actions" &&
        comment.body.startsWith(TEST_MERGE_COMMENT_HEADER),
    );

    const newBody = createComment(servers, existingComment?.body);
    if (!newBody) {
      console.log(`No changes for PR #${prNumber}`);
      continue;
    }

    if (existingComment === undefined) {
      try {
        await github.rest.issues.createComment({
          owner: context.repo.owner,
          repo: context.repo.repo,
          issue_number: prNumber,
          body: newBody,
        });
      } catch (error) {
        if (error.status) {
          console.error(`Failed to create comment for #{prNumber}`);
          console.error(error);
          continue;
        } else {
          throw error;
        }
      }
    } else {
      try {
        await github.rest.issues.updateComment({
          owner: context.repo.owner,
          repo: context.repo.repo,
          comment_id: existingComment.databaseId,
          body: newBody,
        });
      } catch (error) {
        if (error.status) {
          console.error(`Failed to update comment for #{prNumber}`);
          console.error(error);
          continue;
        } else {
          throw error;
        }
      }
    }
  }
}