feat: update ScanBlock function with new API calls (#16)

adapter/block/scan/wire.go

@@ -12,6 +12,7 @@ import (
 	"github.com/blackhorseya/ryze/app/domain/block/biz"
 	"github.com/blackhorseya/ryze/app/infra/configx"
 	"github.com/blackhorseya/ryze/app/infra/otelx"
+	"github.com/blackhorseya/ryze/app/infra/tonx"
 	"github.com/blackhorseya/ryze/app/infra/transports/httpx"
 	"github.com/blackhorseya/ryze/pkg/adapterx"
 	"github.com/blackhorseya/ryze/pkg/contextx"
@@ -37,6 +38,10 @@ func initServer(app *configx.Application) (*httpx.Server, error) {
 	return httpx.NewServer(app.HTTP)
 }
 
+func initTonx() (*tonx.Client, error) {
+	return tonx.NewClient(tonx.Options{Network: "mainnet"})
+}
+
 func New(v *viper.Viper) (adapterx.Restful, error) {
 	panic(wire.Build(
 		wire.Struct(new(wirex.Injector), "*"),
@@ -47,5 +52,6 @@ func New(v *viper.Viper) (adapterx.Restful, error) {
 		initServer,
 
 		biz.NewBlockService,
+		initTonx,
 	))
 }

app/domain/block/biz/block_service.go

@@ -3,16 +3,23 @@ package biz
 import (
 	"context"
 
+	"github.com/blackhorseya/ryze/app/infra/tonx"
 	"github.com/blackhorseya/ryze/entity/domain/block/model"
+	"github.com/blackhorseya/ryze/pkg/contextx"
+	"github.com/xssnick/tonutils-go/ton"
+	"go.uber.org/zap"
 	"google.golang.org/protobuf/types/known/timestamppb"
 )
 
 type impl struct {
+	client *tonx.Client
 }
 
 // NewBlockService is used to create a new model.BlockServiceServer
-func NewBlockService() model.BlockServiceServer {
-	return &impl{}
+func NewBlockService(client *tonx.Client) model.BlockServiceServer {
+	return &impl{
+		client: client,
+	}
 }
 
 func (i *impl) GetBlock(ctx context.Context, request *model.GetBlockRequest) (*model.Block, error) {
@@ -30,7 +37,38 @@ func (i *impl) GetBlocks(request *model.GetBlocksRequest, server model.BlockServ
 	panic("implement me")
 }
 
-func (i *impl) ScanBlock(request *model.ScanBlockRequest, server model.BlockService_ScanBlockServer) error {
-	// TODO: 2024/7/28|sean|implement me
-	panic("implement me")
+func (i *impl) ScanBlock(request *model.ScanBlockRequest, stream model.BlockService_ScanBlockServer) error {
+	api := ton.NewAPIClient(i.client, ton.ProofCheckPolicyFast).WithRetry()
+	api.SetTrustedBlockFromConfig(i.client.Config)
+
+	ctx := contextx.WithContext(stream.Context())
+	master, err := api.GetMasterchainInfo(ctx)
+	if err != nil {
+		ctx.Error("failed to get masterchain info", zap.Error(err))
+		return err
+	}
+	ctx.Info("master proofs chain successfully verified, all data is now safe and trusted!")
+
+	stickyContext := api.Client().StickyContext(ctx)
+	shardLastSeqno := map[string]uint32{}
+	firstShards, err := api.GetBlockShardsInfo(stickyContext, master)
+	if err != nil {
+		ctx.Error("failed to get block shards info", zap.Error(err))
+		return err
+	}
+
+	for _, shard := range firstShards {
+		shardLastSeqno[tonx.GetShardID(shard)] = shard.SeqNo
+	}
+
+	for {
+		ctx.Info("scanning master block", zap.Uint32("seq_no", master.SeqNo))
+
+		next := master.SeqNo + 1
+		master, err = api.WaitForBlock(next).LookupBlock(ctx, master.Workchain, master.Shard, next)
+		if err != nil {
+			ctx.Error("failed to lookup block", zap.Uint32("seq_no", next), zap.Error(err))
+			return err
+		}
+	}
 }

app/domain/block/biz/block_service_test.go

@@ -0,0 +1,44 @@
+package biz
+
+import (
+	"testing"
+	"time"
+
+	"github.com/blackhorseya/ryze/app/infra/tonx"
+	"github.com/blackhorseya/ryze/entity/domain/block/model"
+	"github.com/blackhorseya/ryze/pkg/contextx"
+	"github.com/stretchr/testify/suite"
+	"go.uber.org/mock/gomock"
+)
+
+type suiteTester struct {
+	suite.Suite
+
+	ctrl   *gomock.Controller
+	client *tonx.Client
+	biz    model.BlockServiceServer
+}
+
+func (s *suiteTester) SetupTest() {
+	s.ctrl = gomock.NewController(s.T())
+	s.client, _ = tonx.NewClient(tonx.Options{Network: "testnet"})
+	s.biz = NewBlockService(s.client)
+}
+
+func (s *suiteTester) TearDownTest() {
+	s.ctrl.Finish()
+}
+
+func TestAll(t *testing.T) {
+	suite.Run(t, new(suiteTester))
+}
+
+func (s *suiteTester) Test_impl_ScanBlock() {
+	stream := model.NewMockBlockService_ScanBlockServer(s.ctrl)
+
+	timeout, cancelFunc := contextx.WithTimeout(contextx.Background(), 2*time.Second)
+	defer cancelFunc()
+
+	stream.EXPECT().Context().Return(timeout).Times(1)
+	_ = s.biz.ScanBlock(&model.ScanBlockRequest{}, stream)
+}

app/infra/tonx/utils.go

@@ -0,0 +1,12 @@
+package tonx
+
+import (
+	"fmt"
+
+	"github.com/xssnick/tonutils-go/ton"
+)
+
+// GetShardID is used to get the shard id
+func GetShardID(shard *ton.BlockIDExt) string {
+	return fmt.Sprintf("%d|%d", shard.Workchain, shard.Shard)
+}