allow peers backups
This commit is contained in:
goldsquid
162
restore-peers.sh
Executable file
162
restore-peers.sh
Executable file
@@ -0,0 +1,162 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Script to restore peers from a backup file
|
||||
# Usage: ./restore-peers.sh <backup-file> [rpc-url]
|
||||
# backup-file: Path to backup JSON or TXT file
|
||||
# rpc-url: Optional. If not provided, will extract from backup file
|
||||
|
||||
BASEPATH="$(dirname "$0")"
|
||||
source $BASEPATH/.env
|
||||
|
||||
if [ $# -lt 1 ]; then
|
||||
echo "Usage: $0 <backup-file> [rpc-url]"
|
||||
echo ""
|
||||
echo " backup-file: Path to backup JSON or TXT file"
|
||||
echo " rpc-url: Optional. Target RPC URL to restore peers to"
|
||||
echo " If not provided, will use rpc_url from backup file"
|
||||
echo ""
|
||||
echo "Examples:"
|
||||
echo " $0 peer-backups/ethereum-mainnet__ethereum-mainnet-archive__20240101_120000.json"
|
||||
echo " $0 peer-backups/ethereum-mainnet__ethereum-mainnet-archive__20240101_120000.txt https://domain.com/ethereum-mainnet-archive"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
BACKUP_FILE="$1"
|
||||
TARGET_URL="$2"
|
||||
|
||||
if [ ! -f "$BACKUP_FILE" ]; then
|
||||
echo "Error: Backup file not found: $BACKUP_FILE" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Determine file type and extract enodes
|
||||
if [[ "$BACKUP_FILE" == *.json ]]; then
|
||||
# JSON backup file
|
||||
if ! command -v jq &> /dev/null; then
|
||||
echo "Error: jq is required to parse JSON backup files" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Extract RPC URL from backup if not provided
|
||||
if [ -z "$TARGET_URL" ]; then
|
||||
TARGET_URL=$(jq -r '.rpc_url // empty' "$BACKUP_FILE" 2>/dev/null)
|
||||
if [ -z "$TARGET_URL" ] || [ "$TARGET_URL" = "null" ]; then
|
||||
echo "Error: Could not extract rpc_url from backup file and no target URL provided" >&2
|
||||
exit 1
|
||||
fi
|
||||
fi
|
||||
|
||||
# Extract enodes
|
||||
enodes=$(jq -r '.peers[]?' "$BACKUP_FILE" 2>/dev/null)
|
||||
|
||||
if [ -z "$enodes" ]; then
|
||||
echo "Error: No peers found in backup file" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
peer_count=$(jq -r '.peer_count // 0' "$BACKUP_FILE" 2>/dev/null)
|
||||
compose_file=$(jq -r '.compose_file // "unknown"' "$BACKUP_FILE" 2>/dev/null)
|
||||
rpc_path=$(jq -r '.rpc_path // "unknown"' "$BACKUP_FILE" 2>/dev/null)
|
||||
timestamp=$(jq -r '.timestamp // "unknown"' "$BACKUP_FILE" 2>/dev/null)
|
||||
|
||||
echo "Restoring peers from backup:"
|
||||
echo " Compose file: $compose_file"
|
||||
echo " RPC path: $rpc_path"
|
||||
echo " Timestamp: $timestamp"
|
||||
echo " Peer count: $peer_count"
|
||||
echo " Target URL: $TARGET_URL"
|
||||
echo ""
|
||||
|
||||
elif [[ "$BACKUP_FILE" == *.txt ]]; then
|
||||
# TXT backup file (one enode per line)
|
||||
enodes=$(grep -v '^$' "$BACKUP_FILE" | grep -v '^null$')
|
||||
|
||||
if [ -z "$enodes" ]; then
|
||||
echo "Error: No peers found in backup file" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
peer_count=$(echo "$enodes" | wc -l | tr -d ' ')
|
||||
|
||||
if [ -z "$TARGET_URL" ]; then
|
||||
echo "Error: Target RPC URL required for TXT backup files" >&2
|
||||
echo "Usage: $0 <backup-file> <rpc-url>" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
echo "Restoring peers from backup:"
|
||||
echo " Backup file: $BACKUP_FILE"
|
||||
echo " Peer count: $peer_count"
|
||||
echo " Target URL: $TARGET_URL"
|
||||
echo ""
|
||||
|
||||
else
|
||||
echo "Error: Unsupported backup file format. Expected .json or .txt" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Confirm before proceeding
|
||||
read -p "Do you want to restore $peer_count peer(s) to $TARGET_URL? (y/N) " -n 1 -r
|
||||
echo
|
||||
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
|
||||
echo "Restore cancelled"
|
||||
exit 0
|
||||
fi
|
||||
|
||||
# Restore peers
|
||||
success_count=0
|
||||
failed_count=0
|
||||
skipped_count=0
|
||||
|
||||
echo ""
|
||||
echo "Restoring peers..."
|
||||
|
||||
while IFS= read -r enode; do
|
||||
if [ -z "$enode" ] || [ "$enode" = "null" ]; then
|
||||
skipped_count=$((skipped_count + 1))
|
||||
continue
|
||||
fi
|
||||
|
||||
# Check if peer is reachable (optional, can be slow)
|
||||
if [ -f "$BASEPATH/check-enode.sh" ]; then
|
||||
if ! "$BASEPATH/check-enode.sh" "$enode" --target "$TARGET_URL" > /dev/null 2>&1; then
|
||||
echo "⚠ Skipping unreachable peer: ${enode:0:50}..."
|
||||
skipped_count=$((skipped_count + 1))
|
||||
continue
|
||||
fi
|
||||
fi
|
||||
|
||||
echo -n "Adding peer: ${enode:0:50}... "
|
||||
|
||||
result=$(curl --ipv4 -X POST -H "Content-Type: application/json" \
|
||||
--silent --max-time 10 \
|
||||
--data "{\"jsonrpc\":\"2.0\",\"method\":\"admin_addPeer\",\"params\":[\"${enode}\"],\"id\":1}" \
|
||||
"$TARGET_URL" 2>/dev/null | jq -r '.result // .error.message // "unknown error"' 2>/dev/null)
|
||||
|
||||
if [ "$result" = "true" ] || [ "$result" = "null" ]; then
|
||||
echo "✓ Success"
|
||||
success_count=$((success_count + 1))
|
||||
else
|
||||
echo "✗ Failed: $result"
|
||||
failed_count=$((failed_count + 1))
|
||||
fi
|
||||
|
||||
# Small delay to avoid overwhelming the node
|
||||
sleep 0.1
|
||||
done <<< "$enodes"
|
||||
|
||||
echo ""
|
||||
echo "=========================================="
|
||||
echo "Restore Summary"
|
||||
echo "=========================================="
|
||||
echo "Successful: $success_count/$peer_count"
|
||||
echo "Failed: $failed_count/$peer_count"
|
||||
echo "Skipped: $skipped_count/$peer_count"
|
||||
echo ""
|
||||
|
||||
if [ $failed_count -gt 0 ]; then
|
||||
exit 1
|
||||
fi
|
||||
|
||||
exit 0
|
||||
|
||||
Reference in New Issue
Block a user