# Relational pipes
# Copyright © 2022 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_in_socket_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]}

	CONNECTION_STRINGS=(
		"udp://127.0.0.1:64000"
		"tcp://127.0.0.1:64000"
		"sctp://127.0.0.1:64000"
		"uds:///tmp/relpipe.socket"
	)

	OPTIONS=(
		"protocol"
		"role"
		"mode"
		"host"
		"port"
		"path"
		"delay"
	)

	PROTOCOLS=(
		"tcp"
		"udp"
		"uds"
		"sctp"
	)

	ROLES=(
		"client"
		"server"
	)

	MODES=(
		"stream"
		"datagram"
	)

	if   [[ "$w1" == "--relation"                      && "x$w0" == "x" ]];    then COMPREPLY=("''")
	elif [[ "$w1" == "--message-count"                 && "x$w0" == "x" ]];    then COMPREPLY=("1")
	elif [[ "$w1" == "--connection-string"                              ]];    then COMPREPLY=($(compgen -W "${CONNECTION_STRINGS[*]}" -- "$w0"))
	elif [[ "$w1" == "--connection-option"                              ]];    then COMPREPLY=($(compgen -W "${OPTIONS[*]}" -- "$w0"))
	elif [[ "$w2" == "--connection-option" && "$w1" == "protocol"       ]];    then COMPREPLY=($(compgen -W "${PROTOCOLS[*]}" -- "$w0"))
	elif [[ "$w2" == "--connection-option" && "$w1" == "role"           ]];    then COMPREPLY=($(compgen -W "${ROLES[*]}" -- "$w0"))
	elif [[ "$w2" == "--connection-option" && "$w1" == "mode"           ]];    then COMPREPLY=($(compgen -W "${MODES[*]}" -- "$w0"))
	elif [[ "$w2" == "--connection-option" && "$w1" == "path"           ]];    then COMPREPLY=($(compgen -f -- "$w0"))
	else
		OPTIONS=(
			"--relation"
			"--message-count"
			"--connection-string"
			"--connection-option"
		)
		COMPREPLY=($(compgen -W "${OPTIONS[*]}" -- "$w0"))
	fi
}

complete -F _relpipe_in_socket_completion relpipe-in-socket