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

# 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

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