# shellcheck shell=bash

# ---------------------------------------------------------------------------+
#                                                                            |
# Thanks to Alexander Korznikov                                              |
# http://www.korznikov.com/2014/12/bash-tab-completion-for-awesome-tool.html |
#                                                                            |
# ---------------------------------------------------------------------------+

_command_exists sqlmap || return

function _sqlmap() {
	local cur prev

	COMPREPLY=()
	cur="$(_get_cword)"
	prev="$(_get_pword)"

	case $prev in

		# List directory content
		--tamper)
			COMPREPLY=($(compgen -W "$tamper" -- "$cur"))
			return 0
			;;
		--output-dir | -t | -l | -m | -r | --load-cookies | --proxy-file | --sql-file | --shared-lib | --file-write)
			_filedir
			return 0
			;;
		-c)
			_filedir ini
			return 0
			;;
		--method)
			COMPREPLY=($(compgen -W 'GET POST PUT' -- "$cur"))
			return 0
			;;
		--auth-type)
			COMPREPLY=($(compgen -W 'Basic Digest NTLM PKI' -- "$cur"))
			return 0
			;;
		--tor-type)
			COMPREPLY=($(compgen -W 'HTTP SOCKS4 SOCKS5' -- "$cur"))
			return 0
			;;
		-v)
			COMPREPLY=($(compgen -W '1 2 3 4 5 6' -- "$cur"))
			return 0
			;;
		--dbms)
			COMPREPLY=($(compgen -W 'mysql mssql access postgres' -- "$cur"))
			return 0
			;;
		--level | --crawl)
			COMPREPLY=($(compgen -W '1 2 3 4 5' -- "$cur"))
			return 0
			;;
		--risk)
			COMPREPLY=($(compgen -W '0 1 2 3' -- "$cur"))
			return 0
			;;
		--technique)
			COMPREPLY=($(compgen -W 'B E U S T Q' -- "$cur"))
			return 0
			;;
		-s)
			_filedir sqlite
			return 0
			;;
		--dump-format)
			COMPREPLY=($(compgen -W 'CSV HTML SQLITE' -- "$cur"))
			return 0
			;;
		-x)
			_filedir xml
			return 0
			;;
	esac

	if [[ "$cur" == * ]]; then
		COMPREPLY=($(compgen -W '-h --help -hh --version -v -d -u --url -l -x -m -r -g -c --method \
			--data --param-del --cookie --cookie-del --load-cookies \
			--drop-set-cookie --user-agent --random-agent --host --referer \
			--headers --auth-type --auth-cred --auth-private --ignore-401 \
			--proxy --proxy-cred --proxy-file --ignore-proxy --tor --tor-port \
			--tor-type --check-tor --delay --timeout --retries --randomize \
			--safe-url --safe-freq --skip-urlencode --csrf-token --csrf-url \
			--force-ssl --hpp --eval -o --predict-output --keep-alive \
			--null-connection --threads -p  --skip --dbms --dbms-cred \
			--os --invalid-bignum --invalid-logical --invalid-string \
			--no-cast --no-escape --prefix --suffix --tamper --level \
			--risk --string --not-string --regexp --code --text-only \
			--titles --technique --time-sec --union-cols --union-char \
			--union-from --dns-domain --second-order -f --fingerprint \
			-a --all -b --banner --current-user --current-db --hostname \
			--is-dba --users --passwords --privileges --roles --dbs --tables \
			--columns --schema --count --dump --dump-all --search --comments \
			-D -T -C -X -U --exclude-sysdbs --where --start --stop \
			--first --last --sql-query --sql-shell --sql-file --common-tables \
			--common-columns --udf-inject --shared-lib --file-read --file-write \
			--file-dest --os-cmd --os-shell --os-pwn --os-smbrelay --os-bof \
			--priv-esc --msf-path --tmp-path --reg-read --reg-add --reg-del \
			--reg-key --reg-value --reg-data --reg-type -s -t --batch \
			--charset --crawl --csv-del --dump-format --eta --flush-session \
			--forms --fresh-queries --hex --output-dir --parse-errors \
			--pivot-column --save --scope --test-filter --update \
			-z --alert --answers --beep --check-waf --cleanup \
			--dependencies --disable-coloring --gpage --identify-waf \
			--mobile --page-rank --purge-output --smart \
			--sqlmap-shell --wizard' -- "$cur"))
		# this removes any options from the list of completions that have
		# already been specified somewhere on the command line, as long as
		# these options can only be used once (in a word, "options", in
		# opposition to "tests" and "actions", as in the find(1) manpage).
		onlyonce=' -h --help -hh --version -v -d -u --url -l -x -m -r -g -c \
			--drop-set-cookie --random-agent \
			--ignore-401 \
			--ignore-proxy --tor \
			--check-tor \
			--skip-urlencode \
			--force-ssl --hpp -o --predict-output --keep-alive \
			--null-connection -p \
			--invalid-bignum --invalid-logical --invalid-string \
			--no-cast --no-escape \
			--text-only \
			--titles \
			-f --fingerprint \
			-a --all -b --banner --current-user --current-db --hostname \
			--is-dba --users --passwords --privileges --roles --dbs --tables \
			--columns --schema --count --dump --dump-all --search --comments \
			-D -T -C -X -U --exclude-sysdbs \
			--sql-shell --common-tables \
			--common-columns --udf-inject \
			--os-shell --os-pwn --os-smbrelay --os-bof \
			--priv-esc --reg-read --reg-add --reg-del \
			-s -t --batch \
			--eta --flush-session \
			--forms --fresh-queries --hex --parse-errors \
			--save --update \
			-z --beep --check-waf --cleanup \
			--dependencies --disable-coloring --identify-waf \
			--mobile --page-rank --purge-output --smart \
			--sqlmap-shell --wizard '
		COMPREPLY=($(
			(
				while read -d ' ' i; do
					[[ -z "$i" || "${onlyonce/ ${i%% *} / }" == "$onlyonce" ]] && continue
					# flatten array with spaces on either side,
					# otherwise we cannot grep on word boundaries of
					# first and last word
					COMPREPLY=" ${COMPREPLY[@]} "
					# remove word from list of completions
					COMPREPLY=(${COMPREPLY/ ${i%% *} / })
				done
				printf '%s ' "${COMPREPLY[@]}"
			) <<< "${COMP_WORDS[@]}"
		))

		#else
		#_filedir bat
	fi
}

complete -F _sqlmap sqlmap