From 1099edcb1a529e5920f2af2ec86da3f9301ec812 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 3 Mar 2023 00:40:38 +0800
Subject: refactor: optimize line sperator

---
 src/main.rs | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index b7e3f23..c1535d7 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -99,7 +99,7 @@ fn start() -> Result<()> {
     match text {
         Some(text) => {
             let output = runtime.block_on(async move { acquire(&client, &config, &text).await })?;
-            println!("{output}");
+            println!("{}", output.trim());
         }
         None => run_repl(runtime, client, config, role_name)?,
     }
@@ -159,7 +159,7 @@ fn run_repl(
                 }
                 _ =  tokio::signal::ctrl_c() => {
                     *trigged_ctrlc = true;
-                    dump(" Abort current session")
+                    dump(" Abort current session.")
                 }
             }
         });
@@ -200,9 +200,9 @@ fn run_repl(
                         Some(role_) => {
                             *role = Some(role_.clone());
                         }
-                        None => dump("Unknown role"),
+                        None => dump("Unknown role."),
                     },
-                    None => dump("Usage: .role <name>"),
+                    None => dump("Usage: .role <name>."),
                 },
                 _ => {
                     dump("Unknown command. Type \".help\" for more information.");
@@ -265,8 +265,7 @@ async fn handle_input(client: &Client, config: &Config, text: &str) -> Result<()
     while let Some(part) = stream.next().await {
         let chunk = part?.data;
         if chunk == "[DONE]" {
-            println!();
-            stdout().flush().unwrap();
+            dump("\n");
             break;
         } else {
             let data: Value = serde_json::from_str(&chunk)?;
-- 
cgit v1.2.3