Add ln addr to example cli

Author: benthecarman

examples/cli/src/main.rs

@@ -49,6 +49,13 @@ enum Commands {
 	Transactions,
 	/// List our lightning Channels
 	Channels,
+	/// Get the current lightning address
+	GetLightningAddress,
+	/// Register a lightning address
+	RegisterLightningAddress {
+		/// The lightning address name to register
+		name: String,
+	},
 	/// Clear the screen
 	Clear,
 	/// Exit the application
@@ -346,6 +353,14 @@ fn parse_command(input: &str) -> Result<Commands> {
 		"status" => Ok(Commands::Status),
 		"transactions" | "txs" | "tx" => Ok(Commands::Transactions),
 		"channels" | "chan" => Ok(Commands::Channels),
+		"get-lightning-address" | "get-ln-addr" | "ln-addr" => Ok(Commands::GetLightningAddress),
+		"register-lightning-address" | "register-ln-addr" => {
+			if parts.len() < 2 {
+				return Err(anyhow::anyhow!("Usage: register-lightning-address <name>"));
+			}
+			let name = parts[1].to_string();
+			Ok(Commands::RegisterLightningAddress { name })
+		},
 		"clear" | "cls" => Ok(Commands::Clear),
 		"exit" | "quit" | "q" => Ok(Commands::Exit),
 		"help" => {
@@ -542,6 +557,60 @@ async fn execute_command(command: Commands, state: &mut WalletState) -> Result<(
 				}
 			}
 		},
+		Commands::GetLightningAddress => {
+			let wallet = state.wallet();
+
+			println!("{} Fetching lightning address...", "⚡".bright_yellow());
+
+			match wallet.get_lightning_address().await {
+				Ok(Some(address)) => {
+					println!(
+						"{} Lightning address: {}",
+						"⚡".bright_green(),
+						address.bright_cyan()
+					);
+				},
+				Ok(None) => {
+					println!("{} No lightning address registered yet.", "⚡".bright_yellow());
+					println!(
+						"{} Use 'register-lightning-address <name>' to register one",
+						"Hint:".bright_yellow().bold()
+					);
+				},
+				Err(e) => {
+					return Err(anyhow::anyhow!("Failed to get lightning address: {:?}", e));
+				},
+			}
+		},
+		Commands::RegisterLightningAddress { name } => {
+			let wallet = state.wallet();
+
+			println!(
+				"{} Registering lightning address: {}...",
+				"⚡".bright_yellow(),
+				name.bright_cyan()
+			);
+
+			match wallet.register_lightning_address(name.clone()).await {
+				Ok(()) => {
+					println!("{} Lightning address registered successfully!", "✅".bright_green());
+					// Fetch and display the full address
+					match wallet.get_lightning_address().await {
+						Ok(Some(address)) => {
+							println!(
+								"{} Your lightning address: {}",
+								"⚡".bright_green(),
+								address.bright_cyan()
+							);
+						},
+						_ => {},
+					}
+				},
+				Err(e) => {
+					return Err(anyhow::anyhow!("Failed to register lightning address: {:?}", e));
+				},
+			}
+		},
 		Commands::Clear => {
 			print!("\x1B[2J\x1B[1;1H");
 			std::io::stdout().flush().unwrap();
@@ -577,6 +646,12 @@ fn print_help() {
 	println!("  {}", "channels".bright_green().bold());
 	println!("    List channels");
 	println!();
+	println!("  {}", "get-lightning-address".bright_green().bold());
+	println!("    Get the current lightning address");
+	println!();
+	println!("  {} <name>", "register-lightning-address".bright_green().bold());
+	println!("    Register a lightning address");
+	println!();
 	println!("  {}", "clear".bright_green().bold());
 	println!("    Clear the terminal screen");
 	println!();
@@ -587,7 +662,8 @@ fn print_help() {
 	println!("  balance");
 	println!("  send lnbc1... 10000");
 	println!("  receive 25000");
-	println!("  events");
+	println!("  register-lightning-address myname");
+	println!("  get-lightning-address");
 	println!();
 	println!("{}", "Note:".bright_yellow().bold());
 	println!("  The wallet will be auto-initialized on first use with:");