Attribute Metatxn calldata (#244)

0xProject · Jun 4, 2020 · a756ad5 · a756ad5
1 parent 28e1214
commit a756ad5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/src/handlers/meta_transaction_handlers.ts b/src/handlers/meta_transaction_handlers.ts
@@ -193,6 +193,7 @@ export class MetaTransactionHandlers {
     }
     public async submitZeroExTransactionIfWhitelistedAsync(req: express.Request, res: express.Response): Promise<void> {
         const apiKey = req.header('0x-api-key');
+        const affiliateAddress = req.query.affiliateAddress as string | undefined;
         if (apiKey !== undefined && !isValidUUID(apiKey)) {
             res.status(HttpStatus.BAD_REQUEST).send({
                 code: GeneralErrorCodes.InvalidAPIKey,
@@ -237,6 +238,7 @@ export class MetaTransactionHandlers {
                     zeroExTransaction,
                     signature,
                     protocolFee,
+                    affiliateAddress,
                 );
                 res.status(HttpStatus.OK).send({
                     ethereumTransactionHash,

diff --git a/src/services/meta_transaction_service.ts b/src/services/meta_transaction_service.ts
@@ -307,15 +307,20 @@ export class MetaTransactionService {
         zeroExTransaction: ZeroExTransactionWithoutDomain,
         signature: string,
         protocolFee: BigNumber,
+        affiliateAddress?: string,
     ): Promise<PostTransactionResponse> {
+        const data = serviceUtils.attributeCallData(
+            this._contractWrappers.exchange
+                .executeTransaction(zeroExTransaction, signature)
+                .getABIEncodedTransactionData(),
+            affiliateAddress,
+        );
         const transactionEntity = TransactionEntity.make({
             refHash: zeroExTransactionHash,
             status: TransactionStates.Unsubmitted,
             takerAddress: zeroExTransaction.signerAddress,
             to: this._contractWrappers.exchange.address,
-            data: this._contractWrappers.exchange
-                .executeTransaction(zeroExTransaction, signature)
-                .getABIEncodedTransactionData(),
+            data,
             value: protocolFee,
             gasPrice: zeroExTransaction.gasPrice,
             expectedMinedInSec: META_TXN_RELAY_EXPECTED_MINED_SEC,