Skip to content

Feature Request: Add sudo prefix support for command rewriting #146

@tmaarcxs

Description

@tmaarcxs

Problem

The RTK auto-rewrite hook currently doesn't handle commands prefixed with sudo. This means commands like:

  • sudo docker ps
  • sudo docker logs <container>
  • sudo git status

are not rewritten to their RTK equivalents, resulting in missed token savings when running commands that require elevated privileges.

Proposed Solution

Add logic to strip and preserve the sudo prefix (including flags like sudo -u user) before pattern matching, then restore it in the rewritten command.

Changes to .claude/hooks/rtk-rewrite.sh

  1. Extract sudo prefix after env var extraction:
# Strip leading sudo (with optional flags) for pattern matching
# e.g., "sudo docker ps" → match against "docker ps"
# e.g., "sudo -u root docker ps" → match against "docker ps"
# Handles: sudo, sudo -u user, sudo -i, sudo -E, sudo -v, sudo -k, etc.
SUDO_PREFIX=""
if echo "$MATCH_CMD" | grep -qE '^sudo([[:space:]]|$)'; then
  # Extract sudo and any flags (but stop at the actual command)
  SUDO_PREFIX=$(echo "$MATCH_CMD" | grep -oE '^sudo([[:space:]]+-[A-Za-z]+([[:space:]]+[^[:space:]]+)?)*[[:space:]]+' || echo "")
  if [ -n "$SUDO_PREFIX" ]; then
    MATCH_CMD="${MATCH_CMD:${#SUDO_PREFIX}}"
    CMD_BODY="${CMD_BODY:${#SUDO_PREFIX}}"
  fi
fi
  1. Update all REWRITTEN assignments to include ${SUDO_PREFIX}:
# Before
REWRITTEN="${ENV_PREFIX}rtk $CMD_BODY"

# After
REWRITTEN="${ENV_PREFIX}${SUDO_PREFIX}rtk $CMD_BODY"

Example Transformations

Input Output
sudo docker ps sudo rtk docker ps
sudo docker logs container sudo rtk docker logs container
sudo -u root git status sudo -u root rtk git status

Testing

Verified working with:

echo '{"tool_input":{"command":"sudo docker ps"}}' | ~/.claude/hooks/rtk-rewrite.sh
# Output: {"hookSpecificOutput":{..."updatedInput":{"command":"sudo rtk docker ps"}}}

Notes

  • The regex handles sudo, sudo -u user, sudo -i, sudo -E, and combinations
  • Follows the same pattern already used for env var prefix handling

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions