1use std::fmt;
11
12use crate::arithmetic;
13use crate::ast::{BinaryOp, Expr, FileTestOp, Stmt, StringPart, StringTestOp, TestCmpOp, TestExpr, Value, VarPath};
14use crate::vfs::DirEntry;
15use std::path::Path;
16
17use super::result::ExecResult;
18use super::scope::Scope;
19
20pub fn strip_leading_tabs(s: &str) -> String {
26 let mut out = String::with_capacity(s.len());
27 let mut at_line_start = true;
28 for ch in s.chars() {
29 if at_line_start && ch == '\t' {
30 continue;
32 }
33 out.push(ch);
34 at_line_start = ch == '\n';
35 }
36 out
37}
38
39#[derive(Debug, Clone, PartialEq)]
41pub enum EvalError {
42 UndefinedVariable(String),
44 InvalidPath(String),
46 TypeError { expected: &'static str, got: String },
48 CommandFailed(String),
50 NoExecutor,
52 ArithmeticError(String),
54 RegexError(String),
56}
57
58impl fmt::Display for EvalError {
59 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
60 match self {
61 EvalError::UndefinedVariable(name) => write!(f, "undefined variable: {name}"),
62 EvalError::InvalidPath(path) => write!(f, "invalid path: {path}"),
63 EvalError::TypeError { expected, got } => {
64 write!(f, "type error: expected {expected}, got {got}")
65 }
66 EvalError::CommandFailed(msg) => write!(f, "command failed: {msg}"),
67 EvalError::NoExecutor => write!(f, "no executor available for command substitution"),
68 EvalError::ArithmeticError(msg) => write!(f, "arithmetic error: {msg}"),
69 EvalError::RegexError(msg) => write!(f, "regex error: {msg}"),
70 }
71 }
72}
73
74impl std::error::Error for EvalError {}
75
76pub type EvalResult<T> = Result<T, EvalError>;
78
79pub trait Executor {
85 fn execute(&mut self, stmts: &[Stmt], scope: &mut Scope) -> EvalResult<ExecResult>;
94
95 fn file_stat(&self, path: &Path) -> Option<DirEntry> {
102 std::fs::metadata(path).ok().map(|meta| {
103 if meta.is_dir() {
104 DirEntry::directory(path.file_name().unwrap_or_default().to_string_lossy())
105 } else {
106 #[allow(unused_mut)]
107 let mut entry = DirEntry::file(
108 path.file_name().unwrap_or_default().to_string_lossy(),
109 meta.len(),
110 );
111 #[cfg(unix)]
112 {
113 use std::os::unix::fs::PermissionsExt;
114 entry.permissions = Some(meta.permissions().mode());
115 }
116 entry
117 }
118 })
119 }
120}
121
122pub struct NoOpExecutor;
126
127impl Executor for NoOpExecutor {
128 fn execute(&mut self, _stmts: &[Stmt], _scope: &mut Scope) -> EvalResult<ExecResult> {
129 Err(EvalError::NoExecutor)
130 }
131}
132
133pub struct Evaluator<'a, E: Executor> {
138 scope: &'a mut Scope,
139 executor: &'a mut E,
140}
141
142impl<'a, E: Executor> Evaluator<'a, E> {
143 pub fn new(scope: &'a mut Scope, executor: &'a mut E) -> Self {
145 Self { scope, executor }
146 }
147
148 pub fn eval(&mut self, expr: &Expr) -> EvalResult<Value> {
150 match expr {
151 Expr::Literal(value) => self.eval_literal(value),
152 Expr::VarRef(path) => self.eval_var_ref(path),
153 Expr::Interpolated(parts) => self.eval_interpolated(parts),
154 Expr::HereDocBody { parts, strip_tabs } => {
155 let unwrapped: Vec<StringPart> =
158 parts.iter().map(|sp| sp.part.clone()).collect();
159 let value = self.eval_interpolated(&unwrapped)?;
160 if *strip_tabs {
161 if let Value::String(s) = value {
162 Ok(Value::String(strip_leading_tabs(&s)))
163 } else {
164 Ok(value)
165 }
166 } else {
167 Ok(value)
168 }
169 }
170 Expr::BinaryOp { left, op, right } => self.eval_binary_op(left, *op, right),
171 Expr::CommandSubst(stmts) => self.eval_command_subst(stmts),
172 Expr::Test(test_expr) => self.eval_test(test_expr),
173 Expr::Positional(n) => self.eval_positional(*n),
174 Expr::AllArgs => self.eval_all_args(),
175 Expr::ArgCount => self.eval_arg_count(),
176 Expr::VarLength(name) => self.eval_var_length(name),
177 Expr::VarWithDefault { name, default } => self.eval_var_with_default(name, default),
178 Expr::Arithmetic(expr_str) => self.eval_arithmetic(expr_str),
179 Expr::Command(cmd) => self.eval_command(cmd),
180 Expr::LastExitCode => self.eval_last_exit_code(),
181 Expr::CurrentPid => self.eval_current_pid(),
182 Expr::GlobPattern(s) => Ok(Value::String(s.clone())),
183 }
184 }
185
186 fn eval_last_exit_code(&self) -> EvalResult<Value> {
188 Ok(Value::Int(self.scope.last_result().code))
189 }
190
191 fn eval_current_pid(&self) -> EvalResult<Value> {
193 Ok(Value::Int(self.scope.pid() as i64))
194 }
195
196 fn eval_command(&mut self, cmd: &crate::ast::Command) -> EvalResult<Value> {
198 match cmd.name.as_str() {
201 "true" => return Ok(Value::Bool(true)),
202 "false" => return Ok(Value::Bool(false)),
203 _ => {}
204 }
205
206 let block = [Stmt::Command(cmd.clone())];
208 let result = self.executor.execute(&block, self.scope)?;
209 Ok(Value::Bool(result.code == 0))
211 }
212
213 fn eval_arithmetic(&mut self, expr_str: &str) -> EvalResult<Value> {
215 arithmetic::eval_arithmetic(expr_str, self.scope)
216 .map(Value::Int)
217 .map_err(|e| EvalError::ArithmeticError(e.to_string()))
218 }
219
220 fn eval_test(&mut self, test_expr: &TestExpr) -> EvalResult<Value> {
222 let result = match test_expr {
223 TestExpr::FileTest { op, path } => {
224 let path_value = self.eval(path)?;
225 let path_str = value_to_string(&path_value);
226 let path = Path::new(&path_str);
227 let entry = self.executor.file_stat(path);
228 match op {
229 FileTestOp::Exists => entry.is_some(),
230 FileTestOp::IsFile => entry.as_ref().is_some_and(|e| e.is_file()),
231 FileTestOp::IsDir => entry.as_ref().is_some_and(|e| e.is_dir()),
232 FileTestOp::Readable => entry.is_some(),
233 FileTestOp::Writable => entry.as_ref().is_some_and(|e| {
234 e.permissions.is_none_or(|p| p & 0o222 != 0)
235 }),
236 FileTestOp::Executable => entry.as_ref().is_some_and(|e| {
237 e.permissions.is_some_and(|p| p & 0o111 != 0)
238 }),
239 }
240 }
241 TestExpr::StringTest { op, value } => {
242 let val = self.eval(value)?;
243 let s = value_to_string(&val);
244 match op {
245 StringTestOp::IsEmpty => s.is_empty(),
246 StringTestOp::IsNonEmpty => !s.is_empty(),
247 }
248 }
249 TestExpr::Comparison { left, op, right } => {
250 let left_val = self.eval(left)?;
251 let right_val = self.eval(right)?;
252
253 match op {
254 TestCmpOp::Eq => values_equal(&left_val, &right_val),
255 TestCmpOp::NotEq => !values_equal(&left_val, &right_val),
256 TestCmpOp::Match => {
257 match regex_match(&left_val, &right_val, false) {
259 Ok(Value::Bool(b)) => b,
260 Ok(_) => false,
261 Err(_) => false,
262 }
263 }
264 TestCmpOp::NotMatch => {
265 match regex_match(&left_val, &right_val, true) {
267 Ok(Value::Bool(b)) => b,
268 Ok(_) => true,
269 Err(_) => true,
270 }
271 }
272 TestCmpOp::Gt | TestCmpOp::Lt | TestCmpOp::GtEq | TestCmpOp::LtEq => {
273 let ord = compare_values(&left_val, &right_val)?;
275 match op {
276 TestCmpOp::Gt => ord.is_gt(),
277 TestCmpOp::Lt => ord.is_lt(),
278 TestCmpOp::GtEq => ord.is_ge(),
279 TestCmpOp::LtEq => ord.is_le(),
280 _ => unreachable!(),
281 }
282 }
283 TestCmpOp::NumEq
284 | TestCmpOp::NumNotEq
285 | TestCmpOp::NumGt
286 | TestCmpOp::NumLt
287 | TestCmpOp::NumGtEq
288 | TestCmpOp::NumLtEq => {
289 let ord = numeric_compare(&left_val, &right_val)?;
292 match op {
293 TestCmpOp::NumEq => ord.is_eq(),
294 TestCmpOp::NumNotEq => !ord.is_eq(),
295 TestCmpOp::NumGt => ord.is_gt(),
296 TestCmpOp::NumLt => ord.is_lt(),
297 TestCmpOp::NumGtEq => ord.is_ge(),
298 TestCmpOp::NumLtEq => ord.is_le(),
299 _ => unreachable!(),
300 }
301 }
302 }
303 }
304 TestExpr::And { left, right } => {
305 let left_result = self.eval_test(left)?;
307 if !value_to_bool(&left_result) {
308 false } else {
310 value_to_bool(&self.eval_test(right)?)
311 }
312 }
313 TestExpr::Or { left, right } => {
314 let left_result = self.eval_test(left)?;
316 if value_to_bool(&left_result) {
317 true } else {
319 value_to_bool(&self.eval_test(right)?)
320 }
321 }
322 TestExpr::Not { expr } => {
323 let result = self.eval_test(expr)?;
324 !value_to_bool(&result)
325 }
326 };
327 Ok(Value::Bool(result))
328 }
329
330 fn eval_literal(&mut self, value: &Value) -> EvalResult<Value> {
332 Ok(value.clone())
333 }
334
335 fn eval_var_ref(&mut self, path: &VarPath) -> EvalResult<Value> {
337 self.scope
338 .resolve_path(path)
339 .ok_or_else(|| EvalError::InvalidPath(format_path(path)))
340 }
341
342 fn eval_positional(&self, n: usize) -> EvalResult<Value> {
344 match self.scope.get_positional(n) {
345 Some(s) => Ok(Value::String(s.to_string())),
346 None => Ok(Value::String(String::new())), }
348 }
349
350 fn eval_all_args(&self) -> EvalResult<Value> {
354 let args = self.scope.all_args();
355 Ok(Value::String(args.join(" ")))
356 }
357
358 fn eval_arg_count(&self) -> EvalResult<Value> {
360 Ok(Value::Int(self.scope.arg_count() as i64))
361 }
362
363 fn eval_var_length(&self, name: &str) -> EvalResult<Value> {
365 match self.scope.get(name) {
366 Some(value) => {
367 let s = value_to_string(value);
368 Ok(Value::Int(s.len() as i64))
369 }
370 None => Ok(Value::Int(0)), }
372 }
373
374 fn eval_var_with_default(&mut self, name: &str, default: &[StringPart]) -> EvalResult<Value> {
377 match self.scope.get(name) {
378 Some(value) => {
379 let s = value_to_string(value);
380 if s.is_empty() {
381 self.eval_interpolated(default)
383 } else {
384 Ok(value.clone())
385 }
386 }
387 None => {
388 self.eval_interpolated(default)
390 }
391 }
392 }
393
394 fn eval_interpolated(&mut self, parts: &[StringPart]) -> EvalResult<Value> {
396 let mut result = String::new();
397 for part in parts {
398 match part {
399 StringPart::Literal(s) => result.push_str(s),
400 StringPart::Var(path) => {
401 if let Some(value) = self.scope.resolve_path(path) {
403 result.push_str(&value_to_string(&value));
404 }
405 }
406 StringPart::VarWithDefault { name, default } => {
407 let value = self.eval_var_with_default(name, default)?;
408 result.push_str(&value_to_string(&value));
409 }
410 StringPart::VarLength(name) => {
411 let value = self.eval_var_length(name)?;
412 result.push_str(&value_to_string(&value));
413 }
414 StringPart::Positional(n) => {
415 let value = self.eval_positional(*n)?;
416 result.push_str(&value_to_string(&value));
417 }
418 StringPart::AllArgs => {
419 let value = self.eval_all_args()?;
420 result.push_str(&value_to_string(&value));
421 }
422 StringPart::ArgCount => {
423 let value = self.eval_arg_count()?;
424 result.push_str(&value_to_string(&value));
425 }
426 StringPart::Arithmetic(expr) => {
427 let value = self.eval_arithmetic_string(expr)?;
429 result.push_str(&value_to_string(&value));
430 }
431 StringPart::CommandSubst(stmts) => {
432 let value = self.eval_command_subst(stmts)?;
434 result.push_str(&value_to_string(&value));
435 }
436 StringPart::LastExitCode => {
437 result.push_str(&self.scope.last_result().code.to_string());
438 }
439 StringPart::CurrentPid => {
440 result.push_str(&self.scope.pid().to_string());
441 }
442 }
443 }
444 Ok(Value::String(result))
445 }
446
447 fn eval_arithmetic_string(&mut self, expr: &str) -> EvalResult<Value> {
449 arithmetic::eval_arithmetic(expr, self.scope)
451 .map(Value::Int)
452 .map_err(|e| EvalError::ArithmeticError(e.to_string()))
453 }
454
455 fn eval_binary_op(&mut self, left: &Expr, op: BinaryOp, right: &Expr) -> EvalResult<Value> {
459 match op {
460 BinaryOp::And => {
461 let left_val = self.eval(left)?;
462 if !is_truthy(&left_val) {
463 return Ok(left_val);
464 }
465 self.eval(right)
466 }
467 BinaryOp::Or => {
468 let left_val = self.eval(left)?;
469 if is_truthy(&left_val) {
470 return Ok(left_val);
471 }
472 self.eval(right)
473 }
474 }
475 }
476
477 fn eval_command_subst(&mut self, stmts: &[Stmt]) -> EvalResult<Value> {
479 let result = self.executor.execute(stmts, self.scope)?;
480
481 self.scope.set_last_result(result.clone());
483
484 Ok(result_to_value(&result))
487 }
488}
489
490pub fn value_to_exit_code(value: &Value) -> anyhow::Result<i64> {
497 match value {
498 Value::Int(n) => Ok(*n),
499 Value::Bool(b) => Ok(if *b { 0 } else { 1 }),
500 Value::Float(f) => Ok(*f as i64),
501 Value::String(s) => {
502 let trimmed = s.trim();
503 trimmed.parse::<i64>().map_err(|_| {
504 anyhow::anyhow!("numeric argument required: {:?}", s)
505 })
506 }
507 Value::Null | Value::Json(_) | Value::Bytes(_) => {
508 anyhow::bail!("numeric argument required (got {:?})", value)
509 }
510 }
511}
512
513pub fn value_to_string(value: &Value) -> String {
514 match value {
515 Value::Null => "null".to_string(),
516 Value::Bool(b) => b.to_string(),
517 Value::Int(i) => i.to_string(),
518 Value::Float(f) => f.to_string(),
519 Value::String(s) => s.clone(),
520 Value::Json(json) => json.to_string(),
521 Value::Bytes(b) => format!("[binary: {} bytes]", b.len()),
524 }
525}
526
527pub fn value_to_bool(value: &Value) -> bool {
537 match value {
538 Value::Null => false,
539 Value::Bool(b) => *b,
540 Value::Int(i) => *i != 0,
541 Value::Float(f) => *f != 0.0,
542 Value::String(s) => !s.is_empty(),
543 Value::Json(json) => match json {
544 serde_json::Value::Null => false,
545 serde_json::Value::Array(arr) => !arr.is_empty(),
546 serde_json::Value::Object(obj) => !obj.is_empty(),
547 serde_json::Value::Bool(b) => *b,
548 serde_json::Value::Number(n) => n.as_f64().map(|f| f != 0.0).unwrap_or(false),
549 serde_json::Value::String(s) => !s.is_empty(),
550 },
551 Value::Bytes(b) => !b.is_empty(), }
553}
554
555pub fn expand_tilde(s: &str, home: Option<&str>) -> String {
569 if s == "~" {
570 home.map(|h| h.to_string()).unwrap_or_else(|| "~".to_string())
571 } else if s.starts_with("~/") {
572 match home {
573 Some(home) => format!("{}{}", home, &s[1..]),
574 None => s.to_string(),
575 }
576 } else if s.starts_with('~') {
577 expand_tilde_user(s)
579 } else {
580 s.to_string()
581 }
582}
583
584#[cfg(all(unix, feature = "host"))]
589fn expand_tilde_user(s: &str) -> String {
590 let (username, rest) = if let Some(slash_pos) = s[1..].find('/') {
592 (&s[1..slash_pos + 1], &s[slash_pos + 1..])
593 } else {
594 (&s[1..], "")
595 };
596
597 if username.is_empty() {
598 return s.to_string();
599 }
600
601 let passwd = match std::fs::read_to_string("/etc/passwd") {
604 Ok(content) => content,
605 Err(_) => return s.to_string(),
606 };
607
608 for line in passwd.lines() {
609 let fields: Vec<&str> = line.split(':').collect();
610 if fields.len() >= 6 && fields[0] == username {
611 let home_dir = fields[5];
612 return if rest.is_empty() {
613 home_dir.to_string()
614 } else {
615 format!("{}{}", home_dir, rest)
616 };
617 }
618 }
619
620 s.to_string()
622}
623
624#[cfg(not(all(unix, feature = "host")))]
625fn expand_tilde_user(s: &str) -> String {
626 s.to_string()
629}
630
631pub fn value_to_string_with_tilde(value: &Value, home: Option<&str>) -> String {
636 match value {
637 Value::String(s) if s.starts_with('~') => expand_tilde(s, home),
638 _ => value_to_string(value),
639 }
640}
641
642fn format_path(path: &VarPath) -> String {
644 use crate::ast::VarSegment;
645 let mut result = String::from("${");
646 for (i, seg) in path.segments.iter().enumerate() {
647 match seg {
648 VarSegment::Field(name) => {
649 if i > 0 {
650 result.push('.');
651 }
652 result.push_str(name);
653 }
654 }
655 }
656 result.push('}');
657 result
658}
659
660fn is_truthy(value: &Value) -> bool {
670 value_to_bool(value)
672}
673
674fn values_equal(left: &Value, right: &Value) -> bool {
685 match (left, right) {
686 (Value::Null, Value::Null) => true,
687 (Value::Bool(a), Value::Bool(b)) => a == b,
688 (Value::Int(a), Value::Int(b)) => a == b,
689 (Value::Float(a), Value::Float(b)) => (a - b).abs() < f64::EPSILON,
690 (Value::Int(a), Value::Float(b)) | (Value::Float(b), Value::Int(a)) => {
691 (*a as f64 - b).abs() < f64::EPSILON
692 }
693 (Value::String(a), Value::String(b)) => a == b,
694 (Value::Json(a), Value::Json(b)) => a == b,
695 (Value::Bytes(a), Value::Bytes(b)) => a == b,
696 _ => value_to_string(left) == value_to_string(right),
699 }
700}
701
702fn compare_values(left: &Value, right: &Value) -> EvalResult<std::cmp::Ordering> {
704 match (left, right) {
705 (Value::Int(a), Value::Int(b)) => Ok(a.cmp(b)),
706 (Value::Float(a), Value::Float(b)) => {
707 a.partial_cmp(b).ok_or_else(|| EvalError::ArithmeticError("NaN comparison".into()))
708 }
709 (Value::Int(a), Value::Float(b)) => {
710 (*a as f64).partial_cmp(b).ok_or_else(|| EvalError::ArithmeticError("NaN comparison".into()))
711 }
712 (Value::Float(a), Value::Int(b)) => {
713 a.partial_cmp(&(*b as f64)).ok_or_else(|| EvalError::ArithmeticError("NaN comparison".into()))
714 }
715 (Value::String(a), Value::String(b)) => Ok(a.cmp(b)),
716 _ => Err(EvalError::TypeError {
717 expected: "comparable types (numbers or strings)",
718 got: format!("{:?} vs {:?}", type_name(left), type_name(right)),
719 }),
720 }
721}
722
723enum Num {
728 Int(i64),
729 Float(f64),
730}
731
732fn value_to_num(value: &Value) -> EvalResult<Num> {
733 match value {
734 Value::Int(n) => Ok(Num::Int(*n)),
735 Value::Float(f) => Ok(Num::Float(*f)),
736 Value::String(s) => {
737 let t = s.trim();
738 if let Ok(n) = t.parse::<i64>() {
739 Ok(Num::Int(n))
740 } else if let Ok(f) = t.parse::<f64>() {
741 Ok(Num::Float(f))
742 } else {
743 Err(EvalError::TypeError {
744 expected: "numeric operand",
745 got: format!("non-numeric string {:?}", s),
746 })
747 }
748 }
749 _ => Err(EvalError::TypeError {
750 expected: "numeric operand",
751 got: type_name(value).to_string(),
752 }),
753 }
754}
755
756fn numeric_compare(left: &Value, right: &Value) -> EvalResult<std::cmp::Ordering> {
759 let l = value_to_num(left)?;
760 let r = value_to_num(right)?;
761 match (l, r) {
762 (Num::Int(a), Num::Int(b)) => Ok(a.cmp(&b)),
763 (Num::Float(a), Num::Float(b)) => a
764 .partial_cmp(&b)
765 .ok_or_else(|| EvalError::ArithmeticError("NaN comparison".into())),
766 (Num::Int(a), Num::Float(b)) => (a as f64)
767 .partial_cmp(&b)
768 .ok_or_else(|| EvalError::ArithmeticError("NaN comparison".into())),
769 (Num::Float(a), Num::Int(b)) => a
770 .partial_cmp(&(b as f64))
771 .ok_or_else(|| EvalError::ArithmeticError("NaN comparison".into())),
772 }
773}
774
775fn type_name(value: &Value) -> &'static str {
777 match value {
778 Value::Null => "null",
779 Value::Bool(_) => "bool",
780 Value::Int(_) => "int",
781 Value::Float(_) => "float",
782 Value::String(_) => "string",
783 Value::Json(_) => "json",
784 Value::Bytes(_) => "bytes",
785 }
786}
787
788fn result_to_value(result: &ExecResult) -> Value {
795 if let Some(data) = &result.data {
797 return data.clone();
798 }
799 Value::String(result.text_out().trim_end().to_string())
801}
802
803fn regex_match(left: &Value, right: &Value, negate: bool) -> EvalResult<Value> {
808 let text = match left {
809 Value::String(s) => s.as_str(),
810 _ => {
811 return Err(EvalError::TypeError {
812 expected: "string",
813 got: type_name(left).to_string(),
814 })
815 }
816 };
817
818 let pattern = match right {
819 Value::String(s) => s.as_str(),
820 _ => {
821 return Err(EvalError::TypeError {
822 expected: "string (regex pattern)",
823 got: type_name(right).to_string(),
824 })
825 }
826 };
827
828 let re = regex::Regex::new(pattern).map_err(|e| EvalError::RegexError(e.to_string()))?;
829 let matches = re.is_match(text);
830
831 Ok(Value::Bool(if negate { !matches } else { matches }))
832}
833
834pub fn eval_expr(expr: &Expr, scope: &mut Scope) -> EvalResult<Value> {
838 let mut executor = NoOpExecutor;
839 let mut evaluator = Evaluator::new(scope, &mut executor);
840 evaluator.eval(expr)
841}
842
843#[cfg(test)]
844mod tests {
845 use super::*;
846 use crate::ast::VarSegment;
847
848 fn var_expr(name: &str) -> Expr {
850 Expr::VarRef(VarPath::simple(name))
851 }
852
853 #[test]
854 fn eval_literal_int() {
855 let mut scope = Scope::new();
856 let expr = Expr::Literal(Value::Int(42));
857 assert_eq!(eval_expr(&expr, &mut scope), Ok(Value::Int(42)));
858 }
859
860 #[test]
861 fn eval_literal_string() {
862 let mut scope = Scope::new();
863 let expr = Expr::Literal(Value::String("hello".into()));
864 assert_eq!(eval_expr(&expr, &mut scope), Ok(Value::String("hello".into())));
865 }
866
867 #[test]
868 fn eval_literal_bool() {
869 let mut scope = Scope::new();
870 assert_eq!(
871 eval_expr(&Expr::Literal(Value::Bool(true)), &mut scope),
872 Ok(Value::Bool(true))
873 );
874 }
875
876 #[test]
877 fn eval_literal_null() {
878 let mut scope = Scope::new();
879 assert_eq!(
880 eval_expr(&Expr::Literal(Value::Null), &mut scope),
881 Ok(Value::Null)
882 );
883 }
884
885 #[test]
886 fn eval_literal_float() {
887 let mut scope = Scope::new();
888 let expr = Expr::Literal(Value::Float(3.14));
889 assert_eq!(eval_expr(&expr, &mut scope), Ok(Value::Float(3.14)));
890 }
891
892 #[test]
893 fn eval_variable_ref() {
894 let mut scope = Scope::new();
895 scope.set("X", Value::Int(100));
896 assert_eq!(eval_expr(&var_expr("X"), &mut scope), Ok(Value::Int(100)));
897 }
898
899 #[test]
900 fn eval_undefined_variable() {
901 let mut scope = Scope::new();
902 let result = eval_expr(&var_expr("MISSING"), &mut scope);
903 assert!(matches!(result, Err(EvalError::InvalidPath(_))));
904 }
905
906 #[test]
907 fn eval_interpolated_string() {
908 let mut scope = Scope::new();
909 scope.set("NAME", Value::String("World".into()));
910
911 let expr = Expr::Interpolated(vec![
912 StringPart::Literal("Hello, ".into()),
913 StringPart::Var(VarPath::simple("NAME")),
914 StringPart::Literal("!".into()),
915 ]);
916 assert_eq!(
917 eval_expr(&expr, &mut scope),
918 Ok(Value::String("Hello, World!".into()))
919 );
920 }
921
922 #[test]
923 fn eval_interpolated_with_number() {
924 let mut scope = Scope::new();
925 scope.set("COUNT", Value::Int(42));
926
927 let expr = Expr::Interpolated(vec![
928 StringPart::Literal("Count: ".into()),
929 StringPart::Var(VarPath::simple("COUNT")),
930 ]);
931 assert_eq!(
932 eval_expr(&expr, &mut scope),
933 Ok(Value::String("Count: 42".into()))
934 );
935 }
936
937 #[test]
938 fn eval_and_short_circuit_true() {
939 let mut scope = Scope::new();
940 let expr = Expr::BinaryOp {
941 left: Box::new(Expr::Literal(Value::Bool(true))),
942 op: BinaryOp::And,
943 right: Box::new(Expr::Literal(Value::Int(42))),
944 };
945 assert_eq!(eval_expr(&expr, &mut scope), Ok(Value::Int(42)));
947 }
948
949 #[test]
950 fn eval_and_short_circuit_false() {
951 let mut scope = Scope::new();
952 let expr = Expr::BinaryOp {
953 left: Box::new(Expr::Literal(Value::Bool(false))),
954 op: BinaryOp::And,
955 right: Box::new(Expr::Literal(Value::Int(42))),
956 };
957 assert_eq!(eval_expr(&expr, &mut scope), Ok(Value::Bool(false)));
959 }
960
961 #[test]
962 fn eval_or_short_circuit_true() {
963 let mut scope = Scope::new();
964 let expr = Expr::BinaryOp {
965 left: Box::new(Expr::Literal(Value::Bool(true))),
966 op: BinaryOp::Or,
967 right: Box::new(Expr::Literal(Value::Int(42))),
968 };
969 assert_eq!(eval_expr(&expr, &mut scope), Ok(Value::Bool(true)));
971 }
972
973 #[test]
974 fn eval_or_short_circuit_false() {
975 let mut scope = Scope::new();
976 let expr = Expr::BinaryOp {
977 left: Box::new(Expr::Literal(Value::Bool(false))),
978 op: BinaryOp::Or,
979 right: Box::new(Expr::Literal(Value::Int(42))),
980 };
981 assert_eq!(eval_expr(&expr, &mut scope), Ok(Value::Int(42)));
983 }
984
985 #[test]
986 fn is_truthy_values() {
987 assert!(!is_truthy(&Value::Null));
988 assert!(!is_truthy(&Value::Bool(false)));
989 assert!(is_truthy(&Value::Bool(true)));
990 assert!(!is_truthy(&Value::Int(0)));
991 assert!(is_truthy(&Value::Int(1)));
992 assert!(is_truthy(&Value::Int(-1)));
993 assert!(!is_truthy(&Value::Float(0.0)));
994 assert!(is_truthy(&Value::Float(0.1)));
995 assert!(!is_truthy(&Value::String("".into())));
996 assert!(is_truthy(&Value::String("x".into())));
997 }
998
999 #[test]
1000 fn eval_command_subst_fails_without_executor() {
1001 use crate::ast::Command;
1002
1003 let mut scope = Scope::new();
1004 let expr = Expr::CommandSubst(vec![Stmt::Command(Command {
1005 name: "echo".into(),
1006 args: vec![],
1007 redirects: vec![],
1008 })]);
1009
1010 assert!(matches!(
1011 eval_expr(&expr, &mut scope),
1012 Err(EvalError::NoExecutor)
1013 ));
1014 }
1015
1016 #[test]
1017 fn eval_last_result_bare() {
1018 let mut scope = Scope::new();
1021 scope.set_last_result(ExecResult::failure(42, "test error"));
1022
1023 let expr = Expr::VarRef(VarPath {
1024 segments: vec![VarSegment::Field("?".into())],
1025 });
1026 assert_eq!(eval_expr(&expr, &mut scope), Ok(Value::Int(42)));
1027 }
1028
1029 #[test]
1030 fn value_to_string_all_types() {
1031 assert_eq!(value_to_string(&Value::Null), "null");
1032 assert_eq!(value_to_string(&Value::Bool(true)), "true");
1033 assert_eq!(value_to_string(&Value::Int(42)), "42");
1034 assert_eq!(value_to_string(&Value::Float(3.14)), "3.14");
1035 assert_eq!(value_to_string(&Value::String("hello".into())), "hello");
1036 }
1037
1038 #[test]
1041 fn eval_negative_int() {
1042 let mut scope = Scope::new();
1043 let expr = Expr::Literal(Value::Int(-42));
1044 assert_eq!(eval_expr(&expr, &mut scope), Ok(Value::Int(-42)));
1045 }
1046
1047 #[test]
1048 fn eval_negative_float() {
1049 let mut scope = Scope::new();
1050 let expr = Expr::Literal(Value::Float(-3.14));
1051 assert_eq!(eval_expr(&expr, &mut scope), Ok(Value::Float(-3.14)));
1052 }
1053
1054 #[test]
1055 fn eval_zero_values() {
1056 let mut scope = Scope::new();
1057 assert_eq!(
1058 eval_expr(&Expr::Literal(Value::Int(0)), &mut scope),
1059 Ok(Value::Int(0))
1060 );
1061 assert_eq!(
1062 eval_expr(&Expr::Literal(Value::Float(0.0)), &mut scope),
1063 Ok(Value::Float(0.0))
1064 );
1065 }
1066
1067 #[test]
1068 fn eval_interpolation_empty_var() {
1069 let mut scope = Scope::new();
1070 scope.set("EMPTY", Value::String("".into()));
1071
1072 let expr = Expr::Interpolated(vec![
1073 StringPart::Literal("prefix".into()),
1074 StringPart::Var(VarPath::simple("EMPTY")),
1075 StringPart::Literal("suffix".into()),
1076 ]);
1077 assert_eq!(
1078 eval_expr(&expr, &mut scope),
1079 Ok(Value::String("prefixsuffix".into()))
1080 );
1081 }
1082
1083 #[test]
1084 fn eval_chained_and() {
1085 let mut scope = Scope::new();
1086 let expr = Expr::BinaryOp {
1088 left: Box::new(Expr::BinaryOp {
1089 left: Box::new(Expr::Literal(Value::Bool(true))),
1090 op: BinaryOp::And,
1091 right: Box::new(Expr::Literal(Value::Bool(true))),
1092 }),
1093 op: BinaryOp::And,
1094 right: Box::new(Expr::Literal(Value::Int(42))),
1095 };
1096 assert_eq!(eval_expr(&expr, &mut scope), Ok(Value::Int(42)));
1097 }
1098
1099 #[test]
1100 fn eval_chained_or() {
1101 let mut scope = Scope::new();
1102 let expr = Expr::BinaryOp {
1104 left: Box::new(Expr::BinaryOp {
1105 left: Box::new(Expr::Literal(Value::Bool(false))),
1106 op: BinaryOp::Or,
1107 right: Box::new(Expr::Literal(Value::Bool(false))),
1108 }),
1109 op: BinaryOp::Or,
1110 right: Box::new(Expr::Literal(Value::Int(42))),
1111 };
1112 assert_eq!(eval_expr(&expr, &mut scope), Ok(Value::Int(42)));
1113 }
1114
1115 #[test]
1116 fn eval_mixed_and_or() {
1117 let mut scope = Scope::new();
1118 let expr = Expr::BinaryOp {
1121 left: Box::new(Expr::BinaryOp {
1122 left: Box::new(Expr::Literal(Value::Bool(true))),
1123 op: BinaryOp::Or,
1124 right: Box::new(Expr::Literal(Value::Bool(false))),
1125 }),
1126 op: BinaryOp::And,
1127 right: Box::new(Expr::Literal(Value::Bool(true))),
1128 };
1129 assert_eq!(eval_expr(&expr, &mut scope), Ok(Value::Bool(true)));
1131 }
1132
1133 #[test]
1134 fn eval_interpolation_with_bool() {
1135 let mut scope = Scope::new();
1136 scope.set("FLAG", Value::Bool(true));
1137
1138 let expr = Expr::Interpolated(vec![
1139 StringPart::Literal("enabled: ".into()),
1140 StringPart::Var(VarPath::simple("FLAG")),
1141 ]);
1142 assert_eq!(
1143 eval_expr(&expr, &mut scope),
1144 Ok(Value::String("enabled: true".into()))
1145 );
1146 }
1147
1148 #[test]
1149 fn eval_interpolation_with_null() {
1150 let mut scope = Scope::new();
1151 scope.set("VAL", Value::Null);
1152
1153 let expr = Expr::Interpolated(vec![
1154 StringPart::Literal("value: ".into()),
1155 StringPart::Var(VarPath::simple("VAL")),
1156 ]);
1157 assert_eq!(
1158 eval_expr(&expr, &mut scope),
1159 Ok(Value::String("value: null".into()))
1160 );
1161 }
1162
1163 #[test]
1164 fn eval_format_path_simple() {
1165 let path = VarPath::simple("X");
1166 assert_eq!(format_path(&path), "${X}");
1167 }
1168
1169 #[test]
1170 fn eval_format_path_nested() {
1171 let path = VarPath {
1172 segments: vec![
1173 VarSegment::Field("X".into()),
1174 VarSegment::Field("field".into()),
1175 ],
1176 };
1177 assert_eq!(format_path(&path), "${X.field}");
1178 }
1179
1180 #[test]
1181 fn type_name_all_types() {
1182 assert_eq!(type_name(&Value::Null), "null");
1183 assert_eq!(type_name(&Value::Bool(true)), "bool");
1184 assert_eq!(type_name(&Value::Int(1)), "int");
1185 assert_eq!(type_name(&Value::Float(1.0)), "float");
1186 assert_eq!(type_name(&Value::String("".into())), "string");
1187 }
1188
1189 #[test]
1190 fn expand_tilde_home() {
1191 let home = "/home/session";
1193 assert_eq!(expand_tilde("~", Some(home)), home);
1194 assert_eq!(expand_tilde("~/foo", Some(home)), format!("{}/foo", home));
1195 assert_eq!(
1196 expand_tilde("~/foo/bar", Some(home)),
1197 format!("{}/foo/bar", home)
1198 );
1199 }
1200
1201 #[test]
1202 fn expand_tilde_hermetic_no_home_does_not_leak_host() {
1203 assert_eq!(expand_tilde("~", None), "~");
1206 assert_eq!(expand_tilde("~/foo", None), "~/foo");
1207 }
1208
1209 #[test]
1210 fn expand_tilde_passthrough() {
1211 assert_eq!(expand_tilde("/home/user", Some("/h")), "/home/user");
1213 assert_eq!(expand_tilde("foo~bar", Some("/h")), "foo~bar");
1214 assert_eq!(expand_tilde("", Some("/h")), "");
1215 }
1216
1217 #[test]
1218 #[cfg(all(unix, feature = "host"))]
1219 fn expand_tilde_user() {
1220 let expanded = expand_tilde("~root", None);
1223 assert!(
1225 expanded == "/root" || expanded == "/var/root",
1226 "expected /root or /var/root, got: {}",
1227 expanded
1228 );
1229
1230 let expanded_path = expand_tilde("~root/subdir", None);
1232 assert!(
1233 expanded_path == "/root/subdir" || expanded_path == "/var/root/subdir",
1234 "expected /root/subdir or /var/root/subdir, got: {}",
1235 expanded_path
1236 );
1237
1238 let nonexistent = expand_tilde("~nonexistent_user_12345", None);
1240 assert_eq!(nonexistent, "~nonexistent_user_12345");
1241 }
1242
1243 #[test]
1244 fn value_to_string_with_tilde_expansion() {
1245 let val = Value::String("~/test".into());
1247 assert_eq!(
1248 value_to_string_with_tilde(&val, Some("/home/session")),
1249 "/home/session/test"
1250 );
1251 }
1252
1253 #[test]
1254 fn eval_positional_param() {
1255 let mut scope = Scope::new();
1256 scope.set_positional("my_tool", vec!["hello".into(), "world".into()]);
1257
1258 let expr = Expr::Positional(0);
1260 let result = eval_expr(&expr, &mut scope).unwrap();
1261 assert_eq!(result, Value::String("my_tool".into()));
1262
1263 let expr = Expr::Positional(1);
1265 let result = eval_expr(&expr, &mut scope).unwrap();
1266 assert_eq!(result, Value::String("hello".into()));
1267
1268 let expr = Expr::Positional(2);
1270 let result = eval_expr(&expr, &mut scope).unwrap();
1271 assert_eq!(result, Value::String("world".into()));
1272
1273 let expr = Expr::Positional(3);
1275 let result = eval_expr(&expr, &mut scope).unwrap();
1276 assert_eq!(result, Value::String("".into()));
1277 }
1278
1279 #[test]
1280 fn eval_all_args() {
1281 let mut scope = Scope::new();
1282 scope.set_positional("test", vec!["a".into(), "b".into(), "c".into()]);
1283
1284 let expr = Expr::AllArgs;
1285 let result = eval_expr(&expr, &mut scope).unwrap();
1286
1287 assert_eq!(result, Value::String("a b c".into()));
1289 }
1290
1291 #[test]
1292 fn eval_arg_count() {
1293 let mut scope = Scope::new();
1294 scope.set_positional("test", vec!["x".into(), "y".into()]);
1295
1296 let expr = Expr::ArgCount;
1297 let result = eval_expr(&expr, &mut scope).unwrap();
1298 assert_eq!(result, Value::Int(2));
1299 }
1300
1301 #[test]
1302 fn eval_arg_count_empty() {
1303 let mut scope = Scope::new();
1304
1305 let expr = Expr::ArgCount;
1306 let result = eval_expr(&expr, &mut scope).unwrap();
1307 assert_eq!(result, Value::Int(0));
1308 }
1309
1310 #[test]
1311 fn eval_var_length_string() {
1312 let mut scope = Scope::new();
1313 scope.set("NAME", Value::String("hello".into()));
1314
1315 let expr = Expr::VarLength("NAME".into());
1316 let result = eval_expr(&expr, &mut scope).unwrap();
1317 assert_eq!(result, Value::Int(5));
1318 }
1319
1320 #[test]
1321 fn eval_var_length_empty_string() {
1322 let mut scope = Scope::new();
1323 scope.set("EMPTY", Value::String("".into()));
1324
1325 let expr = Expr::VarLength("EMPTY".into());
1326 let result = eval_expr(&expr, &mut scope).unwrap();
1327 assert_eq!(result, Value::Int(0));
1328 }
1329
1330 #[test]
1331 fn eval_var_length_unset() {
1332 let mut scope = Scope::new();
1333
1334 let expr = Expr::VarLength("MISSING".into());
1336 let result = eval_expr(&expr, &mut scope).unwrap();
1337 assert_eq!(result, Value::Int(0));
1338 }
1339
1340 #[test]
1341 fn eval_var_length_int() {
1342 let mut scope = Scope::new();
1343 scope.set("NUM", Value::Int(12345));
1344
1345 let expr = Expr::VarLength("NUM".into());
1347 let result = eval_expr(&expr, &mut scope).unwrap();
1348 assert_eq!(result, Value::Int(5)); }
1350
1351 #[test]
1352 fn eval_var_with_default_set() {
1353 let mut scope = Scope::new();
1354 scope.set("NAME", Value::String("Alice".into()));
1355
1356 let expr = Expr::VarWithDefault {
1358 name: "NAME".into(),
1359 default: vec![StringPart::Literal("default".into())],
1360 };
1361 let result = eval_expr(&expr, &mut scope).unwrap();
1362 assert_eq!(result, Value::String("Alice".into()));
1363 }
1364
1365 #[test]
1366 fn eval_var_with_default_unset() {
1367 let mut scope = Scope::new();
1368
1369 let expr = Expr::VarWithDefault {
1371 name: "MISSING".into(),
1372 default: vec![StringPart::Literal("fallback".into())],
1373 };
1374 let result = eval_expr(&expr, &mut scope).unwrap();
1375 assert_eq!(result, Value::String("fallback".into()));
1376 }
1377
1378 #[test]
1379 fn eval_var_with_default_empty() {
1380 let mut scope = Scope::new();
1381 scope.set("EMPTY", Value::String("".into()));
1382
1383 let expr = Expr::VarWithDefault {
1385 name: "EMPTY".into(),
1386 default: vec![StringPart::Literal("not empty".into())],
1387 };
1388 let result = eval_expr(&expr, &mut scope).unwrap();
1389 assert_eq!(result, Value::String("not empty".into()));
1390 }
1391
1392 #[test]
1393 fn eval_var_with_default_non_string() {
1394 let mut scope = Scope::new();
1395 scope.set("NUM", Value::Int(42));
1396
1397 let expr = Expr::VarWithDefault {
1399 name: "NUM".into(),
1400 default: vec![StringPart::Literal("default".into())],
1401 };
1402 let result = eval_expr(&expr, &mut scope).unwrap();
1403 assert_eq!(result, Value::Int(42));
1404 }
1405
1406 #[test]
1407 fn eval_unset_variable_is_empty() {
1408 let mut scope = Scope::new();
1409 let parts = vec![
1410 StringPart::Literal("prefix:".into()),
1411 StringPart::Var(VarPath::simple("UNSET")),
1412 StringPart::Literal(":suffix".into()),
1413 ];
1414 let expr = Expr::Interpolated(parts);
1415 let result = eval_expr(&expr, &mut scope).unwrap();
1416 assert_eq!(result, Value::String("prefix::suffix".into()));
1417 }
1418
1419 #[test]
1420 fn eval_unset_variable_multiple() {
1421 let mut scope = Scope::new();
1422 scope.set("SET", Value::String("hello".into()));
1423 let parts = vec![
1424 StringPart::Var(VarPath::simple("UNSET1")),
1425 StringPart::Literal("-".into()),
1426 StringPart::Var(VarPath::simple("SET")),
1427 StringPart::Literal("-".into()),
1428 StringPart::Var(VarPath::simple("UNSET2")),
1429 ];
1430 let expr = Expr::Interpolated(parts);
1431 let result = eval_expr(&expr, &mut scope).unwrap();
1432 assert_eq!(result, Value::String("-hello-".into()));
1433 }
1434}