#!/bin/bash

export CONTAINER_ID=$(basename $(cat /proc/1/cpuset))
GITLAB_TOKEN_SECRET=$(cat /var/run/secrets/dicom_bot_gitlab_token 2>/dev/null)
export GITLAB_TOKEN=${GITLAB_TOKEN_SECRET:=$GITLAB_TOKEN}
export GITLAB_API_URL=https://${CI_SERVER_HOST}/api/v4
export GIT_SSH_PORT=${GIT_SSH_PORT:=222}

mkdir -p ~/.ssh
# only export keys when deploying as a service on swarm
# TODO: should try using gitlab runner mechanism if not
if [ -n "${GITLAB_TOKEN}" ] ; then
	# generate container specific ssh-key
	ssh-keygen -f  /root/.ssh/id_rsa -N ''
	# register it for dicom_bot user
	echo 'registering the ssh key'
	export ssh_key_json=$(curl -X POST -F "private_token=${GITLAB_TOKEN}" \
	  -F "title="${HOSTNAME} -F "key=$(cat ~/.ssh/id_rsa.pub)" \
	  "${GITLAB_API_URL}/user/keys")
	export ssh_key_id=$(jq .id <<< "$ssh_key_json")
fi

git config --global init.defaultBranch main
ssh-keyscan -p ${GIT_SSH_PORT} -H ${CI_SERVER_HOST} | install -m 600 /dev/stdin $HOME/.ssh/known_hosts

# example
# /usr/bin/storescp \
#  -aet DICOM_SERVER_SEQUOIA\
#  -pm\
#  -od $DICOM_TMP_DIR -su ''\
#  --eostudy-timeout ${STORESCP_STUDY_TIMEOUT:=60} \
#  --exec-on-eostudy "python3 $DICOM_ROOT/exec_on_study_received.py #p " 2100 >> $DICOM_DATA_ROOT/storescp.log

# run whatever command was passed (storescp or python index_dicoms directly)
$@

if [ -n "${GITLAB_TOKEN}" ] ; then
	# unregister the temporary ssh key
	curl -X DELETE -F "private_token=${GITLAB_TOKEN}" "${GITLAB_API_URL}/user/keys/${ssh_key_id}"
fi