fix: insurance card isn't shown in some cases

lifinance · May 15, 2023 · a8eb1ef · a8eb1ef
1 parent d31f7e4
commit a8eb1ef
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 15 deletions.
diff --git a/packages/widget/src/components/Insurance/InsuranceCollapsed.tsx b/packages/widget/src/components/Insurance/InsuranceCollapsed.tsx
@@ -1,6 +1,4 @@
-import type { Route } from '@lifi/sdk';
 import { Collapse } from '@mui/material';
-import { useState } from 'react';
 import { useSwapRoutes } from '../../hooks';
 import {
   RouteExecutionStatus,
@@ -18,20 +16,16 @@ export const InsuranceCollapsed: React.FC<InsuranceProps> = ({
   onChange,
   ...props
 }) => {
-  const [insuredRoute, setInsuredRoute] = useState<Route>();
   const setExecutableRoute = useSetExecutableRoute();
   const routeExecution = useRouteExecutionStore(
     (state) => state.routes[insurableRouteId],
   );
-  useSwapRoutes({
+  const { routes } = useSwapRoutes({
     insurableRoute: routeExecution?.route,
-    onSettled(data) {
-      if (data?.routes?.[0]) {
-        setInsuredRoute(data.routes[0]);
-      }
-    },
   });
 
+  const insuredRoute = routes?.[0];
+
   const toggleInsurance = (checked: boolean) => {
     if (insuredRoute) {
       if (checked) {

diff --git a/packages/widget/src/hooks/useSwapRoutes.ts b/packages/widget/src/hooks/useSwapRoutes.ts
@@ -13,14 +13,10 @@ import { useSwapOnly } from './useSwapOnly';
 const refetchTime = 60_000;
 
 interface SwapRoutesProps {
-  onSettled?: (data?: RoutesResponse) => void;
   insurableRoute?: Route;
 }
 
-export const useSwapRoutes = ({
-  onSettled,
-  insurableRoute,
-}: SwapRoutesProps = {}) => {
+export const useSwapRoutes = ({ insurableRoute }: SwapRoutesProps = {}) => {
   const lifi = useLiFi();
   const { variant, sdkConfig, insurance, contractTool } = useWidgetConfig();
   const { account } = useWallet();
@@ -300,7 +296,6 @@ export const useSwapRoutes = ({
             });
           }
         },
-        onSettled,
       },
     );