refactor

1 files changed, 6 insertions, 4 deletions

diff --git a/teleterm/src/cmd/watch.rs b/teleterm/src/cmd/watch.rs
index c3ff4ae..5178fa7 100644
--- a/teleterm/src/cmd/watch.rs
+++ b/teleterm/src/cmd/watch.rs
@@ -581,10 +581,7 @@ impl<S: tokio::io::AsyncRead + tokio::io::AsyncWrite + Send + 'static>
 
     fn poll_input(&mut self) -> component_future::Poll<(), Error> {
         if self.raw_screen.is_none() {
-            self.raw_screen = Some(
-                crossterm::screen::RawScreen::into_raw_mode()
-                    .context(crate::error::ToRawMode)?,
-            );
+            self.raw_screen = Some(new_raw_screen()?);
         }
         if let State::Temporary = self.state {
             self.state = State::LoggingIn {
@@ -662,6 +659,11 @@ impl<S: tokio::io::AsyncRead + tokio::io::AsyncWrite + Send + 'static>
     }
 }
 
+fn new_raw_screen() -> Result<crossterm::screen::RawScreen> {
+    crossterm::screen::RawScreen::into_raw_mode()
+        .context(crate::error::ToRawMode)
+}
+
 fn new_alternate_screen() -> Result<crossterm::screen::AlternateScreen> {
     crossterm::screen::AlternateScreen::to_alternate(false)
         .context(crate::error::ToAlternateScreen)