mirror of https://github.com/miguelmota/cointop
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
251 lines
5.2 KiB
Go
251 lines
5.2 KiB
Go
package cointop
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/BurntSushi/toml"
|
|
)
|
|
|
|
var fileperm = os.FileMode(0644)
|
|
|
|
type config struct {
|
|
Shortcuts map[string]interface{} `toml:"shortcuts"`
|
|
Favorites map[string][]interface{} `toml:"favorites"`
|
|
Portfolio map[string]interface{} `toml:"portfolio"`
|
|
Currency interface{} `toml:"currency"`
|
|
DefaultView interface{} `toml:"defaultView"`
|
|
}
|
|
|
|
func (ct *Cointop) setupConfig() error {
|
|
err := ct.makeConfigDir()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = ct.makeConfigFile()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = ct.parseConfig()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = ct.loadShortcutsFromConfig()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = ct.loadFavoritesFromConfig()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = ct.loadPortfolioFromConfig()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = ct.loadCurrencyFromConfig()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = ct.loadDefaultViewFromConfig()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (ct *Cointop) configDirPath() string {
|
|
homedir := userHomeDir()
|
|
return fmt.Sprintf("%s%s", homedir, "/.cointop")
|
|
}
|
|
|
|
func (ct *Cointop) configPath() string {
|
|
return fmt.Sprintf("%v%v", ct.configDirPath(), "/config")
|
|
}
|
|
|
|
func (ct *Cointop) makeConfigDir() error {
|
|
path := ct.configDirPath()
|
|
if _, err := os.Stat(path); os.IsNotExist(err) {
|
|
_ = os.Mkdir(path, os.ModePerm)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (ct *Cointop) makeConfigFile() error {
|
|
path := ct.configPath()
|
|
if _, err := os.Stat(path); os.IsNotExist(err) {
|
|
fo, err := os.Create(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer fo.Close()
|
|
b, err := ct.configToToml()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if _, err := fo.Write(b); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (ct *Cointop) saveConfig() error {
|
|
ct.savemux.Lock()
|
|
defer ct.savemux.Unlock()
|
|
path := ct.configPath()
|
|
if _, err := os.Stat(path); err == nil {
|
|
b, err := ct.configToToml()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = ioutil.WriteFile(path, b, fileperm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (ct *Cointop) parseConfig() error {
|
|
var conf config
|
|
path := ct.configPath()
|
|
if _, err := toml.DecodeFile(path, &conf); err != nil {
|
|
return err
|
|
}
|
|
|
|
ct.config = conf
|
|
return nil
|
|
}
|
|
|
|
func (ct *Cointop) configToToml() ([]byte, error) {
|
|
shortcutsIfcs := map[string]interface{}{}
|
|
for k, v := range ct.shortcutkeys {
|
|
var i interface{} = v
|
|
shortcutsIfcs[k] = i
|
|
}
|
|
|
|
var favorites []interface{}
|
|
for k, ok := range ct.favorites {
|
|
if ok {
|
|
var i interface{} = k
|
|
favorites = append(favorites, i)
|
|
}
|
|
}
|
|
var favoritesbysymbol []interface{}
|
|
favoritesIfcs := map[string][]interface{}{
|
|
// DEPRECATED: favorites by 'symbol' is deprecated because of collisions. Kept for backward compatibility.
|
|
"symbols": favoritesbysymbol,
|
|
"names": favorites,
|
|
}
|
|
|
|
portfolioIfc := map[string]interface{}{}
|
|
for name := range ct.portfolio.Entries {
|
|
entry, ok := ct.portfolio.Entries[name]
|
|
if !ok || entry.Coin == "" {
|
|
continue
|
|
}
|
|
var i interface{} = entry.Holdings
|
|
portfolioIfc[entry.Coin] = i
|
|
}
|
|
|
|
var currencyIfc interface{} = ct.currencyconversion
|
|
var defaultViewIfc interface{} = ct.defaultView
|
|
|
|
var inputs = &config{
|
|
Shortcuts: shortcutsIfcs,
|
|
Favorites: favoritesIfcs,
|
|
Portfolio: portfolioIfc,
|
|
Currency: currencyIfc,
|
|
DefaultView: defaultViewIfc,
|
|
}
|
|
|
|
var b bytes.Buffer
|
|
encoder := toml.NewEncoder(&b)
|
|
err := encoder.Encode(inputs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return b.Bytes(), nil
|
|
}
|
|
|
|
func (ct *Cointop) loadShortcutsFromConfig() error {
|
|
for k, ifc := range ct.config.Shortcuts {
|
|
if v, ok := ifc.(string); ok {
|
|
if !ct.actionExists(v) {
|
|
continue
|
|
}
|
|
if ct.shortcutkeys[k] == "" {
|
|
continue
|
|
}
|
|
ct.shortcutkeys[k] = v
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (ct *Cointop) loadCurrencyFromConfig() error {
|
|
if currency, ok := ct.config.Currency.(string); ok {
|
|
ct.currencyconversion = strings.ToUpper(currency)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (ct *Cointop) loadDefaultViewFromConfig() error {
|
|
if defaultView, ok := ct.config.DefaultView.(string); ok {
|
|
defaultView = strings.ToLower(defaultView)
|
|
switch defaultView {
|
|
case "portfolio":
|
|
ct.portfoliovisible = true
|
|
case "favorites":
|
|
ct.filterByFavorites = true
|
|
case "default":
|
|
fallthrough
|
|
default:
|
|
ct.portfoliovisible = false
|
|
ct.filterByFavorites = false
|
|
defaultView = "default"
|
|
}
|
|
ct.defaultView = defaultView
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (ct *Cointop) loadFavoritesFromConfig() error {
|
|
for k, arr := range ct.config.Favorites {
|
|
// DEPRECATED: favorites by 'symbol' is deprecated because of collisions. Kept for backward compatibility.
|
|
if k == "symbols" {
|
|
for _, ifc := range arr {
|
|
if v, ok := ifc.(string); ok {
|
|
ct.favoritesbysymbol[strings.ToUpper(v)] = true
|
|
}
|
|
}
|
|
} else if k == "names" {
|
|
for _, ifc := range arr {
|
|
if v, ok := ifc.(string); ok {
|
|
ct.favorites[v] = true
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (ct *Cointop) loadPortfolioFromConfig() error {
|
|
for name, holdingsIfc := range ct.config.Portfolio {
|
|
var holdings float64
|
|
var ok bool
|
|
if holdings, ok = holdingsIfc.(float64); !ok {
|
|
if holdingsInt, ok := holdingsIfc.(int64); ok {
|
|
holdings = float64(holdingsInt)
|
|
}
|
|
}
|
|
|
|
ct.setPortfolioEntry(name, holdings)
|
|
}
|
|
return nil
|
|
}
|