Makefile

.PHONY: compile

compile:
	rebar3 compile

# llama.cpp (OpenAI-compatible server lives in this repo)
LLAMACPP_REPO = https://github.com/ggml-org/llama.cpp
LLAMACPP_DIR = _build/llama.cpp
LLAMACPP_BUILD_DIR = $(LLAMACPP_DIR)/build

.PHONY: llama-server
llama-server: $(LLAMACPP_BUILD_DIR)/bin/llama-server
	@echo "llama-server built at: $(LLAMACPP_BUILD_DIR)/bin/llama-server"

.PHONY: llama-cpp
llama-cpp: $(LLAMACPP_DIR)

# Clone llama.cpp into the build directory (shallow clone)
$(LLAMACPP_DIR):
	git clone --depth 1 $(LLAMACPP_REPO) $(LLAMACPP_DIR)

$(LLAMACPP_BUILD_DIR)/bin/llama-server: $(LLAMACPP_DIR)
	cmake -S $(LLAMACPP_DIR) -B $(LLAMACPP_BUILD_DIR) -DLLAMA_BUILD_SERVER=ON -DGGML_CUDA=ON
	$(MAKE) llama-server -C $(LLAMACPP_BUILD_DIR) -j$$(nproc)

WAMR_VERSION = WAMR-2.2.0-wasi-nn
WAMR_DIR = _build/wamr
WASI_NN_DIR = _build/wasi_nn

GENESIS_WASM_BRANCH = feat/hb-unit
GENESIS_WASM_REPO = https://github.com/permaweb/ao.git
GENESIS_WASM_SERVER_DIR = _build/genesis_wasm/genesis-wasm-server

ifdef HB_DEBUG
	WAMR_FLAGS = -DWAMR_ENABLE_LOG=1 -DWAMR_BUILD_DUMP_CALL_STACK=1 -DCMAKE_BUILD_TYPE=Debug
else
	WAMR_FLAGS = -DCMAKE_BUILD_TYPE=Release
endif

UNAME_S := $(shell uname -s)
UNAME_M := $(shell uname -m)

ifeq ($(UNAME_S),Darwin)
    WAMR_BUILD_PLATFORM = darwin
    ifeq ($(UNAME_M),arm64)
        WAMR_BUILD_TARGET = AARCH64
    else
        WAMR_BUILD_TARGET = X86_64
    endif
else
    WAMR_BUILD_PLATFORM = linux
    WAMR_BUILD_TARGET = X86_64
endif

wamr: $(WAMR_DIR)/lib/libvmlib.a
wasi_nn: $(WASI_NN_DIR)/lib/libwasi_nn_backend.so
debug: debug-clean $(WAMR_DIR)
	HB_DEBUG=1 make $(WAMR_DIR)/lib/libvmlib.a
	CFLAGS="-DHB_DEBUG=1 -fPIC" rebar3 compile

debug-clean:
	rm -rf priv
	rm -rf $(WAMR_DIR)/lib

# Clone the WAMR repository at our target release
$(WAMR_DIR):
	git clone \
		https://github.com/apuslabs/wasm-micro-runtime.git \
		$(WAMR_DIR) \
		-b $(WAMR_VERSION) \
		--single-branch

$(WAMR_DIR)/lib/libvmlib.a: $(WAMR_DIR)
	
	sed -i '742a tbl_inst->is_table64 = 1;' ./_build/wamr/core/iwasm/aot/aot_runtime.c; \
	cmake \
		$(WAMR_FLAGS) \
		-S $(WAMR_DIR) \
		-B $(WAMR_DIR)/lib \
		-DWAMR_BUILD_TARGET=$(WAMR_BUILD_TARGET) \
		-DWAMR_BUILD_PLATFORM=$(WAMR_BUILD_PLATFORM) \
		-DWAMR_BUILD_MEMORY64=1 \
		-DWAMR_DISABLE_HW_BOUND_CHECK=1 \
		-DWAMR_BUILD_EXCE_HANDLING=1 \
		-DWAMR_BUILD_SHARED_MEMORY=0 \
		-DWAMR_BUILD_AOT=1 \
		-DWAMR_BUILD_LIBC_WASI=1 \
		-DWAMR_BUILD_FAST_INTERP=0 \
		-DWAMR_BUILD_INTERP=1 \
		-DWAMR_BUILD_JIT=0 \
		-DWAMR_BUILD_FAST_JIT=0 \
        -DWAMR_BUILD_DEBUG_AOT=1 \
        -DWAMR_BUILD_TAIL_CALL=1 \
        -DWAMR_BUILD_AOT_STACK_FRAME=1 \
        -DWAMR_BUILD_MEMORY_PROFILING=1 \
        -DWAMR_BUILD_DUMP_CALL_STACK=1 \
		-DWAMR_BUILD_SHARED=0 
	make -C $(WAMR_DIR)/lib -j8

$(WASI_NN_DIR):
	git clone \
		https://github.com/apuslabs/wasi_nn_backend.git \
		$(WASI_NN_DIR) \
		-b main \
		--single-branch

$(WASI_NN_DIR)/lib/libwasi_nn_backend.so: $(WASI_NN_DIR)
	make -C $(WASI_NN_DIR) build
	cp $(WASI_NN_DIR)/build/libwasi_nn_backend.so ./native/wasi_nn_llama
clean:
	rebar3 clean

# Add a new target to print the library path
print-lib-path:
	@echo $(CURDIR)/lib/libvmlib.a

$(GENESIS_WASM_SERVER_DIR):
	mkdir -p $(GENESIS_WASM_SERVER_DIR)
	@echo "Cloning genesis-wasm repository..." && \
        tmp_dir=$$(mktemp -d) && \
        git clone --depth=1 -b $(GENESIS_WASM_BRANCH) $(GENESIS_WASM_REPO) $$tmp_dir && \
        mkdir -p $(GENESIS_WASM_SERVER_DIR) && \
        cp -r $$tmp_dir/servers/cu/* $(GENESIS_WASM_SERVER_DIR) && \
        rm -rf $$tmp_dir && \
        echo "Extracted servers/genesis-wasm to $(GENESIS_WASM_SERVER_DIR)"

# Set up genesis-wasm@1.0 environment
setup-genesis-wasm: $(GENESIS_WASM_SERVER_DIR)
	@cp native/genesis-wasm/launch-monitored.sh $(GENESIS_WASM_SERVER_DIR) && \
	if ! command -v node > /dev/null; then \
		echo "Error: Node.js is not installed. Please install Node.js before continuing."; \
		echo "For Ubuntu/Debian, you can install it with:"; \
		echo "  curl -fsSL https://deb.nodesource.com/setup_22.x | sudo -E bash - && \\"; \
		echo "  apt-get install -y nodejs=22.16.0-1nodesource1 --allow-downgrades && \\"; \
		echo "  node -v && npm -v"; \
		exit 1; \
	fi
	@cd $(GENESIS_WASM_SERVER_DIR) && npm install > /dev/null 2>&1 && \
		echo "Installed genesis-wasm@1.0 server."

CC_DIR = native/dev_sev_gpu
# Set up dev_sev_gpu Python environment
setup-cc: $(CC_DIR)
	@echo "Setting up dev_sev_gpu Python environment..." && \
	if ! command -v python3 > /dev/null; then \
		echo "Error: Python 3 is not installed. Please install Python 3 before continuing."; \
		echo "For Ubuntu/Debian, you can install it with:"; \
		echo "  apt-get update && apt-get install -y python3 python3-pip"; \
		exit 1; \
	fi && \
	echo "Installing Python dependencies..." && \
	 pip3 install nv-attestation-sdk  && \
	echo "Installed dev_sev_gpu Python environment successfully."