#!/bin/bash

set -euo pipefail

IMAGE="$2"
TSV="./$3.tsv"
CSV="./csv/$3-vuln.csv"

if [[ "$1" == "--host" ]]; then
	sudo docker pull $IMAGE

	sudo docker run --rm $IMAGE \
		bash -lc "dpkg-query -W -f='\${binary:Package}\t\${Version}\t\${Architecture}\n'" \
		| sort > $TSV

elif [[ "$1" == "--dpu" ]]; then
	TMP=$(mktemp)

	ssh ubuntu@192.168.100.2 bash -s <<EOF
sudo docker pull $IMAGE
sudo docker run --rm $IMAGE bash -lc "dpkg-query -W -f='\\\${binary:Package}\t\\\${Version}\t\\\${Architecture}\n'" \
| sort | tee "$TMP"
EOF
	scp -P 22 ubuntu@192.168.100.2:$TMP $TSV

else
	echo "OH NO!!"
	exit
fi

python3 query.py $TSV \
	--ecosystem "Ubuntu:22.04:LTS" \
	--csv $CSV \
	--json /dev/null