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
}