#!/bin/bash
# https://github.com/Hyy2001X/AutoBuild-Actions
# AutoBuild Module by Hyy2001
# AutoBuild Functions

GET_TARGET_INFO() {
	Diy_Core
	Home=${GITHUB_WORKSPACE}/openwrt
	[ -f ${GITHUB_WORKSPACE}/Openwrt.info ] && . ${GITHUB_WORKSPACE}/Openwrt.info
	Owner_Repo="$(grep "https://github.com/[a-zA-Z0-9]" ${GITHUB_WORKSPACE}/.git/config | cut -c8-100)"
	Source_Repo="$(grep "https://github.com/[a-zA-Z0-9]" ${Home}/.git/config | cut -c8-100)"
	Source_Owner="$(echo "${Source_Repo}" | egrep -o "[a-z]+" | awk 'NR==4')"
	Current_Branch="$(git branch | sed 's/* //g')"
	if [[ ! ${Current_Branch} == master ]];then
		Current_Branch="$(echo ${Current_Branch} | egrep -o "[0-9]+.[0-9]+")"
		Openwrt_Version_="R${Current_Branch}-"
	else
		Openwrt_Version_="R18.06-"
	fi
	AB_Firmware_Info=package/base-files/files/etc/openwrt_info
	case ${Source_Owner} in
	coolsnowwolf)
		Version_File="package/lean/default-settings/files/zzz-default-settings"
		Old_Version="$(egrep -o "R[0-9]+\.[0-9]+\.[0-9]+" ${Version_File})"
		Openwrt_Version="${Old_Version}-${Compile_Date}"
	;;
	immortalwrt)
		Version_File="package/base-files/files/etc/openwrt_release"
		Openwrt_Version="${Openwrt_Version_}${Compile_Date}"
	;;
	*)
		Openwrt_Version="${Openwrt_Version_}${Compile_Date}"
	;;
	esac
	while [[ -z "${x86_Test}" ]]
	do
		x86_Test="$(egrep -o "CONFIG_TARGET.*DEVICE.*=y" .config | sed -r 's/CONFIG_TARGET_(.*)_DEVICE_(.*)=y/\1/')"
		[[ -n "${x86_Test}" ]] && break
		x86_Test="$(egrep -o "CONFIG_TARGET.*Generic=y" .config | sed -r 's/CONFIG_TARGET_(.*)_Generic=y/\1/')"
		[[ -z "${x86_Test}" ]] && TIME "Can not obtain the TARGET_PROFILE !" && exit 1
	done
	if [[ "${x86_Test}" == "x86_64" ]];then
		TARGET_PROFILE="x86_64"
	else
		TARGET_PROFILE="$(egrep -o "CONFIG_TARGET.*DEVICE.*=y" .config | sed -r 's/.*DEVICE_(.*)=y/\1/')"
	fi
	[[ -z "${TARGET_PROFILE}" ]] && TARGET_PROFILE="${Default_Device}"
	case ${TARGET_PROFILE} in
	x86_64)
		if [[ "$(cat ${Home}/.config)" =~ "CONFIG_TARGET_IMAGES_GZIP=y" ]];then
			Firmware_Type="img.gz"
		else
			Firmware_Type="img"
		fi
	;;
	*)
		Firmware_Type="bin"
	;;
	esac
	TARGET_BOARD="$(awk -F '[="]+' '/TARGET_BOARD/{print $2}' .config)"
	TARGET_SUBTARGET="$(awk -F '[="]+' '/TARGET_SUBTARGET/{print $2}' .config)"
	
	echo "Firmware_Type=${Firmware_Type}" > ${Home}/TARGET_INFO
	echo "TARGET_PROFILE=${TARGET_PROFILE}" >> ${Home}/TARGET_INFO
	echo "Openwrt_Version=${Openwrt_Version}" >> ${Home}/TARGET_INFO
	echo "Source_Owner=${Source_Owner}" >> ${Home}/TARGET_INFO
	echo "TARGET_BOARD=${TARGET_BOARD}" >> ${Home}/TARGET_INFO
	echo "TARGET_SUBTARGET=${TARGET_SUBTARGET}" >> ${Home}/TARGET_INFO
	echo "Home=${Home}" >> ${Home}/TARGET_INFO
	echo "Current_Branch=${Current_Branch}" >> ${Home}/TARGET_INFO
	
	echo "${Openwrt_Version}" > ${AB_Firmware_Info}
	echo "${Owner_Repo}" >> ${AB_Firmware_Info}
	echo "${TARGET_PROFILE}" >> ${AB_Firmware_Info}
	echo "${Firmware_Type}" >> ${AB_Firmware_Info}
	
	echo "Author: ${Author}"
	echo "Author Github: ${Owner_Repo}"
	echo "Firmware Version: ${Openwrt_Version}"
	echo "Firmware Type: ${Firmware_Type}"
	echo "Source: ${Source_Repo}"
	echo "Source Author: ${Source_Owner}"
	echo "Source Branch: ${Current_Branch}"
	echo "TARGET_PTOFILE: ${TARGET_PROFILE}"
	echo "TARGET_BOARD: ${TARGET_BOARD}"
	echo "TARGET_SUBTARGET: ${TARGET_SUBTARGET}"
	
	TIME "[Preload Info] All done !"
}

Firmware-Diy_Base() {
	GET_TARGET_INFO
	Auto_ExtraPackages
	chmod +x -R ${GITHUB_WORKSPACE}/Scripts
	chmod +x -R ${GITHUB_WORKSPACE}/CustomFiles
	if [[ "${INCLUDE_AutoBuild_Tools}" == "true" ]];then
		Replace_File Scripts/AutoBuild_Tools.sh package/base-files/files/bin
	fi
	if [[ "${INCLUDE_AutoUpdate}" == "true" ]];then
		ExtraPackages git lean luci-app-autoupdate https://github.com/Hyy2001X main
		Replace_File Scripts/AutoUpdate.sh package/base-files/files/bin
	fi
	if [[ "${INCLUDE_Theme_Argon}" == "true" ]];then
		case ${Source_Owner} in
		coolsnowwolf)
				ExtraPackages git lean luci-theme-argon https://github.com/jerrykuku 18.06
		;;
		*)
			case ${Current_Branch} in
			19.07)
				ExtraPackages git other luci-theme-argon https://github.com/jerrykuku v2.2.5
			;;
			21.02)
				ExtraPackages git other luci-theme-argon https://github.com/jerrykuku
			;;
			18.06)
				ExtraPackages git other luci-theme-argon https://github.com/jerrykuku 18.06
			;;
			*)
				TIME "[ERROR] Unknown source branch: [${Current_Branch}] !"
			;;
			esac
		;;
		esac
	fi
	if [ -f package/base-files/files/bin/AutoUpdate.sh ];then
		AutoUpdate_Version=$(awk 'NR==6' package/base-files/files/bin/AutoUpdate.sh | awk -F '[="]+' '/Version/{print $2}')
	else
		AutoUpdate_Version=OFF
	fi

	Replace_File CustomFiles/Depends/profile package/base-files/files/etc
	case ${Source_Owner} in
	coolsnowwolf)
		Replace_File CustomFiles/Depends/coremark_lede.sh package/lean/coremark coremark.sh
		Replace_File CustomFiles/Depends/cpuinfo_x86 package/lean/autocore/files/x86/sbin cpuinfo
		ExtraPackages git lean helloworld https://github.com/fw876 master
		Update_Makefile xray-core package/lean/helloworld/xray-core
		sed -i 's/143/143,8080/' package/lean/helloworld/luci-app-ssr-plus/root/etc/init.d/shadowsocksr
		sed -i "s?iptables?#iptables?g" ${Version_File} > /dev/null 2>&1
		sed -i "s?${Old_Version}?${Old_Version} Compiled by ${Author} [${Display_Date}]?g" $Version_File
		[[ "${INCLUDE_DRM_I915}" == "true" ]] && Replace_File CustomFiles/Depends/i915-5.4 target/linux/x86 config-5.4
	;;
	immortalwrt)
		sed -i 's/143/143,8080/' package/lean/luci-app-ssr-plus/root/etc/init.d/shadowsocksr
		Replace_File CustomFiles/Depends/coremark_ImmortalWrt.sh package/base-files/files/etc coremark.sh
		Replace_File CustomFiles/Depends/ImmortalWrt package/base-files/files/etc openwrt_release
		Replace_File CustomFiles/Depends/cpuinfo_x86 package/lean/autocore/files/x86/sbin cpuinfo
		sed -i "s?Template?Compiled by ${Author} [${Display_Date}]?g" $Version_File
		[[ "${INCLUDE_DRM_I915}" == "true" ]] && Replace_File CustomFiles/Depends/i915-4.19 target/linux/x86 config-4.19
	;;
	esac
	
	case ${Source_Owner} in
	immortalwrt)
		Replace_File CustomFiles/Depends/banner package/lean/default-settings/files openwrt_banner
		sed -i "s?By?By ${Author}?g" package/lean/default-settings/files/openwrt_banner
		sed -i "s?Openwrt?ImmortalWrt ${Openwrt_Version} / AutoUpdate ${AutoUpdate_Version}?g" package/lean/default-settings/files/openwrt_banner
	;;
	*)
		Replace_File CustomFiles/Depends/banner package/base-files/files/etc
		sed -i "s?By?By ${Author}?g" package/base-files/files/etc/banner
		sed -i "s?Openwrt?Openwrt ${Openwrt_Version} / AutoUpdate ${AutoUpdate_Version}?g" package/base-files/files/etc/banner
	;;
	esac

	if [[ "${INCLUDE_Obsolete_PKG_Compatible}" == "true" ]];then
		TIME "Start to run Obsolete_Package_Compatible Scripts ..."
		if [[ ${Source_Owner} == openwrt ]];then
			case ${Current_Branch} in
			19.07 | 21.02)
				Replace_File CustomFiles/Patches/0003-upx-ucl-${Current_Branch}.patch ./
				cat 0003-upx-ucl-${Current_Branch}.patch | patch -p1 > /dev/null 2>&1
				ExtraPackages svn ../feeds/packages/lang golang https://github.com/coolsnowwolf/packages/trunk/lang
		
				TIME "Start to convert zh-cn translation files to zh_Hans ..."
				Replace_File Scripts/Convert_Translation.sh package
				cd ./package
				bash ./Convert_Translation.sh
				cd ..
			;;
			*)
				TIME "[ERROR] Current branch: [${Current_Branch}] is not supported !"
			;;
			esac
		else
			TIME "[ERROR] Current source: [${Source_Owner}] is not supported !"
		fi
	fi
}

PS_Firmware() {
	. TARGET_INFO
	ls bin/targets/${TARGET_BOARD}/${TARGET_SUBTARGET}
	case ${Source_Owner} in
	immortalwrt)
		_Firmware=immortalwrt
	;;
	*)
		_Firmware=openwrt
	;;
	esac
	case ${Current_Branch} in
	19.07 | 18.06)
		case ${Source_Owner} in
		immortalwrt)
			_Legacy_Firmware=combined-squashfs
			_EFI_Firmware=uefi-gpt-squashfs
		;;
		*)
			_Legacy_Firmware=combined-squashfs
			_EFI_Firmware=combined-squashfs-efi
		;;
		esac
	;;
	*)
		_Legacy_Firmware=generic-squashfs-combined
		_EFI_Firmware=generic-squashfs-combined-efi
	;;
	esac
	Firmware_Path="bin/targets/${TARGET_BOARD}/${TARGET_SUBTARGET}"
	Mkdir bin/Firmware
	case "${TARGET_PROFILE}" in
	x86_64)
		cd ${Firmware_Path}
		Legacy_Firmware=${_Firmware}-${TARGET_BOARD}-${TARGET_SUBTARGET}-${_Legacy_Firmware}.${Firmware_Type}
		EFI_Firmware=${_Firmware}-${TARGET_BOARD}-${TARGET_SUBTARGET}-${_EFI_Firmware}.${Firmware_Type}
		AutoBuild_Firmware="AutoBuild-${TARGET_PROFILE}-${Openwrt_Version}"
		echo "[Preload Info] Legacy_Firmware: ${Legacy_Firmware}"
		echo "[Preload Info] UEFI_Firmware: ${EFI_Firmware}"
		echo "[Preload Info] AutoBuild_Firmware: ${AutoBuild_Firmware}"
		if [ -f "${Legacy_Firmware}" ];then
			_MD5=$(md5sum ${Legacy_Firmware} | cut -d ' ' -f1)
			_SHA256=$(sha256sum ${Legacy_Firmware} | cut -d ' ' -f1)
			touch ${Home}/bin/Firmware/${AutoBuild_Firmware}.detail
			echo -e "\nMD5:${_MD5}\nSHA256:${_SHA256}" > ${Home}/bin/Firmware/${AutoBuild_Firmware}-Legacy.detail
			mv -f ${Legacy_Firmware} ${Home}/bin/Firmware/${AutoBuild_Firmware}-Legacy.${Firmware_Type}
			TIME "Legacy Firmware is detected !"
		else
			TIME "[ERROR] Legacy Firmware is not detected !"
		fi
		if [ -f "${EFI_Firmware}" ];then
			_MD5=$(md5sum ${EFI_Firmware} | cut -d ' ' -f1)
			_SHA256=$(sha256sum ${EFI_Firmware} | cut -d ' ' -f1)
			touch ${Home}/bin/Firmware/${AutoBuild_Firmware}-UEFI.detail
			echo -e "\nMD5:${_MD5}\nSHA256:${_SHA256}" > ${Home}/bin/Firmware/${AutoBuild_Firmware}-UEFI.detail
			cp ${EFI_Firmware} ${Home}/bin/Firmware/${AutoBuild_Firmware}-UEFI.${Firmware_Type}
			TIME "UEFI Firmware is detected !"
		else
			TIME "[ERROR] UEFI Firmware is not detected !"
		fi
	;;
	*)
		cd ${Home}
		Default_Firmware="${_Firmware}-${TARGET_BOARD}-${TARGET_SUBTARGET}-${TARGET_PROFILE}-squashfs-sysupgrade.${Firmware_Type}"
		AutoBuild_Firmware="AutoBuild-${TARGET_PROFILE}-${Openwrt_Version}.${Firmware_Type}"
		AutoBuild_Detail="AutoBuild-${TARGET_PROFILE}-${Openwrt_Version}.detail"
		echo "[Preload Info] Default_Firmware: ${Default_Firmware}"
		echo "[Preload Info] AutoBuild_Firmware: ${AutoBuild_Firmware}"
		if [ -f ${Firmware_Path}/${Default_Firmware} ];then
			mv -f ${Firmware_Path}/${Default_Firmware} bin/Firmware/${AutoBuild_Firmware}
			_MD5=$(md5sum bin/Firmware/${AutoBuild_Firmware} | cut -d ' ' -f1)
			_SHA256=$(sha256sum bin/Firmware/${AutoBuild_Firmware} | cut -d ' ' -f1)
			echo -e "\nMD5:${_MD5}\nSHA256:${_SHA256}" > bin/Firmware/${AutoBuild_Detail}
			TIME "Firmware is detected !"
		else
			TIME "[ERROR] Firmware is not detected !"
		fi
	;;
	esac
	cd ${Home}
	echo "[$(date "+%H:%M:%S")] Actions Avaliable: $(df -h | grep "/dev/root" | awk '{printf $4}')"
}

TIME() {
	echo "[$(date "+%H:%M:%S")] ${*}"
}

Mkdir() {
	_DIR=${1}
	if [ ! -d "${_DIR}" ];then
		TIME "Creating new folder [${_DIR}] ..."
		mkdir -p ${_DIR}
	fi
	unset _DIR
}

PKG_Finder() {
	unset PKG_RESULT
	_PKG_TYPE=${1}
	_PKG_DIR=${2}
	_PKG_NAME=${3}
	[[ -z ${_PKG_TYPE} ]] && [[ -z ${_PKG_NAME} ]] || [[ -z ${_PKG_DIR} ]] && return
	PKG_RESULT=$(find -name ${_PKG_DIR}/${_PKG_NAME} -type ${_PKG_TYPE} -exec echo {} \;)
	[[ -n "${_PKG_RESULT}" ]] && echo "${_PKG_RESULT}"
	unset _PKG_TYPE _PKG_DIR _PKG_NAME
}

Auto_ExtraPackages() {
	COMMON_FILE="${GITHUB_WORKSPACE}/CustomPackages/Common"
	TARGET_FILE="${GITHUB_WORKSPACE}/CustomPackages/${TARGET_PROFILE}"

	if [ -f "${TARGET_FILE}" ];then
		if [ -f "${COMMON_FILE}" ];then
			TIME "Loading Common Packages list ..."
			echo -e "\n$(cat ${COMMON_FILE})" >> ${TARGET_FILE}
		fi
		TIME "Loading Custom Packages list: [${TARGET_PROFILE}] ..."
		cat ${TARGET_FILE} | while read X
		do
			[[ -n "${X}" ]] && {
			if [[ ! "${X}" =~ "#" ]];then
				ExtraPackages ${X}
			else
				TIME "Skip line: ${X}"
			fi
                        }
			unset X
		done
		TIME "[CustomPackages] All done !"
	else
		TIME "Custom Packages list: [${TARGET_PROFILE}] is not detected !"
	fi
}

ExtraPackages() {
	PKG_PROTO=${1}
	PKG_DIR=${2}
	PKG_NAME=${3}
	REPO_URL=${4}
	REPO_BRANCH=${5}

	Mkdir package/${PKG_DIR}
	if [ -d "package/${PKG_DIR}/${PKG_NAME}" ];then
		TIME "Removing old package [${PKG_NAME}] ..."
		rm -rf package/${PKG_DIR}/${PKG_NAME}
	fi
	TIME "Checking out package [${PKG_NAME}] to package/${PKG_DIR} ..."
	case "${PKG_PROTO}" in
	git)
		[[ -z "${REPO_BRANCH}" ]] && REPO_BRANCH=master
		PKG_URL="$(echo ${REPO_URL}/${PKG_NAME} | sed s/[[:space:]]//g)"
		git clone -b ${REPO_BRANCH} ${PKG_URL} ${PKG_NAME} > /dev/null 2>&1
	;;
	svn)
		svn checkout ${REPO_URL}/${PKG_NAME} ${PKG_NAME} > /dev/null 2>&1
	;;
	*)
		TIME "[ERROR] Error option: ${PKG_PROTO} !" && return
	;;
	esac
	if [ -f ${PKG_NAME}/Makefile ] || [ -f ${PKG_NAME}/README* ];then
		TIME "Package [${PKG_NAME}] is detected!"
		mv -f ${PKG_NAME} package/${PKG_DIR}
	fi
	unset PKG_PROTO PKG_DIR PKG_NAME REPO_URL REPO_BRANCH
}

Replace_File() {
	FILE_NAME=${1}
	PATCH_DIR=${GITHUB_WORKSPACE}/openwrt/${2}
	FILE_RENAME=${3}
	
	Mkdir ${PATCH_DIR}
	[ -f "${GITHUB_WORKSPACE}/${FILE_NAME}" ] && _TYPE1="f" && _TYPE2="File"
	[ -d "${GITHUB_WORKSPACE}/${FILE_NAME}" ] && _TYPE1="d" && _TYPE2="Folder"
	if [ -${_TYPE1} "${GITHUB_WORKSPACE}/${FILE_NAME}" ];then
		[[ -n "${FILE_RENAME}" ]] && _RENAME="${FILE_RENAME}" || _RENAME=""
		if [ -${_TYPE1} "${GITHUB_WORKSPACE}/${FILE_NAME}" ];then
			TIME "Moving [${_TYPE2}] ${FILE_NAME} to ${2}/${FILE_RENAME} ..."
			mv -f ${GITHUB_WORKSPACE}/${FILE_NAME} ${PATCH_DIR}/${_RENAME}
		else
			TIME "CustomFiles ${_TYPE2} [${FILE_NAME}] is not detected !"
		fi
	fi
	unset FILE_NAME PATCH_DIR FILE_RENAME
}

Update_Makefile() {
	PKG_NAME=${1}
	Makefile=${2}/Makefile
	[ -f "/tmp/tmp_file" ] && rm -f /tmp/tmp_file
	if [ -f "${Makefile}" ];then
		PKG_URL_MAIN="$(grep "PKG_SOURCE_URL:=" ${Makefile} | cut -c17-100)"
		_process1=${PKG_URL_MAIN##*com/}
		_process2=${_process1%%/tar*}
		api_URL="https://api.github.com/repos/${_process2}/releases"
		PKG_SOURCE_URL="$(grep "PKG_SOURCE_URL:=" ${Makefile} | cut -c17-100)"
		PKG_DL_URL="${PKG_SOURCE_URL%\$(\PKG_VERSION*}"
		Offical_Version="$(curl -s ${api_URL} 2>/dev/null | grep 'tag_name' | egrep -o '[0-9].+[0-9.]+' | awk 'NR==1')"
		if [[ -z "${Offical_Version}" ]];then
			TIME "[ERROR] Failed to obtain the Offical version of [${PKG_NAME}] !"
			return
		fi
		Source_Version="$(grep "PKG_VERSION:=" ${Makefile} | cut -c14-20)"
		Source_HASH="$(grep "PKG_HASH:=" ${Makefile} | cut -c11-100)"
		if [[ -z "${Source_Version}" ]] || [[ -z "${Source_HASH}" ]];then
			TIME "[ERROR] Failed to obtain the Source version or Hash !"
			return
		fi
		if [[ ! "${Source_Version}" == "${Offical_Version}" ]];then
			TIME "Updating package ${PKG_NAME} [${Source_Version}] to [${Offical_Version}] ..."
			sed -i "s?PKG_VERSION:=${Source_Version}?PKG_VERSION:=${Offical_Version}?g" ${Makefile}
			wget -q "${PKG_DL_URL}${Offical_Version}?" -O /tmp/tmp_file
			if [[ "$?" -eq 0 ]];then
				Offical_HASH="$(sha256sum /tmp/tmp_file | cut -d ' ' -f1)"
				sed -i "s?PKG_HASH:=${Source_HASH}?PKG_HASH:=${Offical_HASH}?g" ${Makefile}
			else
				TIME "[ERROR] Failed to update the package [${PKG_NAME}] !"
			fi
		fi
	else
		TIME "Package ${PKG_NAME} is not detected !"
	fi
	unset _process1 _process2 Offical_Version Source_Version
}