From e7062445f49ab5d8fdab2c111b638e4b942bbfe1 Mon Sep 17 00:00:00 2001
From: "user.name" <user.email>
Date: Wed, 29 Jan 2020 13:44:00 +0000
Subject: [PATCH] Adds support for displaying the correct currency of stocks

---
 layout.go       | 72 ++++++++++++++++++++++++++++++++-----------------
 yahoo_quotes.go |  2 ++
 2 files changed, 50 insertions(+), 24 deletions(-)

diff --git a/layout.go b/layout.go
index e7287d7..cac4fde 100644
--- a/layout.go
+++ b/layout.go
@@ -20,7 +20,7 @@ type Column struct {
 	width     int                 // Column width.
 	name      string              // The name of the field in the Stock struct.
 	title     string              // Column title to display in the header.
-	formatter func(string) string // Optional function to format the contents of the column.
+	formatter func(... string) string // Optional function to format the contents of the column.
 }
 
 // Layout is used to format and display all the collected data, i.e. market
@@ -146,7 +146,7 @@ func (layout *Layout) prettify(quotes *Quotes) []Stock {
 			value := reflect.ValueOf(&stock).Elem().FieldByName(column.name).String()
 			if column.formatter != nil {
 				// ex. value = currency(value)
-				value = column.formatter(value)
+				value = column.formatter(value, stock.Currency)
 			}
 			// ex. pretty[i].Change = layout.pad(value, 10)
 			reflect.ValueOf(&pretty[i]).Elem().FieldByName(column.name).SetString(layout.pad(value, column.width))
@@ -256,61 +256,85 @@ func arrowFor(column int, profile *Profile) string {
 }
 
 //-----------------------------------------------------------------------------
-func blank(str string) string {
-	if len(str) == 3 && str[0:3] == `N/A` {
+func blank(str... string) string {
+	if len(str) < 1 {
+		return "ERR"
+	}
+	if len(str[0]) == 3 && str[0][0:3] == `N/A` {
 		return `-`
 	}
 
-	return str
+	return str[0]
 }
 
 //-----------------------------------------------------------------------------
-func zero(str string) string {
-	if str == `0.00` {
+func zero(str... string) string {
+	if len(str) < 2{
+		return "ERR"
+	}
+	if str[0] == `0.00` {
 		return `-`
 	}
 
-	return currency(str)
+	return currency(str[0], str[1])
 }
 
 //-----------------------------------------------------------------------------
-func last(str string) string {
-	if len(str) >= 6 && str[0:6] == `N/A - ` {
-		return str[6:]
+func last(str... string) string {
+	if len(str) < 1 {
+		return "ERR"
+	}
+	if len(str[0]) >= 6 && str[0][0:6] == `N/A - ` {
+		return str[0][6:]
 	}
 
-	return percent(str)
+	return percent(str[0])
 }
 
 //-----------------------------------------------------------------------------
-func currency(str string) string {
-	if str == `N/A` || len(str) == 0 {
+func currency(str... string) string {
+	if len(str) < 2 {
+		return "ERR"
+	}
+	symbol := "$"
+	switch (str[1]){
+		case "EUR":
+			symbol = "€"
+			break
+		case "GBP":
+			symbol = "£"
+			break
+	}
+	if str[0] == `N/A` || len(str[0]) == 0 {
 		return `-`
 	}
-	if sign := str[0:1]; sign == `+` || sign == `-` {
-		return sign + `$` + str[1:]
+	if sign := str[0][0:1]; sign == `+` || sign == `-` {
+		return sign + symbol + str[0][1:]
 	}
 
-	return `$` + str
+	return symbol + str[0]
 }
 
 // Returns percent value truncated at 2 decimal points.
 //-----------------------------------------------------------------------------
-func percent(str string) string {
-	if str == `N/A` || len(str) == 0 {
+func percent(str... string) string {
+	if len(str) < 1 {
+		return "ERR"
+	}
+	if str[0] == `N/A` || len(str[0]) == 0 {
 		return `-`
 	}
 
-	split := strings.Split(str, ".")
+	split := strings.Split(str[0], ".")
 	if len(split) == 2 {
 		digits := len(split[1])
 		if digits > 2 {
 			digits = 2
 		}
-		str = split[0] + "." + split[1][0:digits]
+		str[0] = split[0] + "." + split[1][0:digits]
 	}
-	if str[len(str)-1] != '%' {
-		str += `%`
+	if str[0][len(str)-1] != '%' {
+		str[0] += `%`
 	}
-	return str
+	return str[0]
 }
diff --git a/yahoo_quotes.go b/yahoo_quotes.go
index 762b725..f7cd3fc 100644
--- a/yahoo_quotes.go
+++ b/yahoo_quotes.go
@@ -41,6 +41,7 @@ type Stock struct {
 	Yield      string `json:"trailingAnnualDividendYield"` // y: dividend yield.
 	MarketCap  string `json:"marketCap"` // j3: market cap real time.
 	MarketCapX string `json:"marketCap"` // j1: market cap (fallback when real time is N/A).
+	Currency   string `json:"currency"` // String code for currency of stock.
 	Advancing  bool   // True when change is >= $0.
 }
 
@@ -171,6 +172,7 @@ func (quotes *Quotes) parse2(body []byte) (*Quotes, error) {
 		quotes.stocks[i].MarketCap = result["marketCap"]
 		// TODO calculate rt?
 		quotes.stocks[i].MarketCapX = result["marketCap"]
+		quotes.stocks[i].Currency = result["currency"]
 
  		/*
 			fmt.Println(i)