Fix deploy

Author: mgrabina

src/components/transactions/Swap/helpers/cow/simulation.helpers.ts

@@ -1,13 +1,6 @@
-import {
-  encodeAbiParameters,
-  erc20Abi,
-  encodeFunctionData,
-  keccak256,
-  toHex,
-} from 'viem';
-import { getPublicClient } from 'wagmi/actions';
-
 import { wagmiConfig } from 'src/ui-config/wagmiConfig';
+import { encodeAbiParameters, encodeFunctionData, erc20Abi, keccak256, toHex } from 'viem';
+import { getPublicClient, getWalletClient } from 'wagmi/actions';
 
 const MULTICALL3_ADDRESS = '0xcA11bde05977b3631167028862bE2a173976CA11';
 const MULTICALL3_AGGREGATE3_ABI = [
@@ -60,7 +53,21 @@ type OrderSettlementContext = {
   buyAmount?: bigint;
 };
 
-const DEFAULT_BALANCE_SLOT_CANDIDATES = [0n, 1n, 2n, 3n, 4n, 5n, 6n, 7n, 8n, 9n, 10n, 11n, 12n];
+const DEFAULT_BALANCE_SLOT_CANDIDATES = [
+  BigInt(0),
+  BigInt(1),
+  BigInt(2),
+  BigInt(3),
+  BigInt(4),
+  BigInt(5),
+  BigInt(6),
+  BigInt(7),
+  BigInt(8),
+  BigInt(9),
+  BigInt(10),
+  BigInt(11),
+  BigInt(12),
+];
 
 const getBalanceSlotKeys = (owner: string, slotCandidates = DEFAULT_BALANCE_SLOT_CANDIDATES) => {
   return slotCandidates.map((slot) => {
@@ -87,10 +94,11 @@ const buildBalanceOverride = async ({
   const normalizedRecipient = recipient as `0x${string}` | undefined;
   if (!normalizedToken || !normalizedRecipient || amount === undefined) return undefined;
 
+  const walletClient = await getWalletClient(wagmiConfig, { chainId });
   const publicClient = getPublicClient(wagmiConfig, { chainId });
-  if (!publicClient) return undefined;
+  if (!walletClient || !publicClient) return undefined;
 
-  let currentBalance = 0n;
+  let currentBalance = BigInt(0);
   try {
     currentBalance = await publicClient.readContract({
       address: normalizedToken,
@@ -113,7 +121,9 @@ const buildBalanceOverride = async ({
   return { [normalizedToken]: { stateDiff } };
 };
 
-const mergeOverrides = (...overrides: (Record<string, { stateDiff: Record<string, string> }> | undefined)[]) => {
+const mergeOverrides = (
+  ...overrides: (Record<string, { stateDiff: Record<string, string> }> | undefined)[]
+) => {
   const merged: Record<string, { stateDiff: Record<string, string> }> = {};
   overrides.forEach((override) => {
     if (!override) return;
@@ -136,7 +146,7 @@ export const simulateCollateralSwapPreHook = async ({
   settlementContext,
 }: {
   chainId: number;
-  from?: `0x${string}`;
+  from?: string;
   preHook?: HookDefinition;
   flashloan?: FlashloanMetadata;
   postHook?: HookDefinition;
@@ -149,8 +159,9 @@ export const simulateCollateralSwapPreHook = async ({
     return false;
   }
 
+  const walletClient = await getWalletClient(wagmiConfig, { chainId });
   const publicClient = getPublicClient(wagmiConfig, { chainId });
-  if (!publicClient) {
+  if (!walletClient || !publicClient) {
     console.warn('[CoW][CollateralSwap] No public client available for simulation');
     return;
   }
@@ -186,7 +197,10 @@ export const simulateCollateralSwapPreHook = async ({
         }
       : undefined;
 
-  const callsSequence = [{ ...encodedPreHook, label: 'preHook' }, ...(encodedPostHook ? [{ ...encodedPostHook, label: 'postHook' }] : [])];
+  const callsSequence = [
+    { ...encodedPreHook, label: 'preHook' },
+    ...(encodedPostHook ? [{ ...encodedPostHook, label: 'postHook' }] : []),
+  ];
 
   const aggregateData = encodeFunctionData({
     abi: MULTICALL3_AGGREGATE3_ABI,
@@ -195,15 +209,17 @@ export const simulateCollateralSwapPreHook = async ({
       callsSequence.map((c) => ({
         target: c.to as `0x${string}`,
         allowFailure: true,
-        callData: (c as any).input ?? c.data,
+        callData: c.input,
       })),
     ],
   });
 
   const stateOverrides = mergeOverrides(flashloanOverride, settlementOverride);
 
   const blockStateCall = {
-    calls: [{ from, to: MULTICALL3_ADDRESS as `0x${string}`, input: aggregateData, label: 'multicall' }],
+    calls: [
+      { from, to: MULTICALL3_ADDRESS as `0x${string}`, input: aggregateData, label: 'multicall' },
+    ],
     transactions: [{ from, to: MULTICALL3_ADDRESS as `0x${string}`, data: aggregateData }],
     ...(stateOverrides ? { stateOverrides } : {}),
   };
@@ -216,9 +232,9 @@ export const simulateCollateralSwapPreHook = async ({
   console.log('[CoW][CollateralSwap] PreHook simulation payload', simulationPayload);
 
   try {
-    const result = await publicClient.request({
+    const result = await publicClient.transport.request({
       method: 'eth_simulateV1',
-      params: [simulationPayload] as unknown[],
+      params: [simulationPayload],
     });
     console.log('[CoW][CollateralSwap] PreHook simulation result', result);
     return true;