41 lines
1.4 KiB
Bash
41 lines
1.4 KiB
Bash
#!/bin/bash
|
|
|
|
CONTAINER_ID=$(basename $(cat /proc/1/cpuset))
|
|
GITLAB_TOKEN_SECRET=$(cat /var/run/secrets/dicom_bot_gitlab_token 2>/dev/null)
|
|
GITLAB_TOKEN=${GITLAB_TOKEN_SECRET:=$GITLAB_TOKEN}
|
|
|
|
# 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'
|
|
ssh_key_json=$(curl -X POST -F "private_token=${GITLAB_TOKEN}" \
|
|
-F "title="$(cat /etc/hostname)${CONTAINER_ID:0:12} -F "key=$(cat ~/.ssh/id_rsa.pub)" \
|
|
"${GITLAB_API_URL}/user/keys")
|
|
fi
|
|
|
|
git config --global init.defaultBranch main
|
|
mkdir -p ~/.ssh/known_hosts
|
|
install -m 600 /dev/stdin ~/.ssh/known_hosts <<< "$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
|
|
ssh_key_id=$(jq .id <<< $ssh_key_json)
|
|
curl -X DELETE -F "private_token=${GITLAB_TOKEN}" \
|
|
-F "title="$(cat /etc/hostname)${CONTAINER_ID:0:12}
|
|
"${GITLAB_API_URL}/users/keys/${ssh_key_id}"
|
|
fi
|