From 701d3f10d91d1aa34ed392b30c8cd985a0096ca2 Mon Sep 17 00:00:00 2001
From: Godopu <godopu16@gmail.com>
Date: Wed, 24 Jun 2026 12:24:17 +0900
Subject: [PATCH] Add MAM updater script (update.sh) and integrate into
 installer copy block

---
 deploy/install.sh |   6 ++
 deploy/update.sh  | 137 ++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 143 insertions(+)
 create mode 100644 deploy/update.sh

diff --git a/deploy/install.sh b/deploy/install.sh
index 2371a0e..348e052 100644
--- a/deploy/install.sh
+++ b/deploy/install.sh
@@ -141,6 +141,12 @@ if ! check_assets_present "."; then
     echo "remove.sh" >> "$MANIFEST_FILE"
   fi
 
+  if [ -f "$STAGE_DIR/deploy/update.sh" ] && [ ! -e "update.sh" ]; then
+    cp "$STAGE_DIR/deploy/update.sh" update.sh || { echo "❌ Error: Failed to copy update.sh" >&2; exit 1; }
+    chmod +x update.sh
+    echo "update.sh" >> "$MANIFEST_FILE"
+  fi
+
   if [ -f "$STAGE_DIR/.env.example" ] && [ ! -e ".env.example" ]; then
     cp "$STAGE_DIR/.env.example" . || { echo "❌ Error: Failed to copy .env.example" >&2; exit 1; }
     echo ".env.example" >> "$MANIFEST_FILE"
diff --git a/deploy/update.sh b/deploy/update.sh
new file mode 100644
index 0000000..a265efd
--- /dev/null
+++ b/deploy/update.sh
@@ -0,0 +1,137 @@
+#!/usr/bin/env bash
+# ==============================================================================
+# update.sh — Multi-Agent Mux (MAM) Orchestration Updater
+# ==============================================================================
+# Safely updates MAM skills, virtual environment, and docs to the latest version.
+# Preserves user configuration (.env) and local metadata/jobs database (.mam).
+# ==============================================================================
+set -euo pipefail
+
+TARGET_DIR=""
+FORCE=0
+
+# Parse arguments
+while [[ $# -gt 0 ]]; do
+  case "$1" in
+    -y|--yes|--force)
+      FORCE=1
+      shift
+      ;;
+    *)
+      TARGET_DIR="$1"
+      shift
+      ;;
+  esac
+done
+
+if [ -z "$TARGET_DIR" ]; then
+  TARGET_DIR="$(pwd)"
+fi
+
+echo "===================================================================="
+echo "⚡ Starting Multi-Agent Mux (MAM) Update"
+echo "📂 Target Workspace: $TARGET_DIR"
+echo "===================================================================="
+
+if [ ! -d "$TARGET_DIR" ]; then
+  echo "❌ Error: Target directory '$TARGET_DIR' does not exist." >&2
+  exit 1
+fi
+
+cd "$TARGET_DIR"
+
+# 1. Verification of existing install
+if [ ! -f "remove.sh" ]; then
+  echo "❌ Error: No MAM installation (remove.sh) found in '$TARGET_DIR'." >&2
+  echo "   Please run install.sh first to set up the workspace." >&2
+  exit 1
+fi
+
+# Request confirmation if not forced
+if [ $FORCE -eq 0 ]; then
+  echo "⚠️  WARNING: This will update MAM orchestration skills, virtual environment, "
+  echo "   and docs to the latest version."
+  echo "   (Your configuration, job history, and custom skills will be preserved)."
+  if [ ! -t 0 ]; then
+    echo "❌ Error: Non-interactive terminal detected. Please run with -y/--yes/--force." >&2
+    exit 1
+  fi
+  if ! read -p "❓ Proceed with update? [y/N]: " -r response; then
+    response="n"
+  fi
+  if [[ ! "$response" =~ ^[yY](es)?$ ]]; then
+    echo "❌ Update cancelled by user."
+    exit 0
+  fi
+fi
+
+# 2. Stage backups of user configurations and metadata to prevent deletion
+echo "💾 Backing up configuration and database..."
+HAS_ENV=0
+if [ -f ".env" ]; then
+  HAS_ENV=1
+  mv ".env" ".env.update-tmp"
+fi
+
+HAS_MAM=0
+if [ -d ".mam" ]; then
+  HAS_MAM=1
+  # Move .mam out of the way of remove.sh
+  mv ".mam" ".mam.update-tmp"
+fi
+
+# 3. Perform uninstallation of existing files
+echo "🗑️  Removing existing installation..."
+bash remove.sh --force
+
+# 4. Fetch and run the latest installer from Gitea
+echo "📥 Fetching and running the latest installer..."
+INSTALLER_URL="https://git.godopu.com/tmpl/multi-agent-mux/raw/branch/main/deploy/install.sh"
+if command -v curl &>/dev/null; then
+  curl -fsSL "$INSTALLER_URL" | bash -s -- "$TARGET_DIR"
+elif command -v wget &>/dev/null; then
+  wget -qO- "$INSTALLER_URL" | bash -s -- "$TARGET_DIR"
+else
+  echo "❌ Error: Neither 'curl' nor 'wget' is available to fetch the installer." >&2
+  
+  # Restore backups before failing
+  if [ $HAS_ENV -eq 1 ]; then mv ".env.update-tmp" ".env"; fi
+  if [ $HAS_MAM -eq 1 ]; then mv ".mam.update-tmp" ".mam"; fi
+  exit 1
+fi
+
+# 5. Restore backups of configuration and database
+echo "🔄 Restoring configuration and database..."
+if [ $HAS_ENV -eq 1 ]; then
+  # Overwrite the default .env created by installer (if any) with the user's backup
+  mv -f ".env.update-tmp" ".env"
+fi
+
+if [ $HAS_MAM -eq 1 ]; then
+  # The installer created a new .mam directory with a new manifest.
+  # We want to merge the old .mam database/jobs back while keeping the new manifest.
+  if [ -d ".mam.update-tmp" ]; then
+    # Copy SQLite databases
+    for db in .mam.update-tmp/db.sqlite*; do
+      if [ -f "$db" ]; then
+        cp -f "$db" .mam/
+      fi
+    done
+    # Copy jobs history
+    if [ -d ".mam.update-tmp/jobs" ] && [ "$(ls -A .mam.update-tmp/jobs 2>/dev/null)" ]; then
+      mkdir -p .mam/jobs
+      cp -rf .mam.update-tmp/jobs/* .mam/jobs/
+    fi
+    # Copy delegate logs
+    if [ -d ".mam.update-tmp/delegate_job_logs" ] && [ "$(ls -A .mam.update-tmp/delegate_job_logs 2>/dev/null)" ]; then
+      mkdir -p .mam/delegate_job_logs
+      cp -rf .mam.update-tmp/delegate_job_logs/* .mam/delegate_job_logs/
+    fi
+    # Clean up the backup directory
+    rm -rf ".mam.update-tmp"
+  fi
+fi
+
+echo "===================================================================="
+echo "🎉 Update complete!"
+echo "===================================================================="