blob: 8dfc5f29de19be17c81a513ada0bdd8e18a8dd89 (
plain) (
tree)
|
|
#
# keep -- Main function for zshkeep; provided with a list of NAMEs, whether
# shell functions or variables, writes the current definition of each NAME to a
# directory $ZSHKEEP (defaults to ~/.zshkeep.d) with a .zsh suffix, each of
# which is reloaded each time this file is called. This allows you to quickly
# arrange to keep that useful shell function or variable you made inline on
# subsequent logins.
#
# Consider a shell function declared inline with the NAME 'ayy':
#
# $ ayy() { printf '%s\n' lmao ; }
# $ ayy
# lmao
# $ keep ayy
# $ keep
# ayy
# $ exit
#
# Then, on next login, the function is redefined for you:
#
# $ ayy
# lmao
#
# To get rid of it:
#
# $ keep -d ayy
#
keep() {
# Figure out the directory to which we're reading and writing these scripts
local zshkeep
zshkeep=${ZSHKEEP:-"$HOME"/.zshkeep.d}
mkdir -p -- "$zshkeep" || return
# Parse options
local opt delete
local OPTERR OPTIND OPTARG
while getopts 'dh' opt ; do
case $opt in
# -d given; means delete the keepfiles for the given names
d)
delete=1
;;
# -h given; means show help
h)
cat <<EOF
${FUNCNAME[0]}: Keep variables and functions in shell permanently by writing them to
named scripts iterated on shell start, in \$ZSHKEEP (defaults to
~/.zshkeep.d).
USAGE:
${FUNCNAME[0]}
List all the current kept variables and functions
${FUNCNAME[0]} NAME1 [NAME2 ...]
Write the current definition for the given NAMEs to keep files
${FUNCNAME[0]} -d NAME1 [NAME2 ...]
Delete the keep files for the given NAMEs
${FUNCNAME[0]} -h
Show this help
EOF
return
;;
# Unknown other option
\?)
printf 'zsh: %s -%s: invalid option\n' \
"${FUNCNAME[0]}" "$opt" >&2
return 2
;;
esac
done
shift "$((OPTIND-1))"
# If any arguments left, we must be either keeping or deleting
if (($#)) ; then
# Start keeping count of any errors
local -i errors
errors=0
# Iterate through the NAMEs given
local name
for name ; do
# Check NAMEs for validity
case $name in
# NAME must start with letters or an underscore, and contain no
# characters besides letters, numbers, or underscores
*[!a-zA-Z0-9_]*|[!a-zA-Z_]*)
printf 'zsh: %s: %s not a valid NAME\n' \
"${FUNCNAME[0]}" "$name" >&2
((errors++))
;;
# NAME is valid, proceed
*)
# If -d was given, delete the keep files for the NAME
if ((delete)) ; then
rm -- "$zshkeep"/"$name".zsh ||
((errors++))
# Save a function
elif [[ $(whence -w "$name") = *': function' ]] ; then
declare -f -- "$name" >"$zshkeep"/"$name".zsh ||
((errors++))
# Save a variable
elif declare -p -- "$name" >/dev/null ; then
declare -p -- "$name" >"$zshkeep"/"$name".zsh ||
((errors++))
fi
;;
esac
done
# Return 1 if we accrued any errors, 0 otherwise
return "$((errors > 0))"
fi
# Deleting is an error, since we need at least one argument
if ((delete)) ; then
printf 'zsh: %s: must specify at least one NAME to delete\n' \
"${FUNCNAME[0]}" >&2
return 2
fi
# Otherwise the user must want us to print all the NAMEs kept
(
declare -a keeps
keeps=("$zshkeep"/*.zsh(N))
keeps=("${keeps[@]##*/}")
keeps=("${keeps[@]%.zsh}")
((${#keeps[@]})) || exit 0
printf '%s\n' "${keeps[@]}"
)
}
# Load any existing scripts in zshkeep
for zshkeep in "${ZSHKEEP:-"$HOME"/.zshkeep.d}"/*.zsh(N) ; do
[[ -e $zshkeep ]] && source "$zshkeep"
done
unset -v zshkeep
|