Merge pull request #15 from Dstack-TEE/prelanch-script

Add example prelaunch-script

Author: amiller

prelaunch-script/README.md

@@ -0,0 +1,3 @@
+# DStack Pre-launch Script Example
+
+This directory provides an example of a pre-launch script for the DStack Application. Introduced in Dstack v0.3.5 (as detailed in [#94](https://github.com/Dstack-TEE/dstack/pull/94)), this feature allows the application to perform preliminary setup steps before initiating Docker Compose. The pre-launch script's content is specified in the `pre_launch_script` section of the `app-compose.json` file. The `prelaunch.sh` script demonstrates how to manage container initialization and configure the environment prior to launching your application.

prelaunch-script/docker-compose.yaml

@@ -0,0 +1,7 @@
+services:
+  busybox:
+    image: busybox:latest
+    command: sh -c "cat /etc/motd"
+    volumes:
+      - /tapp/motd:/etc/motd
+    restart: no

prelaunch-script/prelaunch.sh

@@ -0,0 +1,46 @@
+# This is an example of how to write a pre-launch script of DStack App Compose.
+
+# The script is run in /tapp directory. The app-compose.json file is in the same directory.
+
+set -e
+
+# We fully handle the docker compose logic in this script.
+echo "Extracting docker compose file"
+jq -j '.docker_compose_file' app-compose.json >docker-compose.yaml
+echo "Removing orphans"
+tdxctl remove-orphans -f docker-compose.yaml || true
+echo "Restarting docker"
+chmod +x /usr/bin/containerd-shim-runc-v2
+systemctl restart docker
+
+# Login docker account
+echo "Logging into Docker Hub"
+tdxctl notify-host -e "boot.progress" -d "logging into docker hub" || true
+if [ -n "$DOCKER_USERNAME" ] && [ -n "$DOCKER_PASSWORD" ]; then
+    if ! echo "$DOCKER_PASSWORD" | docker login -u "$DOCKER_USERNAME" --password-stdin; then
+        tdxctl notify-host -e "boot.error" -d "failed to login to docker hub"
+        exit 1
+    fi
+fi
+
+# Use a container to setup the environment
+echo "Setting up the environment"
+tdxctl notify-host -e "boot.progress" -d "setting up the environment" || true
+docker run \
+    --rm \
+    --name dstack-app-setup \
+    -v /tapp:/tapp \
+    -w /tapp \
+    -v /var/run/docker.sock:/var/run/docker.sock \
+    curlimages/curl:latest \
+    -s https://raw.githubusercontent.com/Dstack-TEE/meta-dstack/refs/heads/main/meta-dstack/recipes-core/base-files/files/motd -o /tapp/motd
+
+echo "Starting containers"
+tdxctl notify-host -e "boot.progress" -d "starting containers" || true
+if ! docker compose up -d; then
+    tdxctl notify-host -e "boot.error" -d "failed to start containers"
+    exit 1
+fi
+
+# Use exit to skip the original docker compose handling
+exit 0