Early work on scripting features

This commit is contained in:
2025-03-16 02:56:29 -04:00
parent 8699fc408b
commit 48236398f5
14 changed files with 376 additions and 132 deletions

View File

@@ -1,32 +1,40 @@
use crate::{jobs::{ChildProc, JobBldr}, libsh::error::{ShErr, ShErrKind, ShResult}, parse::{execute::prepare_argv, NdRule, Node}, prelude::*};
use crate::{jobs::{ChildProc, JobBldr}, libsh::error::{ShErr, ShErrKind, ShResult}, parse::{execute::prepare_argv, NdRule, Node}, prelude::*, procio::{borrow_fd, IoStack}, state};
pub fn export(node: Node, job: &mut JobBldr) -> ShResult<()> {
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 child_pgid = if let Some(pgid) = job.pgid() {
pgid
let (argv,io_frame) = 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 {
job.set_pgid(Pid::this());
Pid::this()
};
let child = ChildProc::new(Pid::this(), Some("export"), Some(child_pgid))?;
job.push_child(child);
let argv = prepare_argv(argv);
for (arg,span) in argv {
let Some((var,val)) = arg.split_once('=') else {
return Err(
ShErr::full(
ShErrKind::ExecFail,
"Expected an assignment in export args",
span.into()
for (arg,span) in argv {
let Some((var,val)) = arg.split_once('=') else {
return Err(
ShErr::full(
ShErrKind::SyntaxErr,
"export: Expected an assignment in export args",
span.into()
)
)
)
};
env::set_var(var, val);
};
env::set_var(var, val);
}
}
io_frame.unwrap().restore()?;
state::set_status(0);
Ok(())
}