17 feature extract referral and link on first order

schema.graphql

@@ -216,6 +216,7 @@ type PropertyTokenTrade {
   makerTokenFilledAmount: BigInt!
   propertyValuation: BigInt!
   protocol: LimitOrderProtocol!
+  referralCode: String
 }
 
 type PropertyTokenTradeCounts {
@@ -269,6 +270,7 @@ type User {
   chainId: Int!
   wallets: [Wallet!]! @derivedFrom(field: "user")
   certifiedPartners: [UserCertifiedPartner!]! @derivedFrom(field: "user")
+  referrals: [Referral!]! @derivedFrom(field: "referrer")
 }
 
 type UserCertifiedPartner {
@@ -530,3 +532,12 @@ type TokenRecord {
   marketcapBI: BigInt!
   token: Token!
 }
+
+type Referral @entity {
+  id: ID! # chainId-cpAddress-buyerWallet
+  wallet: Wallet!
+  referrer: User!
+  marketplace: CertifiedPartner!
+  firstOrderTrade: PropertyTokenTrade!
+  createdAt: Int!
+}

src/handler/OneInchPostInteraction.ts

@@ -23,22 +23,59 @@ OneInchPostInteraction.PostInteractionOrderFilled.handler(async ({ event, contex
     return;
   }
 
+  const tradeId = `${event.chainId}-${propertyToken.contractAddress}-${event.transaction.hash}-${event.logIndex}`;
+  const buyerWalletId = `${event.chainId}-${event.params.taker}`;
+  const marketplaceId = propertyToken.certifiedPartner_id;
+
+  const extraData = event.params.extraData;
+  const hasValidExtraData = extraData && extraData.length === 66;
+
+  let validReferralCode = '';
+
+  if (hasValidExtraData) {
+    const userBytes = extraData.slice(2);
+    const referrerUserId = `${event.chainId}-${userBytes}`;
+    const referralId = `${event.chainId}-${marketplaceId}-${event.params.taker}`;
+
+    const [referrerUser, buyerWalletEntity, existingReferral] = await Promise.all([
+      context.User.get(referrerUserId),
+      context.Wallet.get(buyerWalletId),
+      context.Referral.get(referralId)
+    ]);
+
+    if (referrerUser && buyerWalletEntity?.user_id !== referrerUserId) {
+      validReferralCode = extraData;
+
+      if (!existingReferral) {
+        context.Referral.set({
+          id: referralId,
+          wallet_id: buyerWalletId,
+          referrer_id: referrerUserId,
+          marketplace_id: marketplaceId,
+          firstOrderTrade_id: tradeId,
+          createdAt: event.block.timestamp
+        });
+      }
+    }
+  }
+
   context.PropertyTokenTrade.set({
-    id: `${event.chainId}-${propertyToken.contractAddress}-${event.transaction.hash}-${event.logIndex}`,
+    id: tradeId,
     chainID: event.chainId,
     transactionHash: event.transaction.hash,
     blockNumber: event.block.number,
     blockTimestamp: event.block.timestamp,
     propertyToken_id: propertyToken.id,
     maker_id: `${event.chainId}-${maker}`,
-    taker_id: `${event.chainId}-${event.params.taker}`,
+    taker_id: buyerWalletId,
     orderHash: event.params.orderHash,
     makerToken: makerToken,
     takerToken: takerToken,
     takerTokenFilledAmount: event.params.takingAmount,
     makerTokenFilledAmount: event.params.makingAmount,
     propertyValuation: propertyToken.propertyValuation,
     protocol: LimitOrderProtocol.OneInch,
+    referralCode: validReferralCode
   });
 
   // Handle case where property token is on maker side (being sold)
@@ -79,5 +116,6 @@ OneInchPostInteraction.PostInteractionOrderFilled.handler(async ({ event, contex
     event.params.taker,
     'takerCount'
   );
+
 });
 

src/handler/ZeroEx.ts

@@ -13,7 +13,6 @@ ZeroEx.LimitOrderFilled.handler(async ({ event, context }) => {
   const propertyToken = makerPropertyToken || takerPropertyToken;
   if (!propertyToken) return;
 
-  // Create PropertyTokenTrade record
   context.PropertyTokenTrade.set({
     id: `${event.chainId}-${propertyToken.contractAddress}-${event.transaction.hash}-${event.logIndex}`,
     chainID: event.chainId,
@@ -30,6 +29,7 @@ ZeroEx.LimitOrderFilled.handler(async ({ event, context }) => {
     makerTokenFilledAmount: event.params.makerTokenFilledAmount,
     propertyValuation: propertyToken.propertyValuation,
     protocol: LimitOrderProtocol.ZeroEx,
+    referralCode: '',
   });
 
   // Handle case where property token is on maker side (being sold)