dfb33681f9
- Add new gRPC endpoint for Heartbeat in the Yao service, enabling communication with the sandbox. - Update Makefile to include a dedicated unit test target for Sandbox V2, ensuring proper testing of new features. - Enhance CI workflows to incorporate Sandbox V2 tests, allowing for dual-mode testing (local and remote) with Docker. - Modify .gitignore to exclude specific Docker files while allowing shell scripts for Sandbox V2. - Update documentation in DESIGN.md to reflect the new architecture and capabilities of the Sandbox V2. These changes enhance the Yao SDK's functionality, providing improved support for sandbox operations and testing.
231 lines
7.2 KiB
Bash
Executable file
231 lines
7.2 KiB
Bash
Executable file
#!/bin/bash
|
|
# Build script for Yao sandbox Docker images
|
|
# Supports multi-architecture builds (amd64 and arm64)
|
|
|
|
set -e
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
TOOL=${1:-claude}
|
|
PUSH=${2:-false}
|
|
REGISTRY=${REGISTRY:-"yaoapp"} # Docker Hub registry
|
|
|
|
echo "=== Building Yao Sandbox Images ==="
|
|
echo "Tool: $TOOL"
|
|
echo "Push: $PUSH"
|
|
echo "Registry: $REGISTRY"
|
|
echo "Script dir: $SCRIPT_DIR"
|
|
|
|
# Build yao-bridge for both architectures
|
|
echo ""
|
|
echo "=== Building yao-bridge (multi-arch) ==="
|
|
cd "$SCRIPT_DIR/../bridge"
|
|
|
|
echo "Building for linux/amd64..."
|
|
CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -ldflags="-s -w" -o "$SCRIPT_DIR/yao-bridge-amd64" .
|
|
|
|
echo "Building for linux/arm64..."
|
|
CGO_ENABLED=0 GOOS=linux GOARCH=arm64 go build -ldflags="-s -w" -o "$SCRIPT_DIR/yao-bridge-arm64" .
|
|
|
|
echo "Built: yao-bridge-amd64, yao-bridge-arm64"
|
|
|
|
cd "$SCRIPT_DIR"
|
|
|
|
# Build claude-proxy for both architectures
|
|
echo ""
|
|
echo "=== Building claude-proxy (multi-arch) ==="
|
|
cd "$SCRIPT_DIR/../proxy/cmd/claude-proxy"
|
|
|
|
echo "Building for linux/amd64..."
|
|
CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -ldflags="-s -w" -o "$SCRIPT_DIR/claude-proxy-amd64" .
|
|
|
|
echo "Building for linux/arm64..."
|
|
CGO_ENABLED=0 GOOS=linux GOARCH=arm64 go build -ldflags="-s -w" -o "$SCRIPT_DIR/claude-proxy-arm64" .
|
|
|
|
echo "Built: claude-proxy-amd64, claude-proxy-arm64"
|
|
|
|
cd "$SCRIPT_DIR"
|
|
|
|
# Check if buildx is available and set up
|
|
setup_buildx() {
|
|
echo ""
|
|
echo "=== Setting up Docker Buildx ==="
|
|
|
|
# Check if buildx is available
|
|
if ! docker buildx version > /dev/null 2>&1; then
|
|
echo "Error: Docker Buildx is not available. Please install it first."
|
|
exit 1
|
|
fi
|
|
|
|
# Create/use multi-arch builder
|
|
BUILDER_NAME="yao-multiarch"
|
|
if ! docker buildx inspect "$BUILDER_NAME" > /dev/null 2>&1; then
|
|
echo "Creating buildx builder: $BUILDER_NAME"
|
|
docker buildx create --name "$BUILDER_NAME" --use --bootstrap
|
|
else
|
|
echo "Using existing builder: $BUILDER_NAME"
|
|
docker buildx use "$BUILDER_NAME"
|
|
fi
|
|
}
|
|
|
|
# Build single-arch image (amd64 only, for Chrome which has no arm64 build)
|
|
build_amd64() {
|
|
local IMAGE_NAME=$1
|
|
local DOCKERFILE=$2
|
|
local PUSH_FLAG=$3
|
|
|
|
echo ""
|
|
echo "=== Building $IMAGE_NAME (linux/amd64 only) ==="
|
|
|
|
BUILD_ARGS="--platform linux/amd64 -t ${REGISTRY}/${IMAGE_NAME}:latest"
|
|
if [ "$PUSH_FLAG" = "true" ]; then
|
|
BUILD_ARGS="$BUILD_ARGS --push"
|
|
else
|
|
BUILD_ARGS="$BUILD_ARGS --load"
|
|
fi
|
|
|
|
docker buildx build $BUILD_ARGS -f "$DOCKERFILE" .
|
|
}
|
|
|
|
# Build multi-arch image
|
|
build_multiarch() {
|
|
local IMAGE_NAME=$1
|
|
local DOCKERFILE=$2
|
|
local PUSH_FLAG=$3
|
|
|
|
echo ""
|
|
echo "=== Building $IMAGE_NAME (linux/amd64,linux/arm64) ==="
|
|
|
|
BUILD_ARGS="--platform linux/amd64,linux/arm64 -t ${REGISTRY}/${IMAGE_NAME}:latest"
|
|
|
|
if [ "$PUSH_FLAG" = "true" ]; then
|
|
BUILD_ARGS="$BUILD_ARGS --push"
|
|
else
|
|
# Load to local Docker (only works for single platform)
|
|
echo "Note: Multi-arch build without push. Building for current platform only."
|
|
BUILD_ARGS="--load -t ${REGISTRY}/${IMAGE_NAME}:latest"
|
|
fi
|
|
|
|
docker buildx build $BUILD_ARGS -f "$DOCKERFILE" .
|
|
}
|
|
|
|
# Setup buildx for multi-arch builds
|
|
setup_buildx
|
|
|
|
# Build base image
|
|
echo ""
|
|
echo "=== Building base image ==="
|
|
build_multiarch "sandbox-base" "base/Dockerfile.base" "$PUSH"
|
|
|
|
# Build tool-specific images
|
|
case $TOOL in
|
|
claude)
|
|
echo ""
|
|
echo "=== Building Claude images ==="
|
|
build_multiarch "sandbox-claude" "claude/Dockerfile" "$PUSH"
|
|
build_multiarch "sandbox-claude-full" "claude/Dockerfile.full" "$PUSH"
|
|
;;
|
|
claude-vnc)
|
|
echo ""
|
|
echo "=== Building Claude VNC images (Browser + Desktop) ==="
|
|
build_multiarch "sandbox-claude-browser" "browser/Dockerfile" "$PUSH"
|
|
build_multiarch "sandbox-claude-desktop" "desktop/Dockerfile" "$PUSH"
|
|
;;
|
|
browser)
|
|
echo ""
|
|
echo "=== Building Claude Browser image ==="
|
|
build_multiarch "sandbox-claude-browser" "browser/Dockerfile" "$PUSH"
|
|
;;
|
|
desktop)
|
|
echo ""
|
|
echo "=== Building Claude Desktop image ==="
|
|
build_multiarch "sandbox-claude-desktop" "desktop/Dockerfile" "$PUSH"
|
|
;;
|
|
chrome)
|
|
echo ""
|
|
echo "=== Building Claude Chrome image (amd64 only) ==="
|
|
build_amd64 "sandbox-claude-chrome" "chrome/Dockerfile" "$PUSH"
|
|
;;
|
|
cursor)
|
|
echo ""
|
|
echo "=== Building Cursor images ==="
|
|
build_multiarch "sandbox-cursor" "cursor/Dockerfile" "$PUSH"
|
|
;;
|
|
all)
|
|
echo ""
|
|
echo "=== Building all images ==="
|
|
# Claude
|
|
build_multiarch "sandbox-claude" "claude/Dockerfile" "$PUSH"
|
|
build_multiarch "sandbox-claude-full" "claude/Dockerfile.full" "$PUSH"
|
|
# Claude VNC variants
|
|
build_multiarch "sandbox-claude-browser" "browser/Dockerfile" "$PUSH"
|
|
build_multiarch "sandbox-claude-desktop" "desktop/Dockerfile" "$PUSH"
|
|
# Chrome (amd64 only - Google Chrome has no arm64 Linux build)
|
|
build_amd64 "sandbox-claude-chrome" "chrome/Dockerfile" "$PUSH"
|
|
# Cursor (uncomment when ready)
|
|
# build_multiarch "sandbox-cursor" "cursor/Dockerfile" "$PUSH"
|
|
;;
|
|
v2)
|
|
echo "V2 images have their own build script: sandbox/v2/docker/build.sh"
|
|
echo "Usage: sandbox/v2/docker/build.sh [true|false]"
|
|
exit 0
|
|
;;
|
|
*)
|
|
echo "Unknown tool: $TOOL"
|
|
echo "Usage: $0 [claude|claude-vnc|browser|desktop|chrome|cursor|v2|all] [true|false]"
|
|
echo " $0 claude # Build Claude images locally"
|
|
echo " $0 claude true # Build and push Claude images"
|
|
echo " $0 claude-vnc # Build Claude VNC images (Browser + Desktop)"
|
|
echo " $0 browser # Build Claude Browser image only"
|
|
echo " $0 desktop # Build Claude Desktop image only"
|
|
echo " $0 chrome # Build Claude Chrome image (amd64 only)"
|
|
echo " $0 v2 # Build Sandbox V2 images (base + test)"
|
|
echo " $0 all true # Build and push all images"
|
|
exit 1
|
|
;;
|
|
esac
|
|
|
|
echo ""
|
|
echo "=== Build complete ==="
|
|
echo "Images built for tool: $TOOL"
|
|
|
|
if [ "$PUSH" = "true" ]; then
|
|
echo ""
|
|
echo "Images pushed to: $REGISTRY"
|
|
echo " - ${REGISTRY}/sandbox-base:latest"
|
|
case $TOOL in
|
|
claude)
|
|
echo " - ${REGISTRY}/sandbox-claude:latest"
|
|
echo " - ${REGISTRY}/sandbox-claude-full:latest"
|
|
;;
|
|
claude-vnc)
|
|
echo " - ${REGISTRY}/sandbox-claude-browser:latest"
|
|
echo " - ${REGISTRY}/sandbox-claude-desktop:latest"
|
|
;;
|
|
browser)
|
|
echo " - ${REGISTRY}/sandbox-claude-browser:latest"
|
|
;;
|
|
desktop)
|
|
echo " - ${REGISTRY}/sandbox-claude-desktop:latest"
|
|
;;
|
|
chrome)
|
|
echo " - ${REGISTRY}/sandbox-claude-chrome:latest"
|
|
;;
|
|
all)
|
|
echo " - ${REGISTRY}/sandbox-claude:latest"
|
|
echo " - ${REGISTRY}/sandbox-claude-full:latest"
|
|
echo " - ${REGISTRY}/sandbox-claude-browser:latest"
|
|
echo " - ${REGISTRY}/sandbox-claude-desktop:latest"
|
|
echo " - ${REGISTRY}/sandbox-claude-chrome:latest"
|
|
;;
|
|
esac
|
|
fi
|
|
|
|
# Show local images
|
|
docker images | grep -E "(sandbox-base|sandbox-claude|sandbox-cursor)" | head -10 || true
|
|
|
|
# Cleanup
|
|
echo ""
|
|
echo "=== Cleanup ==="
|
|
rm -f "$SCRIPT_DIR/yao-bridge-amd64" "$SCRIPT_DIR/yao-bridge-arm64"
|
|
rm -f "$SCRIPT_DIR/claude-proxy-amd64" "$SCRIPT_DIR/claude-proxy-arm64"
|
|
echo "Removed temporary binary files"
|