gocryptotrader/cmd/portfolio/portfolio.go

package main

import (
	"flag"
	"fmt"
	"log"
	"os"

	"github.com/thrasher-corp/gocryptotrader/config"
	"github.com/thrasher-corp/gocryptotrader/currency"
	"github.com/thrasher-corp/gocryptotrader/exchanges/asset"
	"github.com/thrasher-corp/gocryptotrader/exchanges/bitfinex"
	"github.com/thrasher-corp/gocryptotrader/portfolio"
)

var (
	priceMap        map[currency.Code]float64
	displayCurrency currency.Code
)

func printSummary(msg string, amount float64) {
	log.Println()
	log.Println(fmt.Sprintf("%s in USD: $%.2f", msg, amount))

	if displayCurrency != currency.USD {
		conv, err := currency.ConvertCurrency(amount,
			currency.USD,
			displayCurrency)
		if err != nil {
			log.Println(err)
		} else {
			symb, err := currency.GetSymbolByCurrencyName(displayCurrency)
			if err != nil {
				log.Println(fmt.Sprintf("%s in %s: %.2f",
					msg,
					displayCurrency,
					conv))
			} else {
				log.Println(fmt.Sprintf("%s in %s: %s%.2f",
					msg,
					displayCurrency,
					symb,
					conv))
			}
		}
	}
}

func getOnlineOfflinePortfolio(coins []portfolio.Coin, online bool) {
	var totals float64
	for _, x := range coins {
		value := priceMap[x.Coin] * x.Balance
		totals += value
		log.Printf("\t%v %v Subtotal: $%.2f Coin percentage: %.2f%%\n", x.Coin,
			x.Balance, value, x.Percentage)
	}
	if !online {
		printSummary("\tOffline balance", totals)
	} else {
		printSummary("\tOnline balance", totals)
	}
}

func main() {
	var inFile, key string

	defaultCfg, err := config.GetFilePath("")
	if err != nil {
		log.Println(err)
		os.Exit(1)
	}

	flag.StringVar(&inFile, "infile", defaultCfg, "The config input file to process.")
	flag.StringVar(&key, "key", "", "The key to use for AES encryption.")
	flag.Parse()

	log.Println("GoCryptoTrader: portfolio tool.")

	var cfg config.Config
	err = cfg.LoadConfig(inFile, true)
	if err != nil {
		log.Println(err)
		os.Exit(1)
	}
	log.Println("Loaded config file.")

	displayCurrency = cfg.Currency.FiatDisplayCurrency
	port := portfolio.Base{}
	port.Seed(cfg.Portfolio)
	result := port.GetPortfolioSummary()

	log.Println("Fetched portfolio data.")

	type PortfolioTemp struct {
		Balance  float64
		Subtotal float64
	}

	cfg.RetrieveConfigCurrencyPairs(true, asset.Spot)
	portfolioMap := make(map[currency.Code]PortfolioTemp)
	total := float64(0)

	log.Println("Fetching currency data..")
	var fiatCurrencies []currency.Code
	for _, y := range result.Totals {
		if y.Coin.IsFiatCurrency() {
			fiatCurrencies = append(fiatCurrencies, y.Coin)
		}
	}
	err = currency.SeedForeignExchangeData(fiatCurrencies)
	if err != nil {
		log.Println(err)
		os.Exit(1)
	}

	log.Println("Fetched currency data.")
	log.Println("Fetching ticker data and calculating totals..")
	priceMap = make(map[currency.Code]float64)
	priceMap[currency.USD] = 1

	for _, y := range result.Totals {
		pf := PortfolioTemp{}
		pf.Balance = y.Balance
		pf.Subtotal = 0

		if y.Coin.IsDefaultFiatCurrency() {
			if y.Coin != currency.USD {
				conv, err := currency.ConvertCurrency(y.Balance, y.Coin, currency.USD)
				if err != nil {
					log.Println(err)
				} else {
					priceMap[y.Coin] = conv / y.Balance
					pf.Subtotal = conv
				}
			} else {
				pf.Subtotal = y.Balance
			}
		} else {
			bf := bitfinex.Bitfinex{}
			bf.SetDefaults()
			bf.Verbose = false
			ticker, errf := bf.GetTicker(y.Coin.String() + currency.USD.String())
			if errf != nil {
				log.Println(errf)
			} else {
				priceMap[y.Coin] = ticker.Last
				pf.Subtotal = ticker.Last * y.Balance
			}
		}
		portfolioMap[y.Coin] = pf
		total += pf.Subtotal
	}
	log.Println("Done.")
	log.Println()
	log.Println("PORTFOLIO TOTALS:")
	for x, y := range portfolioMap {
		log.Printf("\t%s Amount: %f Subtotal: $%.2f USD (1 %s = $%.2f USD). Percentage of portfolio %.3f%%", x, y.Balance, y.Subtotal, x, y.Subtotal/y.Balance, y.Subtotal/total*100/1)
	}
	printSummary("\tTotal balance", total)

	log.Println("OFFLINE COIN TOTALS:")
	getOnlineOfflinePortfolio(result.Offline, false)

	log.Println("ONLINE COIN TOTALS:")
	getOnlineOfflinePortfolio(result.Online, true)

	log.Println("OFFLINE COIN SUMMARY:")
	var totals float64
	for x, y := range result.OfflineSummary {
		log.Printf("\t%s:", x)
		totals = 0
		for z := range y {
			value := priceMap[x] * y[z].Balance
			totals += value
			log.Printf("\t %s Amount: %f Subtotal: $%.2f Coin percentage: %.2f%%\n",
				y[z].Address, y[z].Balance, value, y[z].Percentage)
		}
		printSummary(fmt.Sprintf("\t %s balance", x), totals)
	}

	log.Println("ONLINE COINS SUMMARY:")
	for x, y := range result.OnlineSummary {
		log.Printf("\t%s:", x)
		totals = 0
		for z, w := range y {
			value := priceMap[z] * w.Balance
			totals += value
			log.Printf("\t %s Amount: %f Subtotal $%.2f Coin percentage: %.2f%%",
				z, w.Balance, value, w.Percentage)
		}
		printSummary("\t Exchange balance", totals)
	}
}