#!/bin/bash
echo -e "\n===== LFGGGGGGGG ======\n"

echo -e "\n"
read -p "Enter directory to install/reinstall to (default ${HOME}/esp):" installDir
installDir=${installDir:-$HOME/esp}

echo -e "\nInstalling prerequisites\n"
sudo apt install -y git wget flex bison gperf python3 python3-pip python3-venv cmake ninja-build ccache libffi-dev libssl-dev dfu-util libusb-1.0-0

runningDir="$( cd "$( dirname "$0" )" && pwd )"
idfDir="${installDir}/esp-idf"
espressifLocation="${HOME}/.espressif"
customBinLocation="${installDir}/.custom_bin"
customBinFrom="${runningDir}/custom_bin"

echo -e "\nInstalling to ${installDir} esp-idf path will be ${idfDir}\n"

echo -e "\nCleaning up environment\n"

if ! [ -d "${installDir}" ]; then
	echo -e "\n${installDir} not found, creating\n"
	mkdir "${installDir}"
else
	echo -e "\n${installDir} found, skipping delete\n"
fi

if [ -d "${idfDir}" ]; then
	echo -e "\n${idfDir} found, deleting\n"
	rm -rf $idfDir
else
	echo -e "\n${idfDir} not found, skipping delete\n"
fi

if [ -d "${espressifLocation}" ]; then
	echo -e "\n${espressifLocation} found, deleting\n"
	rm -rf "${espressifLocation}"
else
	echo -e "\n${espressifLocation} not found, skipping delete\n"
fi

if [ -d "${customBinLocation}" ]; then
	echo -e "\n${customBinLocation} found, deleting\n"
	rm -rf "${customBinLocation}"
else
	echo -e "\n${customBinLocation} not found, skipping delete\n"
fi

echo -e "\nPlacing and enabeling custom scripts at ${customBinLocation}\n"
cp -r "${customBinFrom}" "${customBinLocation}"
chmod -R +x "${customBinLocation}"

echo -e "\nPulling latest esp-idf code from github\n"
git clone --recursive --jobs 5 https://github.com/espressif/esp-idf.git $idfDir

echo -e "\nRunning install script\n"
bash $idfDir/install.sh all

echo -e "\nInstalling optional tools\n"
python $idfDir/tools/idf_tools.py install all

if ! [ -z $(alias | grep get_idf) ]; then
	echo -e "\nget_idf alias not found, appending to ${HOME}/.zshrc\n"
	echo "alias get_idf='. ${idfDir}/export.sh'" >> "${HOME}/.zshrc"
else
	echo -e "\nget_idf alias already installed, skipping\n"
fi

echo -e "\nMaking a backup of ${idfDir}/export.sh to 
${idfDir}/export.sh.bak\n"

cp "${idfDir}/export.sh" "${idfDir}/export.sh.bak"

echo -e "\nEditing ${idfDir}/export.sh\n"
sed -i 's/return 0/# return 0/g' "${idfDir}/export.sh"

echo -e "\nAppending custom additions to ${idfDir}/export.sh\n"
cat "${runningDir}/add-to-export-sh.txt" >> "${idfDir}/export.sh"

echo -e "\nCreating version/commit and date file at ${idfDir}/version-date.txt\n"

datestamp=$(date +"%A, %B %-d %Y at %r %Z (epoch %s)")
commitHash=$(git -C $idfDir rev-parse HEAD)
gitBranch=$(git -C $idfDir rev-parse --abbrev-ref HEAD)

echo -e "Installed at:\n\t${datestamp}\n\tat commit ${commitHash}\n\tfrom branch ${gitBranch}\n\tsource: https://github.com/espressif/esp-idf" > "${idfDir}/version-data.txt"

echo -e '\nRestart shell with `source ~/.zshrc` and run `get_idf` to use\n'
echo  -e "\nEnjoy your new esp-idf install and environment\n"
echo -e "\nAll done :3\n"