rx-attribution

1import type { EngineInterface, Register } from "claude-code";
2
3import { attributionLines } from "./attribution-lines";
4import { commits } from "./commits";
5import { gitDir } from "./git-dir";
6import { opensOrEditsPr } from "./opens-or-edits-pr";
7import { prUrl } from "./pr-url";
8import { withoutAttribution } from "./without-attribution";
9
10/**
11 * Reads a pull request's body and, when it credits Claude, removes those lines with gh pr edit
12 * (or only reports them when `isFixing` is off). Answers the note for the model, if any.
13 */
14const checkPr = async (
15  $: EngineInterface,
16  url: string,
17  isFixing: boolean,
18): Promise<string | undefined> => {
19  const view = await $.process.run(["gh", "pr", "view", url, "--json", "body", "-q", ".body"]);
20  if (view.exitCode !== 0) {
21    return `rx-attribution: could not read ${url} to check its body (${view.stderr.trim()}).`;
22  }
23
24  const found = attributionLines(view.stdout);
25  if (found.length === 0) {
26    return undefined;
27  }
28
29  if (!isFixing) {
30    return `rx-attribution: ${url} carries attribution: ${found.join(" | ")}. Remove it with gh pr edit.`;
31  }
32
33  const edit = await $.process.run(["gh", "pr", "edit", url, "--body-file", "-"], {
34    stdin: withoutAttribution(view.stdout),
35  });
36  if (edit.exitCode !== 0) {
37    return `rx-attribution: ${url} carries attribution (${found.join(" | ")}) and gh pr edit failed: ${edit.stderr.trim()}`;
38  }
39
40  $.ui.toast(`rx-attribution: removed ${found.length} attribution line(s) from ${url}`);
41
42  return `rx-attribution: removed attribution from ${url}'s body: ${found.join(" | ")}.`;
43};
44
45/**
46 * Reads the newest commit and answers a note for the model when its author or committer is not
47 * `email` (an empty `email` skips that) or its message credits Claude.
48 */
49const checkCommit = async (
50  $: EngineInterface,
51  cwd: string | undefined,
52  email: string,
53): Promise<string | undefined> => {
54  const log = await $.process.run(
55    ["git", "log", "-1", "--pretty=%an <%ae>%n%cn <%ce>%n%B"],
56    cwd === undefined ? undefined : { cwd },
57  );
58  if (log.exitCode !== 0) {
59    return undefined;
60  }
61
62  const [author = "", committer = "", ...message] = log.stdout.split("\n");
63  const problems: string[] = [];
64  if (email !== "" && !author.endsWith(`<${email}>`)) {
65    problems.push(`author is ${author}`);
66  }
67  if (email !== "" && !committer.endsWith(`<${email}>`)) {
68    problems.push(`committer is ${committer}`);
69  }
70  const found = attributionLines(message.join("\n"));
71  if (found.length > 0) {
72    problems.push(`message carries ${found.join(" | ")}`);
73  }
74
75  return problems.length === 0
76    ? undefined
77    : `rx-attribution: the new commit is wrong: ${problems.join("; ")}. Amend it before pushing.`;
78};
79
80export const register: Register = (on, options) => {
81  const email = typeof options.authorEmail === "string" ? options.authorEmail.trim() : "";
82  const isFixing = options.fixPrBody !== false;
83
84  on("attribution.text", { kind: "commit" }, () => ({ text: "" }));
85  on("attribution.text", { kind: "pr" }, () => ({ text: "" }));
86
87  on("tool.call", { tool: "Bash" }, async ($, e, next) => {
88    const ran = await next(e);
89    if (ran.deny !== undefined || ran.isError === true) {
90      return ran;
91    }
92
93    const notes: string[] = [];
94    if (opensOrEditsPr(e.command)) {
95      const url = prUrl(ran.text ?? "");
96      const note = url === undefined ? undefined : await checkPr($, url, isFixing);
97      if (note !== undefined) notes.push(note);
98    }
99    if (commits(e.command)) {
100      const note = await checkCommit($, gitDir(e.command), email);
101      if (note !== undefined) notes.push(note);
102    }
103
104    return notes.length === 0 ? ran : { ...ran, context: [...(ran.context ?? []), ...notes] };
105  }).catch(($, e, next) => next(e));
106};
107

1import { carriesAttribution } from "./carries-attribution";
2
3/** The lines of a text that credit Claude. */
4export const attributionLines = (text: string): string[] =>
5  text.split("\n").filter(carriesAttribution);
6

1/** Whether a shell command runs git commit. */
2export const commits = (command: string): boolean =>
3  /\bgit\s+(?:-[cC]\s+\S+\s+)*commit\b/.test(command);
4

1/** The directory a shell command runs git in, from `git -C <dir>` or a leading `cd <dir> &&`. */
2export const gitDir = (command: string): string | undefined => {
3  const word =
4    /\bgit\s+-C\s+("[^"]+"|'[^']+'|\S+)/.exec(command)?.[1] ??
5    /^\s*cd\s+("[^"]+"|'[^']+'|\S+)\s*&&/.exec(command)?.[1];
6
7  return word?.replace(/^["']|["']$/g, "");
8};
9

1/** Whether a shell command runs gh pr create or gh pr edit. */
2export const opensOrEditsPr = (command: string): boolean =>
3  /\bgh\s+pr\s+(create|edit)\b/.test(command);
4

1const PR_URL = /https:\/\/github\.com\/[\w.-]+\/[\w.-]+\/pull\/\d+/;
2
3/** The first GitHub pull request URL in a text, as gh pr create and gh pr edit print it. */
4export const prUrl = (text: string): string | undefined => PR_URL.exec(text)?.[0];
5

1import { carriesAttribution } from "./carries-attribution";
2
3/** The text with every line that credits Claude removed, and the blank runs that leaves collapsed. */
4export const withoutAttribution = (text: string): string =>
5  text
6    .split("\n")
7    .filter((line) => !carriesAttribution(line))
8    .join("\n")
9    .replace(/\n{3,}/g, "\n\n")
10    .trimEnd();
11

1const ATTRIBUTION = [
2  /^\s*co-authored-by:.*\b(claude|anthropic)\b/i,
3  /^\s*claude-session:/i,
4  /generated with \[?claude/i,
5  /🤖/u,
6  /claude\.ai\/code/i,
7  /\bsession_[A-Za-z0-9]{6,}/,
8];
9
10/** Whether a line credits Claude: a co-author trailer, a session link or the generated-with footer. */
11export const carriesAttribution = (line: string): boolean =>
12  ATTRIBUTION.some((pattern) => pattern.test(line));
13