# 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_out_tabular_completion() {
	local w0 w1 w2 w3

	COMPREPLY=()
	w0=${COMP_WORDS[COMP_CWORD]}
	w1=${COMP_WORDS[COMP_CWORD-1]}
	w2=${COMP_WORDS[COMP_CWORD-2]}
	w3=${COMP_WORDS[COMP_CWORD-3]}

	BOOLEAN_VALUES=(
		"true"
		"false"
	)

	TABLE_STYLES=(
		"rounded"
		"sharp"
		"sharp-double"
		"horizontal-only"
		"ascii"
	)

	COLOR_SCHEMES=(
		"greenish"
		"amberish"
		"black-and-white"
		"midnight"
	)

	if   [[ "$w1" == "--relation"                      && "x$w0" == "x" ]];    then COMPREPLY=("'.+'")
	elif [[ "$w1" == "--write-types"                                    ]];    then COMPREPLY=($(compgen -W "${BOOLEAN_VALUES[*]}" -- "$w0"))
	elif [[ "$w1" == "--write-relation-name"                            ]];    then COMPREPLY=($(compgen -W "${BOOLEAN_VALUES[*]}" -- "$w0"))
	elif [[ "$w1" == "--write-record-count"                             ]];    then COMPREPLY=($(compgen -W "${BOOLEAN_VALUES[*]}" -- "$w0"))
	elif [[ "$w1" == "--table-style"                                    ]];    then COMPREPLY=($(compgen -W "${TABLE_STYLES[*]}" -- "$w0"))
	elif [[ "$w1" == "--color-scheme"                                   ]];    then COMPREPLY=($(compgen -W "${COLOR_SCHEMES[*]}" -- "$w0"))
	else
		OPTIONS=(
			"--relation"
			"--write-types"
			"--write-relation-name"
			"--write-record-count"
			"--table-style"
			"--color-scheme"
		)
		COMPREPLY=($(compgen -W "${OPTIONS[*]}" -- "$w0"))
	fi
}

complete -F _relpipe_out_tabular_completion relpipe-out-tabular