Update Dockerfile

Dockerfile

@@ -18,7 +18,7 @@ RUN apt-get update && \
         nano \
     && rm -rf /var/lib/apt/lists/*
 
-# ── Install bore (TCP tunnel — exposes SSH publicly for Termius) ──────────────
+# ── Install bore ──────────────────────────────────────────────────────────────
 RUN wget -q "https://github.com/ekzhang/bore/releases/download/v0.5.0/bore-v0.5.0-x86_64-unknown-linux-musl.tar.gz" \
         -O /tmp/bore.tar.gz && \
     tar -xzf /tmp/bore.tar.gz -C /usr/local/bin bore && \
@@ -37,7 +37,7 @@ ENV HOME=/root \
     VIRTUAL_ENV=/root/venv \
     PIP_NO_CACHE_DIR=1
 
-# ── Create a persistent Python venv ──────────────────────────────────────────
+# ── Create Python venv + install packages (will be persisted via symlink) ─────
 RUN python3 -m venv /root/venv && \
     /root/venv/bin/pip install --upgrade pip && \
     /root/venv/bin/pip install huggingface_hub
@@ -46,7 +46,7 @@ RUN python3 -m venv /root/venv && \
 RUN npm config set prefix /root/.npm-global && \
     npm install -g shellular
 
-# ── Create /data and ensure all shells land there ────────────────────────────
+# ── Ensure /data exists and shells land there ─────────────────────────────────
 RUN mkdir -p /data && \
     echo 'cd /data' >> /root/.bash_profile && \
     echo 'cd /data' >> /root/.bashrc
@@ -56,9 +56,41 @@ COPY package*.json /root/app/
 RUN cd /root/app && npm install --omit=dev
 COPY . /root/app/
 
-# ── Entrypoint — no hostname call (already baked into /etc/hostname) ──────────
-RUN printf '#!/bin/sh\nexec "$@"\n' > /usr/local/bin/docker-entrypoint.sh && \
-    chmod +x /usr/local/bin/docker-entrypoint.sh
+# ── Entrypoint ────────────────────────────────────────────────────────────────
+RUN cat > /usr/local/bin/docker-entrypoint.sh << 'EOF'
+#!/bin/sh
+set -e
+
+# Usage: persist <image_path> <storage_path>
+# - First boot: moves image_path into persistent storage, then symlinks back
+# - Every boot: ensures symlink exists (in case image_path reappeared)
+persist() {
+    SRC="$1"          # original path in image  e.g. /root/venv
+    DEST="$2"         # where to keep it in /data e.g. /data/persist/venv
+
+    mkdir -p "$(dirname "$DEST")"
+
+    if [ ! -e "$DEST" ]; then
+        echo "[persist] First boot — moving $SRC → $DEST"
+        cp -a "$SRC" "$DEST"
+    else
+        echo "[persist] Reusing $DEST"
+    fi
+
+    # Remove original (could be a real dir from the image layer) and symlink
+    rm -rf "$SRC"
+    ln -sf "$DEST" "$SRC"
+    echo "[persist] Symlinked $SRC → $DEST"
+}
+
+# ── Directories to persist ────────────────────────────────────────────────────
+persist /root/venv          /data/persist/venv
+persist /root/.cache        /data/persist/cache
+persist /root/.npm-global   /data/persist/npm-global
+
+exec "$@"
+EOF
+chmod +x /usr/local/bin/docker-entrypoint.sh
 
 # ── Runtime ───────────────────────────────────────────────────────────────────
 WORKDIR /data