From c26d2f0a97ca63c530ace19d453ab3156e63fadc Mon Sep 17 00:00:00 2001
From: Matthias Devlamynck <matthias@chaplean.coop>
Date: Tue, 11 Jul 2017 10:05:36 +0200
Subject: [PATCH] Implement hide cursor wrapper type

---
 src/cursor.rs | 46 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/src/cursor.rs b/src/cursor.rs
index 5cfa72e..541507e 100644
--- a/src/cursor.rs
+++ b/src/cursor.rs
@@ -1,6 +1,7 @@
 //! Cursor movement.
 
 use std::fmt;
+use std::ops;
 use std::io::{self, Write, Error, ErrorKind, Read};
 use async::async_stdin;
 use std::time::{SystemTime, Duration};
@@ -135,3 +136,48 @@ impl<W: Write> DetectCursorPos for W {
         Ok((cx, cy))
     }
 }
+
+/// Hide the cursor for the lifetime of this struct.
+/// It will hide the cursor on creation with from() and show it back on drop().
+pub struct HideCursor<W: Write> {
+    /// The output target.
+    output: W,
+}
+
+impl<W: Write> HideCursor<W> {
+    /// Create a hide cursor wrapper struct for the provided output and hides the cursor.
+    pub fn from(mut output: W) -> Self {
+        write!(output, "{}", Hide).expect("hide the cursor");
+        HideCursor { output: output }
+    }
+}
+
+impl<W: Write> Drop for HideCursor<W> {
+    fn drop(&mut self) {
+        write!(self, "{}", Show).expect("show the cursor");
+    }
+}
+
+impl<W: Write> ops::Deref for HideCursor<W> {
+    type Target = W;
+
+    fn deref(&self) -> &W {
+        &self.output
+    }
+}
+
+impl<W: Write> ops::DerefMut for HideCursor<W> {
+    fn deref_mut(&mut self) -> &mut W {
+        &mut self.output
+    }
+}
+
+impl<W: Write> Write for HideCursor<W> {
+    fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
+        self.output.write(buf)
+    }
+
+    fn flush(&mut self) -> io::Result<()> {
+        self.output.flush()
+    }
+}