#!/usr/bin/env bash

function build() {
	ROOT=$(dirname "$0")
	NAME="edge-dns"
	VERSION=$(lookup-version "$ROOT"/../internal/const/const.go)
	DIST=$ROOT/"../dist/${NAME}"
	OS=${1}
	ARCH=${2}

	if [ -z "$OS" ]; then
		echo "usage: build.sh OS ARCH"
		exit
	fi
	if [ -z "$ARCH" ]; then
		echo "usage: build.sh OS ARCH"
		exit
	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} ..."
	ZIP="${NAME}-${OS}-${ARCH}-v${VERSION}.zip"

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

	cp "$ROOT"/configs/api_dns.template.yaml "$DIST"/configs
	copy_fluent_bit_assets "$ROOT" "$DIST" "$OS" "$ARCH" || exit 1

	echo "building ..."

	MUSL_DIR="/usr/local/opt/musl-cross/bin"
	CC_PATH=""
	CXX_PATH=""
	if [[ $(uname -a) == *"Darwin"* && "${OS}" == "linux" ]]; then
		# /usr/local/opt/musl-cross/bin/
		if [ "${ARCH}" == "amd64" ]; then
			CC_PATH="x86_64-linux-musl-gcc"
			CXX_PATH="x86_64-linux-musl-g++"
		fi
		if [ "${ARCH}" == "386" ]; then
			CC_PATH="i486-linux-musl-gcc"
			CXX_PATH="i486-linux-musl-g++"
		fi
		if [ "${ARCH}" == "arm64" ]; then
			CC_PATH="aarch64-linux-musl-gcc"
			CXX_PATH="aarch64-linux-musl-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="plus" -o "$DIST"/bin/${NAME} -ldflags "-linkmode external -extldflags -static -s -w" "$ROOT"/../cmd/edge-dns/main.go
	else
		env GOOS="${OS}" GOARCH="${ARCH}" CGO_ENABLED=1 go build -trimpath -tags="plus" -o "$DIST"/bin/${NAME} -ldflags="-s -w" "$ROOT"/../cmd/edge-dns/main.go
	fi

	# check build result
	RESULT=$?
    if [ "${RESULT}" != "0" ]; then
    	exit
    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 copy_fluent_bit_assets() {
	ROOT=$1
	DIST=$2
	OS=$3
	ARCH=$4
	FLUENT_ROOT="$ROOT/../../deploy/fluent-bit"
	FLUENT_DIST="$DIST/deploy/fluent-bit"

	if [ ! -d "$FLUENT_ROOT" ]; then
		echo "[error] fluent-bit source directory not found: $FLUENT_ROOT"
		return 1
	fi
	verify_fluent_bit_package_matrix "$FLUENT_ROOT" "$ARCH" || return 1

	rm -rf "$FLUENT_DIST"
	mkdir -p "$FLUENT_DIST"

	for file in fluent-bit.conf fluent-bit-dns.conf fluent-bit-https.conf fluent-bit-dns-https.conf fluent-bit-windows.conf fluent-bit-windows-https.conf parsers.conf clickhouse-upstream.conf clickhouse-upstream-windows.conf README.md; do
		if [ -f "$FLUENT_ROOT/$file" ]; then
			cp "$FLUENT_ROOT/$file" "$FLUENT_DIST/"
		fi
	done

	if [ "$OS" = "linux" ]; then
		PACKAGE_SRC="$FLUENT_ROOT/packages/linux-$ARCH"
		PACKAGE_DST="$FLUENT_DIST/packages/linux-$ARCH"
		if [ -d "$PACKAGE_SRC" ]; then
			mkdir -p "$PACKAGE_DST"
			cp -R "$PACKAGE_SRC/." "$PACKAGE_DST/"
		else
			echo "[error] fluent-bit package directory not found: $PACKAGE_SRC"
			return 1
		fi
	fi

	rm -f "$FLUENT_DIST/.gitignore"
	rm -f "$FLUENT_DIST"/logs.db*
	rm -rf "$FLUENT_DIST/storage"

	return 0
}

function verify_fluent_bit_package_matrix() {
	FLUENT_ROOT=$1
	ARCH=$2
	REQUIRED_FILES=()
	if [ "$ARCH" = "amd64" ]; then
		REQUIRED_FILES=(
			"packages/linux-amd64/fluent-bit_4.2.2_amd64.deb"
			"packages/linux-amd64/fluent-bit-4.2.2-1.x86_64.rpm"
		)
	elif [ "$ARCH" = "arm64" ]; then
		REQUIRED_FILES=(
			"packages/linux-arm64/fluent-bit_4.2.2_arm64.deb"
			"packages/linux-arm64/fluent-bit-4.2.2-1.aarch64.rpm"
		)
	else
		echo "[error] unsupported arch for fluent-bit package validation: $ARCH"
		return 1
	fi

	MISSING=0
	for FILE in "${REQUIRED_FILES[@]}"; do
		if [ ! -f "$FLUENT_ROOT/$FILE" ]; then
			echo "[error] fluent-bit matrix package missing: $FLUENT_ROOT/$FILE"
			MISSING=1
		fi
	done

	if [ "$MISSING" -ne 0 ]; then
		return 1
	fi
	return 0
}

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"