From 47b978785b7828f8e7ab96227c13b49efac71e43 Mon Sep 17 00:00:00 2001
From: Muhammad Atif Ali <me@matifali.dev>
Date: Tue, 26 Sep 2023 05:14:46 +0300
Subject: [PATCH] update

---
 kasmvnc/run.sh | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/kasmvnc/run.sh b/kasmvnc/run.sh
index 882d807..0f88f89 100644
--- a/kasmvnc/run.sh
+++ b/kasmvnc/run.sh
@@ -22,12 +22,23 @@ else
     echo "${DESKTOP_ENVIRONMENT} is already installed."
 fi
 
+# Make sure lsb-release is installed
+sudo apt-get install -y lsb-release
+
+# Fetch Ubuntu distribution codename
+codename=$(lsb_release -cs)
+
 # Check if vncserver is installed
 if ! dpkg -s kasmvncserver &>/dev/null; then
     cd /tmp
-    wget https://github.com/kasmtech/KasmVNC/releases/download/v${VERSION}/kasmvncserver_jammy_${VERSION}_amd64.deb
-    sudo apt-get install -y ./kasmvncserver_focal_${VERSION}_amd64.deb
-    printf "🥳 KasmVNC v${VERSION} has been successfully installed!\n\n"
+    wget "https://github.com/kasmtech/KasmVNC/releases/download/v${VERSION}/kasmvncserver_${codename}_${VERSION}_amd64.deb"
+    sudo apt-get install -y ./kasmvncserver_$codename_${VERSION}_amd64.deb
+    if [ $? -eq 0 ]; then
+        printf "🥳 KasmVNC v${VERSION} has been successfully installed!\n\n"
+    else
+        echo "Failed to install KasmVNC."
+        exit 1
+    fi
 else
     echo "KasmVNC is already installed."
 fi