#!/bin/bash

set -e

cd "$(dirname "$0")"

VERSION=$(date +%Y%m%d)-latest
echo Building: ${VERSION}


###### Training Image ######
# docker build -t dcloud/lerobot-train:${VERSION} -f docker/Dockerfile.train . --build-arg http_proxy=http://192.168.16.68:18000 --build-arg https_proxy=http://192.168.16.68:18000
# docker run -it --rm --gpus '"device=7"' \
#     -v ${PWD}/docker/inputs:/workspace/inputs \
#     -v ${PWD}/docker/outputs:/workspace/outputs/checkpoints \
#     -v ${PWD}/docker/train_task.json:/workspace/inputs/task.json \
#     --shm-size=128G \
#     dcloud/lerobot-train:${VERSION}


###### Merge Image ######
docker build -t dcloud/lerobot-merge:${VERSION} -f docker/Dockerfile.merge . --build-arg http_proxy=http://192.168.16.68:18000 --build-arg https_proxy=http://192.168.16.68:18000
docker run -it --rm \
    -v ${PWD}/docker/inputs:/workspace/inputs \
    -v ${PWD}/docker/outputs:/workspace/outputs \
    -v ${PWD}/docker/merge_task.json:/workspace/inputs/task.json \
    --shm-size=128G \
    dcloud/lerobot-merge:${VERSION}


# # Remove dangling images
docker rmi $(docker images -f "dangling=true" -q)
docker images | grep lerobot | grep -v ${VERSION} | awk '{print $1":"$2}' | xargs docker rmi