start.sh

#!/usr/bin/env bash

set -eu
set -o pipefail
cd $(dirname $0)

source config.sh
cd ..

function yes_no() {
    if [[ "$1" == "yes" || "$1" == "y" ]]; then
        return 0
    else
        return 1
    fi
}

if use_prebuilt; then
    container_name=$PREBUILT_CONTAINER_NAME
else
    container_name=$CONTAINER_NAME
fi


container="$(docker ps --filter="name=^/$container_name$" --latest --quiet)"
if [[ -n "$container" ]]; then
    # Connect to already running container
    log_success "[+] Found running instance: $container, connecting..."
    cmd="docker start $container"
    log_success "[+] $cmd"
    $cmd > /dev/null
    if [[ -v NO_TTY ]]; then
        HAS_TTY=""
    else
        HAS_TTY="-t"
    fi
    cmd="docker exec -i $HAS_TTY --workdir /home/user/fuzztruction $container zsh"
    log_success "[+] $cmd"
    $cmd
    exit 0
fi

touch "$PWD/data/bash_history"
touch "$PWD/data/zsh_history"
mkdir -p "$PWD/data/ccache"
mkdir -p "$PWD/data/vscode-data"
mkdir -p "$PWD/eval-result"

log_success "[+] Creating new container..."

mounts=""
if use_prebuilt; then
    log_success "[+] Using prebuilt image"
else
    log_success "[+] Using locally build image"
    mounts+=" -v $PWD:/home/user/fuzztruction "
fi
mounts+=" -v $PWD:/shared "

cmd="docker run -ti -d --privileged
    $mounts
    -v $PWD/data/zshrc:/home/user/.zshrc
    -v $PWD/data/zsh_history:/home/user/.zsh_history
    -v $PWD/data/bash_history:/home/user/.bash_history
    -v $PWD/data/init.vim:/home/user/.config/nvim/init.vim
    -v $PWD/data/ccache:/ccache
    -v $PWD/data/vscode-data:/home/user/.config/Code
    -v $PWD/eval-result:/home/user/fuzztruction/eval-result
    --mount type=tmpfs,destination=/tmp,tmpfs-mode=777
    --ulimit msgqueue=2097152000
    --shm-size=64G
    --net=host
    --name $container_name
    --env "PREBUILT_ENV_VAR_NAME=$PREBUILT_ENV_VAR_NAME"
    --env "$PREBUILT_ENV_VAR_NAME=${!PREBUILT_ENV_VAR_NAME:-}" "

if [[ ! -z "${SSH_AUTH_SOCK:-}"  ]]; then
    log_success "[+] Forwarding ssh agent ($SSH_AUTH_SOCK -> /ssh-agent)"
    cmd+="-v $(readlink -f "$SSH_AUTH_SOCK"):/ssh-agent --env SSH_AUTH_SOCK=/ssh-agent"
fi

# Use local gitconfig if any
if [[ -f "/home/$USER/.gitconfig" ]]; then
    cmd+=" -v /home/$USER/.gitconfig:/home/user/.gitconfig"
fi

if use_prebuilt; then
    log_success "[+] Using $PREBUILT_IMAGE_NAME as docker image"
    cmd+=" ${PREBUILT_IMAGE_NAME}"
else
    log_success "[+] Using $IMAGE_NAME as docker image"
    cmd+=" ${IMAGE_NAME}"

fi

log_success "[+] $(echo $cmd | xargs)"
$cmd > /dev/null

log_success "[+] Rerun $0 to connect to the new container."