diff --git a/cli/src/cli_key_search_result.cpp b/cli/src/cli_key_search_result.cpp index 0d527de..fa3d009 100644 --- a/cli/src/cli_key_search_result.cpp +++ b/cli/src/cli_key_search_result.cpp @@ -21,6 +21,7 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ +#include #include #include #include @@ -37,8 +38,9 @@ static size_t highlight(console::Color color, const std::string& str, size_t pos return len; } -CliKeySearchResult::CliKeySearchResult(const Dictionary& dict) : -m_dict (dict) +CliKeySearchResult::CliKeySearchResult(const Dictionary& dict, const std::string& prefix) : +m_dict (dict), +m_prefix (prefix) { } @@ -50,7 +52,7 @@ void CliKeySearchResult::onResult(const struct libeosio::ec_keypair* key, const std::cout << "----" << std::endl; std::cout << "Found: " << pub.substr(result.pos, result.len) << std::endl; - std::cout << "Public: EOS"; + std::cout << "Public: " << m_prefix.substr(0, 3); for(size_t i = 3; i < pub.length(); ) { if (i == result.pos) { diff --git a/cli/src/cli_key_search_result.hpp b/cli/src/cli_key_search_result.hpp index ac75cb1..4f1e47b 100644 --- a/cli/src/cli_key_search_result.hpp +++ b/cli/src/cli_key_search_result.hpp @@ -24,6 +24,7 @@ #ifndef EOSIOKEYGEN_KEY_SEARCH_HELPERS_H #define EOSIOKEYGEN_KEY_SEARCH_HELPERS_H +#include #include #include #include @@ -36,13 +37,15 @@ class Dictionary; class CliKeySearchResult : public IKeySearchResult { public: - CliKeySearchResult(const Dictionary& dict); + CliKeySearchResult(const Dictionary& dict, const std::string& prefix); virtual void onResult(const struct libeosio::ec_keypair* key, const struct KeySearch::result& result); protected : const Dictionary& m_dict; + + std::string m_prefix; }; } // namespace eoskeygen