#!/bin/bash -e

set -o errexit # in case caller did not have -e in #! line

function extern () { (exec "$@") || return; }

isset () { test -n "${!1+isset}"; }

realpath () { readlink -f -- "$1"; }

include_guard_var () { realpath "$1"|"extern" sed 's/[^a-zA-Z0-9_]/_/g'; }

self_not_included () # usage: if self_not_included; then ... fi

{

	test $# -ge 1 || set -- "${BASH_SOURCE[1]}"

	local include_guard="$(include_guard_var "$1")"

	alias self_being_included=false

	! isset "$include_guard" && \

	{ eval "$include_guard"=1; alias self_being_included=true; }

}

# to load newly-defined aliases for use in functions in the same file:

## fi # load new aliases

## if self_being_included; then

# this is needed because aliases defined inside an if statement are not

# available inside that if statement

if self_not_included "${BASH_SOURCE[0]}"; then

shopt -s expand_aliases

unalias () { builtin unalias "$@" 2>&- || true; } # no error if undefined

#### exceptions

alias rethrow='return "$e"'

alias rethrow_subshell='exit "$e"'

fi # load new aliases

if self_being_included; then

# usage: save_e cmd... || use $e (or $?)

function save_e () { "$@"; e=$?; rethrow; } # rethrow because e= clears $?

alias save_e='declare e; save_e ' # last space alias-expands next word

function log_e () # usage: log_e cmd... [ || use $e (or $?) ]

{

	if ! "save_e" "$@"; then

		echo "! $*" >&2

		echo "! command exited with error $e" >&2

		rethrow

	fi

}

alias log_e='declare e; log_e ' # last space alias-expands next word

# usage: try cmd...; ignore status; if catch status; then ...; fi; end_try

function try () { "save_e" "$@" || true; }

alias try='declare e; try ' # last space alias-expands next word

catch () { test "$e" -eq "$1"; e=0; }

ignore () { catch "$@" || true; }

alias end_try='rethrow'

alias end_try_subshell='rethrow_subshell'

#### integers

let () { builtin let "$@" || true; }

	# "If the last ARG evaluates to 0, let returns 1" (`help let`)

bool2int () { save_e test -z "$1"; echo "$e"; } # empty->0; non-empty->1

#### arrays

join () { local IFS="$delim"; echo "$*"; } # usage: delim=... join elems...

reverse () # usage: array=($(reverse args...))

{

	local i

	for (( i=$#; i >= 1; i-- )); do printf '%q ' "${!i}"; done

}

#### verbose output

# usage: (stdout2stderr; cmd...) || return

# `|| return` needed on Mac because of bug where -e doesn't apply to ()

stdout2stderr () { exec >&2; }

# set verbosity

if isset verbose; then : "${verbosity:=$(bool2int "$verbose")}"; fi

: "${verbosity=3}" # default

: "${verbosity:=0}" # ensure non-empty

declare -i verbosity # ensure integer

export verbosity # propagate the verbosity to invoked commands

can_log () { test "$verbosity" -gt 0; } # verbosity=0 turns off all logging

: "${log_indent=  }"

# usage: in func:      inc_log_level; ...

#        outside func: inc_log_level; ...; dec_log_level

alias inc_log_level='declare verbosity="$verbosity" PS4="$log_indent$PS4"

let verbosity--'

alias dec_log_level='declare verbosity="$verbosity" PS4="${PS4#$log_indent}"

let verbosity++'

fi # load new aliases

if self_being_included; then

# usage: (limit_stderr; cmd...) || return

# `|| return` needed on Mac because of bug where -e doesn't apply to ()

limit_stderr () { inc_log_level; if ! can_log; then exec 2>/dev/null; fi; }

limit_stderr_cmd () # usage: [stdout2stderr=1] limit_stderr_cmd cmd...

{

	case "$1" in echo_run) shift; echo_cmd "$@";; esac

	(limit_stderr

		if test -n "$stdout2stderr"; then stdout2stderr; fi

		"$@"

	) || return

}

alias limit_stderr_cmd='limit_stderr_cmd ' # last space alias-expands next word

echo_cmd () { if can_log; then echo "$PS4$*" >&2; fi; }

echo_run () { echo_cmd "$@"; "$@"; }

# auto-echo common external commands

for cmd in rm; do alias "$cmd=echo_run $cmd"; done; unset cmd

# echo all external commands

alias extern="echo_run extern " # last space alias-expands next word

alias self='extern "$FUNCNAME"' # usage: wrapper () { self ...; }

# commands that are always external

for cmd in env; do alias "$cmd=extern $cmd"; done; unset cmd

canon_rel_path ()

{

	local path="$1"

	path="$(realpath "$path")" # canonicalize

	path="${path#$(pwd -P)/}" # remove any shared prefix with the current dir

	echo "$path"

}

fi # load new aliases

if self_being_included; then

function echo_func ()

{

	inc_log_level

	local script="$(canon_rel_path "${BASH_SOURCE[1]}")"

	echo_cmd "$script:${BASH_LINENO[0]}" "${FUNCNAME[1]}" "$@"

}

# usage: cmd1 | { pipe_delay; cmd2; }

alias pipe_delay='sleep 0.1' # s; display after leading output of cmd1

fi # load new aliases

if self_being_included; then

echo_stdin () # usage: input|echo_stdin|cmd

{

	inc_log_level

	if can_log; then

		pipe_delay

		echo ----- >&2

		tee -a /dev/stderr;

		echo ----- >&2

	else cat

	fi

}

echo_vars () # usage: echo_vars var...

{

	inc_log_level; inc_log_level

	if can_log; then

		local var

		for var in "${@%%=*}"; do { echo -n "$PS4"; declare -p "$var";} >&2;done

	fi

}

echo_export ()

{

	builtin export "$@"

	echo_vars "$@"

}

if test "$verbosity" -ge 2; then

	alias export="echo_export" # automatically echo env vars when they are set

fi

usage () { echo "Usage: $1" >&2; (exit 2); }

fi # load new aliases

if self_being_included; then

#### strings

sed_ere_flag="$(test "$(uname)" = Darwin && echo E || echo r)"

sed () { self -"$sed_ere_flag" "$@";}

#### vars

set_var () { eval "$1"'="$2"'; }

set_inv () { set_var no_"$1" "$(test -n "${!1}" || echo 1)"; }

# usage: local var=...; local_inv

alias local_inv='declare "no_$var=$(test -n "${!var}" || echo 1)"'

# usage: local prefix=..._; import_vars

alias import_vars="$(cat <<'EOF'

: "${prefix:?}"

local src_var dest_var

for src_var in $(eval echo '${!'$prefix'*}'); do

	dest_var="${src_var#$prefix}"

	local "$dest_var=${!src_var}"; echo_vars "$dest_var"

done

EOF

)"

#### commands

top_script="$0" # outermost script

top_dir="$(dirname "$top_script")"

# auto-removes a command's output file on error (like make's .DELETE_ON_ERROR)

function to_file () # usage: stdout=... to_file cmd...

{ : "${stdout?}"; log_e "$@" >"$stdout" || { rm "$stdout"; rethrow; }; }

alias to_file='to_file ' # last space alias-expands next word

run_args_cmd () # runs the command line args command

{

	test $? -eq 0 || return

	eval set -- "$(reverse "${BASH_ARGV[@]}")"

	test $# -ge 1 || set -- all

	echo_cmd "$(canon_rel_path "$0")" "$@"; "$@"

}

fwd () # usage: subdirs=(...); fwd "$FUNCNAME" "$@"

{

	echo_func

	: "${subdirs?}"

	for subdir in "${subdirs[@]}"; do

		"$(dirname "${BASH_SOURCE[1]}")"/"$subdir"/run "$@"

	done

}

#### make

# usage: target_filename/command () { echo_func; set_make_vars; ...; }

alias set_make_vars="$(cat <<'EOF'

local command="${FUNCNAME##*/}"; echo_vars command

local target_filename="${FUNCNAME%/*}"; echo_vars target_filename

local target="$top_dir/$target_filename"; echo_vars target

EOF

)"

# usage: set_make_vars; check_target_exists

alias check_target_exists='! test -e "$target" || return 0'

# usage: set_make_vars; to_target cmd...

alias to_target='stdout="$target" to_file ' # last space alias-expands next word

make () { echo_func; stdout2stderr=1 limit_stderr_cmd self "$@"; }

if false; then ## usage:

inline_make 3<<'EOF'

target:

	$(self_dir)/cmd >$@

EOF

# target will be run automatically because it's first in the makefile

fi ##

inline_make ()

{

	echo_func

	local self="$(readlink -f "${BASH_SOURCE[1]}")"

	local self_dir="$(dirname "$self")"

	export self self_dir

	make --makefile=<((

		cat /dev/fd/3

		echo -n "

.SUFFIXES: # turn off built-in suffix rules

.SECONDARY: # don't automatically delete intermediate files

.DELETE_ON_ERROR: # delete target if recipe fails

"

	)|echo_stdin) "$@"

}

#### compression

### zip

zip ()

{

	stdout2stderr=1 try limit_stderr_cmd self "$@"

	ignore 12 # "zip has nothing to do" (`man zip`)

	end_try

}

unzip () { stdout2stderr=1 limit_stderr_cmd self "$@"; }

set_inv force

alias zip_newer='zip ${no_force:+-u }'

alias unzip_newer='unzip ${no_force:+-u }-o'

	# -o is safe because -u only extracts newer files

#### URLs

localize_url () { test _"$1" = _"$(hostname -f)" || echo "$1"; }

fi