add get contract info api

Author: taihaofu

src/main/java/org/tron/walletcli/Client.java

@@ -56,6 +56,7 @@
 import org.tron.protos.contract.SmartContractOuterClass.SmartContract;
 import org.tron.protos.Protocol.Transaction;
 import org.tron.protos.Protocol.TransactionInfo;
+import org.tron.protos.contract.SmartContractOuterClass.SmartContractDataWrapper;
 import org.tron.walletserver.WalletApi;
 
 
@@ -108,6 +109,7 @@ public class Client {
       "GetBrokerage",
       "GetChainParameters",
       "GetContract contractAddress",
+      "GetContractInfo contractAddress",
       "GetDelegatedResource",
       "GetDelegatedResourceAccountIndex",
       "GetDiversifier",
@@ -237,6 +239,7 @@ public class Client {
       "GetBrokerage",
       "GetChainParameters",
       "GetContract",
+      "GetContractInfo",
       "GetDelegatedResource",
       "GetDelegatedResourceAccountIndex",
       "GetDiversifier",
@@ -2237,6 +2240,28 @@ private void getContract(String[] parameters) {
     }
   }
 
+  private void getContractInfo(String[] parameters) {
+    if (parameters == null ||
+        parameters.length != 1) {
+      System.out.println("Using getContractInfo needs 1 parameter like: ");
+      System.out.println("GetContractInfo contractAddress");
+      return;
+    }
+
+    byte[] addressBytes = WalletApi.decodeFromBase58Check(parameters[0]);
+    if (addressBytes == null) {
+      System.out.println("GetContractInfo: invalid address !!!");
+      return;
+    }
+
+    SmartContractDataWrapper contractDeployContract = WalletApi.getContractInfo(addressBytes);
+    if (contractDeployContract != null) {
+      System.out.println(Utils.formatMessageString(contractDeployContract));
+    } else {
+      System.out.println("Query contract failed !!!");
+    }
+  }
+
   private void generateAddress() {
     AddressPrKeyPairMessage result = walletApiWrapper.generateAddress();
     if (null != result) {
@@ -4078,6 +4103,10 @@ private void run() {
               getContract(parameters);
               break;
             }
+            case "getcontractinfo": {
+              getContractInfo(parameters);
+              break;
+            }
             case "generateaddress": {
               generateAddress();
               break;

src/main/java/org/tron/walletserver/GrpcClient.java

@@ -62,6 +62,7 @@
 import org.tron.protos.Protocol.Transaction;
 import org.tron.protos.Protocol.TransactionInfo;
 import org.tron.protos.Protocol.TransactionSign;
+import org.tron.protos.contract.SmartContractOuterClass.SmartContractDataWrapper;
 import org.tron.protos.contract.SmartContractOuterClass.TriggerSmartContract;
 import org.tron.protos.contract.SmartContractOuterClass.UpdateEnergyLimitContract;
 import org.tron.protos.contract.SmartContractOuterClass.UpdateSettingContract;
@@ -838,6 +839,12 @@ public SmartContract getContract(byte[] address) {
     return blockingStubFull.getContract(bytesMessage);
   }
 
+  public SmartContractDataWrapper getContractInfo(byte[] address) {
+    ByteString byteString = ByteString.copyFrom(address);
+    BytesMessage bytesMessage = BytesMessage.newBuilder().setValue(byteString).build();
+    return blockingStubFull.getContractInfo(bytesMessage);
+  }
+
   public TransactionExtention accountPermissionUpdate(
       AccountPermissionUpdateContract request) {
     return blockingStubFull.accountPermissionUpdate(request);

src/main/java/org/tron/walletserver/WalletApi.java

@@ -108,6 +108,7 @@
 import org.tron.protos.contract.ShieldContract.OutputPointInfo;
 import org.tron.protos.contract.ShieldContract.ShieldedTransferContract;
 import org.tron.protos.contract.ShieldContract.SpendDescription;
+import org.tron.protos.contract.SmartContractOuterClass.SmartContractDataWrapper;
 import org.tron.protos.contract.SmartContractOuterClass.TriggerSmartContract;
 import org.tron.protos.contract.StorageContract.BuyStorageBytesContract;
 import org.tron.protos.contract.StorageContract.BuyStorageContract;
@@ -139,6 +140,7 @@
 import org.tron.protos.contract.WitnessContract.WitnessUpdateContract;
 import org.tron.protos.contract.MarketContract.MarketCancelOrderContract;
 import org.tron.protos.contract.MarketContract.MarketSellAssetContract;
+import org.tron.protos.contract.MarketContract.MarketSellAssetContract;
 
 import java.io.File;
 import java.io.IOException;
@@ -2208,6 +2210,10 @@ public static SmartContract getContract(byte[] address) {
     return rpcCli.getContract(address);
   }
 
+  public static SmartContractDataWrapper getContractInfo(byte[] address) {
+    return rpcCli.getContractInfo(address);
+  }
+
   public boolean accountPermissionUpdate(byte[] owner, String permissionJson)
       throws CipherException, IOException, CancelException {
     AccountPermissionUpdateContract contract =

src/main/protos/api/api.proto

@@ -411,6 +411,9 @@ service Wallet {
   rpc GetContract (BytesMessage) returns (SmartContract) {
   }
 
+  rpc GetContractInfo (BytesMessage) returns (SmartContractDataWrapper) {
+  }
+
   rpc TriggerContract (TriggerSmartContract) returns (TransactionExtention) {
   }
 

src/main/protos/core/contract/smart_contract.proto

@@ -87,3 +87,8 @@ message UpdateEnergyLimitContract {
     bytes contract_address = 2;
     int64 origin_energy_limit = 3;
 }
+
+message SmartContractDataWrapper {
+    SmartContract smart_contract = 1;
+    bytes runtimecode = 2;
+}