Update TON chain client type, use retry strategy (#761)

Author: krebernisak

.changeset/mighty-chicken-end.md

@@ -0,0 +1,5 @@
+---
+"chainlink-deployments-framework": minor
+---
+
+Changed TON chain client type, use retry strategy

chain/mcms/adapters/chain_access.go

@@ -85,7 +85,7 @@ func (a *ChainAccessAdapter) SuiClient(selector uint64) (sui.ISuiAPI, mcmssui.Su
 }
 
 // TonClient returns the Ton API client for the given selector.
-func (a *ChainAccessAdapter) TonClient(selector uint64) (*ton.APIClient, bool) {
+func (a *ChainAccessAdapter) TonClient(selector uint64) (ton.APIClientWrapped, bool) {
 	ch, ok := a.inner.TonChains()[selector]
 	if !ok {
 		return nil, false

chain/ton/provider/ctf_provider.go

@@ -32,6 +32,7 @@ const (
 
 	// supportedTONImageRepository is the only supported Docker image repository for TON localnet.
 	supportedTONImageRepository = "ghcr.io/neodix42/mylocalton-docker"
+	defaultClientRetryCount     = 5
 )
 
 // CTFChainProviderConfig holds the configuration to initialize the CTFChainProvider.
@@ -136,7 +137,7 @@ func (p *CTFChainProvider) Initialize(ctx context.Context) (chain.BlockChain, er
 	return *p.chain, nil
 }
 
-func (p *CTFChainProvider) startContainer(ctx context.Context, chainID string) (string, *ton.APIClient, error) {
+func (p *CTFChainProvider) startContainer(ctx context.Context, chainID string) (string, ton.APIClientWrapped, error) {
 	var (
 		attempts = uint(10)
 		url      string
@@ -190,7 +191,7 @@ func (p *CTFChainProvider) startContainer(ctx context.Context, chainID string) (
 		return "", nil, fmt.Errorf("failed to create liteclient connection pool: %w", err)
 	}
 
-	client := ton.NewAPIClient(connectionPool, ton.ProofCheckPolicyFast)
+	client := ton.NewAPIClient(connectionPool, ton.ProofCheckPolicyFast).WithRetry(defaultClientRetryCount)
 
 	// check connection, CTFv2 handles the readiness
 	mb, err := getMasterchainBlockID(ctx, client)

chain/ton/provider/rpc_provider.go

@@ -110,13 +110,13 @@ func NewRPCChainProvider(selector uint64, config RPCChainProviderConfig) *RPCCha
 }
 
 // setupConnection creates and tests a connection to the TON liteserver
-func setupConnection(ctx context.Context, liteserverURL string) (*tonlib.APIClient, error) {
+func setupConnection(ctx context.Context, liteserverURL string) (tonlib.APIClientWrapped, error) {
 	connectionPool, err := createLiteclientConnectionPool(ctx, liteserverURL)
 	if err != nil {
 		return nil, fmt.Errorf("failed to connect to liteserver: %w", err)
 	}
 
-	api := tonlib.NewAPIClient(connectionPool, tonlib.ProofCheckPolicyFast)
+	api := tonlib.NewAPIClient(connectionPool, tonlib.ProofCheckPolicyFast).WithRetry(defaultClientRetryCount)
 
 	// Test connection and get current block
 	mb, err := api.GetMasterchainInfo(ctx)

chain/ton/ton_chain.go

@@ -21,12 +21,12 @@ type ChainMetadata = common.ChainMetadata
 
 // Chain represents a TON chain.
 type Chain struct {
-	ChainMetadata                  // Contains canonical chain identifier
-	Client        *ton.APIClient   // APIClient for Lite Server connection
-	Wallet        *wallet.Wallet   // Wallet abstraction (signing, sending)
-	WalletAddress *address.Address // Address of deployer wallet
-	URL           string           // Liteserver URL
-	Confirm       ConfirmFunc      // Function to confirm transactions
+	ChainMetadata                      // Contains canonical chain identifier
+	Client        ton.APIClientWrapped // APIClient for Lite Server connection
+	Wallet        *wallet.Wallet       // Wallet abstraction (signing, sending)
+	WalletAddress *address.Address     // Address of deployer wallet
+	URL           string               // Liteserver URL
+	Confirm       ConfirmFunc          // Function to confirm transactions
 }
 
 // MakeDefaultConfirmFunc creates a default ConfirmFunc that waits for transaction trace.