From 885bfde4fd70d36bbc01225303424b62960c2f4f Mon Sep 17 00:00:00 2001
From: davidjumani <dj.davidjumani1994@gmail.com>
Date: Wed, 17 Jun 2020 05:22:08 +0000
Subject: [PATCH] infra: Adding rolling maintenance (#424)

Fixes #363

Signed-off-by: Rohit Yadav <rohit.yadav@shapeblue.com>
---
 ui/src/config/section/infra/clusters.js | 13 +++++++++++++
 ui/src/config/section/infra/hosts.js    | 16 ++++++++++++++++
 ui/src/config/section/infra/pods.js     | 13 +++++++++++++
 ui/src/config/section/infra/zones.js    | 13 +++++++++++++
 ui/src/locales/en.json                  |  1 +
 5 files changed, 56 insertions(+)

diff --git a/ui/src/config/section/infra/clusters.js b/ui/src/config/section/infra/clusters.js
index ac3868dde75..d2d748da6d5 100644
--- a/ui/src/config/section/infra/clusters.js
+++ b/ui/src/config/section/infra/clusters.js
@@ -147,6 +147,19 @@ export default {
         }
       }
     },
+    {
+      api: 'startRollingMaintenance',
+      icon: 'setting',
+      label: 'label.start.rolling.maintenance',
+      message: 'label.start.rolling.maintenance',
+      dataView: true,
+      args: ['timeout', 'payload', 'forced', 'clusterids'],
+      mapping: {
+        clusterids: {
+          value: (record) => { return record.id }
+        }
+      }
+    },
     {
       api: 'deleteCluster',
       icon: 'delete',
diff --git a/ui/src/config/section/infra/hosts.js b/ui/src/config/section/infra/hosts.js
index de1e8273eca..34dcebb31f4 100644
--- a/ui/src/config/section/infra/hosts.js
+++ b/ui/src/config/section/infra/hosts.js
@@ -243,6 +243,22 @@ export default {
         }
       }
     },
+    {
+      api: 'startRollingMaintenance',
+      icon: 'setting',
+      label: 'label.start.rolling.maintenance',
+      message: 'label.start.rolling.maintenance',
+      dataView: true,
+      show: (record) => {
+        return record.hypervisor === 'KVM' && (record.resourcestate === 'Enabled' || record.resourcestate === 'ErrorInMaintenance')
+      },
+      args: ['timeout', 'payload', 'forced', 'hostids'],
+      mapping: {
+        hostids: {
+          value: (record) => { return record.id }
+        }
+      }
+    },
     {
       api: 'deleteHost',
       icon: 'delete',
diff --git a/ui/src/config/section/infra/pods.js b/ui/src/config/section/infra/pods.js
index 09d69c747db..25ce6f3ad54 100644
--- a/ui/src/config/section/infra/pods.js
+++ b/ui/src/config/section/infra/pods.js
@@ -75,6 +75,19 @@ export default {
         }
       }
     },
+    {
+      api: 'startRollingMaintenance',
+      icon: 'setting',
+      label: 'label.start.rolling.maintenance',
+      message: 'label.start.rolling.maintenance',
+      dataView: true,
+      args: ['timeout', 'payload', 'forced', 'podids'],
+      mapping: {
+        podids: {
+          value: (record) => { return record.id }
+        }
+      }
+    },
     {
       api: 'deletePod',
       icon: 'delete',
diff --git a/ui/src/config/section/infra/zones.js b/ui/src/config/section/infra/zones.js
index 3b769342f81..1b08f8b10ef 100644
--- a/ui/src/config/section/infra/zones.js
+++ b/ui/src/config/section/infra/zones.js
@@ -211,6 +211,19 @@ export default {
         }
       }
     },
+    {
+      api: 'startRollingMaintenance',
+      icon: 'setting',
+      label: 'label.start.rolling.maintenance',
+      message: 'label.start.rolling.maintenance',
+      dataView: true,
+      args: ['timeout', 'payload', 'forced', 'zoneids'],
+      mapping: {
+        zoneids: {
+          value: (record) => { return record.id }
+        }
+      }
+    },
     {
       api: 'deleteZone',
       icon: 'delete',
diff --git a/ui/src/locales/en.json b/ui/src/locales/en.json
index 43ac0d419d1..688836c6228 100644
--- a/ui/src/locales/en.json
+++ b/ui/src/locales/en.json
@@ -1404,6 +1404,7 @@
 "label.path": "Path",
 "label.patp": "Palo Alto Threat Profile",
 "label.pavr": "Virtual Router",
+"label.payload": "Payload",
 "label.pcidevice": "GPU",
 "label.per.account": "Per Account",
 "label.per.zone": "Per Zone",