Implement custody RPC actions

stellar · Oct 31, 2023 · 5263607 · 5263607
1 parent 86d95d4
commit 5263607
Show file tree

Hide file tree

Showing 41 changed files with 1,953 additions and 327 deletions.
diff --git a/core/src/main/java/org/stellar/anchor/config/Sep6Config.java b/core/src/main/java/org/stellar/anchor/config/Sep6Config.java
@@ -11,6 +11,8 @@ public interface Sep6Config {
 
   Features getFeatures();
 
+  DepositInfoGeneratorType getDepositInfoGeneratorType();
+
   @Getter
   @Setter
   @AllArgsConstructor
@@ -22,4 +24,10 @@ class Features {
     @SerializedName("claimable_balances")
     boolean claimableBalances;
   }
+
+  enum DepositInfoGeneratorType {
+    SELF,
+    CUSTODY,
+    NONE
+  }
 }
diff --git a/core/src/main/java/org/stellar/anchor/custody/CustodyService.java b/core/src/main/java/org/stellar/anchor/custody/CustodyService.java
@@ -5,9 +5,18 @@
 import org.stellar.anchor.api.rpc.method.DoStellarRefundRequest;
 import org.stellar.anchor.sep24.Sep24Transaction;
 import org.stellar.anchor.sep31.Sep31Transaction;
+import org.stellar.anchor.sep6.Sep6Transaction;
 
 public interface CustodyService {
 
+  /**
+   * Create custody transaction for SEP6 transaction
+   *
+   * @param txn SEP6 transaction
+   * @throws AnchorException if error happens
+   */
+  void createTransaction(Sep6Transaction txn) throws AnchorException;
+
   /**
    * Create custody transaction for SEP24 transaction
    *

diff --git a/core/src/main/java/org/stellar/anchor/sep6/Sep6DepositInfoGenerator.java b/core/src/main/java/org/stellar/anchor/sep6/Sep6DepositInfoGenerator.java
@@ -0,0 +1,16 @@
+package org.stellar.anchor.sep6;
+
+import org.stellar.anchor.api.exception.AnchorException;
+import org.stellar.anchor.api.shared.SepDepositInfo;
+
+public interface Sep6DepositInfoGenerator {
+
+  /**
+   * Gets the deposit info based on the input parameter.
+   *
+   * @param txn the original SEP-6 transaction the deposit info will be used for.
+   * @return a SepDepositInfo instance containing the destination address, memo and memoType.
+   * @throws AnchorException if the deposit info cannot be generated
+   */
+  SepDepositInfo generate(Sep6Transaction txn) throws AnchorException;
+}
diff --git a/core/src/main/java/org/stellar/anchor/sep6/Sep6Service.java b/core/src/main/java/org/stellar/anchor/sep6/Sep6Service.java
@@ -1,7 +1,6 @@
 package org.stellar.anchor.sep6;
 
 import static org.stellar.anchor.util.MemoHelper.*;
-import static org.stellar.sdk.xdr.MemoType.MEMO_HASH;
 
 import com.google.common.collect.ImmutableMap;
 import java.time.Instant;
@@ -270,10 +269,8 @@ public StartWithdrawResponse withdraw(Sep10Jwt token, StartWithdrawRequest reque
             .startedAt(Instant.now())
             .sep10Account(token.getAccount())
             .sep10AccountMemo(token.getAccountMemo())
-            .memo(generateMemo(id))
-            .memoType(memoTypeAsString(MEMO_HASH))
-            .fromAccount(sourceAccount)
             .withdrawAnchorAccount(asset.getDistributionAccount())
+            .fromAccount(sourceAccount)
             .refundMemo(request.getRefundMemo())
             .refundMemoType(request.getRefundMemoType());
 
@@ -291,8 +288,6 @@ public StartWithdrawResponse withdraw(Sep10Jwt token, StartWithdrawRequest reque
     return StartWithdrawResponse.builder()
         .accountId(asset.getDistributionAccount())
         .id(txn.getId())
-        .memo(txn.getMemo())
-        .memoType(memoTypeAsString(MEMO_HASH))
         .build();
   }
 
@@ -364,10 +359,8 @@ public StartWithdrawResponse withdrawExchange(
             .startedAt(Instant.now())
             .sep10Account(token.getAccount())
             .sep10AccountMemo(token.getAccountMemo())
-            .memo(generateMemo(id))
-            .memoType(memoTypeAsString(MEMO_HASH))
-            .fromAccount(sourceAccount)
             .withdrawAnchorAccount(sellAsset.getDistributionAccount())
+            .fromAccount(sourceAccount)
             .refundMemo(request.getRefundMemo())
             .refundMemoType(request.getRefundMemoType())
             .quoteId(request.getQuoteId());
@@ -386,8 +379,6 @@ public StartWithdrawResponse withdrawExchange(
     return StartWithdrawResponse.builder()
         .accountId(sellAsset.getDistributionAccount())
         .id(txn.getId())
-        .memo(txn.getMemo())
-        .memoType(memoTypeAsString(MEMO_HASH))
         .build();
   }
 

diff --git a/core/src/main/java/org/stellar/anchor/util/MemoHelper.java b/core/src/main/java/org/stellar/anchor/util/MemoHelper.java
@@ -6,7 +6,6 @@
 import java.util.Base64;
 import org.apache.commons.codec.DecoderException;
 import org.apache.commons.codec.binary.Hex;
-import org.apache.commons.lang3.StringUtils;
 import org.stellar.anchor.api.exception.SepException;
 import org.stellar.anchor.api.exception.SepValidationException;
 import org.stellar.sdk.*;
@@ -126,10 +125,4 @@ public static String memoAsString(Memo memo) throws SepException {
         throw new SepException("Unsupported value: " + memoTypeStr);
     }
   }
-
-  public static String generateMemo(String transactionId) {
-    String memo = StringUtils.truncate(transactionId, 32);
-    memo = StringUtils.leftPad(memo, 32, "0");
-    return new String(Base64.getEncoder().encode(memo.getBytes()));
-  }
 }
diff --git a/core/src/main/java/org/stellar/anchor/util/TransactionHelper.java b/core/src/main/java/org/stellar/anchor/util/TransactionHelper.java
@@ -1,9 +1,8 @@
 package org.stellar.anchor.util;
 
-import static org.stellar.anchor.api.platform.PlatformTransactionData.Kind.DEPOSIT;
-import static org.stellar.anchor.api.platform.PlatformTransactionData.Kind.RECEIVE;
-import static org.stellar.anchor.api.platform.PlatformTransactionData.Kind.WITHDRAWAL;
+import static org.stellar.anchor.api.platform.PlatformTransactionData.Kind.*;
 
+import com.google.common.collect.ImmutableSet;
 import java.util.Optional;
 import javax.annotation.Nullable;
 import org.stellar.anchor.api.custody.CreateCustodyTransactionRequest;
@@ -22,6 +21,33 @@
 
 public class TransactionHelper {
 
+  public static CreateCustodyTransactionRequest toCustodyTransaction(Sep6Transaction txn) {
+    PlatformTransactionData.Kind kind = PlatformTransactionData.Kind.from(txn.getKind());
+    return CreateCustodyTransactionRequest.builder()
+        .id(txn.getId())
+        .memo(txn.getMemo())
+        .memoType(txn.getMemoType())
+        .protocol("6")
+        .fromAccount(
+            ImmutableSet.of(WITHDRAWAL, WITHDRAWAL_EXCHANGE).contains(kind)
+                ? txn.getFromAccount()
+                : null)
+        .toAccount(
+            ImmutableSet.of(DEPOSIT, DEPOSIT_EXCHANGE).contains(kind)
+                ? txn.getToAccount()
+                : txn.getWithdrawAnchorAccount())
+        .amount(
+            ImmutableSet.of(DEPOSIT, DEPOSIT_EXCHANGE).contains(kind)
+                ? txn.getAmountOut()
+                : Optional.ofNullable(txn.getAmountExpected()).orElse(txn.getAmountIn()))
+        .asset(
+            ImmutableSet.of(DEPOSIT, DEPOSIT_EXCHANGE).contains(kind)
+                ? txn.getAmountOutAsset()
+                : txn.getAmountInAsset())
+        .kind(txn.getKind())
+        .build();
+  }
+
   public static CreateCustodyTransactionRequest toCustodyTransaction(Sep24Transaction txn) {
     return CreateCustodyTransactionRequest.builder()
         .id(txn.getId())

diff --git a/core/src/test/kotlin/org/stellar/anchor/sep6/Sep6ServiceTest.kt b/core/src/test/kotlin/org/stellar/anchor/sep6/Sep6ServiceTest.kt
@@ -657,8 +657,6 @@ class Sep6ServiceTest {
       JSONCompareMode.LENIENT
     )
     assert(slotTxn.captured.id.isNotEmpty())
-    assert(slotTxn.captured.memo.isNotEmpty())
-    assertEquals(slotTxn.captured.memoType, "hash")
     assertNotNull(slotTxn.captured.startedAt)
 
     JSONAssert.assertEquals(
@@ -668,8 +666,6 @@ class Sep6ServiceTest {
     )
     assert(slotEvent.captured.id.isNotEmpty())
     assert(slotEvent.captured.transaction.id.isNotEmpty())
-    assert(slotEvent.captured.transaction.memo.isNotEmpty())
-    assertEquals(slotEvent.captured.transaction.memoType, "hash")
     assertNotNull(slotEvent.captured.transaction.startedAt)
 
     // Verify response
@@ -738,8 +734,6 @@ class Sep6ServiceTest {
       JSONCompareMode.LENIENT
     )
     assert(slotTxn.captured.id.isNotEmpty())
-    assert(slotTxn.captured.memo.isNotEmpty())
-    assertEquals(slotTxn.captured.memoType, "hash")
     assertNotNull(slotTxn.captured.startedAt)
 
     JSONAssert.assertEquals(
@@ -749,8 +743,6 @@ class Sep6ServiceTest {
     )
     assert(slotEvent.captured.id.isNotEmpty())
     assert(slotEvent.captured.transaction.id.isNotEmpty())
-    assert(slotEvent.captured.transaction.memo.isNotEmpty())
-    assertEquals(slotEvent.captured.transaction.memoType, "hash")
     assertNotNull(slotEvent.captured.transaction.startedAt)
 
     // Verify response
@@ -945,8 +937,6 @@ class Sep6ServiceTest {
       JSONCompareMode.LENIENT
     )
     assert(slotTxn.captured.id.isNotEmpty())
-    assert(slotTxn.captured.memo.isNotEmpty())
-    assertEquals(slotTxn.captured.memoType, "hash")
     assertNotNull(slotTxn.captured.startedAt)
 
     JSONAssert.assertEquals(
@@ -956,8 +946,6 @@ class Sep6ServiceTest {
     )
     assert(slotEvent.captured.id.isNotEmpty())
     assert(slotEvent.captured.transaction.id.isNotEmpty())
-    assert(slotEvent.captured.transaction.memo.isNotEmpty())
-    assertEquals(slotEvent.captured.transaction.memoType, "hash")
     assertNotNull(slotEvent.captured.transaction.startedAt)
 
     // Verify response
@@ -1018,8 +1006,6 @@ class Sep6ServiceTest {
       JSONCompareMode.LENIENT
     )
     assert(slotTxn.captured.id.isNotEmpty())
-    assert(slotTxn.captured.memo.isNotEmpty())
-    assertEquals(slotTxn.captured.memoType, "hash")
     assertNotNull(slotTxn.captured.startedAt)
 
     JSONAssert.assertEquals(
@@ -1029,8 +1015,6 @@ class Sep6ServiceTest {
     )
     assert(slotEvent.captured.id.isNotEmpty())
     assert(slotEvent.captured.transaction.id.isNotEmpty())
-    assert(slotEvent.captured.transaction.memo.isNotEmpty())
-    assertEquals(slotEvent.captured.transaction.memoType, "hash")
     assertNotNull(slotEvent.captured.transaction.startedAt)
 
     // Verify response

diff --git a/core/src/test/kotlin/org/stellar/anchor/sep6/Sep6ServiceTestData.kt b/core/src/test/kotlin/org/stellar/anchor/sep6/Sep6ServiceTestData.kt
@@ -388,10 +388,9 @@ class Sep6ServiceTestData {
     val withdrawResJson =
       """
       {
-          "account_id": "GA7FYRB5VREZKOBIIKHG5AVTPFGWUBPOBF7LTYG4GTMFVIOOD2DWAL7I",
-          "memo_type": "hash"
+          "account_id": "GA7FYRB5VREZKOBIIKHG5AVTPFGWUBPOBF7LTYG4GTMFVIOOD2DWAL7I"
       }
-    """
+      """
         .trimIndent()
 
     val withdrawTxnJson =
@@ -413,7 +412,6 @@ class Sep6ServiceTestData {
           "sep10AccountMemo": "123",
           "withdrawAnchorAccount": "GA7FYRB5VREZKOBIIKHG5AVTPFGWUBPOBF7LTYG4GTMFVIOOD2DWAL7I",
           "fromAccount": "GBLGJA4TUN5XOGTV6WO2BWYUI2OZR5GYQ5PDPCRMQ5XEPJOYWB2X4CJO",
-          "memoType": "hash",
           "refundMemo": "some text",
           "refundMemoType": "text"
         }
@@ -446,7 +444,6 @@ class Sep6ServiceTestData {
               "asset": "stellar:USDC:GDQOE23CFSUMSVQK4Y5JHPPYK73VYCNHZHA7ENKCV37P6SUEO6XQBKPP"
             },
             "source_account": "GBLGJA4TUN5XOGTV6WO2BWYUI2OZR5GYQ5PDPCRMQ5XEPJOYWB2X4CJO",
-            "memo_type": "hash",
             "refund_memo": "some text",
             "refund_memo_type": "text",
             "customers": {
@@ -479,7 +476,6 @@ class Sep6ServiceTestData {
           "sep10AccountMemo": "123",
           "withdrawAnchorAccount": "GA7FYRB5VREZKOBIIKHG5AVTPFGWUBPOBF7LTYG4GTMFVIOOD2DWAL7I",
           "fromAccount": "GBLGJA4TUN5XOGTV6WO2BWYUI2OZR5GYQ5PDPCRMQ5XEPJOYWB2X4CJO",
-          "memoType": "hash",
           "refundMemo": "some text",
           "refundMemoType": "text"
         }
@@ -503,7 +499,6 @@ class Sep6ServiceTestData {
               "asset": "stellar:USDC:GDQOE23CFSUMSVQK4Y5JHPPYK73VYCNHZHA7ENKCV37P6SUEO6XQBKPP"
             },
             "source_account": "GBLGJA4TUN5XOGTV6WO2BWYUI2OZR5GYQ5PDPCRMQ5XEPJOYWB2X4CJO",
-            "memo_type": "hash",
             "refund_memo": "some text",
             "refund_memo_type": "text",
             "customers": {
@@ -540,7 +535,6 @@ class Sep6ServiceTestData {
           "sep10AccountMemo": "123",
           "withdrawAnchorAccount": "GA7FYRB5VREZKOBIIKHG5AVTPFGWUBPOBF7LTYG4GTMFVIOOD2DWAL7I",
           "fromAccount": "GBLGJA4TUN5XOGTV6WO2BWYUI2OZR5GYQ5PDPCRMQ5XEPJOYWB2X4CJO",
-          "memoType": "hash",
           "quoteId": "test-quote-id",
           "refundMemo": "some text",
           "refundMemoType": "text"
@@ -569,7 +563,6 @@ class Sep6ServiceTestData {
             "amount_fee": { "amount": "2", "asset": "iso4217:USD" },
             "quote_id": "test-quote-id",
             "source_account": "GBLGJA4TUN5XOGTV6WO2BWYUI2OZR5GYQ5PDPCRMQ5XEPJOYWB2X4CJO",
-            "memo_type": "hash",
             "refund_memo": "some text",
             "refund_memo_type": "text",
             "customers": {
@@ -606,7 +599,6 @@ class Sep6ServiceTestData {
           "sep10AccountMemo": "123",
           "withdrawAnchorAccount": "GA7FYRB5VREZKOBIIKHG5AVTPFGWUBPOBF7LTYG4GTMFVIOOD2DWAL7I",
           "fromAccount": "GBLGJA4TUN5XOGTV6WO2BWYUI2OZR5GYQ5PDPCRMQ5XEPJOYWB2X4CJO",
-          "memoType": "hash",
           "refundMemo": "some text",
           "refundMemoType": "text"
         }
@@ -637,7 +629,6 @@ class Sep6ServiceTestData {
               "asset": "stellar:USDC:GDQOE23CFSUMSVQK4Y5JHPPYK73VYCNHZHA7ENKCV37P6SUEO6XQBKPP"
             },
             "source_account": "GBLGJA4TUN5XOGTV6WO2BWYUI2OZR5GYQ5PDPCRMQ5XEPJOYWB2X4CJO",
-            "memo_type": "hash",
             "refund_memo": "some text",
             "refund_memo_type": "text",
             "customers": {

diff --git a/...ion-tests/src/test/kotlin/org/stellar/anchor/platform/AnchorPlatformCustodyEnd2EndTest.kt b/...ion-tests/src/test/kotlin/org/stellar/anchor/platform/AnchorPlatformCustodyEnd2EndTest.kt
@@ -28,4 +28,12 @@ class AnchorPlatformCustodyEnd2EndTest :
   fun runSep24Test() {
     singleton.sep24CustodyE2eTests.testAll()
   }
+
+  @Test
+  @Order(11)
+  fun runSep6Test() {
+    // The SEP-6 reference server implementation only implements RPC, so technically this test
+    // should be in the RPC test suite.
+    singleton.sep6E2eTests.testAll()
+  }
 }
diff --git a/integration-tests/src/test/kotlin/org/stellar/anchor/platform/AnchorPlatformEnd2EndTest.kt b/integration-tests/src/test/kotlin/org/stellar/anchor/platform/AnchorPlatformEnd2EndTest.kt
@@ -31,6 +31,8 @@ class AnchorPlatformEnd2EndTest : AbstractIntegrationTest(TestConfig(testProfile
   @Test
   @Order(2)
   fun runSep6Test() {
+    // The SEP-6 reference server implementation only implements RPC, so technically this test
+    // should be in the RPC test suite.
     singleton.sep6E2eTests.testAll()
   }
 }
diff --git a/integration-tests/src/test/kotlin/org/stellar/anchor/platform/test/Sep6End2EndTest.kt b/integration-tests/src/test/kotlin/org/stellar/anchor/platform/test/Sep6End2EndTest.kt
@@ -152,11 +152,14 @@ class Sep6End2EndTest(val config: TestConfig, val jwt: String) {
     anchor.customer(token).add(additionalRequiredFields.associateWith { customerInfo[it]!! })
     waitStatus(withdraw.id, PENDING_USR_TRANSFER_START, sep6Client)
 
+    val withdrawMemo =
+      sep6Client.getTransaction(mapOf("id" to withdraw.id)).transaction.withdrawMemo
+
     // Transfer the withdrawal amount to the Anchor
     val transfer =
       wallet
         .stellar()
-        .transaction(keypair, memo = Pair(MemoType.HASH, withdraw.memo))
+        .transaction(keypair, memo = Pair(MemoType.HASH, withdrawMemo))
         .transfer(withdraw.accountId, USDC, "1")
         .build()
     transfer.sign(keypair)

diff --git a/integration-tests/src/test/kotlin/org/stellar/anchor/platform/test/Sep6Tests.kt b/integration-tests/src/test/kotlin/org/stellar/anchor/platform/test/Sep6Tests.kt
@@ -160,7 +160,7 @@ class Sep6Tests(val toml: TomlContent, jwt: String) {
               "kind": "withdrawal",
               "status": "incomplete",
               "from": "GDJLBYYKMCXNVVNABOE66NYXQGIA5AC5D223Z2KF6ZEYK4UBCA7FKLTG",
-              "withdraw_memo_type": "hash"
+              "withdraw_anchor_account": "GBN4NNCDGJO4XW4KQU3CBIESUJWFVBUZPOKUZHT7W7WRB7CWOA7BXVQF"
           }
       }
     """
@@ -179,8 +179,7 @@ class Sep6Tests(val toml: TomlContent, jwt: String) {
           "amount_fee": "0",
           "amount_fee_asset": "stellar:USDC:GDQOE23CFSUMSVQK4Y5JHPPYK73VYCNHZHA7ENKCV37P6SUEO6XQBKPP",
           "from": "GDJLBYYKMCXNVVNABOE66NYXQGIA5AC5D223Z2KF6ZEYK4UBCA7FKLTG",
-          "withdraw_anchor_account": "GBN4NNCDGJO4XW4KQU3CBIESUJWFVBUZPOKUZHT7W7WRB7CWOA7BXVQF",
-          "withdraw_memo_type": "hash"
+          "withdraw_anchor_account": "GBN4NNCDGJO4XW4KQU3CBIESUJWFVBUZPOKUZHT7W7WRB7CWOA7BXVQF"
         }
       }
     """
@@ -199,8 +198,7 @@ class Sep6Tests(val toml: TomlContent, jwt: String) {
           "amount_fee": "1.00",
           "amount_fee_asset": "stellar:USDC:GDQOE23CFSUMSVQK4Y5JHPPYK73VYCNHZHA7ENKCV37P6SUEO6XQBKPP",
           "from": "GDJLBYYKMCXNVVNABOE66NYXQGIA5AC5D223Z2KF6ZEYK4UBCA7FKLTG",
-          "withdraw_anchor_account": "GBN4NNCDGJO4XW4KQU3CBIESUJWFVBUZPOKUZHT7W7WRB7CWOA7BXVQF",
-          "withdraw_memo_type": "hash"
+          "withdraw_anchor_account": "GBN4NNCDGJO4XW4KQU3CBIESUJWFVBUZPOKUZHT7W7WRB7CWOA7BXVQF"
         }
       }
     """