ARG CUDA_VER=12.2.2
FROM nvidia/cuda:${CUDA_VER}-runtime-ubuntu22.04

ARG CLIENT_VER=0.0.0
ARG BUNDLE_DATE=00000000

LABEL org.opencontainers.image.title="argus-sys-metric-test-node-bundle-gpu" \
      org.opencontainers.image.description="GPU node bundle with embedded Argus client artifact" \
      org.opencontainers.image.version="${CLIENT_VER}" \
      org.opencontainers.image.revision_date="${BUNDLE_DATE}" \
      maintainer="Argus"

ENV DEBIAN_FRONTEND=noninteractive \
    TZ=Asia/Shanghai \
    ARGUS_LOGS_WORLD_WRITABLE=1 \
    ES_HOST=es.log.argus.com \
    ES_PORT=9200 \
    CLUSTER=local \
    RACK=dev

RUN set -eux; \
    apt-get update; \
    apt-get install -y --no-install-recommends \
      ca-certificates curl wget iproute2 iputils-ping net-tools jq tzdata cron procps vim less \
      tar gzip; \
    rm -rf /var/lib/apt/lists/*; \
    ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone

WORKDIR /

# Expect staged build context to provide these directories/files
COPY bundle/ /bundle/
COPY node-bootstrap.sh /usr/local/bin/node-bootstrap.sh
COPY health-watcher.sh /usr/local/bin/health-watcher.sh
COPY private/start-fluent-bit.sh /private/start-fluent-bit.sh
COPY private/etc /private/etc
COPY private/packages /private/packages

RUN chmod +x /usr/local/bin/node-bootstrap.sh /usr/local/bin/health-watcher.sh /private/start-fluent-bit.sh || true; \
    mkdir -p /logs/train /logs/infer /buffers /opt/argus-metric; \
    chmod 1777 /logs/train /logs/infer || true; \
    chmod 770 /buffers || true

ENTRYPOINT ["/usr/local/bin/node-bootstrap.sh"]