#!/bin/bash
#
#  Copyright (c) 2017, The OpenThread Authors.
#  All rights reserved.
#
#  Redistribution and use in source and binary forms, with or without
#  modification, are permitted provided that the following conditions are met:
#  1. Redistributions of source code must retain the above copyright
#     notice, this list of conditions and the following disclaimer.
#  2. Redistributions in binary form must reproduce the above copyright
#     notice, this list of conditions and the following disclaimer in the
#     documentation and/or other materials provided with the distribution.
#  3. Neither the name of the copyright holder nor the
#     names of its contributors may be used to endorse or promote products
#     derived from this software without specific prior written permission.
#
#  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
#  AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
#  IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
#  ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
#  LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
#  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
#  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
#  INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
#  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
#  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
#  POSSIBILITY OF SUCH DAMAGE.
#

set -euxo pipefail

TOOLS_HOME="$HOME"/.cache/tools
[[ -d $TOOLS_HOME ]] || mkdir -p "$TOOLS_HOME"

disable_install_recommends()
{
    OTBR_APT_CONF_FILE=/etc/apt/apt.conf

    if [[ -f ${OTBR_APT_CONF_FILE} ]] && grep Install-Recommends "${OTBR_APT_CONF_FILE}"; then
        return 0
    fi

    sudo tee -a /etc/apt/apt.conf <<EOF
APT::Get::Install-Recommends "false";
APT::Get::Install-Suggests "false";
EOF
}

install_common_dependencies()
{
    # Common dependencies
    sudo apt-get install --no-install-recommends -y \
        libdbus-1-dev \
        ninja-build \
        doxygen \
        expect \
        net-tools \
        libavahi-common-dev \
        libavahi-client-dev \
        libgmock-dev \
        libgtest-dev \
        libreadline-dev \
        libncurses-dev \
        libjsoncpp-dev \
        coreutils \
        git \
        libprotobuf-dev \
        protobuf-compiler
}

install_openthread_binraries()
{
    pip3 install -U pip
    pip3 install -U cmake
    cd third_party/openthread/repo
    mkdir -p build && cd build

    cmake .. -GNinja -DOT_PLATFORM=simulation -DOT_FULL_LOGS=1 -DOT_COMMISSIONER=ON -DOT_JOINER=ON
    ninja
    sudo ninja install
}

configure_network()
{
    echo 0 | sudo tee /proc/sys/net/ipv6/conf/all/disable_ipv6
    echo 1 | sudo tee /proc/sys/net/ipv6/conf/all/forwarding
    echo 1 | sudo tee /proc/sys/net/ipv4/conf/all/forwarding
}

case "$(uname)" in
    "Linux")
        disable_install_recommends
        sudo apt-get update
        install_common_dependencies

        if [ "$BUILD_TARGET" == script-check ] || [ "$BUILD_TARGET" == docker-check ]; then
            sudo bash third_party/openthread/repo/script/install_socat
            install_openthread_binraries
            configure_network
            exit 0
        fi

        if [ "$BUILD_TARGET" == check ] || [ "$BUILD_TARGET" == meshcop ]; then
            sudo bash third_party/openthread/repo/script/install_socat
            install_openthread_binraries
            sudo apt-get install --no-install-recommends -y avahi-daemon avahi-utils
            configure_network
        fi

        if [ "$BUILD_TARGET" == ncp_mode ]; then
            sudo bash third_party/openthread/repo/script/install_socat
            sudo apt-get install --no-install-recommends -y avahi-daemon avahi-utils
        fi

        if [ "$BUILD_TARGET" == scan-build ]; then
            pip3 install -U cmake
            sudo apt-get install --no-install-recommends -y clang clang-tools
        fi

        if [ "$BUILD_TARGET" == pretty-check ]; then
            sudo bash "$(dirname "$0")/install-llvm.sh"
            sudo apt-get install -y shellcheck
            sudo snap install shfmt
            npm install prettier@2.0.4
        fi

        if [ "${OTBR_MDNS-}" == 'mDNSResponder' ]; then
            SOURCE_NAME=mDNSResponder-2600.100.147
            (cd /tmp \
                && wget -c --tries=3 --no-check-certificate -O "$SOURCE_NAME.tar.gz" \
                    "https://github.com/apple-oss-distributions/mDNSResponder/archive/refs/tags/$SOURCE_NAME.tar.gz" \
                && rm -rf "/tmp/$SOURCE_NAME" \
                && mkdir -p "$SOURCE_NAME" \
                && tar xvf "$SOURCE_NAME.tar.gz" -C "$SOURCE_NAME" --overwrite --strip-components=1 \
                && cd "$SOURCE_NAME" \
                && cd mDNSPosix \
                && make os=linux tls=no && sudo make install os=linux tls=no)
        fi

        # Enable IPv6
        if [ "$BUILD_TARGET" == check ]; then
            echo 0 | sudo tee /proc/sys/net/ipv6/conf/all/disable_ipv6
        fi

        # Allow access syslog file for unit test
        if [ "$BUILD_TARGET" == check ]; then
            sudo chmod a+r /var/log/syslog
        fi

        # Prepare Raspbian image
        if [ "$BUILD_TARGET" == raspbian-gcc ]; then
            sudo apt-get install --no-install-recommends --allow-unauthenticated -y qemu-system qemu-user-static binfmt-support parted

            (mkdir -p docker-rpi-emu \
                && cd docker-rpi-emu \
                && (git --git-dir=.git rev-parse --is-inside-work-tree || git --git-dir=.git init .) \
                && git fetch --depth 1 https://github.com/ryankurte/docker-rpi-emu.git master \
                && git checkout FETCH_HEAD)

            pip3 install git-archive-all

            # Determine archive type and image stem name from IMAGE_URL
            IMAGE_STEM_NAME=""
            ARCHIVE_TYPE=""

            if [[ ${IMAGE_URL} == *.zip ]]; then
                ARCHIVE_TYPE="zip"
                IMAGE_STEM_NAME=$(basename "${IMAGE_URL}" .zip)
            elif [[ ${IMAGE_URL} == *.img.xz ]]; then
                ARCHIVE_TYPE="xz"
                IMAGE_STEM_NAME=$(basename "${IMAGE_URL}" .img.xz)
            else
                echo "Unsupported image archive format in IMAGE_URL: ${IMAGE_URL}" >&2
                echo "Must end in .zip or .img.xz" >&2
                exit 1
            fi

            # The final .img filename, e.g., "raspbian.img"
            FINAL_IMAGE_FILENAME="${IMAGE_STEM_NAME}.img"
            # Full path to where the final .img file will be stored
            FINAL_IMAGE_PATH_IN_TOOLS="$TOOLS_HOME/images/$FINAL_IMAGE_FILENAME"

            [ -f "$FINAL_IMAGE_PATH_IN_TOOLS" ] || {
                # unit MB
                EXPAND_SIZE=1024

                [ -d "$TOOLS_HOME"/images ] || mkdir -p "$TOOLS_HOME"/images

                # Name of the archive file as it will be saved by curl -LO
                LOCAL_ARCHIVE_FILENAME=$(basename "${IMAGE_URL}")

                # Download the archive if it's not already present in the current directory
                [[ -f $LOCAL_ARCHIVE_FILENAME ]] || curl -fLO "$IMAGE_URL"

                # Decompress/extract the image to /tmp with the name $FINAL_IMAGE_FILENAME
                # e.g. /tmp/raspbian.img
                if [ "$ARCHIVE_TYPE" == "zip" ]; then
                    # Assumes the zip file "$IMAGE_STEM_NAME.zip" contains "$IMAGE_STEM_NAME.img" at its root
                    unzip -o "$LOCAL_ARCHIVE_FILENAME" "${IMAGE_STEM_NAME}.img" -d /tmp
                elif [ "$ARCHIVE_TYPE" == "xz" ]; then
                    # Decompresses "$IMAGE_STEM_NAME.img.xz" to "/tmp/$IMAGE_STEM_NAME.img"
                    xz -dc "$LOCAL_ARCHIVE_FILENAME" >"/tmp/$FINAL_IMAGE_FILENAME"
                fi

                # Expand the image in /tmp and then move it to its final destination
                (cd /tmp \
                    && dd if=/dev/zero bs=1048576 count="$EXPAND_SIZE" >>"$FINAL_IMAGE_FILENAME" \
                    && mv "$FINAL_IMAGE_FILENAME" "$FINAL_IMAGE_PATH_IN_TOOLS")

                # Clean up the downloaded archive from the current directory
                rm -f "$LOCAL_ARCHIVE_FILENAME"

                (cd docker-rpi-emu/scripts \
                    && sudo ./expand.sh "$FINAL_IMAGE_PATH_IN_TOOLS" "$EXPAND_SIZE")
            }
        fi
        ;;

    "Darwin")
        /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install.sh)"
        ;;

    *)
        echo "Unknown os type"
        exit 1
        ;;
esac
