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

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

# fmc & fmlib
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

# 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

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
COPY kernel-build/dts/*.dts /src/linux/arch/arm64/boot/dts/freescale/

# 1. This step patches the kernel, merges kernel-extra.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

# 4. Stage the 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 /src/ASK/dist/*.ko /out/rootfs/lib/modules/$KERNEL_VER/extra/ && \
    depmod -b /out/rootfs $KERNEL_VER && \
    cd /out && tar zcf rootfs.tar.gz rootfs

# Export stage
FROM scratch AS export
COPY --from=build /src/ASK/dist/ /ask/

# Export the newly staged in-tree modules
COPY --from=build /out/rootfs.tar.gz /
COPY --from=build /src/linux/System.map /kernel/System.map
COPY --from=build /src/linux/.config /kernel/.config
COPY --from=build /src/linux/arch/arm64/boot/Image /kernel/Image
COPY --from=build /src/linux/arch/arm64/boot/dts/freescale/mono-gateway-dk-sdk.dtb /kernel/mono-gateway-dk-sdk.dtb

# Grab the proprietary Mono Gateway XML configs
COPY --from=build /src/ASK/config/gateway-dk/cdx_cfg.xml /xml/cdx_cfg.xml
COPY --from=build /src/ASK/dpa_app/files/etc/cdx_pcd.xml /xml/cdx_pcd.xml