Refactor legacy method getVmPowerState

Reference-Url: https://github.com/kubev2v/forklift-console-plugin/blob/b98d36babe11c14ca53cc0cc5437968fd2b9d4bd/packages/legacy/src/common/components/VMNameWithPowerState.tsx#L24 Signed-off-by: Radoslaw Szwajkowski <[email protected]>
kubev2v · Sep 25, 2023 · 9255bc3 · 9255bc3
1 parent 5a067c1
commit 9255bc3
Show file tree

Hide file tree

Showing 3 changed files with 58 additions and 43 deletions.
diff --git a/...src/modules/Providers/views/details/tabs/VirtualMachines/OpenShiftVirtualMachinesList.tsx b/...src/modules/Providers/views/details/tabs/VirtualMachines/OpenShiftVirtualMachinesList.tsx
@@ -2,8 +2,7 @@ import React from 'react';
 
 import { EnumToTuple, ResourceFieldFactory } from '@kubev2v/common';
 
-import { ProviderVirtualMachinesList, VmData } from './components/ProviderVirtualMachinesList';
-import { getVmPowerState } from './utils/helpers/getVmPowerState';
+import { ProviderVirtualMachinesList } from './components/ProviderVirtualMachinesList';
 import { OpenShiftVirtualMachinesRow } from './OpenShiftVirtualMachinesRow';
 import { ProviderVirtualMachinesProps } from './ProviderVirtualMachines';
 

diff --git a/.../src/modules/Providers/views/details/tabs/VirtualMachines/OpenShiftVirtualMachinesRow.tsx b/.../src/modules/Providers/views/details/tabs/VirtualMachines/OpenShiftVirtualMachinesRow.tsx
@@ -3,8 +3,8 @@ import React from 'react';
 import { ResourceField, RowProps } from '@kubev2v/common';
 import { Td, Tr } from '@patternfly/react-table';
 
-import { VMCellProps, VMConcernsCellRenderer, VmData, VMNameCellRenderer } from './components';
 import { PowerStateCellRenderer } from './components/PowerStateCellRenderer';
+import { VMCellProps, VMConcernsCellRenderer, VmData, VMNameCellRenderer } from './components';
 
 const cellRenderers: Record<string, React.FC<VMCellProps>> = {
   name: VMNameCellRenderer,

diff --git a/...src/modules/Providers/views/details/tabs/VirtualMachines/utils/helpers/getVmPowerState.ts b/...src/modules/Providers/views/details/tabs/VirtualMachines/utils/helpers/getVmPowerState.ts
@@ -8,47 +8,63 @@ import {
 
 export type PowerState = 'on' | 'off' | 'unknown';
 
-// moved from packages/legacy/src/common/components/VMNameWithPowerState.tsx
 export const getVmPowerState = (vm?: ProviderVirtualMachine): PowerState => {
-  let powerStatus: PowerState = 'unknown';
-  if (!vm) return powerStatus;
+  if (!vm) return 'unknown';
+
   switch (vm?.providerType) {
-    case 'ovirt': {
-      if ((vm as OVirtVM).status === 'up') powerStatus = 'on';
-      if ((vm as OVirtVM).status === 'down') powerStatus = 'off';
-      break;
-    }
-    case 'vsphere': {
-      if ((vm as VSphereVM).powerState === 'poweredOn') powerStatus = 'on';
-      if ((vm as VSphereVM).powerState === 'poweredOff') powerStatus = 'off';
-      break;
-    }
-    case 'openstack': {
-      const status = (vm as OpenstackVM).status;
-      if (status === 'ACTIVE') {
-        powerStatus = 'on';
-      }
-      if (status === 'SHUTOFF') {
-        powerStatus = 'off';
-      }
-      break;
-    }
-    case 'openshift': {
-      const status = (vm as OpenshiftVM)?.object?.status?.printableStatus;
-      if (status === 'Running') {
-        powerStatus = 'on';
-      } else {
-        powerStatus = 'off';
-      }
-      break;
-    }
-    case 'ova': {
-      powerStatus = 'off';
-      break;
-    }
-    default: {
-      powerStatus = 'unknown';
-    }
+    case 'ovirt':
+      return getOVirtVmPowerState(vm as OVirtVM);
+    case 'vsphere':
+      return getVSphereVmPowerState(vm as VSphereVM);
+    case 'openstack':
+      return getOpenStackVmPowerState(vm as OpenstackVM);
+    case 'openshift':
+      return getOpenShiftVmPowerState(vm as OpenshiftVM);
+    case 'ova':
+      return 'off';
+    default:
+      return 'unknown';
+  }
+};
+
+const getOVirtVmPowerState = (vm: OVirtVM): PowerState => {
+  switch (vm?.status) {
+    case 'up':
+      return 'on';
+    case 'down':
+      return 'off';
+    default:
+      return 'unknown';
+  }
+};
+
+const getVSphereVmPowerState = (vm: VSphereVM): PowerState => {
+  switch (vm?.powerState) {
+    case 'poweredOn':
+      return 'on';
+    case 'poweredOff':
+      return 'off';
+    default:
+      return 'unknown';
+  }
+};
+
+const getOpenStackVmPowerState = (vm: OpenstackVM): PowerState => {
+  switch (vm?.status) {
+    case 'ACTIVE':
+      return 'on';
+    case 'SHUTOFF':
+      return 'off';
+    default:
+      return 'unknown';
+  }
+};
+
+const getOpenShiftVmPowerState = (vm: OpenshiftVM): PowerState => {
+  switch (vm?.object?.status?.printableStatus) {
+    case 'Running':
+      return 'on';
+    default:
+      return 'off';
   }
-  return powerStatus;
 };