#!/usr/bin/env bash

if [[ $(id -u) != "0" ]]; then
  echo "root privileges required"
  exit 1
fi

set -e

if [[ -z $1 ]]; then
    echo "architecture required: amd64, arm64, or arm" 1>&2
    exit 1
else
    set -eux;
    umask 0000
    curl -sL https://deb.nodesource.com/setup_14.x | bash -

    # Install Chrome & NodeJS
    if [[ $1 == "amd64" ]]; then
        wget -q -O - https://dl-ssl.google.com/linux/linux_signing_key.pub | apt-key add -
        sh -c 'echo "deb [arch=amd64] http://dl.google.com/linux/chrome/deb/ stable main" >> /etc/apt/sources.list.d/google-chrome.list'
        apt-get update
        apt-get -qq install google-chrome-stable nodejs
    elif [[ $1 == "arm64" ]]; then
cat <<EOF > /etc/apt/preferences.d/chromium
Package: *
Pin: release o=LP-PPA-saiarcot895-chromium-dev
Pin-Priority: 1002
EOF
        add-apt-repository -y ppa:saiarcot895/chromium-dev
        apt-get update
        apt-get -qq install chromium-browser chromium-codecs-ffmpeg-extra nodejs
    else
        apt-get update
        apt-get -qq install nodejs
    fi

    # Remove package files
    apt-get -y autoremove
    apt-get -y autoclean
    apt-get clean
    rm -rf /var/lib/apt/lists/*

    # Install NPM
    npm install --unsafe-perm=true --allow-root -g npm
    npm config set cache ~/.cache/npm

    # Install Puppeteer, TestCafe & ChromeDriver
    if [[ $1 == "amd64" ]]; then
        npm install --unsafe-perm=true --allow-root -g puppeteer testcafe testcafe-browser-provider-puppeteer chromedriver
    elif [[ $1 == "arm64" ]]; then
        npm install --unsafe-perm=true --allow-root -g testcafe chromedriver
    fi
fi