Merge pull request #390 from tronprotocol/feature/shielded_TRC20_add_demo

modify the fullnode config for shielded trc20 demo

Author: renchenchang

src/main/java/org/tron/demo/ShieldedTRC20Demo.java

@@ -8,7 +8,9 @@
 import org.junit.Assert;
 import org.spongycastle.util.encoders.Hex;
 import org.tron.api.GrpcAPI;
+import org.tron.api.GrpcAPI.PrivateShieldedTRC20Parameters;
 import org.tron.api.GrpcAPI.Return;
+import org.tron.api.GrpcAPI.SpendNoteTRC20;
 import org.tron.api.GrpcAPI.TransactionExtention;
 import org.tron.common.crypto.ECKey;
 import org.tron.common.crypto.Sha256Sm3Hash;
@@ -38,15 +40,13 @@
 @Slf4j
 public class ShieldedTRC20Demo {
 
-  private static String fullnode = "34.220.77.106:50051";
-  private static String soliditynode = "52.15.93.92:50061";
   private static String trc20ContractAddress = "TR7NHqjeKQxGTCi8q8ZY4pL8otSzgjLj6t";
   private static String shieldedTRC20ContractAddress = "TQEuSEVRk1GtfExm5q9T8a1w84GvgQJ13V";
   private static String privateKey =
-      "f4cc9a4bcd6a2167e734d0e3de349f93c3a0210fcf6c2390f0e20bc43832a13b";
-  private static String pubAddress = "TCuT4T3xhrSuScmhbVuWf3ZAvRnMXvoqC6";
+      "2c8893287a87ac9f4b70af14fbae75e5c898e3b6645e5fed311f5fe60b2dff2f";
+  private static String pubAddress = "TXmiKi5UZ6Pqe22aW5R8LEcNGGpgh2BNMH";
   private static String spendingKey = "004f74ce2bde08f0c936f2929b94cb2ca49111db95001576f99d04c3e671daf6";
-  private static GrpcClient grpcClient = new GrpcClient(fullnode, soliditynode);
+  private static GrpcClient grpcClient = WalletApi.init();
   private static BigInteger scalingFactorBi = getScalingFactorBi();
 
   public static void main(String[] args) throws ZksnarkException, InterruptedException {
@@ -60,9 +60,8 @@ private static String mintDemo() throws ZksnarkException {
 
     SpendingKey sk = new SpendingKey(ByteArray.fromHexString(spendingKey));
     ExpandedSpendingKey expsk = sk.expandedSpendingKey();
-
     byte[] ovk = expsk.getOvk();
-    long fromAmount = 10;
+    long fromAmount = 1;
     FullViewingKey fullViewingKey = sk.fullViewingKey();
     IncomingViewingKey incomingViewingKey = fullViewingKey.inViewingKey();
     PaymentAddress paymentAddress = incomingViewingKey.address(new DiversifierT().random()).get();
@@ -73,7 +72,7 @@ private static String mintDemo() throws ZksnarkException {
     GrpcAPI.ReceiveNote.Builder revNoteBuilder = GrpcAPI.ReceiveNote.newBuilder();
     long revValue = fromAmount;
     byte[] memo = new byte[512];
-    byte[] rcm = grpcClient.getRcm().getValue().toByteArray();
+    byte[] rcm = WalletApi.getRcm().get().getValue().toByteArray();
     GrpcAPI.Note revNote = getNote(revValue, KeyIo.encodePaymentAddress(paymentAddress), rcm, memo);
     revNoteBuilder.setNote(revNote);
 
@@ -85,7 +84,7 @@ private static String mintDemo() throws ZksnarkException {
     paramBuilder.addShieldedReceives(revNoteBuilder.build());
     paramBuilder.setShieldedTRC20ContractAddress(ByteString.copyFrom(contractAddress));
 
-    GrpcAPI.ShieldedTRC20Parameters trc20MintParams = grpcClient
+    GrpcAPI.ShieldedTRC20Parameters trc20MintParams = WalletApi
         .createShieldedContractParameters(paramBuilder.build());
     byte[] callerAddress = WalletApi.decodeFromBase58Check(pubAddress);
     return triggerMint(contractAddress, callerAddress, privateKey,
@@ -97,56 +96,52 @@ public static void transferDemo() throws ZksnarkException, InterruptedException
         .decodeFromBase58Check(shieldedTRC20ContractAddress);
     byte[] callerAddress = WalletApi.decodeFromBase58Check(pubAddress);
     SpendingKey sk = new SpendingKey(ByteArray.fromHexString(spendingKey));
-    setAllowance(10);
+    setAllowance(2);
     GrpcAPI.PrivateShieldedTRC20Parameters mintPrivateParam1 = mintParams(
-        privateKey, 10, shieldedTRC20ContractAddress);
-    GrpcAPI.ShieldedTRC20Parameters mintParam1 = grpcClient.createShieldedContractParameters(
+        privateKey, 2, shieldedTRC20ContractAddress);
+    GrpcAPI.ShieldedTRC20Parameters mintParam1 = WalletApi.createShieldedContractParameters(
         mintPrivateParam1);
     String mintInput = mintParam1.getTriggerContractInput();
     String txid = triggerMint(contractAddress, callerAddress, privateKey, mintInput);
 
-    // SpendNoteTRC20 1
-    logger.info("mint txid: " + txid);
-    Optional<TransactionInfo> infoById = grpcClient.getTransactionInfoById(txid);
-    while (infoById.get().getLogList().size() < 2) {
-      logger.info("Can not get transaction info, please wait....");
-      Thread.sleep(5000);
-      infoById = grpcClient.getTransactionInfoById(txid);
-    }
-    byte[] txData = infoById.get().getLog(1).getData().toByteArray();
-    long pos = bytes32Tolong(ByteArray.subArray(txData, 0, 32));
-    byte[] contractResult = triggerGetPath(contractAddress, pos);
-    byte[] path = ByteArray.subArray(contractResult, 32, 1056);
-    byte[] root = ByteArray.subArray(contractResult, 0, 32);
-    GrpcAPI.SpendNoteTRC20.Builder note1Builder = GrpcAPI.SpendNoteTRC20.newBuilder();
-    note1Builder.setAlpha(ByteString.copyFrom(grpcClient.getRcm().getValue().toByteArray()));
-    note1Builder.setPos(pos);
-    note1Builder.setPath(ByteString.copyFrom(path));
-    note1Builder.setRoot(ByteString.copyFrom(root));
-    note1Builder.setNote(mintPrivateParam1.getShieldedReceives(0).getNote());
+    Optional<TransactionInfo> infoById = waitToGetTransactionInfo(txid);
     GrpcAPI.PrivateShieldedTRC20Parameters.Builder privateTRC20Builder = GrpcAPI
         .PrivateShieldedTRC20Parameters.newBuilder();
-    privateTRC20Builder.addShieldedSpends(note1Builder.build());
+    privateTRC20Builder
+        .addShieldedSpends(getSpendNote(infoById.get(), mintPrivateParam1, contractAddress));
 
     //ReceiveNote 1
     GrpcAPI.ReceiveNote.Builder revNoteBuilder = GrpcAPI.ReceiveNote.newBuilder();
     FullViewingKey fullViewingKey = sk.fullViewingKey();
     IncomingViewingKey incomingViewingKey = fullViewingKey.inViewingKey();
     PaymentAddress paymentAddress = incomingViewingKey.address(new DiversifierT().random()).get();
-    long revValue = 10;
+    long revValue = 1;
     byte[] memo = new byte[512];
-    byte[] rcm = grpcClient.getRcm().getValue().toByteArray();
+    byte[] rcm = WalletApi.getRcm().get().getValue().toByteArray();
     String paymentAddressStr = KeyIo.encodePaymentAddress(paymentAddress);
     GrpcAPI.Note revNote = getNote(revValue, paymentAddressStr, rcm, memo);
     revNoteBuilder.setNote(revNote);
     privateTRC20Builder.addShieldedReceives(revNoteBuilder.build());
 
+    //ReceiveNote 2
+    GrpcAPI.ReceiveNote.Builder revNoteBuilder2 = GrpcAPI.ReceiveNote.newBuilder();
+    PaymentAddress paymentAddress2 = incomingViewingKey.address(new DiversifierT().random()).get();
+    String paymentAddressStr2 = KeyIo.encodePaymentAddress(paymentAddress2);
+    long revValue2 = 1;
+    byte[] memo2 = new byte[512];
+    byte[] rcm2 = WalletApi.getRcm().get().getValue().toByteArray();
+
+    GrpcAPI.Note revNote2 = getNote(revValue2, paymentAddressStr2, rcm2, memo2);
+    revNoteBuilder2.setNote(revNote2);
+    privateTRC20Builder.addShieldedReceives(revNoteBuilder2.build());
+
+
     ExpandedSpendingKey expsk = sk.expandedSpendingKey();
     privateTRC20Builder.setAsk(ByteString.copyFrom(expsk.getAsk()));
     privateTRC20Builder.setNsk(ByteString.copyFrom(expsk.getNsk()));
     privateTRC20Builder.setOvk(ByteString.copyFrom(expsk.getOvk()));
     privateTRC20Builder.setShieldedTRC20ContractAddress(ByteString.copyFrom(contractAddress));
-    GrpcAPI.ShieldedTRC20Parameters transferParam = grpcClient
+    GrpcAPI.ShieldedTRC20Parameters transferParam = WalletApi
         .createShieldedContractParameters(privateTRC20Builder.build());
     triggerTransfer(contractAddress, callerAddress, privateKey,
         transferParam.getTriggerContractInput());
@@ -158,45 +153,30 @@ public static void burnDemo() throws ZksnarkException, InterruptedException {
     byte[] callerAddress = WalletApi.decodeFromBase58Check(pubAddress);
     SpendingKey sk = new SpendingKey(ByteArray.fromHexString(spendingKey));
     GrpcAPI.PrivateShieldedTRC20Parameters mintPrivateParam1 = mintParams(
-        privateKey, 10, shieldedTRC20ContractAddress);
-    setAllowance(10);
-    GrpcAPI.ShieldedTRC20Parameters mintParam1 = grpcClient.createShieldedContractParameters(
+        privateKey, 1, shieldedTRC20ContractAddress);
+    setAllowance(1);
+    GrpcAPI.ShieldedTRC20Parameters mintParam1 = WalletApi.createShieldedContractParameters(
         mintPrivateParam1);
-    long value = 10;
     String mintInput1 = mintParam1.getTriggerContractInput();
     String txid = triggerMint(contractAddress, callerAddress, privateKey, mintInput1);
 
     // SpendNoteTRC20 1
-    Optional<TransactionInfo> infoById = grpcClient.getTransactionInfoById(txid);
-    while (infoById.get().getLogList().size() < 2) {
-      logger.info("Can not get transaction info, please wait....");
-      Thread.sleep(5000);
-      infoById = grpcClient.getTransactionInfoById(txid);
-    }
-    byte[] tx1Data = infoById.get().getLog(1).getData().toByteArray();
-    long pos = bytes32Tolong(ByteArray.subArray(tx1Data, 0, 32));
-    byte[] contractResult = triggerGetPath(contractAddress, pos);
-    byte[] path = ByteArray.subArray(contractResult, 32, 1056);
-    byte[] root = ByteArray.subArray(contractResult, 0, 32);
-    GrpcAPI.SpendNoteTRC20.Builder note1Builder = GrpcAPI.SpendNoteTRC20.newBuilder();
-    note1Builder.setAlpha(ByteString.copyFrom(grpcClient.getRcm().getValue().toByteArray()));
-    note1Builder.setPos(pos);
-    note1Builder.setPath(ByteString.copyFrom(path));
-    note1Builder.setRoot(ByteString.copyFrom(root));
-    note1Builder.setNote(mintPrivateParam1.getShieldedReceives(0).getNote());
+    Optional<TransactionInfo> infoById = waitToGetTransactionInfo(txid);
     GrpcAPI.PrivateShieldedTRC20Parameters.Builder privateTRC20Builder = GrpcAPI
         .PrivateShieldedTRC20Parameters.newBuilder();
-    privateTRC20Builder.addShieldedSpends(note1Builder.build());
+    privateTRC20Builder
+        .addShieldedSpends(getSpendNote(infoById.get(), mintPrivateParam1, contractAddress));
 
     ExpandedSpendingKey expsk = sk.expandedSpendingKey();
     privateTRC20Builder.setAsk(ByteString.copyFrom(expsk.getAsk()));
     privateTRC20Builder.setNsk(ByteString.copyFrom(expsk.getNsk()));
     privateTRC20Builder.setOvk(ByteString.copyFrom(expsk.getOvk()));
-    BigInteger toAmount = BigInteger.valueOf(10).multiply(scalingFactorBi);
+    BigInteger toAmount = BigInteger.valueOf(1).multiply(scalingFactorBi);
     privateTRC20Builder.setToAmount(toAmount.toString());
     privateTRC20Builder.setTransparentToAddress(ByteString.copyFrom(callerAddress));
     privateTRC20Builder.setShieldedTRC20ContractAddress(ByteString.copyFrom(contractAddress));
-    GrpcAPI.ShieldedTRC20Parameters burnParam = grpcClient.createShieldedContractParameters(privateTRC20Builder.build());
+    GrpcAPI.ShieldedTRC20Parameters burnParam = WalletApi
+        .createShieldedContractParameters(privateTRC20Builder.build());
 
     triggerBurn(contractAddress, callerAddress, privateKey,
         burnParam.getTriggerContractInput());
@@ -236,7 +216,7 @@ private static String triggerTransfer(
         "0",
         0,
         callerAddress, privateKey);
-    Optional<TransactionInfo> infoById = grpcClient.getTransactionInfoById(txid);
+    Optional<TransactionInfo> infoById = WalletApi.getTransactionInfoById(txid);
     return txid;
   }
 
@@ -341,7 +321,7 @@ private static String triggerContract(byte[] contractAddress, String method, Str
     String txid = ByteArray.toHexString(Sha256Sm3Hash.hash(
         transaction.getRawData().toByteArray()));
     System.out.println("trigger txid = " + txid);
-    grpcClient.broadcastTransaction(transaction);
+    WalletApi.broadcastTransaction(transaction);
     return txid;
   }
 
@@ -430,7 +410,7 @@ private static GrpcAPI.PrivateShieldedTRC20Parameters mintParams(String privKey,
     IncomingViewingKey incomingViewingKey = fullViewingKey.inViewingKey();
     PaymentAddress paymentAddress = incomingViewingKey.address(new DiversifierT().random()).get();
     byte[] memo = new byte[512];
-    byte[] rcm = grpcClient.getRcm().getValue().toByteArray();
+    byte[] rcm = WalletApi.getRcm().get().getValue().toByteArray();
     String paymentAddressStr = KeyIo.encodePaymentAddress(paymentAddress);
     GrpcAPI.Note revNote = getNote(value, paymentAddressStr, rcm, memo);
     revNoteBuilder.setNote(revNote);
@@ -465,13 +445,41 @@ private static byte[] triggerGetPath(byte[] contractAddress, long pos) {
     return result;
   }
 
+  private static Optional<TransactionInfo> waitToGetTransactionInfo(String txid)
+      throws InterruptedException {
+    logger.info("mint txid: " + txid);
+    Optional<TransactionInfo> infoById = WalletApi.getTransactionInfoById(txid);
+    while (infoById.get().getLogList().size() < 2) {
+      logger.info("Can not get transaction info, please wait....");
+      Thread.sleep(5000);
+      return WalletApi.getTransactionInfoById(txid);
+    }
+    return null;
+  }
+
+  private static SpendNoteTRC20 getSpendNote(TransactionInfo txInfo,
+      PrivateShieldedTRC20Parameters mintPrivateParam1, byte[] contractAddress) {
+    byte[] tx1Data = txInfo.getLog(1).getData().toByteArray();
+    long pos = bytes32ToLong(ByteArray.subArray(tx1Data, 0, 32));
+    byte[] contractResult = triggerGetPath(contractAddress, pos);
+    byte[] path = ByteArray.subArray(contractResult, 32, 1056);
+    byte[] root = ByteArray.subArray(contractResult, 0, 32);
+    GrpcAPI.SpendNoteTRC20.Builder noteBuilder = GrpcAPI.SpendNoteTRC20.newBuilder();
+    noteBuilder.setAlpha(ByteString.copyFrom(WalletApi.getRcm().get().getValue().toByteArray()));
+    noteBuilder.setPos(pos);
+    noteBuilder.setPath(ByteString.copyFrom(path));
+    noteBuilder.setRoot(ByteString.copyFrom(root));
+    noteBuilder.setNote(mintPrivateParam1.getShieldedReceives(0).getNote());
+    return noteBuilder.build();
+  }
+
   private static byte[] longTo32Bytes(long value) {
     byte[] longBytes = ByteArray.fromLong(value);
     byte[] zeroBytes = new byte[24];
     return ByteUtil.merge(zeroBytes, longBytes);
   }
 
-  private static long bytes32Tolong(byte[] value) {
+  private static long bytes32ToLong(byte[] value) {
     return ByteArray.toLong(value);
   }
 }