view balances of "quickview" configured accounts on main page

Author: howeyc

cmd/lweb/handler_accounts.go

@@ -7,9 +7,76 @@ import (
 
 	"github.com/howeyc/ledger"
 
+	"github.com/BurntSushi/toml"
 	"github.com/go-martini/martini"
 )
 
+type quickviewAccountConfig struct {
+	Name      string
+	ShortName string `toml:"short_name"`
+}
+
+type quickviewConfigStruct struct {
+	Accounts []quickviewAccountConfig `toml:"account"`
+}
+
+func quickviewHandler(w http.ResponseWriter, r *http.Request) {
+	if len(quickviewConfigFileName) == 0 {
+		http.Redirect(w, r, "/accounts", http.StatusFound)
+	}
+	var quickviewConfigData quickviewConfigStruct
+	if _, lerr := toml.DecodeFile(quickviewConfigFileName, &quickviewConfigData); lerr != nil || len(quickviewConfigData.Accounts) < 1 {
+		http.Redirect(w, r, "/accounts", http.StatusFound)
+	}
+
+	shorten := func(accname string) string {
+		for _, qvc := range quickviewConfigData.Accounts {
+			if qvc.Name == accname {
+				return qvc.ShortName
+			}
+		}
+		return abbrev(accname)
+	}
+
+	funcMap := template.FuncMap{
+		"abbrev": shorten,
+	}
+
+	t, err := parseAssetsWithFunc(funcMap, "templates/template.accounts.html", "templates/template.nav.html")
+	if err != nil {
+		http.Error(w, err.Error(), 500)
+		return
+	}
+
+	trans, terr := getTransactions()
+	if terr != nil {
+		http.Error(w, terr.Error(), 500)
+		return
+	}
+
+	var pData pageData
+	pData.Reports = reportConfigData.Reports
+	pData.Portfolios = portfolioConfigData.Portfolios
+	pData.Transactions = trans
+
+	includeNames := make(map[string]bool)
+	for _, qvc := range quickviewConfigData.Accounts {
+		includeNames[qvc.Name] = true
+	}
+
+	balances := ledger.GetBalances(trans, []string{})
+	for _, bal := range balances {
+		if includeNames[bal.Name] {
+			pData.Accounts = append(pData.Accounts, bal)
+		}
+	}
+
+	err = t.Execute(w, pData)
+	if err != nil {
+		http.Error(w, err.Error(), 500)
+	}
+}
+
 func accountsHandler(w http.ResponseWriter, r *http.Request) {
 	funcMap := template.FuncMap{
 		"abbrev": abbrev,

cmd/lweb/main.go

@@ -22,6 +22,7 @@ import (
 var ledgerFileName string
 var reportConfigFileName string
 var stockConfigFileName string
+var quickviewConfigFileName string
 var ledgerLock sync.Mutex
 var currentSum []byte
 var currentTrans []*ledger.Transaction
@@ -139,6 +140,7 @@ func main() {
 	flag.StringVar(&ledgerFileName, "f", "", "Ledger file name (*Required).")
 	flag.StringVar(&reportConfigFileName, "r", "", "Report config file name (*Required).")
 	flag.StringVar(&stockConfigFileName, "s", "", "Stock config file name (*Optional).")
+	flag.StringVar(&quickviewConfigFileName, "q", "", "Quickview config file name (*Optional).")
 	flag.IntVar(&serverPort, "port", 8056, "Port to listen on.")
 	flag.BoolVar(&localhost, "localhost", false, "Listen on localhost only.")
 
@@ -181,9 +183,7 @@ func main() {
 	m.Get("/portfolio/:portfolioName", portfolioHandler)
 	m.Get("/account/:accountName", accountHandler)
 	m.Get("/report/:reportName", reportHandler)
-	m.Get("/", func(w http.ResponseWriter, r *http.Request) {
-		http.Redirect(w, r, "/accounts", http.StatusFound)
-	})
+	m.Get("/", quickviewHandler)
 
 	fmt.Println("Listening on port", serverPort)
 	var listenAddress string

cmd/lweb/quickview-sample.toml

@@ -0,0 +1,11 @@
+[[account]]
+name = "Assets:Current Assets:TD Bank:Checking"
+short_name = "TD Checking"
+
+[[account]]
+name = "Assets:Current Assets:Wallet"
+short_name = "Wallet"
+
+[[account]]
+name = "Liabilities:Credit Card:TD Visa"
+short_name = "TD Visa"