Project

General

Profile

1
#!/bin/bash -e
2
realpath () { readlink -f -- "$1"; }
3

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

    
6
self_not_included () # usage: if self_not_included; then ... fi
7
{
8
	test "$#" -ge 1 || set -- "${BASH_SOURCE[1]}"
9
	local include_guard="$(include_guard_var "$1")"
10
	alias self_being_included=false
11
	test -z "${!include_guard+t}" && \
12
	{ eval "$include_guard"=1; alias self_being_included=true; }
13
}
14

    
15
# to load newly-defined aliases for use in functions in the same file:
16
## fi # load new aliases
17
## if self_being_included; then
18
# this is needed because aliases defined inside an if statement are not
19
# available inside that if statement
20

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

    
23
shopt -s expand_aliases
24

    
25
#### arrays
26

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

    
29
reverse () # usage: array=($(reverse args...))
30
{
31
	local i
32
	for (( i=$#; i >= 1; i-- )); do printf '%q ' "${!i}"; done
33
}
34

    
35
#### strings
36

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

    
39
sed () { echo_run env sed -"$sed_ere_flag" "$@";}
40

    
41
#### verbose output
42

    
43
: "${verbosity:=$verbose}" "${verbosity:=0}"
44

    
45
echo_cmd () { echo "$PS4$*" >&2; }
46

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

    
49
canon_rel_path ()
50
{
51
	local path="$1"
52
	path="$(realpath "$path")" # canonicalize
53
	path="${path#$(pwd -P)/}" # remove any shared prefix with the current dir
54
	echo "$path"
55
}
56

    
57
# usage: echo_func "$@"
58
echo_func ()
59
{
60
	local script="$(canon_rel_path "${BASH_SOURCE[1]}")"
61
	echo_cmd "$script:${BASH_LINENO[0]}" "${FUNCNAME[1]}" "$@"
62
}
63

    
64
echo_stdin () # usage: input|echo_stdin|cmd
65
{
66
	echo ----- >&2
67
	tee -a /dev/stderr;
68
	echo ----- >&2
69
}
70

    
71
echo_vars () # usage: echo_vars var...
72
{ { echo -n "$PS4"; declare -p "${@%%=*}";} >&2; }
73

    
74
echo_export ()
75
{
76
	builtin export "$@"
77
	echo_vars "$@"
78
}
79

    
80
if test "$verbosity" -ge 2; then
81
	alias export="echo_export" # automatically echo env vars when they are set
82
fi
83

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

    
86
#### vars
87

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

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

    
92
# usage: local var=...; local_inv
93
alias local_inv='local "no_$var=$(test -n "${!var}" || echo 1)"'
94

    
95
# usage: local prefix=..._; import_vars
96
alias import_vars="$(cat <<'EOF'
97
: "${prefix:?}"
98
local src_var dest_var
99
for src_var in $(eval echo '${!'$prefix'*}'); do
100
	dest_var="${src_var#$prefix}"
101
	local "$dest_var=${!src_var}"; echo_vars "$dest_var"
102
done
103
EOF
104
)"
105

    
106
#### commands
107

    
108
top_dir="$(dirname "$0")" # outermost script
109

    
110
run_args_cmd () # runs the command line args command
111
{
112
	test "$?" -eq 0 || return
113
	eval set -- "$(reverse "${BASH_ARGV[@]}")"
114
	test "$#" -ge 1 || set -- all
115
	echo_cmd "$(canon_rel_path "$0")" "$@"; "$@"
116
}
117

    
118
fwd () # usage: subdirs=(...); fwd "$FUNCNAME" "$@"
119
{
120
	echo_func "$@"
121
	: "${subdirs?}"
122
	
123
	for subdir in "${subdirs[@]}"; do
124
		"$(dirname "${BASH_SOURCE[1]}")"/"$subdir"/run "$@"
125
	done
126
}
127

    
128
#### make
129

    
130
make ()
131
{
132
	echo_func "$@"
133
	echo_run env make --directory="$top_dir" "$@"
134
}
135

    
136
if false; then ## usage:
137
inline_make 3<<'EOF'
138
target:
139
	$(self_dir)/cmd >$@
140
EOF
141
# target will be run automatically because it's first in the makefile
142
fi ##
143
inline_make ()
144
{
145
	echo_func "$@"
146
	local self="$(readlink -f "${BASH_SOURCE[1]}")"
147
	local self_dir="$(dirname "$self")"
148
	export self self_dir
149
	
150
	make --makefile=<((
151
		cat /dev/fd/3
152
		echo -n "
153
.SUFFIXES: # turn off built-in suffix rules
154
.SECONDARY: # don't automatically delete intermediate files
155
.DELETE_ON_ERROR: # delete target if recipe fails
156
"
157
	)|echo_stdin) "$@"
158
}
159

    
160
#### compression
161

    
162
### zip
163

    
164
alias zip="echo_run zip"
165
alias unzip="echo_run unzip"
166
set_inv force
167
alias zip_newer="zip${no_force:+ -u}"
168
alias unzip_newer="unzip${no_force:+ -u} -o"
169
	# -o is safe because -u only extracts newer files
170

    
171
#### databases
172

    
173
# uses the remote_* connection vars
174
alias use_remote="local prefix=remote_; import_vars"
175

    
176
quote='"'
177

    
178
esc_name () { echo "$quote${1//$quote/$quote$quote}$quote"; }
179

    
180
mk_esc_name () { set_var "$1"_esc "$(esc_name "${!1}")"; }
181

    
182
alias mk_schema_esc="local schema_esc; mk_esc_name schema"
183
alias mk_table_esc="local table_esc; mk_esc_name table"
184

    
185
fi # load new aliases
186
if self_being_included; then
187

    
188
log_sql () { test "$verbosity" -ge 2; }
189

    
190
### MySQL
191

    
192
# auto-adds connection/login opts when specified
193
mysql_cmd () # usage: mysql* () { ...; mysql_cmd "$@"; }
194
{
195
	echo_func "$@"
196
	local var=ssh_dest; local_inv
197
	echo_run ${no_ssh_dest:+env }${ssh_dest:+ssh "$ssh_dest" }"${FUNCNAME[1]}" \
198
${server:+ --host="$server" }${user+--user="$user" }\
199
--password${password+="$password"} "$@"
200
}
201

    
202
mysql ()
203
{
204
	echo_func "$@"
205
	mysql_cmd --verbose "$@"
206
}
207

    
208
mysql_ANSI ()
209
{
210
	echo_func "$@"
211
	(echo "SET sql_mode = 'ANSI';"; cat)|mysql "$@"
212
}
213

    
214
### PostgreSQL
215

    
216
pg_copy_to ()
217
{
218
	echo_func "$@"
219
	if test -z "$source"; then
220
		: "${table:?}"; mk_table_esc
221
		if test -z "$limit"; then local source="$table_esc"
222
		else local source="(SELECT * FROM $table_esc LIMIT $limit)"
223
		fi
224
	fi
225
	local pg_copy_format="${pg_copy_format-CSV HEADER}"
226
	
227
	psql "$@" <<<"COPY $source TO STDOUT $pg_copy_format;"
228
}
229

    
230
pg_header ()
231
{
232
	echo_func "$@"
233
	local pg_copy_format="CSV HEADER" limit=0
234
	pg_copy_to "$@"|echo_stdin
235
}
236

    
237
pg_export_table_no_header ()
238
{
239
	echo_func "$@"
240
	local pg_copy_format="CSV"
241
	pg_copy_to "$@"
242
}
243

    
244
pg_export_table_to_dir_no_header ()
245
{
246
	echo_func "$@"
247
	local table="$1"; shift; mk_table_esc
248
	local cols="$(pg_header)"
249
	pg_export_table_no_header "$@" >"$exports_dir/$table.no_header.cols=$cols.csv"
250
}
251

    
252
fi
(45-45/50)