danicoin/src/simplewallet/simplewallet.h

181 lines
6.3 KiB
C
Raw Normal View History

2015-05-27 12:08:46 +00:00
// Copyright (c) 2012-2015, The CryptoNote developers, The Bytecoin developers
2014-08-13 10:38:35 +00:00
//
// This file is part of Bytecoin.
//
// Bytecoin is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Bytecoin is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with Bytecoin. If not, see <http://www.gnu.org/licenses/>.
2014-03-03 22:07:58 +00:00
#pragma once
#include <memory>
#include <future>
2014-03-03 22:07:58 +00:00
#include <boost/program_options/variables_map.hpp>
#include "IWallet.h"
#include "INode.h"
2015-05-27 12:08:46 +00:00
#include "password_container.h"
#include "Common/ConsoleHandler.h"
#include "cryptonote_core/cryptonote_basic_impl.h"
#include "cryptonote_core/Currency.h"
#include "wallet/WalletHelper.h"
2014-03-03 22:07:58 +00:00
2015-05-27 12:08:46 +00:00
#include <Logging/LoggerRef.h>
#include <Logging/LoggerManager.h>
#include <System/Dispatcher.h>
#include <System/Ipv4Address.h>
namespace CryptoNote
2014-03-03 22:07:58 +00:00
{
/************************************************************************/
/* */
/************************************************************************/
class simple_wallet : public CryptoNote::INodeObserver, public CryptoNote::IWalletObserver
2014-03-03 22:07:58 +00:00
{
public:
2015-05-27 12:08:46 +00:00
simple_wallet(System::Dispatcher& dispatcher, const CryptoNote::Currency& currency, Logging::LoggerManager& log);
2014-03-03 22:07:58 +00:00
bool init(const boost::program_options::variables_map& vm);
bool deinit();
bool run();
2014-03-20 11:46:11 +00:00
void stop();
2014-03-03 22:07:58 +00:00
bool process_command(const std::vector<std::string> &args);
std::string get_commands_str();
2014-08-13 10:51:37 +00:00
2015-05-27 12:08:46 +00:00
const CryptoNote::Currency& currency() const { return m_currency; }
2014-08-13 10:51:37 +00:00
2014-03-03 22:07:58 +00:00
private:
2015-05-27 12:08:46 +00:00
Logging::LoggerMessage success_msg_writer(bool color = false) {
return logger(Logging::INFO, color ? Logging::GREEN : Logging::DEFAULT);
}
Logging::LoggerMessage fail_msg_writer() {
auto msg = logger(Logging::ERROR, Logging::BRIGHT_RED);
msg << "Error: ";
return msg;
}
2014-04-02 16:00:17 +00:00
void handle_command_line(const boost::program_options::variables_map& vm);
2014-03-03 22:07:58 +00:00
bool run_console_handler();
bool new_wallet(const std::string &wallet_file, const std::string& password);
bool open_wallet(const std::string &wallet_file, const std::string& password);
bool close_wallet();
2014-03-20 11:46:11 +00:00
bool help(const std::vector<std::string> &args = std::vector<std::string>());
2015-05-27 12:08:46 +00:00
bool exit(const std::vector<std::string> &args);
2014-03-03 22:07:58 +00:00
bool start_mining(const std::vector<std::string> &args);
bool stop_mining(const std::vector<std::string> &args);
2014-03-20 11:46:11 +00:00
bool show_balance(const std::vector<std::string> &args = std::vector<std::string>());
2014-03-03 22:07:58 +00:00
bool show_incoming_transfers(const std::vector<std::string> &args);
2014-04-29 16:26:45 +00:00
bool show_payments(const std::vector<std::string> &args);
2014-03-03 22:07:58 +00:00
bool show_blockchain_height(const std::vector<std::string> &args);
bool listTransfers(const std::vector<std::string> &args);
2014-03-03 22:07:58 +00:00
bool transfer(const std::vector<std::string> &args);
2014-04-02 16:00:17 +00:00
bool print_address(const std::vector<std::string> &args = std::vector<std::string>());
2014-03-03 22:07:58 +00:00
bool save(const std::vector<std::string> &args);
bool reset(const std::vector<std::string> &args);
2014-03-03 22:07:58 +00:00
bool set_log(const std::vector<std::string> &args);
2014-04-29 16:26:45 +00:00
bool ask_wallet_create_if_needed();
2014-03-03 22:07:58 +00:00
//---------------- IWalletObserver -------------------------
virtual void initCompleted(std::error_code result) override;
virtual void externalTransactionCreated(CryptoNote::TransactionId transactionId) override;
//----------------------------------------------------------
//----------------- INodeObserver --------------------------
virtual void localBlockchainUpdated(uint64_t height) override;
2014-04-02 16:00:17 +00:00
//----------------------------------------------------------
friend class refresh_progress_reporter_t;
class refresh_progress_reporter_t
{
public:
2015-05-27 12:08:46 +00:00
refresh_progress_reporter_t(CryptoNote::simple_wallet& simple_wallet)
2014-04-02 16:00:17 +00:00
: m_simple_wallet(simple_wallet)
, m_blockchain_height(0)
, m_blockchain_height_update_time()
, m_print_time()
{
}
void update(uint64_t height, bool force = false)
{
auto current_time = std::chrono::system_clock::now();
2014-08-13 10:51:37 +00:00
if (std::chrono::seconds(m_simple_wallet.currency().difficultyTarget() / 2) < current_time - m_blockchain_height_update_time ||
m_blockchain_height <= height) {
2014-04-02 16:00:17 +00:00
update_blockchain_height();
m_blockchain_height = (std::max)(m_blockchain_height, height);
}
if (std::chrono::milliseconds(1) < current_time - m_print_time || force)
{
2015-05-27 12:08:46 +00:00
std::cout << "Height " << height << " of " << m_blockchain_height << '\r';
2014-04-02 16:00:17 +00:00
m_print_time = current_time;
}
}
private:
void update_blockchain_height()
{
std::string err;
uint64_t blockchain_height = m_simple_wallet.m_node->getLastLocalBlockHeight();
2014-04-02 16:00:17 +00:00
if (err.empty())
{
m_blockchain_height = blockchain_height;
m_blockchain_height_update_time = std::chrono::system_clock::now();
}
else
{
2015-05-27 12:08:46 +00:00
std::cerr << "Failed to get current blockchain height: " << err;
2014-04-02 16:00:17 +00:00
}
}
private:
2015-05-27 12:08:46 +00:00
CryptoNote::simple_wallet& m_simple_wallet;
2014-04-02 16:00:17 +00:00
uint64_t m_blockchain_height;
std::chrono::system_clock::time_point m_blockchain_height_update_time;
std::chrono::system_clock::time_point m_print_time;
};
private:
std::string m_wallet_file_arg;
2014-03-03 22:07:58 +00:00
std::string m_generate_new;
std::string m_import_path;
std::string m_daemon_address;
std::string m_daemon_host;
2015-05-27 12:08:46 +00:00
uint16_t m_daemon_port;
2014-03-03 22:07:58 +00:00
std::string m_wallet_file;
2015-05-27 12:08:46 +00:00
Common::ConsoleHandler m_consoleHandler;
const CryptoNote::Currency& m_currency;
Logging::LoggerManager& logManager;
System::Dispatcher& m_dispatcher;
Logging::LoggerRef logger;
std::unique_ptr<CryptoNote::INode> m_node;
std::unique_ptr<CryptoNote::IWallet> m_wallet;
2014-04-02 16:00:17 +00:00
refresh_progress_reporter_t m_refresh_progress_reporter;
2015-05-27 12:08:46 +00:00
std::unique_ptr<std::promise<std::error_code>> m_initResultPromise;
2014-03-03 22:07:58 +00:00
};
}