mop/lib/profile.go

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

import (
	"sort"
	"encoding/json"
	"io/ioutil"
	"os/user"
	"strings"
)

const moprc = "/.moprc"

type Profile struct {
	MarketRefresh	int
	QuotesRefresh	int
	Tickers         []string
	SortBy          string
	SortOrder       string
}

//-----------------------------------------------------------------------------
func (self *Profile) Initialize() *Profile {
	data, err := ioutil.ReadFile(self.default_file_name())
	if err != nil {
		// Set default values.
		self.MarketRefresh = 12
		self.QuotesRefresh = 5
		self.Tickers = []string{"AAPL", "C", "GOOG", "IBM", "KO", "ORCL", "V"}
		self.SortBy = "Ticker"
		self.SortOrder = "Desc"
		self.Save()
	} else {
		json.Unmarshal(data, self)
	}
	return self
}

//-----------------------------------------------------------------------------
func (self *Profile) Save() error {
	if data, err := json.Marshal(self); err != nil {
		return err
	} else {
		return ioutil.WriteFile(self.default_file_name(), data, 0644)
	}
}

//-----------------------------------------------------------------------------
func (self *Profile) Quotes() string {
	return strings.Join(self.Tickers, "+")
}

//-----------------------------------------------------------------------------
func (self *Profile) AddTickers(tickers []string) {
	existing := make(map[string]bool)

	for _, ticker := range self.Tickers {
		existing[ticker] = true
	}

	for _, ticker := range tickers {
		if _, found := existing[ticker]; !found {
			self.Tickers = append(self.Tickers, ticker)
		}
	}
	sort.Strings(self.Tickers)
	self.Save()
}

//-----------------------------------------------------------------------------
func (self *Profile) RemoveTickers(tickers []string) {
	for _, ticker := range tickers {
		for i, existing := range self.Tickers {
			if ticker == existing { // Requested ticker is there: remove i-th slice item.
				self.Tickers = append(self.Tickers[:i], self.Tickers[i+1:]...)
			}
		}
	}
	self.Save()
}

// private
//-----------------------------------------------------------------------------
func (self *Profile) default_file_name() string {
	usr, err := user.Current()
	if err != nil {
		panic(err)
	}
	return usr.HomeDir + moprc
}