cli/pkg/utils/network_balance.go

// Copyright (C) 2022-2025, Lux Industries Inc. All rights reserved.
// See the file LICENSE for licensing terms.

package utils

import (
	"context"
	"fmt"

	"github.com/luxfi/ids"
	"github.com/luxfi/sdk/models"
	platformvm "github.com/luxfi/sdk/platformvm"
)

// GetNetworkBalance returns the balance of an address on the P-chain
func GetNetworkBalance(address ids.ShortID, network models.Network) (uint64, error) {
	pClient := platformvm.NewClient(network.Endpoint())
	ctx := context.Background()

	response, err := pClient.GetBalance(ctx, []ids.ShortID{address})
	if err != nil {
		return 0, fmt.Errorf("failed to get balance: %w", err)
	}

	return uint64(response.Unlocked), nil
}