562 lines
14 KiB
Bash
Executable File
562 lines
14 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# composure - by erichs
|
|
# light-hearted functions for intuitive shell programming
|
|
|
|
# version: 1.3.1
|
|
# latest source available at http://git.io/composure
|
|
|
|
# install: source this script in your ~/.profile or ~/.${SHELL}rc script
|
|
# known to work on bash, zsh, and ksh93
|
|
|
|
# 'plumbing' functions
|
|
|
|
_bootstrap_composure() {
|
|
_generate_metadata_functions
|
|
_load_composed_functions
|
|
_determine_printf_cmd
|
|
}
|
|
|
|
_get_composure_dir ()
|
|
{
|
|
if [ -n "$XDG_DATA_HOME" ]; then
|
|
echo "$XDG_DATA_HOME/composure"
|
|
else
|
|
echo "$HOME/.local/composure"
|
|
fi
|
|
}
|
|
|
|
_get_author_name ()
|
|
{
|
|
typeset name localname
|
|
localname="$(git --git-dir "$(_get_composure_dir)/.git" config --get user.name)"
|
|
for name in "$GIT_AUTHOR_NAME" "$localname"; do
|
|
if [ -n "$name" ]; then
|
|
echo "$name"
|
|
break
|
|
fi
|
|
done
|
|
}
|
|
|
|
_composure_keywords ()
|
|
{
|
|
echo "about author example group param version"
|
|
}
|
|
|
|
_letterpress ()
|
|
{
|
|
typeset rightcol="$1" leftcol="${2:- }" leftwidth="${3:-20}"
|
|
|
|
if [ -z "$rightcol" ]; then
|
|
return
|
|
fi
|
|
|
|
$_printf_cmd "%-*s%s\n" "$leftwidth" "$leftcol" "$rightcol"
|
|
}
|
|
|
|
_determine_printf_cmd() {
|
|
if [ -z "$_printf_cmd" ]; then
|
|
_printf_cmd=printf
|
|
# prefer GNU gprintf if available
|
|
[ -x "$(which gprintf 2>/dev/null)" ] && _printf_cmd=gprintf
|
|
export _printf_cmd
|
|
fi
|
|
}
|
|
|
|
_longest_function_name_length ()
|
|
{
|
|
echo "$1" | awk 'BEGIN{ maxlength=0 }
|
|
{
|
|
for(i=1;i<=NF;i++)
|
|
if (length($i)>maxlength)
|
|
{
|
|
maxlength=length($i)
|
|
}
|
|
}
|
|
END{ print maxlength}'
|
|
}
|
|
|
|
_temp_filename_for ()
|
|
{
|
|
typeset file=$(mktemp "/tmp/$1.XXXX")
|
|
command rm "$file" 2>/dev/null # ensure file is unlinked prior to use
|
|
echo "$file"
|
|
}
|
|
|
|
_prompt ()
|
|
{
|
|
typeset prompt="$1"
|
|
typeset result
|
|
case "$(_shell)" in
|
|
bash)
|
|
read -r -e -p "$prompt" result;;
|
|
*)
|
|
echo -n "$prompt" >&2; read -r result;;
|
|
esac
|
|
echo "$result"
|
|
}
|
|
|
|
_add_composure_file ()
|
|
{
|
|
typeset func="$1"
|
|
typeset file="$2"
|
|
typeset operation="$3"
|
|
typeset comment="${4:-}"
|
|
typeset composure_dir=$(_get_composure_dir)
|
|
|
|
(
|
|
if ! cd "$composure_dir"; then
|
|
printf "%s\n" "Oops! Can't find $composure_dir!"
|
|
return
|
|
fi
|
|
if git rev-parse 2>/dev/null; then
|
|
if [ ! -f "$file" ]; then
|
|
printf "%s\n" "Oops! Couldn't find $file to version it for you..."
|
|
return
|
|
fi
|
|
cp "$file" "$composure_dir/$func.inc"
|
|
git add --all .
|
|
if [ -z "$comment" ]; then
|
|
comment="$(_prompt 'Git Comment: ')"
|
|
fi
|
|
git commit -m "$operation $func: $comment"
|
|
fi
|
|
)
|
|
}
|
|
|
|
_transcribe ()
|
|
{
|
|
typeset func="$1"
|
|
typeset file="$2"
|
|
typeset operation="$3"
|
|
typeset comment="${4:-}"
|
|
typeset composure_dir=$(_get_composure_dir)
|
|
|
|
if git --version >/dev/null 2>&1; then
|
|
if [ -d "$composure_dir" ]; then
|
|
_add_composure_file "$func" "$file" "$operation" "$comment"
|
|
else
|
|
if [ "$USE_COMPOSURE_REPO" = "0" ]; then
|
|
return # if you say so...
|
|
fi
|
|
printf "%s\n" "I see you don't have a $composure_dir repo..."
|
|
typeset input=''
|
|
typeset valid=0
|
|
while [ $valid != 1 ]; do
|
|
printf "\n%s" 'would you like to create one? y/n: '
|
|
read -r input
|
|
case $input in
|
|
y|yes|Y|Yes|YES)
|
|
(
|
|
echo 'creating git repository for your functions...'
|
|
mkdir -p "$composure_dir" || return 1
|
|
cd "$composure_dir" || return 1
|
|
git init
|
|
echo "composure stores your function definitions here" > README.txt
|
|
git add README.txt
|
|
git commit -m 'initial commit'
|
|
)
|
|
# if at first you don't succeed...
|
|
_transcribe "$func" "$file" "$operation" "$comment"
|
|
valid=1
|
|
;;
|
|
n|no|N|No|NO)
|
|
printf "%s\n" "ok. add 'export USE_COMPOSURE_REPO=0' to your startup script to disable this message."
|
|
valid=1
|
|
;;
|
|
*)
|
|
printf "%s\n" "sorry, didn't get that..."
|
|
;;
|
|
esac
|
|
done
|
|
fi
|
|
fi
|
|
}
|
|
|
|
_typeset_functions ()
|
|
{
|
|
# unfortunately, there does not seem to be a easy, portable way to list just the
|
|
# names of the defined shell functions...
|
|
|
|
case "$(_shell)" in
|
|
sh|bash)
|
|
typeset -F | awk '{print $3}'
|
|
;;
|
|
*)
|
|
# trim everything following '()' in ksh/zsh
|
|
typeset +f | sed 's/().*$//'
|
|
;;
|
|
esac
|
|
}
|
|
|
|
_typeset_functions_about ()
|
|
{
|
|
typeset f
|
|
for f in $(_typeset_functions); do
|
|
typeset -f -- "$f" | grep -qE "^about[[:space:]]|[[:space:]]about[[:space:]]" && echo -- "$f"
|
|
done
|
|
}
|
|
|
|
_shell () {
|
|
# here's a hack I modified from a StackOverflow post:
|
|
# get the ps listing for the current process ($$), and print the last column (CMD)
|
|
# stripping any leading hyphens shells sometimes throw in there
|
|
typeset this=$(ps -o comm -p $$ | tail -1 | awk '{print $NF}' | sed 's/^-*//')
|
|
echo "${this##*/}" # e.g. /bin/bash => bash
|
|
}
|
|
|
|
_generate_metadata_functions() {
|
|
typeset f
|
|
for f in $(_composure_keywords)
|
|
do
|
|
eval "$f() { :; }"
|
|
done
|
|
}
|
|
|
|
_list_composure_files () {
|
|
typeset composure_dir="$(_get_composure_dir)"
|
|
[ -d "$composure_dir" ] && find "$composure_dir" -maxdepth 1 -name '*.inc'
|
|
}
|
|
|
|
_load_composed_functions () {
|
|
# load previously composed functions into shell
|
|
# you may disable this by adding the following line to your shell startup:
|
|
# export LOAD_COMPOSED_FUNCTIONS=0
|
|
|
|
if [ "$LOAD_COMPOSED_FUNCTIONS" = "0" ]; then
|
|
return # if you say so...
|
|
fi
|
|
|
|
typeset inc
|
|
for inc in $(_list_composure_files); do
|
|
# shellcheck source=/dev/null
|
|
. "$inc"
|
|
done
|
|
}
|
|
|
|
_strip_trailing_whitespace () {
|
|
sed -e 's/ \+$//'
|
|
}
|
|
|
|
_strip_semicolons () {
|
|
sed -e 's/;$//'
|
|
}
|
|
|
|
|
|
# 'porcelain' functions
|
|
|
|
cite ()
|
|
{
|
|
about 'creates one or more meta keywords for use in your functions'
|
|
param 'one or more keywords'
|
|
example '$ cite url username'
|
|
example '$ url http://somewhere.com'
|
|
example '$ username alice'
|
|
group 'composure'
|
|
|
|
# this is the storage half of the 'metadata' system:
|
|
# we create dynamic metadata keywords with function wrappers around
|
|
# the NOP command, ':'
|
|
|
|
# anything following a keyword will get parsed as a positional
|
|
# parameter, but stay resident in the ENV. As opposed to shell
|
|
# comments, '#', which do not get parsed and are not available
|
|
# at runtime.
|
|
|
|
# a BIG caveat--your metadata must be roughly parsable: do not use
|
|
# contractions, and consider single or double quoting if it contains
|
|
# non-alphanumeric characters
|
|
|
|
if [ -z "$1" ]; then
|
|
printf '%s\n' 'missing parameter(s)'
|
|
reference cite
|
|
return
|
|
fi
|
|
|
|
typeset keyword
|
|
for keyword in "$@"; do
|
|
eval "$keyword() { :; }"
|
|
done
|
|
}
|
|
|
|
draft ()
|
|
{
|
|
about 'wraps command from history into a new function, default is last command'
|
|
param '1: name to give function'
|
|
param '2: optional history line number'
|
|
example '$ ls'
|
|
example '$ draft list'
|
|
example '$ draft newfunc 1120 # wraps command at history line 1120 in newfunc()'
|
|
group 'composure'
|
|
|
|
typeset func=$1
|
|
typeset num=$2
|
|
|
|
if [ -z "$func" ]; then
|
|
printf '%s\n' 'missing parameter(s)'
|
|
reference draft
|
|
return
|
|
fi
|
|
|
|
# aliases bind tighter than function names, disallow them
|
|
if type -a "$func" 2>/dev/null | grep -q 'is.*alias'; then
|
|
printf '%s\n' "sorry, $(type -a "$func"). please choose another name."
|
|
return
|
|
fi
|
|
|
|
typeset cmd
|
|
if [ -z "$num" ]; then
|
|
# some versions of 'fix command, fc' need corrective lenses...
|
|
typeset lines=$(fc -ln -1 | grep -q draft && echo 2 || echo 1)
|
|
# parse last command from fc output
|
|
# shellcheck disable=SC2086
|
|
cmd=$(fc -ln -$lines | head -1 | sed 's/^[[:blank:]]*//')
|
|
else
|
|
# parse command from history line number
|
|
cmd=$(eval "history | grep '^[[:blank:]]*$num' | head -1" | sed 's/^[[:blank:][:digit:]]*//')
|
|
fi
|
|
eval "function $func {
|
|
author '$(_get_author_name)'
|
|
about ''
|
|
param ''
|
|
example ''
|
|
group ''
|
|
|
|
$cmd;
|
|
}"
|
|
typeset file=$(_temp_filename_for draft)
|
|
typeset -f "$func" | _strip_trailing_whitespace | _strip_semicolons > "$file"
|
|
_transcribe "$func" "$file" Draft "Initial draft"
|
|
command rm "$file" 2>/dev/null
|
|
revise "$func"
|
|
}
|
|
|
|
glossary ()
|
|
{
|
|
about 'displays help summary for all functions, or summary for a group of functions'
|
|
param '1: optional, group name'
|
|
example '$ glossary'
|
|
example '$ glossary misc'
|
|
group 'composure'
|
|
|
|
typeset targetgroup=${1:-}
|
|
typeset functionlist="$(_typeset_functions_about)"
|
|
typeset maxwidth=$(_longest_function_name_length "$functionlist" | awk '{print $1 + 5}')
|
|
|
|
for func in $(echo $functionlist); do
|
|
|
|
if [ "X${targetgroup}X" != "XX" ]; then
|
|
typeset group="$(typeset -f -- $func | metafor group)"
|
|
if [ "$group" != "$targetgroup" ]; then
|
|
continue # skip non-matching groups, if specified
|
|
fi
|
|
fi
|
|
typeset about="$(typeset -f -- $func | metafor about)"
|
|
typeset aboutline=
|
|
echo "$about" | fmt | while read -r aboutline; do
|
|
_letterpress "$aboutline" "$func" "$maxwidth"
|
|
func=" " # only display function name once
|
|
done
|
|
done
|
|
}
|
|
|
|
metafor ()
|
|
{
|
|
about 'prints function metadata associated with keyword'
|
|
param '1: meta keyword'
|
|
example '$ typeset -f glossary | metafor example'
|
|
group 'composure'
|
|
|
|
typeset keyword=$1
|
|
|
|
if [ -z "$keyword" ]; then
|
|
printf '%s\n' 'missing parameter(s)'
|
|
reference metafor
|
|
return
|
|
fi
|
|
|
|
# this sed-fu is the retrieval half of the 'metadata' system:
|
|
# 'grep' for the metadata keyword, and then parse/filter the matching line
|
|
|
|
# grep keyword # strip trailing '|"|; # ignore thru keyword and leading '|"
|
|
sed -n "/$keyword / s/['\";]*\$//;s/^[ ]*$keyword ['\"]*\([^([].*\)*\$/\1/p"
|
|
}
|
|
|
|
reference ()
|
|
{
|
|
about 'displays apidoc help for a specific function'
|
|
param '1: function name'
|
|
example '$ reference revise'
|
|
group 'composure'
|
|
|
|
typeset func=$1
|
|
if [ -z "$func" ]; then
|
|
printf '%s\n' 'missing parameter(s)'
|
|
reference reference
|
|
return
|
|
fi
|
|
|
|
typeset line
|
|
|
|
typeset about="$(typeset -f "$func" | metafor about)"
|
|
_letterpress "$about" "$func"
|
|
|
|
typeset author="$(typeset -f $func | metafor author)"
|
|
if [ -n "$author" ]; then
|
|
_letterpress "$author" 'author:'
|
|
fi
|
|
|
|
typeset version="$(typeset -f $func | metafor version)"
|
|
if [ -n "$version" ]; then
|
|
_letterpress "$version" 'version:'
|
|
fi
|
|
|
|
if [ -n "$(typeset -f $func | metafor param)" ]; then
|
|
printf "parameters:\n"
|
|
typeset -f $func | metafor param | while read -r line
|
|
do
|
|
_letterpress "$line"
|
|
done
|
|
fi
|
|
|
|
if [ -n "$(typeset -f $func | metafor example)" ]; then
|
|
printf "examples:\n"
|
|
typeset -f $func | metafor example | while read -r line
|
|
do
|
|
_letterpress "$line"
|
|
done
|
|
fi
|
|
}
|
|
|
|
revise ()
|
|
{
|
|
about 'loads function into editor for revision'
|
|
param '<optional> -e: revise version stored in ENV'
|
|
param '1: name of function'
|
|
example '$ revise myfunction'
|
|
example '$ revise -e myfunction'
|
|
example 'save a zero-length file to abort revision'
|
|
group 'composure'
|
|
|
|
typeset source='git'
|
|
if [ "$1" = '-e' ]; then
|
|
source='env'
|
|
shift
|
|
fi
|
|
|
|
typeset func=$1
|
|
if [ -z "$func" ]; then
|
|
printf '%s\n' 'missing parameter(s)'
|
|
reference revise
|
|
return
|
|
fi
|
|
|
|
typeset composure_dir=$(_get_composure_dir)
|
|
typeset temp=$(_temp_filename_for revise)
|
|
# populate tempfile...
|
|
if [ "$source" = 'env' ] || [ ! -f "$composure_dir/$func.inc" ]; then
|
|
# ...with ENV if specified or not previously versioned
|
|
typeset -f $func > $temp
|
|
else
|
|
# ...or with contents of latest git revision
|
|
cat "$composure_dir/$func.inc" > "$temp"
|
|
fi
|
|
|
|
if [ -z "$EDITOR" ]
|
|
then
|
|
typeset EDITOR=vi
|
|
fi
|
|
|
|
$EDITOR "$temp"
|
|
if [ -s "$temp" ]; then
|
|
typeset edit='N'
|
|
|
|
# source edited file
|
|
# shellcheck source=/dev/null
|
|
. "$temp" || edit='Y'
|
|
|
|
while [ $edit = 'Y' ]; do
|
|
echo -n "Re-edit? Y/N: "
|
|
read -r edit
|
|
case $edit in
|
|
y|yes|Y|Yes|YES)
|
|
edit='Y'
|
|
$EDITOR "$temp"
|
|
# shellcheck source=/dev/null
|
|
. "$temp" && edit='N';;
|
|
*)
|
|
edit='N';;
|
|
esac
|
|
done
|
|
_transcribe "$func" "$temp" Revise
|
|
else
|
|
# zero-length files abort revision
|
|
printf '%s\n' 'zero-length file, revision aborted!'
|
|
fi
|
|
command rm "$temp"
|
|
}
|
|
|
|
write ()
|
|
{
|
|
about 'writes one or more composed function definitions to stdout'
|
|
param 'one or more function names'
|
|
example '$ write finddown foo'
|
|
example '$ write finddown'
|
|
group 'composure'
|
|
|
|
if [ -z "$1" ]; then
|
|
printf '%s\n' 'missing parameter(s)'
|
|
reference write
|
|
return
|
|
fi
|
|
|
|
echo "#!/usr/bin/env ${SHELL##*/}"
|
|
|
|
# bootstrap metadata
|
|
cat <<END
|
|
for f in $(_composure_keywords)
|
|
do
|
|
eval "\$f() { :; }"
|
|
done
|
|
unset f
|
|
END
|
|
|
|
# write out function definitons
|
|
# shellcheck disable=SC2034
|
|
typeset -f cite "$@"
|
|
|
|
cat <<END
|
|
main() {
|
|
echo "edit me to do something useful!"
|
|
exit 0
|
|
}
|
|
|
|
main \$*
|
|
END
|
|
}
|
|
|
|
_bootstrap_composure
|
|
|
|
: <<EOF
|
|
License: The MIT License
|
|
|
|
Copyright © 2012, 2016 Erich Smith
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy of this
|
|
software and associated documentation files (the "Software"), to deal in the Software
|
|
without restriction, including without limitation the rights to use, copy, modify,
|
|
merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
|
|
permit persons to whom the Software is furnished to do so, subject to the following
|
|
conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in all copies
|
|
or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
|
|
INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
|
|
PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
|
|
HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
|
|
CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE
|
|
OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
EOF
|