#!/bin/bash

set -xeuo pipefail
IFS=$'\n\t'

function build_debian() {
    CHROME_VERSION=${1:-103.0.5060.134-1}
    ARCH=${TARGETARCH:-amd64}

    if [ "${ARCH}" == "amd64" ]; then
        PKG=google-chrome-stable
    else
        PKG=chromium
    fi

    # We hard code the URL rather than using $CI_API_V4_URL $CI_PROJECT_ID,
    # because we would need to forward those variables
    CHROME_DOWNLOAD_URL_BASE="https://gitlab.com/api/v4/projects/1075790/packages/generic/${PKG}"
    CHROME_DEB="${PKG}_${CHROME_VERSION}_${ARCH}.deb"
    CHROME_URL="${CHROME_DOWNLOAD_URL_BASE}/${CHROME_VERSION}/${CHROME_DEB}"

    echo "Downloading deb file from Package registry: $CHROME_URL"
    curl --silent --show-error --fail -O $CHROME_URL
    dpkg -i ./$CHROME_DEB || true
    apt-get install -f -y
    rm -f $CHROME_DEB

    rm -rf /var/lib/apt/lists/*

    # Install ChromeDriver
    CHROME_VERSION_BASE=$(echo $CHROME_VERSION | awk -F "." '{print $1 "." $2 "." $3}')
    CHROME_DRIVER_VERSION=$(curl -q https://chromedriver.storage.googleapis.com/LATEST_RELEASE_$CHROME_VERSION_BASE)
    wget -q https://chromedriver.storage.googleapis.com/$CHROME_DRIVER_VERSION/chromedriver_linux64.zip
    unzip chromedriver_linux64.zip -d /usr/local/bin
    rm -f chromedriver_linux64.zip

    apt-get autoremove -yq
    apt-get clean -yqq
    rm -rf /var/lib/apt/lists/*
    rm -rf /etc/apt/sources.list.d/google*.list
}

BUILD_OS=${BUILD_OS:-debian}

if [[ $BUILD_OS =~ debian ]]; then
    build_debian "$@"
elif [[ $BUILD_OS =~ ubi ]]; then
    build_ubi "$@"
fi