Add block metadata and size calculation

schema.graphql

@@ -14,11 +14,74 @@ type SupplyDenom @entity {
   id: ID!
 }
 
+type BlockIdParts @jsonField(indexed: false) {
+  total: Int!
+  hash: String!
+}
+
+type BlockId @jsonField(indexed: false) {
+  hash: String!
+  parts: BlockIdParts!
+}
+
+type BlockVersion @jsonField(indexed: false) {
+  version: BigInt!
+  app: Int!
+}
+
+type BlockHeader @jsonField {
+  chainId: String!
+  height: BigInt! @index
+  time: Date!
+  version: BlockVersion!
+  lastBlockId: BlockId!
+  lastCommitHash: String!
+  dataHash: String!
+  validatorsHash: String!
+  nextValidatorsHash: String!
+  consensusHash: String!
+  appHash: String!
+  lastResultsHash: String!
+  evidenceHash: String!
+  proposerAddress: String!
+}
+
+type BlockCommitSignature @jsonField(indexed: false) {
+  blockIdFlag: Int!
+  validatorAddress: String!
+  timestamp: Date!
+  signature: String!
+}
+
+type BlockLastCommit @jsonField(indexed: false) {
+  id: ID!
+  blockIdHash: String!
+  blockIdPartsTotal: Int!
+  blockIdPartsHash: String!
+  height: BigInt! @index
+  round: Int!
+  signatures: [BlockCommitSignature]!
+}
+
+# This handle all the information that is coming with the block but is not often used
+# but could help in some debug environment.
+type BlockMetadata @entity {
+  id: ID!
+  blockId: BlockId!
+  header: BlockHeader!
+  # first block has this null on js
+  lastCommit: BlockLastCommit
+}
+
 type Block @entity {
   id: ID! # The block header hash
   chainId: String! @index
   height: BigInt! @index
   timestamp: Date!
+  proposerAddress: String!
+  metadata: BlockMetadata!
+  size: Int!
+  # relations
   transactions: [Transaction] @derivedFrom(field: "block")
   messages: [Message] @derivedFrom(field: "block")
   events: [Event] @derivedFrom(field: "block")

src/mappings/authz/exec.ts

@@ -1,10 +1,19 @@
 import util from "util";
-import {CosmosMessage} from "@subql/types-cosmos";
+import { CosmosMessage } from "@subql/types-cosmos";
 import allModuleTypes from "../../cosmjs/proto";
-import { AuthzExec, AuthzMsgExec, Message } from "../../types";
+import {
+  AuthzExec,
+  AuthzMsgExec,
+  Message,
+} from "../../types";
 import { _handleUpdateParam } from "../poktroll/params";
 import { AuthzExecMsg } from "../types";
-import { attemptHandling, messageId, stringify, unprocessedEventHandler } from "../utils";
+import {
+  attemptHandling,
+  unprocessedEventHandler,
+} from "../utils/handlers";
+import { messageId } from "../utils/ids";
+import { stringify } from "../utils/json";
 
 // This is required for the binary reader to work. It expects TextEncoder and TextDecoder to be set in globalThis.
 globalThis.TextEncoder = util.TextEncoder;
@@ -44,7 +53,7 @@ async function _handleAuthzExec(msg: CosmosMessage<AuthzExecMsg>): Promise<void>
     //_handleUpdateParam will return the decoded message if it is a param update
     // otherwise it will return undefined.
     //_handleUpdateParam will decode and save the message using its specific entity
-    let decodedMsg: unknown = await _handleUpdateParam(encodedMsg, msg.block.block.id)
+    let decodedMsg: unknown = await _handleUpdateParam(encodedMsg, msg.block.block.id);
 
     if (!decodedMsg) {
       for (const [typeUrl, msgType] of allModuleTypes) {
@@ -53,7 +62,7 @@ async function _handleAuthzExec(msg: CosmosMessage<AuthzExecMsg>): Promise<void>
           const bytes = new Uint8Array(Object.values(encodedMsg.value));
 
           decodedMsg = msgType.decode(bytes);
-          break
+          break;
         }
       }
     }

src/mappings/bank/balanceChange.ts

@@ -1,18 +1,50 @@
 import { CosmosEvent } from "@subql/types-cosmos";
 import { parseCoins } from "../../cosmjs/utils";
 import {
+  Account,
+  Balance,
   NativeBalanceChange,
   Transaction,
 } from "../../types";
 import {
   attemptHandling,
-  checkBalancesAccount,
+  unprocessedEventHandler,
+} from "../utils/handlers";
+import {
+  getBalanceId,
   getEventId,
   messageId,
-  stringify,
-  unprocessedEventHandler,
-  updateAccountBalance,
-} from "../utils";
+} from "../utils/ids";
+import { stringify } from "../utils/json";
+
+export async function updateAccountBalance(address: string, denom: string, offset: bigint, blockId: string): Promise<void> {
+  let balance = await Balance.get(getBalanceId(address, denom));
+
+  if (!balance) {
+    balance = Balance.create({
+      id: getBalanceId(address, denom),
+      accountId: address,
+      denom,
+      amount: offset,
+      lastUpdatedBlockId: blockId,
+    });
+  } else {
+    balance.amount = balance.amount + offset;
+    balance.lastUpdatedBlockId = blockId;
+  }
+
+  await balance.save();
+
+  logger.debug(`[updateAccountBalance] (address): ${address}, (denom): ${denom}, (offset): ${offset}, (newBalance): ${balance?.amount}`);
+}
+
+export async function checkBalancesAccount(address: string, chainId: string): Promise<void> {
+  let accountEntity = await Account.get(address);
+  if (typeof (accountEntity) === "undefined") {
+    accountEntity = Account.create({ id: address, chainId });
+    await accountEntity.save();
+  }
+}
 
 export async function saveNativeBalanceEvent(id: string, address: string, amount: bigint, denom: string, event: CosmosEvent): Promise<void> {
   await checkBalancesAccount(address, event.block.block.header.chainId);

src/mappings/bank/supply.ts

@@ -7,7 +7,7 @@ import {
   BlockSupply,
   Supply,
 } from "../../types";
-import { stringify } from "../utils";
+import { stringify } from "../utils/json";
 
 export const getSupplyId = function(denom: string, height: number): string {
   return `${denom}@${height}`;
@@ -26,7 +26,7 @@ export async function queryTotalSupply(): Promise<Coin[]> {
   let paginationKey: Uint8Array | undefined;
 
   try {
-    // Here we force the use of a private property, breaking typescript limitation, due to the need of call a total supply
+    // Here we force the use of a private property, breaking TypeScript limitation, due to the need of call a total supply
     // rpc query of cosmosjs that is not exposed on the implemented client by SubQuery team.
     // To avoid this, we need to move to implement our own rpc client and also use `unsafe` parameter which I prefer to avoid.
     // eslint-disable-next-line @typescript-eslint/ban-ts-comment

src/mappings/bank/transfer.ts

@@ -6,10 +6,10 @@ import { NativeTransfer } from "../../types";
 import { NativeTransferMsg } from "../types";
 import {
   attemptHandling,
-  messageId,
-  stringify,
   unprocessedEventHandler,
-} from "../utils";
+} from "../utils/handlers";
+import { messageId } from "../utils/ids";
+import { stringify } from "../utils/json";
 
 export async function handleNativeTransfer(event: CosmosEvent): Promise<void> {
   await attemptHandling(event, _handleNativeTransfer, unprocessedEventHandler);

src/mappings/constants.ts

@@ -1,6 +1,6 @@
 import { ApplicationUnbondingReason as ApplicationUnbondingReasonEnum } from "../client/poktroll/application/event";
 
-export const PREFIX = "poktroll";
+export const PREFIX = "pokt";
 
 export enum StakeStatus {
   Staked = 0,

src/mappings/genesis.ts

@@ -40,8 +40,9 @@ import {
   getMsgStakeServiceId,
   getParamId,
   getStakeServiceId,
-  stringify,
-} from "./utils";
+} from "./utils/ids";
+import { stringify } from "./utils/json";
+
 
 export async function handleGenesis(block: CosmosBlock): Promise<void> {
   // we MUST load the JSON this way due to the sandboxed environment