Russell Ballestrini
ccd723db99
Changed --restore to take snapshot ID directly and call /snapshots/:id/restore instead of requiring --from SNAPSHOT_ID and calling /sessions/:id/restore or /services/:id/restore. Updated session and service restore in all implementations: - un.py, un.go, un.rb, un.sh, un.ex, un.erl, un.fs, un.hs - un.groovy, un.r, un.m, un.awk Also added snapshot management features where missing.
1172 lines
39 KiB
Bash
1172 lines
39 KiB
Bash
# PUBLIC DOMAIN - NO LICENSE, NO WARRANTY
|
|
#
|
|
# This is free public domain software for the public good of a permacomputer hosted
|
|
# at permacomputer.com - an always-on computer by the people, for the people. One
|
|
# which is durable, easy to repair, and distributed like tap water for machine
|
|
# learning intelligence.
|
|
#
|
|
# The permacomputer is community-owned infrastructure optimized around four values:
|
|
#
|
|
# TRUTH - First principles, math & science, open source code freely distributed
|
|
# FREEDOM - Voluntary partnerships, freedom from tyranny & corporate control
|
|
# HARMONY - Minimal waste, self-renewing systems with diverse thriving connections
|
|
# LOVE - Be yourself without hurting others, cooperation through natural law
|
|
#
|
|
# This software contributes to that vision by enabling code execution across 42+
|
|
# programming languages through a unified interface, accessible to all. Code is
|
|
# seeds to sprout on any abandoned technology.
|
|
#
|
|
# Learn more: https://www.permacomputer.com
|
|
#
|
|
# Anyone is free to copy, modify, publish, use, compile, sell, or distribute this
|
|
# software, either in source code form or as a compiled binary, for any purpose,
|
|
# commercial or non-commercial, and by any means.
|
|
#
|
|
# NO WARRANTY. THE SOFTWARE IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND.
|
|
#
|
|
# That said, our permacomputer's digital membrane stratum continuously runs unit,
|
|
# integration, and functional tests on all of it's own software - with our
|
|
# permacomputer monitoring itself, repairing itself, with minimal human in the
|
|
# loop guidance. Our agents do their best.
|
|
#
|
|
# Copyright 2025 TimeHexOn & foxhop & russell@unturf
|
|
# https://www.timehexon.com
|
|
# https://www.foxhop.net
|
|
# https://www.unturf.com/software
|
|
|
|
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
# un.sh - Unsandbox CLI Client (Bash Implementation)
|
|
#
|
|
# Full-featured CLI matching un.c capabilities:
|
|
# - Execute code with env vars, input files, artifacts
|
|
# - Interactive sessions with shell/REPL support
|
|
# - Persistent services with domains and ports
|
|
#
|
|
# Usage:
|
|
# un.sh [options] <source_file>
|
|
# un.sh session [options]
|
|
# un.sh service [options]
|
|
#
|
|
# Requires: UNSANDBOX_API_KEY environment variable, jq, curl
|
|
|
|
API_BASE="https://api.unsandbox.com"
|
|
PORTAL_BASE="https://unsandbox.com"
|
|
BLUE="\033[34m"
|
|
RED="\033[31m"
|
|
GREEN="\033[32m"
|
|
YELLOW="\033[33m"
|
|
RESET="\033[0m"
|
|
|
|
# Extension to language mapping
|
|
detect_language() {
|
|
local filename="$1"
|
|
local ext="${filename##*.}"
|
|
ext=$(echo "$ext" | tr '[:upper:]' '[:lower:]')
|
|
|
|
case "$ext" in
|
|
py) echo "python" ;;
|
|
js) echo "javascript" ;;
|
|
ts) echo "typescript" ;;
|
|
rb) echo "ruby" ;;
|
|
php) echo "php" ;;
|
|
pl) echo "perl" ;;
|
|
lua) echo "lua" ;;
|
|
sh) echo "bash" ;;
|
|
go) echo "go" ;;
|
|
rs) echo "rust" ;;
|
|
c) echo "c" ;;
|
|
cpp|cc|cxx) echo "cpp" ;;
|
|
java) echo "java" ;;
|
|
kt) echo "kotlin" ;;
|
|
cs) echo "csharp" ;;
|
|
fs) echo "fsharp" ;;
|
|
hs) echo "haskell" ;;
|
|
ml) echo "ocaml" ;;
|
|
clj) echo "clojure" ;;
|
|
scm) echo "scheme" ;;
|
|
lisp) echo "commonlisp" ;;
|
|
erl) echo "erlang" ;;
|
|
ex|exs) echo "elixir" ;;
|
|
jl) echo "julia" ;;
|
|
r|R) echo "r" ;;
|
|
cr) echo "crystal" ;;
|
|
d) echo "d" ;;
|
|
nim) echo "nim" ;;
|
|
zig) echo "zig" ;;
|
|
v) echo "v" ;;
|
|
dart) echo "dart" ;;
|
|
groovy) echo "groovy" ;;
|
|
scala) echo "scala" ;;
|
|
f90|f95) echo "fortran" ;;
|
|
cob) echo "cobol" ;;
|
|
pro) echo "prolog" ;;
|
|
forth|4th) echo "forth" ;;
|
|
tcl) echo "tcl" ;;
|
|
raku) echo "raku" ;;
|
|
m) echo "objc" ;;
|
|
*)
|
|
# Try shebang
|
|
if [[ -f "$filename" ]]; then
|
|
local first_line=$(head -n1 "$filename")
|
|
if [[ "$first_line" =~ ^#! ]]; then
|
|
[[ "$first_line" =~ python ]] && echo "python" && return
|
|
[[ "$first_line" =~ node ]] && echo "javascript" && return
|
|
[[ "$first_line" =~ ruby ]] && echo "ruby" && return
|
|
[[ "$first_line" =~ perl ]] && echo "perl" && return
|
|
[[ "$first_line" =~ (bash|/sh) ]] && echo "bash" && return
|
|
[[ "$first_line" =~ lua ]] && echo "lua" && return
|
|
[[ "$first_line" =~ php ]] && echo "php" && return
|
|
fi
|
|
fi
|
|
echo -e "${RED}Error: Cannot detect language for $filename${RESET}" >&2
|
|
exit 1
|
|
;;
|
|
esac
|
|
}
|
|
|
|
api_request() {
|
|
local endpoint="$1"
|
|
local method="${2:-GET}"
|
|
local data="${3:-}"
|
|
local public_key="${4:-${UNSANDBOX_PUBLIC_KEY:-}}"
|
|
local secret_key="${5:-${UNSANDBOX_SECRET_KEY:-}}"
|
|
|
|
# Fallback to old UNSANDBOX_API_KEY for backwards compat
|
|
if [[ -z "$public_key" ]] && [[ -n "${UNSANDBOX_API_KEY:-}" ]]; then
|
|
public_key="${UNSANDBOX_API_KEY}"
|
|
secret_key=""
|
|
fi
|
|
|
|
if [[ -z "$public_key" ]]; then
|
|
echo -e "${RED}Error: UNSANDBOX_PUBLIC_KEY or UNSANDBOX_API_KEY not set${RESET}" >&2
|
|
exit 1
|
|
fi
|
|
|
|
local url="${API_BASE}${endpoint}"
|
|
local timestamp=$(date +%s)
|
|
local body="${data:-}"
|
|
|
|
# Build HMAC signature: timestamp:METHOD:path:body
|
|
local sig_input="${timestamp}:${method}:${endpoint}:${body}"
|
|
local signature=""
|
|
|
|
if [[ -n "$secret_key" ]]; then
|
|
signature=$(echo -n "$sig_input" | openssl dgst -sha256 -hmac "$secret_key" | sed 's/^.* //')
|
|
fi
|
|
|
|
if [[ -n "$data" ]]; then
|
|
local response
|
|
if [[ -n "$signature" ]]; then
|
|
response=$(curl -s -w "\n%{http_code}" -X "$method" "$url" \
|
|
-H "Authorization: Bearer $public_key" \
|
|
-H "X-Timestamp: $timestamp" \
|
|
-H "X-Signature: $signature" \
|
|
-H "Content-Type: application/json" \
|
|
-d "$data" 2>&1)
|
|
else
|
|
response=$(curl -s -w "\n%{http_code}" -X "$method" "$url" \
|
|
-H "Authorization: Bearer $public_key" \
|
|
-H "Content-Type: application/json" \
|
|
-d "$data" 2>&1)
|
|
fi
|
|
else
|
|
local response
|
|
if [[ -n "$signature" ]]; then
|
|
response=$(curl -s -w "\n%{http_code}" -X "$method" "$url" \
|
|
-H "Authorization: Bearer $public_key" \
|
|
-H "X-Timestamp: $timestamp" \
|
|
-H "X-Signature: $signature" \
|
|
-H "Content-Type: application/json" 2>&1)
|
|
else
|
|
response=$(curl -s -w "\n%{http_code}" -X "$method" "$url" \
|
|
-H "Authorization: Bearer $public_key" \
|
|
-H "Content-Type: application/json" 2>&1)
|
|
fi
|
|
fi
|
|
|
|
local http_code=$(echo "$response" | tail -n1)
|
|
local body=$(echo "$response" | head -n-1)
|
|
|
|
if [[ "$http_code" -lt 200 || "$http_code" -ge 300 ]]; then
|
|
if [[ "$http_code" == "401" ]] && echo "$body" | grep -qi "timestamp"; then
|
|
echo -e "${RED}Error: Request timestamp expired (must be within 5 minutes of server time)${RESET}" >&2
|
|
echo -e "${YELLOW}Your computer's clock may have drifted.${RESET}" >&2
|
|
echo -e "${YELLOW}Check your system time and sync with NTP if needed:${RESET}" >&2
|
|
echo -e " Linux: sudo ntpdate -s time.nist.gov" >&2
|
|
echo -e " macOS: sudo sntp -sS time.apple.com" >&2
|
|
echo -e " Windows: w32tm /resync" >&2
|
|
else
|
|
echo -e "${RED}Error: HTTP $http_code - $body${RESET}" >&2
|
|
fi
|
|
exit 1
|
|
fi
|
|
|
|
echo "$body"
|
|
}
|
|
|
|
cmd_execute() {
|
|
local source_file=""
|
|
local -a env_vars=()
|
|
local -a input_files=()
|
|
local artifacts=false
|
|
local output_dir="."
|
|
local network=""
|
|
local vcpu=""
|
|
local api_key="${UNSANDBOX_API_KEY:-}"
|
|
|
|
# Parse arguments
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
-e)
|
|
env_vars+=("$2")
|
|
shift 2
|
|
;;
|
|
-f)
|
|
input_files+=("$2")
|
|
shift 2
|
|
;;
|
|
-a)
|
|
artifacts=true
|
|
shift
|
|
;;
|
|
-o)
|
|
output_dir="$2"
|
|
shift 2
|
|
;;
|
|
-n)
|
|
network="$2"
|
|
shift 2
|
|
;;
|
|
-v)
|
|
vcpu="$2"
|
|
shift 2
|
|
;;
|
|
-k)
|
|
api_key="$2"
|
|
shift 2
|
|
;;
|
|
-*)
|
|
echo -e "${RED}Unknown option: $1${RESET}" >&2
|
|
exit 1
|
|
;;
|
|
*)
|
|
source_file="$1"
|
|
shift
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if [[ ! -f "$source_file" ]]; then
|
|
echo -e "${RED}Error: File not found: $source_file${RESET}" >&2
|
|
exit 1
|
|
fi
|
|
|
|
local code=$(cat "$source_file")
|
|
local language=$(detect_language "$source_file")
|
|
|
|
# Build JSON payload
|
|
local payload=$(jq -n \
|
|
--arg lang "$language" \
|
|
--arg code "$code" \
|
|
'{language: $lang, code: $code}')
|
|
|
|
# Add environment variables
|
|
if [[ ${#env_vars[@]} -gt 0 ]]; then
|
|
local env_json="{"
|
|
for env_var in "${env_vars[@]}"; do
|
|
local key="${env_var%%=*}"
|
|
local val="${env_var#*=}"
|
|
env_json+="\"$key\":\"$val\","
|
|
done
|
|
env_json="${env_json%,}}"
|
|
payload=$(echo "$payload" | jq --argjson env "$env_json" '. + {env: $env}')
|
|
fi
|
|
|
|
# Add input files
|
|
if [[ ${#input_files[@]} -gt 0 ]]; then
|
|
local files_json="["
|
|
for file in "${input_files[@]}"; do
|
|
if [[ ! -f "$file" ]]; then
|
|
echo -e "${RED}Error: Input file not found: $file${RESET}" >&2
|
|
exit 1
|
|
fi
|
|
local filename=$(basename "$file")
|
|
local content_b64=$(base64 -w0 < "$file")
|
|
files_json+="{\"filename\":\"$filename\",\"content_base64\":\"$content_b64\"},"
|
|
done
|
|
files_json="${files_json%,}]"
|
|
payload=$(echo "$payload" | jq --argjson files "$files_json" '. + {input_files: $files}')
|
|
fi
|
|
|
|
# Add options
|
|
[[ "$artifacts" == true ]] && payload=$(echo "$payload" | jq '. + {return_artifacts: true}')
|
|
[[ -n "$network" ]] && payload=$(echo "$payload" | jq --arg n "$network" '. + {network: $n}')
|
|
[[ -n "$vcpu" ]] && payload=$(echo "$payload" | jq --argjson v "$vcpu" '. + {vcpu: $v}')
|
|
|
|
# Execute
|
|
local result=$(api_request "/execute" "POST" "$payload" "$api_key")
|
|
|
|
# Print output
|
|
local stdout=$(echo "$result" | jq -r '.stdout // empty')
|
|
local stderr=$(echo "$result" | jq -r '.stderr // empty')
|
|
[[ -n "$stdout" ]] && echo -e "${BLUE}${stdout}${RESET}"
|
|
[[ -n "$stderr" ]] && echo -e "${RED}${stderr}${RESET}" >&2
|
|
|
|
# Save artifacts
|
|
if [[ "$artifacts" == true ]]; then
|
|
local artifacts_json=$(echo "$result" | jq -r '.artifacts // []')
|
|
if [[ "$artifacts_json" != "[]" ]]; then
|
|
mkdir -p "$output_dir"
|
|
local num_artifacts=$(echo "$artifacts_json" | jq 'length')
|
|
for ((i=0; i<num_artifacts; i++)); do
|
|
local artifact=$(echo "$artifacts_json" | jq -r ".[$i]")
|
|
local filename=$(echo "$artifact" | jq -r '.filename // "artifact"')
|
|
local content_b64=$(echo "$artifact" | jq -r '.content_base64')
|
|
local filepath="$output_dir/$filename"
|
|
echo "$content_b64" | base64 -d > "$filepath"
|
|
chmod 755 "$filepath"
|
|
echo -e "${GREEN}Saved: $filepath${RESET}" >&2
|
|
done
|
|
fi
|
|
fi
|
|
|
|
local exit_code=$(echo "$result" | jq -r '.exit_code // 0')
|
|
exit "$exit_code"
|
|
}
|
|
|
|
cmd_session() {
|
|
local shell="bash"
|
|
local list=false
|
|
local attach=""
|
|
local kill=""
|
|
local audit=false
|
|
local tmux=false
|
|
local screen=false
|
|
local network=""
|
|
local vcpu=""
|
|
local api_key="${UNSANDBOX_API_KEY:-}"
|
|
local -a input_files=()
|
|
local snapshot=""
|
|
local restore=""
|
|
local from_snapshot=""
|
|
local snapshot_name=""
|
|
local hot=false
|
|
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
-s|--shell)
|
|
shell="$2"
|
|
shift 2
|
|
;;
|
|
-l|--list)
|
|
list=true
|
|
shift
|
|
;;
|
|
--attach)
|
|
attach="$2"
|
|
shift 2
|
|
;;
|
|
--kill)
|
|
kill="$2"
|
|
shift 2
|
|
;;
|
|
--audit)
|
|
audit=true
|
|
shift
|
|
;;
|
|
--tmux)
|
|
tmux=true
|
|
shift
|
|
;;
|
|
--screen)
|
|
screen=true
|
|
shift
|
|
;;
|
|
--snapshot)
|
|
snapshot="$2"
|
|
shift 2
|
|
;;
|
|
--restore)
|
|
restore="$2"
|
|
shift 2
|
|
;;
|
|
--from)
|
|
from_snapshot="$2"
|
|
shift 2
|
|
;;
|
|
--snapshot-name)
|
|
snapshot_name="$2"
|
|
shift 2
|
|
;;
|
|
--hot)
|
|
hot=true
|
|
shift
|
|
;;
|
|
-f)
|
|
input_files+=("$2")
|
|
shift 2
|
|
;;
|
|
-n)
|
|
network="$2"
|
|
shift 2
|
|
;;
|
|
-v)
|
|
vcpu="$2"
|
|
shift 2
|
|
;;
|
|
-k)
|
|
api_key="$2"
|
|
shift 2
|
|
;;
|
|
-*)
|
|
echo -e "${RED}Unknown option: $1${RESET}" >&2
|
|
exit 1
|
|
;;
|
|
*)
|
|
shift
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if [[ "$list" == true ]]; then
|
|
local result=$(api_request "/sessions" "GET" "" "$api_key")
|
|
local sessions=$(echo "$result" | jq -r '.sessions // []')
|
|
if [[ "$sessions" == "[]" ]]; then
|
|
echo "No active sessions"
|
|
else
|
|
printf "%-40s %-10s %-10s %s\n" "ID" "Shell" "Status" "Created"
|
|
echo "$sessions" | jq -r '.[] | "\(.id // "N/A") \(.shell // "N/A") \(.status // "N/A") \(.created_at // "N/A")"' | \
|
|
while read -r id sh status created; do
|
|
printf "%-40s %-10s %-10s %s\n" "$id" "$sh" "$status" "$created"
|
|
done
|
|
fi
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$kill" ]]; then
|
|
api_request "/sessions/$kill" "DELETE" "" "$api_key" > /dev/null
|
|
echo -e "${GREEN}Session terminated: $kill${RESET}"
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$snapshot" ]]; then
|
|
local payload=$(jq -n --arg name "$snapshot_name" --argjson hot "$hot" '{name: $name, hot: $hot}')
|
|
echo -e "${YELLOW}Creating snapshot of session $snapshot...${RESET}"
|
|
local result=$(api_request "/sessions/$snapshot/snapshot" "POST" "$payload" "$api_key")
|
|
local snapshot_id=$(echo "$result" | jq -r '.id // "N/A"')
|
|
echo -e "${GREEN}Snapshot created successfully${RESET}"
|
|
echo "Snapshot ID: $snapshot_id"
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$restore" ]]; then
|
|
# --restore takes snapshot ID directly, calls /snapshots/:id/restore
|
|
echo -e "${YELLOW}Restoring from snapshot $restore...${RESET}"
|
|
local result=$(api_request "/snapshots/$restore/restore" "POST" "{}" "$api_key")
|
|
echo -e "${GREEN}Session restored from snapshot${RESET}"
|
|
local new_id=$(echo "$result" | jq -r '.session_id // empty')
|
|
[[ -n "$new_id" ]] && echo "New session ID: $new_id"
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$attach" ]]; then
|
|
echo -e "${YELLOW}Attaching to session $attach...${RESET}"
|
|
echo -e "${YELLOW}(Interactive sessions require WebSocket - use un2 for full support)${RESET}"
|
|
return
|
|
fi
|
|
|
|
# Create session
|
|
local payload=$(jq -n --arg sh "$shell" '{shell: $sh}')
|
|
[[ -n "$network" ]] && payload=$(echo "$payload" | jq --arg n "$network" '. + {network: $n}')
|
|
[[ -n "$vcpu" ]] && payload=$(echo "$payload" | jq --argjson v "$vcpu" '. + {vcpu: $v}')
|
|
[[ "$tmux" == true ]] && payload=$(echo "$payload" | jq '. + {persistence: "tmux"}')
|
|
[[ "$screen" == true ]] && payload=$(echo "$payload" | jq '. + {persistence: "screen"}')
|
|
[[ "$audit" == true ]] && payload=$(echo "$payload" | jq '. + {audit: true}')
|
|
|
|
# Add input files
|
|
if [[ ${#input_files[@]} -gt 0 ]]; then
|
|
local files_json="["
|
|
for file in "${input_files[@]}"; do
|
|
if [[ ! -f "$file" ]]; then
|
|
echo -e "${RED}Error: Input file not found: $file${RESET}" >&2
|
|
exit 1
|
|
fi
|
|
local filename=$(basename "$file")
|
|
local content_b64=$(base64 -w0 < "$file")
|
|
files_json+="{\"filename\":\"$filename\",\"content_base64\":\"$content_b64\"},"
|
|
done
|
|
files_json="${files_json%,}]"
|
|
payload=$(echo "$payload" | jq --argjson files "$files_json" '. + {input_files: $files}')
|
|
fi
|
|
|
|
echo -e "${YELLOW}Creating session...${RESET}"
|
|
local result=$(api_request "/sessions" "POST" "$payload" "$api_key")
|
|
local session_id=$(echo "$result" | jq -r '.id // "N/A"')
|
|
echo -e "${GREEN}Session created: $session_id${RESET}"
|
|
echo -e "${YELLOW}(Interactive sessions require WebSocket - use un2 for full support)${RESET}"
|
|
}
|
|
|
|
cmd_service() {
|
|
local name=""
|
|
local ports=""
|
|
local domains=""
|
|
local service_type=""
|
|
local bootstrap=""
|
|
local bootstrap_file=""
|
|
local list=false
|
|
local info=""
|
|
local logs=""
|
|
local tail=""
|
|
local sleep=""
|
|
local wake=""
|
|
local destroy=""
|
|
local execute=""
|
|
local command=""
|
|
local network=""
|
|
local vcpu=""
|
|
local api_key="${UNSANDBOX_API_KEY:-}"
|
|
local -a input_files=()
|
|
local snapshot=""
|
|
local restore=""
|
|
local from_snapshot=""
|
|
local snapshot_name=""
|
|
local hot=false
|
|
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
--name)
|
|
name="$2"
|
|
shift 2
|
|
;;
|
|
--ports)
|
|
ports="$2"
|
|
shift 2
|
|
;;
|
|
--domains)
|
|
domains="$2"
|
|
shift 2
|
|
;;
|
|
--type)
|
|
service_type="$2"
|
|
shift 2
|
|
;;
|
|
--bootstrap)
|
|
bootstrap="$2"
|
|
shift 2
|
|
;;
|
|
--bootstrap-file)
|
|
bootstrap_file="$2"
|
|
shift 2
|
|
;;
|
|
-f)
|
|
input_files+=("$2")
|
|
shift 2
|
|
;;
|
|
-l|--list)
|
|
list=true
|
|
shift
|
|
;;
|
|
--info)
|
|
info="$2"
|
|
shift 2
|
|
;;
|
|
--logs)
|
|
logs="$2"
|
|
shift 2
|
|
;;
|
|
--tail)
|
|
tail="$2"
|
|
shift 2
|
|
;;
|
|
--freeze)
|
|
sleep="$2"
|
|
shift 2
|
|
;;
|
|
--unfreeze)
|
|
wake="$2"
|
|
shift 2
|
|
;;
|
|
--destroy)
|
|
destroy="$2"
|
|
shift 2
|
|
;;
|
|
--execute)
|
|
execute="$2"
|
|
shift 2
|
|
;;
|
|
--command)
|
|
command="$2"
|
|
shift 2
|
|
;;
|
|
--dump-bootstrap)
|
|
dump_bootstrap="$2"
|
|
shift 2
|
|
;;
|
|
--dump-file)
|
|
dump_file="$2"
|
|
shift 2
|
|
;;
|
|
--snapshot)
|
|
snapshot="$2"
|
|
shift 2
|
|
;;
|
|
--restore)
|
|
restore="$2"
|
|
shift 2
|
|
;;
|
|
--from)
|
|
from_snapshot="$2"
|
|
shift 2
|
|
;;
|
|
--snapshot-name)
|
|
snapshot_name="$2"
|
|
shift 2
|
|
;;
|
|
--hot)
|
|
hot=true
|
|
shift
|
|
;;
|
|
-n)
|
|
network="$2"
|
|
shift 2
|
|
;;
|
|
-v)
|
|
vcpu="$2"
|
|
shift 2
|
|
;;
|
|
-k)
|
|
api_key="$2"
|
|
shift 2
|
|
;;
|
|
-*)
|
|
echo -e "${RED}Unknown option: $1${RESET}" >&2
|
|
exit 1
|
|
;;
|
|
*)
|
|
shift
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if [[ "$list" == true ]]; then
|
|
local result=$(api_request "/services" "GET" "" "$api_key")
|
|
local services=$(echo "$result" | jq -r '.services // []')
|
|
if [[ "$services" == "[]" ]]; then
|
|
echo "No services"
|
|
else
|
|
printf "%-20s %-15s %-10s %-15s %s\n" "ID" "Name" "Status" "Ports" "Domains"
|
|
echo "$services" | jq -r '.[] | "\(.id // "N/A") \(.name // "N/A") \(.status // "N/A") \((.ports // []) | join(",")) \((.domains // []) | join(","))"' | \
|
|
while read -r id name status ports domains; do
|
|
printf "%-20s %-15s %-10s %-15s %s\n" "$id" "$name" "$status" "$ports" "$domains"
|
|
done
|
|
fi
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$snapshot" ]]; then
|
|
local payload=$(jq -n --arg name "$snapshot_name" --argjson hot "$hot" '{name: $name, hot: $hot}')
|
|
echo -e "${YELLOW}Creating snapshot of service $snapshot...${RESET}"
|
|
local result=$(api_request "/services/$snapshot/snapshot" "POST" "$payload" "$api_key")
|
|
local snapshot_id=$(echo "$result" | jq -r '.id // "N/A"')
|
|
echo -e "${GREEN}Snapshot created successfully${RESET}"
|
|
echo "Snapshot ID: $snapshot_id"
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$restore" ]]; then
|
|
# --restore takes snapshot ID directly, calls /snapshots/:id/restore
|
|
echo -e "${YELLOW}Restoring from snapshot $restore...${RESET}"
|
|
local result=$(api_request "/snapshots/$restore/restore" "POST" "{}" "$api_key")
|
|
echo -e "${GREEN}Service restored from snapshot${RESET}"
|
|
local new_id=$(echo "$result" | jq -r '.service_id // empty')
|
|
[[ -n "$new_id" ]] && echo "New service ID: $new_id"
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$info" ]]; then
|
|
local result=$(api_request "/services/$info" "GET" "" "$api_key")
|
|
echo "$result" | jq '.'
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$logs" ]]; then
|
|
local result=$(api_request "/services/$logs/logs" "GET" "" "$api_key")
|
|
echo "$result" | jq -r '.logs // ""'
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$tail" ]]; then
|
|
local result=$(api_request "/services/$tail/logs?lines=9000" "GET" "" "$api_key")
|
|
echo "$result" | jq -r '.logs // ""'
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$sleep" ]]; then
|
|
api_request "/services/$sleep/sleep" "POST" "" "$api_key" > /dev/null
|
|
echo -e "${GREEN}Service sleeping: $sleep${RESET}"
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$wake" ]]; then
|
|
api_request "/services/$wake/wake" "POST" "" "$api_key" > /dev/null
|
|
echo -e "${GREEN}Service waking: $wake${RESET}"
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$destroy" ]]; then
|
|
api_request "/services/$destroy" "DELETE" "" "$api_key" > /dev/null
|
|
echo -e "${GREEN}Service destroyed: $destroy${RESET}"
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$execute" ]]; then
|
|
local payload=$(jq -n --arg cmd "$command" '{command: $cmd}')
|
|
local result=$(api_request "/services/$execute/execute" "POST" "$payload" "$api_key")
|
|
local stdout=$(echo "$result" | jq -r '.stdout // empty')
|
|
local stderr=$(echo "$result" | jq -r '.stderr // empty')
|
|
[[ -n "$stdout" ]] && echo -e "${BLUE}${stdout}${RESET}"
|
|
[[ -n "$stderr" ]] && echo -e "${RED}${stderr}${RESET}" >&2
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$dump_bootstrap" ]]; then
|
|
echo "Fetching bootstrap script from $dump_bootstrap..." >&2
|
|
local payload=$(jq -n '{command: "cat /tmp/bootstrap.sh"}')
|
|
local result=$(api_request "/services/$dump_bootstrap/execute" "POST" "$payload" "$api_key")
|
|
local bootstrap=$(echo "$result" | jq -r '.stdout // empty')
|
|
|
|
if [[ -n "$bootstrap" ]]; then
|
|
if [[ -n "$dump_file" ]]; then
|
|
# Write to file
|
|
echo "$bootstrap" > "$dump_file"
|
|
chmod 755 "$dump_file"
|
|
echo "Bootstrap saved to $dump_file"
|
|
else
|
|
# Print to stdout
|
|
echo -n "$bootstrap"
|
|
fi
|
|
else
|
|
echo -e "${RED}Error: Failed to fetch bootstrap (service not running or no bootstrap file)${RESET}" >&2
|
|
exit 1
|
|
fi
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$name" ]]; then
|
|
local payload=$(jq -n --arg n "$name" '{name: $n}')
|
|
|
|
if [[ -n "$ports" ]]; then
|
|
local ports_json="[$(echo "$ports" | sed 's/,/,/g')]"
|
|
payload=$(echo "$payload" | jq --argjson p "$ports_json" '. + {ports: $p}')
|
|
fi
|
|
|
|
if [[ -n "$domains" ]]; then
|
|
local domains_json="[\"$(echo "$domains" | sed 's/,/","/g')\"]"
|
|
payload=$(echo "$payload" | jq --argjson d "$domains_json" '. + {domains: $d}')
|
|
fi
|
|
|
|
if [[ -n "$service_type" ]]; then
|
|
payload=$(echo "$payload" | jq --arg t "$service_type" '. + {service_type: $t}')
|
|
fi
|
|
|
|
if [[ -n "$bootstrap" ]]; then
|
|
payload=$(echo "$payload" | jq --arg b "$bootstrap" '. + {bootstrap: $b}')
|
|
fi
|
|
|
|
if [[ -n "$bootstrap_file" ]]; then
|
|
if [[ ! -f "$bootstrap_file" ]]; then
|
|
echo -e "${RED}Error: Bootstrap file not found: $bootstrap_file${RESET}" >&2
|
|
return 1
|
|
fi
|
|
local file_content=$(cat "$bootstrap_file")
|
|
payload=$(echo "$payload" | jq --arg b "$file_content" '. + {bootstrap_content: $b}')
|
|
fi
|
|
|
|
# Add input files
|
|
if [[ ${#input_files[@]} -gt 0 ]]; then
|
|
local files_json="["
|
|
for file in "${input_files[@]}"; do
|
|
if [[ ! -f "$file" ]]; then
|
|
echo -e "${RED}Error: Input file not found: $file${RESET}" >&2
|
|
exit 1
|
|
fi
|
|
local filename=$(basename "$file")
|
|
local content_b64=$(base64 -w0 < "$file")
|
|
files_json+="{\"filename\":\"$filename\",\"content_base64\":\"$content_b64\"},"
|
|
done
|
|
files_json="${files_json%,}]"
|
|
payload=$(echo "$payload" | jq --argjson files "$files_json" '. + {input_files: $files}')
|
|
fi
|
|
|
|
[[ -n "$network" ]] && payload=$(echo "$payload" | jq --arg n "$network" '. + {network: $n}')
|
|
[[ -n "$vcpu" ]] && payload=$(echo "$payload" | jq --argjson v "$vcpu" '. + {vcpu: $v}')
|
|
|
|
local result=$(api_request "/services" "POST" "$payload" "$api_key")
|
|
local service_id=$(echo "$result" | jq -r '.id // "N/A"')
|
|
local service_name=$(echo "$result" | jq -r '.name // "N/A"')
|
|
local service_url=$(echo "$result" | jq -r '.url // ""')
|
|
|
|
echo -e "${GREEN}Service created: $service_id${RESET}"
|
|
echo "Name: $service_name"
|
|
[[ -n "$service_url" ]] && echo "URL: $service_url"
|
|
return
|
|
fi
|
|
|
|
echo -e "${RED}Error: Specify --name to create a service, or use --list, --info, etc.${RESET}" >&2
|
|
exit 1
|
|
}
|
|
|
|
cmd_snapshot() {
|
|
local api_key="${UNSANDBOX_API_KEY:-}"
|
|
local list=false
|
|
local info=""
|
|
local delete=""
|
|
local clone=""
|
|
local clone_type=""
|
|
local clone_name=""
|
|
local clone_shell=""
|
|
local clone_ports=""
|
|
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
-l|--list)
|
|
list=true
|
|
shift
|
|
;;
|
|
--info)
|
|
info="$2"
|
|
shift 2
|
|
;;
|
|
--delete)
|
|
delete="$2"
|
|
shift 2
|
|
;;
|
|
--clone)
|
|
clone="$2"
|
|
shift 2
|
|
;;
|
|
--type)
|
|
clone_type="$2"
|
|
shift 2
|
|
;;
|
|
--name)
|
|
clone_name="$2"
|
|
shift 2
|
|
;;
|
|
--shell)
|
|
clone_shell="$2"
|
|
shift 2
|
|
;;
|
|
--ports)
|
|
clone_ports="$2"
|
|
shift 2
|
|
;;
|
|
-k)
|
|
api_key="$2"
|
|
shift 2
|
|
;;
|
|
-*)
|
|
echo -e "${RED}Unknown option: $1${RESET}" >&2
|
|
exit 1
|
|
;;
|
|
*)
|
|
shift
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if [[ "$list" == true ]]; then
|
|
local result=$(api_request "/snapshots" "GET" "" "$api_key")
|
|
echo "$result" | jq '.'
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$info" ]]; then
|
|
local result=$(api_request "/snapshots/$info" "GET" "" "$api_key")
|
|
echo "$result" | jq '.'
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$delete" ]]; then
|
|
api_request "/snapshots/$delete" "DELETE" "" "$api_key" > /dev/null
|
|
echo -e "${GREEN}Snapshot deleted successfully${RESET}"
|
|
return
|
|
fi
|
|
|
|
if [[ -n "$clone" ]]; then
|
|
if [[ -z "$clone_type" ]]; then
|
|
echo -e "${RED}Error: --type required with --clone (session or service)${RESET}" >&2
|
|
exit 1
|
|
fi
|
|
|
|
local payload=$(jq -n --arg type "$clone_type" '{type: $type}')
|
|
[[ -n "$clone_name" ]] && payload=$(echo "$payload" | jq --arg n "$clone_name" '. + {name: $n}')
|
|
[[ -n "$clone_shell" ]] && payload=$(echo "$payload" | jq --arg s "$clone_shell" '. + {shell: $s}')
|
|
if [[ -n "$clone_ports" ]]; then
|
|
local ports_json="[$(echo "$clone_ports" | sed 's/,/,/g')]"
|
|
payload=$(echo "$payload" | jq --argjson p "$ports_json" '. + {ports: $p}')
|
|
fi
|
|
|
|
echo -e "${YELLOW}Cloning snapshot $clone to create new $clone_type...${RESET}"
|
|
local result=$(api_request "/snapshots/$clone/clone" "POST" "$payload" "$api_key")
|
|
|
|
if [[ "$clone_type" == "session" ]]; then
|
|
local session_id=$(echo "$result" | jq -r '.session_id // "N/A"')
|
|
echo -e "${GREEN}Session created from snapshot${RESET}"
|
|
echo "Session ID: $session_id"
|
|
else
|
|
local service_id=$(echo "$result" | jq -r '.service_id // "N/A"')
|
|
echo -e "${GREEN}Service created from snapshot${RESET}"
|
|
echo "Service ID: $service_id"
|
|
fi
|
|
return
|
|
fi
|
|
|
|
echo -e "${RED}Error: Specify --list, --info, --delete, or --clone${RESET}" >&2
|
|
exit 1
|
|
}
|
|
|
|
validate_key() {
|
|
local api_key="$1"
|
|
local extend_mode="$2"
|
|
|
|
if [[ -z "$api_key" ]]; then
|
|
echo -e "${RED}Error: API key not provided. Use -k flag or set UNSANDBOX_API_KEY${RESET}" >&2
|
|
exit 1
|
|
fi
|
|
|
|
# Call portal validation endpoint
|
|
local response
|
|
local http_code
|
|
response=$(curl -s -w "\n%{http_code}" -X POST "${PORTAL_BASE}/keys/validate" \
|
|
-H "Authorization: Bearer $api_key" \
|
|
-H "Content-Type: application/json" 2>&1)
|
|
|
|
http_code=$(echo "$response" | tail -n1)
|
|
local body=$(echo "$response" | head -n-1)
|
|
|
|
if [[ "$http_code" -eq 200 ]]; then
|
|
# Valid key - parse response
|
|
if command -v jq &> /dev/null; then
|
|
# Use jq for parsing
|
|
local valid=$(echo "$body" | jq -r '.valid // false')
|
|
local public_key=$(echo "$body" | jq -r '.public_key // "N/A"')
|
|
local tier=$(echo "$body" | jq -r '.tier // "N/A"')
|
|
local expires_at=$(echo "$body" | jq -r '.expires_at // "N/A"')
|
|
local expired=$(echo "$body" | jq -r '.expired // false')
|
|
|
|
if [[ "$expired" == "true" ]]; then
|
|
echo -e "${RED}Expired${RESET}"
|
|
echo "Public Key: $public_key"
|
|
echo "Tier: $tier"
|
|
echo "Expired: $expires_at"
|
|
echo -e "${YELLOW}To renew: Visit ${PORTAL_BASE}/keys/extend${RESET}"
|
|
exit 1
|
|
else
|
|
echo -e "${GREEN}Valid${RESET}"
|
|
echo "Public Key: $public_key"
|
|
echo "Tier: $tier"
|
|
echo "Expires: $expires_at"
|
|
|
|
# If extend mode, open browser
|
|
if [[ "$extend_mode" == "true" ]]; then
|
|
local extend_url="${PORTAL_BASE}/keys/extend?pk=${public_key}"
|
|
echo -e "\n${BLUE}Opening browser to extend key...${RESET}"
|
|
|
|
# Detect platform and open browser
|
|
if command -v xdg-open &> /dev/null; then
|
|
xdg-open "$extend_url" &> /dev/null
|
|
elif command -v open &> /dev/null; then
|
|
open "$extend_url" &> /dev/null
|
|
elif command -v start &> /dev/null; then
|
|
start "$extend_url" &> /dev/null
|
|
else
|
|
echo -e "${YELLOW}Cannot detect browser opener. Visit: $extend_url${RESET}"
|
|
fi
|
|
fi
|
|
fi
|
|
else
|
|
# Fallback: use grep/sed for parsing (no jq available)
|
|
local valid=$(echo "$body" | grep -o '"valid"[[:space:]]*:[[:space:]]*[^,}]*' | sed 's/.*:[[:space:]]*//' | tr -d ' "')
|
|
local public_key=$(echo "$body" | grep -o '"public_key"[[:space:]]*:[[:space:]]*"[^"]*"' | sed 's/.*:[[:space:]]*"//' | tr -d '"')
|
|
local tier=$(echo "$body" | grep -o '"tier"[[:space:]]*:[[:space:]]*"[^"]*"' | sed 's/.*:[[:space:]]*"//' | tr -d '"')
|
|
local expires_at=$(echo "$body" | grep -o '"expires_at"[[:space:]]*:[[:space:]]*"[^"]*"' | sed 's/.*:[[:space:]]*"//' | tr -d '"')
|
|
local expired=$(echo "$body" | grep -o '"expired"[[:space:]]*:[[:space:]]*[^,}]*' | sed 's/.*:[[:space:]]*//' | tr -d ' "')
|
|
|
|
[[ -z "$public_key" ]] && public_key="N/A"
|
|
[[ -z "$tier" ]] && tier="N/A"
|
|
[[ -z "$expires_at" ]] && expires_at="N/A"
|
|
|
|
if [[ "$expired" == "true" ]]; then
|
|
echo -e "${RED}Expired${RESET}"
|
|
echo "Public Key: $public_key"
|
|
echo "Tier: $tier"
|
|
echo "Expired: $expires_at"
|
|
echo -e "${YELLOW}To renew: Visit ${PORTAL_BASE}/keys/extend${RESET}"
|
|
exit 1
|
|
else
|
|
echo -e "${GREEN}Valid${RESET}"
|
|
echo "Public Key: $public_key"
|
|
echo "Tier: $tier"
|
|
echo "Expires: $expires_at"
|
|
|
|
# If extend mode, open browser
|
|
if [[ "$extend_mode" == "true" ]]; then
|
|
local extend_url="${PORTAL_BASE}/keys/extend?pk=${public_key}"
|
|
echo -e "\n${BLUE}Opening browser to extend key...${RESET}"
|
|
|
|
# Detect platform and open browser
|
|
if command -v xdg-open &> /dev/null; then
|
|
xdg-open "$extend_url" &> /dev/null
|
|
elif command -v open &> /dev/null; then
|
|
open "$extend_url" &> /dev/null
|
|
elif command -v start &> /dev/null; then
|
|
start "$extend_url" &> /dev/null
|
|
else
|
|
echo -e "${YELLOW}Cannot detect browser opener. Visit: $extend_url${RESET}"
|
|
fi
|
|
fi
|
|
fi
|
|
fi
|
|
else
|
|
# Invalid key or error
|
|
if command -v jq &> /dev/null; then
|
|
local error=$(echo "$body" | jq -r '.error // "Unknown error"')
|
|
echo -e "${RED}Invalid${RESET}"
|
|
echo "Error: $error"
|
|
else
|
|
# Fallback
|
|
local error=$(echo "$body" | grep -o '"error"[[:space:]]*:[[:space:]]*"[^"]*"' | sed 's/.*:[[:space:]]*"//' | tr -d '"')
|
|
[[ -z "$error" ]] && error="Unknown error (HTTP $http_code)"
|
|
echo -e "${RED}Invalid${RESET}"
|
|
echo "Error: $error"
|
|
fi
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
cmd_key() {
|
|
local api_key="${UNSANDBOX_API_KEY:-}"
|
|
local extend=false
|
|
|
|
# Parse arguments
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
-k)
|
|
api_key="$2"
|
|
shift 2
|
|
;;
|
|
--extend)
|
|
extend=true
|
|
shift
|
|
;;
|
|
-*)
|
|
echo -e "${RED}Unknown option: $1${RESET}" >&2
|
|
exit 1
|
|
;;
|
|
*)
|
|
shift
|
|
;;
|
|
esac
|
|
done
|
|
|
|
validate_key "$api_key" "$extend"
|
|
}
|
|
|
|
# Main
|
|
show_help() {
|
|
cat <<EOF
|
|
Unsandbox CLI - Execute code in secure sandboxes
|
|
|
|
Usage:
|
|
$0 [options] <source_file>
|
|
$0 session [options]
|
|
$0 service [options]
|
|
$0 snapshot [options]
|
|
$0 key [options]
|
|
|
|
Execute options:
|
|
-e KEY=VALUE Environment variable (multiple allowed)
|
|
-f FILE Input file (multiple allowed)
|
|
-a Return artifacts
|
|
-o DIR Output directory for artifacts
|
|
-n MODE Network mode (zerotrust|semitrusted)
|
|
-v N vCPU count (1-8)
|
|
-k KEY API key
|
|
|
|
Session options:
|
|
-s, --shell NAME Shell/REPL (default: bash)
|
|
-l, --list List sessions
|
|
--attach ID Attach to session
|
|
--kill ID Terminate session
|
|
--audit Record session
|
|
--tmux Enable tmux persistence
|
|
--screen Enable screen persistence
|
|
--snapshot SESSION_ID Create snapshot of session
|
|
--restore SNAPSHOT_ID Restore from snapshot ID
|
|
--snapshot-name NAME Optional name for snapshot
|
|
--hot Take snapshot without freezing (live snapshot)
|
|
|
|
Service options:
|
|
--name NAME Service name
|
|
--ports PORTS Comma-separated ports
|
|
--domains DOMAINS Custom domains
|
|
--type TYPE Service type for SRV records (minecraft, mumble, teamspeak, source, tcp, udp)
|
|
--bootstrap CMD Bootstrap command or URI
|
|
--bootstrap-file FILE Upload local file as bootstrap script
|
|
-l, --list List services
|
|
--info ID Get service details
|
|
--logs ID Get all logs
|
|
--tail ID Get last 9000 lines
|
|
--freeze ID Freeze service
|
|
--unfreeze ID Unfreeze service
|
|
--destroy ID Destroy service
|
|
--execute ID Execute command in service
|
|
--command CMD Command to execute (with --execute)
|
|
--dump-bootstrap ID Dump bootstrap script
|
|
--dump-file FILE File to save bootstrap (with --dump-bootstrap)
|
|
--snapshot SERVICE_ID Create snapshot of service
|
|
--restore SNAPSHOT_ID Restore from snapshot ID
|
|
--snapshot-name NAME Optional name for snapshot
|
|
--hot Take snapshot without freezing (live snapshot)
|
|
|
|
Snapshot options:
|
|
-l, --list List all snapshots
|
|
--info ID Get snapshot details
|
|
--delete ID Delete a snapshot
|
|
--clone ID Clone snapshot to new session/service (--type required)
|
|
--type TYPE Type for clone: session or service
|
|
--name NAME Name for cloned session/service
|
|
--shell NAME Shell for cloned session
|
|
--ports PORTS Ports for cloned service
|
|
|
|
Key options:
|
|
-k KEY API key to validate
|
|
--extend Validate and open browser to extend key
|
|
EOF
|
|
}
|
|
|
|
# Handle help and no args
|
|
if [[ $# -eq 0 ]] || [[ "$1" == "-h" ]] || [[ "$1" == "--help" ]]; then
|
|
show_help
|
|
exit 0
|
|
fi
|
|
|
|
# Route to command
|
|
if [[ "$1" == "session" ]]; then
|
|
shift
|
|
cmd_session "$@"
|
|
elif [[ "$1" == "service" ]]; then
|
|
shift
|
|
cmd_service "$@"
|
|
elif [[ "$1" == "snapshot" ]]; then
|
|
shift
|
|
cmd_snapshot "$@"
|
|
elif [[ "$1" == "key" ]]; then
|
|
shift
|
|
cmd_key "$@"
|
|
else
|
|
cmd_execute "$@"
|
|
fi
|