#!/bin/bash

EXE=./magellan
SUBNETS=""
PORTS=""
USER=""
PASS=""
SMD_HOST=""
SMD_PORT=""
THREADS="-1"
TIMEOUT="30"
ARGS=""


function build(){
	go mod tidy && go build -C bin/magellan
}

function scan() {
	# ./magellan scan --subnet 172.16.0.0 --port 443
	${EXE} scan --subnet ${SUBNETS} --port ${PORTS} --timeout ${TIMEOUT} --threads ${THREADS}
}

function list(){
	# ./magellan list
	${EXE} list 
}

function collect() {
	# ./magellan collect --user admin --pass password
	${EXE} collect --user ${USER} --pass ${PASS} --timeout ${TIMEOUT} --threads ${THREADS} --host ${SMD_HOST} --port ${SMD_PORT}
}


# parse incoming arguments to set variables
while [[ $# -gt 0 ]]; do
  case $1 in
	--subnet)
		SUBNETS="$2"
		shift # past argument
		shift # past value
		;;
	-p|--port)
		PORTS="$2"
		shift # past argument
		shift # past value
		;;
	--user)
		USER="$2"
		shift # past argument
		shift # past value
		;;
	--pass|--password)
		PASS="$2"
		shift
		shift
		;;
	--smd-host)
		SMD_HOST="$2"
		shift
		shift
		;;
	--smd-port)
		SMD_PORT="$2"
		shift
		shift
		;;
	--timeout)
		TIMEOUT="$2"
		shift
		shift
		;;
	--threads)
		THREADS="$2"
		shift
		shift
		;;
	-*|--*)
		echo "Unknown option $1"
		exit 1
		;;
	*)
		ARGS+=("$1") # save positional arg
		shift # past argument
		;;
  esac
done

set -- "${POSITIONAL_ARGS[@]}" # restore positional parameters

echo "SMD host	= ${SMD_HOST}"
echo "SMD port	= ${SMD_PORT}"
echo "subnets 	= ${SUBNETS}"
echo "ports	 	= ${PORTS}"
echo "user 		= ${USER}"
echo "pass 		= ..."

if [[ -n $1 ]]; then
	echo "Last line of file specified as non-opt/last argument:"
	tail -1 "$1"
fi

scan
collect