@@ -0,0 +1,42 @@
|
||||
#!/bin/sh
|
||||
set -eu
|
||||
ENV_FILE=${ENV_FILE:-.env}
|
||||
SNAPSHOT=${1:-}
|
||||
HELPER_IMAGE=${BACKUP_HELPER_IMAGE:-busybox:1.36}
|
||||
[ -n "$SNAPSHOT" ] && [ -d "$SNAPSHOT" ] || { echo "usage: $0 SNAPSHOT_DIR" >&2; exit 2; }
|
||||
[ -f "$ENV_FILE" ] || { echo "restore: missing $ENV_FILE" >&2; exit 1; }
|
||||
( cd "$SNAPSHOT" && sha256sum -c SHA256SUMS )
|
||||
./scripts/preflight.sh
|
||||
getv() { sed -n "s/^$1=//p" "$ENV_FILE" | tail -n1 | tr -d '\r'; }
|
||||
|
||||
restart() { docker compose --env-file "$ENV_FILE" up -d knowledge neuroforge neuroforge-worker agent >/dev/null 2>&1 || true; }
|
||||
trap restart EXIT INT TERM
|
||||
|
||||
docker compose --env-file "$ENV_FILE" stop neuroforge-worker agent neuroforge knowledge
|
||||
docker compose --env-file "$ENV_FILE" create neuroforge agent >/dev/null
|
||||
volume_for() {
|
||||
service=$1; destination=$2
|
||||
cid=$(docker compose --env-file "$ENV_FILE" ps -aq "$service")
|
||||
[ -n "$cid" ] || return 1
|
||||
docker inspect -f "{{range .Mounts}}{{if eq .Destination \"$destination\"}}{{.Name}}{{end}}{{end}}" "$cid"
|
||||
}
|
||||
restore_volume() {
|
||||
service=$1; destination=$2; source=$3
|
||||
volume=$(volume_for "$service" "$destination")
|
||||
[ -n "$volume" ] || { echo "restore: volume for $service:$destination not found" >&2; exit 1; }
|
||||
src=$(CDPATH= cd -- "$source" && pwd)
|
||||
docker run --rm -v "$volume:/target" -v "$src:/source:ro" "$HELPER_IMAGE" sh -eu -c 'rm -rf /target/* /target/.[!.]* /target/..?* 2>/dev/null || true; cp -a /source/. /target/'
|
||||
}
|
||||
restore_volume neuroforge /app/data "$SNAPSHOT/neuroforge-data"
|
||||
restore_volume agent /app/data "$SNAPSHOT/agent-data"
|
||||
restore_host() {
|
||||
src=$1; dst=$2
|
||||
[ -d "$src" ] || return 0
|
||||
mkdir -p "$dst"
|
||||
find "$dst" -mindepth 1 -maxdepth 1 -exec rm -rf -- {} +
|
||||
cp -a "$src"/. "$dst"/
|
||||
}
|
||||
restore_host "$SNAPSHOT/knowledge" "$(getv KB_DATA_PATH)"
|
||||
restore_host "$SNAPSHOT/staging" "$(getv KB_STAGING_PATH)"
|
||||
restore_host "$SNAPSHOT/knowledge-backups" "$(getv KB_BACKUP_PATH)"
|
||||
echo "restore: snapshot restored; services will be restarted"
|
||||
Reference in New Issue
Block a user