#!/bin/bash
set -o errexit

# define package name
WORK_DIR=/go/src/github.com/goodrain/rainbond
BASE_NAME=rainbond
IMAGE_BASE_NAME=${BUILD_IMAGE_BASE_NAME:-'rainbond'}
DOMESTIC_NAMESPACE=${DOMESTIC_NAMESPACE:-'goodrain'}

GO_VERSION=1.13

GOPROXY=${GOPROXY:-'https://goproxy.io'}

if [ "$DISABLE_GOPROXY" == "true" ]; then
	GOPROXY=
fi
if [ -z "$GOOS" ]; then
	GOOS="linux"
fi
if [ "$DEBUG" ]; then
	set -x
fi
BRANCH=$(git symbolic-ref HEAD 2>/dev/null | cut -d"/" -f 3)
if [ -z "$VERSION" ]; then
	if [ -z "$TRAVIS_TAG" ]; then
		if [ -z "$TRAVIS_BRANCH" ]; then
			VERSION=$BRANCH-dev
		else
			VERSION=$TRAVIS_BRANCH-dev
		fi
	else
		VERSION=$TRAVIS_TAG
	fi
fi

buildTime=$(date +%F-%H)
git_commit=$(git log -n 1 --pretty --format=%h)

release_desc=${VERSION}-${git_commit}-${buildTime}
core_modul_items=(api chaos gateway worker node-proxy eventlog mq)
addon_modul_items=(monitor init-probe mesh-data-panel grctl resource-proxy)

build::binary() {
	echo "---> build binary:$1"
	home=$(pwd)
	local go_mod_cache="${home}/.cache"
	local OUTPATH="./_output/binary/$GOOS/${BASE_NAME}-$1"
	local DOCKER_PATH="./hack/contrib/docker/$2/$1"
	local build_image="golang:${GO_VERSION}"
	local build_args="-w -s -X github.com/goodrain/rainbond/cmd.version=${release_desc}"
	local build_dir="./cmd/$1"
	local build_tag=""
	if [ -f "${DOCKER_PATH}/ignorebuild" ]; then
		return
	fi
	CGO_ENABLED=1
	if [ "$1" = "eventlog" ]; then
		docker build -t goodraim.me/event-build:v1 "${DOCKER_PATH}/build"
		build_image="goodraim.me/event-build:v1"
	elif [ "$1" = "gateway" ]; then
		build_image="golang:1.13-alpine"
	elif [ "$1" = "monitor" ]; then
		CGO_ENABLED=0
	fi
	docker run --rm -e CGO_ENABLED=${CGO_ENABLED} -e GOPROXY=${GOPROXY} -e GOOS="${GOOS}" -v "${go_mod_cache}":/go/pkg/mod -v "$(pwd)":${WORK_DIR} -w ${WORK_DIR} ${build_image} go build -ldflags "${build_args}" -tags "${build_tag}" -o "${OUTPATH}" ${build_dir}
	if [ "$GOOS" = "windows" ]; then
		mv "$OUTPATH" "${OUTPATH}.exe"
	fi
}

build::image() {
	local OUTPATH="./_output/binary/$GOOS/${BASE_NAME}-$1"
	local build_image_dir="./_output/image/$1/"
	local source_dir="./hack/contrib/docker/$3/$1"
	mkdir -p "${build_image_dir}"
	chmod 777 "${build_image_dir}"
	if [ ! -f "${source_dir}/ignorebuild" ]; then
		if [ ! ${CACHE} ] || [ ! -f "${OUTPATH}" ]; then
			build::binary "$1" "$3"
		fi
		cp "${OUTPATH}" "${build_image_dir}"
	fi
	cp -r ${source_dir}/* "${build_image_dir}"
	pushd "${build_image_dir}"
	echo "---> build image:$1"
	docker build --build-arg RELEASE_DESC="${release_desc}" -t "${IMAGE_BASE_NAME}/rbd-$1:${VERSION}" -f Dockerfile .
	docker run --rm "${IMAGE_BASE_NAME}/rbd-$1:${VERSION}" version
	if [ $? -ne 0 ]; then
		echo "image version is different ${release_desc}"
		exit 1
	fi
	if [ -f "${source_dir}/test.sh" ]; then
		"${source_dir}/test.sh" "${IMAGE_BASE_NAME}/rbd-$1:${VERSION}"
	fi
	if [ "$2" = "push" ]; then
		if [ $DOCKER_USERNAME ]; then
			docker login -u "$DOCKER_USERNAME" -p "$DOCKER_PASSWORD"
		fi
		docker push "${IMAGE_BASE_NAME}/rbd-$1:${VERSION}"
		if [ "${DOMESTIC_BASE_NAME}" ]; then
			docker tag "${IMAGE_BASE_NAME}/rbd-$1:${VERSION}" "${DOMESTIC_BASE_NAME}/${DOMESTIC_NAMESPACE}/rbd-$1:${VERSION}"
			docker login -u "$DOMESTIC_DOCKER_USERNAME" -p "$DOMESTIC_DOCKER_PASSWORD" "${DOMESTIC_BASE_NAME}"
			docker push "${DOMESTIC_BASE_NAME}/${DOMESTIC_NAMESPACE}/rbd-$1:${VERSION}"
		fi
	fi
	popd
	rm -rf "${build_image_dir}"
}

build::image::all() {
	for item in "${core_modul_items[@]}"; do
		build::image "$item" "$1" "core"
	done

	for item in "${addon_modul_items[@]}"; do
		build::image "$item" "$1" "addon"
	done
}

build::image::core() {
	for item in "${core_modul_items[@]}"; do
		build::image "$item" "$1" "core"
	done
}

build::binary::all() {
	for item in "${core_modul_items[@]}"; do
		build::binary "$item" "core"
	done

	for item in "${addon_modul_items[@]}"; do
		build::binary "$item" "addon"
	done
}

build::binary::core() {
	for item in "${core_modul_items[@]}"; do
		build::binary "$item" "core"
	done
}

case $1 in
binary)
	if [ "$2" = "all" ]; then
		build::binary::all "$2"
	elif [ "$2" = "core" ]; then
		build::binary::core "$2"
	else
		if [[ "${core_modul_items[*]}" =~ "${1}" ]]; then
			build::binary "$2" "$3" "core"
		else
			build::binary "$2" "$3" "addon"
		fi
	fi
	;;
*)
	if [ "$1" = "all" ]; then
		build::image::all "$2"
	elif [ "$1" = "core" ]; then
		build::image::core "$2"
	else
		if [[ "${core_modul_items[*]}" =~ "${1}" ]]; then
			build::image "$1" "$2" "core"
		else
			build::image "$1" "$2" "addon"
		fi
	fi
	;;
esac