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

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
cd "${ROOT_DIR}"

export http_proxy="${http_proxy:-http://100.72.0.101:8888}"
export https_proxy="${https_proxy:-http://100.72.0.101:8888}"
export HTTP_PROXY="${HTTP_PROXY:-${http_proxy}}"
export HTTPS_PROXY="${HTTPS_PROXY:-${https_proxy}}"
export HF_ENDPOINT="${HF_ENDPOINT:-https://hf-mirror.com}"

source .venv/bin/activate
mkdir -p models

QWEN35_9B_MODEL_ID="${QWEN35_9B_MODEL_ID:-Qwen/Qwen3.5-9B}"
QWEN36_27B_MODEL_ID="${QWEN36_27B_MODEL_ID:-Qwen/Qwen3.6-27B}"
HF_DOWNLOAD_MAX_WORKERS="${HF_DOWNLOAD_MAX_WORKERS:-8}"

download_model() {
  local model_id="$1"
  local local_dir="$2"
  local cmd=(
    hf download "${model_id}"
    --local-dir "${local_dir}"
    --max-workers "${HF_DOWNLOAD_MAX_WORKERS}"
  )
  if [[ -n "${HF_TOKEN:-}" ]]; then
    cmd+=(--token "${HF_TOKEN}")
  fi
  printf '%q ' "${cmd[@]}"
  echo
  if [[ "${DRY_RUN:-0}" == "1" ]]; then
    return 0
  fi
  "${cmd[@]}"
}

download_model "${QWEN35_9B_MODEL_ID}" "models/qwen3.5-9b"
download_model "${QWEN36_27B_MODEL_ID}" "models/qwen3.6-27b"