From 5fdabb4320fad53aa5fb54dc5c9f1e7e6bea965a Mon Sep 17 00:00:00 2001
From: Jeremy Soller <jackpot51@gmail.com>
Date: Sun, 23 Jul 2017 11:38:21 -0600
Subject: [PATCH] Implement size and is_tty with termios on Redox

---
 Cargo.toml  |  5 +++++
 src/lib.rs  |  6 ++++++
 src/size.rs | 21 +++++++++++++--------
 src/tty.rs  | 11 +++++++++--
 4 files changed, 33 insertions(+), 10 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index dd2495a..9f5bbeb 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -11,3 +11,8 @@ exclude = ["target", "CHANGELOG.md", "image.png", "Cargo.lock"]
 
 [target.'cfg(not(target_os = "redox"))'.dependencies]
 libc = "0.2.8"
+
+
+[target.'cfg(target_os = "redox")'.dependencies]
+redox_syscall = "0.1"
+redox_termios = "0.1"
diff --git a/src/lib.rs b/src/lib.rs
index 7e286bb..8dd4d2f 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -17,6 +17,12 @@ extern crate libc;
 #[cfg(not(target_os = "redox"))]
 mod termios;
 
+#[cfg(target_os = "redox")]
+extern crate redox_termios;
+
+#[cfg(target_os = "redox")]
+extern crate syscall;
+
 mod async;
 pub use async::{AsyncReader, async_stdin};
 
diff --git a/src/size.rs b/src/size.rs
index 19f377b..3317ce5 100644
--- a/src/size.rs
+++ b/src/size.rs
@@ -57,16 +57,21 @@ pub fn terminal_size() -> io::Result<(u16, u16)> {
 /// Get the size of the terminal.
 #[cfg(target_os = "redox")]
 pub fn terminal_size() -> io::Result<(u16, u16)> {
-    use std::env;
+    use redox_termios;
+    use syscall;
 
-    let width = try!(env::var("COLUMNS").map_err(|x| io::Error::new(io::ErrorKind::NotFound, x)))
-        .parse()
-        .unwrap_or(0);
-    let height = try!(env::var("LINES").map_err(|x| io::Error::new(io::ErrorKind::NotFound, x)))
-        .parse()
-        .unwrap_or(0);
+    if let Ok(fd) = syscall::dup(1, b"winsize") {
+        let mut winsize = redox_termios::Winsize::default();
+        let res = syscall::read(fd, &mut winsize);
+        let _ = syscall::close(fd);
+        if let Ok(count) = res {
+            if count == winsize.len() {
+                return Ok((winsize.ws_col, winsize.ws_row));
+            }
+        }
+    }
 
-    Ok((width, height))
+    Err(io::Error::new(io::ErrorKind::Other, "Unable to get the terminal size."))
 }
 
 #[cfg(test)]
diff --git a/src/tty.rs b/src/tty.rs
index 9788e54..90a97e0 100644
--- a/src/tty.rs
+++ b/src/tty.rs
@@ -11,8 +11,15 @@ pub fn is_tty<T: AsRawFd>(stream: &T) -> bool {
 
 /// This will panic.
 #[cfg(target_os = "redox")]
-pub fn is_tty<T: AsRawFd>(_stream: &T) -> bool {
-    unimplemented!();
+pub fn is_tty<T: AsRawFd>(stream: &T) -> bool {
+    use syscall;
+
+    if let Ok(fd) = syscall::dup(stream.as_raw_fd(), b"termios") {
+        let _ = syscall::close(fd);
+        true
+    } else {
+        false
+    }
 }
 
 /// Get the TTY device.