cointop/cointop/table.go

package cointop

import (
	"fmt"
	"math"
	"strconv"
	"strings"
	"time"

	"github.com/miguelmota/cointop/pkg/color"
	"github.com/miguelmota/cointop/pkg/humanize"
	"github.com/miguelmota/cointop/pkg/table"
)

func (ct *Cointop) refreshTable() error {
	maxX := ct.Width()
	ct.table = table.New().SetWidth(maxX)
	ct.table.AddCol("")
	ct.table.AddCol("")
	ct.table.AddCol("")
	ct.table.AddCol("")
	ct.table.AddCol("")
	ct.table.AddCol("")
	ct.table.AddCol("")
	ct.table.AddCol("")
	ct.table.AddCol("")
	ct.table.AddCol("")
	ct.table.AddCol("")
	ct.table.AddCol("")
	ct.table.HideColumHeaders = true
	for _, coin := range ct.coins {
		unix, _ := strconv.ParseInt(coin.LastUpdated, 10, 64)
		lastUpdated := time.Unix(unix, 0).Format("15:04:05 Jan 02")
		colorprice := color.Cyan
		color1h := color.White
		color24h := color.White
		color7d := color.White
		if coin.PercentChange1H > 0 {
			color1h = color.Green
		}
		if coin.PercentChange1H < 0 {
			color1h = color.Red
		}
		if coin.PercentChange24H > 0 {
			color24h = color.Green
		}
		if coin.PercentChange24H < 0 {
			color24h = color.Red
		}
		if coin.PercentChange7D > 0 {
			color7d = color.Green
		}
		if coin.PercentChange7D < 0 {
			color7d = color.Red
		}
		name := coin.Name
		lastchar := len(name)
		if lastchar > 20 {
			lastchar = 20
			name = fmt.Sprintf("%s...", name[0:18])
		}
		ct.table.AddRow(
			color.White(fmt.Sprintf("%7v ", coin.Rank)),
			color.White(fmt.Sprintf("%.22s", name)),
			color.White(fmt.Sprintf("%.6s", coin.Symbol)),
			colorprice(fmt.Sprintf("%12s", humanize.Commaf(coin.PriceUSD))),
			color.White(fmt.Sprintf("%17s", humanize.Commaf(coin.MarketCapUSD))),
			color.White(fmt.Sprintf("%15s", humanize.Commaf(coin.USD24HVolume))),
			color1h(fmt.Sprintf("%8.2f%%", coin.PercentChange1H)),
			color24h(fmt.Sprintf("%8.2f%%", coin.PercentChange24H)),
			color7d(fmt.Sprintf("%8.2f%%", coin.PercentChange7D)),
			color.White(fmt.Sprintf("%21s", humanize.Commaf(coin.TotalSupply))),
			color.White(fmt.Sprintf("%18s", humanize.Commaf(coin.AvailableSupply))),
			color.White(fmt.Sprintf("%18s", lastUpdated)),
			// add %percent of cap
		)
	}

	ct.update(func() {
		ct.tableview.Clear()
		ct.table.Format().Fprint(ct.tableview)
	})

	return nil
}

func (ct *Cointop) updateTable() error {
	start := ct.page * ct.perpage
	end := start + ct.perpage
	if start >= len(ct.allCoins())-1 {
		start = 0
	}
	if end >= len(ct.allCoins())-1 {
		start = int(math.Floor(float64(start/100)) * 100)

		end = len(ct.allCoins()) - 1
	}
	if start < 0 {
		start = 0
	}
	if end >= len(ct.allCoins()) {
		end = len(ct.allCoins()) - 1
	}
	sliced := ct.allCoins()[start:end]
	ct.coins = sliced
	ct.sort(ct.sortby, ct.sortdesc, ct.coins)
	ct.refreshTable()
	return nil
}

func (ct *Cointop) highlightedRowIndex() int {
	_, y := ct.tableview.Origin()
	_, cy := ct.tableview.Cursor()
	idx := y + cy
	if idx < 0 {
		idx = 0
	}
	if idx >= len(ct.coins) {
		idx = len(ct.coins) - 1
	}
	return idx
}

func (ct *Cointop) highlightedRowCoin() *coin {
	idx := ct.highlightedRowIndex()
	return ct.coins[idx]
}

func (ct *Cointop) rowLink() string {
	slug := strings.ToLower(strings.Replace(ct.highlightedRowCoin().Name, " ", "-", -1))
	return fmt.Sprintf("https://coinmarketcap.com/currencies/%s", slug)
}

func (ct *Cointop) allCoins() []*coin {
	if ct.filterByFavorites {
		var list []*coin
		for i := range ct.allcoins {
			coin := ct.allcoins[i]
			if coin.Favorite {
				list = append(list, coin)
			}
		}
		return list
	}

	return ct.allcoins
}