claude-dev-env 2.4.0 → 2.7.0

package/hooks/hooks_constants/unscoped_search_blocker_constants.py

@@ -0,0 +1,153 @@
+"""Constants for the unscoped-search PreToolUse Bash/PowerShell blocker.
+
+Holds tool names, program basenames, Get-ChildItem option sets, the clustered
+short-option pattern, unscoped root patterns (drive root, posix root alias, UNC
+share root), the shared flag/value token stride, and the deny message. Find
+global option flags reuse ``destructive_command_segment_constants`` so both find
+parsers stay on one set.
+"""
+
+from __future__ import annotations
+
+import re
+
+from hooks_constants.destructive_command_segment_constants import (
+    ALL_FIND_GLOBAL_OPTION_FLAGS_TAKING_A_VALUE,
+    ALL_FIND_GLOBAL_OPTION_FLAGS_WITHOUT_VALUE,
+    FIND_OPTIMIZATION_LEVEL_OPTION_PREFIX,
+)
+
+__all__ = [
+    "BASH_TOOL_NAME",
+    "POWERSHELL_TOOL_NAME",
+    "ALL_SUPPORTED_TOOL_NAMES",
+    "ALL_FIND_PROGRAM_BASENAMES",
+    "ALL_POWERSHELL_LISTING_PROGRAM_BASENAMES",
+    "ALL_UNIX_LS_PROGRAM_BASENAMES",
+    "ALL_LISTING_PROGRAM_BASENAMES",
+    "ALL_FIND_GLOBAL_OPTION_FLAGS_WITHOUT_VALUE",
+    "ALL_FIND_GLOBAL_OPTION_FLAGS_TAKING_A_VALUE",
+    "FIND_OPTIMIZATION_LEVEL_OPTION_PREFIX",
+    "FLAG_AND_VALUE_TOKEN_STRIDE",
+    "ALL_FIND_EXPRESSION_INTRODUCER_TOKENS",
+    "ALL_POWERSHELL_RECURSE_FLAGS",
+    "ALL_UNIX_LS_RECURSE_FLAGS",
+    "UNIX_LS_RECURSE_SHORT_OPTION_LETTER",
+    "CLUSTERED_SHORT_OPTION_PATTERN",
+    "ALL_POWERSHELL_RECURSE_FLAG_PREFIXES",
+    "ALL_POWERSHELL_PATH_FLAGS",
+    "ALL_POWERSHELL_PATH_FLAG_PREFIXES",
+    "ALL_TRUTHY_RECURSE_COLON_VALUES",
+    "ALL_STRING_EXECUTING_SHELL_BASENAMES",
+    "ALL_STRING_EXEC_COMMAND_FLAGS",
+    "ALL_UNSCOPED_HOME_LITERALS_CASEFOLD",
+    "GIT_BASH_DRIVE_ROOT_PATTERN",
+    "WINDOWS_DRIVE_ROOT_PATTERN",
+    "POSIX_ROOT_ALIAS_PATTERN",
+    "UNC_SHARE_ROOT_PATTERN",
+    "HOOK_EVENT_NAME",
+    "DENY_DECISION",
+    "CALLING_HOOK_NAME",
+    "CORRECTIVE_MESSAGE",
+]
+
+BASH_TOOL_NAME = "Bash"
+POWERSHELL_TOOL_NAME = "PowerShell"
+ALL_SUPPORTED_TOOL_NAMES: frozenset[str] = frozenset(
+    {BASH_TOOL_NAME, POWERSHELL_TOOL_NAME}
+)
+
+ALL_FIND_PROGRAM_BASENAMES: frozenset[str] = frozenset({"find", "find.exe"})
+ALL_POWERSHELL_LISTING_PROGRAM_BASENAMES: frozenset[str] = frozenset(
+    {"get-childitem", "gci", "dir"}
+)
+ALL_UNIX_LS_PROGRAM_BASENAMES: frozenset[str] = frozenset({"ls", "ls.exe"})
+ALL_LISTING_PROGRAM_BASENAMES: frozenset[str] = (
+    ALL_POWERSHELL_LISTING_PROGRAM_BASENAMES | ALL_UNIX_LS_PROGRAM_BASENAMES
+)
+
+FLAG_AND_VALUE_TOKEN_STRIDE = 2
+ALL_FIND_EXPRESSION_INTRODUCER_TOKENS: frozenset[str] = frozenset({"!", "(", ")"})
+
+ALL_POWERSHELL_RECURSE_FLAGS: frozenset[str] = frozenset(
+    {"-recurse", "-r", "-rec", "/s"}
+)
+ALL_UNIX_LS_RECURSE_FLAGS: frozenset[str] = frozenset({"-R", "--recursive"})
+UNIX_LS_RECURSE_SHORT_OPTION_LETTER = "R"
+CLUSTERED_SHORT_OPTION_PATTERN = re.compile(r"^-[A-Za-z]+$")
+ALL_POWERSHELL_RECURSE_FLAG_PREFIXES: tuple[str, ...] = (
+    "-recurse:",
+    "-r:",
+)
+ALL_POWERSHELL_PATH_FLAGS: frozenset[str] = frozenset(
+    {"-path", "-literalpath", "-lp"}
+)
+ALL_POWERSHELL_PATH_FLAG_PREFIXES: tuple[str, ...] = (
+    "-path:",
+    "-literalpath:",
+    "-lp:",
+)
+ALL_TRUTHY_RECURSE_COLON_VALUES: frozenset[str] = frozenset(
+    {"$true", "true", "1", "yes", "on"}
+)
+
+ALL_STRING_EXECUTING_SHELL_BASENAMES: frozenset[str] = frozenset(
+    {
+        "bash",
+        "bash.exe",
+        "sh",
+        "sh.exe",
+        "pwsh",
+        "pwsh.exe",
+        "powershell",
+        "powershell.exe",
+    }
+)
+ALL_STRING_EXEC_COMMAND_FLAGS: frozenset[str] = frozenset(
+    {"-c", "-lc", "-command", "-encodedcommand"}
+)
+
+_ALL_UNSCOPED_HOME_LITERALS: frozenset[str] = frozenset(
+    {
+        "~",
+        "~/",
+        "~\\",
+        "$HOME",
+        "$HOME/",
+        "$HOME\\",
+        "${HOME}",
+        "${HOME}/",
+        "${HOME}\\",
+        "%USERPROFILE%",
+        "%USERPROFILE%\\",
+        "%USERPROFILE%/",
+        "$env:USERPROFILE",
+        "$env:USERPROFILE\\",
+        "$env:USERPROFILE/",
+        "$env:HOME",
+        "$env:HOME\\",
+        "$env:HOME/",
+    }
+)
+ALL_UNSCOPED_HOME_LITERALS_CASEFOLD: frozenset[str] = frozenset(
+    each_literal.casefold() for each_literal in _ALL_UNSCOPED_HOME_LITERALS
+)
+
+GIT_BASH_DRIVE_ROOT_PATTERN = re.compile(r"^/[a-zA-Z]/?$")
+WINDOWS_DRIVE_ROOT_PATTERN = re.compile(r"^[a-zA-Z]:[\\/]*$")
+POSIX_ROOT_ALIAS_PATTERN = re.compile(r"^/(?:\.|/)*$")
+UNC_SHARE_ROOT_PATTERN = re.compile(r"^//[^/]+/[^/]+/*$")
+
+HOOK_EVENT_NAME = "PreToolUse"
+DENY_DECISION = "deny"
+CALLING_HOOK_NAME = "unscoped_search_blocker.py"
+
+CORRECTIVE_MESSAGE = (
+    "Unscoped filesystem search blocked. Never start find/Get-ChildItem at `/`, "
+    "a drive root (`/c`, `C:\\`), bare home (`~`, `$HOME`), or a network share "
+    "root (`\\\\server\\share`). Scope the walk to a project or worktree path "
+    "(for example `find . -name '*.py'` or "
+    "`find packages/claude-dev-env -iname code_rules_gate.py`). On Windows prefer "
+    "es.exe with a path scope, or the Grep/Glob tools. Batch shell work; avoid "
+    "parallel full-tree searches that contend for the shell and lock the host."
+)

package/hooks/lifecycle/enter_worktree_origin_prefetch.py

@@ -1,146 +1,163 @@
-#!/usr/bin/env python3
-"""PreToolUse hook: fetch origin's default branch before EnterWorktree creates one.
-
-``EnterWorktree`` in its default ``fresh`` mode bases a new worktree on the
-locally cached ``refs/remotes/origin/<default-branch>`` ref and only runs
-``git fetch`` when that ref is missing entirely -- so a worktree created
-without a recent fetch silently starts behind the remote. This hook fetches
-that ref immediately before the tool resolves its base, so the ref it reads
-is current.
-
-::
-
-    EnterWorktree({})                              # tool_input has no "path"
-             |
-             v
-    this hook: git fetch origin <default-branch>   # best-effort, never blocks
-             |
-             v
-    EnterWorktree resolves refs/remotes/origin/<default-branch>  # now fresh
-
-The hook is a no-op when ``tool_input`` carries a ``path`` (switching into an
-already-existing worktree needs no fresh base) and always exits 0 -- a failed
-or slow fetch never blocks worktree creation.
-"""
-
-from __future__ import annotations
-
-import json
-import subprocess
-import sys
-from pathlib import Path
-
-_hooks_dir = str(Path(__file__).resolve().parent.parent)
-if _hooks_dir not in sys.path:
-    sys.path.insert(0, _hooks_dir)
-
-from hooks_constants.enter_worktree_prefetch_constants import (  # noqa: E402
-    ENTER_WORKTREE_PATH_INPUT_KEY,
-    ENTER_WORKTREE_TOOL_NAME,
-    GIT_FETCH_TIMEOUT_SECONDS,
-    GIT_SYMBOLIC_REF_TIMEOUT_SECONDS,
-    ORIGIN_HEAD_SYMBOLIC_REF,
-    ORIGIN_REMOTE_NAME,
-    ORIGIN_REMOTE_REF_PREFIX,
-)
-
-
-def is_enter_worktree_creation(payload_by_field: dict[str, object]) -> bool:
-    """Return True when this hook invocation is an EnterWorktree creation call.
-
-    Args:
-        payload_by_field: The full PreToolUse hook payload (already JSON-parsed),
-            keyed by top-level field name.
-
-    Returns:
-        True when ``tool_name == "EnterWorktree"`` and ``tool_input`` carries
-        no ``path`` key. A ``path`` value means the call switches into an
-        already-existing worktree rather than creating one, so it is False
-        for that case and for every other tool.
-    """
-    if payload_by_field.get("tool_name", "") != ENTER_WORKTREE_TOOL_NAME:
-        return False
-    tool_input = payload_by_field.get("tool_input", {})
-    if not isinstance(tool_input, dict):
-        return True
-    return not tool_input.get(ENTER_WORKTREE_PATH_INPUT_KEY)
-
-
-def resolve_origin_default_branch(repo_directory: str) -> str | None:
-    """Return the branch name origin's HEAD points at, or None if unresolved.
-
-    Args:
-        repo_directory: Working directory to run ``git`` in.
-
-    Returns:
-        The branch name (for example ``"main"``) parsed from
-        ``refs/remotes/origin/HEAD``, or None when the symbolic ref is
-        unset, git is unavailable, or the command times out.
-    """
-    try:
-        completed_process = subprocess.run(
-            ["git", "symbolic-ref", ORIGIN_HEAD_SYMBOLIC_REF],
-            capture_output=True,
-            text=True,
-            timeout=GIT_SYMBOLIC_REF_TIMEOUT_SECONDS,
-            cwd=repo_directory,
-            check=False,
-        )
-    except (subprocess.TimeoutExpired, FileNotFoundError, OSError):
-        return None
-    if completed_process.returncode != 0:
-        return None
-    resolved_ref = completed_process.stdout.strip()
-    if not resolved_ref.startswith(ORIGIN_REMOTE_REF_PREFIX):
-        return None
-    return resolved_ref[len(ORIGIN_REMOTE_REF_PREFIX):]
-
-
-def fetch_origin_branch(repo_directory: str, branch_name: str) -> None:
-    """Best-effort ``git fetch origin <branch_name>``; never raises.
-
-    Args:
-        repo_directory: Working directory to run ``git`` in.
-        branch_name: Branch to fetch from the ``origin`` remote.
-    """
-    try:
-        subprocess.run(
-            ["git", "fetch", ORIGIN_REMOTE_NAME, branch_name],
-            capture_output=True,
-            text=True,
-            timeout=GIT_FETCH_TIMEOUT_SECONDS,
-            cwd=repo_directory,
-            check=False,
-        )
-    except (subprocess.TimeoutExpired, FileNotFoundError, OSError):
-        return
-
-
-def main() -> None:
-    """Entry point for the PreToolUse:EnterWorktree hook.
-
-    Reads the PreToolUse payload from stdin, and when it is an EnterWorktree
-    creation call, fetches origin's default branch in the session's ``cwd``
-    before returning. Always exits 0: a fetch failure self-heals on the next
-    fetch rather than blocking worktree creation.
-    """
-    try:
-        hook_payload = json.load(sys.stdin)
-    except json.JSONDecodeError:
-        sys.exit(0)
-    if not isinstance(hook_payload, dict):
-        sys.exit(0)
-    if not is_enter_worktree_creation(hook_payload):
-        sys.exit(0)
-    repo_directory = hook_payload.get("cwd")
-    if not isinstance(repo_directory, str) or not repo_directory:
-        sys.exit(0)
-    default_branch = resolve_origin_default_branch(repo_directory)
-    if default_branch is None:
-        sys.exit(0)
-    fetch_origin_branch(repo_directory, default_branch)
-    sys.exit(0)
-
-
-if __name__ == "__main__":
-    main()
+#!/usr/bin/env python3
+"""PreToolUse hook: fetch origin's default branch before EnterWorktree creates one.
+
+``EnterWorktree`` in its default ``fresh`` mode bases a new worktree on the
+locally cached ``refs/remotes/origin/<default-branch>`` ref and only runs
+``git fetch`` when that ref is missing entirely -- so a worktree created
+without a recent fetch silently starts behind the remote. This hook fetches
+that ref immediately before the tool resolves its base, so the ref it reads
+is current.
+
+::
+
+    EnterWorktree({})                              # tool_input has no "path"
+             |
+             v
+    this hook: git fetch origin <default-branch>   # best-effort, never blocks
+             |
+             v
+    EnterWorktree resolves refs/remotes/origin/<default-branch>  # now fresh
+
+The hook is a no-op when ``tool_input`` carries a ``path`` (switching into an
+already-existing worktree needs no fresh base) and always exits 0 -- a failed
+or slow fetch never blocks worktree creation.
+"""
+
+from __future__ import annotations
+
+import subprocess
+import sys
+from pathlib import Path
+
+_hooks_dir = str(Path(__file__).resolve().parent.parent)
+if _hooks_dir not in sys.path:
+    sys.path.insert(0, _hooks_dir)
+
+from hooks_constants.enter_worktree_prefetch_constants import (  # noqa: E402
+    ENTER_WORKTREE_PATH_INPUT_KEY,
+    ENTER_WORKTREE_TOOL_NAME,
+    GIT_FETCH_TIMEOUT_SECONDS,
+    GIT_SYMBOLIC_REF_TIMEOUT_SECONDS,
+    ORIGIN_HEAD_SYMBOLIC_REF,
+    ORIGIN_REMOTE_NAME,
+    ORIGIN_REMOTE_REF_PREFIX,
+)
+from hooks_constants.pre_tool_use_stdin import read_hook_input_dictionary_from_stdin  # noqa: E402
+
+
+def is_enter_worktree_creation(payload_by_field: dict[str, object]) -> bool:
+    """Return True when this hook invocation is an EnterWorktree creation call.
+
+    Args:
+        payload_by_field: The full PreToolUse hook payload (already JSON-parsed),
+            keyed by top-level field name.
+
+    Returns:
+        True when ``tool_name == "EnterWorktree"`` and ``tool_input`` carries
+        no ``path`` key. A ``path`` value means the call switches into an
+        already-existing worktree rather than creating one, so it is False
+        for that case and for every other tool.
+    """
+    if payload_by_field.get("tool_name", "") != ENTER_WORKTREE_TOOL_NAME:
+        return False
+    tool_input = payload_by_field.get("tool_input", {})
+    if not isinstance(tool_input, dict):
+        return True
+    return not tool_input.get(ENTER_WORKTREE_PATH_INPUT_KEY)
+
+
+def _run_git_quietly(
+    repo_directory: str,
+    all_git_arguments: list[str],
+    timeout_seconds: int,
+) -> subprocess.CompletedProcess[str] | None:
+    """Run ``git`` with the given arguments, or return None when it cannot run.
+
+    Args:
+        repo_directory: Working directory to run ``git`` in.
+        all_git_arguments: The git subcommand and its arguments, without the
+            leading ``"git"`` (for example ``["fetch", "origin", "main"]``).
+        timeout_seconds: Seconds to wait before abandoning the command.
+
+    Returns:
+        The completed process, or None when git is unavailable, the OS refuses
+        to start it, or the command exceeds ``timeout_seconds``.
+    """
+    try:
+        return subprocess.run(
+            ["git", *all_git_arguments],
+            capture_output=True,
+            text=True,
+            timeout=timeout_seconds,
+            cwd=repo_directory,
+            check=False,
+        )
+    except (subprocess.TimeoutExpired, FileNotFoundError, OSError):
+        return None
+
+
+def resolve_origin_default_branch(repo_directory: str) -> str | None:
+    """Return the branch name origin's HEAD points at, or None if unresolved.
+
+    Args:
+        repo_directory: Working directory to run ``git`` in.
+
+    Returns:
+        The branch name (for example ``"main"``) parsed from
+        ``refs/remotes/origin/HEAD``, or None when the symbolic ref is
+        unset, git is unavailable, or the command times out.
+    """
+    completed_process = _run_git_quietly(
+        repo_directory,
+        ["symbolic-ref", ORIGIN_HEAD_SYMBOLIC_REF],
+        GIT_SYMBOLIC_REF_TIMEOUT_SECONDS,
+    )
+    if completed_process is None:
+        return None
+    if completed_process.returncode != 0:
+        return None
+    resolved_ref = completed_process.stdout.strip()
+    if not resolved_ref.startswith(ORIGIN_REMOTE_REF_PREFIX):
+        return None
+    return resolved_ref[len(ORIGIN_REMOTE_REF_PREFIX):]
+
+
+def fetch_origin_branch(repo_directory: str, branch_name: str) -> None:
+    """Best-effort ``git fetch origin <branch_name>``; never raises.
+
+    Args:
+        repo_directory: Working directory to run ``git`` in.
+        branch_name: Branch to fetch from the ``origin`` remote.
+    """
+    _run_git_quietly(
+        repo_directory,
+        ["fetch", ORIGIN_REMOTE_NAME, branch_name],
+        GIT_FETCH_TIMEOUT_SECONDS,
+    )
+
+
+def main() -> None:
+    """Entry point for the PreToolUse:EnterWorktree hook.
+
+    Reads the PreToolUse payload from stdin, and when it is an EnterWorktree
+    creation call, fetches origin's default branch in the session's ``cwd``
+    before returning. Always exits 0: a fetch failure self-heals on the next
+    fetch rather than blocking worktree creation.
+    """
+    hook_payload = read_hook_input_dictionary_from_stdin()
+    if hook_payload is None:
+        sys.exit(0)
+    if not is_enter_worktree_creation(hook_payload):
+        sys.exit(0)
+    repo_directory = hook_payload.get("cwd")
+    if not isinstance(repo_directory, str) or not repo_directory:
+        sys.exit(0)
+    default_branch = resolve_origin_default_branch(repo_directory)
+    if default_branch is None:
+        sys.exit(0)
+    fetch_origin_branch(repo_directory, default_branch)
+    sys.exit(0)
+
+
+if __name__ == "__main__":
+    main()