#!/usr/bin/env bash
set -euo pipefail

UMM_ROOT="${UMM_ROOT:-$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)}"
cd "${UMM_ROOT}"
unset http_proxy https_proxy HTTP_PROXY HTTPS_PROXY all_proxy ALL_PROXY

ENV_PREFIX="${UMM_ENV_PREFIX:-${UMM_ROOT}/.env/unvideo}"
if [[ ! -x "${ENV_PREFIX}/bin/python" ]]; then
  mkdir -p "${ENV_PREFIX}"
  tar -xzf "${UMM_ROOT}/environment/unvideo-conda.tar.gz" -C "${ENV_PREFIX}"
  if [[ -x "${ENV_PREFIX}/bin/conda-unpack" ]]; then
    "${ENV_PREFIX}/bin/conda-unpack"
  fi
fi

PYTHON_BIN="${ENV_PREFIX}/bin/python"
"${PYTHON_BIN}" -m pip install --no-deps --no-build-isolation \
  -e "${UMM_ROOT}/third_party/transformers" \
  -e "${UMM_ROOT}/third_party/diffusers"
"${PYTHON_BIN}" "${UMM_ROOT}/scripts/render_config.py"

mkdir -p "${UMM_ROOT}/data/covt/images"
expected_images="$("${PYTHON_BIN}" -c 'import json; print(json.load(open("data/covt/stats.json"))["image_count"])')"
current_images="$(find "${UMM_ROOT}/data/covt/images" -maxdepth 1 -type f | wc -l)"
if [[ "${current_images}" != "${expected_images}" ]]; then
  if [[ "${current_images}" != "0" ]]; then
    echo "Partial image extraction detected (${current_images}/${expected_images}); rerun with an empty data/covt/images directory." >&2
    exit 1
  fi
  for shard in "${UMM_ROOT}"/data/covt/shards/images-*.tar; do
    (
      cd "${UMM_ROOT}/data/covt/shards"
      sha256sum -c "$(basename "${shard}").sha256"
    )
    tar -xf "${shard}" -C "${UMM_ROOT}/data/covt/images"
  done
fi

"${PYTHON_BIN}" "${UMM_ROOT}/scripts/verify_bundle.py" --materialized
echo "UMM Stage-2 deployment ready at ${UMM_ROOT}"