# Relational pipes

# Copyright © 2019 František Kučera (Frantovo.cz, GlobalCode.info)

#

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation, version 3 of the License.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program. If not, see <http://www.gnu.org/licenses/>.

_relpipe_tr_awk_completion() {

	local w0 w1 w2

	COMPREPLY=()

	w0=${COMP_WORDS[COMP_CWORD]}

	w1=${COMP_WORDS[COMP_CWORD-1]}

	w2=${COMP_WORDS[COMP_CWORD-2]}

	DATA_TYPE=(

		"string"

		"integer"

		"boolean"

	)

	if   [[ "$w1" == "--relation"          && "x$w0" == "x" ]];    then COMPREPLY=("'.*'")

	elif [[ "$w1" == "--before-records"    && "x$w0" == "x" ]];    then COMPREPLY=("''")

	elif [[ "$w1" == "--after-records"     && "x$w0" == "x" ]];    then COMPREPLY=("''")

	elif [[ "$w1" == "--output-attribute"  && "x$w0" == "x" ]];    then COMPREPLY=("''")

	elif [[ "$w2" == "--output-attribute"                   ]];    then COMPREPLY=($(compgen -W "${DATA_TYPE[*]}" -- "$w0"))

	elif [[ "$w1" == "--define"            && "x$w0" == "x" ]];    then COMPREPLY=("''")

	elif [[ "$w2" == "--define"                             ]];    then COMPREPLY=($(compgen -W "${DATA_TYPE[*]}" -- "$w0"))

	elif [[ "$w3" == "--define"            && "x$w0" == "x" ]];    then COMPREPLY=("''")

	elif [[ "$w1" == "--for-each"          && "x$w0" == "x" ]];    then COMPREPLY=("'(a == \"abc\") { a = \"X\"; record(); }'")

	else

		OPTIONS=(

			"--relation"

			"--output-attribute"

			"--input-attributes-append"

			"--input-attributes-prepend"

			"--debug-variable-mapping"

			"--before-records"

			"--after-records"

			"--for-each"

			"--where"

			"--drop"

			"--define"

		)

		COMPREPLY=($(compgen -W "${OPTIONS[*]}" -- "$w0"))

	fi

}

complete -F _relpipe_tr_awk_completion relpipe-tr-awk

complete -F _relpipe_tr_awk_completion relpipe-in-awk