#!/bin/bash -e

# links to locally-available resources

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

.rel db.sh

.rel make.sh

.rel sync.sh

if self_not_included; then

root_dir="$(canon_rel_path "$(dirname "${BASH_SOURCE[0]}")"/../..)" || return

root_rel_path() { echo_func; base_dir="$root_dir" canon_rel_path "$1"; }

bin_dir="$root_dir"/bin

bin_dir_abs="$(realpath "$bin_dir")" || return

: "${PATH_add=1}"

if test "$PATH_add"; then export PATH="$bin_dir_abs:$PATH"; fi

#### make

root_make() { echo_func; make --directory="$root_dir" "$@"; }

#### connection vars

: "${remote_ssh_server=vegbiendev.nceas.ucsb.edu}"

: "${remote_server=}"

: "${local_server=localhost}"

: "${local_user=bien}"

: "${local_password="$(cat "$root_dir"/config/bien_password)"}"

: "${local_pg_database=vegbien}"

: "${root_user=root}"

: "${root_password=$local_password}"

#### ssh

require_remote()

{

	echo_func

	test "$(hostname -f)" = "$remote_ssh_server" \

		|| die "must be run on $remote_ssh_server"

}

#### databases

### MySQL

: "${postgres_compat=1}"

mysql_local() { echo_func; use_local_remote; mysql "$@"; }

mysql_export_local() { echo_func; use_local_remote; mysql_export "$@"; }

mysqldump_local() { echo_func; use_local_remote; mysqldump_diffable "$@"; }

mysql_root() { echo_func; use_local_remote; use_root; mysql "$@"; }

### PostgreSQL

func_override psql__db_sh

psql() # usage: [file=...] [dir=...] self

{

	echo_func; kw_params file dir

	if test "$file"; then local dir="${dir:-$(dirname "$file")}"; fi

	local dir="${dir:-$top_dir}"

	if isset public; then local schema="${schema:-$public}"; fi

	mk_schema_esc

	mk_table_esc

	local is_root="`case "$USER" in *postgres) echo 1;; esac`"

		echo_vars is_root

	if test ! "$is_root"; then use_local; fi

	(if test "$schema" -a "$table"; then cat <<EOF

\cd $dir

\set schema $schema_esc

\set table $table_esc

\set table_str '''$table_esc'''

SET search_path TO $schema_esc;

EOF

	fi

	cat${file:+ "$file"}

	)|psql__db_sh "$@"

}

public_schema_exists() { log++ echo_func; schema=public pg_schema_exists; }

pg_dump_local() { echo_func; use_local; pg_dump "$@"; }

#### sync

: "${sync_local_dir=$root_dir}"

: "${sync_remote_subdir=bien/}"

: "${sync_remote_url=jupiter:/data/dev/aaronmk/$sync_remote_subdir}"

sync_upload() { echo_func; use_sync; upload --no-owner --no-group "$@"; }

#### data loading

# prevent automated tests when the public schema contains the live DB

if test "$(hostname -s)" = vegbiendev; then export can_test=; fi

fi