3 files changed, 58 insertions, 4 deletions
diff --git a/users/tazjin/rlox/src/treewalk/interpreter.rs b/users/tazjin/rlox/src/treewalk/interpreter.rs
index 32822d72face..5b47dc0248f9 100644
--- a/users/tazjin/rlox/src/treewalk/interpreter.rs
+++ b/users/tazjin/rlox/src/treewalk/interpreter.rs
@@ -1,7 +1,7 @@
 use crate::errors::{Error, ErrorKind};
 use crate::parser::{self, Block, Expr, Literal, Statement};
-use crate::treewalk::resolver;
 use crate::scanner::{self, TokenKind};
+use crate::treewalk::resolver;
 use std::collections::HashMap;
 use std::rc::Rc;
 use std::sync::RwLock;
@@ -223,9 +223,13 @@ impl Interpreter {
 
     // Interpreter itself
     pub fn interpret(&mut self, mut program: Block) -> Result<Value, Error> {
-        let globals = self.env.read()
+        let globals = self
+            .env
+            .read()
             .expect("static globals lock poisoned")
-            .values.keys().map(Clone::clone)
+            .values
+            .keys()
+            .map(Clone::clone)
             .collect::<Vec<String>>();
 
         resolver::resolve(&globals, &mut program)?;
diff --git a/users/tazjin/rlox/src/treewalk/interpreter/builtins.rs b/users/tazjin/rlox/src/treewalk/interpreter/builtins.rs
index 614f30ff3b23..0ec6ae08c354 100644
--- a/users/tazjin/rlox/src/treewalk/interpreter/builtins.rs
+++ b/users/tazjin/rlox/src/treewalk/interpreter/builtins.rs
@@ -2,8 +2,8 @@ use std::fmt;
 use std::time::{SystemTime, UNIX_EPOCH};
 
 use crate::errors::Error;
-use crate::treewalk::interpreter::Value;
 use crate::parser::Literal;
+use crate::treewalk::interpreter::Value;
 
 pub trait Builtin: fmt::Debug {
     fn arity(&self) -> usize;
diff --git a/users/tazjin/rlox/src/treewalk/mod.rs b/users/tazjin/rlox/src/treewalk/mod.rs
index d76045b91b9c..95136b2fa141 100644
--- a/users/tazjin/rlox/src/treewalk/mod.rs
+++ b/users/tazjin/rlox/src/treewalk/mod.rs
@@ -1,2 +1,52 @@
+use crate::*;
+
 pub mod interpreter;
 mod resolver;
+
+pub fn main() {
+    let mut args = env::args();
+
+    if args.len() > 2 {
+        println!("Usage: rlox [script]");
+        process::exit(1);
+    } else if let Some(file) = args.nth(1) {
+        run_file(&file);
+    } else {
+        run_prompt();
+    }
+}
+
+// Run Lox code from a file and print results to stdout
+fn run_file(file: &str) {
+    let contents = fs::read_to_string(file).expect("failed to read the input file");
+    let mut lox = treewalk::interpreter::Interpreter::create();
+    run(&mut lox, &contents);
+}
+
+// Evaluate Lox code interactively in a shitty REPL.
+fn run_prompt() {
+    let mut line = String::new();
+    let mut lox = treewalk::interpreter::Interpreter::create();
+
+    loop {
+        print!("> ");
+        io::stdout().flush().unwrap();
+        io::stdin()
+            .read_line(&mut line)
+            .expect("failed to read user input");
+        run(&mut lox, &line);
+        line.clear();
+    }
+}
+
+fn run(lox: &mut treewalk::interpreter::Interpreter, code: &str) {
+    let chars: Vec<char> = code.chars().collect();
+
+    let result = scanner::scan(&chars)
+        .and_then(|tokens| parser::parse(tokens))
+        .and_then(|program| lox.interpret(program).map_err(|e| vec![e]));
+
+    if let Err(errors) = result {
+        report_errors(errors);
+    }
+}