mirror of
https://github.com/OpenRouterTeam/spawn.git
synced 2026-04-30 04:49:33 +00:00
d5461adc16
* feat: SPAWN_CLI_DIR env var to force local source in e2e and shell scripts When SPAWN_CLI_DIR is set, the entire toolchain uses local TypeScript source instead of downloading pre-bundled scripts from GitHub releases: - e2e.sh: auto-sets SPAWN_CLI_DIR to repo root when running locally - provision.sh: exports SPAWN_CLI_DIR into the headless subshell - commands.ts: reads local shell scripts instead of fetching from CDN - All 36 cloud/agent shell scripts: exec local main.ts when set This enables e2e tests to validate local changes before they're released. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> * fix(security): add path traversal defense to SPAWN_CLI_DIR script loading Canonicalize the path via realpathSync and verify it stays inside the resolved CLI directory before reading. Prevents SPAWN_CLI_DIR from being used to read arbitrary files via ../ traversal. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> * fix(security): harden SPAWN_CLI_DIR path traversal defense - Validate cloud/agent names don't contain '..', '/' or '\' before constructing file paths - Fix root-directory edge case in prefix check by handling trailing separator correctly Agent: pr-maintainer Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> --------- Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com> Co-authored-by: B <6723574+louisgv@users.noreply.github.com>
84 lines
3.4 KiB
Bash
Executable file
84 lines
3.4 KiB
Bash
Executable file
#!/bin/bash
|
|
set -eo pipefail
|
|
|
|
# Thin shim: ensures bun is available, runs bundled digitalocean.js (local or from GitHub release)
|
|
# Includes restart loop for SIGTERM recovery on DigitalOcean
|
|
|
|
_AGENT_NAME="kilocode"
|
|
_MAX_RETRIES=3
|
|
|
|
_ensure_bun() {
|
|
if command -v bun &>/dev/null; then return 0; fi
|
|
printf '\033[0;36mInstalling bun...\033[0m\n' >&2
|
|
curl -fsSL --proto '=https' --show-error https://bun.sh/install | bash >/dev/null || { printf '\033[0;31mFailed to install bun\033[0m\n' >&2; exit 1; }
|
|
export PATH="$HOME/.bun/bin:$PATH"
|
|
command -v bun &>/dev/null || { printf '\033[0;31mbun not found after install\033[0m\n' >&2; exit 1; }
|
|
}
|
|
|
|
# Run command in the foreground so bun gets full terminal access (raw mode,
|
|
# arrow keys for interactive prompts). The old pattern backgrounded the child
|
|
# with & + wait so a SIGTERM trap could forward the signal, but that removed
|
|
# bun from the foreground process group and broke @clack/prompts multiselect.
|
|
# Now SIGTERM is detected from exit code 143 (128 + 15) after the child exits.
|
|
_run_with_restart() {
|
|
local attempt=0
|
|
local backoff=2
|
|
while [ "$attempt" -lt "$_MAX_RETRIES" ]; do
|
|
attempt=$((attempt + 1))
|
|
|
|
"$@"
|
|
local exit_code=$?
|
|
|
|
# Normal exit
|
|
if [ "$exit_code" -eq 0 ]; then
|
|
return 0
|
|
fi
|
|
|
|
# SIGTERM (143) or SIGKILL (137) — attempt restart
|
|
if [ "$exit_code" -eq 143 ] || [ "$exit_code" -eq 137 ]; then
|
|
printf '\033[0;33m[spawn/%s] Agent process terminated (exit %s). The droplet is likely still running.\033[0m\n' \
|
|
"$_AGENT_NAME" "$exit_code" >&2
|
|
printf '\033[0;33m[spawn/%s] Check your DigitalOcean dashboard: https://cloud.digitalocean.com/droplets\033[0m\n' \
|
|
"$_AGENT_NAME" >&2
|
|
if [ "$attempt" -lt "$_MAX_RETRIES" ]; then
|
|
printf '\033[0;33m[spawn/%s] Restarting (attempt %s/%s, backoff %ss)...\033[0m\n' \
|
|
"$_AGENT_NAME" "$((attempt + 1))" "$_MAX_RETRIES" "$backoff" >&2
|
|
sleep "$backoff"
|
|
backoff=$((backoff * 2))
|
|
continue
|
|
else
|
|
printf '\033[0;31m[spawn/%s] Max restart attempts reached (%s). Giving up.\033[0m\n' \
|
|
"$_AGENT_NAME" "$_MAX_RETRIES" >&2
|
|
return "$exit_code"
|
|
fi
|
|
fi
|
|
|
|
# Other failure — exit with the original code
|
|
return "$exit_code"
|
|
done
|
|
}
|
|
|
|
_ensure_bun
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" 2>/dev/null && pwd)"
|
|
|
|
# SPAWN_CLI_DIR override — force local source (used by e2e tests)
|
|
if [[ -n "${SPAWN_CLI_DIR:-}" && -f "$SPAWN_CLI_DIR/packages/cli/src/digitalocean/main.ts" ]]; then
|
|
_run_with_restart bun run "$SPAWN_CLI_DIR/packages/cli/src/digitalocean/main.ts" "$_AGENT_NAME" "$@"
|
|
exit $?
|
|
fi
|
|
|
|
# Local checkout — run from source
|
|
if [[ -n "$SCRIPT_DIR" && -f "$SCRIPT_DIR/../../packages/cli/src/digitalocean/main.ts" ]]; then
|
|
_run_with_restart bun run "$SCRIPT_DIR/../../packages/cli/src/digitalocean/main.ts" "$_AGENT_NAME" "$@"
|
|
exit $?
|
|
fi
|
|
|
|
# Remote — download bundled digitalocean.js from GitHub release
|
|
DO_JS=$(mktemp)
|
|
trap 'rm -f "$DO_JS"' EXIT
|
|
curl -fsSL --proto '=https' "https://github.com/OpenRouterTeam/spawn/releases/download/digitalocean-latest/digitalocean.js" -o "$DO_JS" \
|
|
|| { printf '\033[0;31mFailed to download digitalocean.js\033[0m\n' >&2; exit 1; }
|
|
|
|
_run_with_restart bun run "$DO_JS" "$_AGENT_NAME" "$@"
|
|
exit $?
|