From 87fd3bdc6b9c11ced9e50db42047a83c8f11f240 Mon Sep 17 00:00:00 2001
From: Tsu-ba-me <ynho.li.aa.e@gmail.com>
Date: Fri, 25 Nov 2022 18:15:41 -0500
Subject: [PATCH] refactor(striker-ui): repair types in Network

---
 striker-ui/components/Network/Network.tsx     | 32 +++++++++++++------
 .../components/Network/processNetwork.ts      |  3 +-
 2 files changed, 23 insertions(+), 12 deletions(-)

diff --git a/striker-ui/components/Network/Network.tsx b/striker-ui/components/Network/Network.tsx
index 868e4497..0376ee48 100644
--- a/striker-ui/components/Network/Network.tsx
+++ b/striker-ui/components/Network/Network.tsx
@@ -1,17 +1,20 @@
+import { Box, Divider, styled } from '@mui/material';
 import { useContext } from 'react';
-import { Box, Divider } from '@mui/material';
-import { styled } from '@mui/material/styles';
-import { Panel } from '../Panels';
-import { HeaderText, BodyText } from '../Text';
-import periodicFetch from '../../lib/fetchers/periodicFetch';
+
 import {
   DIVIDER,
   LARGE_MOBILE_BREAKPOINT,
 } from '../../lib/consts/DEFAULT_THEME';
-import processNetworkData from './processNetwork';
+
 import { AnvilContext } from '../AnvilContext';
 import Decorator, { Colours } from '../Decorator';
+import { Panel } from '../Panels';
+import periodicFetch from '../../lib/fetchers/periodicFetch';
+import processNetworkData from './processNetwork';
 import Spinner from '../Spinner';
+import { HeaderText, BodyText } from '../Text';
+import useProtect from '../../hooks/useProtect';
+import useProtectedState from '../../hooks/useProtectedState';
 
 const PREFIX = 'Network';
 
@@ -68,26 +71,35 @@ const selectDecorator = (state: string): Colours => {
 
 const Network = (): JSX.Element => {
   const { uuid } = useContext(AnvilContext);
+  const { protect } = useProtect();
+
+  const [processed, setProcessed] = useProtectedState<
+    ProcessedNetwork | undefined
+  >(undefined, protect);
 
-  const { data, isLoading } = periodicFetch<AnvilNetwork>(
+  const { isLoading } = periodicFetch<AnvilNetwork>(
     `${process.env.NEXT_PUBLIC_API_URL}/get_networks?anvil_uuid=${uuid}`,
+    {
+      onSuccess: (data) => {
+        setProcessed(processNetworkData(data));
+      },
+    },
   );
 
-  const processed = processNetworkData(data);
   return (
     <Panel>
       <StyledDiv>
         <HeaderText text="Network" />
         {!isLoading ? (
           <Box className={classes.container}>
-            {data &&
+            {processed &&
               processed.bonds.map((bond: ProcessedBond) => (
                 <>
                   <Box
+                    className={classes.root}
                     display="flex"
                     flexDirection="row"
                     width="100%"
-                    className={classes.root}
                   >
                     <Box p={1} className={classes.noPaddingLeft}>
                       <Decorator colour={selectDecorator(bond.bond_state)} />
diff --git a/striker-ui/components/Network/processNetwork.ts b/striker-ui/components/Network/processNetwork.ts
index 0a321312..c08b11af 100644
--- a/striker-ui/components/Network/processNetwork.ts
+++ b/striker-ui/components/Network/processNetwork.ts
@@ -2,7 +2,7 @@ const processNetworkData = (data: AnvilNetwork): ProcessedNetwork => {
   const processedBonds: string[] = [];
   const displayBonds: ProcessedNetwork = { bonds: [] };
 
-  data?.hosts.forEach((host) => {
+  data.hosts.forEach((host) => {
     host.bonds.forEach((bond) => {
       const index = processedBonds.findIndex(
         (processed: string) => processed === bond.bond_name,
@@ -33,7 +33,6 @@ const processNetworkData = (data: AnvilNetwork): ProcessedNetwork => {
     });
   });
 
-  /* eslint-disable no-param-reassign */
   displayBonds.bonds.forEach((bond) => {
     const hostIndex =
       bond.hosts[0].link.link_speed > bond.hosts[1].link.link_speed ? 1 : 0;