Get Block Utility Function

src/extend-utils.ts

@@ -1,5 +1,5 @@
 import { HardhatPluginError } from "hardhat/plugins";
-import { HardhatRuntimeEnvironment } from "hardhat/types";
+import { Block, HardhatRuntimeEnvironment } from "hardhat/types";
 import * as path from "path";
 
 import {
@@ -26,7 +26,7 @@ export async function getContractFactoryUtil(hre: HardhatRuntimeEnvironment, con
 
     const metadataPath = await findPath(artifactsPath, metadataSearchTarget);
     if (!metadataPath) {
-        throw new HardhatPluginError(PLUGIN_NAME, `Could not find metadata for ${contractPath}`);
+        throw new HardhatPluginError(PLUGIN_NAME, `Could not find metadata for contract ${metadataSearchTarget}`);
     }
 
     const abiSearchTarget = path.join(
@@ -168,3 +168,32 @@ export async function getTransactionReceiptUtil(
     const txReceipt = JSON.parse(executed.stdout.toString()) as TransactionReceipt;
     return txReceipt;
 }
+
+export async function getBlockUtil(
+    hre: HardhatRuntimeEnvironment,
+    blockNumber?: number,
+    hash?: string
+): Promise<Block> {
+    const blockOptions = {
+        feederGatewayUrl: hre.config.starknet.networkUrl,
+        gatewayUrl: hre.config.starknet.networkUrl,
+        number: blockNumber,
+        hash: hash
+    };
+
+    // If blockNumber and hash are both provided, it will get the block by hash
+    // If none of them are provided, it will get the latest block
+    if (!blockOptions.number && !blockOptions.hash) {
+        delete blockOptions.number;
+        delete blockOptions.hash;
+    }
+
+    const executed = await hre.starknetWrapper.getBlock(blockOptions);
+
+    if (executed.statusCode) {
+        const msg = `Could not get block. Error: ${executed.stderr.toString()}`;
+        throw new HardhatPluginError(PLUGIN_NAME, msg);
+    }
+    const block = JSON.parse(executed.stdout.toString()) as Block;
+    return block;
+}

src/index.ts

@@ -37,7 +37,8 @@ import {
     getTransactionUtil,
     getTransactionReceiptUtil,
     getWalletUtil,
-    shortStringToBigIntUtil
+    shortStringToBigIntUtil,
+    getBlockUtil
 } from "./extend-utils";
 import { DevnetUtils } from "./devnet-utils";
 
@@ -228,6 +229,11 @@ extendEnvironment((hre) => {
         getTransactionReceipt: async (txHash) => {
             const txReceipt = await getTransactionReceiptUtil(txHash, hre);
             return txReceipt;
+        },
+
+        getBlock: async (blockNumber?: number, hash?: string) => {
+            const block = await getBlockUtil(hre, blockNumber, hash);
+            return block;
         }
     };
 });

src/starknet-types.ts

@@ -86,3 +86,14 @@ export interface Transaction {
     transaction: TransactionData;
     transaction_index: number;
 }
+
+export interface Block {
+    block_hash: string;
+    parent_block_hash: string;
+    block_number: number;
+    state_root: string;
+    status: string;
+    timestamp: number;
+    transaction_receipts: Record<string, unknown>;
+    transactions: Record<string, unknown>;
+}

src/starknet-wrappers.ts

@@ -58,6 +58,13 @@ interface DeployAccountWrapperOptions {
     network: string;
 }
 
+interface BlockQueryWrapperOptions {
+    number?: number;
+    hash?: string;
+    gatewayUrl: string;
+    feederGatewayUrl: string;
+}
+
 export abstract class StarknetWrapper {
     protected prepareCompileOptions(options: CompileWrapperOptions): string[] {
         const ret = [
@@ -212,6 +219,26 @@ export abstract class StarknetWrapper {
     ): Promise<ProcessResult>;
 
     public abstract getTransaction(options: TxHashQueryWrapperOptions): Promise<ProcessResult>;
+
+    protected prepareBlockQueryOptions(command: string, options: BlockQueryWrapperOptions): string[] {
+
+        const commandArr = [
+            command,
+            options?.hash ? "--hash" : "",
+            options?.hash ? options.hash : "",
+            options?.number ? "--number" : "",
+            options?.number ? options?.number.toString() : "",
+            "--gateway_url",
+            options.gatewayUrl,
+            "--feeder_gateway_url",
+            options.feederGatewayUrl
+        ];
+
+        // delete empty elements/strings
+        return commandArr.filter(e => e);
+    }
+
+    public abstract getBlock(options: BlockQueryWrapperOptions): Promise<ProcessResult>;
 }
 
 function getFullImageName(image: Image): string {
@@ -415,6 +442,25 @@ export class DockerWrapper extends StarknetWrapper {
         );
         return executed;
     }
+
+    public async getBlock(options: BlockQueryWrapperOptions): Promise<ProcessResult> {
+        const binds: String2String = {};
+
+        const dockerOptions = {
+            binds,
+            networkMode: "host"
+        };
+
+        const preparedOptions = this.prepareBlockQueryOptions("get_block", options);
+
+        const docker = await this.getDocker();
+        const executed = await docker.runContainer(
+            this.image,
+            ["starknet", ...preparedOptions],
+            dockerOptions
+        );
+        return executed;
+    }
 }
 
 function checkCommandPath(commandPath: string): void {
@@ -505,4 +551,10 @@ export class VenvWrapper extends StarknetWrapper {
         const executed = await this.execute(this.starknetPath, preparedOptions);
         return executed;
     }
+
+    public async getBlock(options: BlockQueryWrapperOptions): Promise<ProcessResult> {
+        const preparedOptions = this.prepareBlockQueryOptions("get_block", options);
+        const executed = await this.execute(this.starknetPath, preparedOptions);
+        return executed;
+    }
 }

src/type-extensions.ts

@@ -9,7 +9,7 @@ import {
 import { StarknetWrapper } from "./starknet-wrappers";
 import { FlushResponse, LoadL1MessagingContractResponse } from "./devnet-utils";
 import { Account, ArgentAccount, OpenZeppelinAccount } from "./account";
-import { Transaction, TransactionReceipt } from "./starknet-types";
+import { Transaction, TransactionReceipt, Block } from "./starknet-types";
 import { HttpNetworkConfig } from "hardhat/types/config";
 
 type StarknetConfig = {
@@ -71,6 +71,7 @@ type OpenZeppelinAccountType = OpenZeppelinAccount;
 type ArgentAccountType = ArgentAccount;
 type TransactionReceiptType = TransactionReceipt;
 type TransactionType = Transaction;
+type BlockType = Block;
 
 declare module "hardhat/types/runtime" {
     interface Devnet {
@@ -179,6 +180,8 @@ declare module "hardhat/types/runtime" {
             getTransaction: (txHash: string) => Promise<Transaction>;
 
             getTransactionReceipt: (txHash: string) => Promise<TransactionReceipt>;
+
+            getBlock: (blockNumber?: number, hash?: string) => Promise<Block>;
         };
     }
 
@@ -191,4 +194,5 @@ declare module "hardhat/types/runtime" {
     type ArgentAccount = ArgentAccountType;
     type Transaction = TransactionType;
     type TransactionReceipt = TransactionReceiptType;
+    type Block = BlockType;
 }