From bb48fc2c5d8a0b95647782743b9eda564f77dce7 Mon Sep 17 00:00:00 2001
From: simonzhangsz <simonzhangsz@outlook.com>
Date: Sat, 20 Jan 2024 14:39:53 +0800
Subject: [PATCH] fix: rollback software package

---
 misc/rbkpkg.sh | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/misc/rbkpkg.sh b/misc/rbkpkg.sh
index 0b432f68..f8c9868b 100644
--- a/misc/rbkpkg.sh
+++ b/misc/rbkpkg.sh
@@ -2,8 +2,15 @@
 
 filename="$1"
 neType="$2"
+duration=180
+expectBin=/usr/bin/expect
 extension=$(basename "${1}" | awk -F . '{print $NF}')
 
+if [ ! -f ${expectBin} ]; then
+    echo "Please install software expect first"
+    exit 99
+fi
+
 case "${extension}" in
     deb)
         if [ ${neType} == "OMC" ]; then
@@ -11,7 +18,7 @@ case "${extension}" in
         fi    
         expect <<EOF
             spawn dpkg -i --force-all "$filename"
-            set timeout 180
+            set timeout ${duration}
             expect {
                 "y/n" { send "n\n"; exp_continue }
             }
@@ -23,7 +30,7 @@ EOF
         fi    
         expect <<EOF
             spawn rpm -Uvh --oldpackage "$filename"
-            set timeout 180
+            set timeout ${duration}
             expect {
                 "y/n" { send "n\n"; exp_continue }
             }
@@ -32,6 +39,7 @@ EOF
    *)
 
         echo "The file does not have a .deb or .rpm extension"
-        exit 1                 
+        exit 100                 
         ;;
-esac      
+esac
+