main.go

package main

import (
	"context"
	"flag"
	"fmt"
	"strings"
	"wallet-info/client"
	"wallet-info/config"

	"github.com/charmbracelet/lipgloss"
	"github.com/charmbracelet/lipgloss/table"
	log "github.com/charmbracelet/log"
)

func main() {
	wallet := flag.String("wallet", "", "Wallet address (takes priority over value in config file if provided)")
	filter := flag.String("filter", "", "Filter by blockchain name")
	configPath := flag.String("config", "", "Path to configuration file (default: $HOME/wallet-info-config.yaml)")
	flag.Parse()

	cfg, err := config.LoadConfig(*configPath)
	if err != nil {
		log.Error("Error loading config:", "err", err)
		return
	}

	if *wallet == "" {
		*wallet = cfg.Wallet
	}
	if *wallet == "" {
		log.Error("Wallet must be provided")
		return
	}

	rows := buildRows(cfg, *wallet, *filter)

	t := createStyledTable().Headers("Wallet", "Chain", "Token", "Amount").Rows(rows...)
	fmt.Println(t)
}

func buildRows(cfg *config.Config, wallet, filter string) [][]string {
	var rows [][]string
	for _, blockchain := range cfg.Blockchains {
		if filter == "" || strings.Contains(filter, blockchain.Name) {
			rows = append(rows, fetchBlockchainData(blockchain, wallet)...)
		}
	}
	return rows
}

func fetchBlockchainData(blockchain config.Blockchain, wallet string) [][]string {
	var rows [][]string
	ctx := context.Background()
	evmClient := client.NewEvmChainClient(blockchain.RPC, false)

	mainBalance := evmClient.GetBalance(ctx, wallet, blockchain.Decimals)
	rows = append(rows, []string{wallet, blockchain.Name, blockchain.MainTokenLabel, mainBalance})

	for _, token := range blockchain.Tokens {
		tokenBalance := evmClient.GetBalanceForToken(ctx, wallet, token.Address, token.Decimals)
		rows = append(rows, []string{wallet, blockchain.Name, token.Label, tokenBalance})
	}
	return rows
}

func createStyledTable() *table.Table {
	return table.New().
		Border(lipgloss.NormalBorder()).
		BorderStyle(lipgloss.NewStyle().Foreground(lipgloss.Color("#1565c0"))).
		StyleFunc(func(row, col int) lipgloss.Style {
			width := map[int]int{0: 45, 1: 20, 2: 20}[col]
			if width == 0 {
				width = 40
			}

			style := lipgloss.NewStyle().Width(width)
			switch {
			case row == 0:
				return style.Foreground(lipgloss.Color("#f5f5f5")).Background(lipgloss.Color("#1565c0"))
			case row%2 == 0:
				return style.Foreground(lipgloss.Color("#303f9f"))
			default:
				return style.Foreground(lipgloss.Color("#388e3c"))
			}
		})
}