Pass refetch funtion up the context stack

coder · Emyrk · Jun 1, 2023 · Jun 1, 2023 · Jun 1, 2023 · Jun 1, 2023
commit d52d179898c15b03a7fe346867e87b326fea3137
diff --git a/site/src/components/AppLink/AppLink.stories.tsx b/site/src/components/AppLink/AppLink.stories.tsx
@@ -29,6 +29,9 @@ const Template: Story<AppLinkProps> = (args) => (
       clearProxy: () => {
         return
       },
+      refetchProxyLatencies: () => {
+        return
+      },
     }}
   >
     <AppLink {...args} />

diff --git a/site/src/components/Navbar/NavbarView.test.tsx b/site/src/components/Navbar/NavbarView.test.tsx
@@ -19,6 +19,7 @@ const proxyContextValue: ProxyContextValue = {
   isFetched: true,
   setProxy: jest.fn(),
   clearProxy: action("clearProxy"),
+  refetchProxyLatencies: jest.fn(),
   proxyLatencies: {},
 }
 

diff --git a/site/src/components/Resources/AgentRow.stories.tsx b/site/src/components/Resources/AgentRow.stories.tsx
@@ -68,6 +68,9 @@ const TemplateFC = (
         clearProxy: () => {
           return
         },
+        refetchProxyLatencies: () => {
+          return
+        },
       }}
     >
       <AgentRow {...args} />

diff --git a/site/src/components/Resources/ResourceCard.stories.tsx b/site/src/components/Resources/ResourceCard.stories.tsx
@@ -33,6 +33,9 @@ Example.args = {
         clearProxy: () => {
           return
         },
+        refetchProxyLatencies: () => {
+          return
+        },
       }}
     >
       <AgentRow
@@ -103,6 +106,9 @@ BunchOfMetadata.args = {
         clearProxy: () => {
           return
         },
+        refetchProxyLatencies: () => {
+          return
+        },
       }}
     >
       <AgentRow

diff --git a/site/src/components/Workspace/Workspace.stories.tsx b/site/src/components/Workspace/Workspace.stories.tsx
@@ -53,6 +53,9 @@ const Template: Story<WorkspaceProps> = (args) => (
         setProxy: () => {
           return
         },
+        refetchProxyLatencies: () => {
+          return
+        },
       }}
     >
       <Workspace {...args} />

diff --git a/site/src/contexts/ProxyContext.tsx b/site/src/contexts/ProxyContext.tsx
@@ -45,6 +45,9 @@ export interface ProxyContextValue {
   // then the latency has not been fetched yet. Calculations happen async for each proxy in the list.
   // Refer to the returned report for a given proxy for more information.
   proxyLatencies: Record<string, ProxyLatencyReport>
+  // refetchProxyLatencies will trigger refreshing of the proxy latencies. By default the latencies
+  // are loaded once.
+  refetchProxyLatencies: () => void
   // setProxy is a function that sets the user's selected proxy. This function should
   // only be called if the user is manually selecting a proxy. This value is stored in local
   // storage and will persist across reloads and tabs.
@@ -101,7 +104,8 @@ export const ProxyProvider: FC<PropsWithChildren> = ({ children }) => {
 
   // Every time we get a new proxiesResponse, update the latency check
   // to each workspace proxy.
-  const { proxyLatencies } = useProxyLatency(proxiesResp)
+  const { proxyLatencies, refetch: refetchProxyLatencies } =
+    useProxyLatency(proxiesResp)
 
   // updateProxy is a helper function that when called will
   // update the proxy being used.
@@ -128,6 +132,7 @@ export const ProxyProvider: FC<PropsWithChildren> = ({ children }) => {
     <ProxyContext.Provider
       value={{
         proxyLatencies,
+        refetchProxyLatencies,
         userProxy: userSavedProxy,
         proxy: experimentEnabled
           ? proxy

diff --git a/site/src/pages/TerminalPage/TerminalPage.test.tsx b/site/src/pages/TerminalPage/TerminalPage.test.tsx
@@ -55,6 +55,7 @@ const renderTerminal = () => {
               isLoading: false,
               setProxy: jest.fn(),
               clearProxy: jest.fn(),
+              refetchProxyLatencies: jest.fn(),
             }}
           >
             <TerminalPage renderer="dom" />
-Original file line number
+Diff line change
@@ Expand Up / @@ -29,6 +29,9 @@ const Template: Story<AppLinkProps> = (args) => ( @@
           clearProxy: () => {
             return
           },
+          refetchProxyLatencies: () => {
+            return
+          },
         }}
       >
         <AppLink {...args} />
@@ Expand Down @@