Add voucher list handler

dev/api.go

@@ -9,6 +9,7 @@ import (
 
 	"github.com/gofrs/uuid"
 	"git.grassecon.net/grassrootseconomics/sarafu-api/models"
+	dataserviceapi "github.com/grassrootseconomics/ussd-data-service/pkg/api"
 )
 
 const (
@@ -16,10 +17,22 @@ const (
 )
 
 type account struct {
-	balance int
+	track string
 	nonce int
+	defaultVoucher string
+	balances map[string]int
 }
 
+type voucher struct {
+	address string
+	symbol string
+	decimals int
+}
+
+var (
+	vouchers = make(map[string]voucher)
+)
+
 type DevAccountService struct {
 	accounts map[string]account
 	accountsTrack map[string]string
@@ -39,8 +52,15 @@ func (das *DevAccountService) CheckBalance(ctx context.Context, publicKey string
 	if !ok {
 		return nil, fmt.Errorf("account not found (publickey): %v", publicKey)
 	}
+	if acc.defaultVoucher == "" {
+		return nil, fmt.Errorf("no default voucher set for: %v", publicKey)
+	}
+	bal, ok := acc.balances[acc.defaultVoucher]
+	if !ok {
+		return nil, fmt.Errorf("balance not found for default token %s pubkey %v", acc.defaultVoucher, publicKey)
+	}
 	return &models.BalanceResult {
-		Balance: strconv.Itoa(acc.balance),
+		Balance: strconv.Itoa(bal),
 		Nonce: json.Number(strconv.Itoa(acc.nonce)),
 	}, nil
 }
@@ -60,10 +80,44 @@ func (das *DevAccountService) CreateAccount(ctx context.Context) (*models.Accoun
 		return nil, fmt.Errorf("short read: %d", c)
 	}
 	pubKey := fmt.Sprintf("0x%x", b)
-	das.accounts[pubKey] = account{}
+	das.accounts[pubKey] = account{
+		track: uid.String(),
+	}
 	das.accountsTrack[uid.String()] = pubKey
 	return &models.AccountResult{
 		PublicKey: pubKey,
 		TrackingId: uid.String(),
 	}, nil
 }
+
+func (das *DevAccountService) TrackAccountStatus(ctx context.Context, publicKey string) (*models.TrackStatusResult, error) {
+	var ok bool
+	_, ok = das.accounts[publicKey]
+	if !ok {
+		return nil, fmt.Errorf("account not found (publickey): %v", publicKey)
+	}
+	return &models.TrackStatusResult{
+		Active: true,
+	}, nil
+}
+
+func (das *DevAccountService) FetchVouchers(ctx context.Context, publicKey string) ([]dataserviceapi.TokenHoldings, error) {
+	var holdings []dataserviceapi.TokenHoldings
+	acc, ok := das.accounts[publicKey]
+	if !ok {
+		return nil, fmt.Errorf("account not found (publickey): %v", publicKey)
+	}
+	for k, v := range(acc.balances) {
+		voucher, ok := vouchers[k]
+		if !ok {
+			return nil, fmt.Errorf("voucher has balance but object not found: %v", k)
+		}
+		holdings = append(holdings, dataserviceapi.TokenHoldings{
+			ContractAddress: voucher.address,
+			TokenSymbol: voucher.symbol,
+			TokenDecimals: strconv.Itoa(voucher.decimals),
+			Balance: strconv.Itoa(v),
+		})
+	}
+	return holdings, nil
+}