From 9b74527618912919158574740656d42e96b2520f Mon Sep 17 00:00:00 2001
From: greg <greg.shuflin@protonmail.com>
Date: Mon, 9 Jan 2017 04:26:42 -0800
Subject: [PATCH] Control printing eval steps with flags

---
 src/eval.rs | 12 ++++++++++--
 src/main.rs | 24 ++++++++++++++++++------
 2 files changed, 28 insertions(+), 8 deletions(-)

diff --git a/src/eval.rs b/src/eval.rs
index 39753f2..9dd92f4 100644
--- a/src/eval.rs
+++ b/src/eval.rs
@@ -23,15 +23,22 @@ pub struct Evaluator<'a> {
     functions: HashMap<String, Function>,
     variables: HashMap<String, Expression>,
     stdout: BufWriter<Stdout>,
+    pub trace_evaluation: bool,
 }
 
 impl<'a> Evaluator<'a> {
+    pub fn new_with_opts(parent: Option<&'a Evaluator>, trace_evaluation: bool) -> Evaluator<'a> {
+        let mut e = Evaluator::new(parent);
+        e.trace_evaluation = trace_evaluation;
+        e
+    }
     pub fn new(parent: Option<&'a Evaluator>) -> Evaluator<'a> {
         Evaluator {
             functions: HashMap::new(),
             variables: HashMap::new(),
             parent: parent,
             stdout: BufWriter::new(::std::io::stdout()),
+            trace_evaluation: parent.map_or(false, |e| e.trace_evaluation),
         }
     }
 
@@ -126,7 +133,9 @@ impl<'a> Evaluator<'a> {
     }
 
     fn step(&mut self, node: Statement) -> Statement {
-        println!("Step: {:?}", node);
+        if self.trace_evaluation {
+            println!("Step: {:?}", node);
+        }
         let (new_node, side_effect) = self.reduce_astnode(node);
         if let Some(s) = side_effect {
             self.perform_side_effect(s);
@@ -280,7 +289,6 @@ impl<'a> Evaluator<'a> {
     }
 
     fn reduce_binop(&mut self, op: Rc<String>, left: Expression, right: Expression) -> Expression {
-        println!("Got op {:?}, l: {:?}, r: {:?}", op, left, right);
         let truthy = Number(1.0);
         let falsy = Null;
         match (&op[..], left, right) {
diff --git a/src/main.rs b/src/main.rs
index 201df14..be74b6c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -24,12 +24,13 @@ mod llvm_wrap;
 fn main() {
     let option_matches =
         program_options().parse(std::env::args()).expect("Could not parse options");
+    let trace = option_matches.opt_present("t");
     match option_matches.free[..] {
         [] | [_] => {
-            run_repl();
+            run_repl(trace);
         }
         [_, ref filename, _..] => {
-            run_noninteractive(filename, !option_matches.opt_present("i"));
+            run_noninteractive(filename, !option_matches.opt_present("i"), trace);
         }
     };
 }
@@ -39,10 +40,13 @@ fn program_options() -> getopts::Options {
     options.optflag("i",
                     "interpret",
                     "Interpret source file instead of compiling");
+    options.optflag("t",
+                    "trace-evaluation",
+                    "Print out trace of evaluation");
     options
 }
 
-fn run_noninteractive(filename: &str, compile: bool) {
+fn run_noninteractive(filename: &str, compile: bool, trace_evaluation: bool) {
     let mut source_file = File::open(&Path::new(filename)).unwrap();
     let mut buffer = String::new();
     source_file.read_to_string(&mut buffer).unwrap();
@@ -67,7 +71,7 @@ fn run_noninteractive(filename: &str, compile: bool) {
     if compile {
         compilation_sequence(ast, filename);
     } else {
-        let mut evaluator = Evaluator::new(None);
+        let mut evaluator = Evaluator::new_with_opts(None, trace_evaluation);
         let results = evaluator.run(ast);
         for result in results.iter() {
             println!("{}", result);
@@ -75,12 +79,13 @@ fn run_noninteractive(filename: &str, compile: bool) {
     }
 }
 
-fn run_repl() {
+fn run_repl(trace_evaluation: bool) {
     println!("Schala v 0.02");
     let initial_state = InterpreterState {
         show_tokens: false,
         show_parse: false,
-        evaluator: Evaluator::new(None),
+        show_eval: trace_evaluation,
+        evaluator: Evaluator::new_with_opts(None, trace_evaluation),
     };
     REPL::with_prompt_and_state(Box::new(repl_handler), ">> ", initial_state).run();
 }
@@ -88,6 +93,7 @@ fn run_repl() {
 struct InterpreterState<'a> {
     show_tokens: bool,
     show_parse: bool,
+    show_eval: bool,
     evaluator: Evaluator<'a>,
 }
 
@@ -106,6 +112,12 @@ impl<'a> ReplState for InterpreterState<'a> {
             ["set", "show", "parse", "false"] => {
                 self.show_parse = false;
             }
+            ["set", "show", "eval", "true"] => {
+                self.evaluator.trace_evaluation = true;
+            }
+            ["set", "show", "eval", "false"] => {
+                self.evaluator.trace_evaluation = false;
+            }
             _ => (),
         }
     }