#!/bin/bash
set -e

echo "[1/5] 检查spleeter 2stem模型"
if [ -e pretrained_models/2stems/checkpoint ]
then
    echo "模型已存在，跳过下载..."
else
    echo "下载模型..."
    curl -LO https://gitea.cattysteve.top/api/packages/pig2014/generic/spleeter-models/2stems-1/2stems-finetune.tar.gz
    mkdir -p 2stems
    tar -xf 2stems-finetune.tar.gz -C 2stems/
    mkdir -p pretrained_models
    mv 2stems pretrained_models/2stems
    echo "模型下载完成"
fi

echo "[2/5] 检查python3.9..."
if [ -e python39*.pkg.tar.* ]
then
    echo "python3.9已存在，跳过下载..."
else
    curl -LO https://gitea.cattysteve.top/api/packages/pig2014/generic/python39/3.9.19-1/python39-3.9.19-1-x86_64.pkg.tar.zst
    echo "python3.9下载完成"
fi

echo "[3/5] 检查pipx.pyz..."
if [ -e pipx.pyz ]
then
    echo "pipx已存在，跳过下载..."
else
    curl -LO https://github.com/pypa/pipx/releases/download/1.5.0/pipx.pyz
    echo "pipx下载完成"
fi

echo "[4/5] 构建docker镜像..."
docker build -t galaxymusichallv2_convert .

echo "[5/5] 推送docker镜像"
if [ a$1 == apush ]
then
    #docker tag galaxymusichallv2_convert:latest gitea.cattysteve.top/pig2014/galaxymusichallv2_convert:latest
    #docker login gitea.cattysteve.top -u pig2014 -p $2
    #docker push gitea.cattysteve.top/pig2014/galaxymusichallv2_convert:latest
    docker run -d -p 12400:8000 galaxymusichallv2_convert:latest
    echo "推送完成！"
else
    echo "未指定推送参数，跳过中..."
fi