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 \
    crossbuild-essential-arm64 \
    patch bzip2 pkg-config \
    libxml2-dev libtclap-dev

# Provision the musl cross-compiler from musl.cc
RUN curl -O https://musl.cc/aarch64-linux-musl-cross.tgz && \
    tar xzf aarch64-linux-musl-cross.tgz -C /opt && \
    rm aarch64-linux-musl-cross.tgz

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

WORKDIR /src

ARG NXP_TAR
ARG MONO_ASK_TAR
ARG LIBNFNETLINK_TAR
ARG LIBNFCT_TAR
ARG FMLIB_TAR
ARG FMC_TAR

ARG NXP_VERSION
ARG MONO_ASK_VERSION
ARG LIBNFNETLINK_VERSION
ARG LIBNFCT_VERSION
ARG FMLIB_VERSION
ARG FMC_VERSION

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

# Copy the ASK and Netfilter tarballs 
COPY "${MONO_ASK_TAR}" .
COPY "${LIBNFNETLINK_TAR}" ./libnfnetlink-${LIBNFNETLINK_VERSION}.tar.bz2
COPY "${LIBNFCT_TAR}" ./libnetfilter_conntrack-${LIBNFCT_VERSION}.tar.xz
COPY "${FMC_TAR}" ./fmc.tar.gz
COPY "${FMLIB_TAR}" ./fmlib.tar.gz

# Extract ASK and place Netfilter dependencies where the Makefile expects them
RUN mkdir -p ASK/sources/tarballs && \
    tar zxf "${MONO_ASK_VERSION}.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

WORKDIR /src/ASK

# Extract fmc & fmlib to prevent git fetching
RUN mkdir -p sources/fmc && \
    mkdir -p sources/fmlib && \
    tar zxf "/src/fmc.tar.gz" -C "sources/fmc" --strip-components=1 && \
    tar zxf "/src/fmlib.tar.gz" -C "sources/fmlib" --strip-components=1

# 1. Apply the NXP ASK kernel patch to your Linux source tree FIRST
RUN cd /src/linux && \
    patch -p1 < /src/ASK/patches/kernel/002-mono-gateway-ask-kernel_linux_6_12.patch

# 2. Merge the NXP defconfig with your Kubernetes extra config
# Assuming you copy kernel-extra.config into the container at /src/kernel-extra.config
COPY kernel-extra.config /src/kernel-extra.config

RUN cd /src/linux && \
    cp /src/ASK/config/kernel/defconfig .config && \
   ./scripts/kconfig/merge_config.sh -m .config /src/kernel-extra.config && \
    make ARCH=arm64 CROSS_COMPILE=aarch64-linux-gnu- olddefconfig

# 3. Build the Kernel natively in the linux tree (Do NOT run 'make kernel' in the ASK folder)
RUN cd /src/linux && \
    make ARCH=arm64 CROSS_COMPILE=aarch64-linux-gnu- -j$(nproc) Image modules

# 4. Now build the ASK out-of-tree modules
RUN make modules KDIR=/src/linux CROSS_COMPILE=aarch64-linux-gnu- ARCH=arm64

# 5. Build Userspace with the musl cross-compiler
RUN make sources KDIR=/src/linux CROSS_COMPILE=aarch64-linux-musl- ARCH=arm64 HOST=aarch64-linux-musl
RUN make userspace KDIR=/src/linux CROSS_COMPILE=aarch64-linux-musl- ARCH=arm64 \
    HOST=aarch64-linux-musl \
    CC="aarch64-linux-musl-gcc -static" \
    CXX="aarch64-linux-musl-g++ -static"

# Stage the artifacts
RUN make dist

# Export stage
FROM scratch AS export
COPY --from=build /src/ASK/dist/ /ask/
COPY --from=build /src/linux/arch/arm64/boot/Image /kernel/Image