From 86522c10d912ecfb29a5cbe52054a436823fa7a2 Mon Sep 17 00:00:00 2001
From: Michael Feng <mifeng@gmail.com>
Date: Fri, 1 Sep 2023 09:53:35 -0700
Subject: [PATCH] (fix) add pem check

---
 gateway-setup.sh | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/gateway-setup.sh b/gateway-setup.sh
index 41ec6e5e27..a42293ae0a 100755
--- a/gateway-setup.sh
+++ b/gateway-setup.sh
@@ -56,19 +56,28 @@ copy_certs () {
   read -p "Enter path to the Hummingbot certs folder >>> " CERTS_FROM_PATH
   if [ ! -d "$CERTS_FROM_PATH" ]; then
     echo "Error: $CERTS_FROM_PATH does not exist or is not a directory"
-    exit
+    exit 1
+  fi
+
+  # Check if there are any .pem files in the source directory
+  PEM_COUNT=$(find "$CERTS_FROM_PATH" -maxdepth 1 -name "*.pem" | wc -l)
+  if [ "$PEM_COUNT" -eq 0 ]; then
+    echo "Error: No .pem files found in $CERTS_FROM_PATH"
+    exit 1
   fi
- 
+
   # Make destination folder if needed
-  mkdir $CERTS_TO_PATH
+  mkdir -p $CERTS_TO_PATH
+  
   # Copy all files in the source folder to the destination folder
   cp -r $CERTS_FROM_PATH/* $CERTS_TO_PATH/
+  
   # Confirm that the files were copied
   if [ $? -eq 0 ]; then
     echo "Files successfully copied from $CERTS_FROM_PATH to $CERTS_TO_PATH"
   else
     echo "Error copying files from $CERTS_FROM_PATH to $CERTS_TO_PATH"
-    exit
+    exit 1
   fi
 }