ARG BUILD_BASE_TAG=dev
ARG DOCKER_IMAGE_ROOT=monok8s

FROM --platform=$BUILDPLATFORM ${DOCKER_IMAGE_ROOT}/build-base:${BUILD_BASE_TAG} AS build

# Install glibc cross-compiler for kernel and standard build dependencies
RUN apt-get update && apt-get install -y pkg-config

RUN git config --global user.email "monok8s@localhost" && \
	git config --global user.name "monok8s authors" && \
	git config --global --add safe.directory '*'

WORKDIR /src

ARG AARCH64_MUSL_CC_TAR
ARG NXP_TAR
ARG MONO_ASK_TAR
ARG LIBNFNETLINK_TAR
ARG LIBMNL_TAR
ARG LIBNFCT_TAR
ARG FMLIB_TAR
ARG FMC_TAR
ARG LIBXML2_TAR
ARG LIBPCAP_TAR
ARG LIBCLI_TAR
ARG TCLAP_TAR

# ASK's version pins (hardcoded wget)
ARG LIBNFNETLINK_VERSION
ARG LIBNFCT_VERSION

# MUSL Cross Compiler
COPY "${AARCH64_MUSL_CC_TAR}"  ./aarch64_musl_cc.tar.gz

# Linux kernel
COPY "${NXP_TAR}"		  ./kernel.tar.gz

# Copy the ASK deps
COPY "${MONO_ASK_TAR}"	 ./mono-ask.tar.gz
COPY "${FMC_TAR}"		  ./fmc.tar.gz
COPY "${FMLIB_TAR}"		./fmlib.tar.gz
COPY "${LIBXML2_TAR}"	  ./libxml2.tar.xz
COPY "${LIBPCAP_TAR}"	  ./libpcap.tar.xz
COPY "${TCLAP_TAR}"		./tclap.tar.gz
COPY "${LIBMNL_TAR}"	   ./libmnl.tar.bz2
COPY "${LIBCLI_TAR}"	   ./libcli.tar.gz
# Pinned version should keep version names
COPY "${LIBNFNETLINK_TAR}" ./libnfnetlink-${LIBNFNETLINK_VERSION}.tar.bz2
COPY "${LIBNFCT_TAR}"	  ./libnetfilter_conntrack-${LIBNFCT_VERSION}.tar.xz

# Provision the musl cross-compiler from musl.cc
RUN tar zxf "aarch64_musl_cc.tar.gz" -C /opt

# Expose the musl compiler to the PATH
ENV PATH="/opt/aarch64-linux-musl-cross/bin:${PATH}"

# Extract and build the dependency libraries
RUN mkdir -p ASK/sources/tarballs && \
	tar zxf "mono-ask.tar.gz" -C "ASK" --strip-components=1 && \
	mv libnfnetlink-${LIBNFNETLINK_VERSION}.tar.bz2 ASK/sources/tarballs/ && \
	mv libnetfilter_conntrack-${LIBNFCT_VERSION}.tar.xz ASK/sources/tarballs/

RUN mkdir linux && tar zxf "kernel.tar.gz" -C "linux" --strip-components=1 && \
	mkdir -p /src/ASK/patches/kernel/updated-patch

# ASK's Kernel patches
COPY patches/ask/split-kernel-patch.sh /src/split-kernel-patch.sh
COPY patches/ask/kernel/*.patch /src/ASK/patches/kernel/updated-patch/

RUN chmod +x /src/split-kernel-patch.sh && \
	PATCH_FILE="/src/ASK/patches/kernel/002-mono-gateway-ask-kernel_linux_6_12.patch" \
	LINUX_DIR="/src/linux" \
	OUT_DIR="/src/ASK/patches/kernel/split-002" \
	/src/split-kernel-patch.sh split && \
	/src/split-kernel-patch.sh apply

# tclap
RUN mkdir -p tclap && tar zxf "tclap.tar.gz" -C "tclap" --strip-components=1 && \
	cp -r tclap/include/tclap /opt/aarch64-linux-musl-cross/aarch64-linux-musl/include/ && \
	rm -rf tclap

# libxml2
RUN mkdir -p libxml2 && tar xf "libxml2.tar.xz" -C "libxml2" --strip-components=1 && \
	cd libxml2 && \
	CC=aarch64-linux-musl-gcc ./configure --host=aarch64-linux-musl \
		--prefix=/opt/aarch64-linux-musl-cross/aarch64-linux-musl \
		--enable-static --disable-shared --without-python --without-zlib --without-lzma && \
	make -j$(nproc) && make install && \
	cd .. && rm -rf libxml2

# libmnl
RUN mkdir -p libmnl && tar xjf "libmnl.tar.bz2" -C "libmnl" --strip-components=1 && \
	cd libmnl && \
	CC=aarch64-linux-musl-gcc ./configure --host=aarch64-linux-musl \
		--prefix=/opt/aarch64-linux-musl-cross/aarch64-linux-musl \
		--enable-static --disable-shared && \
	make -j$(nproc) && make install && \
	cd .. && rm -rf libmnl

# libcli
RUN mkdir -p libcli && tar zxf "libcli.tar.gz" -C "libcli" --strip-components=1 && \
	cd libcli && \
	make CC=aarch64-linux-musl-gcc AR=aarch64-linux-musl-ar libcli.a && \
	cp libcli.h /opt/aarch64-linux-musl-cross/aarch64-linux-musl/include/ && \
	cp libcli.a /opt/aarch64-linux-musl-cross/aarch64-linux-musl/lib/ && \
	cd .. && rm -rf libcli

# libpcap
RUN mkdir -p libpcap && tar xf "libpcap.tar.xz" -C "libpcap" --strip-components=1 && \
	cd libpcap && \
	CC=aarch64-linux-musl-gcc ./configure --host=aarch64-linux-musl \
		--prefix=/opt/aarch64-linux-musl-cross/aarch64-linux-musl \
		--with-pcap=linux --enable-static --disable-shared \
		--disable-usb --disable-netmap --disable-bluetooth --disable-dbus && \
	make -j$(nproc) && make install && \
	cd .. && rm -rf libpcap

# fmlib, fmc
RUN mkdir fmlib && \
	tar zxf "fmlib.tar.gz" -C "fmlib" --strip-components=1 && \
	cd fmlib && git init -q && git add -A && git commit -q -m "base" && \
	git apply /src/ASK/patches/fmlib/01-mono-ask-extensions.patch && \
	make CROSS_COMPILE=aarch64-linux-musl- KERNEL_SRC=../linux libfm-arm.a && \
	ln -sf libfm-arm.a libfm.a && \
	cd ../ && \
	mkdir -p fmc && \
	tar zxf "fmc.tar.gz" -C "fmc" --strip-components=1 && \
	# Handle fmc: Initialize dummy repo, patch, and build \
	cd fmc && git init -q && git add -A && git commit -q -m "base" && \
	git apply /src/ASK/patches/fmc/01-mono-ask-extensions.patch && \
	make -C "source" CC="aarch64-linux-musl-gcc -static" CXX="aarch64-linux-musl-g++ -static" AR=aarch64-linux-musl-ar \
		MACHINE=ls1046 \
		FMD_USPACE_HEADER_PATH=../../fmlib/include/fmd \
		FMD_USPACE_LIB_PATH=../../fmlib \
		LIBXML2_HEADER_PATH=/opt/aarch64-linux-musl-cross/aarch64-linux-musl/include/libxml2 \
		TCLAP_HEADER_PATH=/opt/aarch64-linux-musl-cross/aarch64-linux-musl/include

# Common paths / flags for the remaining ASK build.
# Keep userspace fully static against the musl sysroot.
ENV ASK_DIR=/src/ASK \
	KERNEL_DIR=/src/linux \
	SYSROOT=/opt/aarch64-linux-musl-cross/aarch64-linux-musl \
	HOST=aarch64-linux-musl \
	CROSS_COMPILE=aarch64-linux-musl- \
	ARCH=arm64 \
	PLATFORM=LS1043A

# Build patched libnfnetlink + libnetfilter_conntrack into the musl sysroot.
# These are needed by cmm through pkg-config.
RUN mkdir -p "${ASK_DIR}/sources" && \
	mkdir -p libnfnetlink && \
	tar xjf "${ASK_DIR}/sources/tarballs/libnfnetlink-${LIBNFNETLINK_VERSION}.tar.bz2" \
		-C libnfnetlink --strip-components=1 && \
	cd libnfnetlink && \
	git init -q && git add -A && git commit -q -m "upstream" && \
	git apply "${ASK_DIR}/patches/libnfnetlink/01-nxp-ask-nonblocking-heap-buffer.patch" && \
	CC=aarch64-linux-musl-gcc AR=aarch64-linux-musl-ar RANLIB=aarch64-linux-musl-ranlib \
	./configure --host="${HOST}" --prefix="${SYSROOT}" \
		--enable-static --disable-shared && \
	make -j$(nproc) && make install && \
	cd /src && rm -rf libnfnetlink && \
	mkdir -p libnetfilter_conntrack && \
	tar xf "${ASK_DIR}/sources/tarballs/libnetfilter_conntrack-${LIBNFCT_VERSION}.tar.xz" \
		-C libnetfilter_conntrack --strip-components=1 && \
	cd libnetfilter_conntrack && \
	git init -q && git add -A && git commit -q -m "upstream" && \
	git apply "${ASK_DIR}/patches/libnetfilter-conntrack/01-nxp-ask-comcerto-fp-extensions.patch" && \
	PKG_CONFIG_PATH="${SYSROOT}/lib/pkgconfig" \
	CC=aarch64-linux-musl-gcc AR=aarch64-linux-musl-ar RANLIB=aarch64-linux-musl-ranlib \
	./configure --host="${HOST}" --prefix="${SYSROOT}" \
		--enable-static --disable-shared \
		CFLAGS="-I${SYSROOT}/include" \
		LDFLAGS="-L${SYSROOT}/lib" && \
	make -j$(nproc) && make install && \
	cd /src && rm -rf libnetfilter_conntrack

# Build libfci, used by cmm.
RUN make -C "${ASK_DIR}/fci/lib" \
	CC=aarch64-linux-musl-gcc \
	AR=aarch64-linux-musl-ar

# Build the kernel first, without the custom DTS
COPY kernel-extra.config /src/kernel-extra.config
COPY kernel-build/ensure-kconfig.sh /src/

RUN cd /src/linux \
 && make ARCH="${ARCH}" CROSS_COMPILE="${CROSS_COMPILE}" defconfig lsdk.config \
 && ./scripts/kconfig/merge_config.sh -m .config "${ASK_DIR}/config/kernel/defconfig" /src/kernel-extra.config \
 && make ARCH="${ARCH}" CROSS_COMPILE="${CROSS_COMPILE}" olddefconfig \
 && /src/ensure-kconfig.sh .config /src/kernel-extra.config \
 && make ARCH="${ARCH}" CROSS_COMPILE="${CROSS_COMPILE}" -j"$(nproc)"

# DTS changes happen after the expensive kernel build layer
COPY kernel-build/dts/*.dts /src/linux/arch/arm64/boot/dts/freescale/

ARG DEVICE_TREE_TARGET

RUN cd /src/linux \
 && grep -q "^dtb-\\\$(CONFIG_ARCH_LAYERSCAPE) += ${DEVICE_TREE_TARGET}.dtb$" \
	  arch/arm64/boot/dts/freescale/Makefile \
	|| echo "dtb-\$(CONFIG_ARCH_LAYERSCAPE) += ${DEVICE_TREE_TARGET}.dtb" \
	  >> arch/arm64/boot/dts/freescale/Makefile \
 && make ARCH="${ARCH}" CROSS_COMPILE="${CROSS_COMPILE}" \
	  "freescale/${DEVICE_TREE_TARGET}.dtb"

# Patch verdor's modules
RUN mkdir -p /src/patches/ask
COPY patches/ask/ /src/patches/ask/

RUN cd /src/ASK && \
	git init -q && git add -A && git commit -q -m "base" && \
	find /src/patches/ask \
		-path /src/patches/ask/kernel -prune -o \
		-name '*.patch' -print \
		| sort > /tmp/ask-module-patches.list && \
	test ! -s /tmp/ask-module-patches.list || xargs -a /tmp/ask-module-patches.list git apply --check && \
	test ! -s /tmp/ask-module-patches.list || xargs -a /tmp/ask-module-patches.list git apply

# Build out-of-tree ASK kernel modules: cdx -> fci, plus auto_bridge.
RUN make -C "${ASK_DIR}/cdx" \
		CROSS_COMPILE="${CROSS_COMPILE}" \
		ARCH="${ARCH}" \
		KERNELDIR="${KERNEL_DIR}" \
		PLATFORM="${PLATFORM}" \
		modules && \
	make -C "${ASK_DIR}/fci" \
		CROSS_COMPILE="${CROSS_COMPILE}" \
		ARCH="${ARCH}" \
		KERNEL_SOURCE="${KERNEL_DIR}" \
		BOARD_ARCH="${ARCH}" \
		KBUILD_EXTRA_SYMBOLS="${ASK_DIR}/cdx/Module.symvers" \
		modules && \
	make -C "${ASK_DIR}/auto_bridge" \
		CROSS_COMPILE="${CROSS_COMPILE}" \
		ARCH="${ARCH}" \
		KERNEL_SOURCE="${KERNEL_DIR}" \
		PLATFORM="${PLATFORM}"

# Patch vendor's binaries
RUN sed -i '/^[[:space:]]*CFLAGS[[:space:]]*+=.*-Wall.*-Werror/a CFLAGS += -Wno-address-of-packed-member' "${ASK_DIR}/cmm/Makefile"

# Build remaining userspace binaries: cmm and dpa_app.
RUN make -C "${ASK_DIR}/cmm" \
		CC="aarch64-linux-musl-gcc -static" \
		LIBFCI_DIR="${ASK_DIR}/fci/lib" \
		ABM_DIR="${ASK_DIR}/auto_bridge" \
		SYSROOT="${SYSROOT}" \
		PKG_CONFIG_PATH="${SYSROOT}/lib/pkgconfig" && \
	make -C "${ASK_DIR}/dpa_app" \
		CC="aarch64-linux-musl-gcc -static" \
		CFLAGS="-DDPAA_DEBUG_ENABLE -DLS1043 -DNCSW_LINUX -D__STDC_LIMIT_MACROS \
			-I/src/fmc/source \
			-I${ASK_DIR}/cdx \
			-I/src/fmlib/include/fmd \
			-I/src/fmlib/include/fmd/Peripherals \
			-I/src/fmlib/include/fmd/integrations" \
		LDFLAGS="-static -L/src/fmc/source -L/src/fmlib -L${SYSROOT}/lib \
			-lfmc -lfm -lcli -lxml2 -lstdc++ -lpthread -lm"

# Stage outputs
RUN mkdir -p /out/ASK/dist && \
	cp "${ASK_DIR}/cdx/cdx.ko" /out/ASK/dist && \
	cp "${ASK_DIR}/fci/fci.ko" /out/ASK/dist && \
	cp "${ASK_DIR}/auto_bridge/auto_bridge.ko" /out/ASK/dist && \
	mkdir -p /out/ASK/bin && \
	cp /src/fmc/source/fmc /out/ASK/bin/ && \
	cp "${ASK_DIR}/cmm/src/cmm" /out/ASK/bin/ && \
	cp "${ASK_DIR}/dpa_app/dpa_app" /out/ASK/bin/
#	aarch64-linux-musl-strip /out/ASK/bin/fmc /out/ASK/bin/cmm /out/ASK/bin/dpa_app

# in-tree Linux kernel modules
RUN mkdir -p /out/rootfs && \
	make -C /src/linux ARCH=arm64 CROSS_COMPILE=aarch64-linux-gnu- \
	modules_install INSTALL_MOD_PATH=/out/rootfs

RUN KERNEL_VER=$(ls /out/rootfs/lib/modules/) && \
	mkdir -p /out/rootfs/lib/modules/$KERNEL_VER/extra && \
	mv /out/ASK/dist/*.ko /out/rootfs/lib/modules/$KERNEL_VER/extra/ && \
	depmod -b /out/rootfs $KERNEL_VER && \
	cd /out && tar zcf rootfs.tar.gz rootfs

RUN mkdir -p /out/rootfs-cfg/etc/dpa && \
	cp -r "${ASK_DIR}/dpa_app/files/etc/"* /out/rootfs-cfg/etc/dpa/ && \
	cp "${ASK_DIR}/config/gateway-dk/cdx_cfg.xml" /out/rootfs-cfg/etc/dpa/ && \
	cp -r /src/fmc/etc/* /out/rootfs-cfg/etc

FROM scratch AS export

ARG DEVICE_TREE_TARGET

# Export the newly staged in-tree modules
COPY --from=build /out/rootfs.tar.gz /
COPY --from=build \
	/src/linux/System.map \
	/src/linux/.config \
	/src/linux/arch/arm64/boot/Image.gz \
	/src/linux/arch/arm64/boot/dts/freescale/${DEVICE_TREE_TARGET}.dtb \
	/kernel/

COPY --from=build /out/ASK/bin/ /bin/

# Export the configs for Gateway Development Kit
COPY --from=build /out/rootfs-cfg/ /rootfs-cfg/