From f3a174b29c47d7222da44b940a892a532cb33571 Mon Sep 17 00:00:00 2001
From: Rohit Yadav <rohit.yadav@shapeblue.com>
Date: Mon, 10 Feb 2020 15:21:49 +0530
Subject: [PATCH] infra: add support for ILBVM migration

Signed-off-by: Rohit Yadav <rohit.yadav@shapeblue.com>
---
 ui/src/components/widgets/Console.vue |  2 +-
 ui/src/config/section/infra/ilbvms.js | 13 +++++++++++++
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/ui/src/components/widgets/Console.vue b/ui/src/components/widgets/Console.vue
index daed176e533..7f269966c52 100644
--- a/ui/src/components/widgets/Console.vue
+++ b/ui/src/components/widgets/Console.vue
@@ -17,7 +17,7 @@
 
 <template>
   <a
-    v-if="['vm', 'systemvm', 'router'].includes($route.meta.name)"
+    v-if="['vm', 'systemvm', 'router', 'ilbvm'].includes($route.meta.name)"
     :href="'/client/console?cmd=access&vm=' + resource.id"
     target="_blank">
     <a-button style="margin-right: 5px" shape="circle" type="dashed" :size="size" :disabled="['Stopped', 'Error', 'Destroyed'].includes(resource.state)" >
diff --git a/ui/src/config/section/infra/ilbvms.js b/ui/src/config/section/infra/ilbvms.js
index c412ac610be..df96dba1d11 100644
--- a/ui/src/config/section/infra/ilbvms.js
+++ b/ui/src/config/section/infra/ilbvms.js
@@ -37,6 +37,19 @@ export default {
       dataView: true,
       args: ['forced'],
       show: (record) => { return record.state === 'Running' }
+    },
+    {
+      api: 'migrateSystemVm',
+      icon: 'drag',
+      label: 'label.action.migrate.router',
+      dataView: true,
+      show: (record) => { return record.state === 'Running' },
+      args: ['virtualmachineid', 'hostid'],
+      mapping: {
+        virtualmachineid: {
+          value: (record) => { return record.id }
+        }
+      }
     }
   ]
 }