Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion cmd/sim/evm/activity.go
Original file line number Diff line number Diff line change
Expand Up @@ -152,7 +152,7 @@ func runActivity(cmd *cobra.Command, args []string) error {
a.Type,
a.AssetType,
activitySymbol(a),
formatUSD(a.ValueUSD),
output.FormatUSD(a.ValueUSD),
truncateHash(a.TxHash),
a.BlockTime,
}
Expand Down
4 changes: 2 additions & 2 deletions cmd/sim/evm/balance.go
Original file line number Diff line number Diff line change
Expand Up @@ -89,8 +89,8 @@ func runBalance(cmd *cobra.Command, args []string) error {
}
fmt.Fprintf(w, "Decimals: %d\n", b.Decimals)
fmt.Fprintf(w, "Amount: %s\n", formatAmount(b.Amount, b.Decimals))
fmt.Fprintf(w, "Price USD: %s\n", formatUSD(b.PriceUSD))
fmt.Fprintf(w, "Value USD: %s\n", formatUSD(b.ValueUSD))
fmt.Fprintf(w, "Price USD: %s\n", output.FormatUSD(b.PriceUSD))
fmt.Fprintf(w, "Value USD: %s\n", output.FormatUSD(b.ValueUSD))

return nil
}
Expand Down
12 changes: 2 additions & 10 deletions cmd/sim/evm/balances.go
Original file line number Diff line number Diff line change
Expand Up @@ -183,8 +183,8 @@ func runBalancesEndpoint(cmd *cobra.Command, args []string, pathPrefix, pathSuff
b.Chain,
b.Symbol,
formatAmount(b.Amount, b.Decimals),
formatUSD(b.PriceUSD),
formatUSD(b.ValueUSD),
output.FormatUSD(b.PriceUSD),
output.FormatUSD(b.ValueUSD),
}
}
output.PrintTable(w, columns, rows)
Expand Down Expand Up @@ -245,14 +245,6 @@ func formatAmount(raw string, decimals int) string {
return intPart + "." + fracPart
}

// formatUSD formats a USD value for display.
func formatUSD(v float64) string {
if v == 0 {
return "0.00"
}
return fmt.Sprintf("%.2f", v)
}

// printBalanceErrors writes balance-level errors to stderr.
func printBalanceErrors(cmd *cobra.Command, errs *balanceErrors) {
if errs == nil {
Expand Down
6 changes: 3 additions & 3 deletions cmd/sim/evm/defi_positions.go
Original file line number Diff line number Diff line change
Expand Up @@ -150,7 +150,7 @@ func runDefiPositions(cmd *cobra.Command, args []string) error {
p.Type,
fmt.Sprintf("%d", p.ChainID),
p.Protocol,
formatUSD(p.USDVal),
output.FormatUSD(p.USDVal),
positionDetails(p),
}
}
Expand Down Expand Up @@ -233,7 +233,7 @@ func printAggregations(w io.Writer, agg *defiAggregations) {
return
}

fmt.Fprintf(w, "\nTotal USD Value: %s\n", formatUSD(agg.TotalUSDValue))
fmt.Fprintf(w, "\nTotal USD Value: %s\n", output.FormatUSD(agg.TotalUSDValue))

if len(agg.TotalByChain) > 0 {
fmt.Fprintln(w, "Breakdown by chain:")
Expand All @@ -253,7 +253,7 @@ func printAggregations(w io.Writer, agg *defiAggregations) {
})

for _, cid := range chainIDs {
fmt.Fprintf(w, " Chain %s: %s\n", cid, formatUSD(agg.TotalByChain[cid]))
fmt.Fprintf(w, " Chain %s: %s\n", cid, output.FormatUSD(agg.TotalByChain[cid]))
}
}
}
6 changes: 3 additions & 3 deletions cmd/sim/evm/token_info.go
Original file line number Diff line number Diff line change
Expand Up @@ -111,18 +111,18 @@ func runTokenInfo(cmd *cobra.Command, args []string) error {
fmt.Fprintf(w, "Name: %s\n", t.Name)
}
fmt.Fprintf(w, "Decimals: %d\n", t.Decimals)
fmt.Fprintf(w, "Price USD: %s\n", formatUSD(t.PriceUSD))
fmt.Fprintf(w, "Price USD: %s\n", output.FormatUSD(t.PriceUSD))
if t.TotalSupply != "" {
fmt.Fprintf(w, "Total Supply: %s\n", t.TotalSupply)
}
if t.MarketCap > 0 {
fmt.Fprintf(w, "Market Cap: %s\n", formatUSD(t.MarketCap))
fmt.Fprintf(w, "Market Cap: %s\n", output.FormatUSD(t.MarketCap))
}
if t.Logo != "" {
fmt.Fprintf(w, "Logo: %s\n", t.Logo)
}
for _, hp := range t.HistoricalPrices {
fmt.Fprintf(w, "Price %dh ago: %s\n", hp.OffsetHours, formatUSD(hp.PriceUSD))
fmt.Fprintf(w, "Price %dh ago: %s\n", hp.OffsetHours, output.FormatUSD(hp.PriceUSD))
}
}

Expand Down
128 changes: 128 additions & 0 deletions cmd/sim/svm/balances.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,128 @@
package svm

import (
"encoding/json"
"fmt"
"net/url"

"github.com/spf13/cobra"

"github.com/duneanalytics/cli/output"
)

// NewBalancesCmd returns the `sim svm balances` command.
func NewBalancesCmd() *cobra.Command {
cmd := &cobra.Command{
Use: "balances <address>",
Short: "Get SVM token balances for a wallet address",
Long: "Return token balances for the given SVM wallet address across\n" +
"Solana and Eclipse chains, including USD valuations.\n\n" +
"Examples:\n" +
" dune sim svm balances 86xCnPeV69n6t3DnyGvkKobf9FdN2H9oiVDdaMpo2MMY\n" +
" dune sim svm balances 86xCnPeV... --chains solana,eclipse\n" +
" dune sim svm balances 86xCnPeV... --limit 50 -o json",
Args: cobra.ExactArgs(1),
RunE: runBalances,
}

cmd.Flags().String("chains", "", "Comma-separated chains: solana, eclipse (default: solana)")
cmd.Flags().Int("limit", 0, "Max results (1-1000, default 1000)")
cmd.Flags().String("offset", "", "Pagination cursor from previous response")
output.AddFormatFlag(cmd, "text")

return cmd
}

// --- Response types ---

type svmBalancesResponse struct {
ProcessingTimeMs float64 `json:"processing_time_ms,omitempty"`
WalletAddress string `json:"wallet_address"`
NextOffset string `json:"next_offset,omitempty"`
BalancesCount float64 `json:"balances_count,omitempty"`
Balances []svmBalanceEntry `json:"balances"`
}

type svmBalanceEntry struct {
Chain string `json:"chain"`
Address string `json:"address"`
Amount string `json:"amount"`
Balance string `json:"balance,omitempty"`
RawBalance string `json:"raw_balance,omitempty"`
ValueUSD float64 `json:"value_usd,omitempty"`
ProgramID *string `json:"program_id,omitempty"`
Decimals float64 `json:"decimals,omitempty"`
TotalSupply string `json:"total_supply,omitempty"`
Name string `json:"name,omitempty"`
Symbol string `json:"symbol,omitempty"`
URI *string `json:"uri,omitempty"`
PriceUSD float64 `json:"price_usd,omitempty"`
LiquidityUSD float64 `json:"liquidity_usd,omitempty"`
PoolType *string `json:"pool_type,omitempty"`
PoolAddress *string `json:"pool_address,omitempty"`
MintAuthority *string `json:"mint_authority,omitempty"`
}

func runBalances(cmd *cobra.Command, args []string) error {
client, err := requireSimClient(cmd)
if err != nil {
return err
}

address := args[0]
params := url.Values{}

if v, _ := cmd.Flags().GetString("chains"); v != "" {
params.Set("chains", v)
}
if v, _ := cmd.Flags().GetInt("limit"); v > 0 {
params.Set("limit", fmt.Sprintf("%d", v))
}
if v, _ := cmd.Flags().GetString("offset"); v != "" {
params.Set("offset", v)
}

data, err := client.Get(cmd.Context(), "/beta/svm/balances/"+address, params)
if err != nil {
return err
}

w := cmd.OutOrStdout()
switch output.FormatFromCmd(cmd) {
case output.FormatJSON:
var raw json.RawMessage = data
return output.PrintJSON(w, raw)
default:
var resp svmBalancesResponse
if err := json.Unmarshal(data, &resp); err != nil {
return fmt.Errorf("parsing response: %w", err)
}

if len(resp.Balances) == 0 {
fmt.Fprintln(w, "No balances found.")
return nil
}

columns := []string{"CHAIN", "SYMBOL", "BALANCE", "PRICE_USD", "VALUE_USD"}
rows := make([][]string, len(resp.Balances))
for i, b := range resp.Balances {
bal := b.Balance
if bal == "" {
bal = b.Amount
}
rows[i] = []string{
b.Chain,
b.Symbol,
bal,
output.FormatUSD(b.PriceUSD),
output.FormatUSD(b.ValueUSD),
}
}
output.PrintTable(w, columns, rows)

if resp.NextOffset != "" {
fmt.Fprintf(w, "\nNext offset: %s\n", resp.NextOffset)
}
return nil
}
}
125 changes: 125 additions & 0 deletions cmd/sim/svm/balances_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,125 @@
package svm_test

import (
"bytes"
"encoding/json"
"testing"

"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)

func TestSvmBalances_Text(t *testing.T) {
key := simAPIKey(t)

root := newSimTestRoot()
var buf bytes.Buffer
root.SetOut(&buf)
root.SetArgs([]string{"sim", "--sim-api-key", key, "svm", "balances", svmTestAddress})

require.NoError(t, root.Execute())

out := buf.String()
assert.Contains(t, out, "CHAIN")
assert.Contains(t, out, "SYMBOL")
assert.Contains(t, out, "BALANCE")
assert.Contains(t, out, "PRICE_USD")
assert.Contains(t, out, "VALUE_USD")
}

func TestSvmBalances_JSON(t *testing.T) {
key := simAPIKey(t)

root := newSimTestRoot()
var buf bytes.Buffer
root.SetOut(&buf)
root.SetArgs([]string{"sim", "--sim-api-key", key, "svm", "balances", svmTestAddress, "-o", "json"})

require.NoError(t, root.Execute())

var resp map[string]interface{}
require.NoError(t, json.Unmarshal(buf.Bytes(), &resp))
assert.Contains(t, resp, "wallet_address")
assert.Contains(t, resp, "balances")

balances, ok := resp["balances"].([]interface{})
require.True(t, ok)
if len(balances) > 0 {
b, ok := balances[0].(map[string]interface{})
require.True(t, ok)
assert.Contains(t, b, "chain")
assert.Contains(t, b, "address")
assert.Contains(t, b, "amount")
}
}

func TestSvmBalances_WithChains(t *testing.T) {
key := simAPIKey(t)

root := newSimTestRoot()
var buf bytes.Buffer
root.SetOut(&buf)
root.SetArgs([]string{"sim", "--sim-api-key", key, "svm", "balances", svmTestAddress, "--chains", "solana", "-o", "json"})

require.NoError(t, root.Execute())

var resp map[string]interface{}
require.NoError(t, json.Unmarshal(buf.Bytes(), &resp))
assert.Contains(t, resp, "balances")

// All balances should be on solana chain.
balances, ok := resp["balances"].([]interface{})
require.True(t, ok)
for _, bal := range balances {
b, ok := bal.(map[string]interface{})
require.True(t, ok)
assert.Equal(t, "solana", b["chain"])
}
}

func TestSvmBalances_Limit(t *testing.T) {
key := simAPIKey(t)

root := newSimTestRoot()
var buf bytes.Buffer
root.SetOut(&buf)
root.SetArgs([]string{"sim", "--sim-api-key", key, "svm", "balances", svmTestAddress, "--limit", "3", "-o", "json"})

require.NoError(t, root.Execute())

var resp map[string]interface{}
require.NoError(t, json.Unmarshal(buf.Bytes(), &resp))

balances, ok := resp["balances"].([]interface{})
require.True(t, ok)
assert.LessOrEqual(t, len(balances), 3)
}

func TestSvmBalances_Pagination(t *testing.T) {
key := simAPIKey(t)

root := newSimTestRoot()
var buf bytes.Buffer
root.SetOut(&buf)
root.SetArgs([]string{"sim", "--sim-api-key", key, "svm", "balances", svmTestAddress, "--limit", "2", "-o", "json"})

require.NoError(t, root.Execute())

var resp map[string]interface{}
require.NoError(t, json.Unmarshal(buf.Bytes(), &resp))
assert.Contains(t, resp, "balances")

// If next_offset is present, fetch page 2.
if offset, ok := resp["next_offset"].(string); ok && offset != "" {
root2 := newSimTestRoot()
var buf2 bytes.Buffer
root2.SetOut(&buf2)
root2.SetArgs([]string{"sim", "--sim-api-key", key, "svm", "balances", svmTestAddress, "--limit", "2", "--offset", offset, "-o", "json"})

require.NoError(t, root2.Execute())

var resp2 map[string]interface{}
require.NoError(t, json.Unmarshal(buf2.Bytes(), &resp2))
assert.Contains(t, resp2, "balances")
}
}
31 changes: 31 additions & 0 deletions cmd/sim/svm/helpers_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
package svm_test

import (
"context"
"os"
"testing"

"github.com/duneanalytics/cli/cmd/sim"
"github.com/spf13/cobra"
)

// simAPIKey returns the DUNE_SIM_API_KEY env var or skips the test.
func simAPIKey(t *testing.T) string {
t.Helper()
key := os.Getenv("DUNE_SIM_API_KEY")
if key == "" {
t.Skip("DUNE_SIM_API_KEY not set, skipping e2e test")
}
return key
}

const svmTestAddress = "86xCnPeV69n6t3DnyGvkKobf9FdN2H9oiVDdaMpo2MMY"

// newSimTestRoot builds the full command tree: dune -> sim -> svm -> <subcommands>.
// Used for authenticated E2E tests. Pass the API key via --sim-api-key in SetArgs.
func newSimTestRoot() *cobra.Command {
root := &cobra.Command{Use: "dune"}
root.SetContext(context.Background())
root.AddCommand(sim.NewSimCmd())
return root
}
Loading
Loading