[Docker] Load env from tags when using AWS ECS

Author: Herklos

Dockerfile

@@ -30,21 +30,21 @@ COPY --from=base /opt/venv /opt/venv
 # Add default config files
 COPY octobot/config /octobot/octobot/config
 
-COPY docker-entrypoint.sh docker-entrypoint.sh
+COPY docker/*.sh /octobot/
 
 # 1. Install requirements
 # 2. Add cloudflare gpg key and add cloudflare repo in apt repositories (from https://pkg.cloudflare.com/index.html)
 # 3. Install required packages
 # 4. Finish env setup
 SHELL ["/bin/bash", "-o", "pipefail", "-c"]
-RUN apt-get update \ 
+RUN apt-get update \
     && apt-get install -y --no-install-recommends curl \
-    && mkdir -p /usr/share/keyrings \ 
+    && mkdir -p /usr/share/keyrings \
     && chmod 0755 /usr/share/keyrings \
     && curl -fsSL https://pkg.cloudflare.com/cloudflare-main.gpg | tee /usr/share/keyrings/cloudflare-main.gpg >/dev/null \ 
     && echo 'deb [signed-by=/usr/share/keyrings/cloudflare-main.gpg] https://pkg.cloudflare.com/cloudflared buster main' | tee /etc/apt/sources.list.d/cloudflared.list \
     && apt-get update \ 
-    && apt-get install -y --no-install-recommends curl cloudflared libxslt-dev libxcb-xinput0 libjpeg62-turbo-dev zlib1g-dev libblas-dev liblapack-dev libatlas-base-dev libopenjp2-7 libtiff-dev \
+    && apt-get install -y --no-install-recommends jq curl cloudflared libxslt-dev libxcb-xinput0 libjpeg62-turbo-dev zlib1g-dev libblas-dev liblapack-dev libatlas-base-dev libopenjp2-7 libtiff-dev \
     && rm -rf /var/lib/apt/lists/* \
     && ln -s /opt/venv/bin/OctoBot OctoBot # Make sure we use the virtualenv \
     && chmod +x docker-entrypoint.sh

docker/aws.sh

@@ -0,0 +1,10 @@
+#!/bin/bash
+
+source util.sh
+
+# https://docs.aws.amazon.com/AmazonECS/latest/developerguide/task-metadata-endpoint-v4.html
+if [[ -n "$ECS_CONTAINER_METADATA_URI_V4" ]]; then
+  AWS_TASK_WITH_TAGS_DETAILS=$(curl --silent "$ECS_CONTAINER_METADATA_URI_V4/taskWithTags")
+  AWS_TASK_TAGS=$(echo $AWS_TASK_WITH_TAGS_DETAILS | jq -r ".TaskTags")
+  json_to_env "$AWS_TASK_TAGS"
+fi

docker/docker-entrypoint.sh

@@ -0,0 +1,11 @@
+#!/bin/bash
+
+##### OctoBot config #####
+if [[ -n "${OCTOBOT_CONFIG}" ]]; then
+  echo "$OCTOBOT_CONFIG" | tee /octobot/user/config.json >/dev/null
+fi
+
+bash aws.sh
+bash tunnel.sh
+
+./OctoBot

docker-entrypoint.sh renamed to docker/tunnel.sh

@@ -1,17 +1,10 @@
 #!/bin/bash
 
-# OctoBot user config from env
-if [[ -n "${OCTOBOT_CONFIG}" ]]; then
-  echo "$OCTOBOT_CONFIG" | tee /octobot/user/config.json > /dev/null
-fi
-
-##### CLOUDFLARED #####
 export TUNNEL_LOGFILE=/root/cloudflared.log
 export TUNNEL_LOGLEVEL=info
 
 # start cloudflared if token is provided
+# https://developers.cloudflare.com/cloudflare-one/tutorials/cli/
 if [[ -n "$CLOUDFLARE_TOKEN" ]]; then
   cloudflared tunnel --url http://localhost:5001 --no-autoupdate run --token $CLOUDFLARE_TOKEN &
 fi
-
-./OctoBot

docker/util.sh

@@ -0,0 +1,5 @@
+#!/bin/bash
+
+json_to_env() {
+  for s in $(echo $1 | jq -r 'keys[] as $k | "export \($k)=\(.[$k])"'); do export $s; done
+}