#!/bin/bash
################################################## ATTENTION ###################################################
# Instruction: Run on the build server as a BUILD CAPABLE USER (ROOT is assumed in this example).
# This script is modified from OWRT_Builder script.
# This script is only intended to work with Teltonika SDK RUT9_R_00.07.xx
################################################################################################################

###################################################
# Set initial variables/functions.
ZT_TUNVER="latest"
ZT_TUNBRANCH=""
ZT_TUNURL="https://github.com/openziti/ziti-tunnel-sdk-c"
VCPKG_URL="https://github.com/microsoft/vcpkg"
LIBCAP_URL="https://github.com/OpenLD/libcap"
ZT_WORKDIR="/home/ziggy/rutos-ath79-rut9-gpl"
SDK_DIRNAME="rutos-ath79-rut9-gpl"

################################################################################################################
# DO NOT MODIFY BELOW THIS LINE
################################################################################################################
ZT_CMAKEVER="0.0.0"
VCPKG_ROOT="${ZT_WORKDIR}/vcpkg"
LIBCAP_ROOT="${ZT_WORKDIR}/libcap"
ZT_STEP="0"
ZT_ADDLPKG=("gawk" "sed" "git" "zip" "unzip" "wget" "build-essential" "autoconf" "libtool" "pkg-config" "autopoint")
for ((i=0;i<100;i++)); do PRINT_PADDING+='          '; done
function CPrint() {
	local OUT_COLOR=(${1/:/ }) IN_TEXT="${2}" OUT_MAXWIDTH="${3}" OUT_SCREENWIDTH NL_INCLUDE i x z
	shopt -s checkwinsize; (:); OUT_SCREENWIDTH="${COLUMNS:-$(tput cols 2>/dev/null || echo 80)}";
	if [[ -z ${OUT_MAXWIDTH} ]]; then
		OUT_MAXWIDTH="${OUT_SCREENWIDTH:-80}"
		NL_INCLUDE='\n'
	elif [[ ${OUT_MAXWIDTH} -eq "0" ]]; then
		OUT_MAXWIDTH="${#IN_TEXT}"
		NL_INCLUDE='\n'
	elif [[ ${OUT_MAXWIDTH} -lt "0" ]]; then
		OUT_MAXWIDTH="${#IN_TEXT}"
		NL_INCLUDE=''
	fi
	[[ ${#IN_TEXT} -gt ${OUT_MAXWIDTH} ]] && IN_TEXT="${IN_TEXT:0:${OUT_MAXWIDTH}}"
	if [[ ${OUT_COLOR} == "COLORTEST" ]]; then
		OUT_MAXWIDTH="10"
		for i in {1..107}; do
			for x in {1..107}; do
				[[ $((++z%(OUT_SCREENWIDTH/OUT_MAXWIDTH))) -eq 0 ]] && echo
				IN_TEXT="${i}:${x}"
				printf "\e[${i};${x}m%-${OUT_MAXWIDTH}s\e[1;0m" "${PRINT_PADDING:0:$(((OUT_MAXWIDTH/2)-${#IN_TEXT}/2))}${IN_TEXT}"
			done
		done && echo
	else
		printf "\e[${OUT_COLOR[0]};${OUT_COLOR[1]}m%-${OUT_MAXWIDTH}s\e[1;0m${NL_INCLUDE}" "${PRINT_PADDING:0:$(((OUT_MAXWIDTH/2)-${#IN_TEXT}/2))}${IN_TEXT}"
	fi
}
function GTE() {
	CPrint "30:41" "ERROR: Early Exit at Step ${1}."
	exit ${1}
}

###################################################
CPrint "30:42" "ZITI EDGE TUNNEL VERSION: ${ZT_TUNVER:=UNKNOWN} (BRANCH:${ZT_TUNBRANCH:-MAIN})"

###################################################
CPrint "30:43" "Begin Step $((++ZT_STEP)): Create Staging Area."
CPrint "30:47" "Location:" "-1" && echo " ${ZT_WORKDIR}"
if [[ -d "${ZT_WORKDIR}" ]]; then
	CPrint "30:45" "WARNING: Staging Area Already Exists." "0"
fi

###################################################
CPrint "30:43" "Begin Step $((++ZT_STEP)): Acquire Additional Software Part One."
apt update
apt install -y ${ZT_ADDLPKG[@]} || GTE ${ZT_STEP}
IFS='\.' read -r -d '' -a ZT_CMAKEVER < <(\
	awk '/^cmake version/ {
		CMAKEVER=gensub(/cmake version (.*)/,"\\1","1")
		print CMAKEVER
	}' <(cmake --version 2>/dev/null)
)
if [[ ${ZT_CMAKEVER[0]:-0} -ge 3 ]] && [[ ${ZT_CMAKEVER[1]:-0} -ge 24 ]]; then
	CPrint "30:47" "CMake Version:" "-1" && echo " ${ZT_CMAKEVER[0]:-0}.${ZT_CMAKEVER[1]:-0}.${ZT_CMAKEVER[2]:-0}"
else
	CPrint "30:41" "ERROR: CMake Version Installed [${ZT_CMAKEVER[0]:-0}.${ZT_CMAKEVER[1]:-0}.${ZT_CMAKEVER[2]:-0}]." "0"
	CPrint "30:41" "ERROR: CMake Version Required [3.24.0+]." "0"
	CPrint "30:41" "See [https://apt.kitware.com] for Latest Version." "0"
	GTE ${ZT_STEP}
fi

###################################################
CPrint "30:43" "Begin Step $((++ZT_STEP)): Acquire ZITI EDGE TUNNEL Source [Version ${ZT_TUNVER}]."
if [[ ${ZT_TUNVER:-latest} == "latest" ]]; then
	IFS=$'\n' read -r -d '' -a ZT_ALLVERSIONS < <(\
		wget "${ZT_TUNURL}/tags" -O- 2>/dev/null \
		| awk '/tags/ {
			if (match($0,/v[0-9].[0-9]+.[0-9]+/)) {
				ALLVERSIONS[substr($0,RSTART,RLENGTH)]++}
			} END {
				for (EACHVERSION in ALLVERSIONS) {
					gsub("v","",EACHVERSION)
					print EACHVERSION
				}
			}' \
		| sort -rnt '.' -k1,1 -k2,2 -k3,3
	)
	ZT_TUNVER="${ZT_ALLVERSIONS[0]}"
fi

# VCPKG support began with ZITI-TUNNEL-SDK-C version 0.21.1.
ZT_USEVCPKG="TRUE"

if [[ -d "${ZT_WORKDIR}/ziti-tunnel-sdk-c-${ZT_TUNVER}" ]]; then
	CPrint "30:45" "Tunnel source code Already Exists: ${ZT_WORKDIR}/ziti-tunnel-sdk-c-${ZT_TUNVER}, do not overwite." "0"
else
	git clone --branch "${ZT_TUNBRANCH:-v${ZT_TUNVER}}" "${ZT_TUNURL}" "${ZT_WORKDIR}/ziti-tunnel-sdk-c-${ZT_TUNVER}" || GTE ${ZT_STEP}
	mkdir -vp "${ZT_WORKDIR}/ziti-tunnel-sdk-c-${ZT_TUNVER}/build" || GTE ${ZT_STEP}

fi


###################################################
if [[ ${ZT_USEVCPKG} == "TRUE" ]]; then
	CPrint "30:43" "Begin Step $((++ZT_STEP)): Acquire Additional Software Part Two - VCPKG Required."
	if [[ -d "${VCPKG_ROOT}" ]]; then
		CPrint "30:45" "VCPKG Already Exists: ${VCPKG_ROOT}" "0"
	else
		git clone "${VCPKG_URL}" "${VCPKG_ROOT}" || GTE ${ZT_STEP}
		${VCPKG_ROOT}/bootstrap-vcpkg.sh -disableMetrics || GTE ${ZT_STEP}
		${VCPKG_ROOT}/vcpkg version >&1 >/dev/null || GTE ${ZT_STEP}
		mkdir -vp "${VCPKG_ROOT}/custom-triplets" || GTE ${ZT_STEP}
	fi
else
	CPrint "30:43" "Skipping Step $((++ZT_STEP)): Acquire Additional Software Part Two - VCPKG Not Required."
fi

###################################################
CPrint "30:43" "Begin Step $((++ZT_STEP)): Checking Teltonika SDK "
export STAGING_DIR="${ZT_WORKDIR}/${SDK_DIRNAME}/staging_dir"
echo ${STAGING_DIR}
find "${STAGING_DIR}" -maxdepth 1 -name "target*" -printf "%P"
find "${STAGING_DIR}" -maxdepth 1 -name "toolchain*"
ZT_OWRT_BUILDTARGET="$(find "${STAGING_DIR}" -maxdepth 1 -name "target*" -printf "%P" || GTE ${ZT_STEP})"
ZT_OWRT_BUILDTOOLCHAIN[0]="$(find "${STAGING_DIR}" -maxdepth 1 -name "toolchain*" -printf "%P" || GTE ${ZT_STEP})"
ZT_OWRT_BUILDTOOLCHAIN[1]="${STAGING_DIR}/${ZT_OWRT_BUILDTOOLCHAIN[0]}"

###################################################
CPrint "30:43" "Begin Step $((++ZT_STEP)): Setup Build Environment Part One [Target ${ZT_OWRT_BUILDTARGET}]."
source <(find "${ZT_WORKDIR}" -type f -name info.mk -exec cat {} \;) || GTE ${ZT_STEP}
ZT_OWRT_TCINFO=(${TARGET_CROSS//-/ })
ZT_OWRT_TRIPLE="${ZT_OWRT_TCINFO[0]}-${ZT_OWRT_TCINFO[1]}-${ZT_OWRT_TCINFO[2]}"
ZT_OWRT_GCC="$(find "${ZT_WORKDIR}" -type l -name ${ZT_OWRT_TRIPLE}-gcc || GTE ${ZT_STEP})"
ZT_OWRT_GPP="$(find "${ZT_WORKDIR}" -type l -name ${ZT_OWRT_TRIPLE}-g++ || GTE ${ZT_STEP})"
cat << EOFEOF > "${ZT_WORKDIR}/toolchain.cmake"
set(triple "${ZT_OWRT_TRIPLE}")
set(CMAKE_SYSTEM_NAME Linux)
set(CMAKE_SYSTEM_PROCESSOR "${ZT_OWRT_TCINFO[0]}")
set(CMAKE_SYSROOT "${ZT_OWRT_BUILDTOOLCHAIN[1]}")
set(CMAKE_C_COMPILER "${ZT_OWRT_GCC}")
set(CMAKE_CXX_COMPILER "${ZT_OWRT_GPP}")
set(INCLUDE_DIRECTORIES "${VCPKG_ROOT}/installed/${ZT_OWRT_TCINFO[0]}-linux")
set(CMAKE_FIND_ROOT_PATH_MODE_PROGRAM NEVER)
set(CMAKE_FIND_ROOT_PATH_MODE_LIBRARY ONLY)
set(CMAKE_FIND_ROOT_PATH_MODE_INCLUDE ONLY)
set(CMAKE_FIND_ROOT_PATH_MODE_PACKAGE ONLY)
EOFEOF
CPrint "30:47" "TOOLCHAIN:" "0"
awk '{print "\t"$0}' "${ZT_WORKDIR}/toolchain.cmake" || GTE ${ZT_STEP}
if [[ ${ZT_USEVCPKG} == "TRUE" ]]; then
cat << EOFEOF > "${VCPKG_ROOT}/custom-triplets/${ZT_OWRT_TCINFO[0]}-linux.cmake"
set(VCPKG_TARGET_ARCHITECTURE "${ZT_OWRT_TCINFO[0]}")
set(VCPKG_CRT_LINKAGE "dynamic")
set(VCPKG_LIBRARY_LINKAGE "static")
set(VCPKG_CMAKE_SYSTEM_NAME "Linux")
set(VCPKG_CHAINLOAD_TOOLCHAIN_FILE "${ZT_WORKDIR}/toolchain.cmake")
EOFEOF
CPrint "30:47" "VCPKG_TRIPLE:" "0"
awk '{print "\t"$0}' "${VCPKG_ROOT}/custom-triplets/${ZT_OWRT_TCINFO[0]}-linux.cmake" || GTE ${ZT_STEP}
fi

###################################################
if grep -q "LIBCAP REQUIRED" ${ZT_WORKDIR}/ziti-tunnel-sdk-c-${ZT_TUNVER}/programs/ziti-edge-tunnel/CMakeLists.txt; then
	CPrint "30:43" "Begin Step $((++ZT_STEP)): Acquire Additional Software Part Three - LIBCAP."
	git clone "${LIBCAP_URL}" "${LIBCAP_ROOT}" || GTE ${ZT_STEP}
	make -C "${LIBCAP_ROOT}" BUILD_CC="gcc" CC="${ZT_OWRT_GCC}" || GTE ${ZT_STEP}
	cp -vr "${LIBCAP_ROOT}/libcap/"* "${ZT_OWRT_BUILDTOOLCHAIN[1]}/lib" || GTE ${ZT_STEP}
	cp -vr "${LIBCAP_ROOT}/libcap/include" "${ZT_OWRT_BUILDTOOLCHAIN[1]}" || GTE ${ZT_STEP}
else
	CPrint "30:43" "Skipping Step $((++ZT_STEP)): Acquire Additional Software Part Three - LIBCAP not required."
fi

###################################################
CPrint "30:43" "Begin Step $((++ZT_STEP)): Setup Build Environment Part Two [Target ${ZT_OWRT_BUILDTARGET}]."
ZT_OWRT_CMAKEOPTS[((i++))]="-DDISABLE_LIBSYSTEMD_FEATURE=ON"
[[ -x "/usr/bin/ninja" ]] && ZT_OWRT_CMAKEOPTS[((i++))]="-G Ninja"
ZT_OWRT_CMAKEOPTS[((i++))]="-DTLSUV_TLSLIB=mbedtls"
ZT_OWRT_CMAKEOPTS[((i++))]="-DUSE_MBEDTLS=ON"
ZT_OWRT_CMAKEOPTS[((i++))]="-DHAVE_LIBSODIUM=ON"
ZT_OWRT_CMAKEOPTS[((i++))]="-DCMAKE_TOOLCHAIN_FILE=${ZT_WORKDIR}/toolchain.cmake"
ZT_OWRT_CMAKEOPTS[((i++))]="-DGIT_VERSION=${ZT_TUNVER}"
ZT_OWRT_CMAKEOPTS[((i++))]="-S ${ZT_WORKDIR}/ziti-tunnel-sdk-c-${ZT_TUNVER}"
ZT_OWRT_CMAKEOPTS[((i++))]="-B ${ZT_WORKDIR}/ziti-tunnel-sdk-c-${ZT_TUNVER}/build"

###################################################
if [[ ${ZT_USEVCPKG} == "TRUE" ]]; then
	CPrint "30:43" "Begin Step $((++ZT_STEP)): Build Dependencies via VCPKG [Target ${ZT_OWRT_BUILDTARGET}]."
	cd "${VCPKG_ROOT}"
	./vcpkg install zlib mbedtls llhttp libsodium libuv acl attr --triplet ${ZT_OWRT_TCINFO[0]}-linux --overlay-triplets=custom-triplets || GTE ${ZT_STEP}
	cp -vr "installed/${ZT_OWRT_TCINFO[0]}-linux"/* "${ZT_OWRT_BUILDTOOLCHAIN[1]}" || GTE ${ZT_STEP}
	cd -
else
	CPrint "30:43" "Skipping Step $((++ZT_STEP)): Build Dependencies via VCPKG - Not Required."
fi

###################################################
CPrint "30:43" "Begin Step $((++ZT_STEP)): Configure Build [Target ${ZT_OWRT_BUILDTARGET}]."
CPrint "30:47" "CMAKE SYNTAX:" "-1" && echo " [cmake ${ZT_OWRT_CMAKEOPTS[@]}]"
cmake ${ZT_OWRT_CMAKEOPTS[@]} || GTE ${ZT_STEP}
# Note: This is only required on pre-0.21.6 releases of ZET-CSDK.
#  The prior versions failed to build due to preprocessor error on metrics.h as it was expecting a macro to be present.
#  In other toolchains the included features.h has the macro, and it seems only OpenWRT doesn't include it.
#  This will only fire if it exactly matches, which applies only to pre-0.21.6 releases.
CPrint "30:43" "Begin Step $((++ZT_STEP)): Pre-Build Modifications."
sed -i '/# if ! __GNUC_PREREQ(4,9)/,+2d' $(find "${ZT_WORKDIR}/ziti-tunnel-sdk-c-${ZT_TUNVER}" -name "metrics.h") || GTE ${ZT_STEP}
if [[ ${ZT_USEVCPKG} != "TRUE" ]]; then
	cp -vr "/usr/include/sodium.h" "/usr/include/sodium" "${ZT_WORKDIR}/ziti-tunnel-sdk-c-${ZT_TUNVER}/build/_deps/ziti-sdk-c-src/includes" || GTE ${ZT_STEP}
fi

###################################################
CPrint "30:43" "Begin Step $((++ZT_STEP)): Build [Target ${ZT_OWRT_BUILDTARGET}]."
CPrint "30:47" "CMAKE SYNTAX:" "-1" && echo " [cmake --build \"${ZT_WORKDIR}/ziti-tunnel-sdk-c-${ZT_TUNVER}/build\" --target \"ziti-edge-tunnel\"]"
cmake --build "${ZT_WORKDIR}/ziti-tunnel-sdk-c-${ZT_TUNVER}/build" --target "ziti-edge-tunnel" || GTE ${ZT_STEP}

###################################################
CPrint "30:43" "Begin Step $((++ZT_STEP)): Compress and Move Binary."
CPrint "30:47" "Location:" "-1" && echo " [${ZT_WORKDIR%\/*}/${ZT_WORKDIR##*\/}.gz]"
gzip -ck9 "${ZT_WORKDIR}/ziti-tunnel-sdk-c-${ZT_TUNVER}/build/programs/ziti-edge-tunnel/ziti-edge-tunnel" > "${ZT_WORKDIR%\/*}/${ZT_WORKDIR##*\/}.gz" || GTE ${ZT_STEP}

###################################################
CPrint "30:42" "Compile and Build Complete."