mop/lib/format.go

// Copyright (c) 2013 by Michael Dvorkin. All Rights Reserved.
//=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=
package mop

import (
	"bytes"
	"fmt"
	"regexp"
	"strings"
	"text/template"
	"time"
)

//-----------------------------------------------------------------------------
func FormatMarket(m Market) string {
	markup := `{{.Dow.name}}: `
	if m.Dow[`change`][0:1] != "-" {
		markup += `<green>{{.Dow.change}} ({{.Dow.percent}})</green> at {{.Dow.latest}}, `
	} else {
		markup += `"{{.Dow.change}}" ({{.Dow.percent}}) at {{.Dow.latest}}, `
	}
	markup += `{{.Sp500.name}}: `
	if m.Sp500[`change`][0:1] != "-" {
		markup += `<green>{{.Sp500.change}} ({{.Sp500.percent}})</green> at {{.Sp500.latest}}, `
	} else {
		markup += `{{.Sp500.change}} ({{.Sp500.percent}}) at {{.Sp500.latest}}, `
	}
	markup += `{{.Nasdaq.name}}: `
	if m.Nasdaq[`change`][0:1] != "-" {
		markup += `<green>{{.Nasdaq.change}} ({{.Nasdaq.percent}})</green> at {{.Nasdaq.latest}}`
	} else {
		markup += `{{.Nasdaq.change}} ({{.Nasdaq.percent}}) at {{.Nasdaq.latest}}`
	}
	markup += "\n"
	markup += `{{.Advances.name}}: {{.Advances.nyse}} ({{.Advances.nysep}}) on NYSE and {{.Advances.nasdaq}} ({{.Advances.nasdaqp}}) on Nasdaq. `
	markup += `{{.Declines.name}}: {{.Declines.nyse}} ({{.Declines.nysep}}) on NYSE and {{.Declines.nasdaq}} ({{.Declines.nasdaqp}}) on Nasdaq`
	markup += "\n"
	markup += `New highs: {{.Highs.nyse}} on NYSE and {{.Highs.nasdaq}} on Nasdaq. `
	markup += `New lows: {{.Lows.nyse}} on NYSE and {{.Lows.nasdaq}} on Nasdaq.`
	template, err := template.New("market").Parse(markup)
	if err != nil {
		panic(err)
	}

	buffer := new(bytes.Buffer)
	err = template.Execute(buffer, m)
	if err != nil {
		panic(err)
	}

	return buffer.String()
}

//-----------------------------------------------------------------------------
func FormatQuotes(quotes Quotes) string {
	vars := struct {
		Now    string
		Header string
		Stocks Quotes
	}{
		time.Now().Format("3:04:05pm PST"),
		header(),
		prettify(quotes),
	}

	markup := `<right><white>{{.Now}}</white></right>



{{.Header}}
{{range .Stocks}}{{.Color}}{{.Ticker}} {{.LastTrade}} {{.Change}} {{.ChangePercent}} {{.Open}} {{.Low}} {{.High}} {{.Low52}} {{.High52}} {{.Volume}} {{.AvgVolume}} {{.PeRatio}} {{.Dividend}} {{.Yield}} {{.MarketCap}}
{{end}}...`

	template, err := template.New("quotes").Parse(markup)
	if err != nil {
		panic(err)
	}

	buffer := new(bytes.Buffer)
	err = template.Execute(buffer, vars)
	if err != nil {
		panic(err)
	}

	return buffer.String()
}

func header() string {
	str := fmt.Sprintf("%-7s ", "Ticker")
	str += fmt.Sprintf("%9s ", "Last")
	str += fmt.Sprintf("%9s ", "Change")
	str += fmt.Sprintf("%9s ", "%Change")
	str += fmt.Sprintf("%9s ", "Open")
	str += fmt.Sprintf("%9s ", "Low")
	str += fmt.Sprintf("%9s ", "High")
	str += fmt.Sprintf("%9s ", "52w Low")
	str += fmt.Sprintf("%9s ", "52w High")
	str += fmt.Sprintf("%10s ", "Volume")
	str += fmt.Sprintf("%10s ", "AvgVolume")
	str += fmt.Sprintf("%9s ", "P/E")
	str += fmt.Sprintf("%9s ", "Dividend")
	str += fmt.Sprintf("%9s ", "Yield")
	str += fmt.Sprintf("%10s", "MktCap")

	return str
}

func prettify(quotes Quotes) Quotes {
	pretty := make(Quotes, len(quotes))
	for i, q := range quotes {
		pretty[i].Ticker        = pad(q.Ticker, -7)
		pretty[i].LastTrade     = pad(with_currency(q.LastTrade), 9)
		pretty[i].Change        = pad(with_currency(q.Change), 9)
		pretty[i].ChangePercent = pad(last_of_pair(q.ChangePercent), 9)
		pretty[i].Open          = pad(with_currency(q.Open), 9)
		pretty[i].Low           = pad(with_currency(q.Low), 9)
		pretty[i].High          = pad(with_currency(q.High), 9)
		pretty[i].Low52         = pad(with_currency(q.Low52), 9)
		pretty[i].High52        = pad(with_currency(q.High52), 9)
		pretty[i].Volume        = pad(q.Volume, 10)
		pretty[i].AvgVolume     = pad(q.AvgVolume, 10)
		pretty[i].PeRatio       = pad(nullify(q.PeRatioX), 9)
		pretty[i].Dividend      = pad(with_currency(q.Dividend), 9)
		pretty[i].Yield         = pad(with_percent(q.Yield), 9)
		pretty[i].MarketCap     = pad(with_currency(q.MarketCapX), 10)
	}
	return pretty
}

func nullify(str string) string {
	if len(str) == 3 && str[0:3] == "N/A" {
		return "-"
	} else {
		return str
	}
}

func last_of_pair(str string) string {
	if len(str) >= 6 && str[0:6] != "N/A - " {
		return str
	} else {
		return str[6:]
	}
}

func with_currency(str string) string {
	if str == "N/A" || str == "0.00" {
		return "-"
	} else {
		switch str[0:1] {
		case "+", "-":
			return str[0:1] + "$" + str[1:]
		default:
			return "$" + str
		}
	}
}

func with_percent(str string) string {
	if str == "N/A" {
		return "-"
	} else {
		return str + "%"
	}
}

func colorize(str string) string {
	if str == "N/A" {
		return "-"
	} else if str[0:1] == "-" {
		return "<red>" + str + "</red>"
	} else {
		return "<green>" + str + "</green>"
	}
}

func ticker(str string, change string) string {
	if change[0:1] == "-" {
		return "<red>" + str + "</red>"
	} else {
		return "<green>" + str + "</green>"
	}
}

func pad(str string, width int) string {
	re := regexp.MustCompile(`(\.\d+)[MB]?$`)
	match := re.FindStringSubmatch(str)
	if len(match) > 0 {
		switch len(match[1]) {
		case 2:
			str = strings.Replace(str, match[1], match[1]+"0", 1)
		case 4, 5:
			str = strings.Replace(str, match[1], match[1][0:3], 1)
		}
	}

	return fmt.Sprintf("%*s", width, str)
}