#!/usr/bin/env bash

source ./scripts/test_lib.sh

GIT_SHA=$(git rev-parse --short HEAD || echo "GitNotFound")
if [[ -n "$FAILPOINTS" ]]; then
	GIT_SHA="$GIT_SHA"-FAILPOINTS
fi

VERSION_SYMBOL="go.etcd.io/etcd/api/v3/version.GitSHA"

# Set GO_LDFLAGS="-s" for building without symbols for debugging.
# shellcheck disable=SC2206
GO_LDFLAGS=(${GO_LDFLAGS} "-X=${VERSION_SYMBOL}=${GIT_SHA}")
GO_BUILD_ENV=("CGO_ENABLED=0" "GO_BUILD_FLAGS=${GO_BUILD_FLAGS}" "GOOS=${GOOS}" "GOARCH=${GOARCH}")

# enable/disable failpoints
toggle_failpoints() {
	mode="$1"
	if command -v gofail >/dev/null 2>&1; then
		run gofail "$mode" etcdserver/ mvcc/backend/
	elif [[ "$mode" != "disable" ]]; then
		log_error "FAILPOINTS set but gofail not found"
		exit 1
	fi
}

toggle_failpoints_default() {
	mode="disable"
	if [[ -n "$FAILPOINTS" ]]; then mode="enable"; fi
	toggle_failpoints "$mode"
}

etcd_build() {
	out="bin"
	if [[ -n "${BINDIR}" ]]; then out="${BINDIR}"; fi
	toggle_failpoints_default

  run rm -f "${out}/etcd"
	# Static compilation is useful when etcd is run in a container. $GO_BUILD_FLAGS is OK
	# shellcheck disable=SC2086
	run env "${GO_BUILD_ENV[@]}" go build $GO_BUILD_FLAGS \
		-installsuffix=cgo \
		"-ldflags=${GO_LDFLAGS[*]}" \
		-o="${out}/etcd" . || return 2

	run rm -f "${out}/etcdctl"
	# shellcheck disable=SC2086
	run env CGO_ENABLED=0 GO_BUILD_FLAGS="${GO_BUILD_FLAGS}" go build $GO_BUILD_FLAGS \
		-installsuffix=cgo \
		"-ldflags=${GO_LDFLAGS[*]}" \
		-o="${out}/etcdctl" ./etcdctl || return 2

	# Verify whether symbol we overriden exists
  # For cross-compiling we cannot run: ${out}/etcd --version | grep -q "Git SHA: ${GIT_SHA}"

  # We need symbols to do this check:
  if [[ "${GO_LDFLAGS[*]}" != *"-s"* ]]; then
    go tool nm "${out}/etcd" | grep "${VERSION_SYMBOL}" > /dev/null
    if [[ "${PIPESTATUS[*]}" != "0 0" ]]; then
      log_error "FAIL: Symbol ${VERSION_SYMBOL} not found in binary: ${out}/etcd"
      return 2
    fi
  fi
}

tools_build() {
	out="bin"
	if [[ -n "${BINDIR}" ]]; then out="${BINDIR}"; fi
	tools_path="tools/benchmark
		tools/etcd-dump-db
		tools/etcd-dump-logs
		tools/local-tester/bridge"
	for tool in ${tools_path}
	do
		echo "Building" "'${tool}'"...
		run rm -f "${out}/${tool}"
		# shellcheck disable=SC2086
		run env GO_BUILD_FLAGS="${GO_BUILD_FLAGS}" CGO_ENABLED=0 go build ${GO_BUILD_FLAGS} \
		  -installsuffix=cgo \
		  "-ldflags='${GO_LDFLAGS[*]}'" \
		  -o="${out}/${tool}" "./${tool}" || return 2
	done
	tests_build "${@}"
}

tests_build() {
	out="bin"
	if [[ -n "${BINDIR}" ]]; then out="${BINDIR}"; fi
	tools_path="
		functional/cmd/etcd-agent
		functional/cmd/etcd-proxy
		functional/cmd/etcd-runner
		functional/cmd/etcd-tester"
	(
		cd tests || exit 2
		for tool in ${tools_path}; do
			echo "Building" "'${tool}'"...
			run rm -f "../${out}/${tool}"

			# shellcheck disable=SC2086
			run env CGO_ENABLED=0 GO_BUILD_FLAGS="${GO_BUILD_FLAGS}" go build ${GO_BUILD_FLAGS} \
			  -installsuffix=cgo \
			  "-ldflags='${GO_LDFLAGS[*]}'" \
			  -o="../${out}/${tool}" "./${tool}" || return 2
    done
	)
}

toggle_failpoints_default

# only build when called directly, not sourced
if echo "$0" | grep "build$" >/dev/null; then
	if etcd_build; then
	  log_success "SUCCESS: etcd_build"
	else
	  log_error "FAIL: etcd_build"
	  exit 2
  fi
fi