feat: add tier tracking

VaporFi · Jun 10, 2024 · 606701b · 606701b
1 parent 3fb13b7
commit 606701b
Show file tree

Hide file tree

Showing 3 changed files with 54 additions and 0 deletions.
diff --git a/ponder.schema.ts b/ponder.schema.ts
@@ -145,4 +145,12 @@ export default createSchema((p) => ({
     pointsSpent: p.bigint(),
     lastUpdated: p.bigint(),
   }),
+
+  // @dev: Id is the tierId + chainId
+  // @dev: The tierId is the tier number
+  // @dev: wallets is the number of wallets in the tier
+  WalletsPerTier: p.createTable({
+    id: p.string(),
+    wallets: p.string().list(),
+  }),
 }));
diff --git a/src/helpers.ts b/src/helpers.ts
@@ -392,3 +392,37 @@ export async function getOrUpdateTokenIdData(
 
   return tokenIdData;
 }
+
+export async function getOrUpdateWalletsPerTier(
+  context: Context,
+  tierId: bigint,
+  userAddress: string
+): Promise<any> {
+  const { WalletsPerTier } = context.db;
+  const { chainId } = context.network;
+
+  const id = `${tierId}-${chainId}`;
+
+  let walletsPerTier = await WalletsPerTier.findUnique({ id });
+
+  if (!walletsPerTier) {
+    walletsPerTier = await WalletsPerTier.create({
+      id,
+      data: {
+        wallets: [userAddress],
+      },
+    });
+  }
+
+  if (!walletsPerTier.wallets.includes(userAddress)) {
+    walletsPerTier.wallets.push(userAddress);
+    walletsPerTier = await WalletsPerTier.update({
+      id,
+      data: {
+        wallets: walletsPerTier.wallets,
+      },
+    });
+  }
+
+  return walletsPerTier;
+}
diff --git a/src/index.ts b/src/index.ts
@@ -5,6 +5,7 @@ import {
   getTokenId,
   getOrUpdateTokenIdData,
   handleChainFirstWallet,
+  getOrUpdateWalletsPerTier,
 } from "./helpers";
 import {
   BIGINT_HUNDRED_THOUSAND,
@@ -43,6 +44,8 @@ ponder.on("Stratosphere:Transfer", async ({ event, context }) => {
   await getOrUpdateTokenIdData(context, tokenId, timestamp, {
     pointsEarned: pointsMap.Enrollment,
   });
+
+  await getOrUpdateWalletsPerTier(context, 0n, userAddressLowerCase);
 });
 
 ponder.on("LiquidMining:Deposit", async ({ event, context }) => {
@@ -493,4 +496,13 @@ ponder.on("RewardsController:ClaimPoints", async ({ event, context }) => {
   await getOrUpdateTokenIdData(context, tokenId, timestamp, {
     pointsClaimed: points,
   });
+
+  const tier = await context.client.readContract({
+    abi: context.contracts.RewardsController.abi,
+    address: context.contracts.RewardsController.address as `0x${string}`,
+    functionName: "tierOf",
+    args: [tokenId],
+  });
+
+  await getOrUpdateWalletsPerTier(context, tier, event.args.member);
 });