From 39a769db72111c953383e3d7823f0f27b205dd0c Mon Sep 17 00:00:00 2001
From: 21pages <pages21@163.com>
Date: Mon, 27 Nov 2023 09:36:48 +0800
Subject: [PATCH] use conf.d if sites-available and sites-enabled don't exist

Signed-off-by: 21pages <pages21@163.com>
---
 install.sh | 19 +++++++++++++++----
 restore.sh | 13 +++++++++++--
 2 files changed, 26 insertions(+), 6 deletions(-)

diff --git a/install.sh b/install.sh
index c0c14bc..8a11018 100644
--- a/install.sh
+++ b/install.sh
@@ -401,10 +401,21 @@ Do you want to install Certbot with snap? (recommended)"
         fi
 
         # Add Nginx config
-        if [ ! -f "/etc/nginx/sites-available/rustdesk.conf" ]
+        if [ -d "/etc/nginx/sites-available" ] && [ -d "/etc/nginx/sites-enabled" ]
         then
-            touch "/etc/nginx/sites-available/rustdesk.conf"
-            cat << NGINX_RUSTDESK_CONF > "/etc/nginx/sites-available/rustdesk.conf"
+            SITES_CONF_DIR="sites-available"
+        elif [ -d "/etc/nginx/conf.d" ]
+        then
+            SITES_CONF_DIR="conf.d"
+        else
+            msg_box "Couldn't find the Nginx config directory. Please check your system!"
+            exit 1
+        fi
+
+        if [ ! -f "/etc/nginx/$SITES_CONF_DIR/rustdesk.conf" ]
+        then
+            touch "/etc/nginx/$SITES_CONF_DIR/rustdesk.conf"
+            cat << NGINX_RUSTDESK_CONF > "/etc/nginx/$SITES_CONF_DIR/rustdesk.conf"
 server {
   server_name ${RUSTDESK_DOMAIN};
       location / {
@@ -417,7 +428,7 @@ NGINX_RUSTDESK_CONF
         fi
 
         # Enable the Nginx config file
-        if [ ! -f /etc/nginx/sites-enabled/rustdesk.conf ]
+        if [ "$SITES_CONF_DIR" = "sites-available" ] && [ ! -f /etc/nginx/sites-enabled/rustdesk.conf ]
         then
             ln -s /etc/nginx/sites-available/rustdesk.conf /etc/nginx/sites-enabled/rustdesk.conf
         fi
diff --git a/restore.sh b/restore.sh
index f8c8d43..3c1c3cb 100644
--- a/restore.sh
+++ b/restore.sh
@@ -272,9 +272,18 @@ server {
 }
 EOF
 )"
-echo "${rustdesknginx}" | sudo tee /etc/nginx/sites-available/rustdesk.conf >/dev/null
 
-sudo ln -s /etc/nginx/sites-available/rustdesk.conf /etc/nginx/sites-enabled/rustdesk.conf
+if [ -d "/etc/nginx/sites-available" ] && [ -d "/etc/nginx/sites-enabled" ]
+then
+    echo "${rustdesknginx}" | sudo tee /etc/nginx/sites-available/rustdesk.conf >/dev/null
+    sudo ln -s /etc/nginx/sites-available/rustdesk.conf /etc/nginx/sites-enabled/rustdesk.conf
+elif [ -d "/etc/nginx/conf.d" ]
+then
+    echo "${rustdesknginx}" | sudo tee /etc/nginx/conf.d/rustdesk.conf >/dev/null
+else
+    msg_box "Couldn't find the Nginx config directory. Please check your system!"
+    exit 1
+fi
 
 sudo ufw allow 80/tcp
 sudo ufw allow 443/tcp