From 58d09bd794cd5092d553122136ccb3c9ae17ee9c Mon Sep 17 00:00:00 2001
From: Sam Calder-Mason <sam@puritydev.io>
Date: Thu, 23 Nov 2023 11:31:17 +1000
Subject: [PATCH] feat(cloudflare-tunnel): Add PDB

---
 charts/cloudflare-tunnel/Chart.yaml         |  2 +-
 charts/cloudflare-tunnel/templates/pdb.yaml | 18 ++++++++++++++++++
 charts/cloudflare-tunnel/values.yaml        |  6 ++++++
 3 files changed, 25 insertions(+), 1 deletion(-)
 create mode 100644 charts/cloudflare-tunnel/templates/pdb.yaml

diff --git a/charts/cloudflare-tunnel/Chart.yaml b/charts/cloudflare-tunnel/Chart.yaml
index 3f4a3ca..2b9d8bd 100644
--- a/charts/cloudflare-tunnel/Chart.yaml
+++ b/charts/cloudflare-tunnel/Chart.yaml
@@ -2,7 +2,7 @@ apiVersion: v2
 name: cloudflare-tunnel
 description: Creation of a cloudflared deployment - a reverse tunnel for an environment
 type: application
-version: 0.1.3
+version: 0.2.0
 icon: https://developers.cloudflare.com/cloudflare-one/favicon-32x32.png
 maintainers:
   - name: barnabasbusa
diff --git a/charts/cloudflare-tunnel/templates/pdb.yaml b/charts/cloudflare-tunnel/templates/pdb.yaml
new file mode 100644
index 0000000..7aa4f8d
--- /dev/null
+++ b/charts/cloudflare-tunnel/templates/pdb.yaml
@@ -0,0 +1,18 @@
+{{- if or .Values.podDisruptionBudget.minAvailable .Values.podDisruptionBudget.maxUnavailable -}}
+apiVersion: policy/v1
+kind: PodDisruptionBudget
+metadata:
+  name: {{ include "cloudflare-tunnel.fullname" . }}
+  labels:
+    {{- include "cloudflare-tunnel.labels" . | nindent 4 }}
+spec:
+  selector:
+    matchLabels:
+      {{- include "cloudflare-tunnel.selectorLabels" . | nindent 6 }}
+  {{- if .Values.podDisruptionBudget.minAvailable }}
+  minAvailable: {{ .Values.podDisruptionBudget.minAvailable }}
+  {{- end  }}
+  {{- if .Values.podDisruptionBudget.maxUnavailable }}
+  maxUnavailable: {{ .Values.podDisruptionBudget.maxUnavailable }}
+  {{- end  }}
+{{- end -}}
\ No newline at end of file
diff --git a/charts/cloudflare-tunnel/values.yaml b/charts/cloudflare-tunnel/values.yaml
index 952336e..c27b210 100644
--- a/charts/cloudflare-tunnel/values.yaml
+++ b/charts/cloudflare-tunnel/values.yaml
@@ -78,3 +78,9 @@ tolerations: []
 
 # Default affinity is to spread out over nodes; use this to override.
 affinity: {}
+
+# -- Define the PodDisruptionBudget spec
+# If not set then a PodDisruptionBudget will not be created
+podDisruptionBudget:
+  minAvailable: 1
+# maxUnavailable: 1
\ No newline at end of file