From 68a1082e6dd11b97856dbf1ee66d66fecf01e3b5 Mon Sep 17 00:00:00 2001
From: Shuo Wu <shuo@rancher.com>
Date: Mon, 12 Oct 2020 12:38:12 +0800
Subject: [PATCH] chart: Add settings for CSI Snapshotter

Signed-off-by: Shuo Wu <shuo@rancher.com>
---
 chart/questions.yml                    | 22 ++++++++++++++++++++++
 chart/templates/deployment-driver.yaml |  9 +++++++++
 chart/values.yaml                      |  4 ++++
 3 files changed, 35 insertions(+)

diff --git a/chart/questions.yml b/chart/questions.yml
index 86cf453..eead561 100644
--- a/chart/questions.yml
+++ b/chart/questions.yml
@@ -106,6 +106,12 @@ questions:
     type: string
     label: Longhorn CSI Driver Resizer Image Tag
     group: "Longhorn CSI Driver Images"
+  - variable: image.csi.snapshotter.tag
+    default: v2.1.1
+    description: "Specify CSI Driver Snapshotter image tag. Leave blank to autodetect."
+    type: string
+    label: Longhorn CSI Driver Snapshotter Image Tag
+    group: "Longhorn CSI Driver Images"
 - variable: privateRegistry.registryUrl
   label: Private registry URL
   description: "URL of private registry. Leave blank to apply system default registry."
@@ -160,6 +166,22 @@ questions:
     description: "Specify replica count of CSI Provisioner. By default 3."
     label: Longhorn CSI Provisioner replica count
     group: "Longhorn CSI Driver Settings"
+  - variable: csi.resizerReplicaCount
+    type: int
+    default:
+    min: 1
+    max: 10
+    description: "Specify replica count of CSI Resizer. By default 3."
+    label: Longhorn CSI Resizer replica count
+    group: "Longhorn CSI Driver Settings"
+  - variable: csi.snapshotterReplicaCount
+    type: int
+    default:
+    min: 1
+    max: 10
+    description: "Specify replica count of CSI Snapshotter. By default 3."
+    label: Longhorn CSI Snapshotter replica count
+    group: "Longhorn CSI Driver Settings"
   - variable: persistence.defaultClass
     default: "true"
     description: "Set as default StorageClass"
diff --git a/chart/templates/deployment-driver.yaml b/chart/templates/deployment-driver.yaml
index 628822b..ec3a867 100644
--- a/chart/templates/deployment-driver.yaml
+++ b/chart/templates/deployment-driver.yaml
@@ -63,6 +63,10 @@ spec:
           - name: CSI_RESIZER_IMAGE
             value: "{{ template "registry_url" . }}{{ .Values.image.csi.resizer.repository }}:{{ .Values.image.csi.resizer.tag }}"
           {{- end }}
+          {{- if and .Values.image.csi.snapshotter.repository .Values.image.csi.snapshotter.tag }}
+          - name: CSI_SNAPSHOTTER_IMAGE
+            value: "{{ template "registry_url" . }}{{ .Values.image.csi.snapshotter.repository }}:{{ .Values.image.csi.snapshotter.tag }}"
+          {{- end }}
           {{- if .Values.csi.attacherReplicaCount }}
           - name: CSI_ATTACHER_REPLICA_COUNT
             value: {{ .Values.csi.attacherReplicaCount | quote }}
@@ -75,6 +79,11 @@ spec:
           - name: CSI_RESIZER_REPLICA_COUNT
             value: {{ .Values.csi.resizerReplicaCount | quote }}
           {{- end }}
+          {{- if .Values.csi.snapshotterReplicaCount }}
+          - name: CSI_SNAPSHOTTER_REPLICA_COUNT
+            value: {{ .Values.csi.snapshotterReplicaCount | quote }}
+          {{- end }}
+
       {{- if .Values.defaultSettings.registrySecret }}
       imagePullSecrets:
       - name: {{ .Values.defaultSettings.registrySecret }}
diff --git a/chart/values.yaml b/chart/values.yaml
index 83369d8..398a659 100644
--- a/chart/values.yaml
+++ b/chart/values.yaml
@@ -32,6 +32,9 @@ image:
     resizer:
       repository: longhornio/csi-resizer
       tag: v0.3.0
+    snapshotter:
+      repository: longhornio/csi-snapshotter
+      tag: v2.1.1
   pullPolicy: IfNotPresent
 
 service:
@@ -51,6 +54,7 @@ csi:
   attacherReplicaCount: ~
   provisionerReplicaCount: ~
   resizerReplicaCount: ~
+  snapshotterReplicaCount: ~
 
 defaultSettings:
   backupTarget: ~