Makefile

# Build variables
VERSION ?= latest
BUILD_DATE = $(shell date -u +%Y-%m-%dT%H:%M:%SZ)
BINARY_NAME = topology-scheduler
GOARCH = amd64
MODULE_NAME = github.com/nod-ai/topology-aware-gpu-scheduler

# Determine OS-specific variables
ifeq ($(OS),Windows_NT)
    BINARY_SUFFIX = .exe
    RM = del /F /Q
    MKDIR = mkdir
else
    BINARY_SUFFIX =
    RM = rm -f
    MKDIR = mkdir -p
endif

# Directories
BIN_DIR = bin
PKG_DIR = pkg
CMD_DIR = cmd

# Go build flags
LDFLAGS = -ldflags "-X main.version=$(VERSION) -X main.buildDate=$(BUILD_DATE)"
GOFLAGS = CGO_ENABLED=0

.PHONY: all
all: clean deps build

.PHONY: clean
clean:
	$(RM) $(BIN_DIR)$(BINARY_NAME)$(BINARY_SUFFIX)
	go clean -modcache
	go clean -cache
	rm -rf vendor/

.PHONY: generate-client
generate-client:
	chmod +x codegen.sh
	./codegen.sh
	go mod tidy

.PHONY: generate
generate: generate-client
	go generate ./...

.PHONY: setup-codegen
setup-codegen:
	chmod +x setup-codegen.sh
	./setup-codegen.sh

.PHONY: fix-imports
fix-imports:
	@echo "Fixing import paths..."
	@find . -type f -name "*.go" -exec sed -i 's|github.com/iamakanshab/topology-aware-scheduler|$(MODULE_NAME)|g' {} \;

.PHONY: deps
deps: fix-imports
	@echo "Updating go.mod..."
	@if [ -f go.mod ]; then \
		sed -i 's|module.*|module $(MODULE_NAME)|' go.mod; \
	fi
	go mod tidy
	go mod download
	go mod vendor

.PHONY: build
build:
	$(MKDIR) $(BIN_DIR)
	$(GOFLAGS) go build $(LDFLAGS) -o $(BIN_DIR)/$(BINARY_NAME)$(BINARY_SUFFIX) $(CMD_DIR)/scheduler/main.go

.PHONY: test
test:
	go test -v ./...

.PHONY: test-coverage
test-coverage:
	go test -v -coverprofile=coverage.out ./...
	go tool cover -html=coverage.out

.PHONY: lint
lint:
	golangci-lint run

.PHONY: run
run: build
	./$(BIN_DIR)/$(BINARY_NAME)$(BINARY_SUFFIX)

.PHONY: docker
docker:
	docker build -t $(BINARY_NAME):$(VERSION) .

.PHONY: help
help:
	@echo "Available targets:"
	@echo "  all            - Clean, get dependencies, and build"
	@echo "  clean          - Remove built binary and clean go cache"
	@echo "  fix-imports    - Fix import paths in all Go files"
	@echo "  deps           - Download and tidy dependencies"
	@echo "  build          - Build the binary"
	@echo "  test           - Run tests"
	@echo "  test-coverage  - Run tests with coverage report"
	@echo "  lint           - Run linter"
	@echo "  run            - Build and run the binary"
	@echo "  generate       - Generate code and run go generate"
	@echo "  generate-client- Generate only client code"
	@echo "  docker         - Build Docker image"