63 lines
1.7 KiB
Rust
63 lines
1.7 KiB
Rust
use ariadne::Fmt;
|
|
use yansi::Color;
|
|
|
|
use crate::{
|
|
jobs::JobBldr,
|
|
libsh::error::{ShErr, ShErrKind, ShResult, next_color},
|
|
parse::{NdRule, Node},
|
|
prelude::*,
|
|
state::{self},
|
|
};
|
|
|
|
use super::setup_builtin;
|
|
|
|
pub fn cd(node: Node, job: &mut JobBldr) -> ShResult<()> {
|
|
let span = node.get_span();
|
|
let NdRule::Command {
|
|
assignments: _,
|
|
argv,
|
|
} = node.class
|
|
else {
|
|
unreachable!()
|
|
};
|
|
let cd_span = argv.first().unwrap().span.clone();
|
|
|
|
let (argv, _) = setup_builtin(Some(argv), job, None)?;
|
|
let argv = argv.unwrap();
|
|
|
|
let (new_dir,arg_span) = if let Some((arg, span)) = argv.into_iter().next() {
|
|
(PathBuf::from(arg),Some(span))
|
|
} else {
|
|
(PathBuf::from(env::var("HOME").unwrap()),None)
|
|
};
|
|
|
|
if !new_dir.exists() {
|
|
let mut err = ShErr::new(
|
|
ShErrKind::ExecFail,
|
|
span.clone(),
|
|
).labeled(cd_span.clone(), "Failed to change directory");
|
|
if let Some(span) = arg_span {
|
|
err = err.labeled(span, format!("No such file or directory '{}'", new_dir.display().fg(next_color())));
|
|
}
|
|
return Err(err);
|
|
}
|
|
|
|
if !new_dir.is_dir() {
|
|
return Err(ShErr::new(ShErrKind::ExecFail, span.clone())
|
|
.labeled(cd_span.clone(), format!("cd: Not a directory '{}'", new_dir.display().fg(next_color()))));
|
|
}
|
|
|
|
if let Err(e) = env::set_current_dir(new_dir) {
|
|
return Err(ShErr::new(ShErrKind::ExecFail, span.clone())
|
|
.labeled(cd_span.clone(), format!("cd: Failed to change directory: '{}'", e.fg(Color::Red))));
|
|
}
|
|
let new_dir = env::current_dir().map_err(|e| {
|
|
ShErr::new(ShErrKind::ExecFail, span.clone())
|
|
.labeled(cd_span.clone(), format!("cd: Failed to get current directory: '{}'", e.fg(Color::Red)))
|
|
})?;
|
|
unsafe { env::set_var("PWD", new_dir) };
|
|
|
|
state::set_status(0);
|
|
Ok(())
|
|
}
|