feat: seed genesis data at project initiation

Agoric · May 3, 2024 · 80b8891 · 80b8891
1 parent 9fc1b43
commit 80b8891
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 8 deletions.
diff --git a/project.ts b/project.ts
@@ -52,6 +52,20 @@ const project: CosmosProject = {
     ]),
   },
   dataSources: [
+    {
+      kind: CosmosDatasourceKind.Runtime,
+      startBlock: 1,
+      endBlock: 1,
+      mapping: {
+        file: "./dist/index.js",
+        handlers: [
+          {
+            kind: CosmosHandlerKind.Block,
+            handler: "initiateBalancesTable",
+          },
+        ],
+      },
+    },
     {
       kind: CosmosDatasourceKind.Runtime,
       // First block of mainnet is 2115669
@@ -118,13 +132,6 @@ const project: CosmosProject = {
               type: "burn",
             },
           },
-          // {
-          //   handler: "handleTransferEvent",
-          //   kind: CosmosHandlerKind.Event,
-          //   filter: {
-          //     type: "transfer",
-          //   },
-          // },
           // Distribution Events
           {
             handler: "handleBalanceEvent",

diff --git a/src/mappings/mappingHandlers.ts b/src/mappings/mappingHandlers.ts
@@ -16,8 +16,9 @@ import {
   VaultManagerMetricsDaily,
   PsmMetricDaily,
   ReserveAllocationMetricsDaily,
+  Balances,
 } from "../types";
-import { CosmosEvent } from "@subql/types-cosmos";
+import { CosmosBlock, CosmosEvent } from "@subql/types-cosmos";
 import {
   b64encode,
   b64decode,
@@ -230,3 +231,16 @@ export async function handleBalanceEvent(
   const formattedAmount = BigInt(Math.round(Number(amount.slice(0, -4))));
   await balancesKit.updateBalance(address, formattedAmount, operation);
 }
+
+export async function initiateBalancesTable(block: CosmosBlock): Promise<void> {
+  const newBalance = new Balances(
+    'agoric1estsewt6jqsx77pwcxkn5ah0jqgu8rhgflwfdl'
+  );
+  newBalance.address = 'agoric1estsewt6jqsx77pwcxkn5ah0jqgu8rhgflwfdl';
+  newBalance.balance = BigInt(999999995000000000);
+  newBalance.denom = 'ubld';
+
+  await newBalance.save();
+
+  logger.info(`Balances Table Initiated`);
+}