fix: fix

fedimint · Sep 23, 2024 · a1b2f56 · a1b2f56
1 parent 7b0aacf
commit a1b2f56
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/apps/router/src/components/Header.tsx b/apps/router/src/components/Header.tsx
@@ -9,12 +9,8 @@ export const Header = React.memo(function Header() {
   const { guardians, gateways } = useAppContext();
   const { type, id } = useActiveService();
   const activeService = useMemo(() => {
-    if (type === 'guardian') {
-      return guardians[id];
-    } else if (type === 'gateway') {
-      return gateways[id];
-    }
-    return null;
+    const serviceMap = { guardian: guardians, gateway: gateways };
+    return serviceMap[type]?.[id] ?? null;
   }, [type, id, guardians, gateways]);
 
   const hasServices =

diff --git a/apps/router/src/hooks/index.tsx b/apps/router/src/hooks/index.tsx
@@ -1,9 +1,14 @@
 import { useLocation } from 'react-router-dom';
 
-export const useActiveService = () => {
+export const useActiveService = (): {
+  type: 'guardian' | 'gateway';
+  id: string;
+} => {
   const location = useLocation();
   const [type, id] = location.pathname.split('/').filter(Boolean);
-
+  if (type !== 'guardian' && type !== 'gateway') {
+    throw new Error('Invalid service type');
+  }
   return {
     type,
     id,