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

MODEL_DIR=${MODEL_DIR:-/data/models/qwen3.8-27b-dirk}
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
REVISION=12362f2b3d7dc11044e99c9e7e99fb9f530528c0
BASE_URL="https://huggingface.co/peculiar-ragdoll/Dirk-Qwen3.8-27B-GGUF/resolve/$REVISION"
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 -c "$SCRIPT_DIR/MODEL_ARTIFACTS.sha256"
  cp "$SCRIPT_DIR/MODEL_ARTIFACTS.sha256" SHA256SUMS
)

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