86 lines
2.4 KiB
Rust
86 lines
2.4 KiB
Rust
use crate::{jobs::JobBldr, libsh::error::{ShErr, ShErrKind, ShResult}, parse::{NdRule, Node}, prelude::*, procio::{borrow_fd, IoStack}, state::{self, read_logic, write_logic}};
|
|
|
|
use super::setup_builtin;
|
|
|
|
pub fn alias(node: Node, io_stack: &mut IoStack, job: &mut JobBldr) -> ShResult<()> {
|
|
let NdRule::Command { assignments: _, argv } = node.class else {
|
|
unreachable!()
|
|
};
|
|
|
|
let (argv,io_frame) = setup_builtin(argv, job, Some((io_stack,node.redirs)))?;
|
|
|
|
if argv.is_empty() {
|
|
// Display the environment variables
|
|
let mut alias_output = read_logic(|l| {
|
|
l.aliases()
|
|
.iter()
|
|
.map(|ent| format!("{} = \"{}\"", ent.0, ent.1))
|
|
.collect::<Vec<_>>()
|
|
});
|
|
alias_output.sort(); // Sort them alphabetically
|
|
let mut alias_output = alias_output.join("\n"); // Join them with newlines
|
|
alias_output.push('\n'); // Push a final newline
|
|
|
|
let stdout = borrow_fd(STDOUT_FILENO);
|
|
write(stdout, alias_output.as_bytes())?; // Write it
|
|
} else {
|
|
for (arg,span) in argv {
|
|
let Some((name,body)) = arg.split_once('=') else {
|
|
return Err(
|
|
ShErr::full(
|
|
ShErrKind::SyntaxErr,
|
|
"alias: Expected an assignment in alias args",
|
|
span
|
|
)
|
|
)
|
|
};
|
|
write_logic(|l| l.insert_alias(name, body));
|
|
}
|
|
}
|
|
io_frame.unwrap().restore()?;
|
|
state::set_status(0);
|
|
Ok(())
|
|
}
|
|
|
|
pub fn unalias(node: Node, io_stack: &mut IoStack, job: &mut JobBldr) -> ShResult<()> {
|
|
let NdRule::Command { assignments: _, argv } = node.class else {
|
|
unreachable!()
|
|
};
|
|
|
|
let (argv, io_frame) = setup_builtin(argv, job, Some((io_stack, node.redirs)))?;
|
|
|
|
|
|
if argv.is_empty() {
|
|
// Display the environment variables
|
|
let mut alias_output = read_logic(|l| {
|
|
l.aliases()
|
|
.iter()
|
|
.map(|ent| format!("{} = \"{}\"", ent.0, ent.1))
|
|
.collect::<Vec<_>>()
|
|
});
|
|
alias_output.sort(); // Sort them alphabetically
|
|
let mut alias_output = alias_output.join("\n"); // Join them with newlines
|
|
alias_output.push('\n'); // Push a final newline
|
|
|
|
let stdout = borrow_fd(STDOUT_FILENO);
|
|
write(stdout, alias_output.as_bytes())?; // Write it
|
|
} else {
|
|
for (arg,span) in argv {
|
|
flog!(DEBUG, arg);
|
|
if read_logic(|l| l.get_alias(&arg)).is_none() {
|
|
return Err(
|
|
ShErr::full(
|
|
ShErrKind::SyntaxErr,
|
|
format!("unalias: alias '{arg}' not found"),
|
|
span
|
|
)
|
|
)
|
|
};
|
|
write_logic(|l| l.remove_alias(&arg))
|
|
}
|
|
}
|
|
io_frame.unwrap().restore()?;
|
|
state::set_status(0);
|
|
Ok(())
|
|
}
|