From a1f8ddd8aa4c107998cb8c5e749baea249493b69 Mon Sep 17 00:00:00 2001
From: Alexandre <44178713+alexbelgium@users.noreply.github.com>
Date: Sun, 8 Oct 2023 12:02:28 +0200
Subject: [PATCH] Improve code

---
 omada/rootfs/etc/cont-init.d/00-folders.sh | 18 +++++++-----------
 1 file changed, 7 insertions(+), 11 deletions(-)

diff --git a/omada/rootfs/etc/cont-init.d/00-folders.sh b/omada/rootfs/etc/cont-init.d/00-folders.sh
index 4cce0941d..2950c4ea6 100755
--- a/omada/rootfs/etc/cont-init.d/00-folders.sh
+++ b/omada/rootfs/etc/cont-init.d/00-folders.sh
@@ -21,19 +21,15 @@ fi
 cp -rnf /opt/tplink/EAPController/data/* "$CONFIGSOURCE"
 
 echo "Creating symlink"
+# Clean existing folder
 rm -r /opt/tplink/EAPController/data/*
 
-mkdir -p "$CONFIGSOURCE"/pdf
-mkdir -p "$CONFIGSOURCE"/omada/html
-mkdir -p "$CONFIGSOURCE"/db
-mkdir -p "$CONFIGSOURCE"/map
-mkdir -p "$CONFIGSOURCE"/portal
-
-ln -s "$CONFIGSOURCE"/pdf /opt/tplink/EAPController/data
-ln -s "$CONFIGSOURCE"/omada/html /opt/tplink/EAPController/data
-ln -s "$CONFIGSOURCE"/db /opt/tplink/EAPController/data
-ln -s "$CONFIGSOURCE"/map /opt/tplink/EAPController/data
-ln -s "$CONFIGSOURCE"/portal /opt/tplink/EAPController/data
+# Create symlinks for all files in /data
+for files in $(ls -d "$CONFIGSOURCE"/*); do
+    if [ -e "$files" ]; then
+        ln -s "$files" /opt/tplink/EAPController/data
+    fi
+done
 
 # Make sure permissions are right
 echo "Updating permissions"