Files
upstream-atomic-crm/.claude/hooks/block-duplicate-dispatch.mjs
upstream-mirror 93ecc1a6f8
Some checks failed
✅ Check / 🔬 ESLint (push) Failing after 21s
✅ Check / 🏷️ Typecheck (push) Failing after 11s
✅ Check / 🔎 Test (push) Failing after 11s
✅ Check / e2e-test (push) Failing after 11s
✅ Check / 🔨 Build (push) Failing after 11s
🚀 Deploy / 🚀 Deploy (doc) (push) Failing after 12s
🚀 Deploy / 🚀 Deploy (demo) (push) Failing after 12s
🚀 Deploy / 🚀 Deploy (supabase) (push) Failing after 18s
snapshot of https://github.com/marmelab/atomic-crm.git (main, 2026-09-18; blobs > 131072B omitted)
2026-09-18 01:48:50 +05:00

155 lines
6.8 KiB
JavaScript
Executable File

#!/usr/bin/env node
// PreToolUse(Agent) — block accidental duplicate dispatches. Two concerns,
// grouped in one hook because both are "don't dispatch the same thing twice":
//
// 1. planner: at most ONE `planner` dispatch per request.
// Scope = caller orchestrator agent_id + TICKETS_DIR (one planning round).
// A 2nd planner would overwrite the TASK-*.json the wave is already running
// against.
//
// 2. developer / quality-reviewer / merger: debounce identical re-dispatches
// inside a short window. A foreground Agent call is SUPPOSED to block and
// return the subagent's final line inline; in some runtimes (interactive
// Claude Code) it instead returns immediately with "Async agent launched …
// agentId: <id>" and the real result arrives later as a task-notification.
import {
existsSync,
mkdirSync,
readFileSync,
statSync,
unlinkSync,
writeFileSync,
} from "node:fs";
import { join } from "node:path";
import { createHash } from "node:crypto";
import { createHookContext } from "./lib/context.mjs";
import { parseDispatch } from "./lib/dispatch-parse.mjs";
import { pipelineRoleSet, debounceRoleSet } from "./lib/teams.mjs";
// Sourced from harness.config.json (config.roles debounce/pipeline flags) via
// teams.mjs, so these sets and force-foreground's share one source of truth.
const DEBOUNCE_ROLES = debounceRoleSet();
const DEBOUNCE_WINDOW_MS = 90 * 1000;
const PLANNER_STALE_MS = 60 * 60 * 1000;
const input = JSON.parse(readFileSync(0, "utf8"));
const ctx = createHookContext(input, "block-duplicate-dispatch");
const d = parseDispatch(input);
const prompt = String(input.tool_input?.prompt ?? "");
// Without a caller agent_id we can't scope a marker safely — allow rather than
// risk over-blocking the only dispatch.
const caller = ctx.agentId;
if (!caller) process.exit(0);
const markerDir = join(ctx.sessionDir, "breaker");
try {
mkdirSync(markerDir, { recursive: true });
} catch {
process.exit(0); // can't persist a marker → don't risk blocking a real dispatch
}
const sha = (s) => createHash("sha1").update(s).digest("hex").slice(0, 16);
// A pipeline dispatch WITHOUT an explicit run_in_background:false will be DENIED by
// force-foreground-orchestrator-dispatch and re-issued with false. Do NOT record or
// check a marker for it here: otherwise the denied attempt's marker would reject the
// corrective retry as a "duplicate" (a regression from force-foreground's deny-and-retry,
// seen when the planner never ran yet its marker blocked the
// retry for 60 min). Only debounce dispatches that will actually proceed (rib === false).
const PIPELINE_ROLES = pipelineRoleSet();
const childRole = PIPELINE_ROLES.has(d.subagentType)
? d.subagentType
: PIPELINE_ROLES.has(d.role)
? d.role
: "";
if (childRole && input.tool_input?.run_in_background !== false) process.exit(0);
// ---- Concern 1: at most one planner per request -------------------------------
if (d.subagentType === "planner") {
// The STATE A planner template writes `TICKETS_DIR=<path>` (equals);
// parseDispatch only captures the `TICKETS_DIR:` (colon) form, so accept both.
const promptTicketsDir =
(prompt.match(/^TICKETS_DIR[:=]\s*(\S+)/m) || [])[1] || "";
const ticketsDir = d.ticketsDir || promptTicketsDir || ctx.ticketsDir || "";
const marker = join(markerDir, `planner-${sha(`${caller}::${ticketsDir}`)}`);
if (existsSync(marker)) {
try {
// Stale if older than the window — OR if its mtime is in the FUTURE
// (clock skew / NFS drift): a negative age would otherwise read as "fresh"
// forever and permanently block re-planning, with no escape (the
// orchestrator can't rm a breaker marker — bash-guard blocks that).
const age = Date.now() - statSync(marker).mtimeMs;
if (age > PLANNER_STALE_MS || age < 0) unlinkSync(marker);
} catch {
// ignore — fall through to the existence check
}
}
if (existsSync(marker)) {
ctx.block({
reason:
`A \`planner\` has ALREADY been dispatched for this request (TICKETS_DIR=${ticketsDir || "(session default)"}). ` +
`Do NOT dispatch a second planner — a re-plan overwrites the existing TASK-*.json while the wave may already be running against them. ` +
`Read the tickets already in TICKETS_DIR and continue into STATE B with them. ` +
`If the first plan looks wrong, fix the ticket JSON in place — do not re-run the planner.`,
log: `BLOCK 2nd planner caller=${caller} ticketsDir=${ticketsDir || "(default)"}`,
});
}
try {
writeFileSync(marker, `${caller} ${ticketsDir}\n`);
} catch {
// if we can't record the marker, still allow this (first) planner through
}
ctx.log(
`ALLOW 1st planner caller=${caller} ticketsDir=${ticketsDir || "(default)"}`,
);
process.exit(0);
}
// ---- Concern 2: debounce duplicate developer/reviewer/merger dispatches -------
if (DEBOUNCE_ROLES.has(d.subagentType)) {
// Key on the ticket identity AND the prompt content. The async-ack duplicate
// re-issues the IDENTICAL dispatch prompt, so an identical (caller, role,
// ticket, prompt) inside the window collides and is blocked. A genuine retry
// carries different prompt text (RETRY_FEEDBACK / FIX findings) → different
// key → allowed even inside the window (keying on the TASK id alone would
// wrongly debounce a fast-failing ticket's legitimate re-dispatch). The
// TASK id, when present, is kept as the human-readable label for messages.
const label = d.taskId || `prompt:${sha(prompt)}`;
const idPart = `${d.taskId || "_"}::${sha(prompt)}`;
const marker = join(
markerDir,
`dispatch-${sha(`${caller}::${d.subagentType}::${idPart}`)}`,
);
if (existsSync(marker)) {
let ageMs = Infinity;
try {
ageMs = Date.now() - statSync(marker).mtimeMs;
} catch {
// unreadable mtime → treat as stale, fall through to refresh + allow
}
if (ageMs < DEBOUNCE_WINDOW_MS) {
ctx.block({
reason:
`A \`${d.subagentType}\` dispatch for ${label} was made ${Math.round(ageMs / 1000)}s ago and is still in flight. ` +
`A dispatch can return immediately as "Async agent launched … agentId: <id>" WITHOUT blocking — that acknowledgement means "dispatched", not "done". ` +
`Do NOT re-dispatch the same role for the same ticket: wait for its task-notification, then read the agent's output file and parse its contract line. ` +
`Two ${d.subagentType}s on the same ticket race on one worktree/branch — that is always a bug.`,
log: `BLOCK duplicate ${d.subagentType} key=${idPart} age=${Math.round(ageMs / 1000)}s caller=${caller}`,
});
}
}
try {
writeFileSync(marker, `${caller} ${d.subagentType} ${idPart}\n`);
} catch {
// can't record → allow this dispatch through
}
ctx.log(`ALLOW ${d.subagentType} dispatch key=${idPart} caller=${caller}`);
process.exit(0);
}
process.exit(0);