Add sd
parent
6aaee34153
commit
47c019dbde
GPG Key ID:
A343F43342EB6E2A
|
|
@ -0,0 +1,314 @@
|
||||||
|
#!/usr/bin/env bash
|
||||||
|
|
||||||
|
# https://github.com/ianthehenry/sd
|
||||||
|
# ecd1ab8d3fc3a829d8abfb8bf1e3722c9c99407b
|
||||||
|
|
||||||
|
__sd_join_path() {
|
||||||
|
echo "$@" | tr ' ' /
|
||||||
|
}
|
||||||
|
|
||||||
|
__sd_which() {
|
||||||
|
echo "$1"
|
||||||
|
}
|
||||||
|
|
||||||
|
__sd_cat() {
|
||||||
|
${SD_CAT:-cat} "$1"
|
||||||
|
}
|
||||||
|
|
||||||
|
__sd_edit() {
|
||||||
|
${SD_EDITOR:-${VISUAL:-${EDITOR:-vi}}} "$1"
|
||||||
|
}
|
||||||
|
|
||||||
|
__sd_directory_help() {
|
||||||
|
local target="$1"
|
||||||
|
local i file helpfile help command
|
||||||
|
local -a commands helps
|
||||||
|
if [[ -d "$target" ]]; then
|
||||||
|
helpfile="$target/help"
|
||||||
|
else
|
||||||
|
helpfile="$target.help"
|
||||||
|
fi
|
||||||
|
# in case you have `sd help` as a command alias, we don't
|
||||||
|
# want to print that when you just run "sd" by itself.
|
||||||
|
# Admittedly we don't check this in any other place, which
|
||||||
|
# could be considered a bug, but whatever it's a weird little
|
||||||
|
# hack and no one uses this.
|
||||||
|
if [[ -e "$helpfile" && ! -x "$helpfile" ]]; then
|
||||||
|
__sd_cat "$helpfile"
|
||||||
|
echo
|
||||||
|
else
|
||||||
|
command=$(basename "$target")
|
||||||
|
echo "$command commands"
|
||||||
|
echo
|
||||||
|
fi
|
||||||
|
commands=()
|
||||||
|
helps=()
|
||||||
|
for file in "$target"/*; do
|
||||||
|
if [[ ! -x "$file" ]]; then
|
||||||
|
continue
|
||||||
|
fi
|
||||||
|
command=$(basename "$file")
|
||||||
|
|
||||||
|
if [[ -d "$file" ]]; then
|
||||||
|
helpfile="$file/help"
|
||||||
|
if [[ -f "$helpfile" ]]; then
|
||||||
|
help=$(head -n1 "$helpfile")
|
||||||
|
else
|
||||||
|
help="$command commands"
|
||||||
|
fi
|
||||||
|
command="$command ..."
|
||||||
|
else
|
||||||
|
helpfile="$file.help"
|
||||||
|
if [[ -f "$helpfile" ]]; then
|
||||||
|
help=$(head -n1 "$helpfile")
|
||||||
|
else
|
||||||
|
help=$(sed -nE -e '/^#!/d' -e '/^#/{s/^# *//; p; q;}' "$file")
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
commands+=("$command")
|
||||||
|
helps+=("$help")
|
||||||
|
done
|
||||||
|
|
||||||
|
if [[ ${#commands[@]} -eq 0 ]]; then
|
||||||
|
echo "(no subcommands found)"
|
||||||
|
else
|
||||||
|
local max_length=0
|
||||||
|
local length
|
||||||
|
for command in "${commands[@]}"; do
|
||||||
|
length=${#command}
|
||||||
|
max_length=$((length > max_length ? length : max_length))
|
||||||
|
done
|
||||||
|
|
||||||
|
if [[ -n ${ZSH_EVAL_CONTEXT+x} ]]; then
|
||||||
|
set -o LOCAL_OPTIONS
|
||||||
|
# we need to ensure we have 0-indexed arrays
|
||||||
|
# in order for this loop to work properly.
|
||||||
|
# zsh doesn't support ${!commands[@]} expansion
|
||||||
|
set -o KSH_ARRAYS
|
||||||
|
fi
|
||||||
|
|
||||||
|
for ((i = 0; i < ${#commands[@]}; i++)); do
|
||||||
|
printf "%-${max_length}s -- %s\n" "${commands[i]}" "${helps[i]}"
|
||||||
|
done
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
__sd_help() {
|
||||||
|
local target=$1
|
||||||
|
|
||||||
|
if [[ -d "$target" ]]; then
|
||||||
|
__sd_directory_help "$target"
|
||||||
|
elif [[ -f "$target.help" ]]; then
|
||||||
|
__sd_cat "$target.help"
|
||||||
|
else
|
||||||
|
help=$(sed -nE -e '/^#!/d' -e $':start\n /^#/{ s/^# ?//; p; \nb start\n }' "$target")
|
||||||
|
if [[ -z "$help" ]]; then
|
||||||
|
echo "there is no help for you here" >&2
|
||||||
|
exit 1
|
||||||
|
else
|
||||||
|
echo "$help"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
__sd_print_template() {
|
||||||
|
if [[ -f "$1/template" ]]; then
|
||||||
|
cat "$1/template"
|
||||||
|
else
|
||||||
|
if [[ "$1" = "${SD_ROOT:-$HOME/sd}" ]]; then
|
||||||
|
cat <<EOF
|
||||||
|
#!/usr/bin/env bash
|
||||||
|
|
||||||
|
set -euo pipefail
|
||||||
|
EOF
|
||||||
|
else
|
||||||
|
__sd_print_template "$(dirname "$1")"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
__sd_new() {
|
||||||
|
local script dir target body
|
||||||
|
target="$1"
|
||||||
|
shift
|
||||||
|
|
||||||
|
local -a command_path=()
|
||||||
|
|
||||||
|
for arg in "$@"; do
|
||||||
|
case "$arg" in
|
||||||
|
--new) shift; break ;;
|
||||||
|
*) command_path+=("$arg"); shift ;;
|
||||||
|
esac
|
||||||
|
done
|
||||||
|
|
||||||
|
if [[ ${#command_path[@]} -eq 0 ]]; then
|
||||||
|
echo "error: $target already exists" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ -f "$target" ]]; then
|
||||||
|
echo "error: command prefix $target is a regular file" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
body="$*"
|
||||||
|
|
||||||
|
script="$target"/"$(__sd_join_path "${command_path[@]}")"
|
||||||
|
|
||||||
|
if [[ -e "$script" ]]; then
|
||||||
|
echo "$script already exists!" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
dir="$(dirname "$script")"
|
||||||
|
mkdir -p "$dir"
|
||||||
|
(
|
||||||
|
__sd_print_template "$dir"
|
||||||
|
if [[ -n "$body" ]]; then
|
||||||
|
printf "\n%s\n" "$body"
|
||||||
|
fi
|
||||||
|
) >"$script"
|
||||||
|
|
||||||
|
chmod +x "$script"
|
||||||
|
|
||||||
|
if [[ -z "$body" ]]; then
|
||||||
|
__sd_edit "$script"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
__sd_new_user_help() {
|
||||||
|
root=$1
|
||||||
|
echo >&2 "error: $root not found"
|
||||||
|
echo >&2
|
||||||
|
echo >&2 "It looks like you don't have a script directory yet!"
|
||||||
|
echo >&2
|
||||||
|
echo >&2 "Get started by creating your first script:"
|
||||||
|
echo >&2
|
||||||
|
echo >&2 " sd hello --new 'echo \"Hello, sd!\"'"
|
||||||
|
echo >&2
|
||||||
|
echo >&2 "And then run it like this:"
|
||||||
|
echo >&2
|
||||||
|
echo >&2 " sd hello"
|
||||||
|
}
|
||||||
|
|
||||||
|
__sd() {
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
local root=${SD_ROOT:-$HOME/sd}
|
||||||
|
|
||||||
|
if [[ -e "$root" && ! -d "$root" ]]; then
|
||||||
|
echo "error: $root is not a directory" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
local target=$root
|
||||||
|
local arg
|
||||||
|
|
||||||
|
while [[ $# -gt 0 ]]; do
|
||||||
|
arg="$1"
|
||||||
|
if [[ -d "$target/$arg" ]]; then
|
||||||
|
target="$target/$arg"
|
||||||
|
shift
|
||||||
|
elif [[ -f "$target/$arg" ]]; then
|
||||||
|
target="$target/$arg"
|
||||||
|
shift
|
||||||
|
break
|
||||||
|
else
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
local found_help="false"
|
||||||
|
local found_new="false"
|
||||||
|
local found_edit="false"
|
||||||
|
local found_cat="false"
|
||||||
|
local found_which="false"
|
||||||
|
local found_really="false"
|
||||||
|
|
||||||
|
for arg in "$@"; do
|
||||||
|
case "$arg" in
|
||||||
|
--help) found_help=true ;;
|
||||||
|
--new) found_new=true ;;
|
||||||
|
--edit) found_edit=true ;;
|
||||||
|
--cat) found_cat=true ;;
|
||||||
|
--which) found_which=true ;;
|
||||||
|
--really) found_really=true ;;
|
||||||
|
esac
|
||||||
|
done
|
||||||
|
|
||||||
|
# you're allowed to run --new even if there is no
|
||||||
|
# script directory root, in order to bootstrap it
|
||||||
|
if [[ "$found_really" = "true" || "$found_new" = "false" ]]; then
|
||||||
|
if [[ ! -d "$root" ]]; then
|
||||||
|
__sd_new_user_help "$root"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ "$found_really" = "true" ]]; then
|
||||||
|
local -a preserved=()
|
||||||
|
for arg in "$@"; do
|
||||||
|
case "$arg" in
|
||||||
|
'--really') shift; break ;;
|
||||||
|
*) preserved+=("$arg"); shift ;;
|
||||||
|
esac
|
||||||
|
done
|
||||||
|
if [[ ${#preserved[@]} -gt 0 ]]; then
|
||||||
|
set -- "${preserved[@]}" "$@"
|
||||||
|
fi
|
||||||
|
elif [[ "$found_new" = "true" ]]; then
|
||||||
|
__sd_new "$target" "$@"
|
||||||
|
exit 0
|
||||||
|
elif [[ "$found_help" = "true" ]]; then
|
||||||
|
__sd_help "$target"
|
||||||
|
exit 0
|
||||||
|
elif [[ "$found_edit" = "true" ]]; then
|
||||||
|
__sd_edit "$target"
|
||||||
|
exit 0
|
||||||
|
elif [[ "$found_cat" = "true" ]]; then
|
||||||
|
__sd_cat "$target"
|
||||||
|
exit 0
|
||||||
|
elif [[ "$found_which" = "true" ]]; then
|
||||||
|
__sd_which "$target"
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ -d "$target" ]]; then
|
||||||
|
__sd_directory_help "$target"
|
||||||
|
if [[ $# -gt 0 ]]; then
|
||||||
|
echo >&2
|
||||||
|
echo "$target/$(__sd_join_path "$@") not found" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
elif [[ -x "$target" ]]; then
|
||||||
|
SD="$(dirname "$target")" exec "$target" "$@"
|
||||||
|
else
|
||||||
|
__sd_cat "$@"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# If you source this file and use function-mode sd, we want to
|
||||||
|
# wrap the execution in a subshell so that set -e doesn't kill
|
||||||
|
# the interactive shell, and so that we can exec the subshell
|
||||||
|
# without destroying the interactive shell.
|
||||||
|
sd() (
|
||||||
|
__sd "$@"
|
||||||
|
)
|
||||||
|
|
||||||
|
# If you source this file, it will define the function sd,
|
||||||
|
# which you can use from your shell. If you *run* this file
|
||||||
|
# as an executable, it will just invoke that function. We
|
||||||
|
# have to do some tricks to detect that reliably on bash/zsh.
|
||||||
|
|
||||||
|
if [[ -n ${ZSH_EVAL_CONTEXT+x} ]]; then
|
||||||
|
# we are in zsh
|
||||||
|
if [[ "$ZSH_EVAL_CONTEXT" = toplevel ]]; then
|
||||||
|
__sd "$@"
|
||||||
|
fi
|
||||||
|
elif [[ -n ${BASH_SOURCE[0]+x} ]]; then
|
||||||
|
# we are in bash
|
||||||
|
if [[ "${BASH_SOURCE[0]}" = "$0" ]]; then
|
||||||
|
__sd "$@"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
Loading…
Reference in New Issue