Adds New Kgen fee contracts (#4562)

* Adds Kgen native token revenue event

* addd new kgen revenue contracts

* fix issue

* Revert "package-lock.json"

---------

Co-authored-by: YogeshIndigg <[email protected]>

Author: Akhileshkgen

fees/kgen.ts

@@ -15,32 +15,62 @@ interface DepositFungible {
   type: string;
 }
 
-const config: Record<string, string> = {
-  [CHAIN.APTOS]: "0x5a96fab415f43721a44c5a761ecfcccc3dae9c21f34313f0e594b49d8d4564f4",
-  [CHAIN.POLYGON]: "0x9Df4C994d8d8c440d87da8BA94D355BB85706f51",
-}
+const aptosConfig = {
+  contracts: [
+    "0x5a96fab415f43721a44c5a761ecfcccc3dae9c21f34313f0e594b49d8d4564f4",
+    "0x61b28909165252d7d21dbcb16572eaf13a660ad3d6d9884358894e0ea88d1e1f"
+  ]
+};
+
+const polygonConfig = {
+  contracts: [
+    {
+      address: "0x9Df4C994d8d8c440d87da8BA94D355BB85706f51",
+      events: [
+        {
+          abi: "event ItemSoldV1(uint256 tokenId, uint256 quantity, uint256 totalPrice)",
+          amountField: "totalPrice"
+        }
+      ]
+    },
+    {
+      address: "0x1Fcfa7866Eb4361E322aFbcBcB426B27a29d90Bd",
+      events: [
+        {
+          abi: "event OrderPlaced(string orderId, string dpId, string productId, string purchaseUtr, string purchaseDate, uint256 quantity, uint256 amount, address customer)",
+          amountField: "amount"
+        },
+        // Add more events for this contract here
+      ]
+    },
+    // Add more contracts here
+  ]
+};
 
-const POLYGON_USDT_ADDRESS = ADDRESSES.polygon.USDT
-const ItemSoldEvent = "event ItemSoldV1(uint256 tokenId, uint256 quantity, uint256 totalPrice)";
-const PAGE_SIZE = 100;                        
-const APT_DECIMALS = 1e8;                     
+const POLYGON_USDT_ADDRESS = ADDRESSES.polygon.USDT;
+const APTOS_USDT_ADDRESS = ADDRESSES.aptos.USDC
+const PAGE_SIZE = 100;
+const APT_DECIMALS = 1e8;
+const USDC_DECIMALS = 1e6;
 const toUnixTime = (timestamp: string): number =>
   Math.floor(Number(timestamp) / 1e6);
 
 const version2tsCache: Record<string, number> = {};
+
 async function versionToTimestamp(version: string): Promise<number> {
   if (version2tsCache[version] !== undefined) return version2tsCache[version];
   const block = await fetchURL(`${APTOS_RPC}/v1/blocks/by_version/${version}`);
   const ts = block?.block_timestamp ? toUnixTime(block.block_timestamp) : 0;
   version2tsCache[version] = ts;
   return ts;
 }
+
 async function getEventData(
   resource: any,
   fromTimestamp: number,
   toTimestamp: number,
-  chain: string,
-  eventKeys: string[] = ["deposit_fungible", "deposit_native"],
+  contractAddress: string,
+  eventKeys: string[] = ["deposit_fungible", "deposit_native", "order_placed"],
 ): Promise<DepositFungible[]> {
 
   const collected: DepositFungible[] = [];
@@ -55,14 +85,16 @@ async function getEventData(
 
     for (let seq = totalEvents - 1; seq >= 0; seq -= PAGE_SIZE) {
       const batchStart = Math.max(seq - PAGE_SIZE + 1, 0);
-      const url = `${APTOS_RPC}/v1/accounts/${config[chain]}/events/${creationNum}?start=${batchStart}&limit=${PAGE_SIZE}`;
+
+      const url = `${APTOS_RPC}/v1/accounts/${contractAddress}/events/${creationNum}?start=${batchStart}&limit=${PAGE_SIZE}`;
+
       const events: any[] = await fetchURL(url);
       if (!events?.length) break;
 
       for (const e of events.reverse()) {
         const ts = await versionToTimestamp(e.version);
-        if (ts > toTimestamp) continue;         
-        if (ts < fromTimestamp) break;         
+        if (ts > toTimestamp) continue;
+        if (ts < fromTimestamp) break;
 
         collected.push({
           amount: e.data.amount,
@@ -82,22 +114,30 @@ async function getEventData(
 
 const fetchAptosRevenue: FetchV2 = async (options: FetchOptions) => {
   const dailyFees = options.createBalances();
-  const resources = await getResources(config[options.chain]);
-
-  const revHolders = resources.filter((resource) =>
-    resource.type.includes("RevenueContractV2::RevenueEventHolder")
-    || resource.type.includes("RevenueContractV2::RevenueEventHolderV1"));
-
-  const eventArrays = await Promise.all(
-    revHolders.map((r) =>
-      getEventData(r, options.fromTimestamp, options.toTimestamp, options.chain))
-  );
-
-  for (const event of eventArrays.flat()) {
-    if (event.token) {
-      dailyFees.add(event.token, event.amount);
-    } else {
-      dailyFees.addCGToken("aptos", Number(event.amount) / APT_DECIMALS);
+
+  for (const contractAddress of aptosConfig.contracts) {
+    const resources = await getResources(contractAddress);
+
+    const revHolders = resources.filter((resource) =>
+      resource.type.includes("RevenueContractV2::RevenueEventHolder")
+      || resource.type.includes("RevenueContractV2::RevenueEventHolderV1")
+      || resource.type.includes("order_management_v1::B2bRevenueEventHolder")
+    );
+
+    const eventArrays = await Promise.all(
+      revHolders.map((r) =>
+        getEventData(r, options.fromTimestamp, options.toTimestamp, contractAddress)
+      )
+    );
+    for (const event of eventArrays.flat()) {
+      if (event.token) {
+        dailyFees.add(event.token, event.amount);
+      } else if (event.type = "0x61b28909165252d7d21dbcb16572eaf13a660ad3d6d9884358894e0ea88d1e1f::order_management_v1::OrderPlacedEvent") {
+        dailyFees.addUSDValue(Number(event.amount) / USDC_DECIMALS);
+      }
+      else {
+        dailyFees.addCGToken("aptos", Number(event.amount) / APT_DECIMALS);
+      }
     }
   }
 
@@ -109,20 +149,28 @@ const fetchAptosRevenue: FetchV2 = async (options: FetchOptions) => {
 };
 
 const fetchPolygonRevenue: FetchV2 = async (options: FetchOptions) => {
-  const logs: any[] = await options.getLogs({
-    target: config[options.chain],
-    eventAbi: ItemSoldEvent,
-  });
-
-  const dailyFees = options.createBalances()
+  const dailyFees = options.createBalances();
 
-  for (const log of logs) {
-    const amount = Number(log.totalPrice);
-    if (!isNaN(amount)) {
-      dailyFees.add(POLYGON_USDT_ADDRESS, amount);
+  for (const contract of polygonConfig.contracts) {
+    for (const event of contract.events) {
+      const logs: any[] = await options.getLogs({
+        target: contract.address,
+        eventAbi: event.abi,
+      });
+      for (const log of logs) {
+        let amount: number = 0;
+        if (log.totalPrice) {
+          amount = Number(log.totalPrice);
+        }
+        else {
+          amount = Number(log.amount);
+        }
+        if (!isNaN(amount)) {
+          dailyFees.add(POLYGON_USDT_ADDRESS, amount);
+        }
+      }
     }
   }
-
   return {
     dailyFees,
     dailyRevenue: dailyFees,
@@ -142,7 +190,8 @@ const adapter: SimpleAdapter = {
   adapter: {
     [CHAIN.APTOS]: {
       fetch: fetchAptosRevenue,
-      start: "2025-06-02",
+      start: "2025-06-02", 
+
     },
     [CHAIN.POLYGON]: {
       fetch: fetchPolygonRevenue,
@@ -151,5 +200,4 @@ const adapter: SimpleAdapter = {
   },
 };
 
-export default adapter;
-
+export default adapter;