#!/bin/bash 
# Ruta almacenamiento de BBDD de Vulnerabilidades
RUTA=${HOME}/vuls/config

if [ $# -eq 0 ]; then
	echo "indique [--nvd --jvn --fortinet]"
	exit 1
fi

target=$1
shift

if [[ -z "${DOCKER_NETWORK}" ]]; then
	DOCKER_NETWORK_OPT=""
else
	DOCKER_NETWORK_OPT="--network ${DOCKER_NETWORK}"
fi

if [[ $(tty) =~ "not a tty" ]]
then
    t=''
else
    t="-t"
fi

# Actualizar + version
docker pull vuls/go-cve-dictionary
docker run --rm -i $t vuls/go-cve-dictionary version

case "$target" in
	--nvd) docker run --rm -i $t \
		-v $RUTA:/go-cve-dictionary \
		${DOCKER_NETWORK_OPT} \
		vuls/go-cve-dictionary fetch ${@} nvd
		;;
	--jvn) docker run --rm -i $t \
		-v $RUTA:/go-cve-dictionary \
		${DOCKER_NETWORK_OPT} \
		vuls/go-cve-dictionary fetch ${@} jvn
		;;
	--fortinet) docker run --rm -i $t \
		-v $RUTA:/go-cve-dictionary \
		${DOCKER_NETWORK_OPT} \
		vuls/go-cve-dictionary fetch ${@} fortinet
		;;
	--*)  echo "indique [--nvd --jvn --fortinet]"
		exit 1
	    ;;
	*) echo "indique [--nvd --jvn --fortinet]"
		exit 1
	    ;;
esac

exit 0