imp: integrate with dstack-util notify-host

Author: Leechael

phala-cloud-prelaunch-script/prelaunch.sh

@@ -1,15 +1,34 @@
 #!/bin/bash
 echo "----------------------------------------------"
-echo "Running Phala Cloud Pre-Launch Script v0.0.4"
+echo "Running Phala Cloud Pre-Launch Script v0.0.5"
 echo "----------------------------------------------"
 set -e
 
+# Function: notify host
+
+notify_host() {
+    if command -v dstack-util >/dev/null 2>&1; then
+        dstack-util notify-host -e "$1" -d "$2"
+    else
+        tdxctl notify-host -e "$1" -d "$2"
+    fi
+}
+
+notify_host_hoot_info() {
+    notify_host "boot.progress" "$1"
+}
+
+notify_host_hoot_error() {
+    notify_host "boot.error" "$1"
+}
+
 # Function: Perform Docker cleanup
 perform_cleanup() {
     echo "Pruning unused images"
     docker image prune -af
     echo "Pruning unused volumes"
     docker volume prune -f
+    notify_host_hoot_info "docker cleanup completed"
 }
 
 # Function: Check Docker login status without exposing credentials
@@ -22,16 +41,6 @@ check_docker_login() {
     fi
 }
 
-# Function: Check AWS ECR login status
-check_ecr_login() {
-    # Check if we can access the registry without exposing credentials
-    if aws ecr get-authorization-token --region $DSTACK_AWS_REGION &>/dev/null; then
-        return 0
-    else
-        return 1
-    fi
-}
-
 # Main logic starts here
 echo "Starting login process..."
 
@@ -55,6 +64,7 @@ if [[ -n "$DSTACK_DOCKER_USERNAME" && -n "$DSTACK_DOCKER_PASSWORD" ]]; then
             echo "Docker login successful"
         else
             echo "Docker login failed"
+            notify_host_hoot_error "docker login failed"
             exit 1
         fi
     fi
@@ -64,11 +74,13 @@ elif [[ -n "$DSTACK_AWS_ACCESS_KEY_ID" && -n "$DSTACK_AWS_SECRET_ACCESS_KEY" &&
 
     # Check if AWS CLI is installed
     if ! command -v aws &> /dev/null; then
+        notify_host_hoot_info "awscli not installed, installing..."
         echo "AWS CLI not installed, installing..."
         curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64-2.24.14.zip" -o "awscliv2.zip"
         echo "6ff031a26df7daebbfa3ccddc9af1450 awscliv2.zip" | md5sum -c
         if [ $? -ne 0 ]; then
             echo "MD5 checksum failed"
+            notify_host_hoot_error "awscli install failed"
             exit 1
         fi
         unzip awscliv2.zip &> /dev/null
@@ -95,15 +107,18 @@ elif [[ -n "$DSTACK_AWS_ACCESS_KEY_ID" && -n "$DSTACK_AWS_SECRET_ACCESS_KEY" &&
     echo "Testing AWS credentials..."
     if ! aws sts get-caller-identity &> /dev/null; then
         echo "AWS credentials test failed"
+        notify_host_hoot_error "Invalid AWS credentials"
         exit 1
     fi
 
     echo "Logging in to AWS ECR..."
     aws ecr get-login-password --region $DSTACK_AWS_REGION | docker login --username AWS --password-stdin "$DSTACK_AWS_ECR_REGISTRY"
     if [ $? -eq 0 ]; then
         echo "AWS ECR login successful"
+        notify_host_hoot_info "AWS ECR login successful"
     else
         echo "AWS ECR login failed"
+        notify_host_hoot_error "AWS ECR login failed"
         exit 1
     fi
 fi