77 lines
2.2 KiB
Rust
77 lines
2.2 KiB
Rust
use std::{io::Write, sync::atomic::Ordering};
|
|
use rustyline::{Config, EditMode, ColorMode, Editor, CompletionType, error::ReadlineError};
|
|
use matrix_lang::prelude::*;
|
|
|
|
use crate::{State, helper::MatrixHelper};
|
|
|
|
pub struct Repl<'s, 'a> {
|
|
state: &'s mut State<'a>
|
|
}
|
|
|
|
impl<'s, 'a> Repl<'s, 'a> {
|
|
|
|
pub fn new(state: &'s mut State<'a>) -> Self {
|
|
Self { state }
|
|
}
|
|
|
|
fn execute(&mut self, line: String) -> Result<Value> {
|
|
let fun = self.state.compile(line)?;
|
|
let val = self.state.execute(fun)?;
|
|
Ok(val)
|
|
}
|
|
|
|
pub fn run(&mut self) {
|
|
|
|
let interupt = self.state.vm.borrow().interupt();
|
|
ctrlc::set_handler(move || {
|
|
interupt.store(Interupt::KeyboardInterupt as usize, Ordering::SeqCst);
|
|
}).unwrap();
|
|
|
|
let config = Config::builder()
|
|
.indent_size(4)
|
|
.edit_mode(EditMode::Emacs)
|
|
.check_cursor_position(true)
|
|
.completion_type(CompletionType::List)
|
|
.color_mode(if self.state.color { ColorMode::Enabled } else { ColorMode::Disabled })
|
|
.build();
|
|
|
|
let helper = MatrixHelper::new(self.state.vm.clone());
|
|
|
|
let histfile = std::env::var("MATRIX_HISTORY").ok();
|
|
|
|
let mut rl = Editor::with_config(config).unwrap();
|
|
rl.set_helper(Some(helper));
|
|
if let Some(hf) = &histfile {
|
|
rl.load_history(hf).ok();
|
|
}
|
|
|
|
loop {
|
|
let line = match rl.readline(">> ") {
|
|
Ok(line) => line,
|
|
Err(ReadlineError::Eof) => break,
|
|
Err(_) => continue,
|
|
};
|
|
|
|
rl.add_history_entry(&line).ok();
|
|
|
|
match self.execute(line) {
|
|
Ok(val) => {
|
|
if val != Value::Nil {
|
|
if self.state.color {
|
|
println!("{val:#}");
|
|
} else {
|
|
println!("{val}");
|
|
}
|
|
}
|
|
}
|
|
Err(err) => crate::error(err, &self.state),
|
|
};
|
|
std::io::stdout().flush().ok();
|
|
}
|
|
|
|
if let Some(hf) = &histfile {
|
|
rl.save_history(hf).ok();
|
|
}
|
|
}
|
|
|
|
}
|