improvement: use helper function for terminal cleanup

src/main.rs

@@ -10,6 +10,7 @@ use crate::tui::app::App;
 
 mod tui;
 mod net;
+mod util;
 
 fn main() -> Result<()>  {
     let args: Vec<String> = env::args().collect();
@@ -32,7 +33,6 @@ fn main() -> Result<()>  {
     let app = App::default();
     app.run(&mut terminal, host);
 
-    stdout().execute(LeaveAlternateScreen)?;
-    disable_raw_mode()?;
+    util::terminal::cleanup();
     Ok(())
 }

src/tui/app.rs

@@ -1,12 +1,10 @@
-use std::io::stdout;
 use std::sync::{Arc, RwLock};
 use std::time::Duration;
 use std::net::TcpStream;
 use std::sync::mpsc::{channel};
 
-use crossterm::{event, ExecutableCommand};
+use crossterm::event;
 use crossterm::event::{Event, KeyCode};
-use crossterm::terminal::{disable_raw_mode, LeaveAlternateScreen};
 use ratatui::backend::Backend;
 use ratatui::style::{Color, Style};
 use ratatui::Terminal;
@@ -16,7 +14,7 @@ use stblib::stbchat::net::{IncomingPacketStream, OutgoingPacketStream};
 
 use tui_textarea::TextArea;
 
-use crate::net;
+use crate::{net, util};
 use crate::tui::app::AppEvent::SendMessage;
 
 #[derive(Default, Clone)]
@@ -82,8 +80,7 @@ impl App {
         let stream = match TcpStream::connect(host) {
             Ok(tcp_stream) => tcp_stream,
             Err(_) => {
-                stdout().execute(LeaveAlternateScreen).unwrap();
-                disable_raw_mode().unwrap();
+                util::terminal::cleanup();
                 eprintln!("Server {}:{} is unreachable. Check if the server is online.", self.address, self.port);
                 std::process::exit(1);
             }

src/util/mod.rs

@@ -0,0 +1 @@
+pub mod terminal;

src/util/terminal.rs

@@ -0,0 +1,8 @@
+use std::io::stdout;
+use crossterm::ExecutableCommand;
+use crossterm::terminal::{disable_raw_mode, LeaveAlternateScreen};
+
+pub fn cleanup() {
+    stdout().execute(LeaveAlternateScreen).unwrap();
+    disable_raw_mode().unwrap();
+}