Add restore workspace script

Signed-off-by: Artem Savchenko <armisav@gmail.com>
2026-08-17 21:25:55 +02:00 · 2026-07-04 10:20:58 +07:00
parent 767fd47adc
commit 40b73cc145
1 changed files with 144 additions and 0 deletions
@@ -0,0 +1,144 @@
+#!/usr/bin/env bash
+#
+# One-shot workspace restore: create the admin account, create + assign the
+# workspace, then restore a backup into it (via ./backup-restore.sh).
+#
+# Usage:
+#   ./restore-workspace.sh <backup-dir> <workspace> [options] [-- <extra backup-restore args>]
+#
+# Options:
+#   -e, --email <email>       Admin account email     (default: admin@local)
+#   -p, --password <pass>     Admin account password  (default: generated, printed at the end)
+#   -f, --first <name>        Admin first name        (default: Admin)
+#   -l, --last <name>         Admin last name         (default: Platform)
+#       --skip-account        Don't create the account (it already exists)
+#       --skip-workspace      Don't create/assign the workspace (it already exists)
+#       --date <ms>           Backup snapshot timestamp, passed to backup-restore.sh
+#
+# Examples:
+#   ./restore-workspace.sh ./backups/myws myws
+#   ./restore-workspace.sh ./backups/myws myws -e me@example.com -p secret
+#   ./restore-workspace.sh ./backups/myws myws --skip-account -- --merge
+set -euo pipefail
+
+CONFIG_FILE="huly_v7.conf"
+if [ ! -f "$CONFIG_FILE" ]; then
+    echo -e "\033[1;31mConfig not found: $CONFIG_FILE. Run ./setup.sh first.\033[0m"
+    exit 1
+fi
+# shellcheck disable=SC1090
+source "$CONFIG_FILE"
+
+BACKUP_DIR="${1:-}"
+WORKSPACE="${2:-}"
+shift $(( $# >= 2 ? 2 : $# )) || true
+
+EMAIL="admin@local"
+PASSWORD=""
+FIRST="Admin"
+LAST="Platform"
+SKIP_ACCOUNT=false
+SKIP_WORKSPACE=false
+DATE=""
+EXTRA_ARGS=()
+
+while [ $# -gt 0 ] && [ "${1:-}" != "--" ]; do
+    case "$1" in
+        -e|--email)       EMAIL="$2";    shift 2 ;;
+        -p|--password)    PASSWORD="$2"; shift 2 ;;
+        -f|--first)       FIRST="$2";    shift 2 ;;
+        -l|--last)        LAST="$2";     shift 2 ;;
+        --skip-account)   SKIP_ACCOUNT=true;   shift ;;
+        --skip-workspace) SKIP_WORKSPACE=true; shift ;;
+        --date)           DATE="$2";     shift 2 ;;
+        *)
+            echo -e "\033[1;31mUnknown option: $1\033[0m"
+            exit 1
+            ;;
+    esac
+done
+if [ "${1:-}" == "--" ]; then
+    shift
+    EXTRA_ARGS=("$@")
+fi
+
+if [ -z "$BACKUP_DIR" ] || [ -z "$WORKSPACE" ]; then
+    echo "Usage: $0 <backup-dir> <workspace> [options] [-- <extra backup-restore args>]"
+    echo ""
+    echo "  <backup-dir>        Local directory with downloaded backup files"
+    echo "  <workspace>         Workspace id/url to create and restore into"
+    echo "  -e, --email         Admin account email (default: admin@local)"
+    echo "  -p, --password      Admin password (default: generated and printed)"
+    echo "  -f, --first         Admin first name (default: Admin)"
+    echo "  -l, --last          Admin last name (default: Platform)"
+    echo "      --skip-account    Account already exists, don't create it"
+    echo "      --skip-workspace  Workspace already exists, don't create/assign it"
+    echo "      --date <ms>       Backup snapshot timestamp (default: latest)"
+    echo "  -- <args>           Extra args passed to backup-restore.sh tool call (e.g. --merge)"
+    exit 1
+fi
+
+if [ ! -d "$BACKUP_DIR" ]; then
+    echo -e "\033[1;31mBackup directory not found: $BACKUP_DIR\033[0m"
+    exit 1
+fi
+
+# Generate a password if none was provided
+GENERATED_PASSWORD=false
+if [ -z "$PASSWORD" ] && [ "$SKIP_ACCOUNT" != true ]; then
+    PASSWORD="$(LC_ALL=C tr -dc 'A-Za-z0-9' </dev/urandom | head -c 16)"
+    GENERATED_PASSWORD=true
+fi
+
+echo -e "\033[1;34mRestore workspace:\033[0m"
+echo "  Backup:    $BACKUP_DIR"
+echo "  Workspace: $WORKSPACE"
+echo "  Admin:     $EMAIL ($FIRST $LAST)"
+echo "  Account:   $([ "$SKIP_ACCOUNT" == true ] && echo 'skip (exists)' || echo 'create')"
+echo "  Workspace: $([ "$SKIP_WORKSPACE" == true ] && echo 'skip (exists)' || echo 'create + assign')"
+echo ""
+
+# 1. Create the admin account (tolerate "already exists")
+if [ "$SKIP_ACCOUNT" != true ]; then
+    echo -e "\033[1;34m[1/4] Creating account $EMAIL...\033[0m"
+    if ! ./run-tool.sh create-account "$EMAIL" -p "$PASSWORD" -f "$FIRST" -l "$LAST"; then
+        echo -e "\033[1;33mcreate-account failed - account may already exist. Continuing.\033[0m"
+        echo -e "\033[1;33m(Use --skip-account to silence this, or -p to match the existing password.)\033[0m"
+        GENERATED_PASSWORD=false
+    fi
+else
+    echo -e "\033[1;34m[1/4] Skipping account creation.\033[0m"
+fi
+
+# 2 + 3. Create the workspace and assign the admin to it
+if [ "$SKIP_WORKSPACE" != true ]; then
+    echo -e "\n\033[1;34m[2/4] Creating workspace $WORKSPACE...\033[0m"
+    if ! ./run-tool.sh create-workspace "$WORKSPACE" "email:$EMAIL"; then
+        echo -e "\033[1;31mcreate-workspace failed. If it already exists, rerun with --skip-workspace.\033[0m"
+        exit 1
+    fi
+
+    echo -e "\n\033[1;34m[3/4] Assigning $EMAIL to $WORKSPACE...\033[0m"
+    if ! ./run-tool.sh assign-workspace "$WORKSPACE" "$EMAIL"; then
+        echo -e "\033[1;33massign-workspace failed - the owner may already be assigned. Continuing.\033[0m"
+    fi
+else
+    echo -e "\n\033[1;34m[2/4] [3/4] Skipping workspace creation/assignment.\033[0m"
+fi
+
+# 4. Restore the backup
+echo -e "\n\033[1;34m[4/4] Restoring backup into $WORKSPACE...\033[0m"
+RESTORE_CMD=(./backup-restore.sh "$BACKUP_DIR" "$WORKSPACE")
+[ -n "$DATE" ] && RESTORE_CMD+=("$DATE")
+[ ${#EXTRA_ARGS[@]} -gt 0 ] && RESTORE_CMD+=(-- "${EXTRA_ARGS[@]}")
+"${RESTORE_CMD[@]}"
+
+# Summary
+SCHEME="http"
+[ "${SECURE:-}" == "true" ] && SCHEME="https"
+echo -e "\n\033[1;32mDone. Workspace '$WORKSPACE' restored.\033[0m"
+echo "  URL:      ${SCHEME}://${HOST_ADDRESS:-localhost}"
+echo "  Login:    $EMAIL"
+if [ "$GENERATED_PASSWORD" == true ]; then
+    echo -e "  Password: $PASSWORD  \033[1;33m(generated - save it now)\033[0m"
+fi