feat(Dataworker): Support pre-fill refunds and duplicate deposits

package.json

@@ -12,7 +12,7 @@
   "dependencies": {
     "@across-protocol/constants": "^3.1.30",
     "@across-protocol/contracts": "^3.0.25",
-    "@across-protocol/sdk": "^3.4.20",
+    "@across-protocol/sdk": "^4.0.0",
     "@arbitrum/sdk": "^4.0.2",
     "@consensys/linea-sdk": "^0.2.1",
     "@defi-wonderland/smock": "^2.3.5",

scripts/withdrawFromOpStack.ts

@@ -97,8 +97,9 @@ export async function run(): Promise<void> {
           "0x", // _data
         ];
 
+  const functionNameToCall = l1TokenInfo.symbol === "ETH" ? "bridgeETHTo" : "bridgeERC20To";
   console.log(
-    `Submitting bridgeETHTo on the OVM standard bridge @ ${ovmStandardBridge.address} with the following args: `,
+    `Submitting ${functionNameToCall} on the OVM standard bridge @ ${ovmStandardBridge.address} with the following args: `,
     ...bridgeArgs
   );
 
@@ -122,9 +123,7 @@ export async function run(): Promise<void> {
   if (!(await askYesNoQuestion("\nDo you want to proceed?"))) {
     return;
   }
-  const withdrawal = await ovmStandardBridge[l1TokenInfo.symbol === "ETH" ? "bridgeETHTo" : "bridgeERC20To"](
-    ...bridgeArgs
-  );
+  const withdrawal = await ovmStandardBridge[functionNameToCall](...bridgeArgs);
   console.log(`Submitted withdrawal: ${blockExplorerLink(withdrawal.hash, chainId)}.`);
   const receipt = await withdrawal.wait();
   console.log("Receipt", receipt);

test/mocks/MockBundleDataClient.ts

@@ -1,9 +1,11 @@
-import { BundleDataClient } from "../../src/clients";
+import { BundleDataClient, SpokePoolClient } from "../../src/clients";
 import { CombinedRefunds } from "../../src/dataworker/DataworkerUtils";
+import { DepositWithBlock, FillWithBlock } from "../../src/interfaces";
 
 export class MockBundleDataClient extends BundleDataClient {
   private pendingBundleRefunds: CombinedRefunds = {};
   private nextBundleRefunds: CombinedRefunds = {};
+  private matchingFillEvents: Record<string, FillWithBlock> = {};
 
   async getPendingRefundsFromValidBundles(): Promise<CombinedRefunds[]> {
     return [this.pendingBundleRefunds];
@@ -28,4 +30,19 @@ export class MockBundleDataClient extends BundleDataClient {
   getPersistedNextBundleRefunds(): Promise<CombinedRefunds | undefined> {
     return Promise.resolve(undefined);
   }
+
+  setMatchingFillEvent(deposit: DepositWithBlock, fill: FillWithBlock): void {
+    const relayDataHash = this.getRelayHashFromEvent(deposit);
+    this.matchingFillEvents[relayDataHash] = fill;
+  }
+
+  findMatchingFillEvent(
+    deposit: DepositWithBlock,
+    spokePoolClient: SpokePoolClient
+  ): Promise<FillWithBlock | undefined> {
+    const relayDataHash = this.getRelayHashFromEvent(deposit);
+    return this.matchingFillEvents[relayDataHash]
+      ? Promise.resolve(this.matchingFillEvents[relayDataHash])
+      : super.findMatchingFillEvent(deposit, spokePoolClient);
+  }
 }

test/mocks/MockSpokePoolClient.ts

@@ -1,7 +1,9 @@
-import { clients } from "@across-protocol/sdk";
+import { clients, interfaces } from "@across-protocol/sdk";
+import { Deposit } from "../../src/interfaces";
 export class MockSpokePoolClient extends clients.mocks.MockSpokePoolClient {
   public maxFillDeadlineOverride?: number;
   public oldestBlockTimestampOverride?: number;
+  private relayFillStatuses: Record<string, interfaces.FillStatus> = {};
 
   public setMaxFillDeadlineOverride(maxFillDeadlineOverride?: number): void {
     this.maxFillDeadlineOverride = maxFillDeadlineOverride;
@@ -18,4 +20,19 @@ export class MockSpokePoolClient extends clients.mocks.MockSpokePoolClient {
   public getOldestTime(): number {
     return this.oldestBlockTimestampOverride ?? super.getOldestTime();
   }
+
+  public setRelayFillStatus(deposit: Deposit, fillStatus: interfaces.FillStatus): void {
+    const relayDataHash = deposit.depositId.toString();
+    this.relayFillStatuses[relayDataHash] = fillStatus;
+  }
+  public relayFillStatus(
+    relayData: interfaces.RelayData,
+    blockTag?: number | "latest",
+    destinationChainId?: number
+  ): Promise<interfaces.FillStatus> {
+    const relayDataHash = relayData.depositId.toString();
+    return this.relayFillStatuses[relayDataHash]
+      ? Promise.resolve(this.relayFillStatuses[relayDataHash])
+      : super.relayFillStatus(relayData, blockTag, destinationChainId);
+  }
 }

yarn.lock

@@ -53,10 +53,10 @@
     yargs "^17.7.2"
     zksync-web3 "^0.14.3"
 
-"@across-protocol/sdk@^3.4.20":
-  version "3.4.20"
-  resolved "https://registry.yarnpkg.com/@across-protocol/sdk/-/sdk-3.4.20.tgz#0fa2b223e264fc8ff9ea47ab9e65ad9cd176848d"
-  integrity sha512-HUgWYfbH0haa2r9nl892IE6U+z+QtowcYwHzcimiBlux2+tn6Ztq80CEOpp1367GB4gBiWz1stL5TQfRi+vxtg==
+"@across-protocol/sdk@^4.0.0":
+  version "4.0.0"
+  resolved "https://registry.yarnpkg.com/@across-protocol/sdk/-/sdk-4.0.0.tgz#83242907471577a8fe670dcee3d633b2338d15b9"
+  integrity sha512-qDkOYHlQy8KhT5WStRXJybp84pFuxQc4MgMwZzOOBTvvrGWtklSpWY9NT8Uu3Rd9v9Yn+oa/MRURdFmeMsJnrg==
   dependencies:
     "@across-protocol/across-token" "^1.0.0"
     "@across-protocol/constants" "^3.1.30"