utils.ts

import { expect } from "@jest/globals";
import { RequestError } from "@octokit/request-error";
import EventEmitter from "events";
import { Octokit } from "octokit";
import YAML from "yaml";
import { BotConfig } from "../types/configuration-types";

export const webhookEventEmitter = new EventEmitter();

export function waitForNWebhooks(n = 1) {
  return new Promise<void>((resolve, reject) => {
    let i = 0;

    const timer = setTimeout(() => {
      reject(new Error(`timeout, received ${i} webhooks, expected ${n}`));
    }, 30000);

    webhookEventEmitter.on("event", () => {
      i += 1;
      if (i == n) {
        clearTimeout(timer);
        setTimeout(resolve, 1000);
      }
    });
  });
}

export async function createLabel({ octokit, owner, repo, label, color }: CreateLabel) {
  try {
    await octokit.rest.issues.createLabel({
      owner,
      repo,
      name: label,
      color,
    });
  } catch (err: unknown) {
    if (err instanceof RequestError) {
      expect(err).toBeDefined();
      expect(err?.status).toBe(422);
    }
  }
}

export async function addLabelToIssue({ octokit, owner, repo, issueNumber, label }: LabelParams) {
  await octokit.rest.issues.addLabels({
    owner,
    repo,
    issue_number: issueNumber,
    labels: [label],
  });
}

export async function removeLabelFromIssue({ octokit, owner, repo, issueNumber, label }: LabelParams) {
  await octokit.rest.issues.removeLabel({
    owner,
    repo,
    issue_number: issueNumber,
    name: label,
  });
}
export async function createAndAddLabel({ octokit, owner, repo, issueNumber, label }: LabelParams) {
  try {
    await octokit.rest.issues.createLabel({
      owner,
      repo,
      name: label,
    });
  } catch (err: unknown) {
    if (err instanceof RequestError) {
      expect(err).toBeDefined();
      expect(err?.status).toBe(422);
    }
  } finally {
    await octokit.rest.issues.addLabels({
      owner,
      repo,
      issue_number: issueNumber,
      labels: [label],
    });
    await waitForNWebhooks(1);
  }
}

export async function updateConfig({ octokit, owner, repo, path, config }: UpdateConfig) {
  let sha: string | undefined = undefined;
  try {
    const fileContent = await octokit.rest.repos.getContent({
      owner,
      repo,
      path,
    });
    if (Array.isArray(fileContent.data)) {
      throw new Error("ubiquibot-config.yml should not be directory");
    }
    if (fileContent.data.type !== "file") {
      throw new Error("ubiquibot-config.yml is not a file");
    }
    sha = fileContent.data.sha;
  } catch (err: unknown) {
    if (err instanceof RequestError) {
      expect(err).toBeDefined();
      expect(err?.status).toBe(404);
    }
  }

  await octokit.rest.repos.createOrUpdateFileContents({
    owner,
    repo,
    path,
    message: "test(e2e): automated update config for test",
    content: Buffer.from(YAML.stringify(config)).toString("base64"),
    sha,
  });
}

export async function createComment({ octokit, owner, repo, issueNumber, body }: CreateComment) {
  await octokit.rest.issues.createComment({
    repo,
    owner,
    issue_number: issueNumber,
    body: body,
  });
}
interface GetLastComment {
  octokit: Octokit;
  owner: string;
  repo: string;
  issueNumber: number;
}
export async function getLastComment({ octokit, owner, repo, issueNumber }: GetLastComment) {
  const { data } = await octokit.rest.issues.listComments({
    repo,
    owner,
    issue_number: issueNumber,
    per_page: 100,
  });
  expect(data.length).toBeGreaterThan(0);
  return data[data.length - 1];
}

export default async function checkLastComment({
  octokit,
  owner,
  repo,
  issueNumber,
  expectedComment,
}: CheckLastComment) {
  const lastComment = await getLastComment({ octokit, owner, repo, issueNumber });
  expect(lastComment.body).toBe(expectedComment);
}

interface OctokitParams {
  octokit: Octokit;
  owner: string;
  repo: string;
}

interface IssueParams extends OctokitParams {
  issueNumber: number;
}

interface LabelParams extends IssueParams {
  label: string;
}

interface CreateLabel extends LabelParams {
  color?: string;
}

interface UpdateConfig extends OctokitParams {
  path: string;
  config: BotConfig;
}

interface CreateComment extends IssueParams {
  body: string;
}

interface CheckLastComment extends IssueParams {
  expectedComment: string;
}