From c4a2829fb47493a6f53ef76183ede00cec788897 Mon Sep 17 00:00:00 2001
From: Westly Wright <westly.wright@gmail.com>
Date: Thu, 5 Mar 2020 15:11:02 -0700
Subject: [PATCH] Add drain options to k3s imported clusters edit

---
 .../components/k3s-cluster-info/template.hbs  | 29 +++++++++++++++++++
 translations/en-us.yaml                       |  2 ++
 2 files changed, 31 insertions(+)

diff --git a/lib/shared/addon/components/k3s-cluster-info/template.hbs b/lib/shared/addon/components/k3s-cluster-info/template.hbs
index a2ae629ec..e31ed4c68 100644
--- a/lib/shared/addon/components/k3s-cluster-info/template.hbs
+++ b/lib/shared/addon/components/k3s-cluster-info/template.hbs
@@ -17,6 +17,35 @@
     </div>
   </div>
 </div>
+<div class="row">
+  <div class="col span-6">
+    <div class="form-control checkbox">
+      {{input
+        id="drain-master-nodes"
+        class="form-control"
+        type="checkbox"
+        checked=k3sConfig.k3supgradeStrategy.drainServerNodes
+      }}
+      <label for="drain-master-nodes">
+        {{t "k3sClusterInfo.drainServerNodes"}}
+      </label>
+    </div>
+  </div>
+  <div class="col span-6">
+    <div class="form-control checkbox">
+      {{input
+        id="drain-worker-nodes"
+        class="form-control"
+        type="checkbox"
+        checked=k3sConfig.k3supgradeStrategy.drainWorkerNodes
+        disabled=(not editing)
+      }}
+      <label for="drain-worker-nodes">
+        {{t "k3sClusterInfo.drainWorkerNodes"}}
+      </label>
+    </div>
+  </div>
+</div>
 <div class="row">
   <div class="col span-6">
     <label class="acc-label" for="node-server-concurrency">
diff --git a/translations/en-us.yaml b/translations/en-us.yaml
index fd3715742..a702d5a0e 100644
--- a/translations/en-us.yaml
+++ b/translations/en-us.yaml
@@ -7142,6 +7142,8 @@ k3sClusterInfo:
   detail: Customize the K3S cluster options
   workerConcurrency: Worker Concurrency
   serverConcurrency: Server Concurrency
+  drainServerNodes: Drain Server Nodes
+  drainWorkerNodes: Drain Worker Nodes
   nodeInfo:
     title: K3s Node Information
     detail: Read-only view of K3S master node arguments and environment variables.