mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-10-26 21:26:39 +00:00
107 lines
3.5 KiB
Bash
Executable File
107 lines
3.5 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
#
|
|
# cleanup-containers.sh - Clean up DeerFlow sandbox containers
|
|
#
|
|
# This script cleans up both Docker and Apple Container runtime containers
|
|
# to ensure compatibility across different container runtimes.
|
|
#
|
|
|
|
set -e
|
|
|
|
PREFIX="${1:-deer-flow-sandbox}"
|
|
|
|
# Colors for output
|
|
RED='\033[0;31m'
|
|
GREEN='\033[0;32m'
|
|
YELLOW='\033[1;33m'
|
|
NC='\033[0m' # No Color
|
|
|
|
echo "Cleaning up sandbox containers with prefix: ${PREFIX}"
|
|
|
|
# Function to clean up Docker containers
|
|
cleanup_docker() {
|
|
if command -v docker &> /dev/null; then
|
|
echo -n "Checking Docker containers... "
|
|
# Docker's name filter matches substrings. Check the returned names as
|
|
# well so a container like not-deer-flow-sandbox is never stopped.
|
|
DOCKER_LIST=$(docker ps --filter "name=${PREFIX}-" --format '{{.ID}} {{.Names}} {{.Status}}' 2>/dev/null || true)
|
|
docker_ids=()
|
|
docker_rows=()
|
|
while read -r cid name status; do
|
|
case "$name" in
|
|
"$PREFIX"-*) docker_ids+=("$cid"); docker_rows+=("$cid $name $status") ;;
|
|
esac
|
|
done <<< "$DOCKER_LIST"
|
|
|
|
if [ "${#docker_ids[@]}" -gt 0 ]; then
|
|
echo ""
|
|
echo "Found Docker containers to clean up:"
|
|
printf ' %s\n' "${docker_rows[@]}"
|
|
echo "Stopping Docker containers..."
|
|
docker stop "${docker_ids[@]}" >/dev/null 2>&1 || true
|
|
echo -e "${GREEN}✓ Docker containers stopped${NC}"
|
|
else
|
|
echo -e "${GREEN}none found${NC}"
|
|
fi
|
|
else
|
|
echo "Docker not found, skipping..."
|
|
fi
|
|
}
|
|
|
|
# Function to clean up Apple Container containers
|
|
cleanup_apple_container() {
|
|
if command -v container &> /dev/null; then
|
|
echo -n "Checking Apple Container containers... "
|
|
|
|
# List all containers and filter by name
|
|
CONTAINER_LIST=$(container list --format json 2>/dev/null || echo "[]")
|
|
|
|
if [ "$CONTAINER_LIST" != "[]" ] && [ -n "$CONTAINER_LIST" ]; then
|
|
# Extract container IDs that match our prefix
|
|
CONTAINER_IDS=$(echo "$CONTAINER_LIST" | CLEANUP_CONTAINER_PREFIX="$PREFIX" python3 -c '
|
|
import json
|
|
import os
|
|
import sys
|
|
try:
|
|
containers = json.load(sys.stdin)
|
|
if isinstance(containers, list):
|
|
prefix = os.environ["CLEANUP_CONTAINER_PREFIX"] + "-"
|
|
for c in containers:
|
|
if isinstance(c, dict):
|
|
configuration = c.get("configuration")
|
|
cid = configuration.get("id") if isinstance(configuration, dict) else None
|
|
if isinstance(cid, str) and cid.startswith(prefix):
|
|
print(cid)
|
|
except:
|
|
pass
|
|
' 2>/dev/null || echo "")
|
|
|
|
if [ -n "$CONTAINER_IDS" ]; then
|
|
echo ""
|
|
echo "Found Apple Container containers to clean up:"
|
|
echo "$CONTAINER_IDS" | while read -r cid; do
|
|
echo " - $cid"
|
|
done
|
|
|
|
echo "Stopping Apple Container containers..."
|
|
echo "$CONTAINER_IDS" | while read -r cid; do
|
|
container stop "$cid" 2>/dev/null || true
|
|
done
|
|
echo -e "${GREEN}✓ Apple Container containers stopped${NC}"
|
|
else
|
|
echo -e "${GREEN}none found${NC}"
|
|
fi
|
|
else
|
|
echo -e "${GREEN}none found${NC}"
|
|
fi
|
|
else
|
|
echo "Apple Container not found, skipping..."
|
|
fi
|
|
}
|
|
|
|
# Clean up both runtimes
|
|
cleanup_docker
|
|
cleanup_apple_container
|
|
|
|
echo -e "${GREEN}✓ Container cleanup complete${NC}"
|