#!/bin/sh
# ShiMeta SVP 模型安装脚本
# 用法: sh install_models.sh [mirror] [version]
#       sh install_models.sh -v 0.2.2

VERSION=""
MIRROR="https://mirrors.shimetapi.cn/repos/GK7206/shimeta-ai"

case "$1" in -v) VERSION="$2"; shift 2 ;; esac
if [ -n "$1" ] && [ "$1" != "-v" ]; then MIRROR="$1"; fi
case "$2" in [0-9]*.*) VERSION="$2" ;; esac

DEST="/mnt/models/svp"
TARBALL="svp_models.tar"
if [ -n "${VERSION}" ]; then TARBALL="svp_models_${VERSION}.tar"; fi
URL="${MIRROR}/${TARBALL}"

echo "=== ShiMeta SVP Models Installer${VERSION:+ v${VERSION}} ==="

echo "=== ShiMeta SVP Models Installer v${VERSION} ==="
echo "Server: ${MIRROR}"
echo ""

mkdir -p "${DEST}"

DL="${DEST}/svp_models.tar"
echo "[1/2] Downloading svp_models.tar (62MB)..."

if ! command -v wget >/dev/null 2>&1; then
    echo "ERROR: 未找到 wget，无法下载"
    exit 1
fi
wget "${URL}" -O "${DL}"

if [ ! -f "${DL}" ]; then
    echo "ERROR: download failed"
    exit 1
fi

echo "[2/2] Extracting to ${DEST}..."
tar xf "${DL}" -C "${DEST}"
rm -f "${DL}"

echo ""
echo "Done. $(ls ${DEST}/*.bin 2>/dev/null | wc -l) models installed."
grep -q "mount /dev/mmcblk1p1" /etc/profile 2>/dev/null || echo "mount /dev/mmcblk1p1 /mnt 2>/dev/null" >> /etc/profile
echo "Run: shimeta_svp <task>"