From ed72b2b8499eb52e787b3db73951a2c27b2feae0 Mon Sep 17 00:00:00 2001
From: romanornr <romanornr@gmail.com>
Date: Sat, 18 Nov 2017 23:47:45 +0100
Subject: [PATCH] Viacoin:[RPC] getCryptoAddress fix

---
 .../wallets/viacoind/ViacoindRPCWallet.java   | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/server_extensions_extra/src/main/java/com/generalbytes/batm/server/extensions/extra/viacoin/wallets/viacoind/ViacoindRPCWallet.java b/server_extensions_extra/src/main/java/com/generalbytes/batm/server/extensions/extra/viacoin/wallets/viacoind/ViacoindRPCWallet.java
index 106b9e8..38c94fe 100644
--- a/server_extensions_extra/src/main/java/com/generalbytes/batm/server/extensions/extra/viacoin/wallets/viacoind/ViacoindRPCWallet.java
+++ b/server_extensions_extra/src/main/java/com/generalbytes/batm/server/extensions/extra/viacoin/wallets/viacoind/ViacoindRPCWallet.java
@@ -74,17 +74,22 @@ public class ViacoindRPCWallet implements IWallet{
     }
 
     @Override
-    public String getCryptoAddress(String cryptoCurrency){
-        if(!CRYPTO_CURRENCY.equalsIgnoreCase(cryptoCurrency)){
+    public String getCryptoAddress(String cryptoCurrency) {
+        if (!CRYPTO_CURRENCY.equalsIgnoreCase(cryptoCurrency)) {
             log.error("Viacoind wallet error: unknown cryptocurrency.");
             return null;
         }
-        try{
-            double balance = getClient(rpcURL).getBalance(accountName);
-            return new BigDecimal(balance);
-        }catch(BitcoinException e){
+
+        try {
+            List<String> addressesByAccount = getClient(rpcURL).getAddressesByAccount(accountName);
+            if (addressesByAccount == null || addressesByAccount.size() == 0) {
+                return null;
+            }else{
+                return addressesByAccount.get(0);
+            }
+        } catch (BitcoinException e) {
             e.printStackTrace();
-            return null
+            return null;
         }
     }