#!/bin/bash
if [ -f /etc/os-release ]; then
	. /etc/os-release
	OS=$NAME
else
	OS="none"
fi

if [ "$OS" == "Ubuntu" ]; then
	if [ "$(dpkg -l | awk '/ubuntu-desktop/ {print }' | wc -l)" -ge 1 ]; then
		while [ ! -f /media/$USER/RPI-RP2/INFO_UF2.TXT ]; do
			sleep 1
		done
		set -e
		cargo run --release
	else
		sudo umount /mnt/usb
		while [ ! -f /mnt/usb/INFO_UF2.TXT ]; do
			sudo mount /dev/sda1 /mnt/usb -o umask=000
			sleep 1
		done
		set -e
		cargo run --release
		sudo umount /mnt/usb
	fi
elif [ "$OS" == "Arch Linux" ]; then
	sudo umount /mnt/usb
	while [ ! -f /mnt/usb/INFO_UF2.TXT ]; do
		sudo mkdir -p /mnt/usb
		sudo mount /dev/sda1 /mnt/usb -o umask=000
		sleep 1
	done
	set -e
	cargo run --release
	sudo umount /mnt/usb
fi