182 lines
6.1 KiB
Bash
182 lines
6.1 KiB
Bash
#!/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)"
|
|
else
|
|
if [ ! -d "$TARGET_DIR" ]; then
|
|
echo "❌ Error: Target directory '$TARGET_DIR' does not exist." >&2
|
|
exit 1
|
|
fi
|
|
TARGET_DIR="$(cd "$TARGET_DIR" && pwd)"
|
|
fi
|
|
|
|
echo "===================================================================="
|
|
echo "⚡ Starting Multi-Agent Mux (MAM) Update"
|
|
echo "📂 Target Workspace: $TARGET_DIR"
|
|
echo "===================================================================="
|
|
|
|
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
|
|
# Copy database and jobs to temporary backup outside of .mam.
|
|
# We do NOT move the .mam folder away so that remove.sh can still read .mam/install_manifest.txt!
|
|
mkdir -p .mam.update-tmp
|
|
# Copy SQLite databases and session files
|
|
for db in .mam/agent-sessions.*; do
|
|
if [ -f "$db" ]; then
|
|
cp -f "$db" .mam.update-tmp/
|
|
fi
|
|
done
|
|
# Copy jobs history
|
|
if [ -d ".mam/jobs" ] && [ "$(ls -A .mam/jobs 2>/dev/null)" ]; then
|
|
mkdir -p .mam.update-tmp/jobs
|
|
cp -rf .mam/jobs/* .mam.update-tmp/jobs/
|
|
fi
|
|
# Copy delegate logs
|
|
if [ -d ".mam/delegate_job_logs" ] && [ "$(ls -A .mam/delegate_job_logs 2>/dev/null)" ]; then
|
|
mkdir -p .mam.update-tmp/delegate_job_logs
|
|
cp -rf .mam/delegate_job_logs/* .mam.update-tmp/delegate_job_logs/
|
|
fi
|
|
# Copy manifest so we have a backup
|
|
if [ -f ".mam/install_manifest.txt" ]; then
|
|
cp -f .mam/install_manifest.txt .mam.update-tmp/
|
|
fi
|
|
fi
|
|
|
|
# Define trap to restore backup files on failure
|
|
restore_on_failure() {
|
|
echo "❌ Update failed. Reverting configuration and database to previous state..."
|
|
if [ $HAS_ENV -eq 1 ] && [ -f ".env.update-tmp" ]; then
|
|
mv -f ".env.update-tmp" ".env" 2>/dev/null || true
|
|
fi
|
|
if [ $HAS_MAM -eq 1 ] && [ -d ".mam.update-tmp" ]; then
|
|
# Revert to old database/jobs backup by restoring .mam directory
|
|
rm -rf .mam 2>/dev/null || true
|
|
mkdir -p .mam
|
|
cp -f .mam.update-tmp/agent-sessions.* .mam/ 2>/dev/null || true
|
|
if [ -d ".mam.update-tmp/jobs" ]; then
|
|
cp -rf .mam.update-tmp/jobs .mam/ 2>/dev/null || true
|
|
fi
|
|
if [ -d ".mam.update-tmp/delegate_job_logs" ]; then
|
|
cp -rf .mam.update-tmp/delegate_job_logs .mam/ 2>/dev/null || true
|
|
fi
|
|
if [ -f ".mam.update-tmp/install_manifest.txt" ]; then
|
|
cp -f .mam.update-tmp/install_manifest.txt .mam/ 2>/dev/null || true
|
|
fi
|
|
rm -rf .mam.update-tmp 2>/dev/null || true
|
|
fi
|
|
}
|
|
trap restore_on_failure EXIT
|
|
|
|
# 3. Perform uninstallation of existing files
|
|
echo "🗑️ Removing existing installation..."
|
|
# remove.sh will run in manifest mode because .mam/install_manifest.txt is still present.
|
|
# It will delete .agents/, documents, scripts, .venv, and .mam folder.
|
|
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
|
|
exit 1
|
|
fi
|
|
|
|
# Disable failure trap since installation succeeded
|
|
trap - EXIT
|
|
|
|
# 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
|
|
if [ -d ".mam.update-tmp" ]; 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.
|
|
for db in .mam.update-tmp/agent-sessions.*; do
|
|
if [ -f "$db" ]; then
|
|
cp -f "$db" .mam/
|
|
fi
|
|
done
|
|
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
|
|
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
|
|
rm -rf ".mam.update-tmp"
|
|
fi
|
|
fi
|
|
|
|
echo "===================================================================="
|
|
echo "🎉 Update complete!"
|
|
echo "===================================================================="
|