Merge pull request 'fix: feat: PreToolUse hook guards destructive operations in dev-agent sessions (#277)' (#329) from fix/issue-277 into main

This commit is contained in:
johba 2026-03-20 01:25:19 +01:00
commit e3b4c733b9
2 changed files with 113 additions and 0 deletions

View file

@ -45,6 +45,7 @@ agent_inject_into_session() {
# Create a tmux session running Claude in the given workdir. # Create a tmux session running Claude in the given workdir.
# Installs a Stop hook for idle detection (see monitor_phase_loop). # Installs a Stop hook for idle detection (see monitor_phase_loop).
# Installs a PreToolUse hook to guard destructive Bash operations.
# Optionally installs a PostToolUse hook for phase file write detection. # Optionally installs a PostToolUse hook for phase file write detection.
# Args: session workdir [phase_file] # Args: session workdir [phase_file]
# Returns 0 if session is ready, 1 otherwise. # Returns 0 if session is ready, 1 otherwise.
@ -120,6 +121,37 @@ create_agent_session() {
fi fi
fi fi
# Install PreToolUse hook for destructive operation guard: blocks force push
# to primary branch, rm -rf outside worktree, direct API merge calls, and
# checkout/switch to primary branch. Claude sees the denial reason on exit 2
# and can self-correct.
local guard_hook_script="${FACTORY_ROOT}/lib/hooks/on-pretooluse-guard.sh"
if [ -x "$guard_hook_script" ]; then
local abs_workdir
abs_workdir=$(cd "$workdir" 2>/dev/null && pwd) || abs_workdir="$workdir"
local guard_hook_cmd="${guard_hook_script} ${PRIMARY_BRANCH:-main} ${abs_workdir}"
if [ -f "$settings" ]; then
jq --arg cmd "$guard_hook_cmd" '
if (.hooks.PreToolUse // [] | any(.[]; .hooks[]?.command == $cmd))
then .
else .hooks.PreToolUse = (.hooks.PreToolUse // []) + [{
matcher: "Bash",
hooks: [{type: "command", command: $cmd}]
}]
end
' "$settings" > "${settings}.tmp" && mv "${settings}.tmp" "$settings"
else
jq -n --arg cmd "$guard_hook_cmd" '{
hooks: {
PreToolUse: [{
matcher: "Bash",
hooks: [{type: "command", command: $cmd}]
}]
}
}' > "$settings"
fi
fi
# Install Stop hook for Matrix streaming: when MATRIX_THREAD_ID is set, # Install Stop hook for Matrix streaming: when MATRIX_THREAD_ID is set,
# each Claude response is posted to the Matrix thread so humans can follow. # each Claude response is posted to the Matrix thread so humans can follow.
local matrix_hook_script="${FACTORY_ROOT}/lib/hooks/on-stop-matrix.sh" local matrix_hook_script="${FACTORY_ROOT}/lib/hooks/on-stop-matrix.sh"

View file

@ -0,0 +1,81 @@
#!/bin/bash
# on-pretooluse-guard.sh — PreToolUse hook: guard destructive operations.
#
# Called by Claude Code before executing a Bash command in agent sessions.
# Blocks:
# - git push --force / -f to primary branch
# - rm -rf targeting paths outside the worktree
# - Direct Codeberg API merge calls (should go through phase protocol)
# - git checkout / git switch to primary branch (stay on feature branch)
#
# Usage (in .claude/settings.json):
# {"type":"command","command":"this-script <primary_branch> <worktree_path>"}
#
# Args: $1 = primary branch (default: main), $2 = worktree absolute path
#
# Exit 0: allow (tool proceeds)
# Exit 2: deny (reason on stdout — Claude sees it and can self-correct)
primary_branch="${1:-main}"
worktree_path="${2:-}"
input=$(cat)
# Extract the command string from hook JSON
command_str=$(printf '%s' "$input" | jq -r '.tool_input.command // empty' 2>/dev/null)
[ -z "$command_str" ] && exit 0
# --- Guard 1: force push to primary branch ---
# Also blocks bare "git push --force" (no branch arg) since the upstream
# tracking branch might point to the primary branch.
if printf '%s' "$command_str" | grep -qE '\bgit\s+push\b' \
&& printf '%s' "$command_str" | grep -qE '(--force|--force-with-lease|\s-[a-zA-Z]*f)\b'; then
if printf '%s' "$command_str" | grep -qw "$primary_branch"; then
printf 'BLOCKED: Force-pushing to %s is not allowed. Push to your feature branch instead.\n' "$primary_branch"
exit 2
fi
# Bare force push with no explicit branch — could target primary via upstream
if ! printf '%s' "$command_str" | grep -qE '\bgit\s+push\s+\S+\s+\S'; then
printf 'BLOCKED: Bare force-push without an explicit branch is not allowed (upstream may point to %s). Specify your feature branch: git push --force-with-lease origin <branch>\n' "$primary_branch"
exit 2
fi
fi
# --- Guard 2: rm -rf outside worktree ---
if [ -n "$worktree_path" ] \
&& printf '%s' "$command_str" | grep -qE '\brm\s+(-[a-zA-Z]*r[a-zA-Z]*f|-[a-zA-Z]*f[a-zA-Z]*r)\b'; then
# Extract absolute paths from the command
abs_paths=$(printf '%s' "$command_str" | grep -oE "/[^[:space:];|&>\"']+" || true)
if [ -n "$abs_paths" ]; then
while IFS= read -r p; do
[ -z "$p" ] && continue
case "$p" in
"${worktree_path}"/*|"${worktree_path}") ;; # Inside worktree — allow
/tmp/*|/tmp) ;; # Temp files — allow (agents use /tmp for scratch)
/dev/*) ;; # Device paths — allow
*)
printf 'BLOCKED: rm -rf targets %s which is outside the worktree (%s). Only delete files within your worktree.\n' "$p" "$worktree_path"
exit 2
;;
esac
done <<< "$abs_paths"
fi
fi
# --- Guard 3: Direct Codeberg API merge calls ---
if printf '%s' "$command_str" | grep -qE '/pulls/[0-9]+/merge'; then
printf 'BLOCKED: Direct API merge calls must go through the phase protocol. Push your changes and write PHASE:awaiting_ci — the orchestrator handles merges.\n'
exit 2
fi
# --- Guard 4: checkout/switch to primary branch ---
# Blocks: git checkout main, git switch main, git switch --detach main, etc.
# Allows: git checkout -b branch main, git checkout -- file
escaped_branch=$(printf '%s' "$primary_branch" | sed 's/[.[\*^$()+?{|]/\\&/g')
if printf '%s' "$command_str" | grep -qE "\bgit\s+(checkout|switch)\s+(-[^ ]+\s+)*${escaped_branch}\b" \
&& ! printf '%s' "$command_str" | grep -qE '(\s--\s|\s-[bBcC]\s)'; then
printf 'BLOCKED: Switching to %s is not allowed. Stay on your feature branch.\n' "$primary_branch"
exit 2
fi
exit 0