From af1857a0f1d8d452affdaa769ef03bb5cd54d741 Mon Sep 17 00:00:00 2001
From: brianp <brian.o.pearce@gmail.com>
Date: Fri, 12 Jan 2024 13:02:34 +0100
Subject: [PATCH] Add TariAddress as valid address for discovering a peer

---
 applications/minotari_app_utilities/src/utilities.rs | 8 ++++++++
 applications/minotari_console_wallet/README.md       | 2 +-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/applications/minotari_app_utilities/src/utilities.rs b/applications/minotari_app_utilities/src/utilities.rs
index cbc6a22438..b6daff88d2 100644
--- a/applications/minotari_app_utilities/src/utilities.rs
+++ b/applications/minotari_app_utilities/src/utilities.rs
@@ -27,6 +27,7 @@ use log::*;
 use tari_common::exit_codes::{ExitCode, ExitError};
 use tari_common_types::{
     emoji::EmojiId,
+    tari_address::TariAddress,
     types::{BlockHash, PublicKey},
 };
 use tari_comms::{peer_manager::NodeId, types::CommsPublicKey};
@@ -82,6 +83,8 @@ impl FromStr for UniPublicKey {
             Ok(Self(emoji_id.to_public_key()))
         } else if let Ok(public_key) = PublicKey::from_hex(key) {
             Ok(Self(public_key))
+        } else if let Ok(tari_address) = TariAddress::from_hex(key) {
+            Ok(Self(tari_address.public_key().clone()))
         } else {
             Err(UniIdError::UnknownIdType)
         }
@@ -98,6 +101,7 @@ impl From<UniPublicKey> for PublicKey {
 pub enum UniNodeId {
     PublicKey(PublicKey),
     NodeId(NodeId),
+    TariAddress(TariAddress),
 }
 
 #[derive(Debug, Error)]
@@ -118,6 +122,8 @@ impl FromStr for UniNodeId {
             Ok(Self::PublicKey(public_key))
         } else if let Ok(node_id) = NodeId::from_hex(key) {
             Ok(Self::NodeId(node_id))
+        } else if let Ok(tari_address) = TariAddress::from_hex(key) {
+            Ok(Self::TariAddress(tari_address))
         } else {
             Err(UniIdError::UnknownIdType)
         }
@@ -130,6 +136,7 @@ impl TryFrom<UniNodeId> for PublicKey {
     fn try_from(id: UniNodeId) -> Result<Self, Self::Error> {
         match id {
             UniNodeId::PublicKey(public_key) => Ok(public_key),
+            UniNodeId::TariAddress(tari_address) => Ok(tari_address.public_key().clone()),
             UniNodeId::NodeId(_) => Err(UniIdError::Nonconvertible),
         }
     }
@@ -140,6 +147,7 @@ impl From<UniNodeId> for NodeId {
         match id {
             UniNodeId::PublicKey(public_key) => NodeId::from_public_key(&public_key),
             UniNodeId::NodeId(node_id) => node_id,
+            UniNodeId::TariAddress(tari_address) => NodeId::from_public_key(tari_address.public_key()),
         }
     }
 }
diff --git a/applications/minotari_console_wallet/README.md b/applications/minotari_console_wallet/README.md
index a296987fdb..69ab53b4d3 100644
--- a/applications/minotari_console_wallet/README.md
+++ b/applications/minotari_console_wallet/README.md
@@ -226,7 +226,7 @@ Maximum value UTXO   : 5538.616395 T
 
 Discover a peer on the network by public key or emoji id.
 
-`minotari_console_wallet --command "discover-peer <public key or emoji id>"`
+`minotari_console_wallet --command "discover-peer <public key or emoji id or tari address>"`
 
 example output: