brut.sh
· 3.2 KiB · Bash
Исходник
#!/bin/sh
# Brutal Router for Unix Tools - https://brut.sh
set -eu
# Find ourselves
OLDENV="$(export -p)"
SELF="$(realpath "$0")"
NAME="${0##*/}"
PROGRAM_NAME="${SELF##*/}"
ROOT="${SELF%/*/*}"
# Set up the environment
LIB="$ROOT/lib/$NAME"
LIBEXEC="$ROOT/libexec"
SHARE="$ROOT/share/$NAME"
VENDOR="$SHARE/vendor"
PATH="$LIB:$LIBEXEC:$VENDOR:$PATH"
export SELF
USAGE="<COMMAND>"
unset VERSION
# ---
# Define shared helpers
error() {
NAME="${1##*/}"
shift
printf "%s: " "$(invocation)"
printf "%s\n" "$@"
return 1
} >&2
usage() {
if [ $# -eq 0 ]; then
set -- "$NAME" "${USAGE:-}" "commands: $(flatten commands)"
fi
NAME="${1##*/}"
shift
printf "usage: %s " "$(invocation)"
printf "%s\n" "$@"
return 1
} >&2
version() {
printf "%s %s\n" "$PROGRAM_NAME" "$VERSION"
}
# ---
commands() {
executables | sed -e "s/^.*\/$NAME-//" | sort
}
executables() {
for ns in $(namespaces "$NAME-"); do
set -- "$@" \! -name "$ns-*"
done
searchpath -name "$NAME-*" "$@" | awk -F / '!a[$NF]++'
}
flatten() {
"$@" | { tr "\n" " "; printf "\n"; }
}
invocation() {
{ printf "%s\n" "$PROGRAM_NAME"; namespaces; } \
| sort -r | awk -v name="$NAME" \
'index(name, $0 "-") == 1 { name = $0 " " substr(name, length($0) + 2) }
END { print name }'
}
namespaces() {
searchpath -type l -name "${1:-$PROGRAM_NAME-}*" \
-exec sh -c '[ "$0" = "$(realpath "$1")" ] 2>/dev/null' "$SELF" {} \; \
| sed -e "s/^.*\///" | sort -u
}
resolve() {
case "${1:-}" in
*-* ) executables -name "$NAME-$1" ;;
?* ) searchpath -name "$NAME-$1" | head -n 1
esac | grep .
}
searchpath() {
{ IFS=:
dirs=
set -- \) -o -prune "$@"
for dir in $PATH; do
[ -d "$dir" ] || continue
set -- -o -path "$dir" "$@"
dirs="$dirs:$dir"
done
set -- ${dirs#:} \( \! -name '*' "$@"
unset IFS dir dirs
} 2>/dev/null
find -H "$@" 2>/dev/null \
\( -perm -001 -o -perm -010 -o -perm -100 \) \
\( -type f -o -type l \) \
\! -name "?*--?*" \
-print || true
}
unenv() {
exec env -i sh -euc 'eval "$1"; shift; exec "$@"' "$0" "$OLDENV" "$@"
}
# ---
# Load _init.sh, if it exists
if [ -r "$LIB"/_init.sh ]; then
. "$LIB"/_init.sh
fi
# Rewrite `--version` to `--- version`
if [ $# -eq 1 ] && [ -n "${VERSION:-}" ] && [ "$1" = "--version" ]; then
set -- --- version
fi
# When invoked as `--- <COMMAND> ...`, run that command and exit
if [ "${1:-}" = "---" ]; then
[ $# -gt 1 ] || usage
shift
"$@"
exit
fi
# Scan the argument list to find the command name
COMMAND=
args=
index=0
for arg; do
index=$((index+1))
case "$arg" in
-* ) ;;
?*--?* ) [ -n "$COMMAND" ] || usage ;;
* )
if [ -z "$COMMAND" ]; then
COMMAND="$arg"
continue
fi
esac
args="$args"' "${'"$index"'}"'
done
# Match the command name to an executable in the current namespace;
# if one exists, exec it
if executable="$(resolve "$COMMAND")"; then
eval "set --$args"
exec "$executable" "$@"
fi
# No matching executable; load _unhandled.sh, if it exists
if [ -r "$LIB"/_unhandled.sh ]; then
. "$LIB"/_unhandled.sh
fi
# No matching command or no command specified
if [ -n "$COMMAND" ]; then
error "$0" "$COMMAND: command not found" || exit 127
fi
usage
| 1 | #!/bin/sh |
| 2 | # Brutal Router for Unix Tools - https://brut.sh |
| 3 | set -eu |
| 4 | |
| 5 | |
| 6 | # Find ourselves |
| 7 | OLDENV="$(export -p)" |
| 8 | SELF="$(realpath "$0")" |
| 9 | NAME="${0##*/}" |
| 10 | PROGRAM_NAME="${SELF##*/}" |
| 11 | ROOT="${SELF%/*/*}" |
| 12 | |
| 13 | |
| 14 | # Set up the environment |
| 15 | LIB="$ROOT/lib/$NAME" |
| 16 | LIBEXEC="$ROOT/libexec" |
| 17 | SHARE="$ROOT/share/$NAME" |
| 18 | VENDOR="$SHARE/vendor" |
| 19 | |
| 20 | PATH="$LIB:$LIBEXEC:$VENDOR:$PATH" |
| 21 | export SELF |
| 22 | |
| 23 | USAGE="<COMMAND>" |
| 24 | unset VERSION |
| 25 | |
| 26 | |
| 27 | # --- |
| 28 | # Define shared helpers |
| 29 | |
| 30 | error() { |
| 31 | NAME="${1##*/}" |
| 32 | shift |
| 33 | |
| 34 | printf "%s: " "$(invocation)" |
| 35 | printf "%s\n" "$@" |
| 36 | return 1 |
| 37 | } >&2 |
| 38 | |
| 39 | usage() { |
| 40 | if [ $# -eq 0 ]; then |
| 41 | set -- "$NAME" "${USAGE:-}" "commands: $(flatten commands)" |
| 42 | fi |
| 43 | |
| 44 | NAME="${1##*/}" |
| 45 | shift |
| 46 | |
| 47 | printf "usage: %s " "$(invocation)" |
| 48 | printf "%s\n" "$@" |
| 49 | return 1 |
| 50 | } >&2 |
| 51 | |
| 52 | version() { |
| 53 | printf "%s %s\n" "$PROGRAM_NAME" "$VERSION" |
| 54 | } |
| 55 | |
| 56 | # --- |
| 57 | |
| 58 | commands() { |
| 59 | executables | sed -e "s/^.*\/$NAME-//" | sort |
| 60 | } |
| 61 | |
| 62 | executables() { |
| 63 | for ns in $(namespaces "$NAME-"); do |
| 64 | set -- "$@" \! -name "$ns-*" |
| 65 | done |
| 66 | |
| 67 | searchpath -name "$NAME-*" "$@" | awk -F / '!a[$NF]++' |
| 68 | } |
| 69 | |
| 70 | flatten() { |
| 71 | "$@" | { tr "\n" " "; printf "\n"; } |
| 72 | } |
| 73 | |
| 74 | invocation() { |
| 75 | { printf "%s\n" "$PROGRAM_NAME"; namespaces; } \ |
| 76 | | sort -r | awk -v name="$NAME" \ |
| 77 | 'index(name, $0 "-") == 1 { name = $0 " " substr(name, length($0) + 2) } |
| 78 | END { print name }' |
| 79 | } |
| 80 | |
| 81 | namespaces() { |
| 82 | searchpath -type l -name "${1:-$PROGRAM_NAME-}*" \ |
| 83 | -exec sh -c '[ "$0" = "$(realpath "$1")" ] 2>/dev/null' "$SELF" {} \; \ |
| 84 | | sed -e "s/^.*\///" | sort -u |
| 85 | } |
| 86 | |
| 87 | resolve() { |
| 88 | case "${1:-}" in |
| 89 | *-* ) executables -name "$NAME-$1" ;; |
| 90 | ?* ) searchpath -name "$NAME-$1" | head -n 1 |
| 91 | esac | grep . |
| 92 | } |
| 93 | |
| 94 | searchpath() { |
| 95 | { IFS=: |
| 96 | dirs= |
| 97 | set -- \) -o -prune "$@" |
| 98 | for dir in $PATH; do |
| 99 | [ -d "$dir" ] || continue |
| 100 | set -- -o -path "$dir" "$@" |
| 101 | dirs="$dirs:$dir" |
| 102 | done |
| 103 | set -- ${dirs#:} \( \! -name '*' "$@" |
| 104 | unset IFS dir dirs |
| 105 | } 2>/dev/null |
| 106 | |
| 107 | find -H "$@" 2>/dev/null \ |
| 108 | \( -perm -001 -o -perm -010 -o -perm -100 \) \ |
| 109 | \( -type f -o -type l \) \ |
| 110 | \! -name "?*--?*" \ |
| 111 | -print || true |
| 112 | } |
| 113 | |
| 114 | unenv() { |
| 115 | exec env -i sh -euc 'eval "$1"; shift; exec "$@"' "$0" "$OLDENV" "$@" |
| 116 | } |
| 117 | |
| 118 | # --- |
| 119 | |
| 120 | |
| 121 | # Load _init.sh, if it exists |
| 122 | if [ -r "$LIB"/_init.sh ]; then |
| 123 | . "$LIB"/_init.sh |
| 124 | fi |
| 125 | |
| 126 | |
| 127 | # Rewrite `--version` to `--- version` |
| 128 | if [ $# -eq 1 ] && [ -n "${VERSION:-}" ] && [ "$1" = "--version" ]; then |
| 129 | set -- --- version |
| 130 | fi |
| 131 | |
| 132 | |
| 133 | # When invoked as `--- <COMMAND> ...`, run that command and exit |
| 134 | if [ "${1:-}" = "---" ]; then |
| 135 | [ $# -gt 1 ] || usage |
| 136 | shift |
| 137 | "$@" |
| 138 | exit |
| 139 | fi |
| 140 | |
| 141 | |
| 142 | # Scan the argument list to find the command name |
| 143 | COMMAND= |
| 144 | args= |
| 145 | index=0 |
| 146 | |
| 147 | for arg; do |
| 148 | index=$((index+1)) |
| 149 | case "$arg" in |
| 150 | -* ) ;; |
| 151 | ?*--?* ) [ -n "$COMMAND" ] || usage ;; |
| 152 | * ) |
| 153 | if [ -z "$COMMAND" ]; then |
| 154 | COMMAND="$arg" |
| 155 | continue |
| 156 | fi |
| 157 | esac |
| 158 | args="$args"' "${'"$index"'}"' |
| 159 | done |
| 160 | |
| 161 | |
| 162 | # Match the command name to an executable in the current namespace; |
| 163 | # if one exists, exec it |
| 164 | if executable="$(resolve "$COMMAND")"; then |
| 165 | eval "set --$args" |
| 166 | exec "$executable" "$@" |
| 167 | fi |
| 168 | |
| 169 | |
| 170 | # No matching executable; load _unhandled.sh, if it exists |
| 171 | if [ -r "$LIB"/_unhandled.sh ]; then |
| 172 | . "$LIB"/_unhandled.sh |
| 173 | fi |
| 174 | |
| 175 | |
| 176 | # No matching command or no command specified |
| 177 | if [ -n "$COMMAND" ]; then |
| 178 | error "$0" "$COMMAND: command not found" || exit 127 |
| 179 | fi |
| 180 | |
| 181 | usage |
| 182 |