From f7ac2a1a431f69d239cf22487621af9f1fc58acc Mon Sep 17 00:00:00 2001
From: Alex Bethel <abethel@cs.nmt.edu>
Date: Fri, 18 Jun 2021 16:05:50 -0500
Subject: [PATCH] Implement `read` semantics in interpreter

Also added a test file `iotest.able` to just read data and print
it (the `cat` of AbleScript I guess).
---
 able-script-test/iotest.able |  5 +++
 src/error.rs                 | 16 +++++++--
 src/interpret.rs             | 63 +++++++++++++++++++++++++++++-------
 3 files changed, 70 insertions(+), 14 deletions(-)
 create mode 100644 able-script-test/iotest.able

diff --git a/able-script-test/iotest.able b/able-script-test/iotest.able
new file mode 100644
index 0000000..b0406df
--- /dev/null
+++ b/able-script-test/iotest.able
@@ -0,0 +1,5 @@
+var data;
+loop {
+  data read;
+  data print;
+}
diff --git a/src/error.rs b/src/error.rs
index 7d1e489..940b479 100644
--- a/src/error.rs
+++ b/src/error.rs
@@ -1,14 +1,14 @@
-use std::ops::Range;
+use std::{io, ops::Range};
 
 use crate::{brian::InterpretError, lexer::Token};
 
-#[derive(Debug, Clone)]
+#[derive(Debug)]
 pub struct Error {
     pub kind: ErrorKind,
     pub span: Range<usize>,
 }
 
-#[derive(Debug, Clone)]
+#[derive(Debug)]
 pub enum ErrorKind {
     SyntaxError(String),
     UnexpectedEof,
@@ -22,6 +22,7 @@ pub enum ErrorKind {
     BfInterpretError(InterpretError),
     MismatchedArgumentError,
     MissingLhs,
+    IOError(io::Error),
 }
 
 impl Error {
@@ -35,3 +36,12 @@ impl Error {
         Self::new(ErrorKind::UnexpectedEof, index..index)
     }
 }
+
+impl From<io::Error> for Error {
+    fn from(e: io::Error) -> Self {
+        Self {
+            kind: ErrorKind::IOError(e),
+            span: 0..0,
+        }
+    }
+}
diff --git a/src/interpret.rs b/src/interpret.rs
index 29466ec..9d0bda9 100644
--- a/src/interpret.rs
+++ b/src/interpret.rs
@@ -9,12 +9,11 @@
 #![deny(missing_docs)]
 use std::{
     cell::RefCell,
-    collections::HashMap,
-    io::{stdout, Write},
+    collections::{HashMap, VecDeque},
+    io::{stdin, stdout, Read, Write},
     ops::Range,
     process::exit,
     rc::Rc,
-    usize,
 };
 
 use rand::random;
@@ -32,6 +31,13 @@ pub struct ExecEnv {
     /// top-most (newest) stack frame, and `stack[0]` is the
     /// bottom-most (oldest) stack frame.
     stack: Vec<Scope>,
+
+    /// The `read` statement maintains a buffer of up to 7 bits,
+    /// because input comes from the operating system 8 bits at a time
+    /// (via stdin) but gets delivered to AbleScript 3 bits at a time
+    /// (via the `read` statement). We store each of those bits as
+    /// booleans to facilitate easy manipulation.
+    read_buf: VecDeque<bool>,
 }
 
 /// A set of visible variable and function definitions in a single
@@ -58,6 +64,10 @@ enum HaltStatus {
     Hopback(Range<usize>),
 }
 
+/// The number of bits the `read` statement reads at once from
+/// standard input.
+pub const READ_BITS: u8 = 3;
+
 impl ExecEnv {
     /// Create a new Scope with no predefined variable definitions or
     /// other information.
@@ -65,6 +75,7 @@ impl ExecEnv {
         Self {
             // We always need at least one stackframe.
             stack: vec![Default::default()],
+            read_buf: Default::default(),
         }
     }
 
@@ -205,13 +216,15 @@ impl ExecEnv {
 
                 self.decl_var(&iden.iden, init);
             }
-            StmtKind::Functio { iden, params, body } => self.decl_var(
-                &iden.iden,
-                Value::Functio(Functio::AbleFunctio {
-                    params: params.iter().map(|iden| iden.iden.to_string()).collect(),
-                    body: body.block.to_owned(),
-                }),
-            ),
+            StmtKind::Functio { iden, params, body } => {
+                self.decl_var(
+                    &iden.iden,
+                    Value::Functio(Functio::AbleFunctio {
+                        params: params.iter().map(|iden| iden.iden.to_string()).collect(),
+                        body: body.block.to_owned(),
+                    }),
+                );
+            }
             StmtKind::BfFunctio {
                 iden,
                 tape_len,
@@ -280,7 +293,15 @@ impl ExecEnv {
                     .write_all(include_str!("rickroll").as_bytes())
                     .expect("Failed to write to stdout");
             }
-            StmtKind::Read(_) => todo!(),
+            StmtKind::Read(iden) => {
+                let mut value = 0;
+                for _ in 0..READ_BITS {
+                    value <<= 1;
+                    value += self.get_bit()? as i32;
+                }
+
+                self.get_var_mut(&iden)?.value.replace(Value::Int(value));
+            }
         }
 
         Ok(HaltStatus::Finished)
@@ -356,6 +377,26 @@ impl ExecEnv {
         Ok(())
     }
 
+    /// Get a single bit from the bit buffer, or refill it from
+    /// standard input if it is empty.
+    fn get_bit(&mut self) -> Result<bool, Error> {
+        const BITS_PER_BYTE: u8 = 8;
+
+        if self.read_buf.is_empty() {
+            let mut data = [0];
+            stdin().read_exact(&mut data)?;
+
+            for n in (0..BITS_PER_BYTE).rev() {
+                self.read_buf.push_back(((data[0] >> n) & 1) != 0);
+            }
+        }
+
+        Ok(self
+            .read_buf
+            .pop_front()
+            .expect("We just pushed to the buffer if it was empty"))
+    }
+
     /// Get the value of a variable. Throw an error if the variable is
     /// inaccessible or banned.
     fn get_var(&self, name: &Iden) -> Result<Value, Error> {