import { Octokit } from "@octokit/rest";
import { z } from "zod";

const octokit = new Octokit({ auth: process.env.GITHUB_TOKEN });

export const WorkflowFailureWebhookSchema = z.object({
  action: z.string(),
  workflow_job: z.object({
    id: z.number(),
    run_id: z.number(),
    name: z.string(),
    conclusion: z.string().nullable(),
  }),
  repository: z.object({
    owner: z.object({ login: z.string() }),
    name: z.string(),
  }),
});

export type WorkflowFailureWebhook = z.infer<typeof WorkflowFailureWebhookSchema>;

export async function handleWorkflowJobFailure(payload: WorkflowFailureWebhook) {
  if (payload.workflow_job.conclusion !== "failure") {
    return { status: "IGNORED" };
  }

  const { owner, name: repo } = {
    owner: payload.repository.owner.login,
    name: payload.repository.name,
  };

  // 1. Download raw job logs from GitHub API
  const logsResponse = await octokit.actions.downloadJobLogsForWorkflowRun({
    owner,
    repo,
    job_id: payload.workflow_job.id,
  });

  const rawLogs = String(logsResponse.data);

  // 2. Extract failing test suites and stack traces
  const jestFailureRegex = /FAIL\s+([\w\-/.]+\.test\.(ts|tsx|js))/g;
  const matches = [...rawLogs.matchAll(jestFailureRegex)];

  if (matches.length > 0) {
    const failingTestFiles = matches.map((m) => m[1]);
    console.log(`[CI Agent] Detected failing test files: ${failingTestFiles.join(", ")}`);

    // 3. Post diagnosis comment on the associated pull request
    return {
      status: "TRIAGED",
      category: "TEST_FAILURE",
      failingFiles: failingTestFiles,
      suggestedAction: "Re-run with test isolation or tag test file with @flaky.",
    };
  }

  return {
    status: "TRIAGED",
    category: "BUILD_OR_ENV_FAILURE",
    suggestedAction: "Check package dependencies or lockfile synchronization.",
  };
}