#!/bin/bash
# Copyright 2025 Alibaba Group Holding Ltd.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

set -euxo pipefail

TAG=${TAG:-latest}
RUN_CODE_INTERPRETER_E2E=${RUN_CODE_INTERPRETER_E2E:-false}

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

source "${REPO_ROOT}/scripts/credential-vault-e2e-target.sh"

cleanup_server() {
  if [ -n "${SERVER_PID}" ]; then
    kill "${SERVER_PID}" 2>/dev/null || true
    wait "${SERVER_PID}" 2>/dev/null || true
  fi
}

cleanup() {
  cleanup_server
  cleanup_credential_vault_e2e_target
}
trap cleanup EXIT

# build execd image locally (context must include internal/)
docker build -f components/execd/Dockerfile -t opensandbox/execd:local "${REPO_ROOT}"

# prepare required images from registry
docker pull opensandbox/code-interpreter:${TAG}
echo "-------- Eval test images --------"
docker images

# prepare hostpath volume for e2e test
mkdir -p /tmp/opensandbox-e2e/host-volume-test
mkdir -p /tmp/opensandbox-e2e/logs
echo "opensandbox-e2e-marker" > /tmp/opensandbox-e2e/host-volume-test/marker.txt
chmod -R 755 /tmp/opensandbox-e2e

# prepare Docker named volume for pvc e2e test
docker volume rm opensandbox-e2e-pvc-test 2>/dev/null || true
docker volume create opensandbox-e2e-pvc-test
# seed the named volume with a marker file and subpath test data via a temporary container
docker run --rm -v opensandbox-e2e-pvc-test:/data alpine sh -c "\
  echo 'pvc-marker-data' > /data/marker.txt && \
  mkdir -p /data/datasets/train && \
  echo 'pvc-subpath-marker' > /data/datasets/train/marker.txt"
echo "-------- JAVASCRIPT E2E test logs for execd --------" > /tmp/opensandbox-e2e/logs/execd.log

export OPENSANDBOX_CREDENTIAL_VAULT_E2E_SANDBOX_IMAGE="${OPENSANDBOX_CREDENTIAL_VAULT_E2E_SANDBOX_IMAGE:-opensandbox/code-interpreter:${TAG}}"
setup_credential_vault_e2e_target

# setup server
cd server
export OPENSANDBOX_INSECURE_SERVER=YES
uv sync
uv run python -m opensandbox_server.main > server.log 2>&1 &
SERVER_PID=$!
cd ..

# wait for server
sleep 10

# run JavaScript/TypeScript e2e (SDK builds are handled by the test script)
cd tests/javascript

# Pin pnpm via corepack (repo expects pnpm@9.x)
corepack enable
corepack prepare pnpm@9.15.0 --activate

pnpm install

# Ensure SDK workspace deps exist before running build steps (CI does not have prebuilt node_modules).
pnpm -C ../../sdks install --frozen-lockfile

# Align with other E2E jobs: local server does not require API key by default.
# Ensure tests do not send an auth header.
export OPENSANDBOX_TEST_API_KEY=""
export OPENSANDBOX_SANDBOX_DEFAULT_IMAGE="opensandbox/code-interpreter:${TAG}"

if [ "${RUN_CODE_INTERPRETER_E2E}" = "true" ]; then
  pnpm test:ci
else
  pnpm run prep:sdk
  test_files=$(find tests -name "*.test.ts" ! -name "test_code_interpreter_e2e.test.ts" -print | sort)
  pnpm exec vitest run ${test_files} \
    --reporter=default \
    --reporter=junit \
    --outputFile=build/test-results/junit.xml
fi