Merge pull request #743 from pastelnetwork/PSL-1031_checkSNBalance

[PSL-1031] check SN balance before reg ops

Author: mateeullahmalik

dupedetection/dd-server.pb.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/base64"
 	"fmt"
+	"math"
 	"net"
 	"strconv"
 	"strings"
@@ -806,6 +807,30 @@ func (client *client) MasterNodesExtra(ctx context.Context) (MasterNodes, error)
 	return masterNodes, nil
 }
 
+// ZGetTotalBalance returns total balance
+// Command `z_gettotalbalance`
+func (client *client) ZGetTotalBalance(ctx context.Context) (*GetTotalBalanceResponse, error) {
+	resp := &GetTotalBalanceResponse{}
+
+	if err := client.callFor(ctx, &resp, "z_gettotalbalance"); err != nil {
+		return nil, errors.Errorf("failed to get total balance: %w", err)
+	}
+
+	return resp, nil
+}
+
+// NFTStorageFee returns the fee of NFT storage
+// Command `tickets tools estimatenftstoragefee <sizeInMB>`
+func (client *client) NFTStorageFee(ctx context.Context, sizeInMB float64) (*NFTStorageFeeEstimate, error) {
+	resp := &NFTStorageFeeEstimate{}
+
+	if err := client.callFor(ctx, &resp, "tickets", "tools", "estimatenftstoragefee", int(math.Ceil(sizeInMB))); err != nil {
+		return nil, errors.Errorf("failed to get estimated nft storage: %w", err)
+	}
+
+	return resp, nil
+}
+
 // GetRawMempool returns the list of in-progress transaction ids
 func (client *client) GetRawMempool(ctx context.Context) ([]string, error) {
 	var transactionIDs []string

dupedetection/dd-server_grpc.pb.go

@@ -134,3 +134,17 @@ type GetActionFeesResult struct {
 	SenseFee   float64 `json:"sensefee"`
 	CascadeFee float64 `json:"cascadefee"`
 }
+
+// GetTotalBalanceResponse ...
+type GetTotalBalanceResponse struct {
+	Transparent float64 `json:"transparent"`
+	Private     float64 `json:"private"`
+	Total       float64 `json:"total"`
+}
+
+// NFTStorageFeeEstimate ...
+type NFTStorageFeeEstimate struct {
+	EstimatedNftStorageFeeMin     float64 `json:"estimatedNftStorageFeeMin"`
+	EstimatedNftStorageFeeAverage float64 `json:"estimatedNftStorageFeeAverage"`
+	EstimatedNftStorageFeeMax     float64 `json:"estimatedNftStorageFeeMax"`
+}

pastel/client.go

@@ -231,4 +231,12 @@ type Client interface {
 
 	// BurnAddress ...
 	BurnAddress() string
+
+	//ZGetTotalBalance returns total balance
+	//Command `z_gettotalbalance`
+	ZGetTotalBalance(ctx context.Context) (*GetTotalBalanceResponse, error)
+
+	//NFTStorageFee returns the fee of NFT storage
+	//Command `tickets tools estimatenftstoragefee <sizeInMB>`
+	NFTStorageFee(ctx context.Context, sizeInMB float64) (*NFTStorageFeeEstimate, error)
 }