From b586b4b2a77c626249185dd447aa0051fddcbfa4 Mon Sep 17 00:00:00 2001
From: Jin Kang <jinkang23@users.noreply.github.com>
Date: Tue, 17 Jan 2023 05:45:45 -0600
Subject: [PATCH] Updated script to support UDM/UDMPro 2.x firmware (#451)

* Updated script to support UDM/UDMPro 2.x firmware while preserving legacy UDM/UDMPro 1.x firmware

* Updated script to make the message clearer of which device and version was detected
---
 on-boot-script/remote_install.sh | 22 ++++++++++++++++------
 1 file changed, 16 insertions(+), 6 deletions(-)

diff --git a/on-boot-script/remote_install.sh b/on-boot-script/remote_install.sh
index 62de92c..0a15ca4 100755
--- a/on-boot-script/remote_install.sh
+++ b/on-boot-script/remote_install.sh
@@ -61,10 +61,18 @@ udm_model() {
       echo "udmse"
       ;;
     "UniFi Dream Machine Pro")
-      echo "udmpro"
+      if test $(ubnt-device-info firmware) \< "2.0.0"; then 
+        echo "udmprolegacy"
+      else 
+        echo "udmpro"
+      fi
       ;;
     "UniFi Dream Machine")
-      echo "udm"
+      if test $(ubnt-device-info firmware) \< "2.0.0"; then 
+        echo "udmlegacy"
+      else 
+        echo "udm"
+      fi
       ;;
     "UniFi Dream Router")
       echo "udr"
@@ -160,8 +168,9 @@ depends_on curl
 ON_BOOT_D_PATH="$DATA_DIR/on_boot.d"
 
 case "$(udm_model)" in
-  udm|udmpro)
-    echo "UDM/Pro detected, installing on-boot script..."
+  udmlegacy|udmprolegacy)
+    echo "$(ubnt-device-info model) version $(ubnt-device-info firmware) was detected"
+    echo "Installing on-boot script..."
     depends_on podman
 
     if ! install_on_boot_udm_series; then
@@ -172,8 +181,9 @@ case "$(udm_model)" in
 
     echo "UDM Boot Script installed"
     ;;
-  udr|udmse)
-    echo "UDR/UDMSE detected, installing on-boot script..."
+  udr|udmse|udm|udmpro)
+    echo "$(ubnt-device-info model) version $(ubnt-device-info firmware) was detected"
+    echo "Installing on-boot script..."
     depends_on systemctl
 
     if ! install_on_boot_udr_se; then