# vim: set ft=sh:

run_hook() {
    if [[ -n "${ip}" && -n "${archiso_nbd_srv}" ]]; then

        archiso_nbd_srv=$(eval echo ${archiso_nbd_srv})
        [[ -z "${archiso_nbd_name}" ]] && archiso_nbd_name="archiso"

        mount_handler="archiso_pxe_nbd_mount_handler"
    fi
}

archiso_pxe_nbd_mount_handler () {
    newroot="${1}"

    # Module autoloading like with loop devices does not work, doing manually...
    modprobe nbd 2> /dev/null
    msg ":: Waiting for boot device..."
    while ! poll_device /dev/nbd0 30; do
        echo "ERROR: boot device didn't show up after 30 seconds..."
        echo "   Falling back to interactive prompt"
        echo "   You can try to fix the problem manually, log out when you are finished"
        launch_interactive_shell
    done

    msg ":: Setup NBD from ${archiso_nbd_srv} at /dev/nbd0"
    if [[ "${copytoram}" = "y" ]]; then
        nbd-client ${archiso_nbd_srv} -N ${archiso_nbd_name} /dev/nbd0
    else
        nbd-client ${archiso_nbd_srv} -N ${archiso_nbd_name} /dev/nbd0 -persist
    fi

    archisodevice=/dev/nbd0

    archiso_mount_handler ${newroot}

    if [[ "${copytoram}" = "y" ]]; then
        msg ":: Disconnect NBD from ${archiso_nbd_srv} at /dev/nbd0"
        nbd-client -d /dev/nbd0
    else
        mkdir -p /run/archiso
        pidof nbd-client > /run/archiso/nbd_client.pid
        cp /archiso_pxe_nbd ${newroot}/etc/rc.d/functions.d/
    fi
}