#!/usr/bin/make -f
# -*- makefile -*-

## default globals
TARGET ?= emqx/emqx
QEMU_ARCH ?= x86_64
ARCH ?= amd64
QEMU_VERSION ?= v5.0.0-2
OS ?= alpine
export PKG_VSN ?= $(shell $(CURDIR)/pkg-vsn.sh)

ifeq ($(findstring emqx-ee, $(TARGET)), emqx-ee)
	ARCH_LIST := amd64 arm64v8 arm32v7
	EMQX_NAME := emqx-ee
else ifeq ($(findstring emqx-edge, $(TARGET)), emqx-edge)
	ARCH_LIST := amd64 arm64v8 arm32v7 i386 s390x
	EMQX_NAME := emqx-edge
else
	ARCH_LIST := amd64 arm64v8 arm32v7 i386 s390x
	EMQX_NAME := emqx
endif

.PHONY: docker
docker: docker-build docker-tag docker-save

.PHONY: docker-prepare
docker-prepare:
	## Prepare the machine before any code installation scripts
	# @echo "PREPARE: Setting up dependencies."
	# @apt update -y
	# @apt install --only-upgrade docker-ce -y

	## Update docker configuration to enable docker manifest command
	@echo "PREPARE: Updating docker configuration"
	@mkdir -p $$HOME/.docker

	# enable experimental to use docker manifest command
	@echo '{ "experimental": "enabled" }' | tee $$HOME/.docker/config.json
	# enable experimental
	@echo '{ "experimental": true, "storage-driver": "overlay2", "max-concurrent-downloads": 50, "max-concurrent-uploads": 50 }' | tee /etc/docker/daemon.json 
	@service docker restart

.PHONY: docker-build
docker-build:
	## Build Docker image
	@echo "DOCKER BUILD: Build Docker image."
	@echo "DOCKER BUILD: build version -> $(PKG_VSN)."
	@echo "DOCKER BUILD: arch - $(ARCH)."
	@echo "DOCKER BUILD: qemu arch - $(QEMU_ARCH)."
	@echo "DOCKER BUILD: docker repo - $(TARGET) "
	@echo "DOCKER BUILD: emqx name - $(EMQX_NAME)."

	## Prepare qemu to build images other then x86_64 on travis
	@echo "PREPARE: Qemu" \
	&& docker run --rm --privileged multiarch/qemu-user-static:register --reset

	@mkdir -p tmp \
	&& cd tmp \
	&& curl -L -o qemu-$(QEMU_ARCH)-static.tar.gz https://github.com/multiarch/qemu-user-static/releases/download/$(QEMU_VERSION)/qemu-$(QEMU_ARCH)-static.tar.gz \
	&& tar xzf qemu-$(QEMU_ARCH)-static.tar.gz \
	&& cd -

	@docker build --no-cache \
		--build-arg PKG_VSN=$(PKG_VSN)  \
		--build-arg BUILD_FROM=emqx/build-env:erl23.2.7.2-emqx-2-alpine-$(ARCH)  \
		--build-arg RUN_FROM=$(ARCH)/alpine:3.12 \
		--build-arg EMQX_NAME=$(EMQX_NAME) \
		--build-arg QEMU_ARCH=$(QEMU_ARCH) \
		--tag $(TARGET):build-$(OS)-$(ARCH) \
		-f deploy/docker/Dockerfile .

.PHONY: docker-tag
docker-tag:
	@echo "DOCKER TAG: Tag Docker image."
	@for arch in $(ARCH_LIST); do \
		if [ -n  "$$(docker images -q $(TARGET):build-$(OS)-$${arch})" ]; then \
			docker tag $(TARGET):build-$(OS)-$${arch}  $(TARGET):$(PKG_VSN)-$(OS)-$${arch}; \
			echo "DOCKER TAG: $(TARGET):$(PKG_VSN)-$(OS)-$${arch}"; \
			if [ $${arch} = amd64 ]; then \
				docker tag $(TARGET):$(PKG_VSN)-$(OS)-amd64 $(TARGET):$(PKG_VSN); \
				echo "DOCKER TAG: $(TARGET):$(PKG_VSN)"; \
			fi; \
		fi; \
	done

.PHONY: docker-save
docker-save:
	@echo "DOCKER SAVE: Save Docker image." 

	@mkdir -p _packages/$(EMQX_NAME)

	@if [ -n  "$$(docker images -q $(TARGET):$(PKG_VSN))" ]; then \
		docker save $(TARGET):$(PKG_VSN) > $(EMQX_NAME)-docker-$(PKG_VSN); \
		zip -r -m $(EMQX_NAME)-docker-$(PKG_VSN).zip $(EMQX_NAME)-docker-$(PKG_VSN); \
		mv ./$(EMQX_NAME)-docker-$(PKG_VSN).zip _packages/$(EMQX_NAME)/$(EMQX_NAME)-docker-$(PKG_VSN).zip; \
	fi
	
	@for arch in $(ARCH_LIST); do \
		if [ -n  "$$(docker images -q  $(TARGET):$(PKG_VSN)-$(OS)-$${arch})" ]; then \
			docker save  $(TARGET):$(PKG_VSN)-$(OS)-$${arch} > $(EMQX_NAME)-docker-$(PKG_VSN)-$(OS)-$${arch}; \
			zip -r -m $(EMQX_NAME)-docker-$(PKG_VSN)-$(OS)-$${arch}.zip $(EMQX_NAME)-docker-$(PKG_VSN)-$(OS)-$${arch}; \
			mv ./$(EMQX_NAME)-docker-$(PKG_VSN)-$(OS)-$${arch}.zip _packages/$(EMQX_NAME)/$(EMQX_NAME)-docker-$(PKG_VSN)-$(OS)-$${arch}.zip; \
		fi; \
	done

.PHONY: docker-push
docker-push:
	@echo "DOCKER PUSH: Push Docker image."; 
	@echo "DOCKER PUSH: pushing - $(TARGET):$(PKG_VSN)."; 

	@if [ -n "$$(docker images -q $(TARGET):$(PKG_VSN))" ]; then \
		docker push $(TARGET):$(PKG_VSN); \
		docker tag $(TARGET):$(PKG_VSN) $(TARGET):latest; \
		docker push $(TARGET):latest; \
	fi;

	@for arch in $(ARCH_LIST); do \
		if [ -n "$$(docker images -q  $(TARGET):$(PKG_VSN)-$(OS)-$${arch})" ]; then \
			docker push  $(TARGET):$(PKG_VSN)-$(OS)-$${arch}; \
		fi; \
	done

.PHONY: docker-manifest-list
docker-manifest-list:
	version="docker manifest create --amend $(TARGET):$(PKG_VSN)"; \
	latest="docker manifest create --amend $(TARGET):latest"; \
	for arch in $(ARCH_LIST); do \
		if [ -n "$$(docker images -q  $(TARGET):$(PKG_VSN)-$(OS)-$${arch})" ];then \
			version="$${version} $(TARGET):$(PKG_VSN)-$(OS)-$${arch} "; \
			latest="$${latest} $(TARGET):$(PKG_VSN)-$(OS)-$${arch} "; \
		fi; \
	done; \
	eval $$version; \
	eval $$latest; 

	for arch in $(ARCH_LIST); do \
		case $${arch} in \
			"amd64") \
				if [ -n "$$(docker images -q  $(TARGET):$(PKG_VSN)-$(OS)-$${arch})" ]; then \
					docker manifest annotate $(TARGET):$(PKG_VSN) $(TARGET):$(PKG_VSN)-$(OS)-amd64 --os=linux --arch=amd64; \
					docker manifest annotate $(TARGET):latest $(TARGET):$(PKG_VSN)-$(OS)-amd64 --os=linux --arch=amd64; \
				fi; \
				;; \
			"arm64v8") \
				if [ -n "$$(docker images -q  $(TARGET):$(PKG_VSN)-$(OS)-$${arch})" ]; then \
					docker manifest annotate $(TARGET):$(PKG_VSN) $(TARGET):$(PKG_VSN)-$(OS)-arm64v8 --os=linux --arch=arm64 --variant=v8; \
					docker manifest annotate $(TARGET):latest $(TARGET):$(PKG_VSN)-$(OS)-arm64v8 --os=linux --arch=arm64 --variant=v8; \
				fi; \
				;; \
			"arm32v7") \
				if [ -n "$$(docker images -q  $(TARGET):$(PKG_VSN)-$(OS)-$${arch})" ]; then \
					docker manifest annotate $(TARGET):$(PKG_VSN) $(TARGET):$(PKG_VSN)-$(OS)-arm32v7 --os=linux --arch=arm --variant=v7; \
					docker manifest annotate $(TARGET):latest $(TARGET):$(PKG_VSN)-$(OS)-arm32v7 --os=linux --arch=arm --variant=v7; \
				fi; \
				;; \
			"i386") \
				if [ -n "$$(docker images -q  $(TARGET):$(PKG_VSN)-$(OS)-$${arch})" ]; then \
					docker manifest annotate $(TARGET):$(PKG_VSN) $(TARGET):$(PKG_VSN)-$(OS)-i386 --os=linux --arch=386; \
					docker manifest annotate $(TARGET):latest $(TARGET):$(PKG_VSN)-$(OS)-i386 --os=linux --arch=386; \
				fi; \
				;; \
			"s390x") \
				if [ -n "$$(docker images -q  $(TARGET):$(PKG_VSN)-$(OS)-$${arch})" ]; then \
					docker manifest annotate $(TARGET):$(PKG_VSN) $(TARGET):$(PKG_VSN)-$(OS)-s390x --os=linux --arch=s390x; \
					docker manifest annotate $(TARGET):latest $(TARGET):$(PKG_VSN)-$(OS)-s390x --os=linux --arch=s390x; \
				fi; \
				;; \
		esac; \
	done; 

	docker manifest inspect $(TARGET):$(PKG_VSN)
	docker manifest push $(TARGET):$(PKG_VSN); 
	docker manifest inspect $(TARGET):latest
	docker manifest push $(TARGET):latest;

.PHONY: docker-clean
docker-clean:
	@echo "DOCKER CLEAN: Clean Docker image."

	@if [ -n "$$(docker images -q  $(TARGET):$(PKG_VSN))" ]; then docker rmi -f $$(docker images -q  $(TARGET):$(PKG_VSN)); fi

	@for arch in $(ARCH_LIST); do \
		if [ -n "$$(docker images -q  $(TARGET):$(PKG_VSN)-$(OS)-$${arch})" ]; then \
			docker rmi -f $$(docker images -q  $(TARGET):$(PKG_VSN)-$(OS)-$${arch}); \
		fi \
	done