MODULE:=function
BIN:=baetyl-$(MODULE)
SRC_FILES:=$(shell find . -type f -name '*.go')
PLATFORM_ALL:=darwin/amd64 linux/amd64 linux/arm64 linux/arm/v7

export DOCKER_CLI_EXPERIMENTAL=enabled

GIT_TAG:=$(shell git tag --contains HEAD|awk 'END {print}')
GIT_REV:=git-$(shell git rev-parse --short HEAD)
VERSION:=$(if $(GIT_TAG),$(GIT_TAG),$(GIT_REV))
ifeq ($(findstring race,$(BUILD_ARGS)),race)
VERSION:=$(VERSION)-race
endif

GO_OS:=$(shell go env GOOS)
GO_ARCH:=$(shell go env GOARCH)
GO_ARM:=$(shell go env GOARM)

ifndef PLATFORMS
    PLATFORMS:=$(if $(GO_ARM),$(GO_OS)/$(GO_ARCH)/$(GO_ARM),$(GO_OS)/$(GO_ARCH))
    ifeq ($(GO_OS),darwin)
        PLATFORMS+=linux/amd64
    endif
else ifeq ($(PLATFORMS),all)
    override PLATFORMS:=$(PLATFORM_ALL)
endif


GO       := go
GO_MOD   := $(GO) mod
GO_ENV   := env GO111MODULE=on CGO_ENABLED=0
GO_FLAGS := $(BUILD_ARGS) -ldflags '-X "github.com/baetyl/baetyl-go/v2/utils.REVISION=$(GIT_REV)" -X "github.com/baetyl/baetyl-go/v2/utils.VERSION=$(VERSION)"'
ifeq ($(findstring race,$(BUILD_ARGS)),race)
GO_ENV   := env GO111MODULE=on CGO_ENABLED=1
GO_FLAGS := $(BUILD_ARGS) -ldflags '-s -w -X "github.com/baetyl/baetyl-go/v2/utils.REVISION=$(GIT_REV)" -X "github.com/baetyl/baetyl-go/v2/utils.VERSION=$(VERSION)"  -linkmode external -w -extldflags "-static"'
override PLATFORMS:= $(filter-out linux/arm/v7,$(PLATFORMS))
endif
GO_BUILD := $(GO_ENV) $(GO) build $(GO_FLAGS)
GOTEST   := $(GO) test
GOPKGS   := $$($(GO) list ./... | grep -vE "vendor")

REGISTRY:=
XFLAGS:=--load
XPLATFORMS:=$(shell echo $(filter-out darwin/amd64,$(PLATFORMS)) | sed 's: :,:g')

OUTPUT     :=output
OUTPUT_DIRS:=$(PLATFORMS:%=$(OUTPUT)/%/$(BIN))
OUTPUT_BINS:=$(OUTPUT_DIRS:%=%/$(BIN))
PKG_PLATFORMS := $(shell echo $(PLATFORMS) | sed 's:/:-:g')
OUTPUT_PKGS:=$(PKG_PLATFORMS:%=$(OUTPUT)/$(BIN)_%_$(VERSION).zip)

.PHONY: all
all: build test

.PHONY: build
build: $(OUTPUT_BINS)

$(OUTPUT_BINS): $(SRC_FILES)
	@echo "BUILD $@"
	@mkdir -p $(dir $@)
	@cp program.yml $(dir $@)
	@$(shell echo $(@:$(OUTPUT)/%/$(BIN)/$(BIN)=%)  | sed 's:/v:/:g' | awk -F '/' '{print "GOOS="$$1" GOARCH="$$2" GOARM="$$3""}') $(GO_BUILD) -o $@ cmd/main.go

.PHONY: build-local
build-local: $(SRC_FILES)
	@echo "BUILD $(BIN)"
	$(GO_BUILD) -o $(BIN) cmd/main.go
	@chmod +x $(BIN)

.PHONY: image
image:
	@echo "BUILDX: $(REGISTRY)$(MODULE):$(VERSION)"
	@-docker buildx create --name baetyl
	@docker buildx use baetyl
	@docker run --privileged --rm tonistiigi/binfmt --install all
	docker buildx build $(XFLAGS) --platform $(XPLATFORMS) -t $(REGISTRY)$(MODULE):$(VERSION) --build-arg BUILD_ARGS=$(BUILD_ARGS) -f Dockerfile .

.PHONY: test
test: fmt
	$(GOTEST) -race -short -covermode=atomic -coverprofile=coverage.txt $(GOPKGS)
	@go tool cover -func=coverage.txt | grep total

.PHONY: fmt
fmt:
	$(GO_MOD) tidy
	@go fmt ./...

.PHONY: clean
clean:
	@rm -rf $(OUTPUT) $(BIN)

.PHONY: package
package: build $(OUTPUT_PKGS)

$(OUTPUT_PKGS):
	@echo "PACKAGE $@"
	@cd $(OUTPUT)/$(shell echo $(@:$(OUTPUT)/$(BIN)_%_$(VERSION).zip=%) | sed 's:-:/:g')/$(BIN) && zip -q -r $(notdir $@) $(BIN) program.yml

.PHONY: python-image
python-image:
	make -C python3 image

.PHONY: python-image-opencv
python-image-opencv:
	make -C python3 image-opencv

.PHONY: node-image
node-image:
	make -C node image

.PHONY: python-package
python-package:
	make -C python3 package

.PHONY: node-package
node-package:
	make -C node package