commands_add.go

package main

import (
	"fmt"
	"strconv"
	"time"

	"github.com/distributeddesigns/currency"
	types "github.com/distributeddesigns/shared_types"
)

type addCmd struct {
	id     uint64
	userID string
	amount currency.Currency
}

func parseAddCmd(parts []string) addCmd {
	if len(parts) != 4 {
		abortTx("ADD needs 4 parts")
	}

	txID, err := strconv.ParseUint(parts[0], 10, 64)
	abortTxOnError(err, "Could not parse ID")

	amount, err := currency.NewFromString(parts[3])
	abortTxOnError(err, "Could not parse amount in transaction")

	return addCmd{
		id:     txID,
		userID: parts[2],
		amount: amount,
	}
}

func (a addCmd) Name() string {
	return fmt.Sprintf("[%d] ADD", a.id)
}

func (a addCmd) ToAuditEvent() types.AuditEvent {
	xmlElement := fmt.Sprintf(`
	<userCommand>
		<timestamp>%d</timestamp>
		<server>%s</server>
		<transactionNum>%d</transactionNum>
		<command>ADD</command>
		<username>%s</username>
		<funds>%.02f</funds>
	</userCommand>`,
		time.Now().UnixNano()/1e6, redisBaseKey, a.id, a.userID, a.amount.ToFloat(),
	)

	return types.AuditEvent{
		UserID:    a.userID,
		ID:        a.id,
		EventType: "command",
		Content:   xmlElement,
	}
}

func (a addCmd) Execute() {
	// Create an account if one does not exist
	if _, accountExists := accountStore[a.userID]; !accountExists {
		consoleLog.Infof("Creating account for %s", a.userID)
		accountStore[a.userID] = newAccountForUser(a.userID)
	}

	acct := accountStore[a.userID]

	consoleLog.Infof("Adding %s to %s", a.amount, a.userID)
	acct.Lock()
	acct.AddFunds(a.amount)
	acct.AddSummaryItem("Finished " + a.Name())
	acct.Unlock()

	consoleLog.Notice(" [✔] Finished", a.Name())
}