#include <eosio/wallet_plugin/wallet_plugin.hpp>

#include <eosio/wallet_plugin/wallet_manager.hpp>

#include <eosio/wallet_plugin/yubihsm_wallet.hpp>

#include <eosio/chain/exceptions.hpp>

#include <boost/filesystem/path.hpp>

#include <chrono>

#include <fc/io/json.hpp>

namespace fc { class variant; }

namespace eosio {

static appbase::abstract_plugin& _wallet_plugin = app().register_plugin<wallet_plugin>();

wallet_plugin::wallet_plugin() {}

wallet_manager& wallet_plugin::get_wallet_manager() {

return *wallet_manager_ptr;

}

void wallet_plugin::set_program_options(options_description& cli, options_description& cfg) {

cfg.add_options()

("wallet-dir", bpo::value<boost::filesystem::path>()->default_value("."),

"The path of the wallet files (absolute path or relative to application data dir)")

("unlock-timeout", bpo::value<int64_t>()->default_value(900),

"Timeout for unlocked wallet in seconds (default 900 (15 minutes)). "

"Wallets will automatically lock after specified number of seconds of inactivity. "

"Activity is defined as any wallet command e.g. list-wallets.")

("yubihsm-url", bpo::value<string>()->value_name("URL"),

"Override default URL of http://localhost:12345 for connecting to yubihsm-connector")

("yubihsm-authkey", bpo::value<uint16_t>()->value_name("key_num"),

"Enables YubiHSM support using given Authkey")

;

}

void wallet_plugin::plugin_initialize(const variables_map& options) {

ilog("initializing wallet plugin");

try {

wallet_manager_ptr = std::make_unique<wallet_manager>();

if (options.count("wallet-dir")) {

auto dir = options.at("wallet-dir").as<boost::filesystem::path>();

if (dir.is_relative())

dir = app().data_dir() / dir;

if( !bfs::exists(dir) )

bfs::create_directories(dir);

wallet_manager_ptr->set_dir(dir);

}

if (options.count("unlock-timeout")) {

auto timeout = options.at("unlock-timeout").as<int64_t>();

EOS_ASSERT(timeout > 0, chain::invalid_lock_timeout_exception, "Please specify a positive timeout {t}", ("t", timeout));

std::chrono::seconds t(timeout);

wallet_manager_ptr->set_timeout(t);

}

if (options.count("yubihsm-authkey")) {

uint16_t key = options.at("yubihsm-authkey").as<uint16_t>();

string connector_endpoint = "http://localhost:12345";

if(options.count("yubihsm-url"))

connector_endpoint = options.at("yubihsm-url").as<string>();

try {

wallet_manager_ptr->own_and_use_wallet("YubiHSM", make_unique<yubihsm_wallet>(connector_endpoint, key));

}FC_LOG_AND_RETHROW()

}

} FC_LOG_AND_RETHROW()

}

} // namespace eosio