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 LIBXML2_TAR
ARG TCLAP_TAR

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

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

# Copy the ASK and Netfilter tarballs 
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 "${TCLAP_TAR}"        ./tclap.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

# Extract ASK and place Netfilter dependencies where the Makefile expects them
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

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

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

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

WORKDIR /src/ASK

COPY patches/mono-ask.mk .
COPY kernel-extra.config /src/kernel-extra.config
COPY kernel-build/ensure-kconfig.sh /src/ensure-kconfig.sh

# 1. This step patches the kernel, merges your k8s config, and builds the modules
RUN make -f mono-ask.mk modules

# 2. This step cross-compiles fmlib, fmc, cmm, and dpa_app for Alpine
RUN make -f mono-ask.mk userspace

# 3. Stage the artifacts
RUN make -f mono-ask.mk 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