fix: only update pmc repo when installing deb pkgs (#7332)

Author: lilypan26

parts/linux/cloud-init/artifacts/ubuntu/cse_helpers_ubuntu.sh

@@ -46,6 +46,10 @@ _apt_get_update() {
 apt_get_update() {
     _apt_get_update 10 ""
 }
+apt_get_update_with_opts() {
+    local apt_opts=$1
+    _apt_get_update 10 "${apt_opts}"
+}
 _apt_get_install() {
     local retries=$1
     local wait_sleep=$2

parts/linux/cloud-init/artifacts/ubuntu/cse_install_ubuntu.sh

@@ -72,6 +72,11 @@ updateAptWithMicrosoftPkg() {
     apt_get_update || exit $ERR_APT_UPDATE_TIMEOUT
 }
 
+updatePMCRepository() {
+    local opts="-o Dir::Etc::sourcelist=/etc/apt/sources.list.d/microsoft-prod.list -o Dir::Etc::sourceparts=-"
+    apt_get_update_with_opts "${opts}" || exit $ERR_APT_UPDATE_TIMEOUT
+}
+
 updateAptWithNvidiaPkg() {
     readonly nvidia_gpg_keyring_path="/etc/apt/keyrings/nvidia.pub"
     mkdir -p "$(dirname "${nvidia_gpg_keyring_path}")"
@@ -272,8 +277,9 @@ installPkgWithAptGet() {
             return 0
         fi
 
+        # update pmc repo to get latest package versions
+        updatePMCRepository
         # query all package versions and get the latest version for matching k8s version
-        updateAptWithMicrosoftPkg
         fullPackageVersion=$(apt list ${packageName} --all-versions | grep ${packageVersion}- | awk '{print $2}' | sort -V | tail -n 1)
         if [ -z "${fullPackageVersion}" ]; then
             echo "Failed to find valid ${packageName} version for ${packageVersion}"