From e2525f23fb085fa7549ab7c4b11da15a7c4dad50 Mon Sep 17 00:00:00 2001
From: Dan <dan@noreply.localhost>
Date: Thu, 16 Apr 2026 09:34:06 +0800
Subject: [PATCH] Update mermaids.sh

---
 mermaids.sh | 23 ++++++++++++++---------
 1 file changed, 14 insertions(+), 9 deletions(-)

diff --git a/mermaids.sh b/mermaids.sh
index b2897bd..4b1b500 100644
--- a/mermaids.sh
+++ b/mermaids.sh
@@ -129,15 +129,20 @@ log() { echo -e "${GREEN}[$(date +%H:%M:%S)]${NC} $1"; }
 PORT="${1:-80}"
 
 # 1. Update & install Caddy + deps
-log "Updating system..."
-apt-get clean -y
-rm -rf /var/lib/apt/lists/*
-apt-get update -o Acquire::Retries=3 || {
-  warn "First apt-get update failed. Checking network..."
-  cat /etc/resolv.conf
-  exit 1
-}
-apt-get install -y -qq wget ufw
+log "Updating package lists..."
+# Enable non-free for ufw (Debian 12 requirement)
+if grep -q "deb debianmain" /etc/apt/sources.list; then
+  sed -i 's/deb \(.*\) main/deb \1 main non-free non-free-firmware/' /etc/apt/sources.list
+fi
+apt-get update -o Acquire::Retries=3
+
+log "Installing dependencies..."
+apt-get install -y -qq \
+  wget \
+  ufw \
+  curl \
+  apt-transport-https \
+  ca-certificates
 
 # 2. Download Mermaid (offline bundled)
 log "Downloading Mermaid v10.6.1 (offline)..."