feat(web): refetch discovery after bridge url update (#6005)

novuhq · Jul 7, 2024 · 688eee3 · 688eee3
1 parent bc0130e
commit 688eee3
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/apps/api/src/app/environments/usecases/update-environment/update-environment.usecase.ts b/apps/api/src/app/environments/usecases/update-environment/update-environment.usecase.ts
@@ -24,8 +24,9 @@ export class UpdateEnvironment {
       updatePayload[`dns.inboundParseDomain`] = command.dns.inboundParseDomain;
     }
 
-    if ((await this.shouldUpdateBridgeConfiguration(command)) && command.bridge) {
+    if ((await this.shouldUpdateBridgeConfiguration()) && command.bridge) {
       updatePayload['echo.url'] = command.bridge?.url || '';
+      updatePayload['bridge.url'] = command.bridge?.url || '';
     }
 
     return await this.environmentRepository.update(
@@ -36,7 +37,7 @@ export class UpdateEnvironment {
       { $set: updatePayload }
     );
   }
-  async shouldUpdateBridgeConfiguration(command: UpdateEnvironmentCommand): Promise<boolean> {
+  async shouldUpdateBridgeConfiguration(): Promise<boolean> {
     return process.env.NOVU_ENTERPRISE === 'true' || process.env.CI_EE_TEST === 'true';
   }
 }
diff --git a/apps/web/src/studio/hooks/useBridgeAPI.ts b/apps/web/src/studio/hooks/useBridgeAPI.ts
@@ -1,4 +1,4 @@
-import { useMemo } from 'react';
+import { useEffect, useMemo } from 'react';
 import { useMutation, useQuery } from '@tanstack/react-query';
 import {
   buildBridgeHTTPClient,
@@ -18,9 +18,10 @@ function useBridgeAPI() {
 const BRIDGE_STATUS_REFRESH_INTERVAL_IN_MS = 5 * 1000;
 
 export const useDiscover = (options?: any) => {
+  const { bridgeURL, setBridgeURL } = useStudioState();
   const api = useBridgeAPI();
 
-  return useQuery(
+  const discoverQuery = useQuery(
     ['bridge-workflows'],
     async () => {
       return api.discover();
@@ -30,6 +31,12 @@ export const useDiscover = (options?: any) => {
       ...(options || {}),
     }
   );
+
+  useEffect(() => {
+    discoverQuery.refetch();
+  }, [bridgeURL, setBridgeURL, discoverQuery]);
+
+  return discoverQuery;
 };
 
 export const useHealthCheck = (options?: any) => {