#!/usr/bin/env bash

function build() {
	ROOT=$(dirname $0)
	NAME="edge-node"
	VERSION=$(lookup-version "$ROOT"/../internal/const/const.go)
	DIST=$ROOT/"../dist/${NAME}"
	MUSL_DIR="/usr/local/opt/musl-cross/bin"
	GCC_X86_64_DIR="/usr/local/Cellar/x86_64-unknown-linux-gnu/10.3.0/bin"
	GCC_ARM64_DIR="/usr/local/Cellar/aarch64-unknown-linux-gnu/10.3.0/bin"

	OS=${1}
	ARCH=${2}
	TAG=${3}

	if [ -z "$OS" ]; then
		echo "usage: build.sh OS ARCH"
		exit
	fi
	if [ -z "$ARCH" ]; then
		echo "usage: build.sh OS ARCH"
		exit
	fi
	if [ -z "$TAG" ]; then
		TAG="community"
	fi

	echo "checking ..."
	ZIP_PATH=$(which zip)
	if [ -z "$ZIP_PATH" ]; then
		echo "we need 'zip' command to compress files"
		exit
	fi

	echo "building v${VERSION}/${OS}/${ARCH}/${TAG} ..."
	ZIP="${NAME}-${OS}-${ARCH}-${TAG}-v${VERSION}.zip"

	echo "copying ..."
	if [ ! -d "$DIST" ]; then
		mkdir "$DIST"
		mkdir "$DIST"/bin
		mkdir "$DIST"/configs
		mkdir "$DIST"/logs
		mkdir "$DIST"/data

		if [ "$TAG" = "plus" ]; then
			mkdir "$DIST"/scripts
			mkdir "$DIST"/scripts/js
		fi
	fi

	cp "$ROOT"/configs/api.template.yaml "$DIST"/configs
	cp "$ROOT"/configs/cluster.template.yaml "$DIST"/configs
	cp -R "$ROOT"/www "$DIST"/
	cp -R "$ROOT"/pages "$DIST"/

	# we support TOA on linux/amd64 only
	if [ "$OS" == "linux" ] && [ "$ARCH" == "amd64" ]
	then
		cp -R "$ROOT"/edge-toa "$DIST"
	fi

	echo "building ..."

	CC_PATH=""
	CXX_PATH=""
	BUILD_TAG=$TAG
	if [[ `uname -a` == *"Darwin"* && "${OS}" == "linux" ]]; then
		if [ "${ARCH}" == "amd64" ]; then
			# build with script support
			if [ -d $GCC_X86_64_DIR ]; then
				MUSL_DIR=$GCC_X86_64_DIR
				CC_PATH="x86_64-unknown-linux-gnu-gcc"
				CXX_PATH="x86_64-unknown-linux-gnu-g++"
				if [ "$TAG" = "plus" ]; then
					BUILD_TAG="plus,script"
				fi
			else
				CC_PATH="x86_64-linux-musl-gcc"
				CXX_PATH="x86_64-linux-musl-g++"
			fi
		fi
		if [ "${ARCH}" == "386" ]; then
			CC_PATH="i486-linux-musl-gcc"
			CXX_PATH="i486-linux-musl-g++"
		fi
		if [ "${ARCH}" == "arm64" ]; then
			# build with script support
			if [ -d $GCC_ARM64_DIR ]; then
				MUSL_DIR=$GCC_ARM64_DIR
				CC_PATH="aarch64-unknown-linux-gnu-gcc"
				CXX_PATH="aarch64-unknown-linux-gnu-g++"
				if [ "$TAG" = "plus" ]; then
					BUILD_TAG="plus,script"
				fi
			else
				CC_PATH="aarch64-linux-musl-gcc"
				CXX_PATH="aarch64-linux-musl-g++"
			fi
		fi
		if [ "${ARCH}" == "arm" ]; then
			CC_PATH="arm-linux-musleabi-gcc"
			CXX_PATH="arm-linux-musleabi-g++"
		fi
		if [ "${ARCH}" == "mips64" ]; then
			CC_PATH="mips64-linux-musl-gcc"
			CXX_PATH="mips64-linux-musl-g++"
		fi
		if [ "${ARCH}" == "mips64le" ]; then
			CC_PATH="mips64el-linux-musl-gcc"
			CXX_PATH="mips64el-linux-musl-g++"
		fi
	fi
	if [ ! -z $CC_PATH ]; then
		env CC=$MUSL_DIR/$CC_PATH CXX=$MUSL_DIR/$CXX_PATH GOOS="${OS}" GOARCH="${ARCH}" CGO_ENABLED=1 go build -trimpath -tags $BUILD_TAG -o "$DIST"/bin/${NAME} -ldflags "-linkmode external -extldflags -static -s -w" "$ROOT"/../cmd/edge-node/main.go
	else
		if [[ `uname` == *"Linux"* ]] && [ "$OS" == "linux" ] && [[ "$ARCH" == "amd64" || "$ARCH" == "arm64" ]] &&  [ "$TAG" == "plus" ]; then
			BUILD_TAG="plus,script"
		fi
		env GOOS="${OS}" GOARCH="${ARCH}" CGO_ENABLED=1 go build -trimpath -tags $BUILD_TAG -o "$DIST"/bin/${NAME} -ldflags="-s -w" "$ROOT"/../cmd/edge-node/main.go
	fi

	# delete hidden files
	find "$DIST" -name ".DS_Store" -delete
	find "$DIST" -name ".gitignore" -delete

	echo "zip files"
	cd "${DIST}/../" || exit
	if [ -f "${ZIP}" ]; then
		rm -f "${ZIP}"
	fi
	zip -r -X -q "${ZIP}" ${NAME}/
	rm -rf ${NAME}
	cd - || exit

	echo "OK"
}

function lookup-version() {
	FILE=$1
	VERSION_DATA=$(cat "$FILE")
	re="Version[ ]+=[ ]+\"([0-9.]+)\""
	if [[ $VERSION_DATA =~ $re ]]; then
		VERSION=${BASH_REMATCH[1]}
		echo "$VERSION"
	else
		echo "could not match version"
		exit
	fi
}

build "$1" "$2" "$3"
