From 95003e88e63ff6a9aa5e4648ba86d5995cd4e6c6 Mon Sep 17 00:00:00 2001
From: Howard20181 <40033067+Howard20181@users.noreply.github.com>
Date: Tue, 23 Aug 2022 14:03:56 +0800
Subject: [PATCH] Handle unexpected exit

---
 scripts/run.sh | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/scripts/run.sh b/scripts/run.sh
index 0c95d48..cdbe450 100755
--- a/scripts/run.sh
+++ b/scripts/run.sh
@@ -1,6 +1,6 @@
 #!/bin/bash
-
-WORK_DIR=$(mktemp -d -t wsa-build-XXXXXXXXXX_)
+trap 'rm -rf -- "$WORK_DIR"' EXIT
+WORK_DIR=$(mktemp -d -t wsa-build-XXXXXXXXXX_) || exit 1
 DOWNLOAD_DIR=../download
 OUTPUT_DIR=../output
 MOUNT_DIR="$WORK_DIR"/system
@@ -34,6 +34,7 @@ abort() {
     fi
     exit 1
 }
+trap abort EXIT
 
 if [ ! "$BASH_VERSION" ] ; then
     echo "Please do not use sh to run this script, just execute it directly" 1>&2