From 802b4383086b67b01088d3e834795d03de4ff8ee Mon Sep 17 00:00:00 2001
From: Alain Nicolas <alain.nicolas@consensys.net>
Date: Tue, 10 Dec 2024 21:29:39 +0100
Subject: [PATCH] fix(sdk): AttestV2 doesn't support customABI

---
 sdk/package.json                       |  2 +-
 sdk/src/dataMapper/PortalDataMapper.ts | 16 +++++++++++++---
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/sdk/package.json b/sdk/package.json
index 082cb3d8..3eced1a9 100644
--- a/sdk/package.json
+++ b/sdk/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@verax-attestation-registry/verax-sdk",
-  "version": "2.2.1",
+  "version": "2.2.2",
   "description": "Verax Attestation Registry SDK to interact with the subgraph and the contracts",
   "keywords": [
     "linea-attestation-registry",
diff --git a/sdk/src/dataMapper/PortalDataMapper.ts b/sdk/src/dataMapper/PortalDataMapper.ts
index 252c1715..d305bc23 100644
--- a/sdk/src/dataMapper/PortalDataMapper.ts
+++ b/sdk/src/dataMapper/PortalDataMapper.ts
@@ -2,13 +2,12 @@ import { AttestationPayload, Portal } from "../types";
 import { ActionType } from "../utils/constants";
 import BaseDataMapper from "./BaseDataMapper";
 import { abiDefaultPortal } from "../abi/DefaultPortal";
-import { Address } from "viem";
+import { Abi, Address } from "viem";
 import { encode } from "../utils/abiCoder";
 import { Portal_filter, Portal_orderBy } from "../../.graphclient";
 import { abiPortalRegistry } from "../abi/PortalRegistry";
 import { handleError } from "../utils/errorHandler";
 import { executeTransaction } from "../utils/transactionSender";
-import { Abi } from "viem";
 
 export default class PortalDataMapper extends BaseDataMapper<Portal, Portal_filter, Portal_orderBy> {
   typeName = "portal";
@@ -95,6 +94,7 @@ export default class PortalDataMapper extends BaseDataMapper<Portal, Portal_filt
     attestationPayload: AttestationPayload,
     validationPayloads: string[],
     value: bigint = 0n,
+    customAbi?: Abi,
   ) {
     const matchingSchema = await this.veraxSdk.schema.findOneById(attestationPayload.schemaId);
     if (!matchingSchema) {
@@ -109,6 +109,7 @@ export default class PortalDataMapper extends BaseDataMapper<Portal, Portal_filt
         validationPayloads,
       ],
       value,
+      customAbi,
     );
   }
 
@@ -118,8 +119,15 @@ export default class PortalDataMapper extends BaseDataMapper<Portal, Portal_filt
     validationPayloads: string[],
     waitForConfirmation: boolean = false,
     value: bigint = 0n,
+    customAbi?: Abi,
   ) {
-    const request = await this.simulateAttestV2(portalAddress, attestationPayload, validationPayloads, value);
+    const request = await this.simulateAttestV2(
+      portalAddress,
+      attestationPayload,
+      validationPayloads,
+      value,
+      customAbi,
+    );
     return executeTransaction(request, this.web3Client, this.walletClient, waitForConfirmation);
   }
 
@@ -327,6 +335,7 @@ export default class PortalDataMapper extends BaseDataMapper<Portal, Portal_filt
   async isPortalRegistered(id: Address) {
     return this.executePortalRegistryReadMethod("isRegistered", [id]);
   }
+
   private async executePortalRegistryReadMethod(functionName: string, args: unknown[]) {
     return this.web3Client.readContract({
       abi: abiPortalRegistry,
@@ -352,6 +361,7 @@ export default class PortalDataMapper extends BaseDataMapper<Portal, Portal_filt
       handleError(ActionType.Simulation, err);
     }
   }
+
   private async simulatePortalContract(
     portalAddress: Address,
     functionName: string,