qemuDir=$(cd "${BASH_SOURCE[0]%/*}" && pwd)

source "$qemuDir/wait-until.sh"

tmpDir=/tmp/nix-bitcoin-qemu-vm
mkdir -p $tmpDir

# Cleanup on exit
cleanup() {
    set +eu
    if [[ $qemuPID ]]; then
        kill -9 $qemuPID
    fi
    rm -rf $tmpDir
}
trap "cleanup" EXIT

identityFile=$qemuDir/id-vm
chmod 0600 $identityFile

runVM() {
    vm=$1
    vmNumCPUs=$2
    vmMemoryMiB=$3
    sshPort=$4

    export NIX_DISK_IMAGE=$tmpDir/img
    export QEMU_NET_OPTS=hostfwd=tcp::$sshPort-:22
    </dev/null $vm/bin/run-*-vm -m $vmMemoryMiB -smp $vmNumCPUs &>/dev/null &
    qemuPID=$!
}

vmWaitForSSH() {
    echo
    printf "Waiting for SSH connection..."
    waitUntil "c : 2>/dev/null" 500
    echo
}

# Run command in VM
c() {
    ssh -p $sshPort -i $identityFile -o ConnectTimeout=1 \
        -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -o LogLevel=ERROR \
        -o ControlMaster=auto -o ControlPath=$tmpDir/ssh-connection -o ControlPersist=60 \
        root@127.0.0.1 "$@"
}
export identityFile
export sshPort
export tmpDir
export -f c