Encode destinations in public distribution intent record

Author: Jeff Yanta

pkg/code/data/intent/intent.go

@@ -100,10 +100,17 @@ type ReceivePaymentsPubliclyMetadata struct {
 
 type PublicDistributionMetadata struct {
 	Source         string
+	Distributions  []*Distribution
 	Quantity       uint64
 	UsdMarketValue float64
 }
 
+type Distribution struct {
+	DestinationOwnerAccount string
+	DestinationTokenAccount string
+	Quantity                uint64
+}
+
 func (r *Record) IsCompleted() bool {
 	return r.State == StateConfirmed
 }
@@ -416,15 +423,29 @@ func (m *ReceivePaymentsPubliclyMetadata) Validate() error {
 }
 
 func (m *PublicDistributionMetadata) Clone() PublicDistributionMetadata {
+	clonedDistributions := make([]*Distribution, len(m.Distributions))
+	for i, distribution := range m.Distributions {
+		cloned := distribution.Clone()
+		clonedDistributions[i] = &cloned
+	}
+
 	return PublicDistributionMetadata{
 		Source:         m.Source,
+		Distributions:  clonedDistributions,
 		Quantity:       m.Quantity,
 		UsdMarketValue: m.UsdMarketValue,
 	}
 }
 
 func (m *PublicDistributionMetadata) CopyTo(dst *PublicDistributionMetadata) {
+	clonedDistributions := make([]*Distribution, len(m.Distributions))
+	for i, distribution := range m.Distributions {
+		cloned := distribution.Clone()
+		clonedDistributions[i] = &cloned
+	}
+
 	dst.Source = m.Source
+	dst.Distributions = clonedDistributions
 	dst.Quantity = m.Quantity
 	dst.UsdMarketValue = m.UsdMarketValue
 }
@@ -434,6 +455,45 @@ func (m *PublicDistributionMetadata) Validate() error {
 		return errors.New("source is required")
 	}
 
+	if len(m.Distributions) == 0 {
+		return errors.New("distributions are required")
+	}
+	for _, distribution := range m.Distributions {
+		if err := distribution.Validate(); err != nil {
+			return err
+		}
+	}
+
+	if m.Quantity == 0 {
+		return errors.New("quantity is required")
+	}
+
+	return nil
+}
+
+func (m *Distribution) Clone() Distribution {
+	return Distribution{
+		DestinationOwnerAccount: m.DestinationOwnerAccount,
+		DestinationTokenAccount: m.DestinationTokenAccount,
+		Quantity:                m.Quantity,
+	}
+}
+
+func (m *Distribution) CopyTo(dst *Distribution) {
+	dst.DestinationOwnerAccount = m.DestinationOwnerAccount
+	dst.DestinationTokenAccount = m.DestinationTokenAccount
+	dst.Quantity = m.Quantity
+}
+
+func (m *Distribution) Validate() error {
+	if len(m.DestinationOwnerAccount) == 0 {
+		return errors.New("destination owner account is required")
+	}
+
+	if len(m.DestinationTokenAccount) == 0 {
+		return errors.New("destination token account is required")
+	}
+
 	if m.Quantity == 0 {
 		return errors.New("quantity is required")
 	}

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

@@ -73,13 +73,21 @@ func (s *store) findByOwner(owner string) []*intent.Record {
 
 		if item.SendPublicPaymentMetadata != nil && item.SendPublicPaymentMetadata.DestinationOwnerAccount == owner {
 			res = append(res, item)
-			continue
+		}
+
+		if item.PublicDistributionMetadata != nil {
+			for _, distribution := range item.PublicDistributionMetadata.Distributions {
+				if distribution.DestinationOwnerAccount == owner {
+					res = append(res, item)
+				}
+			}
 		}
 	}
 
 	return res
 }
 
+// todo: doesn't support PublicDistribution, but it's not required
 func (s *store) findByDestination(destination string) []*intent.Record {
 	res := make([]*intent.Record, 0)
 	for _, item := range s.records {
@@ -97,6 +105,7 @@ func (s *store) findByDestination(destination string) []*intent.Record {
 	return res
 }
 
+// todo: doesn't support PublicDistribution, but it's not required
 func (s *store) findBySource(source string) []*intent.Record {
 	res := make([]*intent.Record, 0)
 	for _, item := range s.records {
@@ -110,7 +119,8 @@ func (s *store) findBySource(source string) []*intent.Record {
 	return res
 }
 
-func (s *store) findByOwnerSinceTimestamp(owner string, since time.Time) []*intent.Record {
+// todo: doesn't support PublicDistribution, but it's not required
+func (s *store) findByInitiatorOwnerSinceTimestamp(owner string, since time.Time) []*intent.Record {
 	res := make([]*intent.Record, 0)
 	for _, item := range s.records {
 		if item.CreatedAt.Before(since) {
@@ -348,7 +358,7 @@ func (s *store) GetTransactedAmountForAntiMoneyLaundering(ctx context.Context, o
 	s.mu.Lock()
 	defer s.mu.Unlock()
 
-	items := s.findByOwnerSinceTimestamp(owner, since)
+	items := s.findByInitiatorOwnerSinceTimestamp(owner, since)
 	items = s.filterByType(items, intent.SendPublicPayment)
 	items = s.filterByState(items, false, intent.StateRevoked)
 	items = s.filterByWithdrawalFlag(items, false)