#!/bin/bash

# Copyright (C) 2018-2022 Intel Corporation
# SPDX-License-Identifier: Apache-2.0

params=$1

yes_or_no() {
    if [ "$params" == "-y" ]; then
        return 0
    fi

    while true; do
        read -p "Add third-party Nux Dextop repository and install FFmpeg package (y) / Skip this step (N)" yn
        case $yn in
            [Yy]*) return 0 ;;
            [Nn]*) return 1 ;;
        esac
    done
}

# install dependencies
if [ -f /etc/lsb-release ]; then
    # Ubuntu
    host_cpu=$(uname -m)
    if [ "$host_cpu" = "x86_64" ]; then
        x86_64_specific_packages=(gcc-multilib g++-multilib)
    else
        x86_64_specific_packages=()
    fi

    if ! command -v cmake &> /dev/null; then
        cmake_packages=(cmake)
    fi

    sudo -E apt update
    sudo -E apt-get install -y \
            build-essential \
            "${cmake_packages[@]}" \
            ccache \
            curl \
            wget \
            libssl-dev \
            ca-certificates \
            git \
            git-lfs \
            "${x86_64_specific_packages[@]}" \
            libgtk2.0-dev \
            unzip \
            shellcheck \
            patchelf \
            fdupes \
            lintian \
            file \
            gzip \
            `# openvino` \
            libtbb-dev \
            libpugixml-dev \
            `# gpu plugin extensions` \
            libva-dev \
            `# python` \
            python3-pip \
            python3-venv \
            python3-enchant \
            python3-setuptools \
            libpython3-dev \
            `# samples` \
            pkg-config \
            libgflags-dev \
            zlib1g-dev \
            `# hddl` \
            libudev1 \
            libusb-1.0-0 \
            `# myriad` \
            libusb-1.0-0-dev \
            `# cl_compiler` \
            libtinfo5
    # hddl
    if apt-cache search --names-only '^libjson-c3'| grep -q libjson-c3; then
        # ubuntu 18.04
        sudo -E apt-get install -y \
            libjson-c3 \
            libboost-filesystem1.65.1 \
            libboost-program-options1.65.1 \
            libboost-system1.65.1
    elif apt-cache search --names-only '^libjson-c4'| grep -q libjson-c4; then
        # ubuntu 20.04
        sudo -E apt-get install -y \
            libjson-c4 \
            libboost-filesystem1.71.0 \
            libboost-program-options1.71.0
    fi
    # for python3-enchant
    if apt-cache search --names-only 'libenchant1c2a'| grep -q libenchant1c2a; then
        sudo -E apt-get install -y libenchant1c2a
    fi
    # samples
    if apt-cache search --names-only '^nlohmann-json3-dev'| grep -q nlohmann-json3; then
        sudo -E apt-get install -y nlohmann-json3-dev
    else
        sudo -E apt-get install -y nlohmann-json-dev
    fi
elif [ -f /etc/redhat-release ]; then
    # RHEL 8 / CentOS 7
    sudo -E yum install -y centos-release-scl epel-release
    sudo -E yum install -y \
            wget \
            curl \
            cmake3 \
            tar \
            xz \
            p7zip \
            ccache \
            rpm-build \
            rpmlint \
            ShellCheck \
            unzip \
            which \
            ca-certificates \
            git \
            git-lfs \
            boost-devel \
            python-pip \
            python-devel \
            libtool \
            tbb-devel \
            pugixml-devel \
            gcc \
            gcc-c++ \
            make \
            patchelf \
            fdupes \
            libusbx-devel \
            file \
            zlib-devel \
            gflags-devel \
            python3-pip \
            libva-devel
elif [ -f /etc/os-release ] && grep -q "raspbian" /etc/os-release; then
    # Raspbian
    sudo -E apt update
    sudo -E apt-get install -y \
            build-essential \
            wget \
            libssl-dev \
            ca-certificates \
            git \
            pkg-config \
            libgflags-dev \
            zlib1g-dev \
            nlohmann-json-dev \
            unzip \
            libusb-1.0-0-dev
else
    echo "Unknown OS, please install build dependencies manually"
fi

# cmake 3.20 or higher is required to build OpenVINO
current_cmake_ver=$(cmake --version | sed -ne 's/[^0-9]*\(\([0-9]\.\)\{0,4\}[0-9][^.]\).*/\1/p')
required_cmake_ver=3.20.0
if [ ! "$(printf '%s\n' "$required_cmake_ver" "$current_cmake_ver" | sort -V | head -n1)" = "$required_cmake_ver" ]; then
    installed_cmake_ver=3.23.2
    wget "https://github.com/Kitware/CMake/releases/download/v${installed_cmake_ver}/cmake-${installed_cmake_ver}.tar.gz"
    tar xf cmake-${installed_cmake_ver}.tar.gz
    (cd cmake-${installed_cmake_ver} && ./bootstrap --parallel="$(nproc --all)" && make --jobs="$(nproc --all)" && sudo make install)
    rm -rf cmake-${installed_cmake_ver} cmake-${installed_cmake_ver}.tar.gz
fi