1 |
8291
|
aaronmk
|
#!/bin/bash -e
|
2 |
9328
|
aaronmk
|
set -o errexit -o pipefail # errexit in case caller's #! line missing -e
|
3 |
8952
|
aaronmk
|
|
4 |
9018
|
aaronmk
|
if test ! "$_util_sh_include_guard_utils"; then
|
5 |
|
|
_util_sh_include_guard_utils=1
|
6 |
|
|
|
7 |
9074
|
aaronmk
|
isset() { test "${!1+isset}"; }
|
8 |
9002
|
aaronmk
|
|
9 |
9074
|
aaronmk
|
realpath() { readlink -f -- "$1"; }
|
10 |
8703
|
aaronmk
|
|
11 |
9314
|
aaronmk
|
str2varname() { echo "${1//[^a-zA-Z0-9_]/_}"; }
|
12 |
8716
|
aaronmk
|
|
13 |
9312
|
aaronmk
|
include_guard_var() { str2varname "$(realpath "$1")"; }
|
14 |
|
|
|
15 |
9074
|
aaronmk
|
self_not_included() # usage: if self_not_included; then ... fi
|
16 |
8703
|
aaronmk
|
{
|
17 |
8971
|
aaronmk
|
test $# -ge 1 || set -- "${BASH_SOURCE[1]}"
|
18 |
8703
|
aaronmk
|
local include_guard="$(include_guard_var "$1")"
|
19 |
8793
|
aaronmk
|
alias self_being_included=false
|
20 |
9003
|
aaronmk
|
! isset "$include_guard" && \
|
21 |
8793
|
aaronmk
|
{ eval "$include_guard"=1; alias self_being_included=true; }
|
22 |
8703
|
aaronmk
|
}
|
23 |
|
|
|
24 |
8793
|
aaronmk
|
# to load newly-defined aliases for use in functions in the same file:
|
25 |
|
|
## fi # load new aliases
|
26 |
|
|
## if self_being_included; then
|
27 |
|
|
# this is needed because aliases defined inside an if statement are not
|
28 |
|
|
# available inside that if statement
|
29 |
|
|
|
30 |
9018
|
aaronmk
|
fi
|
31 |
9017
|
aaronmk
|
|
32 |
9018
|
aaronmk
|
|
33 |
8704
|
aaronmk
|
if self_not_included "${BASH_SOURCE[0]}"; then
|
34 |
|
|
|
35 |
9017
|
aaronmk
|
|
36 |
|
|
#### options
|
37 |
|
|
|
38 |
8709
|
aaronmk
|
shopt -s expand_aliases
|
39 |
|
|
|
40 |
9017
|
aaronmk
|
|
41 |
9226
|
aaronmk
|
#### vars
|
42 |
|
|
|
43 |
|
|
set_var() { eval "$1"'="$2"'; }
|
44 |
|
|
|
45 |
9420
|
aaronmk
|
set_default() { if ! isset "$1"; then set_var "$@"; fi; }
|
46 |
|
|
|
47 |
9226
|
aaronmk
|
set_inv() { set_var no_"$1" "$(test "${!1}" || echo 1)"; }
|
48 |
|
|
|
49 |
|
|
# usage: local var=...; local_inv
|
50 |
|
|
alias local_inv='declare "no_$var=$(test "${!var}" || echo 1)"'
|
51 |
|
|
|
52 |
9227
|
aaronmk
|
unexport() { export -n "$@"; }
|
53 |
|
|
# `declare +x` won't work because it defines the var if it isn't set
|
54 |
|
|
|
55 |
9236
|
aaronmk
|
alias local_export='declare -x' # combines effects of local and export
|
56 |
|
|
|
57 |
9226
|
aaronmk
|
get_prefix_vars() { : "${prefix:?}"; eval echo '${!'$prefix'*}'; }
|
58 |
|
|
|
59 |
|
|
# usage: local prefix=..._; import_vars
|
60 |
|
|
# when used inside another alias 2+ levels deep, *must* be run inside a function
|
61 |
|
|
alias import_vars="$(cat <<'EOF'
|
62 |
|
|
: "${prefix:?}"
|
63 |
|
|
declare src_var dest_var
|
64 |
|
|
for src_var in $(get_prefix_vars); do
|
65 |
|
|
dest_var="${src_var#$prefix}"
|
66 |
|
|
declare "$dest_var=${!src_var}"; echo_vars "$dest_var"
|
67 |
|
|
done
|
68 |
|
|
EOF
|
69 |
|
|
)"
|
70 |
|
|
|
71 |
|
|
|
72 |
9311
|
aaronmk
|
#### caching
|
73 |
|
|
|
74 |
|
|
## shell-variable-based caching
|
75 |
|
|
|
76 |
9330
|
aaronmk
|
# usage: local cache_key=...; load_cache; \
|
77 |
9331
|
aaronmk
|
# if ! cached; then save_cache value || return; fi; echo_cached_value
|
78 |
9561
|
aaronmk
|
# cache_key for function inputs: "$(declare -p kw_param...) $*"
|
79 |
9311
|
aaronmk
|
alias load_cache='declare cache_var="$(str2varname "${FUNCNAME}___$cache_key")"'
|
80 |
|
|
alias cached='isset "$cache_var"'
|
81 |
|
|
alias save_cache='set_var "$cache_var"'
|
82 |
|
|
alias echo_cached_value='echo "${!cache_var}"'
|
83 |
|
|
|
84 |
|
|
clear_cache() # usage: func=... clear_cache
|
85 |
|
|
{ : "${func:?}"; unset $(prefix="${func}___" get_prefix_vars); }
|
86 |
|
|
|
87 |
|
|
fi # load new aliases
|
88 |
|
|
if self_being_included; then
|
89 |
|
|
|
90 |
|
|
|
91 |
9017
|
aaronmk
|
#### aliases
|
92 |
|
|
|
93 |
9074
|
aaronmk
|
unalias() { builtin unalias "$@" 2>&- || true; } # no error if undefined
|
94 |
8889
|
aaronmk
|
|
95 |
9191
|
aaronmk
|
# usage: alias alias_='var=value run_cmd '
|
96 |
|
|
function run_cmd() { "$@"; }
|
97 |
|
|
alias run_cmd='"run_cmd" ' # last space alias-expands next word
|
98 |
9017
|
aaronmk
|
|
99 |
9191
|
aaronmk
|
|
100 |
9082
|
aaronmk
|
#### functions
|
101 |
|
|
|
102 |
9235
|
aaronmk
|
kw_params() # usage: func() { kw_params param_var...; }; ...; param_var=... cmd
|
103 |
9229
|
aaronmk
|
# removes keyword-param-only vars from the environment
|
104 |
|
|
{ unexport "$@"; }
|
105 |
9228
|
aaronmk
|
|
106 |
9082
|
aaronmk
|
alias self='command "$FUNCNAME"' # usage: wrapper() { self ...; }
|
107 |
|
|
|
108 |
9673
|
aaronmk
|
pf() { declare -f "$@"; } # usage: pf function (prints func decl for debugging)
|
109 |
9082
|
aaronmk
|
|
110 |
9673
|
aaronmk
|
|
111 |
8886
|
aaronmk
|
#### exceptions
|
112 |
|
|
|
113 |
9032
|
aaronmk
|
# usage: cmd || { save_e; ...; rethrow; }
|
114 |
9031
|
aaronmk
|
alias export_e='e=$?'
|
115 |
9032
|
aaronmk
|
alias save_e='declare e=$?'
|
116 |
8976
|
aaronmk
|
alias rethrow='return "$e"'
|
117 |
|
|
alias rethrow_subshell='exit "$e"'
|
118 |
|
|
|
119 |
8978
|
aaronmk
|
fi # load new aliases
|
120 |
|
|
if self_being_included; then
|
121 |
|
|
|
122 |
9535
|
aaronmk
|
# usage: try cmd...; ignore_e status; if catch status; then ...; fi; end_try
|
123 |
8886
|
aaronmk
|
|
124 |
9448
|
aaronmk
|
function try() { e=0; benign_error=1 "$@" || { export_e; true; }; }
|
125 |
9177
|
aaronmk
|
alias try='declare e; "try" ' # last space alias-expands next word
|
126 |
8886
|
aaronmk
|
|
127 |
9534
|
aaronmk
|
catch() { test "$e" -eq "$1" && e=0; }
|
128 |
8886
|
aaronmk
|
|
129 |
9535
|
aaronmk
|
ignore_e() { catch "$@" || true; }
|
130 |
8886
|
aaronmk
|
|
131 |
8977
|
aaronmk
|
alias end_try='rethrow'
|
132 |
|
|
alias end_try_subshell='rethrow_subshell'
|
133 |
8886
|
aaronmk
|
|
134 |
9536
|
aaronmk
|
ignore() { save_e; ignore_e "$@"; rethrow; } # usage: try cmd || ignore status
|
135 |
|
|
|
136 |
9551
|
aaronmk
|
### signals
|
137 |
|
|
|
138 |
|
|
sig_e() { echo $(( 128+$(kill -l "$1") )); } # usage: sig_e SIGINT, etc.
|
139 |
|
|
|
140 |
|
|
ignore_sig() { ignore "$(sig_e "$1")"; }
|
141 |
|
|
|
142 |
|
|
# usage: piped_cmd cmd1...|cmd2... (which doesn't read all its input)
|
143 |
|
|
function piped_cmd() { "$@" || ignore_sig SIGPIPE; }
|
144 |
|
|
alias piped_cmd='"piped_cmd" ' # last space alias-expands next word
|
145 |
|
|
|
146 |
9020
|
aaronmk
|
fi # load new aliases
|
147 |
|
|
if self_being_included; then
|
148 |
9017
|
aaronmk
|
|
149 |
9020
|
aaronmk
|
|
150 |
8899
|
aaronmk
|
#### integers
|
151 |
|
|
|
152 |
9074
|
aaronmk
|
let!() { let "$@" || true; } # always returns true; safe to use for setting
|
153 |
8899
|
aaronmk
|
# "If the last ARG evaluates to 0, let returns 1" (`help let`)
|
154 |
|
|
|
155 |
9074
|
aaronmk
|
bool2int() { try test ! "$1"; echo "$e"; } # empty->0; non-empty->1
|
156 |
9004
|
aaronmk
|
|
157 |
9538
|
aaronmk
|
int2bool() { (( "$1" != 0 )); }
|
158 |
9017
|
aaronmk
|
|
159 |
9538
|
aaronmk
|
|
160 |
9364
|
aaronmk
|
#### floats
|
161 |
|
|
|
162 |
|
|
int_part() { echo "${1%%.*}"; }
|
163 |
|
|
|
164 |
|
|
dec_suffix() { echo "${1#$(int_part "$1")}"; }
|
165 |
|
|
|
166 |
|
|
round_down() { int_part "$1"; }
|
167 |
|
|
|
168 |
|
|
float+int() { echo "$(($(int_part "$1")+$2))$(dec_suffix "$1")"; }
|
169 |
|
|
|
170 |
9544
|
aaronmk
|
float_set_min() { if (($(int_part $1) >= $2)); then echo $1; else echo $2; fi; }
|
171 |
9364
|
aaronmk
|
|
172 |
9544
|
aaronmk
|
|
173 |
9079
|
aaronmk
|
#### strings
|
174 |
|
|
|
175 |
9653
|
aaronmk
|
starts_with() { test "${2#$1}" != "$2"; } # usage: starts_with pattern str
|
176 |
|
|
|
177 |
9081
|
aaronmk
|
repeat() # usage: str=... n=... repeat
|
178 |
|
|
{
|
179 |
9304
|
aaronmk
|
: "${str?}" "${n:?}"; local result= n="$n" # n will be modified in function
|
180 |
|
|
for (( ; n > 0; n-- )); do result="$result$str"; done
|
181 |
|
|
echo "$result"
|
182 |
9081
|
aaronmk
|
}
|
183 |
|
|
|
184 |
9550
|
aaronmk
|
sed_cmd="sed -`case "$(uname)" in Darwin) echo E;; *) echo r;; esac`"
|
185 |
9366
|
aaronmk
|
alias sed="$sed_cmd"
|
186 |
9079
|
aaronmk
|
|
187 |
9366
|
aaronmk
|
fi # load new aliases
|
188 |
|
|
if self_being_included; then
|
189 |
9079
|
aaronmk
|
|
190 |
9425
|
aaronmk
|
rtrim() { log+ 3; sed 's/[[:space:]]+$//' <<<"$1"; }
|
191 |
9079
|
aaronmk
|
|
192 |
9085
|
aaronmk
|
|
193 |
8854
|
aaronmk
|
#### arrays
|
194 |
8694
|
aaronmk
|
|
195 |
9074
|
aaronmk
|
join() { local IFS="$delim"; echo "$*"; } # usage: delim=... join elems...
|
196 |
8816
|
aaronmk
|
|
197 |
9074
|
aaronmk
|
reverse() # usage: array=($(reverse args...))
|
198 |
8691
|
aaronmk
|
{
|
199 |
|
|
local i
|
200 |
9080
|
aaronmk
|
for (( i=$#; i > 0; i-- )); do printf '%q ' "${!i}"; done
|
201 |
8691
|
aaronmk
|
}
|
202 |
|
|
|
203 |
9097
|
aaronmk
|
contains() # usage: contains value in_array...
|
204 |
|
|
{
|
205 |
|
|
local value="$1"; shift
|
206 |
|
|
local elem
|
207 |
|
|
for elem in "$@"; do if test "$elem" = "$value"; then return 0; fi; done
|
208 |
|
|
return 1
|
209 |
|
|
}
|
210 |
9017
|
aaronmk
|
|
211 |
9097
|
aaronmk
|
|
212 |
8854
|
aaronmk
|
#### verbose output
|
213 |
|
|
|
214 |
9110
|
aaronmk
|
|
215 |
9208
|
aaronmk
|
err_fd=2 # stderr
|
216 |
|
|
|
217 |
9111
|
aaronmk
|
usage() { echo "Usage: $1" >&2; return 2; }
|
218 |
9110
|
aaronmk
|
|
219 |
|
|
|
220 |
9452
|
aaronmk
|
### log++
|
221 |
9110
|
aaronmk
|
|
222 |
9361
|
aaronmk
|
log_fd=2 # initially stderr
|
223 |
9209
|
aaronmk
|
|
224 |
9142
|
aaronmk
|
if test "$explicit_errors_only"; then verbosity=0; fi # hide startup logging
|
225 |
|
|
|
226 |
9006
|
aaronmk
|
# set verbosity
|
227 |
9005
|
aaronmk
|
if isset verbose; then : "${verbosity:=$(bool2int "$verbose")}"; fi
|
228 |
9145
|
aaronmk
|
if isset vb; then : "${verbosity:=$vb}"; fi
|
229 |
9272
|
aaronmk
|
: "${verbosity=1}" # default
|
230 |
9006
|
aaronmk
|
: "${verbosity:=0}" # ensure non-empty
|
231 |
9119
|
aaronmk
|
export verbosity # propagate to invoked commands
|
232 |
9381
|
aaronmk
|
export PS4 # follows verbosity, so also propagate this
|
233 |
8710
|
aaronmk
|
|
234 |
9542
|
aaronmk
|
# set log_level
|
235 |
|
|
: "${log_level=$(( ${#PS4}-1 ))}" # defaults to # non-space symbols in PS4
|
236 |
|
|
export log_level # propagate to invoked commands
|
237 |
|
|
|
238 |
9396
|
aaronmk
|
verbosity_int() { round_down "$verbosity"; }
|
239 |
|
|
|
240 |
9450
|
aaronmk
|
# verbosities (and `make` equivalents):
|
241 |
9359
|
aaronmk
|
# 0: just print errors. useful for cron jobs.
|
242 |
9451
|
aaronmk
|
# vs. make: equivalent to --silent, but suppresses external command output
|
243 |
9359
|
aaronmk
|
# 1: also external commands run. useful for running at the command line.
|
244 |
9450
|
aaronmk
|
# vs. make: not provided (but sorely needed to avoid excessive output)
|
245 |
9359
|
aaronmk
|
# 2: full graphical call tree. useful for determining where error occurred.
|
246 |
9450
|
aaronmk
|
# vs. make: equivalent to default verbosity, but with much-needed indents
|
247 |
9360
|
aaronmk
|
# 3: also values of kw params and variables. useful for low-level debugging.
|
248 |
9450
|
aaronmk
|
# vs. make: not provided; need to manually use $(error $(var))
|
249 |
9359
|
aaronmk
|
# 4: also variables in util.sh commands. useful for debugging util.sh.
|
250 |
9450
|
aaronmk
|
# vs. make: somewhat similar to --print-data-base
|
251 |
9359
|
aaronmk
|
# 5: also variables in logging commands themselves. useful for debugging echo_*.
|
252 |
9450
|
aaronmk
|
# vs. make: not provided; need to search Makefile for @ at beginning of cmd
|
253 |
9359
|
aaronmk
|
# 6+: not currently used (i.e. same as 5)
|
254 |
|
|
|
255 |
9287
|
aaronmk
|
# definition: the log_level is the minimum verbosity needed to display a message
|
256 |
|
|
# for messages that use can_log(), the log_level starts with *1*, not 0
|
257 |
|
|
# for unfiltered messages, the log_level is 0 (i.e. still output at verbosity=0)
|
258 |
9288
|
aaronmk
|
# to view a message's log_level, count the # of + signs before it in the output
|
259 |
9284
|
aaronmk
|
|
260 |
9305
|
aaronmk
|
fi # load new aliases
|
261 |
|
|
if self_being_included; then
|
262 |
|
|
|
263 |
9299
|
aaronmk
|
# usage: in func: log++; ... OR log_local; "log++"; ...
|
264 |
9176
|
aaronmk
|
# outside func: log++; ...; log--
|
265 |
9301
|
aaronmk
|
# before cmd: log++ cmd...
|
266 |
9302
|
aaronmk
|
# without a cmd, "$@" expands to nothing and assignments are applied to caller
|
267 |
9305
|
aaronmk
|
# "${@:2}" expands to all of $@ after *1st* arg, not 2nd ($@ indexes start at 1)
|
268 |
9373
|
aaronmk
|
log+()
|
269 |
|
|
{
|
270 |
|
|
# no local vars because w/o cmd, assignments should be applied to caller
|
271 |
9545
|
aaronmk
|
PS4="$(str="${PS4:0:1}" n=$((log_level+$1-1)) repeat)${PS4: -2}"; \
|
272 |
9543
|
aaronmk
|
log_level=$((log_level+$1)) \
|
273 |
9377
|
aaronmk
|
verbosity="$(float+int "$verbosity" "-$1")" "${@:2}"
|
274 |
9373
|
aaronmk
|
}
|
275 |
9374
|
aaronmk
|
log++() { log+ 1 "$@"; }
|
276 |
|
|
log--() { log+ -1 "$@"; }
|
277 |
9542
|
aaronmk
|
alias log_local=\
|
278 |
|
|
'declare PS4="$PS4" log_level="$log_level" verbosity="$verbosity"'
|
279 |
9395
|
aaronmk
|
alias log+='log_local; "log+"' # don't expand next word because it's not a cmd
|
280 |
9301
|
aaronmk
|
alias log++='log_local; "log++" ' # last space alias-expands next word
|
281 |
|
|
alias log--='log_local; "log--" ' # last space alias-expands next word
|
282 |
9176
|
aaronmk
|
|
283 |
9397
|
aaronmk
|
verbosity_min() # usage: verbosity_min min
|
284 |
|
|
{ if test "$(verbosity_int)" -lt "$1"; then verbosity="$1"; fi; }
|
285 |
|
|
alias verbosity_min='log_local; "verbosity_min"'
|
286 |
9289
|
aaronmk
|
|
287 |
9397
|
aaronmk
|
|
288 |
9289
|
aaronmk
|
# indent for call tree. this is *not* the log_level (below).
|
289 |
9382
|
aaronmk
|
: "${log_indent_step=| }" "${log_indent=}"
|
290 |
|
|
export log_indent_step log_indent # propagate to invoked commands
|
291 |
9289
|
aaronmk
|
|
292 |
9382
|
aaronmk
|
__caller_indent='log_indent="$log_indent$log_indent_step"'
|
293 |
9289
|
aaronmk
|
alias caller_indent="$__caller_indent"
|
294 |
|
|
alias indent="declare $__caller_indent"
|
295 |
|
|
|
296 |
|
|
|
297 |
9190
|
aaronmk
|
fi # load new aliases
|
298 |
|
|
if self_being_included; then
|
299 |
|
|
|
300 |
9396
|
aaronmk
|
can_log() { test "$(verbosity_int)" -gt 0; }
|
301 |
9380
|
aaronmk
|
# verbosity=0 turns off all logging
|
302 |
8895
|
aaronmk
|
|
303 |
9212
|
aaronmk
|
log() { if can_log; then echo "$log_indent$PS4$1" >&"$log_fd"; fi; }
|
304 |
9064
|
aaronmk
|
|
305 |
9270
|
aaronmk
|
log_custom() # usage: symbol=... log_custom msg
|
306 |
9242
|
aaronmk
|
{ log_indent="${log_indent//[^ ]/$symbol}" PS4="${PS4//[^ ]/$symbol}" log "$@";}
|
307 |
9070
|
aaronmk
|
|
308 |
9251
|
aaronmk
|
log_err() { symbol='#' verbosity=1 log_fd="$err_fd" log_custom "$@"; }
|
309 |
9070
|
aaronmk
|
|
310 |
9250
|
aaronmk
|
log_info() { symbol=: log_custom "$@"; }
|
311 |
9070
|
aaronmk
|
|
312 |
9446
|
aaronmk
|
die() # usage: cmd || [type=...] die msg (msg can use $? but not $())
|
313 |
9270
|
aaronmk
|
{ save_e; kw_params type; "log_${type:-err}" "$1"; rethrow; }
|
314 |
9069
|
aaronmk
|
|
315 |
9447
|
aaronmk
|
die_e() # usage: cmd || [benign_error=1] die_e [|| handle error]
|
316 |
|
|
{
|
317 |
|
|
save_e; kw_params benign_error
|
318 |
9449
|
aaronmk
|
if test "$benign_error"; then log++; fi
|
319 |
9447
|
aaronmk
|
type="${benign_error:+info}" die "command exited with \
|
320 |
|
|
$(if test "$benign_error"; then echo status; else echo error; fi) $e"
|
321 |
|
|
rethrow
|
322 |
|
|
}
|
323 |
8919
|
aaronmk
|
|
324 |
9437
|
aaronmk
|
|
325 |
9233
|
aaronmk
|
#### functions
|
326 |
|
|
|
327 |
|
|
func_exists() { declare -f "$1" >/dev/null; }
|
328 |
|
|
|
329 |
|
|
copy_func() # usage: from=... to=... copy_func
|
330 |
|
|
# $to must not exist. to get around the no-clobber restriction, use `unset -f`.
|
331 |
|
|
{
|
332 |
|
|
: "${from:?}" "${to:?}"
|
333 |
|
|
func_exists "$from" || die "function does not exist: $from"
|
334 |
|
|
! func_exists "$to" || die "function already exists: $to"
|
335 |
|
|
local from_def="$(declare -f "$from")"
|
336 |
|
|
eval "$to${from_def#$from}"
|
337 |
|
|
}
|
338 |
|
|
|
339 |
|
|
func_override() # usage: func_override old_name__suffix
|
340 |
9559
|
aaronmk
|
{ from="${1%__*}" to="$1" copy_func; }
|
341 |
9233
|
aaronmk
|
|
342 |
|
|
ensure_nested_func() # usage: func__nested_func() { ensure_nested_func; ... }
|
343 |
|
|
{
|
344 |
|
|
local nested_func="${FUNCNAME[1]}"
|
345 |
|
|
local func="${nested_func%%__*}"
|
346 |
|
|
contains "$func" "${FUNCNAME[@]}" || \
|
347 |
|
|
die "$nested_func() must be used by $func()"
|
348 |
|
|
}
|
349 |
|
|
|
350 |
|
|
|
351 |
9240
|
aaronmk
|
#### paths
|
352 |
|
|
|
353 |
9313
|
aaronmk
|
# cache realpath
|
354 |
|
|
: "${realpath_cache=}" # default off because slower than without
|
355 |
|
|
if test "$realpath_cache"; then
|
356 |
|
|
func_override realpath__no_cache
|
357 |
|
|
realpath() # caches the last result for efficiency
|
358 |
|
|
{
|
359 |
|
|
local cache_key="$*"; load_cache
|
360 |
9331
|
aaronmk
|
if ! cached; then save_cache "$(realpath__no_cache "$@")" || return; fi
|
361 |
9313
|
aaronmk
|
echo_cached_value
|
362 |
|
|
}
|
363 |
|
|
fi
|
364 |
|
|
|
365 |
9244
|
aaronmk
|
rel_path() # usage: base_dir=... path=... rel_path
|
366 |
9240
|
aaronmk
|
{
|
367 |
9244
|
aaronmk
|
log++; kw_params base_dir path
|
368 |
|
|
: "${base_dir:?}" "${path:?}"
|
369 |
|
|
|
370 |
9246
|
aaronmk
|
local path="$path/" # add *extra* / to match path when exactly = base_dir
|
371 |
|
|
path="${path#$base_dir/}" # remove prefix shared with base_dir
|
372 |
|
|
path="${path%/}" # remove any remaining extra trailing /
|
373 |
9244
|
aaronmk
|
|
374 |
9245
|
aaronmk
|
if test ! "$path"; then path=.; fi # ensure non-empty
|
375 |
|
|
|
376 |
9244
|
aaronmk
|
echo_vars path
|
377 |
9240
|
aaronmk
|
echo "$path"
|
378 |
|
|
}
|
379 |
|
|
|
380 |
9309
|
aaronmk
|
cd -P . # expand symlinks in $PWD so it matches the output of realpath
|
381 |
|
|
# do before setting $top_script_abs so realpath has less symlinks to resolve
|
382 |
|
|
|
383 |
9331
|
aaronmk
|
canon_rel_path()
|
384 |
|
|
{
|
385 |
|
|
local path; path="$(realpath "$1")" || return
|
386 |
|
|
base_dir="$PWD" rel_path
|
387 |
|
|
}
|
388 |
9244
|
aaronmk
|
|
389 |
9240
|
aaronmk
|
# makes $1 a canon_rel_path if it's a filesystem path
|
390 |
|
|
alias cmd2rel_path="$(cat <<'EOF'
|
391 |
|
|
if test "$(type -t "$1")" = file && test -e "$1"; then # not relative to PATH
|
392 |
|
|
declare _1="$1"; shift
|
393 |
9331
|
aaronmk
|
_1="$(canon_rel_path "$_1")" || return
|
394 |
|
|
set -- "$_1" "$@"
|
395 |
9240
|
aaronmk
|
fi
|
396 |
|
|
EOF
|
397 |
|
|
)"
|
398 |
|
|
|
399 |
|
|
|
400 |
|
|
#### verbose output
|
401 |
|
|
|
402 |
|
|
|
403 |
9165
|
aaronmk
|
### command echoing
|
404 |
|
|
|
405 |
|
|
alias echo_params='log "$*"'
|
406 |
|
|
|
407 |
8995
|
aaronmk
|
fi # load new aliases
|
408 |
|
|
if self_being_included; then
|
409 |
|
|
|
410 |
9197
|
aaronmk
|
echo_cmd() { echo_params; }
|
411 |
9056
|
aaronmk
|
|
412 |
9549
|
aaronmk
|
function echo_run() { echo_params; "$@"; }
|
413 |
|
|
alias echo_run='"echo_run" ' # last space alias-expands next word
|
414 |
9539
|
aaronmk
|
|
415 |
9278
|
aaronmk
|
echo_eval() { echo_params; builtin eval "$@"; }
|
416 |
|
|
|
417 |
9655
|
aaronmk
|
# usage: redirs=(...); echo_redirs_cmd
|
418 |
|
|
function echo_redirs_cmd()
|
419 |
|
|
{
|
420 |
9657
|
aaronmk
|
# print <>file redirs before cmd, because they introduce it
|
421 |
|
|
echo_cmd "$@" $(
|
422 |
|
|
set -- "${redirs[@]}" # operate on @redirs
|
423 |
|
|
while test "$#" -gt 0 && starts_with '[<>][^&]' "$1"
|
424 |
|
|
do log "$1 \\"; shift; done # log() will run *before* echo_cmd itself
|
425 |
|
|
echo "$@"
|
426 |
9655
|
aaronmk
|
)
|
427 |
|
|
}
|
428 |
|
|
alias echo_redirs_cmd='"echo_redirs_cmd" "$@"'
|
429 |
|
|
|
430 |
9263
|
aaronmk
|
## vars
|
431 |
|
|
|
432 |
|
|
echo_vars() # usage: echo_vars var...
|
433 |
|
|
{
|
434 |
9320
|
aaronmk
|
log+ 2
|
435 |
9263
|
aaronmk
|
if can_log; then
|
436 |
|
|
local var
|
437 |
|
|
for var in "${@%%=*}"; do
|
438 |
|
|
if isset "$var"; then log "$(declare -p "$var")"; fi
|
439 |
|
|
done
|
440 |
|
|
fi
|
441 |
|
|
}
|
442 |
|
|
|
443 |
|
|
echo_export() { builtin export "$@"; echo_vars "$@"; }
|
444 |
|
|
|
445 |
9379
|
aaronmk
|
alias export="echo_export" # automatically echo env vars when they are set
|
446 |
9263
|
aaronmk
|
|
447 |
|
|
func_override kw_params__lang
|
448 |
|
|
kw_params() { kw_params__lang "$@"; echo_vars "$@"; } # echo all keyword params
|
449 |
|
|
|
450 |
9273
|
aaronmk
|
## functions
|
451 |
9264
|
aaronmk
|
|
452 |
9315
|
aaronmk
|
# usage: local func=...; set_func_loc; use $file, $line
|
453 |
|
|
alias set_func_loc="$(cat <<'EOF'
|
454 |
|
|
: "${func:?}"
|
455 |
|
|
local func_info="$(shopt -s extdebug; declare -F "$func")" # 'func line file'
|
456 |
|
|
func_info="${func_info#$func }"
|
457 |
|
|
local line="${func_info%% *}"
|
458 |
|
|
local file="${func_info#$line }"
|
459 |
|
|
EOF
|
460 |
|
|
)"
|
461 |
|
|
|
462 |
|
|
fi # load new aliases
|
463 |
|
|
if self_being_included; then
|
464 |
|
|
|
465 |
|
|
func_loc() # gets where function declared in the format file:line
|
466 |
9331
|
aaronmk
|
{
|
467 |
|
|
local func="$1"; set_func_loc
|
468 |
|
|
file="$(canon_rel_path "$file")" || return
|
469 |
|
|
echo "$file:$line"
|
470 |
|
|
}
|
471 |
9315
|
aaronmk
|
|
472 |
9273
|
aaronmk
|
# usage: func() { [minor=1] echo_func; ... }
|
473 |
9317
|
aaronmk
|
function echo_func()
|
474 |
|
|
# usage: [minor=1] "echo_func" "$FUNCNAME" "$@" && indent || true
|
475 |
9273
|
aaronmk
|
# exit status: whether function call was echoed
|
476 |
|
|
{
|
477 |
|
|
kw_params minor
|
478 |
9317
|
aaronmk
|
local func="$1"; shift
|
479 |
9273
|
aaronmk
|
|
480 |
|
|
log++; if test "$minor"; then log++; fi
|
481 |
9331
|
aaronmk
|
local loc; loc="$(func_loc "$func")" || return
|
482 |
|
|
echo_cmd "$loc" "$func" "$@"
|
483 |
9273
|
aaronmk
|
can_log
|
484 |
|
|
}
|
485 |
9317
|
aaronmk
|
alias echo_func='"echo_func" "$FUNCNAME" "$@" && indent || true'
|
486 |
9273
|
aaronmk
|
|
487 |
9276
|
aaronmk
|
fi # load new aliases
|
488 |
|
|
if self_being_included; then
|
489 |
9273
|
aaronmk
|
|
490 |
9276
|
aaronmk
|
|
491 |
9279
|
aaronmk
|
#### streams
|
492 |
|
|
|
493 |
|
|
fd_exists() { (: <&"$1") 2>/dev/null; }
|
494 |
|
|
|
495 |
|
|
require_fd_not_exists() # usage: require_fd_not_exists fd || return 0
|
496 |
|
|
{ ! fd_exists "$1" || type=info die "fd $1 already exists, skipping"; }
|
497 |
|
|
|
498 |
|
|
set_fds() { echo_func; echo_eval exec "$@"; } # usage: set_fds redirect...
|
499 |
|
|
|
500 |
|
|
fd_set_default() # usage: fd_set_default 'dest[<>]src'
|
501 |
|
|
{
|
502 |
|
|
echo_func
|
503 |
|
|
local dest="${1%%[<>]*}"
|
504 |
|
|
require_fd_not_exists "$dest" || return 0
|
505 |
|
|
set_fds "$1"
|
506 |
|
|
}
|
507 |
|
|
|
508 |
9651
|
aaronmk
|
# convention: use fd 40/41/42 for command-specific alternate stdin/stdout/stderr
|
509 |
|
|
# do NOT use 1x, which are used by eval (which is used by set_fds())
|
510 |
|
|
# do NOT use 2x, which are used as global stdin/stdout/stderr
|
511 |
|
|
# do NOT use 3x, which are used for logging
|
512 |
9279
|
aaronmk
|
|
513 |
9282
|
aaronmk
|
setup_log_fd() # view logging output at verbosity >= 5
|
514 |
|
|
{
|
515 |
9319
|
aaronmk
|
log+ 4; log-- echo_func
|
516 |
9282
|
aaronmk
|
fd_set_default '30>&2' || true # stdlog
|
517 |
|
|
log_fd=30 # stdlog
|
518 |
|
|
}
|
519 |
|
|
setup_log_fd
|
520 |
|
|
|
521 |
9279
|
aaronmk
|
set_global_fds()
|
522 |
|
|
# allows commands to access global stdin/stdout/stderr using fd 20/21/22
|
523 |
|
|
# this works even when /dev/tty isn't available
|
524 |
9283
|
aaronmk
|
# view logging output at verbosity >= 3
|
525 |
9279
|
aaronmk
|
{
|
526 |
9319
|
aaronmk
|
log+ 2; log-- echo_func
|
527 |
9279
|
aaronmk
|
# ignore errors if a source fd isn't open
|
528 |
|
|
fd_set_default '20<&0' || true
|
529 |
|
|
fd_set_default '21>&1' || true
|
530 |
|
|
fd_set_default '22>&2' || true
|
531 |
|
|
}
|
532 |
|
|
set_global_fds
|
533 |
|
|
err_fd=22 # global stderr
|
534 |
|
|
|
535 |
|
|
# usage: explicit_errors_only=1 script...
|
536 |
|
|
# show only explicitly-displayed errors (which have been redirected to fd 22)
|
537 |
|
|
# most of the time this has the same effect as `verbosity=0 script...`,
|
538 |
|
|
# which displays everything that isn't explicitly hidden
|
539 |
|
|
# this option should only be used for testing the explicit error displaying
|
540 |
|
|
if test "$explicit_errors_only"; then disable_logging; fi
|
541 |
|
|
|
542 |
|
|
|
543 |
9264
|
aaronmk
|
#### paths
|
544 |
|
|
|
545 |
9269
|
aaronmk
|
top_script_abs="$(realpath "$0")"; echo_vars top_script_abs # outermost script
|
546 |
|
|
# realpath this before doing any cd so this points to the right place
|
547 |
|
|
|
548 |
9265
|
aaronmk
|
set_paths()
|
549 |
|
|
{
|
550 |
9331
|
aaronmk
|
top_script="$(canon_rel_path "$top_script_abs")" || return
|
551 |
|
|
echo_vars top_script
|
552 |
|
|
top_dir="$(dirname "$top_script")" || return; echo_vars top_dir
|
553 |
9265
|
aaronmk
|
}
|
554 |
|
|
set_paths
|
555 |
9264
|
aaronmk
|
|
556 |
|
|
|
557 |
|
|
#### verbose output
|
558 |
|
|
|
559 |
|
|
|
560 |
9123
|
aaronmk
|
## internal commands
|
561 |
|
|
|
562 |
9166
|
aaronmk
|
.()
|
563 |
|
|
{
|
564 |
9275
|
aaronmk
|
cmd2rel_path; set -- "$FUNCNAME" "$@"
|
565 |
9435
|
aaronmk
|
if (log++; echo_params; can_log); then indent; fi
|
566 |
9166
|
aaronmk
|
builtin "$@"
|
567 |
|
|
}
|
568 |
9158
|
aaronmk
|
|
569 |
9253
|
aaronmk
|
cd() # indent is permanent within subshell cd was used in
|
570 |
|
|
{
|
571 |
9393
|
aaronmk
|
log++ echo_func
|
572 |
9306
|
aaronmk
|
cmd2rel_path; echo_cmd "$FUNCNAME" "$@"
|
573 |
9435
|
aaronmk
|
if can_log; then caller_indent; fi
|
574 |
9307
|
aaronmk
|
# -P: expand symlinks so $PWD matches the output of realpath
|
575 |
9306
|
aaronmk
|
builtin "$FUNCNAME" -P "$@"
|
576 |
9313
|
aaronmk
|
|
577 |
|
|
func=realpath clear_cache
|
578 |
9266
|
aaronmk
|
set_paths
|
579 |
9253
|
aaronmk
|
}
|
580 |
|
|
|
581 |
9110
|
aaronmk
|
## external commands
|
582 |
|
|
|
583 |
9262
|
aaronmk
|
disable_logging() { set_fds "$log_fd>/dev/null"; }
|
584 |
9224
|
aaronmk
|
|
585 |
9407
|
aaronmk
|
function command()
|
586 |
9473
|
aaronmk
|
# usage: [cmd_log_fd=|1|2|#] [redirs=...] [verbosity_min=] command extern_cmd...
|
587 |
9138
|
aaronmk
|
# to view only explicitly-displayed errors: explicit_errors_only=1 script...
|
588 |
9134
|
aaronmk
|
{
|
589 |
9474
|
aaronmk
|
log++ echo_func; kw_params cmd_log_fd redirs verbosity_min
|
590 |
9291
|
aaronmk
|
# if no cmd_log_fd, limit log_fd in case command uses util.sh
|
591 |
|
|
local cmd_log_fd="${cmd_log_fd-$log_fd}"
|
592 |
9473
|
aaronmk
|
local redirs=("${redirs[@]}")
|
593 |
9231
|
aaronmk
|
|
594 |
9436
|
aaronmk
|
# determine redirections now so they can be logged along with the command
|
595 |
|
|
# the following redirections must happen in exactly this order
|
596 |
|
|
if test "$cmd_log_fd"; then
|
597 |
|
|
if can_log; then
|
598 |
|
|
if test "$cmd_log_fd" != "$log_fd"; then
|
599 |
9473
|
aaronmk
|
redirs+=("$cmd_log_fd>&$log_fd")
|
600 |
9436
|
aaronmk
|
fi # else no redir needed
|
601 |
9473
|
aaronmk
|
else redirs+=("$cmd_log_fd>/dev/null");
|
602 |
9436
|
aaronmk
|
fi
|
603 |
|
|
fi
|
604 |
9444
|
aaronmk
|
|
605 |
9655
|
aaronmk
|
cmd2rel_path; echo_redirs_cmd
|
606 |
9444
|
aaronmk
|
|
607 |
|
|
# add after echoing command so it isn't echoed at the end of every command
|
608 |
9436
|
aaronmk
|
if test "$cmd_log_fd" != 2; then # fd 2 not used for logging
|
609 |
9473
|
aaronmk
|
redirs+=("2>&$err_fd") # assume fd 2 used for errors
|
610 |
9436
|
aaronmk
|
fi
|
611 |
|
|
|
612 |
9192
|
aaronmk
|
(
|
613 |
9438
|
aaronmk
|
if can_log; then indent; fi
|
614 |
9473
|
aaronmk
|
log++ set_fds "${redirs[@]}"
|
615 |
9200
|
aaronmk
|
|
616 |
9407
|
aaronmk
|
if test "$verbosity_min"; then verbosity_min "$verbosity_min"; fi
|
617 |
|
|
|
618 |
9203
|
aaronmk
|
exec -- "$@" # -- so cmd name not treated as `exec` option
|
619 |
9439
|
aaronmk
|
) || die_e
|
620 |
9134
|
aaronmk
|
}
|
621 |
9133
|
aaronmk
|
|
622 |
9160
|
aaronmk
|
# auto-echo common external commands
|
623 |
|
|
for cmd in env rm; do alias "$cmd=command $cmd"; done; unset cmd
|
624 |
9110
|
aaronmk
|
|
625 |
|
|
|
626 |
|
|
### external command input/output
|
627 |
8907
|
aaronmk
|
|
628 |
9318
|
aaronmk
|
pipe_delay() # usage: cmd1 | { pipe_delay; cmd2; }
|
629 |
9270
|
aaronmk
|
{ sleep 0.1; } # s; display after leading output of cmd1
|
630 |
8997
|
aaronmk
|
|
631 |
|
|
fi # load new aliases
|
632 |
|
|
if self_being_included; then
|
633 |
|
|
|
634 |
9074
|
aaronmk
|
echo_stdin() # usage: input|echo_stdin|cmd
|
635 |
8702
|
aaronmk
|
{
|
636 |
8897
|
aaronmk
|
if can_log; then
|
637 |
8998
|
aaronmk
|
pipe_delay
|
638 |
9212
|
aaronmk
|
echo ----- >&"$log_fd"
|
639 |
|
|
tee -a /dev/fd/"$log_fd";
|
640 |
|
|
echo ----- >&"$log_fd"
|
641 |
8897
|
aaronmk
|
else cat
|
642 |
|
|
fi
|
643 |
8702
|
aaronmk
|
}
|
644 |
8275
|
aaronmk
|
|
645 |
9318
|
aaronmk
|
echo_stdout() { echo_stdin; } # usage: cmd|echo_stdout
|
646 |
9044
|
aaronmk
|
|
647 |
8873
|
aaronmk
|
|
648 |
8854
|
aaronmk
|
#### commands
|
649 |
|
|
|
650 |
9622
|
aaronmk
|
already_exists_msg() # usage: cond || what=... already_exists_msg || return 0
|
651 |
|
|
{ type=info die "$what already exists, skipping"; }
|
652 |
9620
|
aaronmk
|
|
653 |
9122
|
aaronmk
|
require_not_exists() # usage: require_not_exists file || return 0
|
654 |
9622
|
aaronmk
|
{ test ! -e "$1" || what="file \"$1\"" already_exists_msg; }
|
655 |
9062
|
aaronmk
|
|
656 |
9271
|
aaronmk
|
function to_file() # usage: stdout=... [if_not_exists=1] to_file cmd...
|
657 |
8986
|
aaronmk
|
# auto-removes a command's output file on error (like make's .DELETE_ON_ERROR)
|
658 |
9048
|
aaronmk
|
{
|
659 |
9235
|
aaronmk
|
echo_func; kw_params stdout
|
660 |
9234
|
aaronmk
|
: "${stdout?}"
|
661 |
9475
|
aaronmk
|
if test "$if_not_exists"; then require_not_exists "$stdout" || return 0; fi
|
662 |
9481
|
aaronmk
|
|
663 |
9652
|
aaronmk
|
local redirs=("${redirs[@]}" ">$stdout")
|
664 |
9669
|
aaronmk
|
"$@" || { save_e; unset redirs; rm "$stdout"; rethrow; }
|
665 |
9048
|
aaronmk
|
}
|
666 |
9177
|
aaronmk
|
alias to_file='"to_file" ' # last space alias-expands next word
|
667 |
8986
|
aaronmk
|
|
668 |
9074
|
aaronmk
|
run_args_cmd() # runs the command line args command
|
669 |
8272
|
aaronmk
|
{
|
670 |
8693
|
aaronmk
|
eval set -- "$(reverse "${BASH_ARGV[@]}")"
|
671 |
8971
|
aaronmk
|
test $# -ge 1 || set -- all
|
672 |
9189
|
aaronmk
|
echo_cmd "$top_script" "$@"; "$@"
|
673 |
8272
|
aaronmk
|
}
|
674 |
|
|
|
675 |
9074
|
aaronmk
|
fwd() # usage: subdirs=(...); fwd "$FUNCNAME" "$@"
|
676 |
8272
|
aaronmk
|
{
|
677 |
8881
|
aaronmk
|
echo_func
|
678 |
8284
|
aaronmk
|
: "${subdirs?}"
|
679 |
|
|
|
680 |
9297
|
aaronmk
|
for subdir in "${subdirs[@]}"; do "$top_dir"/"$subdir"/run "$@"; done
|
681 |
8272
|
aaronmk
|
}
|
682 |
|
|
|
683 |
9017
|
aaronmk
|
|
684 |
9552
|
aaronmk
|
#### filesystem
|
685 |
|
|
|
686 |
9556
|
aaronmk
|
alias file_size=\
|
687 |
9558
|
aaronmk
|
"stat `case "$(uname)" in Darwin) echo -f %z;; *) echo --format=%s;; esac`"
|
688 |
9552
|
aaronmk
|
|
689 |
|
|
|
690 |
8966
|
aaronmk
|
#### URLs
|
691 |
|
|
|
692 |
9074
|
aaronmk
|
localize_url() { test _"$1" = _"$(hostname -f)" || echo "$1"; }
|
693 |
8966
|
aaronmk
|
|
694 |
8704
|
aaronmk
|
fi
|