mirror of
https://github.com/OpenRouterTeam/spawn.git
synced 2026-05-05 23:50:48 +00:00
91c466c008
- Add trap 'rm -f "${ENV_TEMP}"' EXIT after mktemp creation
- Scripts with DOTENV_TEMP get combined trap for both files
- Remove manual rm calls that are now redundant
- Prevents temp file leaks on early script exit (errors, signals)
- Affects 67 agent scripts across all providers
Impact: Prevents /tmp pollution in production deployments
Score: 90 (Impact: 9, Confidence: 10, Risk: 1)
56 lines
1.6 KiB
Bash
Executable file
56 lines
1.6 KiB
Bash
Executable file
#!/bin/bash
|
|
set -eo pipefail
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" 2>/dev/null && pwd)"
|
|
# shellcheck source=digitalocean/lib/common.sh
|
|
if [[ -f "${SCRIPT_DIR}/lib/common.sh" ]]; then
|
|
source "${SCRIPT_DIR}/lib/common.sh"
|
|
else
|
|
eval "$(curl -fsSL https://raw.githubusercontent.com/OpenRouterTeam/spawn/main/digitalocean/lib/common.sh)"
|
|
fi
|
|
|
|
log_info "Gemini CLI on DigitalOcean"
|
|
echo ""
|
|
|
|
ensure_do_token
|
|
ensure_ssh_key
|
|
|
|
DROPLET_NAME=$(get_server_name)
|
|
create_server "${DROPLET_NAME}"
|
|
verify_server_connectivity "${DO_SERVER_IP}"
|
|
wait_for_cloud_init "${DO_SERVER_IP}" 60
|
|
|
|
log_warn "Installing Gemini CLI..."
|
|
run_server "${DO_SERVER_IP}" "npm install -g @google/gemini-cli"
|
|
log_info "Gemini CLI installed"
|
|
|
|
echo ""
|
|
if [[ -n "${OPENROUTER_API_KEY:-}" ]]; then
|
|
log_info "Using OpenRouter API key from environment"
|
|
else
|
|
OPENROUTER_API_KEY=$(get_openrouter_api_key_oauth 5180)
|
|
fi
|
|
|
|
log_warn "Setting up environment variables..."
|
|
ENV_TEMP=$(mktemp)
|
|
trap 'rm -f "${ENV_TEMP}"' EXIT
|
|
cat > "${ENV_TEMP}" << EOF
|
|
|
|
# [spawn:env]
|
|
export OPENROUTER_API_KEY="${OPENROUTER_API_KEY}"
|
|
export GEMINI_API_KEY="${OPENROUTER_API_KEY}"
|
|
export OPENAI_API_KEY="${OPENROUTER_API_KEY}"
|
|
export OPENAI_BASE_URL="https://openrouter.ai/api/v1"
|
|
EOF
|
|
upload_file "${DO_SERVER_IP}" "${ENV_TEMP}" "/tmp/env_config"
|
|
run_server "${DO_SERVER_IP}" "cat /tmp/env_config >> ~/.zshrc && rm /tmp/env_config"
|
|
|
|
echo ""
|
|
log_info "DigitalOcean droplet setup completed successfully!"
|
|
log_info "Droplet: ${DROPLET_NAME} (ID: ${DO_DROPLET_ID}, IP: ${DO_SERVER_IP})"
|
|
echo ""
|
|
|
|
log_warn "Starting Gemini..."
|
|
sleep 1
|
|
clear
|
|
interactive_session "${DO_SERVER_IP}" "source ~/.zshrc && gemini"
|