# FROM nvcr.io/nvidia/deepstream-l4t:6.0.1-base
FROM nvcr.io/nvidia/deepstream-l4t:6.0.1-iot

# 1. Install dependencies
RUN apt-get update && \
      DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
      rsyslog git \
           tzdata \
           libgstrtspserver-1.0-0 \
           libjansson4 \
           libglib2.0 \
           libjson-glib-1.0-0 \
           librabbitmq4 \
           gstreamer1.0-rtsp \
           libcurl4-openssl-dev ca-certificates
RUN apt install -y libssl1.0.0 libgstreamer1.0-0 gstreamer1.0-tools gstreamer1.0-plugins-good \
      gstreamer1.0-plugins-bad gstreamer1.0-plugins-ugly gstreamer1.0-libav libgstrtspserver-1.0-0 \
      libjansson4=2.11-1

RUN apt install -y python3-pip
RUN python3 -m pip install --upgrade pip
RUN apt-get install -y gstreamer-1.0 gstreamer1.0-dev git autoconf automake libtool build-essential libpython3.8 \
    python3-dev libpython3-dev
RUN apt install -y libgstreamer1.0-dev python-gi-dev libpython3.6-dev python3.6-dev cmake
RUN apt-get install -y libgstreamer-plugins-base1.0-dev curl python3-tk
RUN apt install -y python-gi-dev libpython3.8-dev python3.8-dev

# 2. Install DeepStream Python
WORKDIR /opt/nvidia/deepstream/deepstream-6.0
RUN git clone https://github.com/NVIDIA-AI-IOT/deepstream_python_apps.git
WORKDIR /opt/nvidia/deepstream/deepstream-6.0/deepstream_python_apps
RUN git submodule update --init
WORKDIR /opt/nvidia/deepstream/deepstream-6.0/deepstream_python_apps/3rdparty/gst-python
RUN ./autogen.sh PYTHON=python3
RUN make && make install

# 3. Build python bindings
WORKDIR /opt/nvidia/deepstream/deepstream-6.0/deepstream_python_apps/bindings
RUN mkdir build
WORKDIR /opt/nvidia/deepstream/deepstream-6.0/deepstream_python_apps/bindings/build
RUN curl -O -L https://github.com/NVIDIA-AI-IOT/deepstream_python_apps/releases/download/v1.1.1/pyds-1.1.1-py3-none-linux_aarch64.whl

# 4. Install pip wheel
RUN pip3 install ./pyds-1.1.1-py3-none*.whl

# 5. Install kafka
WORKDIR /opt/nvidia/deepstream/deepstream-6.0
RUN git clone https://github.com/edenhill/librdkafka.git
WORKDIR /opt/nvidia/deepstream/deepstream-6.0/librdkafka
RUN git reset --hard 7101c2310341ab3f4675fc565f64f0967e135a6a
RUN ./configure
RUN make && make install

# # 6. Copy kafka to correct directory
RUN mkdir -p /opt/nvidia/deepstream/deepstream-6.0/lib
RUN cp /usr/local/lib/librdkafka* /opt/nvidia/deepstream/deepstream-6.0/lib

# 8. Clean up and copy
WORKDIR /opt/nvidia/deepstream/deepstream-6.0/deepstream_python_apps
RUN rm -rf notebooks/
RUN rm -rf apps/
COPY . .


ENTRYPOINT [ "python3", "/opt/nvidia/deepstream/deepstream-6.0/deepstream_python_apps/apps/app/main.py","-c", "-p", "-i"]
# ENTRYPOINT [ "python3", "-u", "/opt/nvidia/deepstream/deepstream-6.0/deepstream_python_apps/apps/app/main.py","-c", "-p", "-i"]
# CMD ["/bin/bash"]
# WORKDIR /opt/nvidia/deepstream/deepstream-6.0

ENV LD_LIBRARY_PATH /usr/local/cuda-10.2/lib64
ENV NVIDIA_VISIBLE_DEVICES all
ENV NVIDIA_DRIVER_CAPABILITIES all