183 lines
6.4 KiB
Bash
183 lines
6.4 KiB
Bash
|
#!/bin/bash
|
||
|
|
# ==============================================================================
|
||
|
|
# OpenCode Persistence Restore Script
|
||
|
|
# ==============================================================================
|
||
|
|
# This script restores session data from GitHub Actions artifacts.
|
||
|
|
# It handles the restoration of:
|
||
|
|
# - Configuration files (~/.config/opencode/)
|
||
|
|
# - Session data, messages, chats (~/.local/share/opencode/storage/)
|
||
|
|
# - Project snapshots (~/.local/share/opencode/snapshot/)
|
||
|
|
# - Authentication tokens (~/.local/share/opencode/auth.json)
|
||
|
|
#
|
||
|
|
# If the artifact is encrypted (session.enc exists), it will be decrypted
|
||
|
|
# using OPENCODE_SERVER_PASSWORD.
|
||
|
|
# ==============================================================================
|
||
|
|
|
||
|
|
set -euo pipefail
|
||
|
|
|
||
|
|
RESTORE_DIR="${RESTORE_DIR:-/tmp/opencode-restore}"
|
||
|
|
CONFIG_DIR="$HOME/.config/opencode"
|
||
|
|
SHARE_DIR="$HOME/.local/share/opencode"
|
||
|
|
ENCRYPTION_PASSWORD="${OPENCODE_SERVER_PASSWORD:-}"
|
||
|
|
|
||
|
|
echo "=== OpenCode Session Restore ==="
|
||
|
|
echo "Restore directory: $RESTORE_DIR"
|
||
|
|
echo "Config directory: $CONFIG_DIR"
|
||
|
|
echo "Share directory: $SHARE_DIR"
|
||
|
|
echo ""
|
||
|
|
|
||
|
|
# ------------------------------------------------------------------------------
|
||
|
|
# Check for encrypted artifact and decrypt if needed
|
||
|
|
# ------------------------------------------------------------------------------
|
||
|
|
if [ -f "$RESTORE_DIR/session.enc" ]; then
|
||
|
|
echo "=== Encrypted Artifact Detected ==="
|
||
|
|
|
||
|
|
if [ -z "$ENCRYPTION_PASSWORD" ]; then
|
||
|
|
echo "ERROR: Encrypted artifact found but OPENCODE_SERVER_PASSWORD is not set."
|
||
|
|
echo "Cannot decrypt session data. Starting fresh."
|
||
|
|
rm -rf "$RESTORE_DIR"
|
||
|
|
exit 0
|
||
|
|
fi
|
||
|
|
|
||
|
|
echo "Decrypting session data..."
|
||
|
|
TEMP_ARCHIVE="/tmp/opencode-session-data.tar.gz"
|
||
|
|
|
||
|
|
if openssl enc -aes-256-cbc -d -salt -pbkdf2 -iter 100000 \
|
||
|
|
-in "$RESTORE_DIR/session.enc" \
|
||
|
|
-out "$TEMP_ARCHIVE" \
|
||
|
|
-pass pass:"$ENCRYPTION_PASSWORD" 2>/dev/null; then
|
||
|
|
|
||
|
|
rm -rf "$RESTORE_DIR"
|
||
|
|
mkdir -p "$RESTORE_DIR"
|
||
|
|
tar -xzf "$TEMP_ARCHIVE" -C "$RESTORE_DIR"
|
||
|
|
rm -f "$TEMP_ARCHIVE"
|
||
|
|
echo "Decryption successful."
|
||
|
|
else
|
||
|
|
echo "ERROR: Decryption failed. Password may be incorrect."
|
||
|
|
echo "Starting fresh session."
|
||
|
|
rm -rf "$RESTORE_DIR"
|
||
|
|
rm -f "$TEMP_ARCHIVE"
|
||
|
|
exit 0
|
||
|
|
fi
|
||
|
|
echo ""
|
||
|
|
fi
|
||
|
|
|
||
|
|
# ------------------------------------------------------------------------------
|
||
|
|
# Debug: Show what we're working with
|
||
|
|
# ------------------------------------------------------------------------------
|
||
|
|
echo "=== RESTORE DEBUG ==="
|
||
|
|
echo "Contents of $RESTORE_DIR:"
|
||
|
|
if [ -d "$RESTORE_DIR" ]; then
|
||
|
|
ls -la "$RESTORE_DIR/" 2>/dev/null || echo "Directory exists but empty"
|
||
|
|
echo ""
|
||
|
|
echo "Full tree of restore directory:"
|
||
|
|
find "$RESTORE_DIR" -type f 2>/dev/null | head -50 || echo "No files found"
|
||
|
|
else
|
||
|
|
echo "Restore directory does not exist - fresh start"
|
||
|
|
exit 0
|
||
|
|
fi
|
||
|
|
echo ""
|
||
|
|
|
||
|
|
# ------------------------------------------------------------------------------
|
||
|
|
# Create target directories
|
||
|
|
# ------------------------------------------------------------------------------
|
||
|
|
mkdir -p "$CONFIG_DIR"
|
||
|
|
mkdir -p "$SHARE_DIR"
|
||
|
|
|
||
|
|
# ------------------------------------------------------------------------------
|
||
|
|
# Restore configuration files
|
||
|
|
# ------------------------------------------------------------------------------
|
||
|
|
restore_config() {
|
||
|
|
local src="$RESTORE_DIR/config"
|
||
|
|
|
||
|
|
if [ -d "$src" ] && [ "$(ls -A "$src" 2>/dev/null)" ]; then
|
||
|
|
echo "Restoring configuration files..."
|
||
|
|
|
||
|
|
# Restore all config files except node_modules (will be reinstalled)
|
||
|
|
find "$src" -maxdepth 1 -type f -exec cp -v {} "$CONFIG_DIR/" \; 2>/dev/null || true
|
||
|
|
|
||
|
|
# Count restored files
|
||
|
|
local count=$(find "$src" -maxdepth 1 -type f 2>/dev/null | wc -l)
|
||
|
|
echo "Restored $count configuration file(s)"
|
||
|
|
else
|
||
|
|
echo "No configuration files to restore"
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# ------------------------------------------------------------------------------
|
||
|
|
# Restore share data (sessions, messages, snapshots, etc.)
|
||
|
|
# ------------------------------------------------------------------------------
|
||
|
|
restore_share() {
|
||
|
|
local src="$RESTORE_DIR/share"
|
||
|
|
|
||
|
|
if [ -d "$src" ] && [ "$(ls -A "$src" 2>/dev/null)" ]; then
|
||
|
|
echo "Restoring share data..."
|
||
|
|
|
||
|
|
# Restore auth.json
|
||
|
|
if [ -f "$src/auth.json" ]; then
|
||
|
|
cp -v "$src/auth.json" "$SHARE_DIR/" 2>/dev/null || true
|
||
|
|
echo "Authentication data restored"
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Restore storage directory (sessions, messages, parts, projects)
|
||
|
|
if [ -d "$src/storage" ]; then
|
||
|
|
mkdir -p "$SHARE_DIR/storage"
|
||
|
|
cp -rv "$src/storage/"* "$SHARE_DIR/storage/" 2>/dev/null || true
|
||
|
|
local storage_count=$(find "$SHARE_DIR/storage" -type f 2>/dev/null | wc -l)
|
||
|
|
echo "Storage data restored: $storage_count file(s)"
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Restore snapshot directory (project snapshots)
|
||
|
|
if [ -d "$src/snapshot" ]; then
|
||
|
|
mkdir -p "$SHARE_DIR/snapshot"
|
||
|
|
cp -rv "$src/snapshot/"* "$SHARE_DIR/snapshot/" 2>/dev/null || true
|
||
|
|
local snapshot_count=$(find "$SHARE_DIR/snapshot" -type f 2>/dev/null | wc -l)
|
||
|
|
echo "Snapshot data restored: $snapshot_count file(s)"
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Restore log directory
|
||
|
|
if [ -d "$src/log" ]; then
|
||
|
|
mkdir -p "$SHARE_DIR/log"
|
||
|
|
cp -rv "$src/log/"* "$SHARE_DIR/log/" 2>/dev/null || true
|
||
|
|
echo "Log data restored"
|
||
|
|
fi
|
||
|
|
else
|
||
|
|
echo "No share data to restore"
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# ==============================================================================
|
||
|
|
# Main Restoration Process
|
||
|
|
# ==============================================================================
|
||
|
|
|
||
|
|
echo "=== Starting Restoration ==="
|
||
|
|
|
||
|
|
restore_config
|
||
|
|
echo ""
|
||
|
|
|
||
|
|
restore_share
|
||
|
|
echo ""
|
||
|
|
|
||
|
|
# ------------------------------------------------------------------------------
|
||
|
|
# Summary
|
||
|
|
# ------------------------------------------------------------------------------
|
||
|
|
echo "=== Restoration Summary ==="
|
||
|
|
|
||
|
|
echo "Config directory contents:"
|
||
|
|
ls -la "$CONFIG_DIR/" 2>/dev/null | head -10 || echo "Empty"
|
||
|
|
echo ""
|
||
|
|
|
||
|
|
echo "Share directory structure:"
|
||
|
|
if [ -d "$SHARE_DIR" ]; then
|
||
|
|
du -sh "$SHARE_DIR"/* 2>/dev/null || echo "Empty"
|
||
|
|
fi
|
||
|
|
echo ""
|
||
|
|
|
||
|
|
# Count total restored items
|
||
|
|
config_files=$(find "$CONFIG_DIR" -maxdepth 1 -type f 2>/dev/null | wc -l)
|
||
|
|
share_files=$(find "$SHARE_DIR" -type f 2>/dev/null | wc -l)
|
||
|
|
|
||
|
|
echo "Total restored: $config_files config files, $share_files share files"
|
||
|
|
echo ""
|
||
|
|
echo "Session restoration complete!"
|