#!/bin/bash

export MSYS_NO_PATHCONV=1
set -e

function chown_iterations() {
    if [ -d "./iterations" ] && [ "$(find ./iterations/ ! -user $USER -o ! -group $USER 2>/dev/null)" ]; then
        echo "Chowning iterations directory"
        sudo chown -R $USER:$USER ./iterations/
    fi
}

function clean_docker() {
    docker images -f "dangling=true" -q | xargs -r docker rmi
}

trap chown_iterations EXIT
trap clean_docker EXIT

docker build \
    --tag sia \
    .

docker run \
    --init \
    --rm \
    -ti \
    --gpus=all \
    -p 8080:8080 \
    -p 8000:8000 \
    --env-file .env \
    -v /$(pwd)/models/:/root/models/ \
    -v /$(pwd)/iterations/:/root/data/iterations/ \
    -v /$(pwd)/tasks/:/root/data/tasks/ \
    -v /$(pwd)/user/:/root/data/user/ \
    -v /$(pwd)/environment/:/root/data/environment/ \
    -v /$(pwd)/cache/:/root/.cache/ \
    -v /$(pwd)/:/root/sia/ \
    sia "$@"

exit $?