use litestorage as blocksource

Author: erokhinav

cmd/api/main.go

@@ -69,6 +69,7 @@ func main() {
 		api.WithExecutor(storage),
 		api.WithMessageSender(msgSender),
 		api.WithSpamFilter(spamFilter),
+		api.WithBlocksSource(storage),
 		api.WithTonConnectSecret(cfg.TonConnect.Secret),
 	)
 	if err != nil {

pkg/api/blockchain_handlers.go

@@ -79,7 +79,7 @@ func (h *Handler) DownloadBlockchainBlockBoc(ctx context.Context, params oas.Dow
 		return nil, toError(http.StatusBadRequest, err)
 	}
 
-	bocBytes, err := h.blockSource.GetBlockchainBlock(ctx, blockID.Workchain, blockID.Shard, blockID.Seqno)
+	bocBytes, err := h.blockSource.GetBlockchainBlock(ctx, blockID)
 	if err != nil {
 		return nil, toError(http.StatusInternalServerError, err)
 	}

pkg/api/interfaces.go

@@ -215,5 +215,5 @@ type mempoolEmulate struct {
 }
 
 type blockSource interface {
-	GetBlockchainBlock(ctx context.Context, workchain int32, shard uint64, seqno uint32) ([]byte, error)
+	GetBlockchainBlock(ctx context.Context, id tongo.BlockID) ([]byte, error)
 }

pkg/litestorage/litestorage.go

@@ -574,3 +574,15 @@ func (s *LiteStorage) SaveTraceWithState(ctx context.Context, msgHash string, tr
 func (s *LiteStorage) GetTraceWithState(ctx context.Context, msgHash string) (*core.Trace, int, []abi.MethodInvocation, error) {
 	return nil, 0, nil, fmt.Errorf("not implemented")
 }
+
+func (s *LiteStorage) GetBlockchainBlock(ctx context.Context, id tongo.BlockID) ([]byte, error) {
+	idExt, _, err := s.client.LookupBlock(ctx, id, 1, nil, nil)
+	if err != nil {
+		return nil, err
+	}
+	block, err := s.client.GetBlockRaw(ctx, idExt)
+	if err != nil {
+		return nil, err
+	}
+	return block.Data, nil
+}