From df41bff4fee80eddea2a25b9e96970a8d4d0af8f Mon Sep 17 00:00:00 2001
From: Gerd Naschenweng <magicdude4eva@users.noreply.github.com>
Date: Sun, 18 Dec 2022 11:37:41 +0100
Subject: [PATCH] Supporting new firmwares

Supporting new data-directories for new firmwares. Tested un UDM Pro 2.4.23
---
 install.sh | 26 ++++++++++++++++++++++----
 1 file changed, 22 insertions(+), 4 deletions(-)

diff --git a/install.sh b/install.sh
index 6407a58..9f621be 100644
--- a/install.sh
+++ b/install.sh
@@ -1,15 +1,33 @@
 #!/bin/sh
 
-user_authorized_keys_file="/mnt/data/ssh/authorized_keys"
+# Get DataDir location
+DATA_DIR="/mnt/data"
+case "$(ubnt-device-info firmware || true)" in
+    1*)
+      DATA_DIR="/mnt/data"
+      ;;
+    2*)
+      DATA_DIR="/data"
+      ;;
+    3*)
+      DATA_DIR="/data"
+      ;;
+    *)
+      echo "ERROR: No persistent storage found." 1>&2
+      exit 1
+      ;;
+  esac
+
+user_authorized_keys_file="$DATA_DIR/ssh/authorized_keys"
 
 curl -sO https://raw.githubusercontent.com/fire1ce/UDM-Persistent-SSH-Keys/main/99-ssh-keys.sh
-mv 99-ssh-keys.sh /mnt/data/on_boot.d/99-ssh-keys.sh
-chmod +x /mnt/data/on_boot.d/99-ssh-keys.sh
+mv 99-ssh-keys.sh $DATA_DIR/on_boot.d/99-ssh-keys.sh
+chmod +x $DATA_DIR/on_boot.d/99-ssh-keys.sh
 echo "99-ssh-keys.sh installed"
 
 if [ ! -f "$user_authorized_keys_file" ]; then
   echo "Creating $user_authorized_keys_file"
-  mkdir -p /mnt/data/ssh
+  mkdir -p $DATA_DIR/ssh
   touch "$user_authorized_keys_file"
   chmod 0644 "$user_authorized_keys_file"
 fi