Wallet::createTransaction API introduced

3 files changed, 191 insertions, 126 deletions

diff --git a/src/wallet/wallet2_api.cpp b/src/wallet/wallet2_api.cpp
index 60e75802c..ae26a1eb3 100644
--- a/src/wallet/wallet2_api.cpp
+++ b/src/wallet/wallet2_api.cpp
@@ -66,67 +66,45 @@ using namespace cryptonote;
 
 Wallet::~Wallet() {}
 
+Transaction::~Transaction() {}
+
+
+class WalletImpl;
+
 ///////////////////////// Transaction implementation ///////////////////////////
 
 class TransactionImpl : public Transaction
 {
 public:
-    TransactionImpl(Wallet * wallet);
+    TransactionImpl(WalletImpl * wallet);
     ~TransactionImpl();
     int status() const;
     std::string errorString() const;
     bool commit();
-
-
-private:
-    std::vector<tools::wallet2::pending_tx> & transactions();
+    uint64_t dust() const;
+    uint64_t fee() const;
+    // TODO: continue with interface;
 
 private:
     friend class WalletImpl;
-    Wallet * m_wallet;
+    WalletImpl * m_wallet;
+
     int  m_status;
     std::string m_errorString;
     std::vector<tools::wallet2::pending_tx> m_pending_tx;
-
-
 };
 
 
-TransactionImpl::TransactionImpl(Wallet *wallet)
-    : m_wallet(wallet)
-{
 
-}
 
-TransactionImpl::~TransactionImpl()
+/////////////////////////////////////////////////////////////////////////////////
+string Wallet::displayAmount(uint64_t amount)
 {
-
-}
-
-int TransactionImpl::status() const
-{
-    return m_status;
-}
-
-string TransactionImpl::errorString() const
-{
-    return m_errorString;
-}
-
-bool TransactionImpl::commit()
-{
-//    while (!m_pending_tx.empty()) {
-
-//    }
-    return false;
-
-
+    return cryptonote::print_money(amount);
 }
 
 
 
-
-
 ///////////////////////// Wallet implementation ////////////////////////////////
 class WalletImpl : public Wallet
 {
@@ -151,16 +129,15 @@ public:
     bool connectToDaemon();
     uint64_t balance() const;
     uint64_t unlockedBalance() const;
-    std::string displayAmount(uint64_t amount) const;
     bool refresh();
-    bool transfer(const std::string &dst_addr, uint64_t amount);
-
+    Transaction * createTransaction(const std::string &dst_addr, uint64_t amount);
+    virtual void disposeTransaction(Transaction * t);
 
 private:
     void clearStatus();
 
 private:
-    //std::unique_ptr<tools::wallet2> m_wallet;
+    friend class TransactionImpl;
     tools::wallet2 * m_wallet;
     int  m_status;
     std::string m_errorString;
@@ -366,10 +343,6 @@ uint64_t WalletImpl::unlockedBalance() const
     return m_wallet->unlocked_balance();
 }
 
-std::string WalletImpl::displayAmount(uint64_t amount) const
-{
-    return cryptonote::print_money(amount);
-}
 
 bool WalletImpl::refresh()
 {
@@ -383,7 +356,8 @@ bool WalletImpl::refresh()
     return m_status == Status_Ok;
 }
 
-bool WalletImpl::transfer(const std::string &dst_addr, uint64_t amount)
+
+Transaction *WalletImpl::createTransaction(const string &dst_addr, uint64_t amount)
 {
     clearStatus();
     vector<cryptonote::tx_destination_entry> dsts;
@@ -395,32 +369,154 @@ bool WalletImpl::transfer(const std::string &dst_addr, uint64_t amount)
     if (fake_outs_count == 0)
         fake_outs_count = DEFAULT_MIX;
 
+    TransactionImpl * transaction = new TransactionImpl(this);
+    do {
 
-    if(!cryptonote::get_account_integrated_address_from_str(de.addr, has_payment_id, new_payment_id, m_wallet->testnet(), dst_addr)) {
-        // TODO: copy-paste 'if treating as an address fails, try as url' from simplewallet.cpp:1982
-        m_status = Status_Error;
-        m_errorString = "Invalid destination address";
-        return false;
-    }
+        if(!cryptonote::get_account_integrated_address_from_str(de.addr, has_payment_id, new_payment_id, m_wallet->testnet(), dst_addr)) {
+            // TODO: copy-paste 'if treating as an address fails, try as url' from simplewallet.cpp:1982
+            m_status = Status_Error;
+            m_errorString = "Invalid destination address";
+            break;
 
-    de.amount = amount;
-    if (de.amount <= 0) {
-        m_status = Status_Error;
-        m_errorString = "Invalid amount";
-        return false;
+        }
+
+        de.amount = amount;
+        if (de.amount <= 0) {
+            m_status = Status_Error;
+            m_errorString = "Invalid amount";
+            break;
+        }
+
+        dsts.push_back(de);
+        //std::vector<tools::wallet2::pending_tx> ptx_vector;
+        std::vector<uint8_t> extra;
+
+
+        try {
+            transaction->m_pending_tx = m_wallet->create_transactions(dsts, fake_outs_count, 0 /* unlock_time */, 0 /* unused fee arg*/, extra);
+            //      TODO: move it to transaction class
+
+        } catch (const tools::error::daemon_busy&) {
+            // TODO: make it translatable with "tr"?
+            m_errorString = tr("daemon is busy. Please try again later.");
+            m_status = Status_Error;
+        } catch (const tools::error::no_connection_to_daemon&) {
+            m_errorString = tr("no connection to daemon. Please make sure daemon is running.");
+            m_status = Status_Error;
+        } catch (const tools::error::wallet_rpc_error& e) {
+            m_errorString = tr("RPC error: ") +  e.to_string();
+            m_status = Status_Error;
+        } catch (const tools::error::get_random_outs_error&) {
+            m_errorString = tr("failed to get random outputs to mix");
+            m_status = Status_Error;
+
+        } catch (const tools::error::not_enough_money& e) {
+            m_status = Status_Error;
+            std::ostringstream writer(m_errorString);
+
+            writer << boost::format(tr("not enough money to transfer, available only %s, transaction amount %s = %s + %s (fee)")) %
+                      print_money(e.available()) %
+                      print_money(e.tx_amount() + e.fee())  %
+                      print_money(e.tx_amount()) %
+                      print_money(e.fee());
+
+        } catch (const tools::error::not_enough_outs_to_mix& e) {
+            std::ostringstream writer(m_errorString);
+            writer << tr("not enough outputs for specified mixin_count") << " = " << e.mixin_count() << ":";
+            for (const cryptonote::COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::outs_for_amount& outs_for_amount : e.scanty_outs()) {
+                writer << "\n" << tr("output amount") << " = " << print_money(outs_for_amount.amount) << ", " << tr("found outputs to mix") << " = " << outs_for_amount.outs.size();
+            }
+            m_status = Status_Error;
+        } catch (const tools::error::tx_not_constructed&) {
+            m_errorString = tr("transaction was not constructed");
+            m_status = Status_Error;
+        } catch (const tools::error::tx_rejected& e) {
+            std::ostringstream writer(m_errorString);
+            writer << (boost::format(tr("transaction %s was rejected by daemon with status: ")) % get_transaction_hash(e.tx())) <<  e.status();
+            m_status = Status_Error;
+        } catch (const tools::error::tx_sum_overflow& e) {
+            m_errorString = e.what();
+            m_status = Status_Error;
+        } catch (const tools::error::zero_destination&) {
+            m_errorString =  tr("one of destinations is zero");
+            m_status = Status_Error;
+        } catch (const tools::error::tx_too_big& e) {
+            m_errorString =  tr("failed to find a suitable way to split transactions");
+            m_status = Status_Error;
+        } catch (const tools::error::transfer_error& e) {
+            m_errorString = string(tr("unknown transfer error: ")) + e.what();
+            m_status = Status_Error;
+        } catch (const tools::error::wallet_internal_error& e) {
+            m_errorString =  string(tr("internal error: ")) + e.what();
+            m_status = Status_Error;
+        } catch (const std::exception& e) {
+            m_errorString =  string(tr("unexpected error: ")) + e.what();
+            m_status = Status_Error;
+        } catch (...) {
+            m_errorString = tr("unknown error");
+            m_status = Status_Error;
+        }
+    } while (false);
+
+    transaction->m_status = m_status;
+    transaction->m_errorString = m_errorString;
+    return transaction;
+}
+
+void WalletImpl::disposeTransaction(Transaction *t)
+{
+    delete t;
+}
+
+bool WalletImpl::connectToDaemon()
+{
+    bool result = m_wallet->check_connection();
+    m_status = result ? Status_Ok : Status_Error;
+    if (!result) {
+        m_errorString = "Error connecting to daemon at " + m_wallet->get_daemon_address();
     }
-    dsts.push_back(de);
-    std::vector<tools::wallet2::pending_tx> ptx_vector;
-    std::vector<uint8_t> extra;
+    return result;
+}
+
+void WalletImpl::clearStatus()
+{
+    m_status = Status_Ok;
+    m_errorString.clear();
+}
+
+
+
+
+TransactionImpl::TransactionImpl(WalletImpl *wallet)
+    : m_wallet(wallet)
+{
+
+}
+
+TransactionImpl::~TransactionImpl()
+{
+
+}
+
+int TransactionImpl::status() const
+{
+    return m_status;
+}
+
+string TransactionImpl::errorString() const
+{
+    return m_errorString;
+}
+
+bool TransactionImpl::commit()
+{
     try {
-        ptx_vector = m_wallet->create_transactions(dsts, fake_outs_count, 0 /* unlock_time */, 0 /* unused fee arg*/, extra);
-//      TODO: move it to transaction class
-        while (!ptx_vector.empty()) {
-            auto & ptx = ptx_vector.back();
-            m_wallet->commit_tx(ptx);
+        while (!m_pending_tx.empty()) {
+            auto & ptx = m_pending_tx.back();
+            m_wallet->m_wallet->commit_tx(ptx);
             // success_msg_writer(true) << tr("Money successfully sent, transaction ") << get_transaction_hash(ptx.tx);
             // if no exception, remove element from vector
-            ptx_vector.pop_back();
+            m_pending_tx.pop_back();
         } // TODO: extract method;
     } catch (const tools::error::daemon_busy&) {
         // TODO: make it translatable with "tr"?
@@ -429,76 +525,38 @@ bool WalletImpl::transfer(const std::string &dst_addr, uint64_t amount)
     } catch (const tools::error::no_connection_to_daemon&) {
         m_errorString = tr("no connection to daemon. Please make sure daemon is running.");
         m_status = Status_Error;
-    } catch (const tools::error::wallet_rpc_error& e) {
-        m_errorString = tr("RPC error: ") +  e.to_string();
-        m_status = Status_Error;
-    } catch (const tools::error::get_random_outs_error&) {
-        m_errorString = tr("failed to get random outputs to mix");
-        m_status = Status_Error;
-
-    } catch (const tools::error::not_enough_money& e) {
-        m_status = Status_Error;
-        std::ostringstream writer(m_errorString);
-
-        writer << boost::format(tr("not enough money to transfer, available only %s, transaction amount %s = %s + %s (fee)")) %
-        print_money(e.available()) %
-        print_money(e.tx_amount() + e.fee())  %
-        print_money(e.tx_amount()) %
-        print_money(e.fee());
-
-    } catch (const tools::error::not_enough_outs_to_mix& e) {
-        std::ostringstream writer(m_errorString);
-        writer << tr("not enough outputs for specified mixin_count") << " = " << e.mixin_count() << ":";
-        for (const cryptonote::COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::outs_for_amount& outs_for_amount : e.scanty_outs()) {
-            writer << "\n" << tr("output amount") << " = " << print_money(outs_for_amount.amount) << ", " << tr("found outputs to mix") << " = " << outs_for_amount.outs.size();
-        }
-        m_status = Status_Error;
-    } catch (const tools::error::tx_not_constructed&) {
-        m_errorString = tr("transaction was not constructed");
-        m_status = Status_Error;
     } catch (const tools::error::tx_rejected& e) {
         std::ostringstream writer(m_errorString);
         writer << (boost::format(tr("transaction %s was rejected by daemon with status: ")) % get_transaction_hash(e.tx())) <<  e.status();
         m_status = Status_Error;
-    } catch (const tools::error::tx_sum_overflow& e) {
-        m_errorString = e.what();
-        m_status = Status_Error;
-    } catch (const tools::error::zero_destination&) {
-        m_errorString =  tr("one of destinations is zero");
-        m_status = Status_Error;
-    } catch (const tools::error::tx_too_big& e) {
-        m_errorString =  tr("failed to find a suitable way to split transactions");
+    } catch (std::exception &e) {
+        m_errorString = string(tr("Unknown exception: ")) + e.what();
         m_status = Status_Error;
-    } catch (const tools::error::transfer_error& e) {
-       m_errorString = string(tr("unknown transfer error: ")) + e.what();
-       m_status = Status_Error;
-    } catch (const tools::error::wallet_internal_error& e) {
-       m_errorString =  string(tr("internal error: ")) + e.what();
-       m_status = Status_Error;
-    } catch (const std::exception& e) {
-      m_errorString =  string(tr("unexpected error: ")) + e.what();
-      m_status = Status_Error;
     } catch (...) {
-      m_errorString = tr("unknown error");
-      m_status = Status_Error;
+        m_errorString = tr("Unhandled exception");
+        LOG_ERROR(m_errorString);
+        m_status = Status_Error;
     }
+
     return m_status == Status_Ok;
 }
 
-bool WalletImpl::connectToDaemon()
+uint64_t TransactionImpl::dust() const
 {
-    bool result = m_wallet->check_connection();
-    m_status = result ? Status_Ok : Status_Error;
-    if (!result) {
-        m_errorString = "Error connecting to daemon at " + m_wallet->get_daemon_address();
+    uint32_t result = 0;
+    for (auto ptx : m_pending_tx) {
+        result += ptx.dust;
     }
     return result;
 }
 
-void WalletImpl::clearStatus()
+uint64_t TransactionImpl::fee() const
 {
-    m_status = Status_Ok;
-    m_errorString.clear();
+    uint32_t result = 0;
+    for (auto ptx : m_pending_tx) {
+        result += ptx.fee;
+    }
+    return result;
 }
 
 
diff --git a/src/wallet/wallet2_api.h b/src/wallet/wallet2_api.h
index 662264f50..c85cda2de 100644
--- a/src/wallet/wallet2_api.h
+++ b/src/wallet/wallet2_api.h
@@ -45,10 +45,12 @@ struct Transaction
         Status_Ok,
         Status_Error
     };
-
+    virtual ~Transaction() = 0;
     virtual int status() const = 0;
     virtual std::string errorString() const = 0;
     virtual bool commit() = 0;
+    virtual uint64_t dust() const = 0;
+    virtual uint64_t fee() const = 0;
 };
 
 /**
@@ -85,13 +87,12 @@ struct Wallet
     virtual bool connectToDaemon() = 0;
     virtual uint64_t balance() const = 0;
     virtual uint64_t unlockedBalance() const = 0;
-    virtual std::string displayAmount(uint64_t amount) const = 0;
+    static std::string displayAmount(uint64_t amount);
     // TODO?
     // virtual uint64_t unlockedDustBalance() const = 0;
     virtual bool refresh() = 0;
-    // TODO transfer
-    virtual bool transfer(const std::string &dst_addr, uint64_t amount) = 0;
-
+    virtual Transaction * createTransaction(const std::string &dst_addr, uint64_t amount) = 0;
+    virtual void disposeTransaction(Transaction * t) = 0;
 };
 
 /**
diff --git a/tests/libwallet_api_tests/main.cpp b/tests/libwallet_api_tests/main.cpp
index 970d9a74e..f073b47b9 100644
--- a/tests/libwallet_api_tests/main.cpp
+++ b/tests/libwallet_api_tests/main.cpp
@@ -264,14 +264,20 @@ TEST_F(WalletManagerTest, WalletRefresh)
     ASSERT_TRUE(wmgr->closeWallet(wallet1));
 }
 
-TEST_F(WalletManagerTest, WalletTransfer)
+TEST_F(WalletManagerTest, WalletTransaction)
 {
     Bitmonero::Wallet * wallet1 = wmgr->openWallet(TESTNET_WALLET_NAME, TESTNET_WALLET_PASS, true);
     // make sure testnet daemon is running
     ASSERT_TRUE(wallet1->init(TESTNET_DAEMON_ADDRESS, 0));
     ASSERT_TRUE(wallet1->refresh());
     uint64_t balance = wallet1->balance();
-    ASSERT_TRUE(wallet1->transfer(RECIPIENT_WALLET_ADDRESS, AMOUNT_10XMR));
+
+    Bitmonero::Transaction * transaction = wallet1->createTransaction(
+                RECIPIENT_WALLET_ADDRESS, AMOUNT_10XMR);
+    ASSERT_TRUE(transaction->status() == Bitmonero::Transaction::Status_Ok);
+
+    ASSERT_TRUE(wallet1->balance() == balance);
+    ASSERT_TRUE(transaction->commit());
     ASSERT_FALSE(wallet1->balance() == balance);
     ASSERT_TRUE(wmgr->closeWallet(wallet1));
 }