CC = gcc

# Source of extensions compiled w/ Lua's source.
# Only include .c files that can't be directly included in ext/all.c.
SRC = \
	ext/all.c \
	ext/libinjection/libinjection_html5.c \
	ext/libinjection/libinjection_xss.c \
	ext/libinjection/libinjection_sqli.c \
	ext/lpeg/lpcap.c \
	ext/lpeg/lpcode.c \
	ext/lpeg/lpprint.c \
	ext/lpeg/lpvm.c

OBJ = ${SRC:.c=.o}

# Library archive. Used for compiling along agent bindings.
SO_OUT = libimmunio.so

# CLI for running tests
CLI = lua
CLI_SRC = ext/luajit/src/luajit.c ${SRC}

CFLAGS = -DLUA_USE_APICHECK -DLUAJIT -Dlua_assert=assert -O3 -fPIC
INCS = -Iext -Iext/luajit/src
LIBS = -lm -ldl

INIT_HOOK = hooks/__init__.lua
HOOK_SRCS := $(wildcard hooks/*.lua) hooks/__init__.lua
MIN_SRCS = $(HOOK_SRCS:hooks/%.lua=build/%.lua)
HOOKS_TARBALL = hooks.tgz
HOOKS_SRCS_TARBALL = hooks_srcs.tgz

LUAJIT_OBJ = ext/luajit/src/libluajit.a
LUAJIT_XCFLAGS = -fPIC

ifeq (${shell uname}, Darwin)
	# Disable the JIT on OS X
	LUAJIT_XCFLAGS += -DLUAJIT_ENABLE_GC64
endif

# Build lua, run tests, and create hooks archive
all: ${CLI} ${INIT_HOOK} ${HOOKS_TARBALL} ${HOOKS_SRCS_TARBALL}

.c.o:
	${CC} ${CFLAGS} -c ${INCS} -o $@ $<

${SO_OUT}: ${OBJ} ${LUAJIT_OBJ}
	${CC} -shared ${CFLAGS} ${LIBS} -o $@ -lc $^

${LUAJIT_OBJ}:
	cd ext/luajit && make XCFLAGS="${LUAJIT_XCFLAGS}"

# Build lua executable for testing and compilation
# Seperate compilation as we need the LUA_UNSAFE_MODE flag set...
${CLI}: ${CLI_SRC} ${LUAJIT_OBJ}
	${CC} ${CFLAGS} -DLUA_UNSAFE_MODE ${INCS} -o $@ $^ ${LIBS}

# Concatenate init hooks into one __init__.lua hook with two newlines in between
${INIT_HOOK}: hooks/init/*.lua hooks/init/__header__
	rm -f hooks/__init__.lua
	cat hooks/init/__header__ >> hooks/__init__.lua
	for file in hooks/init/*.lua; do cat "$$file" >> hooks/__init__.lua; printf "\n\n" >> hooks/__init__.lua; done
	echo "return utils" >> hooks/__init__.lua

build/%.lua: hooks/%.lua ${CLI}
	@mkdir -p build
	@echo Minifying $<
	@cd LuaMinify; ../lua CommandLineMinify.lua ../$< ../$@ >/dev/null

# Create tarball of hooks to publish
${HOOKS_TARBALL}: ${MIN_SRCS}
	tar -czf $@ -C build .

${HOOKS_SRCS_TARBALL}: ${HOOK_SRCS}
	tar -czf $@ -C hooks . --exclude="init"

clean:
	rm -f ${CLI} ${OBJ} ${SO_OUT}
	cd ext/luajit && make clean
	rm -f ${INIT_HOOK}
	rm -f test_failed
	rm -rf build
	find . -name \*.o -delete

test: ${CLI} ${INIT_HOOK} lint ${MIN_SRCS}
	@rm -f test_failed
	@for file in test/*_test.lua; do printf "\nRunning $$file\n"; TEST_BUILT_HOOKS=1 ./${CLI} $$file || touch test_failed; done
	@test ! -f test_failed

lint: ${INIT_HOOK}
	@# Scan all lua files for lines with trailing spaces
	@# The leading `!` negates the logic, so this target fails if trailing
	@# spaces are found in any Lua file.
	@! find . -path ./LuaMinify -prune -o -name "*.lua" -exec grep -E -n "[[:space:]]+$$" {} +

.PHONY: all test clean lint