From 0fed41e6afbf32d274660cc3ec435686901d12c4 Mon Sep 17 00:00:00 2001
From: Arnaud Bailly <arnaud.bailly@iohk.io>
Date: Wed, 25 Sep 2024 14:37:06 +0200
Subject: Introduce basic REPL

---
 rust/src/lambda.rs | 33 ++++++++++++++++++++++++++++++++-
 1 file changed, 32 insertions(+), 1 deletion(-)

(limited to 'rust/src/lambda.rs')

diff --git a/rust/src/lambda.rs b/rust/src/lambda.rs
index 97b6f08..416c31e 100644
--- a/rust/src/lambda.rs
+++ b/rust/src/lambda.rs
@@ -1,5 +1,9 @@
 use rand::Rng;
-use std::{collections::HashMap, fs::read_to_string};
+use std::{
+    collections::HashMap,
+    fs::read_to_string,
+    io::{BufRead, BufReader, Read, Write},
+};
 
 mod ast;
 use ast::*;
@@ -58,6 +62,33 @@ pub fn eval_file(file_name: &str) -> String {
         .join(" ")
 }
 
+pub fn repl<I: Read, O: Write>(inp: &mut I, outp: &mut O) {
+    let mut env = Environment::new();
+    let mut reader = BufReader::new(inp);
+    loop {
+        let mut input = String::new();
+        write!(outp, "> ").unwrap();
+        outp.flush().unwrap();
+        match reader.read_line(&mut input) {
+            Ok(0) => break,
+            Ok(_) => (),
+            Err(e) => {
+                writeln!(outp, "{}", e).unwrap();
+                break;
+            }
+        }
+        let values = parse(&input);
+        let results = values
+            .iter()
+            .map(|v| eval(v, &mut env))
+            .collect::<Vec<Value>>();
+        for result in results {
+            writeln!(outp, "{}", result).unwrap();
+            outp.flush().unwrap();
+        }
+    }
+}
+
 fn eval_all(values: &[Value]) -> Vec<Value> {
     let mut env = Environment::new();
     values.iter().map(|v| eval(v, &mut env)).collect()
-- 
cgit v1.2.3