src/model-routing-pr.ts

Functions

Imports

It imports src/env.ts, src/github/client.ts, src/github/pr-files.ts, src/model-routing.ts. Imported by src/jobs-claim.ts.

Source

1import { parsePrUrl, resolveRepo } from "./github/client";
2import { readPrFiles } from "./github/pr-files";
3import type { Env } from "./env";
4import { riskOf, type Risk } from "./model-routing";
5
6// The risk of the pull request a job already carries, read at claim so a job that came
7// back for a correction is routed on what it actually changed (riskOf in
8// src/model-routing.ts). Every page of files, read-only, through the one reader
9// (src/github/pr-files.ts). A pull request whose files were not read whole is high
10// risk, never silently routine, and the note says why.
11
12const unknownRisk = (reason: string, read: Risk | null): Risk => ({
13  level: "high",
14  reasons: [reason, ...(read?.level === "high" ? read.reasons : [])].slice(0, 12),
15  files_read: read?.files_read ?? 0,
16});
17
18export async function pullRequestRisk(env: Env, namespace: string, ref: string | null): Promise<{ risk: Risk | null; note: string | null }> {
19  const pr = parsePrUrl(ref);
20  if (!pr) return { risk: null, note: null };
21  try {
22    const { owner, repo } = await resolveRepo(env, namespace, `${pr.owner}/${pr.repo}`);
23    const files = await readPrFiles(env, owner, repo, pr.number);
24    const risk = riskOf(files.rows);
25    if (files.problem) {
26      return { risk: unknownRisk(`the files of ${ref} were not read whole`, risk), note: `routing could not read every file of ${ref} (${files.problem}), so the job was routed as high risk` };
27    }
28    if (risk.level === "unread") return { risk: unknownRisk(`no changed files of ${ref} were read`, risk), note: `routing read no changed files of ${ref}, so the job was routed as high risk` };
29    return { risk, note: null };
30  } catch (err) {
31    return { risk: unknownRisk(`the files of ${ref} could not be read`, null), note: `routing could not read ${ref}: ${(err instanceof Error ? err.message : String(err)).slice(0, 160)}, so the job was routed as high risk` };
32  }
33}