#compdef ssh-add

_ssh-add() {
	local -a keys
	for key in $HOME/.ssh/*; do
		# only accept files that do not end in .pub
		if [[ ${key#*.pub} == "$key" ]]; then
			# exclude files that are not ssh keys
			if [[ ${key#*/config} == "$key" && ${key#*/known_hosts} == "$key" ]]; then
				keys+=($key)
			fi
		fi
	done
	compadd "$@" $keys
}

_ssh-add "$@"