feat(btc): add get block header from db for reorg handling

Co-authored-by: Gaze <dev@gaze.network>

modules/bitcoin/btcclient/client_db.go

@@ -167,7 +167,11 @@ func (c *ClientDatabase) FetchAsync(ctx context.Context, from, to int64, ch chan
 }
 
 func (c *ClientDatabase) GetBlockHeader(ctx context.Context, height int64) (types.BlockHeader, error) {
-	return types.BlockHeader{}, nil
+	header, err := c.bitcoinDg.GetBlockHeaderByHeight(ctx, height)
+	if err != nil {
+		return types.BlockHeader{}, errors.WithStack(err)
+	}
+	return header, nil
 }
 
 func (c *ClientDatabase) prepareRange(ctx context.Context, fromHeight, toHeight int64) (start, end int64, skip bool, err error) {

modules/bitcoin/database/postgresql/queries/data.sql

@@ -1,10 +1,6 @@
 -- name: GetLatestBlockHeader :one
 SELECT * FROM bitcoin_blocks ORDER BY block_height DESC LIMIT 1;
 
--- TODO: GetBlockHeaderByRange
-
--- TODO: GetBlockByHeight/Hash (Join block with transactions, txins, txouts)
-
 -- name: InsertBlock :exec
 INSERT INTO bitcoin_blocks ("block_height","block_hash","version","merkle_root","prev_block_hash","timestamp","bits","nonce") VALUES ($1, $2, $3, $4, $5, $6, $7, $8);
 
@@ -42,6 +38,9 @@ WITH delete_tx AS (
 )
 DELETE FROM "bitcoin_blocks" WHERE "bitcoin_blocks"."block_height" >= @from_height;
 
+-- name: GetBlockByHeight :one
+SELECT * FROM bitcoin_blocks WHERE block_height = $1;
+
 -- name: GetBlocksByHeightRange :many
 SELECT * FROM bitcoin_blocks WHERE block_height >= @from_height AND block_height <= @to_height ORDER BY block_height ASC;
 

modules/bitcoin/datagateway/bitcoin_data.go

@@ -18,5 +18,6 @@ type BitcoinWriterDataDataGateway interface {
 
 type BitcoinReaderDataDataGateway interface {
 	GetLatestBlockHeader(context.Context) (types.BlockHeader, error)
+	GetBlockHeaderByHeight(ctx context.Context, blockHeight int64) (types.BlockHeader, error)
 	GetBlocksByHeightRange(ctx context.Context, from int64, to int64) ([]*types.Block, error)
 }

modules/bitcoin/repository/postgres/block.go

@@ -81,6 +81,19 @@ func (r *Repository) RevertBlocks(ctx context.Context, from int64) error {
 	return nil
 }
 
+func (r *Repository) GetBlockHeaderByHeight(ctx context.Context, blockHeight int64) (types.BlockHeader, error) {
+	blockModel, err := r.queries.GetBlockByHeight(ctx, int32(blockHeight))
+	if err != nil {
+		return types.BlockHeader{}, errors.Wrap(err, "failed to get block by height")
+	}
+
+	data, err := mapBlockHeaderModelToType(blockModel)
+	if err != nil {
+		return types.BlockHeader{}, errors.Wrap(err, "failed to map block header model to type")
+	}
+	return data, nil
+}
+
 func (r *Repository) GetBlocksByHeightRange(ctx context.Context, from int64, to int64) ([]*types.Block, error) {
 	blocks, err := r.queries.GetBlocksByHeightRange(ctx, gen.GetBlocksByHeightRangeParams{
 		FromHeight: int32(from),

modules/bitcoin/repository/postgres/gen/data.sql.go

@@ -11,6 +11,26 @@ import (
 	"github.com/jackc/pgx/v5/pgtype"
 )
 
+const getBlockByHeight = `-- name: GetBlockByHeight :one
+SELECT block_height, block_hash, version, merkle_root, prev_block_hash, timestamp, bits, nonce FROM bitcoin_blocks WHERE block_height = $1
+`
+
+func (q *Queries) GetBlockByHeight(ctx context.Context, blockHeight int32) (BitcoinBlock, error) {
+	row := q.db.QueryRow(ctx, getBlockByHeight, blockHeight)
+	var i BitcoinBlock
+	err := row.Scan(
+		&i.BlockHeight,
+		&i.BlockHash,
+		&i.Version,
+		&i.MerkleRoot,
+		&i.PrevBlockHash,
+		&i.Timestamp,
+		&i.Bits,
+		&i.Nonce,
+	)
+	return i, err
+}
+
 const getBlocksByHeightRange = `-- name: GetBlocksByHeightRange :many
 SELECT block_height, block_hash, version, merkle_root, prev_block_hash, timestamp, bits, nonce FROM bitcoin_blocks WHERE block_height >= $1 AND block_height <= $2 ORDER BY block_height ASC
 `
@@ -169,8 +189,6 @@ func (q *Queries) GetTransactionsByHeightRange(ctx context.Context, arg GetTrans
 }
 
 const insertBlock = `-- name: InsertBlock :exec
-
-
 INSERT INTO bitcoin_blocks ("block_height","block_hash","version","merkle_root","prev_block_hash","timestamp","bits","nonce") VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
 `
 
@@ -185,8 +203,6 @@ type InsertBlockParams struct {
 	Nonce         int32
 }
 
-// TODO: GetBlockHeaderByRange
-// TODO: GetBlockByHeight/Hash (Join block with transactions, txins, txouts)
 func (q *Queries) InsertBlock(ctx context.Context, arg InsertBlockParams) error {
 	_, err := q.db.Exec(ctx, insertBlock,
 		arg.BlockHeight,