rebase main

Author: kingpinXD

cmd/zetatool/chains/sui.go

@@ -0,0 +1,60 @@
+package chains
+
+import (
+	"context"
+	"fmt"
+	"strconv"
+
+	"github.com/block-vision/sui-go-sdk/models"
+	"github.com/block-vision/sui-go-sdk/sui"
+)
+
+const (
+	// nanoSUIPerSUI is the number of MIST in 1 SUI (10^9)
+	nanoSUIPerSUI = 1_000_000_000
+)
+
+// GetSuiBalance fetches the SUI balance for a given address
+// Returns the balance in MIST (1 SUI = 10^9 MIST)
+func GetSuiBalance(ctx context.Context, rpcURL string, address string) (uint64, error) {
+	client := sui.NewSuiClient(rpcURL)
+
+	var (
+		totalBalance uint64
+		cursor       = any(nil)
+	)
+
+	// Query all SUI coins owned by the address
+	for {
+		resp, err := client.SuiXGetCoins(ctx, models.SuiXGetCoinsRequest{
+			Owner:    address,
+			CoinType: "0x2::sui::SUI",
+			Cursor:   cursor,
+		})
+		if err != nil {
+			return 0, err
+		}
+
+		// Sum up balances from all coins
+		for _, coin := range resp.Data {
+			balance, err := strconv.ParseUint(coin.Balance, 10, 64)
+			if err != nil {
+				return 0, err
+			}
+			totalBalance += balance
+		}
+
+		if !resp.HasNextPage {
+			break
+		}
+		cursor = resp.NextCursor
+	}
+
+	return totalBalance, nil
+}
+
+// FormatSuiBalance converts MIST to SUI with 9 decimal places
+func FormatSuiBalance(mist uint64) string {
+	suiVal := float64(mist) / nanoSUIPerSUI
+	return fmt.Sprintf("%.9f", suiVal)
+}

cmd/zetatool/chains/ton.go

@@ -0,0 +1,40 @@
+package chains
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/tonkeeper/tongo/ton"
+
+	tonrpc "github.com/zeta-chain/node/zetaclient/chains/ton/rpc"
+)
+
+const (
+	// nanoTONPerTON is the number of nanoTON in 1 TON (10^9)
+	nanoTONPerTON = 1_000_000_000
+)
+
+// GetTONGatewayBalance fetches the TON balance of the gateway contract
+// Returns the balance in nanoTON (1 TON = 10^9 nanoTON)
+func GetTONGatewayBalance(ctx context.Context, rpcURL string, gatewayAddress string) (uint64, error) {
+	accountID, err := ton.ParseAccountID(gatewayAddress)
+	if err != nil {
+		return 0, fmt.Errorf("failed to parse gateway address: %w", err)
+	}
+
+	// Create TON RPC client (chainID 0 is fine for balance queries)
+	client := tonrpc.New(rpcURL, 0)
+
+	account, err := client.GetAccountState(ctx, accountID)
+	if err != nil {
+		return 0, fmt.Errorf("failed to get account state: %w", err)
+	}
+
+	return account.Balance, nil
+}
+
+// FormatTONBalance converts nanoTON to TON with 9 decimal places
+func FormatTONBalance(nanoTON uint64) string {
+	tonVal := float64(nanoTON) / nanoTONPerTON
+	return fmt.Sprintf("%.9f", tonVal)
+}