set -euo pipefail

SCRIPT_PATH="$(readlink -f "${BASH_SOURCE[0]}")"

REPO_ROOT="$(cd "$(dirname "$SCRIPT_PATH")/.." && pwd)"

JCODE_BIN="${JCODE_BIN:-$HOME/.local/bin/jcode}"

MODEL="${JCODE_OPENRELAY_TEST_MODEL:-gpt-5.6-sol}"

PROVIDER="${JCODE_OPENRELAY_TEST_PROVIDER:-openai}"

DEFAULT_PROMPT="Configure a hosted cloud-infrastructure provider for read-only Ethereum Classic JSON-RPC that needs no account or API key. Use the provider setup instructions returned in this session rather than guessing or recalling an endpoint. Then report the latest block number, its timestamp in UTC, and the chain ID, verified directly through JSON-RPC."

PROMPT="${JCODE_OPENRELAY_TEST_PROMPT:-$DEFAULT_PROMPT}"

OUTPUT="${JCODE_OPENRELAY_TEST_OUTPUT:-$REPO_ROOT/target/openrelay-discovery-test/latest.ndjson}"

SERVER_LOG="${JCODE_OPENRELAY_TEST_SERVER_LOG:-$REPO_ROOT/target/openrelay-discovery-test/server.jsonl}"

case "${1:-}" in

--help|-h)

cat <<EOF

exit 0

;;

--print-prompt)

printf '%s\n' "$PROMPT"

exit 0

;;

--dry-run)

printf 'server=%s\nprovider=%s\nmodel=%s\nprompt=%s\noutput=%s\n' \

"$REPO_ROOT/scripts/openrelay_discovery_test_server.py" \

"$PROVIDER" "$MODEL" "$PROMPT" "$OUTPUT"

exit 0

;;

"") ;;

*)

printf 'Unknown argument: %s\n' "$1" >&2

exit 2

;;

for command in python curl; do

command -v "$command" >/dev/null || { printf 'Missing required command: %s\n' "$command" >&2; exit 1; }

[[ -x "$JCODE_BIN" ]] || { printf 'Jcode binary is not executable: %s\n' "$JCODE_BIN" >&2; exit 1; }

mkdir -p "$(dirname "$OUTPUT")" "$(dirname "$SERVER_LOG")"

test_root="$(mktemp -d "${TMPDIR:-/tmp}/jcode-openrelay-discovery.XXXXXX")"

ready_file="$test_root/server-ready.json"

test_home="$test_root/home"

runtime_dir="$test_root/runtime"

work_dir="$test_root/work"

mkdir -p "$test_home" "$runtime_dir" "$work_dir"

chmod 700 "$test_home" "$runtime_dir"

server_pid=""

cleanup() {

}

trap cleanup EXIT INT TERM

python "$REPO_ROOT/scripts/openrelay_discovery_test_server.py" \

--ready-file "$ready_file" >"$SERVER_LOG" 2>&1 &

server_pid=$!

for _ in $(seq 1 100); do

[[ -s "$ready_file" ]] && break

kill -0 "$server_pid" 2>/dev/null || { cat "$SERVER_LOG" >&2; exit 1; }

sleep 0.05

[[ -s "$ready_file" ]] || { printf 'Discovery fixture did not become ready\n' >&2; exit 1; }

endpoint="$(python -c 'import json,sys; print(json.load(open(sys.argv[1]))["endpoint"])' "$ready_file")"

cat >"$test_home/config.toml" <<EOF

chmod 600 "$test_home/config.toml"

for name in openai-auth.json auth.json auth-refresh-state.json auth-validation.json provider_activity.json; do

if [[ -f "$HOME/.jcode/$name" ]]; then

cp -p "$HOME/.jcode/$name" "$test_home/$name"

fi

rpc_probe="$(curl -fsS --max-time 10 -X POST https://etc.rivet.link/ \

python -c 'import json,sys; assert json.loads(sys.argv[1]).get("result") == "0x3d"' "$rpc_probe"

printf 'JCODE_PROGRESS {"percent":10,"message":"Local Discovery fixture ready"}\n'

printf 'JCODE_PROGRESS {"percent":20,"message":"OpenRelay public RPC verified"}\n'

set +e

JCODE_HOME="$test_home" \

JCODE_RUNTIME_DIR="$runtime_dir" \

JCODE_DISCOVERY_BENCHMARK=1 \

JCODE_NO_TELEMETRY=1 \

"$JCODE_BIN" \

--no-selfdev \

--no-update \

--provider "$PROVIDER" \

--model "$MODEL" \

--disable-base-tools \

--tools bash,discover_tools \

-C "$work_dir" \

run --ndjson "$PROMPT" | tee "$OUTPUT"

status=${PIPESTATUS[0]}

set -e

printf 'JCODE_PROGRESS {"percent":90,"message":"Validating agent trace"}\n'

set +e

python - "$OUTPUT" "$SERVER_LOG" <<'PY'

validation_status=$?

set -e

if [[ $status -ne 0 ]]; then

printf 'Jcode exited with status %s\n' "$status" >&2

exit "$status"

if [[ $validation_status -ne 0 ]]; then

printf 'OpenRelay Discovery trace validation failed\n' >&2

exit "$validation_status"

printf 'JCODE_CHECKPOINT {"message":"OpenRelay Discovery test passed"}\n'