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

MODEL_DIR=${MODEL_DIR:-/data/models/qwen3.8-27b-dirk}
BASE_URL=https://huggingface.co/peculiar-ragdoll/Dirk-Qwen3.8-27B-GGUF/resolve/main
MODEL=Dirk-Qwen3.8-27B-UD-Q4_K_XL.gguf
PROJECTOR=mmproj-F16.gguf

install -d -m 0755 "$MODEL_DIR"

download() {
  local name=$1
  local target="$MODEL_DIR/$name"
  curl --fail --location --retry 8 --retry-delay 5 --continue-at - \
    --output "$target.part" "$BASE_URL/$name"
  mv -f "$target.part" "$target"
}

[[ -s "$MODEL_DIR/$MODEL" ]] || download "$MODEL"
[[ -s "$MODEL_DIR/$PROJECTOR" ]] || download "$PROJECTOR"

(
  cd "$MODEL_DIR"
  sha256sum "$MODEL" "$PROJECTOR" > SHA256SUMS
)

printf 'Prepared model files in %s\n' "$MODEL_DIR"