From de94b8cb06fc937b47c5134e7da1e79253744bc8 Mon Sep 17 00:00:00 2001
From: Bill Liang <bilia@microsoft.com>
Date: Sun, 26 Jul 2020 23:12:33 -0700
Subject: [PATCH] allow overriding package URL via env, patch docker if env
 exists

---
 .../setup-arc-data-controller.sh              | 30 ++++++++++++++-----
 1 file changed, 23 insertions(+), 7 deletions(-)

diff --git a/samples/features/azure-arc/deployment/kubeadm/ubuntu-single-node-vm/setup-arc-data-controller.sh b/samples/features/azure-arc/deployment/kubeadm/ubuntu-single-node-vm/setup-arc-data-controller.sh
index 736313c2..6435ad03 100644
--- a/samples/features/azure-arc/deployment/kubeadm/ubuntu-single-node-vm/setup-arc-data-controller.sh
+++ b/samples/features/azure-arc/deployment/kubeadm/ubuntu-single-node-vm/setup-arc-data-controller.sh
@@ -147,8 +147,12 @@ sudo apt install -y libodbc1 odbcinst odbcinst1debian2 unixodbc apt-transport-ht
 #
 echo ""
 
-if [[ -v AZDATA_DEB_PATH ]]; then
-  sudo dpkg -i $AZDATA_DEB_PATH
+if [[ -v AZDATA_DEB_PACKAGE_PATH ]]; then
+  sudo dpkg -i $AZDATA_DEB_PACKAGE_PATH
+elif [[ -v AZDATA_DEB_PACKAGE_URL ]]; then
+  echo "Downloading azdata installer from" $AZDATA_DEB_PACKAGE_URL 
+  curl --location $AZDATA_DEB_PACKAGE_URL --output azdata_setup.deb
+  sudo dpkg -i azdata_setup.deb
 else
   echo "Downloading azdata installer from" $AZDATA_PRIVATE_PREVIEW_DEB_PACKAGE 
   curl --location $AZDATA_PRIVATE_PREVIEW_DEB_PACKAGE --output azdata_setup.deb
@@ -325,11 +329,23 @@ echo "Starting to deploy azdata cluster..."
 
 # Command to create cluster for single node cluster.
 #
-azdata arc dc config init --source azure-arc-kubeadm-dev-test --path azure-arc-custom --force
-azdata arc dc config replace --path azure-arc-custom/control.json --json-values '$.spec.docker.registry=$DOCKER_REGISTRY'
-azdata arc dc config replace --path azure-arc-custom/control.json --json-values '$.spec.docker.repository=$DOCKER_REPOSITORY'
-azdata arc dc config replace --path azure-arc-custom/control.json --json-values '$.spec.docker.imageTag=$DOCKER_IMAGE_TAG'
-azdata arc dc config replace --path azure-arc-custom/control.json --json-values '$.spec.docker.imagePullPolicy=IfNotPresent'
+azdata arc dc config init --source azure-arc-kubeadm --path azure-arc-custom --force
+
+if [[ -v DOCKER_REGISTRY ]]; then
+    azdata arc dc config replace --path azure-arc-custom/control.json --json-values '$.spec.docker.registry=$DOCKER_REGISTRY'
+fi
+
+if [[ -v DOCKER_REPOSITORY ]]; then
+    azdata arc dc config replace --path azure-arc-custom/control.json --json-values '$.spec.docker.repository=$DOCKER_REPOSITORY'
+fi
+
+if [[ -v DOCKER_IMAGE_TAG ]]; then
+    azdata arc dc config replace --path azure-arc-custom/control.json --json-values '$.spec.docker.imageTag=$DOCKER_IMAGE_TAG'
+fi
+
+# azdata arc dc config replace --path azure-arc-custom/control.json --json-values '$.spec.docker.imagePullPolicy=IfNotPresent'
+azdata arc dc config replace --path azure-arc-custom/control.json --json-values '$.spec.storage.data.className=local-storage'
+azdata arc dc config replace --path azure-arc-custom/control.json --json-values '$.spec.storage.logs.className=local-storage'
 
 azdata arc dc create --name $ARC_DC_NAME --path azure-arc-custom --namespace $CLUSTER_NAME --location $ARC_DC_REGION --resource-group $ARC_DC_RG --subscription $ARC_DC_SUBSCRIPTION --connectivity-mode indirect
 echo "Azure Arc Data Controller cluster created."