pagedmov
aed0e6fb8c
- Added disown builtin - Fixed job table not hanging up child processes on exit - Added target architecture and os to --version output - Added local builtin for creating variables scoped to functions
87 lines
2.3 KiB
Rust
87 lines
2.3 KiB
Rust
use crate::{
|
|
jobs::JobBldr,
|
|
libsh::error::ShResult,
|
|
parse::{NdRule, Node},
|
|
prelude::*,
|
|
procio::{IoStack, borrow_fd},
|
|
state::{self, VarFlags, read_vars, write_vars},
|
|
};
|
|
|
|
use super::setup_builtin;
|
|
|
|
pub fn export(node: Node, io_stack: &mut IoStack, job: &mut JobBldr) -> ShResult<()> {
|
|
let NdRule::Command {
|
|
assignments: _,
|
|
argv,
|
|
} = node.class
|
|
else {
|
|
unreachable!()
|
|
};
|
|
|
|
let (argv, _guard) = setup_builtin(argv, job, Some((io_stack, node.redirs)))?;
|
|
|
|
if argv.is_empty() {
|
|
// Display the environment variables
|
|
let mut env_output = env::vars()
|
|
.map(|var| format!("{}={}", var.0, var.1)) // Get all of them, zip them into one string
|
|
.collect::<Vec<_>>();
|
|
env_output.sort(); // Sort them alphabetically
|
|
let mut env_output = env_output.join("\n"); // Join them with newlines
|
|
env_output.push('\n'); // Push a final newline
|
|
|
|
let stdout = borrow_fd(STDOUT_FILENO);
|
|
write(stdout, env_output.as_bytes())?; // Write it
|
|
} else {
|
|
for (arg, _) in argv {
|
|
if let Some((var, val)) = arg.split_once('=') {
|
|
write_vars(|v| v.set_var(var, val, VarFlags::EXPORT)); // Export an assignment like
|
|
// 'foo=bar'
|
|
} else {
|
|
write_vars(|v| v.export_var(&arg)); // Export an existing variable, if
|
|
// any
|
|
}
|
|
}
|
|
}
|
|
state::set_status(0);
|
|
Ok(())
|
|
}
|
|
|
|
pub fn local(node: Node, io_stack: &mut IoStack, job: &mut JobBldr) -> ShResult<()> {
|
|
let NdRule::Command {
|
|
assignments: _,
|
|
argv,
|
|
} = node.class
|
|
else {
|
|
unreachable!()
|
|
};
|
|
|
|
let (argv, _guard) = setup_builtin(argv, job, Some((io_stack, node.redirs)))?;
|
|
|
|
if argv.is_empty() {
|
|
// Display the local variables
|
|
let vars_output = read_vars(|v| {
|
|
let mut vars = v.flatten_vars()
|
|
.into_iter()
|
|
.map(|(k, v)| format!("{}={}", k, v))
|
|
.collect::<Vec<String>>();
|
|
vars.sort();
|
|
let mut vars_joined = vars.join("\n");
|
|
vars_joined.push('\n');
|
|
vars_joined
|
|
});
|
|
|
|
let stdout = borrow_fd(STDOUT_FILENO);
|
|
write(stdout, vars_output.as_bytes())?; // Write it
|
|
} else {
|
|
for (arg, _) in argv {
|
|
if let Some((var, val)) = arg.split_once('=') {
|
|
write_vars(|v| v.set_var(var, val, VarFlags::LOCAL));
|
|
} else {
|
|
write_vars(|v| v.set_var(&arg, "", VarFlags::LOCAL)); // Create an uninitialized local variable
|
|
}
|
|
}
|
|
}
|
|
state::set_status(0);
|
|
Ok(())
|
|
}
|