rove/pkg/accounts/accounts.go

package accounts

import (
	"fmt"
	"time"
)

// Account represents a registered user
type Account struct {
	// Name simply describes the account and must be unique
	Name string `json:"name"`

	// Data represents internal account data
	Data map[string]string `json:"data"`
}

// Accountant manages a set of accounts
type Accountant struct {
	Accounts map[string]Account `json:"accounts"`
}

// NewAccountant creates a new accountant
func NewAccountant() *Accountant {
	return &Accountant{
		Accounts: make(map[string]Account),
	}
}

// RegisterAccount adds an account to the set of internal accounts
func (a *Accountant) RegisterAccount(name string) (acc Account, err error) {

	// Set the account name
	acc.Name = name
	acc.Data = make(map[string]string)

	// Verify this acount isn't already registered
	for _, a := range a.Accounts {
		if a.Name == acc.Name {
			return Account{}, fmt.Errorf("account name already registered: %s", a.Name)
		}
	}

	// Set the creation time
	acc.Data["created"] = time.Now().String()

	// Simply add the account to the map
	a.Accounts[acc.Name] = acc

	return
}

// AssignData assigns data to an account
func (a *Accountant) AssignData(account string, key string, value string) error {

	// Find the account matching the ID
	if this, ok := a.Accounts[account]; ok {
		this.Data[key] = value
		a.Accounts[account] = this
	} else {
		return fmt.Errorf("no account found for id: %s", account)
	}

	return nil
}

// GetValue gets the rover rover for the account
func (a *Accountant) GetValue(account string, key string) (string, error) {
	// Find the account matching the ID
	this, ok := a.Accounts[account]
	if !ok {
		return "", fmt.Errorf("no account found for id: %s", account)
	}
	return this.Data[key], nil
}