zenfun
64d8daa67d
- Generated config uses allow_anonymous: false (triggers auto-provision) - Set BAY_DATA_DIR so credentials.json writes to pkgs/bay/ - Add read_bay_credentials() to extract auto-generated key after boot - Display API key in config hints for easy AstrBot setup
299 lines
10 KiB
Bash
Executable File
299 lines
10 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
# ──────────────────────────────────────────────────────────────
|
||
# start-with-neo.sh — 一键启动 Shipyard Neo Bay + AstrBot
|
||
#
|
||
# Usage:
|
||
# bash scripts/start-with-neo.sh # 默认 Bay :8114
|
||
# BAY_PORT=9000 bash scripts/start-with-neo.sh # 自定义端口
|
||
# ──────────────────────────────────────────────────────────────
|
||
set -euo pipefail
|
||
|
||
# ── 路径 ──────────────────────────────────────────────────────
|
||
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
||
ASTRBOT_DIR="$(cd "$SCRIPT_DIR/.." && pwd)"
|
||
# shipyard-neo mono-repo root is one level above AstrBot
|
||
NEO_ROOT="$(cd "$ASTRBOT_DIR/.." && pwd)"
|
||
BAY_DIR="$NEO_ROOT/pkgs/bay"
|
||
|
||
BAY_PORT="${BAY_PORT:-8114}"
|
||
BAY_HOST="0.0.0.0"
|
||
BAY_PID=""
|
||
BAY_API_KEY="" # Populated after Bay starts from credentials.json
|
||
|
||
# ── 颜色 ──────────────────────────────────────────────────────
|
||
RED='\033[0;31m'
|
||
GREEN='\033[0;32m'
|
||
YELLOW='\033[1;33m'
|
||
CYAN='\033[0;36m'
|
||
NC='\033[0m' # No Color
|
||
|
||
log() { echo -e "${CYAN}[neo]${NC} $*"; }
|
||
ok() { echo -e "${GREEN}[neo]${NC} $*"; }
|
||
warn() { echo -e "${YELLOW}[neo]${NC} $*"; }
|
||
err() { echo -e "${RED}[neo]${NC} $*" >&2; }
|
||
|
||
# ── 清理函数 ──────────────────────────────────────────────────
|
||
cleanup() {
|
||
log "Shutting down..."
|
||
if [[ -n "$BAY_PID" ]] && kill -0 "$BAY_PID" 2>/dev/null; then
|
||
log "Stopping Bay (PID $BAY_PID)..."
|
||
kill "$BAY_PID" 2>/dev/null || true
|
||
wait "$BAY_PID" 2>/dev/null || true
|
||
fi
|
||
ok "All services stopped."
|
||
}
|
||
trap cleanup EXIT INT TERM
|
||
|
||
# ── 检查前置条件 ──────────────────────────────────────────────
|
||
check_prerequisites() {
|
||
log "Checking prerequisites..."
|
||
|
||
if [[ ! -d "$BAY_DIR" ]]; then
|
||
err "Bay directory not found: $BAY_DIR"
|
||
err "Expected shipyard-neo mono-repo at: $NEO_ROOT"
|
||
exit 1
|
||
fi
|
||
|
||
if ! command -v uv &>/dev/null; then
|
||
err "'uv' is not installed. Please install it first."
|
||
exit 1
|
||
fi
|
||
|
||
# Check Docker access (try without sudo first, then with sudo)
|
||
if docker info &>/dev/null 2>&1; then
|
||
ok "Docker is accessible."
|
||
elif sudo docker info &>/dev/null 2>&1; then
|
||
warn "Docker requires sudo. Bay may need socket permissions."
|
||
warn "If Bay fails to connect to Docker, run: sudo chmod 666 /var/run/docker.sock"
|
||
else
|
||
err "Docker is not accessible. Please install Docker or fix permissions."
|
||
exit 1
|
||
fi
|
||
|
||
# Check Bay venv
|
||
if [[ ! -d "$BAY_DIR/.venv" ]]; then
|
||
log "Bay venv not found. Running 'uv sync' in $BAY_DIR ..."
|
||
(cd "$BAY_DIR" && uv sync)
|
||
fi
|
||
|
||
ok "Prerequisites OK."
|
||
}
|
||
|
||
# ── 生成 Bay config.yaml（如不存在）────────────────────────────
|
||
ensure_bay_config() {
|
||
local config_file="$BAY_DIR/config.yaml"
|
||
|
||
if [[ -f "$config_file" ]]; then
|
||
ok "Bay config.yaml already exists."
|
||
return
|
||
fi
|
||
|
||
log "Generating Bay config.yaml for local development..."
|
||
|
||
cat > "$config_file" << 'BAYCONFIG'
|
||
# Bay Local Development Config (auto-generated by start-with-neo.sh)
|
||
# For full reference see config.yaml.example
|
||
|
||
server:
|
||
host: "0.0.0.0"
|
||
port: 8114
|
||
|
||
database:
|
||
url: "sqlite+aiosqlite:///./bay.db"
|
||
echo: false
|
||
|
||
driver:
|
||
type: docker
|
||
image_pull_policy: if_not_present
|
||
docker:
|
||
socket: "unix:///var/run/docker.sock"
|
||
connect_mode: host_port
|
||
host_address: "127.0.0.1"
|
||
publish_ports: true
|
||
host_port: null
|
||
network: null
|
||
|
||
cargo:
|
||
root_path: "/var/lib/bay/cargos"
|
||
default_size_limit_mb: 1024
|
||
mount_path: "/workspace"
|
||
|
||
# Security: auto-provision mode
|
||
# Bay generates sk-bay-* key on first boot → credentials.json
|
||
security:
|
||
allow_anonymous: false
|
||
|
||
profiles:
|
||
- id: python-default
|
||
description: "Standard Python sandbox"
|
||
image: "ghcr.io/astrbotdevs/shipyard-neo-ship:latest"
|
||
runtime_type: ship
|
||
runtime_port: 8123
|
||
resources:
|
||
cpus: 1.0
|
||
memory: "1g"
|
||
capabilities:
|
||
- filesystem
|
||
- shell
|
||
- python
|
||
idle_timeout: 1800
|
||
env: {}
|
||
|
||
gc:
|
||
enabled: true
|
||
run_on_startup: true
|
||
interval_seconds: 300
|
||
idle_session:
|
||
enabled: true
|
||
expired_sandbox:
|
||
enabled: true
|
||
orphan_cargo:
|
||
enabled: true
|
||
orphan_container:
|
||
enabled: false
|
||
BAYCONFIG
|
||
|
||
ok "Bay config.yaml created at $config_file"
|
||
}
|
||
|
||
# ── 拉取 Ship 镜像 ───────────────────────────────────────────
|
||
ensure_ship_image() {
|
||
local image="ghcr.io/astrbotdevs/shipyard-neo-ship:latest"
|
||
log "Checking Ship image: $image ..."
|
||
|
||
if docker image inspect "$image" &>/dev/null 2>&1 || \
|
||
sudo docker image inspect "$image" &>/dev/null 2>&1; then
|
||
ok "Ship image is available locally."
|
||
else
|
||
log "Pulling Ship image (this may take a while)..."
|
||
if docker pull "$image" 2>/dev/null || sudo docker pull "$image" 2>/dev/null; then
|
||
ok "Ship image pulled successfully."
|
||
else
|
||
warn "Failed to pull Ship image. Bay will try to pull it on first sandbox creation."
|
||
fi
|
||
fi
|
||
}
|
||
|
||
# ── 启动 Bay ──────────────────────────────────────────────────
|
||
start_bay() {
|
||
log "Starting Bay on :$BAY_PORT ..."
|
||
|
||
(cd "$BAY_DIR" && BAY_DATA_DIR="$BAY_DIR" uv run uvicorn app.main:app \
|
||
--host "$BAY_HOST" \
|
||
--port "$BAY_PORT" \
|
||
--reload \
|
||
2>&1 | sed "s/^/ ${CYAN}[bay]${NC} /") &
|
||
BAY_PID=$!
|
||
|
||
log "Bay started (PID $BAY_PID), waiting for health check..."
|
||
|
||
# Wait for Bay to become healthy
|
||
local max_wait=30
|
||
local waited=0
|
||
while [[ $waited -lt $max_wait ]]; do
|
||
if curl -sf "http://127.0.0.1:$BAY_PORT/health" &>/dev/null; then
|
||
ok "Bay is healthy at http://127.0.0.1:$BAY_PORT"
|
||
return
|
||
fi
|
||
# Check if process is still alive
|
||
if ! kill -0 "$BAY_PID" 2>/dev/null; then
|
||
err "Bay process died unexpectedly. Check the output above."
|
||
exit 1
|
||
fi
|
||
sleep 1
|
||
waited=$((waited + 1))
|
||
done
|
||
|
||
err "Bay did not become healthy within ${max_wait}s."
|
||
err "It may still be starting — check http://127.0.0.1:$BAY_PORT/health"
|
||
}
|
||
|
||
# ── 读取 Bay 自动生成的凭证 ───────────────────────────────────
|
||
read_bay_credentials() {
|
||
local cred_file="$BAY_DIR/credentials.json"
|
||
|
||
# Wait briefly for credentials.json to appear (Bay writes it during startup)
|
||
local max_wait=5
|
||
local waited=0
|
||
while [[ $waited -lt $max_wait ]]; do
|
||
if [[ -f "$cred_file" ]]; then
|
||
break
|
||
fi
|
||
sleep 1
|
||
waited=$((waited + 1))
|
||
done
|
||
|
||
if [[ -f "$cred_file" ]]; then
|
||
# Extract api_key using python (always available) — no jq dependency
|
||
BAY_API_KEY=$(python3 -c "
|
||
import json, sys
|
||
try:
|
||
d = json.load(open('$cred_file'))
|
||
print(d.get('api_key', ''))
|
||
except Exception:
|
||
print('')
|
||
" 2>/dev/null || echo "")
|
||
|
||
if [[ -n "$BAY_API_KEY" ]]; then
|
||
ok "Auto-provisioned API key loaded from credentials.json"
|
||
else
|
||
warn "credentials.json found but api_key is empty"
|
||
fi
|
||
else
|
||
warn "credentials.json not found — Bay may be using an existing key or anonymous mode"
|
||
warn "Check Bay logs above for the API key, or look at: $cred_file"
|
||
fi
|
||
}
|
||
|
||
# ── 打印 AstrBot 配置提示 ────────────────────────────────────
|
||
print_astrbot_config_hint() {
|
||
echo ""
|
||
echo -e "${GREEN}════════════════════════════════════════════════════════════${NC}"
|
||
echo -e "${GREEN} Shipyard Neo Bay is running at http://127.0.0.1:$BAY_PORT ${NC}"
|
||
echo -e "${GREEN}════════════════════════════════════════════════════════════${NC}"
|
||
echo ""
|
||
if [[ -n "$BAY_API_KEY" ]]; then
|
||
echo -e " ${CYAN}Bay API Key (auto-generated):${NC}"
|
||
echo -e " ${YELLOW}$BAY_API_KEY${NC}"
|
||
echo ""
|
||
fi
|
||
echo -e " ${CYAN}AstrBot Dashboard 配置指引：${NC}"
|
||
echo -e " 1. AI 配置 → Agent Computer Use"
|
||
echo -e " • Computer Use Runtime → ${YELLOW}沙箱${NC}"
|
||
echo -e " • 沙箱环境驱动器 → ${YELLOW}Shipyard Neo${NC}"
|
||
echo -e " • Shipyard Neo API Endpoint → ${YELLOW}http://127.0.0.1:$BAY_PORT${NC}"
|
||
if [[ -n "$BAY_API_KEY" ]]; then
|
||
echo -e " • Shipyard Neo Access Token → ${YELLOW}$BAY_API_KEY${NC}"
|
||
else
|
||
echo -e " • Shipyard Neo Access Token → ${YELLOW}（查看 Bay 日志获取 key）${NC}"
|
||
fi
|
||
echo -e " • Shipyard Neo Profile → ${YELLOW}python-default${NC}"
|
||
echo ""
|
||
}
|
||
|
||
# ── 启动 AstrBot ──────────────────────────────────────────────
|
||
start_astrbot() {
|
||
log "Starting AstrBot..."
|
||
cd "$ASTRBOT_DIR"
|
||
uv run main.py
|
||
}
|
||
|
||
# ── 主流程 ────────────────────────────────────────────────────
|
||
main() {
|
||
echo ""
|
||
echo -e "${CYAN}╔══════════════════════════════════════════╗${NC}"
|
||
echo -e "${CYAN}║ Shipyard Neo + AstrBot Quick Start ║${NC}"
|
||
echo -e "${CYAN}╚══════════════════════════════════════════╝${NC}"
|
||
echo ""
|
||
|
||
check_prerequisites
|
||
ensure_bay_config
|
||
ensure_ship_image
|
||
start_bay
|
||
read_bay_credentials
|
||
print_astrbot_config_hint
|
||
start_astrbot
|
||
}
|
||
|
||
main "$@"
|