Track details for fiat onramp purchases

Author: jeffyanta

go.mod

@@ -6,7 +6,7 @@ require (
 	firebase.google.com/go/v4 v4.8.0
 	github.com/aws/aws-sdk-go-v2 v0.17.0
 	github.com/bits-and-blooms/bloom/v3 v3.1.0
-	github.com/code-payments/code-protobuf-api v1.11.0
+	github.com/code-payments/code-protobuf-api v1.12.2
 	github.com/emirpasic/gods v1.12.0
 	github.com/envoyproxy/protoc-gen-validate v0.1.0
 	github.com/golang-jwt/jwt/v5 v5.0.0

go.sum

@@ -108,8 +108,8 @@ github.com/cncf/xds/go v0.0.0-20210922020428-25de7278fc84/go.mod h1:eXthEFrGJvWH
 github.com/cncf/xds/go v0.0.0-20211011173535-cb28da3451f1/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=
 github.com/cockroachdb/apd v1.1.0 h1:3LFP3629v+1aKXU5Q37mxmRxX/pIu1nijXydLShEq5I=
 github.com/cockroachdb/apd v1.1.0/go.mod h1:8Sl8LxpKi29FqWXR16WEFZRNSz3SoPzUzeMeY4+DwBQ=
-github.com/code-payments/code-protobuf-api v1.11.0 h1:xlE4cwDAkyzsa8Jxmvy5D0trtTGLCw+wqs142tYhTXo=
-github.com/code-payments/code-protobuf-api v1.11.0/go.mod h1:pHQm75vydD6Cm2qHAzlimW6drysm489Z4tVxC2zHSsU=
+github.com/code-payments/code-protobuf-api v1.12.2 h1:jU3CBQubbgjycyYkfwcm4v+Y0Q2mYZJJPDAQ4AXgZjI=
+github.com/code-payments/code-protobuf-api v1.12.2/go.mod h1:pHQm75vydD6Cm2qHAzlimW6drysm489Z4tVxC2zHSsU=
 github.com/containerd/continuity v0.0.0-20190827140505-75bee3e2ccb6 h1:NmTXa/uVnDyp0TY5MKi197+3HWcnYWfnHGyaFthlnGw=
 github.com/containerd/continuity v0.0.0-20190827140505-75bee3e2ccb6/go.mod h1:GL3xCUCBDV3CZiTSEKksMWbLE66hEyuu9qyDOOqM47Y=
 github.com/coreos/bbolt v1.3.2/go.mod h1:iRUV2dpdMOn7Bo10OQBFzIJO9kkE559Wcmn+qkEiiKk=

pkg/code/data/internal.go

@@ -33,6 +33,7 @@ import (
 	"github.com/code-payments/code-server/pkg/code/data/login"
 	"github.com/code-payments/code-server/pkg/code/data/merkletree"
 	"github.com/code-payments/code-server/pkg/code/data/nonce"
+	"github.com/code-payments/code-server/pkg/code/data/onramp"
 	"github.com/code-payments/code-server/pkg/code/data/payment"
 	"github.com/code-payments/code-server/pkg/code/data/paymentrequest"
 	"github.com/code-payments/code-server/pkg/code/data/paywall"
@@ -65,6 +66,7 @@ import (
 	messaging "github.com/code-payments/code-server/pkg/code/data/messaging"
 	messaging_memory_client "github.com/code-payments/code-server/pkg/code/data/messaging/memory"
 	nonce_memory_client "github.com/code-payments/code-server/pkg/code/data/nonce/memory"
+	onramp_memory_client "github.com/code-payments/code-server/pkg/code/data/onramp/memory"
 	payment_memory_client "github.com/code-payments/code-server/pkg/code/data/payment/memory"
 	paymentrequest_memory_client "github.com/code-payments/code-server/pkg/code/data/paymentrequest/memory"
 	paywall_memory_client "github.com/code-payments/code-server/pkg/code/data/paywall/memory"
@@ -95,6 +97,7 @@ import (
 	merkletree_postgres_client "github.com/code-payments/code-server/pkg/code/data/merkletree/postgres"
 	messaging_postgres_client "github.com/code-payments/code-server/pkg/code/data/messaging/postgres"
 	nonce_postgres_client "github.com/code-payments/code-server/pkg/code/data/nonce/postgres"
+	onramp_postgres_client "github.com/code-payments/code-server/pkg/code/data/onramp/postgres"
 	payment_postgres_client "github.com/code-payments/code-server/pkg/code/data/payment/postgres"
 	paymentrequest_postgres_client "github.com/code-payments/code-server/pkg/code/data/paymentrequest/postgres"
 	paywall_postgres_client "github.com/code-payments/code-server/pkg/code/data/paywall/postgres"
@@ -393,6 +396,11 @@ type DatabaseData interface {
 	SaveBalanceCheckpoint(ctx context.Context, record *balance.Record) error
 	GetBalanceCheckpoint(ctx context.Context, account string) (*balance.Record, error)
 
+	// Onramp
+	// --------------------------------------------------------------------------------
+	PutFiatOnrampPurchase(ctx context.Context, record *onramp.Record) error
+	GetFiatOnrampPurchase(ctx context.Context, nonce uuid.UUID) (*onramp.Record, error)
+
 	// ExecuteInTx executes fn with a single DB transaction that is scoped to the call.
 	// This enables more complex transactions that can span many calls across the provider.
 	//
@@ -432,6 +440,7 @@ type DatabaseProvider struct {
 	badgecount     badgecount.Store
 	login          login.Store
 	balance        balance.Store
+	onramp         onramp.Store
 
 	exchangeCache cache.Cache
 	timelockCache cache.Cache
@@ -490,6 +499,7 @@ func NewDatabaseProvider(dbConfig *pg.Config) (DatabaseData, error) {
 		badgecount:     badgecount_postgres_client.New(db),
 		login:          login_postgres_client.New(db),
 		balance:        balance_postgres_client.New(db),
+		onramp:         onramp_postgres_client.New(db),
 
 		exchangeCache: cache.NewCache(maxExchangeRateCacheBudget),
 		timelockCache: cache.NewCache(maxTimelockCacheBudget),
@@ -529,6 +539,7 @@ func NewTestDatabaseProvider() DatabaseData {
 		badgecount:     badgecount_memory_client.New(),
 		login:          login_memory_client.New(),
 		balance:        balance_memory_client.New(),
+		onramp:         onramp_memory_client.New(),
 
 		exchangeCache: cache.NewCache(maxExchangeRateCacheBudget),
 		timelockCache: nil, // Shouldn't be used for tests
@@ -1415,3 +1426,12 @@ func (dp *DatabaseProvider) SaveBalanceCheckpoint(ctx context.Context, record *b
 func (dp *DatabaseProvider) GetBalanceCheckpoint(ctx context.Context, account string) (*balance.Record, error) {
 	return dp.balance.GetCheckpoint(ctx, account)
 }
+
+// Onramp
+// --------------------------------------------------------------------------------
+func (dp *DatabaseProvider) PutFiatOnrampPurchase(ctx context.Context, record *onramp.Record) error {
+	return dp.onramp.Put(ctx, record)
+}
+func (dp *DatabaseProvider) GetFiatOnrampPurchase(ctx context.Context, nonce uuid.UUID) (*onramp.Record, error) {
+	return dp.onramp.Get(ctx, nonce)
+}

pkg/code/data/onramp/memory/store.go

@@ -0,0 +1,90 @@
+package memory
+
+import (
+	"bytes"
+	"context"
+	"sync"
+	"time"
+
+	"github.com/google/uuid"
+
+	"github.com/code-payments/code-server/pkg/code/data/onramp"
+)
+
+type store struct {
+	mu      sync.Mutex
+	records []*onramp.Record
+	last    uint64
+}
+
+// New returns a new in memory onramp.Store
+func New() onramp.Store {
+	return &store{}
+}
+
+// Put implements onramp.Store.Put
+func (s *store) Put(_ context.Context, data *onramp.Record) error {
+	if err := data.Validate(); err != nil {
+		return err
+	}
+
+	s.mu.Lock()
+	defer s.mu.Unlock()
+
+	s.last++
+	if item := s.find(data); item != nil {
+		return onramp.ErrPurchaseAlreadyExists
+	} else {
+		if data.Id == 0 {
+			data.Id = s.last
+		}
+		if data.CreatedAt.IsZero() {
+			data.CreatedAt = time.Now()
+		}
+		c := data.Clone()
+		s.records = append(s.records, &c)
+	}
+
+	return nil
+}
+
+// Get implements onramp.Store.Get
+func (s *store) Get(_ context.Context, nonce uuid.UUID) (*onramp.Record, error) {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+
+	if item := s.findByNonce(nonce); item != nil {
+		cloned := item.Clone()
+		return &cloned, nil
+	}
+	return nil, onramp.ErrPurchaseNotFound
+}
+
+func (s *store) find(data *onramp.Record) *onramp.Record {
+	for _, item := range s.records {
+		if item.Id == data.Id {
+			return item
+		}
+		if bytes.Equal(data.Nonce[:], item.Nonce[:]) {
+			return item
+		}
+	}
+	return nil
+}
+
+func (s *store) findByNonce(nonce uuid.UUID) *onramp.Record {
+	for _, item := range s.records {
+		if bytes.Equal(nonce[:], item.Nonce[:]) {
+			return item
+		}
+	}
+	return nil
+}
+
+func (s *store) reset() {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+
+	s.records = nil
+	s.last = 0
+}

pkg/code/data/onramp/memory/store_test.go

@@ -0,0 +1,15 @@
+package memory
+
+import (
+	"testing"
+
+	"github.com/code-payments/code-server/pkg/code/data/onramp/tests"
+)
+
+func TestOnrampMemoryStore(t *testing.T) {
+	testStore := New()
+	teardown := func() {
+		testStore.(*store).reset()
+	}
+	tests.RunTests(t, testStore, teardown)
+}

pkg/code/data/onramp/postgres/model.go

@@ -0,0 +1,94 @@
+package postgres
+
+import (
+	"context"
+	"database/sql"
+	"time"
+
+	"github.com/google/uuid"
+	"github.com/jmoiron/sqlx"
+
+	"github.com/code-payments/code-server/pkg/code/data/onramp"
+	pgutil "github.com/code-payments/code-server/pkg/database/postgres"
+)
+
+const (
+	tableName = "codewallet__core_onramppurchase"
+)
+
+type model struct {
+	Id sql.NullInt64 `db:"id"`
+
+	Owner    string    `db:"owner"`
+	Currency string    `db:"currency"`
+	Amount   float64   `db:"amount"`
+	Nonce    uuid.UUID `db:"nonce"`
+
+	CreatedAt time.Time `db:"created_at"`
+}
+
+func toModel(obj *onramp.Record) (*model, error) {
+	if err := obj.Validate(); err != nil {
+		return nil, err
+	}
+
+	return &model{
+		Owner:     obj.Owner,
+		Currency:  obj.Currency,
+		Amount:    obj.Amount,
+		Nonce:     obj.Nonce,
+		CreatedAt: obj.CreatedAt,
+	}, nil
+}
+
+func fromModel(obj *model) *onramp.Record {
+	return &onramp.Record{
+		Id:        uint64(obj.Id.Int64),
+		Owner:     obj.Owner,
+		Currency:  obj.Currency,
+		Amount:    obj.Amount,
+		Nonce:     obj.Nonce,
+		CreatedAt: obj.CreatedAt,
+	}
+}
+
+func (m *model) dbPut(ctx context.Context, db *sqlx.DB) error {
+	return pgutil.ExecuteInTx(ctx, db, sql.LevelDefault, func(tx *sqlx.Tx) error {
+		query := `INSERT INTO ` + tableName + `
+			(owner, currency, amount, nonce, created_at)
+			VALUES ($1, $2, $3, $4, $5)
+			RETURNING id, owner, currency, amount, nonce, created_at`
+
+		if m.CreatedAt.IsZero() {
+			m.CreatedAt = time.Now()
+		}
+
+		err := tx.QueryRowxContext(
+			ctx,
+			query,
+			m.Owner,
+			m.Currency,
+			m.Amount,
+			m.Nonce,
+			m.CreatedAt.UTC(),
+		).StructScan(m)
+
+		return pgutil.CheckUniqueViolation(err, onramp.ErrPurchaseAlreadyExists)
+	})
+}
+
+func dbGet(ctx context.Context, db *sqlx.DB, nonce uuid.UUID) (*model, error) {
+	res := &model{}
+
+	query := `SELECT
+		id, owner, currency, amount, nonce, created_at
+		FROM ` + tableName + `
+		WHERE nonce = $1
+		LIMIT 1`
+
+	err := db.GetContext(ctx, res, query, nonce)
+	if err != nil {
+		return nil, pgutil.CheckNoRows(err, onramp.ErrPurchaseNotFound)
+	}
+	return res, nil
+}

pkg/code/data/onramp/postgres/store.go

@@ -0,0 +1,48 @@
+package postgres
+
+import (
+	"context"
+	"database/sql"
+
+	"github.com/google/uuid"
+	"github.com/jmoiron/sqlx"
+
+	"github.com/code-payments/code-server/pkg/code/data/onramp"
+)
+
+type store struct {
+	db *sqlx.DB
+}
+
+// New returns a new postgres-backed onramp.Store
+func New(db *sql.DB) onramp.Store {
+	return &store{
+		db: sqlx.NewDb(db, "pgx"),
+	}
+}
+
+// Put implements onramp.Store.Put
+func (s *store) Put(ctx context.Context, record *onramp.Record) error {
+	model, err := toModel(record)
+	if err != nil {
+		return err
+	}
+
+	if err := model.dbPut(ctx, s.db); err != nil {
+		return err
+	}
+
+	res := fromModel(model)
+	res.CopyTo(record)
+
+	return nil
+}
+
+// Get implements onramp.Store.Get
+func (s *store) Get(ctx context.Context, nonce uuid.UUID) (*onramp.Record, error) {
+	model, err := dbGet(ctx, s.db, nonce)
+	if err != nil {
+		return nil, err
+	}
+	return fromModel(model), nil
+}