Add getBlockHeader

packages/contracts/source/contracts/database.ts

@@ -1,4 +1,4 @@
-import { Block, Commit, Transaction } from "./crypto/index.js";
+import { Block, BlockHeader, Commit, Transaction } from "./crypto/index.js";
 
 export interface State {
 	height: number;
@@ -22,6 +22,9 @@ export interface DatabaseService {
 	getBlockById(id: string): Promise<Block | undefined>;
 	findBlocks(start: number, end: number): Promise<Block[]>;
 
+	getBlockHeader(height: number): Promise<BlockHeader | undefined>;
+	getBlockHeaderById(id: string): Promise<BlockHeader | undefined>;
+
 	getTransactionById(id: string): Promise<Transaction | undefined>;
 
 	addCommit(block: Commit): void;

packages/database/source/database-service.ts

@@ -32,6 +32,9 @@ export class DatabaseService implements Contracts.Database.DatabaseService {
 	@inject(Identifiers.Cryptography.Block.Factory)
 	private readonly blockFactory!: Contracts.Crypto.BlockFactory;
 
+	@inject(Identifiers.Cryptography.Block.Deserializer)
+	private readonly blockDeserializer!: Contracts.Crypto.BlockDeserializer;
+
 	@inject(Identifiers.Cryptography.Transaction.Factory)
 	private readonly transactionFactory!: Contracts.Crypto.TransactionFactory;
 
@@ -138,6 +141,31 @@ export class DatabaseService implements Contracts.Database.DatabaseService {
 		return undefined;
 	}
 
+	public async getBlockHeader(height: number): Promise<Contracts.Crypto.BlockHeader | undefined> {
+		const bytes = this.#readBlockHeaderBytes(height);
+
+		if (bytes) {
+			return await this.blockDeserializer.deserializeHeader(bytes);
+		}
+
+		return undefined;
+	}
+
+	public async getBlockHeaderById(id: string): Promise<Contracts.Crypto.BlockHeader | undefined> {
+		const height = this.#getHeightById(id);
+
+		if (height === undefined) {
+			return undefined;
+		}
+
+		const bytes = this.#readBlockHeaderBytes(height);
+		if (bytes) {
+			return await this.blockDeserializer.deserializeHeader(bytes);
+		}
+
+		return undefined;
+	}
+
 	public async findBlocks(start: number, end: number): Promise<Contracts.Crypto.Block[]> {
 		return await this.#map<Contracts.Crypto.Block>(
 			await this.findCommitBuffers(start, end),
@@ -273,6 +301,17 @@ export class DatabaseService implements Contracts.Database.DatabaseService {
 		return Buffer.concat([blockBuffer, ...transactions]);
 	}
 
+	#readBlockHeaderBytes(height: number): Buffer | undefined {
+		if (this.#commitCache.has(height)) {
+			return Buffer.from(this.#commitCache.get(height)!.serialized, "hex").subarray(
+				this.proofSize(),
+				this.proofSize() + this.headerSize(),
+			);
+		}
+
+		return this.blockStorage.get(height);
+	}
+
 	async #map<T>(data: unknown[], callback: (...arguments_: any[]) => Promise<T>): Promise<T[]> {
 		const result: T[] = [];