Ignore Ctrl-C and enable Ctrl-D

[ldk-sample] / src / cli.rs

diff --git a/src/cli.rs b/src/cli.rs
index 7d865158e6e7e786a93b52ff99a513d766f50b08..8dbba3d31b0f5af1eabf095da2fc73b13d361d75 100644 (file)
--- a/src/cli.rs
+++ b/src/cli.rs
@@ -147,7 +147,9 @@ pub(crate) async fn poll_for_user_input<E: EventHandler>(
        inbound_payments: PaymentInfoStorage, outbound_payments: PaymentInfoStorage,
        ldk_data_dir: String, network: Network,
 ) {
-       println!("LDK startup successful. To view available commands: \"help\".");
+       println!(
+               "LDK startup successful. Enter \"help\" to view available commands. Press Ctrl-D to quit."
+       );
        println!("LDK logs are available at <your-supplied-ldk-data-dir-path>/.ldk/logs");
        println!("Local Node ID is {}.", channel_manager.get_our_node_id());
        loop {
@@ -158,6 +160,11 @@ pub(crate) async fn poll_for_user_input<E: EventHandler>(
                        break println!("ERROR: {e:#}");
                }
 
+               if line.len() == 0 {
+                       // We hit EOF / Ctrl-D
+                       break;
+               }
+
                let mut words = line.split_whitespace();
                if let Some(word) = words.next() {
                        match word {
@@ -459,6 +466,7 @@ pub(crate) async fn poll_for_user_input<E: EventHandler>(
                                                Err(e) => println!("ERROR: failed to send onion message: {:?}", e),
                                        }
                                }
+                               "quit" | "exit" => break,
                                _ => println!("Unknown command. See `\"help\" for available commands."),
                        }
                }
@@ -479,6 +487,7 @@ fn help() {
        println!("listpeers");
        println!("signmessage <message>");
        println!("sendonionmessage <node_id_1,node_id_2,..,destination_node_id>");
+       println!("quit")
 }
 
 fn node_info(channel_manager: &Arc<ChannelManager>, peer_manager: &Arc<PeerManager>) {