mirror of
https://github.com/nushell/nushell.git
synced 2024-12-11 09:43:24 +01:00
26f31da711
* Remove comment * Split delta and environment merging * Move table mode to a more logical place * Cleanup * Merge environment after reading default_env.nu * Fmt
131 lines
4.0 KiB
Rust
131 lines
4.0 KiB
Rust
use nu_cli::{eval_env_change_hook, eval_hook};
|
|
use nu_command::create_default_context;
|
|
use nu_engine::eval_block;
|
|
use nu_parser::parse;
|
|
use nu_protocol::engine::{EngineState, Stack, StateWorkingSet};
|
|
use nu_protocol::{CliError, PipelineData, Span, Value};
|
|
use nu_test_support::fs::in_directory;
|
|
use nu_test_support::Outcome;
|
|
|
|
fn outcome_err(
|
|
engine_state: &EngineState,
|
|
error: &(dyn miette::Diagnostic + Send + Sync + 'static),
|
|
) -> Outcome {
|
|
let working_set = StateWorkingSet::new(engine_state);
|
|
|
|
eprintln!("Error: {:?}", CliError(error, &working_set));
|
|
|
|
Outcome {
|
|
out: String::new(),
|
|
err: format!("{:?}", error),
|
|
}
|
|
}
|
|
|
|
fn outcome_ok(msg: String) -> Outcome {
|
|
Outcome {
|
|
out: msg,
|
|
err: String::new(),
|
|
}
|
|
}
|
|
|
|
pub fn nu_repl(cwd: &str, source_lines: &[&str]) -> Outcome {
|
|
let cwd = in_directory(cwd);
|
|
|
|
let mut engine_state = create_default_context();
|
|
let mut stack = Stack::new();
|
|
|
|
stack.add_env_var(
|
|
"PWD".to_string(),
|
|
Value::String {
|
|
val: cwd.to_string(),
|
|
span: Span::test_data(),
|
|
},
|
|
);
|
|
|
|
let mut last_output = String::new();
|
|
|
|
for (i, line) in source_lines.iter().enumerate() {
|
|
let cwd = match nu_engine::env::current_dir(&engine_state, &stack) {
|
|
Ok(d) => d,
|
|
Err(err) => {
|
|
return outcome_err(&engine_state, &err);
|
|
}
|
|
};
|
|
|
|
// Before doing anything, merge the environment from the previous REPL iteration into the
|
|
// permanent state.
|
|
if let Err(err) = engine_state.merge_env(&mut stack, &cwd) {
|
|
return outcome_err(&engine_state, &err);
|
|
}
|
|
|
|
// Check for pre_prompt hook
|
|
let config = engine_state.get_config();
|
|
if let Some(hook) = config.hooks.pre_prompt.clone() {
|
|
if let Err(err) = eval_hook(&mut engine_state, &mut stack, vec![], &hook) {
|
|
return outcome_err(&engine_state, &err);
|
|
}
|
|
}
|
|
|
|
// Check for env change hook
|
|
let config = engine_state.get_config();
|
|
if let Err(err) = eval_env_change_hook(
|
|
config.hooks.env_change.clone(),
|
|
&mut engine_state,
|
|
&mut stack,
|
|
) {
|
|
return outcome_err(&engine_state, &err);
|
|
}
|
|
|
|
// Check for pre_execution hook
|
|
let config = engine_state.get_config();
|
|
if let Some(hook) = config.hooks.pre_execution.clone() {
|
|
if let Err(err) = eval_hook(&mut engine_state, &mut stack, vec![], &hook) {
|
|
return outcome_err(&engine_state, &err);
|
|
}
|
|
}
|
|
|
|
// Eval the REPL line
|
|
let (block, delta) = {
|
|
let mut working_set = StateWorkingSet::new(&engine_state);
|
|
let (block, err) = parse(
|
|
&mut working_set,
|
|
Some(&format!("line{}", i)),
|
|
line.as_bytes(),
|
|
false,
|
|
&[],
|
|
);
|
|
|
|
if let Some(err) = err {
|
|
return outcome_err(&engine_state, &err);
|
|
}
|
|
(block, working_set.render())
|
|
};
|
|
|
|
if let Err(err) = engine_state.merge_delta(delta) {
|
|
return outcome_err(&engine_state, &err);
|
|
}
|
|
|
|
let input = PipelineData::new(Span::test_data());
|
|
let config = engine_state.get_config();
|
|
|
|
match eval_block(&engine_state, &mut stack, &block, input, false, false) {
|
|
Ok(pipeline_data) => match pipeline_data.collect_string("", config) {
|
|
Ok(s) => last_output = s,
|
|
Err(err) => return outcome_err(&engine_state, &err),
|
|
},
|
|
Err(err) => return outcome_err(&engine_state, &err),
|
|
}
|
|
|
|
if let Some(cwd) = stack.get_env_var(&engine_state, "PWD") {
|
|
let path = match cwd.as_string() {
|
|
Ok(p) => p,
|
|
Err(err) => return outcome_err(&engine_state, &err),
|
|
};
|
|
let _ = std::env::set_current_dir(path);
|
|
engine_state.add_env_var("PWD".into(), cwd);
|
|
}
|
|
}
|
|
|
|
outcome_ok(last_output)
|
|
}
|