From 0ea49109d118fa9a3063f82494d2ca353539f76f Mon Sep 17 00:00:00 2001
From: ryan <ryan@distrust.co>
Date: Thu, 21 Dec 2023 12:12:52 -0500
Subject: [PATCH] keyfork-prompt: refactor to use lifetimes

---
 keyfork-prompt/src/alternate_screen.rs | 32 ++++++++++----------------
 keyfork-prompt/src/lib.rs              | 27 +++++++++-------------
 keyfork-prompt/src/raw_mode.rs         | 27 +++++++++-------------
 3 files changed, 34 insertions(+), 52 deletions(-)

diff --git a/keyfork-prompt/src/alternate_screen.rs b/keyfork-prompt/src/alternate_screen.rs
index eb4e486..ed22e50 100644
--- a/keyfork-prompt/src/alternate_screen.rs
+++ b/keyfork-prompt/src/alternate_screen.rs
@@ -1,7 +1,6 @@
 use std::{
     io::Write,
     os::fd::AsRawFd,
-    sync::{Arc, Mutex},
 };
 
 use crossterm::{
@@ -12,50 +11,44 @@ use crossterm::{
 
 use crate::Result;
 
-pub(crate) struct AlternateScreen<W>
+pub(crate) struct AlternateScreen<'a, W>
 where
     W: Write + AsRawFd + Sized,
 {
-    write: Arc<Mutex<W>>,
+    write: &'a mut W,
 }
 
-impl<W> AlternateScreen<W>
+impl<'a, W> AlternateScreen<'a, W>
 where
     W: Write + AsRawFd + Sized,
 {
-    pub(crate) fn new(write_handle: Arc<Mutex<W>>) -> Result<Self> {
-        let mut write = write_handle.lock().unwrap();
-        write.execute(EnterAlternateScreen)?.execute(MoveTo(0, 0))?;
-        drop(write);
+    pub(crate) fn new(write_handle: &'a mut W) -> Result<Self> {
+        write_handle.execute(EnterAlternateScreen)?.execute(MoveTo(0, 0))?;
         Ok(Self {
             write: write_handle,
         })
     }
-
-    pub(crate) fn arc_mutex(self) -> Arc<Mutex<Self>> {
-        Arc::new(Mutex::new(self))
-    }
 }
 
-impl<W> Write for AlternateScreen<W>
+impl<W> Write for AlternateScreen<'_, W>
 where
     W: Write + AsRawFd + Sized,
 {
     fn write(&mut self, buf: &[u8]) -> std::io::Result<usize> {
-        self.write.lock().unwrap().write(buf)
+        self.write.write(buf)
     }
 
     fn flush(&mut self) -> std::io::Result<()> {
-        self.write.lock().unwrap().flush()
+        self.write.flush()
     }
 }
 
-impl<W> AsRawFd for AlternateScreen<W>
+impl<W> AsRawFd for AlternateScreen<'_, W>
 where
     W: Write + AsRawFd + Sized,
 {
     fn as_raw_fd(&self) -> std::os::fd::RawFd {
-        self.write.lock().unwrap().as_raw_fd()
+        self.write.as_raw_fd()
     }
 }
 
@@ -77,12 +70,11 @@ where
 }
 */
 
-impl<W> Drop for AlternateScreen<W>
+impl<W> Drop for AlternateScreen<'_, W>
 where
     W: Write + AsRawFd + Sized,
 {
     fn drop(&mut self) {
-        let mut write_handle = self.write.lock().unwrap();
-        write_handle.execute(LeaveAlternateScreen).unwrap();
+        self.write.execute(LeaveAlternateScreen).unwrap();
     }
 }
diff --git a/keyfork-prompt/src/lib.rs b/keyfork-prompt/src/lib.rs
index 124bdda..34a5627 100644
--- a/keyfork-prompt/src/lib.rs
+++ b/keyfork-prompt/src/lib.rs
@@ -1,7 +1,6 @@
 use std::{
     io::{BufRead, BufReader, Read, Write},
     os::fd::AsRawFd,
-    sync::{Arc, Mutex},
 };
 
 use crossterm::{
@@ -29,8 +28,8 @@ pub enum Error {
 pub type Result<T, E = Error> = std::result::Result<T, E>;
 
 pub struct PromptManager<R, W> {
-    read: Arc<Mutex<BufReader<R>>>,
-    write: Arc<Mutex<W>>,
+    read: BufReader<R>,
+    write: W,
 }
 
 impl<R, W> PromptManager<R, W>
@@ -43,31 +42,27 @@ where
             return Err(Error::NotATTY);
         }
         Ok(Self {
-            read: Arc::new(Mutex::new(BufReader::new(read_handle))),
-            write: Arc::new(Mutex::new(write_handle)),
+            read: BufReader::new(read_handle),
+            write: write_handle,
         })
     }
 
-    fn alt_screen(&self) -> Result<AlternateScreen<W>> {
-        AlternateScreen::new(self.write.clone())
-    }
-
     pub fn prompt_input(&mut self, prompt: &str) -> Result<String> {
-        let mut alt_screen = self.alt_screen()?;
-        alt_screen
+        let mut terminal = AlternateScreen::new(&mut self.write)?;
+        terminal
             .execute(terminal::Clear(terminal::ClearType::All))?
             .execute(Print(prompt))?;
         let mut line = String::new();
-        self.read.lock().unwrap().read_line(&mut line)?;
+        self.read.read_line(&mut line)?;
         Ok(line)
     }
 
     // TODO: return secrecy::Secret<String>
     // TODO: write a guard drop system for raw mode
     pub fn prompt_passphrase(&mut self, prompt: &str) -> Result<String> {
-        let write = AlternateScreen::new(self.write.clone())?;
-        let mut write = RawMode::new(write.arc_mutex())?;
-        write
+        let mut terminal = AlternateScreen::new(&mut self.write)?;
+        let mut terminal = RawMode::new(&mut terminal)?;
+        terminal
             .execute(terminal::Clear(terminal::ClearType::All))?
             .execute(Print(prompt))?;
         let mut passphrase = String::new();
@@ -79,7 +74,7 @@ where
                         break;
                     }
                     KeyCode::Char(c) => {
-                        write.execute(Print("*"))?;
+                        terminal.execute(Print("*"))?;
                         passphrase.push(c);
                     }
                     _ => (),
diff --git a/keyfork-prompt/src/raw_mode.rs b/keyfork-prompt/src/raw_mode.rs
index 7cfdde9..60be05c 100644
--- a/keyfork-prompt/src/raw_mode.rs
+++ b/keyfork-prompt/src/raw_mode.rs
@@ -1,56 +1,51 @@
 use std::{
     io::Write,
     os::fd::AsRawFd,
-    sync::{Arc, Mutex},
 };
 
-use crossterm::{terminal, ExecutableCommand};
+use crossterm::terminal;
 
 use crate::Result;
 
-pub(crate) struct RawMode<W>
+pub(crate) struct RawMode<'a, W>
 where
     W: Write + AsRawFd + Sized,
 {
-    write: Arc<Mutex<W>>,
+    write: &'a mut W,
 }
 
 // TODO: fork crossterm to allow using FD from as_raw_fd()
-impl<W> RawMode<W>
+impl<'a, W> RawMode<'a, W>
 where
     W: Write + AsRawFd + Sized,
 {
-    pub(crate) fn new(write_handle: Arc<Mutex<W>>) -> Result<Self> {
+    pub(crate) fn new(write_handle: &'a mut W) -> Result<Self> {
         terminal::enable_raw_mode()?;
         Ok(Self {
             write: write_handle,
         })
     }
-
-    pub(crate) fn arc_mutex(self) -> Arc<Mutex<Self>> {
-        Arc::new(Mutex::new(self))
-    }
 }
 
-impl<W> Write for RawMode<W>
+impl<W> Write for RawMode<'_, W>
 where
     W: Write + AsRawFd + Sized,
 {
     fn write(&mut self, buf: &[u8]) -> std::io::Result<usize> {
-        self.write.lock().unwrap().write(buf)
+        self.write.write(buf)
     }
 
     fn flush(&mut self) -> std::io::Result<()> {
-        self.write.lock().unwrap().flush()
+        self.write.flush()
     }
 }
 
-impl<W> AsRawFd for RawMode<W>
+impl<W> AsRawFd for RawMode<'_, W>
 where
     W: Write + AsRawFd + Sized,
 {
     fn as_raw_fd(&self) -> std::os::fd::RawFd {
-        self.write.lock().unwrap().as_raw_fd()
+        self.write.as_raw_fd()
     }
 }
 
@@ -72,7 +67,7 @@ where
 }
 */
 
-impl<W> Drop for RawMode<W>
+impl<W> Drop for RawMode<'_, W>
 where
     W: Write + AsRawFd + Sized,
 {