From a0a7edd1047a01de9861bb030618964313141345 Mon Sep 17 00:00:00 2001
From: Frostebite <jas.f.ukcmti@gmail.com>
Date: Sun, 9 Aug 2020 02:53:12 +0100
Subject: [PATCH] Fix

---
 .github/workflows/main.yml       | 18 +-------
 ApplyClusterAndAcquireLock.sh    | 70 ++++++++++++++++++++++++++++++++
 ReleaseLockAndAttemptShutdown.sh | 13 ++++++
 3 files changed, 85 insertions(+), 16 deletions(-)
 create mode 100755 ApplyClusterAndAcquireLock.sh
 create mode 100755 ReleaseLockAndAttemptShutdown.sh

diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 2d2c5701..11cfdceb 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -99,13 +99,7 @@ jobs:
           version: '288.0.0'
           service_account_email: ${{ secrets.GOOGLE_SERVICE_ACCOUNT_EMAIL }}
           service_account_key: ${{ secrets.GOOGLE_SERVICE_ACCOUNT_KEY }}
-      - name: Get Gist
-        id: get-apply
-        uses: andymckay/get-gist-action@master
-        with:
-          gistURL: 'https://gist.github.com/frostebite/58fdeb02d45f6e7a24bcbe36f6bdc051'
-      - run: chmod +x ${{ steps.get-apply.outputs.file }}
-      - run: ${{ steps.get-apply.outputs.file }} ${{ env.GKE_PROJECT }} ${{ env.GKE_CLUSTER }} ${{ env.GKE_ZONE }}
+      - run: ./ApplyClusterAndAcquireLock.sh ${{ env.GKE_PROJECT }} ${{ env.GKE_CLUSTER }} ${{ env.GKE_ZONE }}
       - uses: frostebite/File-To-Base64@master
         id: read-base64
         with:
@@ -129,13 +123,5 @@ jobs:
         with:
           name: Kubernetes Build (${{ matrix.targetPlatform }})
           path: k8s-volume-download
-      - name: Get Gist
-        id: get-shutdown
-        uses: andymckay/get-gist-action@master
-        with:
-          gistURL: 'https://gist.github.com/frostebite/5964fdd4d1c84507673ddef035d344ca'
-        if: ${{ always() }}
-      - run: |
-          chmod +x ${{ steps.get-shutdown.outputs.file }}
-          ${{ steps.get-shutdown.outputs.file }} ${{ env.GKE_PROJECT }} ${{ env.GKE_CLUSTER }} ${{ env.GKE_ZONE }}
+      - run: ./ReleaseLockAndAttemptShutdown.sh ${{ env.GKE_PROJECT }} ${{ env.GKE_CLUSTER }} ${{ env.GKE_ZONE }}
         if: ${{ always() }}
diff --git a/ApplyClusterAndAcquireLock.sh b/ApplyClusterAndAcquireLock.sh
new file mode 100755
index 00000000..7c42b32c
--- /dev/null
+++ b/ApplyClusterAndAcquireLock.sh
@@ -0,0 +1,70 @@
+#!/bin/sh
+
+# This creates a GKE Cluster
+# - Will wait for any deletion to complete on a cluster with the same name before creating
+# - Will wait for completion before continuing
+# - If the script is run concurrently multiple times, only one cluster will be created, all instances will wait for availability
+# Requires GCP Cloud SDK
+# Installs retry https://github.com/kadwanev/retry
+
+export GKE_PROJECT=$1
+export GKE_CLUSTER=$2
+export GKE_ZONE=$3
+
+attempts=0
+while [ attempts == 0 ]
+  do
+
+# may update this to avoid repeated install, drop me a comment if needed
+sudo sh -c "curl https://raw.githubusercontent.com/kadwanev/retry/master/retry -o /usr/local/bin/retry && chmod +x /usr/local/bin/retry"
+
+retry -s 15 -t 20 -v '
+    STATUS=$(gcloud container clusters list --format="json" --project $GKE_PROJECT |
+    jq "
+    .[] |
+    {name: .name, status: .status} |
+    select(.name == \"$GKE_CLUSTER\")
+    " |
+    jq ".status")
+    if [ "$STATUS" == "\"STOPPING\"" ]; then echo "Cluster stopping waiting for completion" && exit 1; fi
+    exit 0
+  '
+
+cluster=$(gcloud container clusters list --project $GKE_PROJECT --format="json" | jq '.[] | select(.name == "${GKE_CLUSTER}")')
+
+if [ -z "$cluster" ];
+then
+  echo "No clusters found for \"$GKE_CLUSTER\" in project \"$GKE_CLUSTER\" in zone \"$GKE_ZONE\""
+  # you may not need this, it installs GCP beta for additional command line options
+  gcloud components install beta -q
+  # replace this line with whatever type of cluster you want to create
+  gcloud beta container --project $GKE_PROJECT clusters create $GKE_CLUSTER --zone $GKE_ZONE --no-enable-basic-auth --cluster-version "1.15.12-gke.2" --machine-type "custom-1-3072" --image-type "COS" --disk-type "pd-standard" --disk-size "15" --metadata disable-legacy-endpoints=true --scopes "https://www.googleapis.com/auth/devstorage.read_only","https://www.googleapis.com/auth/logging.write","https://www.googleapis.com/auth/monitoring","https://www.googleapis.com/auth/servicecontrol","https://www.googleapis.com/auth/service.management.readonly","https://www.googleapis.com/auth/trace.append" --num-nodes "1" --enable-stackdriver-kubernetes --enable-ip-alias --default-max-pods-per-node "110" --enable-autoscaling --min-nodes "0" --max-nodes "3" --no-enable-master-authorized-networks --addons HorizontalPodAutoscaling,HttpLoadBalancing --enable-autoupgrade --enable-autorepair --max-surge-upgrade 1 --max-unavailable-upgrade 0
+fi;
+
+retry -s 15 -t 20 -v '
+  STATUS=$(gcloud container clusters list --format="json" --project $GKE_PROJECT |
+  jq "
+  .[] |
+  {name: .name, status: .status} |
+  select(.name == \"$GKE_CLUSTER\")
+  " |
+  jq ".status")
+  if [ "$STATUS" == "\"PROVISIONING\"" ]; then echo "Cluster provisioning waiting for available" && exit 1; fi
+  exit 0
+'
+
+echo "Cluster is available"
+gcloud container clusters get-credentials $GKE_CLUSTER --zone $GKE_ZONE --project $GKE_PROJECT
+kubectl version
+NSID=$(cat /proc/sys/kernel/random/uuid)
+echo "::set-env name=NSID::"$NSID
+cat <<EOF | kubectl apply -f -
+apiVersion: v1
+kind: Namespace
+metadata:
+  name: ns-unity-builder-$NSID
+  labels:
+    app: unity-builder
+EOF && exit 0
+attempts=attempts+1
+done
diff --git a/ReleaseLockAndAttemptShutdown.sh b/ReleaseLockAndAttemptShutdown.sh
new file mode 100755
index 00000000..a25b61d8
--- /dev/null
+++ b/ReleaseLockAndAttemptShutdown.sh
@@ -0,0 +1,13 @@
+kubectl delete ns ns-unity-builder-$NSID
+
+# do any unity-builder namespaces remain?
+namespaceCount=$(kubectl get ns --output json | jq ".items | .[] | select(.metadata.labels.app == \"unity-builder\") | select(.status.phase != \"TERMINATING\")" | jq -s "length")
+echo $namespaceCount
+if [ "$namespaceCount" != "0" ]
+then
+   echo "let next cluster delete"
+   exit 0
+else
+   echo "delete cluster"
+   gcloud container clusters delete $GKE_CLUSTER
+fi