# --------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
# --------------------------------------------------------------
# Dockerfile for ONNX Runtime Android package build environment

FROM ubuntu:18.04

ENV DEBIAN_FRONTEND=noninteractive

# install utilities
RUN apt-get update && apt-get install --yes --no-install-recommends \
  aria2 \
  unzip

# install Java
RUN apt-get install --yes --no-install-recommends openjdk-8-jdk-headless

ENV ANDROID_HOME=/opt/android-sdk
ENV NDK_VERSION=24.0.8215888
ENV ANDROID_NDK_HOME=${ANDROID_HOME}/ndk/${NDK_VERSION}

# install Android command line tools
RUN aria2c -q -d /tmp -o cmdline-tools.zip \
  --checksum=sha-256=d71f75333d79c9c6ef5c39d3456c6c58c613de30e6a751ea0dbd433e8f8b9cbf \
  https://dl.google.com/android/repository/commandlinetools-linux-8092744_latest.zip && \
  unzip /tmp/cmdline-tools.zip -d /tmp/cmdline-tools && \
  mkdir -p ${ANDROID_HOME}/cmdline-tools && \
  mv /tmp/cmdline-tools/cmdline-tools ${ANDROID_HOME}/cmdline-tools/latest

RUN yes | ${ANDROID_HOME}/cmdline-tools/latest/bin/sdkmanager --licenses
RUN ${ANDROID_HOME}/cmdline-tools/latest/bin/sdkmanager --install \
  "platforms;android-32" \
  "ndk;${NDK_VERSION}"

# install ORT dependencies
RUN apt-get install --yes --no-install-recommends \
  ca-certificates \
  build-essential \
  git \
  ninja-build \
  python3-dev \
  python3-numpy \
  python3-pip \
  python3-setuptools \
  python3-wheel

# cmake
RUN CMAKE_VERSION=3.21.0 && \
  aria2c -q -d /tmp -o cmake-${CMAKE_VERSION}-linux-x86_64.tar.gz \
  --checksum=sha-256=d54ef6909f519740bc85cec07ff54574cd1e061f9f17357d9ace69f61c6291ce \
  https://github.com/Kitware/CMake/releases/download/v${CMAKE_VERSION}/cmake-${CMAKE_VERSION}-linux-x86_64.tar.gz && \
  tar -zxf /tmp/cmake-${CMAKE_VERSION}-linux-x86_64.tar.gz --strip=1 -C /usr

# gradle
RUN GRADLE_VERSION=6.8.3 && \
  aria2c -q -d /tmp -o gradle-${GRADLE_VERSION}-bin.zip \
  --checksum=sha-256=7faa7198769f872826c8ef4f1450f839ec27f0b4d5d1e51bade63667cbccd205 \
  https://services.gradle.org/distributions/gradle-${GRADLE_VERSION}-bin.zip && \
  mkdir /opt/gradle && \
  unzip -d /opt/gradle /tmp/gradle-${GRADLE_VERSION}-bin.zip && \
  ln -s /opt/gradle/gradle-${GRADLE_VERSION}/bin/gradle /usr/bin

# flatbuffers
RUN python3 -m pip install flatbuffers==2.0

WORKDIR /workspace

# get ORT repo
ARG ONNXRUNTIME_REPO=https://github.com/microsoft/onnxruntime.git
ARG ONNXRUNTIME_BRANCH_OR_TAG=main
RUN git clone --single-branch --branch=${ONNXRUNTIME_BRANCH_OR_TAG} --recurse-submodules ${ONNXRUNTIME_REPO} \
  /workspace/onnxruntime