#!/bin/bash

mydir=$(readlink -f "$0")
mydir=$(dirname "$mydir")
codebasedir=$(readlink -f "$mydir"/..) # build process needs absolute paths

target="$1"
buildtype="$2"

if [ -z "$target" ]; then
	echo "Usage: $0 target"
	echo "Usage: $0 <esp32 or unix or macOS>"
	echo "Example: $0 unix"
	echo "Example: $0 macOS"
	echo "Example: $0 esp32"
	echo "Example: $0 esp32s3"
	echo "Example: $0 unphone"
	exit 1
fi

# This assumes all the git submodules have been checked out recursively

echo "Fetch tags for lib/SDL, otherwise lvgl_micropython's make.py script can't checkout a specific tag..."
pushd "$codebasedir"/lvgl_micropython/lib/SDL
git fetch --unshallow origin 2>/dev/null # will give error if already done
# Or fetch all refs without unshallowing (keeps it shallow but adds refs)
git fetch origin 'refs/tags/*:refs/tags/*'
popd

idfile="$codebasedir"/lvgl_micropython/lib/micropython/ports/esp32/main/idf_component.yml
echo "Patching $idfile"...

echo "Check need to add esp32-camera to $idfile"
if ! grep esp32-camera "$idfile"; then
	echo "Adding esp32-camera to $idfile"
	echo "  mpos/esp32-camera:
    git: https://github.com/MicroPythonOS/esp32-camera" >> "$idfile"
else
	echo "No need to add esp32-camera to $idfile"
fi

echo "Check need to add adc_mic to $idfile"
if ! grep adc_mic "$idfile"; then
	echo "Adding adc_mic to $idfile"
        echo '  espressif/adc_mic: "*"' >> "$idfile"
else
	echo "No need to add adc_mic to $idfile"
fi

echo "Resulting $idfile file:"
cat "$idfile"

echo "Check need to add lvgl_micropython manifest to micropython-camera-API's manifest..."
camani="$codebasedir"/micropython-camera-API/src/manifest.py
rellvglmani=lvgl_micropython/build/manifest.py
abslvglmani="$codebasedir"/"$rellvglmani"
if ! grep "$rellvglmani" "$camani"; then
	echo "Adding include(\"$abslvglmani\") to $camani"
	echo >> "$camani" # needs newline because file doesn't have newline at the end
	echo "include(\"$abslvglmani\") # workaround to prevent micropython-camera-API from overriding the lvgl_micropython manifest..." >> "$camani"
	echo "Resulting file:"
	cat "$camani"
else
	echo "No need to add include(\"$abslvglmani\") to $camani"
fi

echo "Check need to add asyncio..."
manifile="$codebasedir"/lvgl_micropython/lib/micropython/ports/unix/variants/manifest.py
if ! grep asyncio "$manifile"; then
	echo "Adding asyncio to $manifile"
	echo 'include("$(MPY_DIR)/extmod/asyncio") # needed to have asyncio, which is used by aiohttp, which has used by websockets' >> "$manifile"
	echo "Resulting file:"
	cat "$manifile"
else
	echo "No need to add asyncio to $manifile"
fi

# unix and macOS builds need these symlinks because make.py doesn't handle USER_C_MODULE arguments for them:
echo "Symlinking secp256k1-embedded-ecdh for unix and macOS builds..."
ln -sf ../../secp256k1-embedded-ecdh "$codebasedir"/lvgl_micropython/ext_mod/secp256k1-embedded-ecdh
echo "Symlinking c_mpos for unix and macOS builds..."
ln -sf ../../c_mpos "$codebasedir"/lvgl_micropython/ext_mod/c_mpos
# Only for MicroPython 1.26.1 workaround:
#echo "Applying lvgl_micropython i2c patch..."
#patch -p0 --forward < "$codebasedir"/patches/i2c_ng.patch

echo "Minifying and inlining HTML..."
pushd "$codebasedir"/webrepl/
python3 inline_minify_webrepl.py
result=$0
if [ $? -ne 0 ]; then
	echo "ERROR: webrepl/inline_minify_webrepl.py failed with exit code $result, webrepl won't work"
else
	mv webrepl_inlined_minified.html ../internal_filesystem/builtin/html/
fi
popd

echo "Refreshing freezefs..."
"$codebasedir"/scripts/freezefs_mount_builtin.sh

if [ "$target" == "esp32" -o "$target" == "esp32s3" -o "$target" == "unphone" ]; then
    partition_size="4194304"
    flash_size="16"
	extra_configs=""
    if [ "$target" == "esp32" ]; then
		BOARD=ESP32_GENERIC
		BOARD_VARIANT=SPIRAM
	else # esp32s3 or unphone
	    if [ "$target" == "unphone" ]; then
	        partition_size="3900000"
	        flash_size="8"
        fi
		BOARD=ESP32_GENERIC_S3
		BOARD_VARIANT=SPIRAM_OCT
		# These options disable hardware AES, SHA and MPI because they give warnings in QEMU: [AES] Error reading from GDMA buffer
		# There's a 25% https download speed penalty for this, but that's usually not the bottleneck.
		extra_configs="CONFIG_MBEDTLS_HARDWARE_AES=n CONFIG_MBEDTLS_HARDWARE_SHA=n CONFIG_MBEDTLS_HARDWARE_MPI=n"
		# --py-freertos: add MicroPython FreeRTOS module to expose internals
		extra_configs="$extra_configs --py-freertos"
	fi
	manifest=$(readlink -f "$codebasedir"/manifests/manifest.py)
	frozenmanifest="FROZEN_MANIFEST=$manifest" # Comment this out if you want to make a build without any frozen files, just an empty MicroPython + whatever files you have on the internal storage
	echo "Note that you can also prevent the builtin filesystem from being mounted by umounting it and creating a builtin/ folder."
	pushd "$codebasedir"/lvgl_micropython/
	rm -rf lib/micropython/ports/esp32/build-$BOARD-$BOARD_VARIANT

	# For more info on the options, see https://github.com/lvgl-micropython/lvgl_micropython
	# --ota: support Over-The-Air updates
	# --partition size: both OTA partitions are 4MB
	# --flash-size: total flash size is 16MB
	# --debug: enable debugging from ESP-IDF but makes copying files to it very slow so that's not added
	# --dual-core-threads: disabled GIL, run code on both CPUs
	# --task-stack-size={stack size in bytes}
	# CONFIG_* sets ESP-IDF options
	# listing processes on the esp32 still doesn't work because no esp32.vtask_list_threads() or something
	# CONFIG_FREERTOS_USE_TRACE_FACILITY=y
	# CONFIG_FREERTOS_VTASKLIST_INCLUDE_COREID=y
	# CONFIG_FREERTOS_GENERATE_RUN_TIME_STATS=y
	# CONFIG_ADC_MIC_TASK_CORE=1 because with the default (-1) it hangs the CPU
	# CONFIG_SPIRAM_XIP_FROM_PSRAM: load entire firmware into RAM to reduce SD vs PSRAM contention (recommended at https://github.com/MicroPythonOS/MicroPythonOS/issues/17)
#	python3 make.py --ota --partition-size=$partition_size --flash-size=$flash_size esp32 BOARD=$BOARD BOARD_VARIANT=$BOARD_VARIANT \
	python3 make.py --optimize-size --partition-size=$partition_size --flash-size=$flash_size esp32 BOARD=$BOARD BOARD_VARIANT=$BOARD_VARIANT \
	    USER_C_MODULE="$codebasedir"/micropython-camera-API/src/micropython.cmake \
	    USER_C_MODULE="$codebasedir"/secp256k1-embedded-ecdh/micropython.cmake \
	    USER_C_MODULE="$codebasedir"/c_mpos/micropython.cmake \
	    CONFIG_FREERTOS_USE_TRACE_FACILITY=y \
	    CONFIG_FREERTOS_VTASKLIST_INCLUDE_COREID=y \
	    CONFIG_FREERTOS_GENERATE_RUN_TIME_STATS=y \
	    CONFIG_ADC_MIC_TASK_CORE=1 \
	    $extra_configs \
	    "$frozenmanifest"

	popd
elif [ "$target" == "unix" -o "$target" == "macOS" ]; then
	manifest=$(readlink -f "$codebasedir"/manifests/manifest.py)
	frozenmanifest="FROZEN_MANIFEST=$manifest"

	# Comment out @micropython.viper decorator for Unix/macOS builds
	# (cross-compiler doesn't support Viper native code emitter)
	echo "Temporarily commenting out @micropython.viper decorator for Unix/macOS build..."
	stream_wav_file="$codebasedir"/internal_filesystem/lib/mpos/audio/stream_wav.py
	sed -i.backup 's/^@micropython\.viper$/#@micropython.viper/' "$stream_wav_file"

	# If it's still running, kill it, otherwise "text file busy"
	pkill -9 -f /lvgl_micropy_unix
	# LV_CFLAGS are passed to USER_C_MODULES (compiler flags only, no linker flags)
	# STRIP= makes it so that debug symbols are kept
	pushd "$codebasedir"/lvgl_micropython/
	# USER_C_MODULE doesn't seem to work properly so there are symlinks in lvgl_micropython/extmod/
	python3 make.py "$target" LV_CFLAGS="-g -O0 -ggdb" STRIP=  DISPLAY=sdl_display INDEV=sdl_pointer "$frozenmanifest"
	popd

	# Restore @micropython.viper decorator after build
	echo "Restoring @micropython.viper decorator..."
	sed -i.backup 's/^#@micropython\.viper$/@micropython.viper/' "$stream_wav_file"
	rm "$stream_wav_file".backup
else
	echo "invalid target $target"
fi

if uname | grep -q Linux; then # only compile it on Linux because MacOS complains about No module named 'elftools', even thought it gets pip installed...
	echo "Building breakout app..."
	"$codebasedir"/c_mpos/breakout/build.sh
fi