@ensofinance/checkout-widget 0.0.16 → 0.0.18

package/src/enso-api/api.ts

@@ -1,6 +1,5 @@
 import { useAccount, useConfig } from "wagmi";
 import {
-    useRouterControllerPostRouteShortcutTransaction,
     useRouterControllerRouteShortcutTransaction,
     useTokensControllerTokens,
     useWalletControllerCreateApproveTransaction,
@@ -15,11 +14,9 @@ import { Token } from "@/util/common";
 import { useAppStore } from "@/store";
 import { Address } from "viem";
 import { formatNumber, normalizeValue } from "@/util";
-import { VITALIK_ADDRESS } from "@/util/constants";
 import {
     RouterControllerRouteShortcutTransactionParams,
     NontokenizedControllerRouteNontokenizedShorcutTransactionParams,
-    NonTokenizedModel,
     ProtocolModel,
 } from "./model";
 
@@ -284,54 +281,34 @@ export const useProtocolData = (protocolSlug?: string) => {
     return { protocolData, isLoading };
 };
 
-export const useNontokenizedTokens = () => {
-    const config = useConfig();
-
-    const supportedChainIds = useMemo(
-        () => config.chains.map((chain) => chain.id),
-        [config.chains],
-    );
-
+export const useIsNontokenized = (tokenAddress?: string, chainId?: number) => {
     const { data, isLoading } = useNonTokenizedControllerTokens(
-        {},
+        {
+            positionId: tokenAddress ? [tokenAddress] : undefined,
+            chainId,
+        },
         {
             query: {
-                enabled: true,
+                enabled: !!tokenAddress && !!chainId,
+            },
+        },
+    );
+    const { data: addressData } = useNonTokenizedControllerTokens(
+        {
+            address: tokenAddress ? [tokenAddress] : undefined,
+            chainId,
+        },
+        {
+            query: {
+                enabled: !!tokenAddress && !!chainId,
             },
         },
     );
 
-    const nontokenizedMap = useMemo(() => {
-        if (!data?.data) return new Map<string, NonTokenizedModel>();
-
-        const map = new Map<string, NonTokenizedModel>();
-        data.data.forEach((token) => {
-            if (supportedChainIds.includes(token.chainId)) {
-                const key = `${token.chainId}-${token.address.toLowerCase()}`;
-                map.set(key, token);
-            }
-        });
-
-        return map;
-    }, [data, supportedChainIds]);
-
-    console.log(nontokenizedMap);
-
-    return { nontokenizedMap, isLoading };
-};
-
-export const useIsNontokenized = (tokenAddress?: string, chainId?: number) => {
-    const { nontokenizedMap, isLoading } = useNontokenizedTokens();
-
-    const nontokenizedData = useMemo(() => {
-        if (!tokenAddress || !chainId) return null;
-
-        const key = `${chainId}-${tokenAddress.toLowerCase()}`;
-        return nontokenizedMap.get(key) || null;
-    }, [tokenAddress, chainId, nontokenizedMap]);
+    const nontokenizedData = data?.data?.[0] || addressData?.data?.[0] || null;
 
     return {
-        isNontokenized: !!nontokenizedData,
+        isNontokenized: Boolean(nontokenizedData),
         nontokenizedData,
         isLoading,
     };