Save chrome and chromium deb files in package registry

.gitlab-ci.yml

@@ -1,6 +1,5 @@
 # we support merge request workflow only
 include:
-  - template: 'Workflows/MergeRequest-Pipelines.gitlab-ci.yml'
   - local: '.gitlab/ci/*.yml'
 
 default:
@@ -23,13 +22,5 @@ stages:
 variables:
   DOCKER_HOST: tcp://docker:2375
 
-cache-google-chrome:
-  image: debian:bullseye-slim
-  stage: automation
-  # Starts the job immediately
-  needs: []
-  script:
-    - bash ./scripts/cache-google-chrome
-  # disable DinD
-  before_script: []
-  services: []
+cache-chrome-chromium:
+  extends: .cache-google-chrome

.gitlab/ci/definitions.yml

@@ -30,3 +30,23 @@
       variables:
         PUSH_CUSTOM_IMAGE: "true"
         BASE_BUILD_REGISTRY_IMAGE: $CI_REGISTRY_IMAGE
+
+.cache-google-chrome:
+  stage: automation
+  needs: []
+  before_script:
+    - docker buildx create --use
+  script:
+    - |
+      docker buildx build \
+        --build-arg CI_API_V4_URL=$CI_API_V4_URL \
+        --build-arg CI_PROJECT_ID=$CI_PROJECT_ID \
+        --build-arg CI_JOB_TOKEN=$CI_JOB_TOKEN \
+        --platform linux/amd64,linux/arm64 \
+        -f "Dockerfile.cache-chrome" \
+        .
+  rules:
+    - if: '$CI_COMMIT_BRANCH == $CI_DEFAULT_BRANCH && $CI_SERVER_HOST == "gitlab.com"'
+    - if: '$CI_PIPELINE_SOURCE == "schedule" && $CI_SERVER_HOST == "gitlab.com"'
+    - if: '$CI_PIPELINE_SOURCE == "merge_request_event" && $CI_SERVER_HOST == "gitlab.com"'
+      when: manual

Dockerfile.cache-chrome

@@ -0,0 +1,11 @@
+# Save amd64 chrome and arm64 .deb files to package registry
+FROM debian:latest
+
+ARG TARGETARCH
+ARG CI_API_V4_URL
+ARG CI_PROJECT_ID
+ARG CI_JOB_TOKEN
+
+ADD scripts/cache-google-chrome /cache-google-chrome
+
+RUN /cache-google-chrome

scripts/cache-google-chrome

@@ -1,27 +1,27 @@
 #!/bin/bash
+
 # This script attempts to copy the latest version of the Google Chrome Debian
 # package into our own package registry. Google yanks old versions regularly,
 # making it hard to keep up with all the new versions.
 
 set -e
 
+export DEBIAN_FRONTEND=noninteractive
+
+function cache-chrome() {
   PKG=google-chrome-stable
 
-export DEBIAN_FRONTEND=noninteractive
-apt-get -y update
-apt-get -y install apt-utils curl bash gnupg2
-
   curl -sS -L https://dl.google.com/linux/linux_signing_key.pub | apt-key add -
-echo "deb [arch=$(dpkg --print-architecture)] http://dl.google.com/linux/chrome/deb/ stable main" > /etc/apt/sources.list.d/google.list
+  echo "deb http://dl.google.com/linux/chrome/deb/ stable main" >/etc/apt/sources.list.d/google.list
 
   echo "Updating apt to get Google Chrome packages..."
 
-apt-get -y -q update
+  apt-get -y -qq update
 
   echo "Checking for latest Chrome version in apt repository..."
 
   LATEST_VERSION=$(apt-cache show $PKG | grep Version | sort | tail -1 | sed -e "s/Version: //")
-CHROME_DEB="google-chrome-stable_${LATEST_VERSION}_$(dpkg --print-architecture).deb"
+  CHROME_DEB="google-chrome-stable_${LATEST_VERSION}_amd64.deb"
   CHROME_URL="${CI_API_V4_URL}/projects/${CI_PROJECT_ID}/packages/generic/${PKG}/${LATEST_VERSION}/${CHROME_DEB}"
 
   echo "Checking if cache has $CHROME_DEB"
@@ -41,7 +41,54 @@ else
     fi
 
     echo "Transferring $CHROME_DEB to GitLab packages"
-  curl --header "JOB-TOKEN: ${CI_JOB_TOKEN}" \
+    curl --fail --header "JOB-TOKEN: ${CI_JOB_TOKEN}" \
       --upload-file "./{$CHROME_DEB}" \
       "$CHROME_URL"
   fi
+}
+
+function cache-chromium() {
+  PKG=chromium
+
+  echo "Checking for latest Chromium version in apt repository..."
+
+  LATEST_VERSION=$(apt-cache show $PKG | grep Version | sort | tail -1 | sed -e "s/Version: //")
+  VERSION_NUMBER=$(echo $LATEST_VERSION | sed -e "s/~deb.*//") # remove debian version part to have chrome and chromium compatible version numbers
+  CHROMIUM_DEB="chromium_${VERSION_NUMBER}_arm64.deb"
+  CHROMIUM_URL="${CI_API_V4_URL}/projects/${CI_PROJECT_ID}/packages/generic/${PKG}/${VERSION_NUMBER}/${CHROMIUM_DEB}"
+
+  echo "Checking if cache has ${CHROMIUM_DEB}"
+  FILE_CHECK=$(curl --silent --location --head --output /dev/null --write "%{http_code}\\n" "$CHROMIUM_URL")
+
+  if [ "$FILE_CHECK" -eq "200" ]; then
+    echo "Latest version ${LATEST_VERSION} is already cached!"
+  else
+    echo "Downloading latest Chromium version (${LATEST_VERSION}) from apt repository..."
+    cd /tmp
+    apt-get download "$PKG"
+
+    DEB_FILE="chromium_${LATEST_VERSION}_arm64.deb"
+
+    if ! [ -f "$DEB_FILE" ]; then
+      echo "Downloaded file didn't have expected name: ${DEB_FILE}"
+      ls
+      exit 1
+    fi
+
+    mv "$DEB_FILE" "$CHROMIUM_DEB"
+
+    echo "Transferring ${CHROMIUM_DEB} to GitLab packages"
+    curl --fail --header "JOB-TOKEN: ${CI_JOB_TOKEN}" \
+      --upload-file "./{$CHROMIUM_DEB}" \
+      "$CHROMIUM_URL"
+  fi
+}
+
+apt-get -y -qq update
+apt-get -y install apt-utils curl bash gnupg2 > /dev/null
+
+if [ "$TARGETARCH" == "amd64" ]; then
+  cache-chrome
+else
+  cache-chromium
+fi

scripts/install-chrome

@@ -4,47 +4,26 @@ set -xeuo pipefail
 IFS=$'\n\t'
 
 function build_debian() {
-    ARCH="${TARGETARCH:-amd64}"
+    CHROME_VERSION=${1:-103.0.5060.134-1}
+    ARCH=${TARGETARCH:-amd64}
 
-    if [[ "${ARCH}" == "arm64" ]]; then
-        echo "The arm64 does not have prebuilt chrome. Using chromium instead."
-        apt-get update -q -y
-        apt-get install -y chromium chromium-driver
-        apt-get autoremove -yq
-        apt-get clean -yqq
-        rm -rf /var/lib/apt/lists/*
-        exit 0
+    if [ "${ARCH}" == "amd64" ]; then
+        PKG=google-chrome-stable
+    else
+        PKG=chromium
     fi
 
-    CHROME_VERSION=${1:-99.0.4844.74-1}
     # 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/google-chrome-stable"
-    export DEBIAN_FRONTEND=noninteractive
-
-    curl -sS -L https://dl.google.com/linux/linux_signing_key.pub | apt-key add -
-    echo "deb http://dl.google.com/linux/chrome/deb/ stable main" >/etc/apt/sources.list.d/google.list
-
-    apt-get update -q -y
-
-    # Download from our package registry if we can't find the package in the apt repository
-    echo "Searching for $CHROME_VERSION in apt repository"
-    CHECK_VERSION=$(apt-cache show google-chrome-stable | grep Version | grep "$CHROME_VERSION") || true
-
-    apt-cache policy google-chrome-stable
-
-    if [[ -z $CHECK_VERSION ]]; then
-        CHROME_DEB="google-chrome-stable_${CHROME_VERSION}_${ARCH}.deb"
+    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 from our Package registry: $CHROME_URL"
+
+    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
-    else
-        echo "Installing via apt-get"
-        apt-get install -y google-chrome-stable=$CHROME_VERSION
-    fi
 
     rm -rf /var/lib/apt/lists/*