#!/usr/bin/env bash

# Check associative arrays support
[ "${BASH_VERSION:0:1}" -ge 4 ] || { echo "Bash 4.0 or later is required"; exit 1; }

DATADIR="$(dirname "$0")"
CONFDIR=./conf

MODS_CONF=$CONFDIR/mods.conf
LDS_CONF=$CONFDIR/lds.conf
BUILD_CONF=$CONFDIR/build.conf

CONFIG_ARCH=$(cat $BUILD_CONF | gcc -P -E - | sed -n 's/ARCH *= *//p')
CONFIG_PLATFORM=$(cat $BUILD_CONF | gcc -P -E - | sed -n 's/PLATFORM *= *//p')

info() {
	echo -e "$@" >&2
}

error() {
	echo -e "$@" >&2
	exit 1
}

guessed_info() {
	info "$1": "${!1}" "$2"
}

error_empty() {
	if [ -z "${!1}" ]; then
		error "$1" is empty
	fi
}

get_user_undef_guess() {
	label="(got)"
	if [ -z "${!1+defined}" ]; then
		eval "$1=\"$($2)\""
		label="(guessed)"
	fi
	guessed_info "$1" $label
}

get_user_empty_guess() {
	get_user_undef_guess "$@"
	error_empty "$1"
}

guess_arch() {
	file ./build/base/bin/embox | grep -q "64-bit"
	[ $? -eq 0 ] && elf_64="true" || elf_64="false"
	file ./build/base/bin/embox | grep -q "LSB"
	[ $? -eq 0 ] && elf_lsb="true" || elf_lsb="false"

	if [ "$CONFIG_ARCH" == "x86" ]; then
		qemu_arch="i386"
	else
		qemu_arch="$CONFIG_ARCH"
	fi

	if [ "$CONFIG_ARCH" == "riscv" ]; then
		if [ "$elf_64" == "true" ]; then
			qemu_arch="${qemu_arch}64"
		else
			qemu_arch="${qemu_arch}32"
		fi
	fi

	if [ "$CONFIG_ARCH" == "mips" ]; then
		if [ "$elf_64" == "true" ]; then
			qemu_arch="${qemu_arch}64"
		fi
		if [ "$elf_lsb" == "true" ]; then
			qemu_arch="${qemu_arch}el"
		fi
	fi

	echo "$qemu_arch"
}
get_user_empty_guess AUTOQEMU_ARCH guess_arch

guess_mem() {
	ram=$(cat $LDS_CONF | gcc -P -E - | sed -n 's/^RAM *([0-9x]*, *\([0-9]*[MK]\))/\1/p')
	if [ "$ram" ]; then
		echo "$ram"
	else
		echo 128
	fi
}
get_user_empty_guess AUTOQEMU_MEM guess_mem

guess_nics() {
	cat $MODS_CONF | gcc -P -E - | sed -n 's#.*include\ embox.driver.net.\([a-z0-9_]\+\)\((.*)\|\)$#\1#p' | grep -v loopback | head -n 1;
}
get_user_undef_guess AUTOQEMU_NICS guess_nics

guessed_info AUTOQEMU_NICS_CONFIG "(got)"

guess_sound() {
	cat $MODS_CONF | gcc -P -E - | sed -n 's#.*include\ embox.driver.audio.\([a-z0-9_]\+\)\((.*)\|\)$#\1#p' | \
	grep -v "portaudio_stub\|portaudio_lib" | head -n 1;
}
get_user_undef_guess AUTOQEMU_AUDIO guess_sound

guessed_info AUTOQEMU_AUDIO_CONFIG "(got)"

guess_usb() {
	cat $MODS_CONF | gcc -P -E - | sed -n 's#.*include\ embox.driver.usb.hc.\([a-z0-9_]\+\)\((.*)\|\)$#\1#p' | head -n 1;
}
get_user_undef_guess AUTOQEMU_USB guess_usb

guessed_info AUTOQEMU_USB_CONFIG "(got)"

guess_graphic() {
	cat $MODS_CONF | gcc -P -E - | sed -n 's#.*include\ embox.driver.video.\([a-z0-9_]\+\)\((.*)\|\)$#\1#p' | head -n 3;
}
get_user_undef_guess AUTOQEMU_GRAPHIC guess_graphic

guessed_info AUTOQEMU_GRAPHIC_CONFIG "(got)"

guess_diag() {
	cat $MODS_CONF | gcc -P -E - | sed -n 's/.*embox.driver.diag*[^"]*"\([^"]*\).*/\1/p' | head -n 1;
}
get_user_undef_guess AUTOQEMU_DIAG guess_diag

guessed_info AUTOQEMU_DIAG_CONFIG "(got)"


guess_kvm() {
	ret=""
	# Build kvm argument
	if [ "$CONFIG_ARCH" == "x86" ] && [ "$OSTYPE" == "linux-gnu" ]; then
		if grep -q -E '(vmx|svm)' /proc/cpuinfo; then
			ret="-enable-kvm"
		else
			info VT is not supported by CPU
		fi

		if dmesg | tail | grep -q "kvm: disabled by bios"; then
			info "kvm disabled by bios. You can enable VT in bios"
			ret=
		fi

		if ! lsmod | grep -q -E '(kvm_intel|kvm_amd)'; then
			info no kvm kernel module loaded
			ret=
		fi
	fi
	echo "$ret";
}
get_user_undef_guess AUTOQEMU_KVM_ARG guess_kvm

guess_load_arg() {
	NAND_SIZE=$(cat $LDS_CONF | gcc -P -E - | sed -n 's/.*\<AUTOQEMU_UIMAGE_SIZE_MB=\([0-9]\+\).*$/\1/p')
	if [ ! "$NAND_SIZE" ]; then
		if [ "$CONFIG_ARCH" == "riscv" ] && [ "$CONFIG_PLATFORM" != "esp32c3" ]; then
			echo "-bios ${KERNEL:-./build/base/bin/embox}"
		elif [ "$CONFIG_PLATFORM" == "niisi_vk01" ] || [ "$CONFIG_PLATFORM" == "niisi_vm10" ]; then
			echo "-bios ${KERNEL:-./build/base/bin/embox.bin}"
		else
			echo "-kernel ${KERNEL:-./build/base/bin/embox}"
		fi
	else
		if [ ! "$AUTOQEMU_NAND_BUILD_SKIP" ];  then
			echo "Making uImage..." >&2
			OMAP_UBOOT_IMAGE_BOARD=overo ./scripts/uboot-uimage >/dev/null
			if [ 0 -ne $? ]; then
			    exit 1
			fi
			echo "Making nand image..." >&2
			./scripts/qemu/beagle/run.sh beagle_nand.img ./uImage "$NAND_SIZE" >/dev/null 2>/dev/null
			rm ./uImage
		fi
		echo "-mtdblock beagle_nand.img"
	fi
}
get_user_empty_guess AUTOQEMU_LOAD_ARG guess_load_arg

guess_machine() {
	case $CONFIG_ARCH in
		arm)
			if [ "$CONFIG_PLATFORM" == "overo" ]; then
				echo "-M overo"
			elif [ "$CONFIG_PLATFORM" == "SECO-Q7-796" ]; then
				echo "-M overo"
			elif [ "$CONFIG_PLATFORM" == "integratorcp" ]; then
				echo "-M integratorcp"
			elif [ "$CONFIG_PLATFORM" == "vexpress-a9" ]; then
				echo "-M vexpress-a9"
			elif [ "$CONFIG_PLATFORM" == "sabrelite" ]; then
				echo "-M sabrelite"
			elif [ "$CONFIG_PLATFORM" == "lm3s811evb" ]; then
				echo "-M lm3s811evb"
			elif [ "$CONFIG_PLATFORM" == "lm3s6965evb" ]; then
				echo "-M lm3s6965evb"
			elif [ "$CONFIG_PLATFORM" == "raspberry" ]; then
				echo "-M raspi1ap"
			elif [ "$CONFIG_PLATFORM" == "raspberry2" ]; then
				echo "-M raspi2"
			elif [ "$CONFIG_PLATFORM" == "netduinoplus2" ]; then
				echo "-M netduinoplus2"
			elif [ "$CONFIG_PLATFORM" == "stm32vl_discovery" ]; then
				echo "-M stm32vldiscovery"
			fi
			;;
		aarch64)
			echo "-M virt,gic_version=3 -cpu cortex-a53"
			;;
		microblaze)
			echo "-M petalogix-s3adsp1800"
			;;
		mips)
			if [ "$CONFIG_PLATFORM" == "mipssim" ]; then
				echo "-M mipssim"
			elif [ "$CONFIG_PLATFORM" == "baikal-t" ]; then
				echo "-M baikal-t"
			elif [ "$CONFIG_PLATFORM" == "niisi_vk01" ]; then
				echo "-M vk01 -plc1-board bt75-403a"
			elif [ "$CONFIG_PLATFORM" == "niisi_vm10" ]; then
				echo "-M vm10"
			fi
			;;
		ppc)
			echo "-M virtex-ml507"
			;;
		sparc)
			echo "-M leon3_generic -cpu LEON3"
			;;
		riscv)
			if [ "$CONFIG_PLATFORM" == "virt" ]; then
				echo "-M virt"
			elif [ "$CONFIG_PLATFORM" == "sifive_e" ]; then
				echo "-M sifive_e"
			elif [ "$CONFIG_PLATFORM" == "sifive_u" ]; then
				echo "-M sifive_u"
			elif [ "$CONFIG_PLATFORM" == "syntacore_scr1" ]; then
				echo "-M scr1"
			elif [ "$CONFIG_PLATFORM" == "syntacore_scr3" ]; then
				echo "-M scr3"
			elif [ "$CONFIG_PLATFORM" == "syntacore_scr4" ]; then
				echo "-M scr4"
			elif [ "$CONFIG_PLATFORM" == "syntacore_scr5" ]; then
				echo "-M scr5"
			elif [ "$CONFIG_PLATFORM" == "esp32c3" ]; then
				echo "-M esp32c3"
			fi
			;;
		x86)
			;;
	esac
}
get_user_undef_guess AUTOQEMU_MACHINE guess_machine

guess_cpuhartnum() {
	cpu_num=$(cat "$MODS_CONF" | gcc -P -E - | sed -n 's/.*embox\.arch\.[^\.]*\.kernel\.cpu([^)]*cpu_count=\([0-9]\+\)).*/\1/p')
	echo "${cpu_num:-1}"
}
get_user_undef_guess AUTOQEMU_CPUHARTNUM guess_cpuhartnum

QEMU=${AUTOQEMU_PREFIX}qemu-system-${AUTOQEMU_ARCH}
AWK=$(which gawk >/dev/null && echo gawk || echo awk)
QEMU_VER=$($QEMU --version | $AWK -f "$DATADIR"/qemuver.awk)

if [ "$QEMU_VER" -gt 010200 ]; then
	MB_XEMACLINE="xlnx.xps-ethernetlite"
else
	MB_XEMACLINE="xilinx-ethlite"
fi

declare -A NIC2QEMU
NIC2QEMU=(
	[e1000]="e1000"
	[ne2k_pci]="ne2k_pci"
	[rtl8139]="rtl8139"
	[virtio]="virtio"
	[xemaclite]=$MB_XEMACLINE
	[lan9118]="lan9118"
	[lan91c111]="smc91c111"
	[mipsnet]="mipsnet"
	[fec]="imx.enet"
	[dwc_gmac]="dwgmac"
	[usbnet]="usb-net"
	[cadence_gem]="cadence_gem"
	[sc64_tulip]="tulip-sc64"
)

IFS=" " read -r -a qemu_nics <<< "$AUTOQEMU_NICS"
IFS=" " read -r -a qemu_nics_config <<< "$AUTOQEMU_NICS_CONFIG"
sudo=""
nic_lines=""
nic_n=${#qemu_nics[*]}
if [ "$nic_n" -gt 0 ]; then
	sudo='sudo -E'
	for ni in "${!qemu_nics[@]}"; do

		nic_model=${NIC2QEMU[${qemu_nics[$ni]}]}
		echo
		if [ ! "$nic_model" ]; then
			error "nic model ${qemu_nics[$ni]} is not supported"
		fi

		if [ "${qemu_nics_config[$ni]}" ] && [ "${qemu_nics_config[$ni]}" != "-" ]; then
			host_nic_config=${qemu_nics_config[$ni]}
		else
			host_nic_config="tap,script=$DATADIR/start_script,downscript=$DATADIR/stop_script"
			if [ "$nic_model" == "virtio" ]; then
				host_nic_config="$host_nic_config,vnet_hdr=no"
			fi
		fi

		nic_macaddr="AA:BB:CC:DD:EE:0$(($ni + 2))"

		if [ "$nic_model" == "usb-net" ]; then
			nic_lines="$nic_lines \
				-device usb-net,netdev=n$ni,mac=$nic_macaddr"
		else
			nic_lines="$nic_lines \
				-net nic,netdev=n$ni,model=$nic_model,macaddr=$nic_macaddr"
		fi

		# For new version of QEMU it should be in future:
		# -nic $host_nic_config,model=$nic_model,mac=$nic_macaddr
		#
		nic_lines="$nic_lines \
			-netdev $host_nic_config,id=n$ni"
	done
fi

#############################################
# AUDIO
#############################################
declare -A AUDIO2QEMU
AUDIO2QEMU=(
	[intel_ac_pa]="ac97"
	[es1370]="es1370"
	[aaci_pl041]="default"
)

AUDIO_DEV=" " read -r -a qemu_sound <<< "$AUTOQEMU_AUDIO"

sound_lines=""
sound_n=${#qemu_sound[*]}
if [ "$sound_n" -gt 0 ]; then

	for so in "${!qemu_sound[@]}"; do
		sound_model=${AUDIO2QEMU[${qemu_sound[$so]}]}
		echo
		if [ "$sound_model" != "default" ]; then
			if [ ! "$sound_model" ]; then
				error "sound model ${qemu_sound[$ni]} is not supported"
			fi
			sound_lines="$sound_lines -soundhw $sound_model"
		fi
	done
	if [[ -z "${QEMU_AUDIO_DRV}" ]]; then
		# Use ALSA by default
		export QEMU_AUDIO_DRV=alsa
		info 'Using host ALSA subsystem. You can change it by setting QEMU_AUDIO_DRV'
		info 'For example, "export QEMU_AUDIO_DRV=oss"\n'
	fi
	info 'QEMU_AUDIO_DRV is now' "${QEMU_AUDIO_DRV}" '\n'
fi
#############################################
# END AUDIO
#############################################

#############################################
# USB
#############################################
declare -A USB2QEMU
USB2QEMU=(
	[ehci_pci]="usb-ehci"
	[ohci_pci]="pci-ohci"
	[uhci_pci]="piix3-usb-uhci"
)
declare -A USB2QEMUID
USB2QEMUID=(
	[ehci_pci]="ehci"
	[ohci_pci]="ohci"
	[uhci_pci]="uhci"
)

USB_DEV=" " read -r -a qemu_usb <<< "$AUTOQEMU_USB"

usb_lines=""
usb_n=${#qemu_usb[*]}
if [ "$usb_n" -gt 0 ]; then

	for so in "${!qemu_usb[@]}"; do
		if [ "${qemu_usb[$ni]}" != "ehci_hcd" ]; then
			usb_model=${USB2QEMU[${qemu_usb[$so]}]}
			hci_id=${USB2QEMUID[${qemu_usb[$so]}]}
			echo
			if [ ! "$usb_model" ]; then
				error "usb model ${qemu_usb[$so]} ${qemu_usb[$ni]} is not supported"
			else
				usb_lines="$usb_lines -device $usb_model,id=$hci_id"
			fi
		fi
	done
fi
#############################################
# END USB
#############################################

#-vga [std|cirrus|vmware|qxl|xenfb|tcx|cg3|none]
declare -A GRAPHIC2QEMU
GRAPHIC2QEMU=(
	[bochs]="std"
	[cirrus_logic]="cirrus"
	[pl110]="pl110"
	[ssd0323]="ssd0323"
)

qemu_graphic=$(guess_graphic)
graphic_lines="-nographic"
graphic_n=${#qemu_graphic[*]}
if [ "$graphic_n" -gt 0 ]; then
	if [ -z "${AUTOQEMU_NOGRAPHIC_ARG+defined}" ]; then
		for so in $qemu_graphic; do
			graphic_model=${GRAPHIC2QEMU[$so]}
			if [ ! "$graphic_model" ]; then
				graphic_lines="-nographic"
			else
				if [ "$graphic_model" == "pl110" ] || [ "$graphic_model" == "ssd0323" ]; then
					graphic_lines="-serial mon:stdio"
				else
					graphic_lines="-vga $graphic_model -serial mon:stdio"
				fi
				break
			fi
		done
		if [ "$graphic_lines" == "-nographic" ]; then
			if [ "$(guess_diag)" == "embox__driver__console__vc__vga" ]; then
				graphic_lines="-vga std"
			fi
		fi

	else
		graphic_lines=$AUTOQEMU_NOGRAPHIC_ARG
	fi

fi

runas=""
# Don't try to drop privileges for root user; qemu fails otherwise
if [[ "$(id -u)" -ne 0 ]] ; then
	if [ "$QEMU_VER" -ge 90000 ]; then
		runas="-run-with user=$(whoami)"
	else 
		runas="-runas $(whoami)"
	fi
	sudo='sudo -E'
fi

if [ "$AUTOQEMU_DISABLE_SUDO" == "y" ]; then
	sudo=""
	runas=""
fi

ARG_LINE="$sudo $QEMU $runas $AUTOQEMU_MACHINE $AUTOQEMU_LOAD_ARG $AUTOQEMU_KVM_ARG -m $AUTOQEMU_MEM -smp $AUTOQEMU_CPUHARTNUM $usb_lines $nic_lines $sound_lines $graphic_lines $@"

info $ARG_LINE
$ARG_LINE

if [ -f beagle_nand.img ]; then
	rm beagle_nand.img
fi
