feat(pg-node): prompt user before overwriting existing installation in restore.sh

Author: s7net

pg-node/restore.sh

@@ -219,29 +219,66 @@ resolve_backup_url() {
     fi
 }
 
+FORCE_CONFIRM=0
+
 parse_args() {
-    while getopts ":b:p:h" opt; do
+    while getopts ":b:p:yfh" opt; do
         case "${opt}" in
             b) BACKUP_URL="${OPTARG}" ;;
             p) BACKUP_PASS="${OPTARG}" ;;
+            y|f) FORCE_CONFIRM=1 ;;
             h)
-                echo "Usage: sudo $0 -b <backup-url> [-p <backup-password>]"
+                echo "Usage: sudo $0 -b <backup-url> [-p <backup-password>] [-y]"
                 exit 0
                 ;;
             \?)
                 log_err "Invalid option: -${OPTARG}"
-                echo "Usage: sudo $0 -b <backup-url> [-p <backup-password>]"
+                echo "Usage: sudo $0 -b <backup-url> [-p <backup-password>] [-y]"
                 exit 1
                 ;;
             :)
                 log_err "Option -${OPTARG} requires an argument."
-                echo "Usage: sudo $0 -b <backup-url> [-p <backup-password>]"
+                echo "Usage: sudo $0 -b <backup-url> [-p <backup-password>] [-y]"
                 exit 1
                 ;;
         esac
     done
 }
 
+check_existing_installation() {
+    if [[ -d "${APP_DIR}" ]] || [[ -d "${DATA_DIR}" ]]; then
+        log_warn "An existing pg-node installation was detected (${APP_DIR})."
+
+        if [[ "${FORCE_CONFIRM}" -eq 1 ]]; then
+            log_info "Non-interactive confirmation flag (-y/-f) set. Proceeding with overwrite..."
+        elif [[ -t 0 ]]; then
+            echo -e "${COLOR_YELLOW}[WARNING] An existing pg-node installation was found on this server.${COLOR_RESET}"
+            read -r -p "$(echo -e "${COLOR_BLUE}[INPUT]${COLOR_RESET} Do you want to remove existing pg-node and restore this backup? [y/N]: ")" choice
+            case "${choice}" in
+                [yY][eE][sS]|[yY])
+                    log_info "User confirmed. Cleaning up existing installation..."
+                    ;;
+                *)
+                    log_info "Restore cancelled by user."
+                    exit 0
+                    ;;
+            esac
+        else
+            log_err "Existing pg-node installation found. Run with -y to force overwrite in non-interactive mode."
+            exit 1
+        fi
+
+        if [[ -f "${APP_DIR}/docker-compose.yml" ]] && command -v docker &>/dev/null && docker compose version &>/dev/null; then
+            log_info "Stopping existing pg-node stack..."
+            (cd "${APP_DIR}" && docker compose down) 2>/dev/null || true
+        fi
+
+        log_info "Removing existing application and data directories..."
+        rm -rf "${APP_DIR}" "${DATA_DIR}"
+        log_ok "Existing installation cleaned up."
+    fi
+}
+
 DOWNLOADER=""
 detect_downloader() {
     if command -v curl &>/dev/null; then
@@ -430,10 +467,11 @@ main() {
     detect_distro
     repair_broken_packages
     ensure_prerequisites
+    resolve_backup_url
+    check_existing_installation
     install_pg_node_official
     stop_default_stack
     detect_downloader
-    resolve_backup_url
     download_backup
     extract_backup
     start_stack