#!/bin/bash -e

. "$(dirname "${BASH_SOURCE[0]}")"/util.sh

if self_not_included; then

# using prefixed connection vars

alias use_local='declare prefix=local_; import_vars'

alias use_remote='declare prefix=remote_; import_vars'

alias use_local_remote='{ use_local; use_remote; }'

	# *must* be run inside a function

quote='"'

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

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

alias mk_schema_esc='declare schema_esc; mk_esc_name schema'

alias mk_table_esc='declare table_esc; mk_esc_name table'

alias mk_select='mk_table_esc

declare query="${query:-SELECT * FROM $table_esc}${limit:+ LIMIT $limit}"'

fi # load new aliases

if self_being_included; then

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

### MySQL

alias set_database=\

'if test "$schema"; then declare database="${database-$schema}"; fi'

fi # load new aliases

if self_being_included; then

# auto-adds connection/login opts when specified

# usage: mysql*() { ...; mysql_cmd "$@"; } (with alias)

function mysql_cmd() # usage: mysql*() { ...; mysql_cmd cmd "$@"; }

{

	echo_func

	local cmd="$1"; shift

	local ssh_server="$(localize_url "$ssh_server")"

	local server="$(localize_url "$server")"

	if test "$ssh_server"; then

		local ssh_dest="${ssh_dest-${ssh_user:+$ssh_user@}$ssh_server}"

	fi

	local var=ssh_dest; local_inv

	command ${ssh_dest:+ssh "$ssh_dest" }"$cmd" \

${server:+ --host="$server" }${user:+--user="$user" } --password\

${password+="$password"} "$@"

}

alias mysql_cmd='mysql_cmd "$FUNCNAME"'

fi # load new aliases

if self_being_included; then

mysql() # usage: [output_data=1] mysql ...

{

	echo_func

	set_database

	set -- ${database:+--database="$database" }--column-names "$@"

	if test "$output_data"; then echo_stdin|mysql_cmd --batch "$@"

	else cmd_log_fd=1 mysql_cmd --verbose "$@"

	fi

}

mysql_ANSI()

{

	echo_func

	(echo "SET sql_mode = 'ANSI';"; cat)|mysql "$@"

}

mysql_export() # does not support CSV

{

	echo_func

	mk_select

	output_data=1 mysql_ANSI "$@" <<<"$query"

}

mysql_export_outfile() # supports CSV, but requires the FILE privilege

{

	echo_func

	: "${file:?}"

	mk_select

	local mysql_load_data_format="${mysql_load_data_format-\

FIELDS TERMINATED BY ','

OPTIONALLY ENCLOSED BY '\"'

}"

	local head="${query%%FROM*}" # includes trailing newline

	head="${head%

}"

	local tail="${query#$head}"

	mysql_load_data_format="${mysql_load_data_format%

}"

	mysql_ANSI "$@" <<EOF

$head

INTO OUTFILE '$file'

$mysql_load_data_format

$tail

EOF

}

mysqldump() # usage: [schema=1 | data=1] mysqldump db [table...]

{

	echo_func

	mysql_cmd ${database:+--databases "$database" --tables } \

--quick --lock-tables=false --set-charset \

${postgres_compat:+--compatible=postgresql --add-locks=false }\

${schema:+--no-data }${data:+--no-create-info }"$@"

}

mysqldump_diffable()

{

	echo_func

	mysqldump "$@"|{ pipe_delay; sed 's/^(-- Dump completed).*$/\1/'; }

}

### PostgreSQL

pg_export()

{

	echo_func

	mk_select

	local pg_copy_format="${pg_copy_format-CSV HEADER}"

	psql "$@" <<<"COPY ($query) TO STDOUT $pg_copy_format;"

}

pg_header()

{

	echo_func

	local pg_copy_format="CSV HEADER" limit=0

	pg_export "$@"|echo_stdout

}

pg_export_table_no_header()

{

	echo_func

	local pg_copy_format="CSV"

	pg_export "$@"

}

pg_export_table_to_dir_no_header()

{

	echo_func

	local table="$1"; shift; mk_table_esc

	local cols="$(pg_header)"

	stdout="$exports_dir/$table.no_header.cols=$cols.csv" to_file \

pg_export_table_no_header "$@"

}

fi