rush-sh 0.8.0

A POSIX sh-compatible shell written in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

use std::io::Write;

use crate::parser::ShellCommand;
use crate::state::ShellState;

pub struct ExportBuiltin;

impl super::Builtin for ExportBuiltin {
    fn name(&self) -> &'static str {
        "export"
    }

    fn names(&self) -> Vec<&'static str> {
        vec![self.name()]
    }

    fn description(&self) -> &'static str {
        "Export variables to environment"
    }

    fn run(
        &self,
        cmd: &ShellCommand,
        shell_state: &mut ShellState,
        output_writer: &mut dyn Write,
    ) -> i32 {
        if cmd.args.len() < 2 {
            // Print all exported variables
            for var_name in &shell_state.exported {
                if let Some(value) = shell_state.variables.get(var_name) {
                    let _ = writeln!(output_writer, "export {}={}", var_name, value);
                }
            }
            0
        } else {
            let arg = &cmd.args[1];
            if let Some(eq_pos) = arg.find('=') {
                // export VAR=value
                let var = arg[..eq_pos].to_string();
                let value = arg[eq_pos + 1..].to_string();
                shell_state.set_exported_var(&var, value);
                0
            } else {
                // export VAR (mark existing var as exported)
                shell_state.export_var(arg);
                0
            }
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::builtins::Builtin;

    #[test]
    fn test_export_builtin_list() {
        let cmd = ShellCommand {
            args: vec!["export".to_string()],
            redirections: Vec::new(),
            compound: None,
        };
        let mut shell_state = ShellState::new();
        shell_state.set_exported_var("TEST_VAR", "test_value".to_string());
        let builtin = ExportBuiltin;
        let mut output = Vec::new();
        let exit_code = builtin.run(&cmd, &mut shell_state, &mut output);
        assert_eq!(exit_code, 0);
        let output_str = String::from_utf8(output).unwrap();
        assert!(output_str.contains("export TEST_VAR=test_value"));
    }

    #[test]
    fn test_export_builtin_set() {
        let cmd = ShellCommand {
            args: vec!["export".to_string(), "NEW_VAR=new_value".to_string()],
            redirections: Vec::new(),
            compound: None,
        };
        let mut shell_state = ShellState::new();
        let builtin = ExportBuiltin;
        let mut output = Vec::new();
        let exit_code = builtin.run(&cmd, &mut shell_state, &mut output);
        assert_eq!(exit_code, 0);
        assert_eq!(
            shell_state.get_var("NEW_VAR"),
            Some("new_value".to_string())
        );
        assert!(shell_state.exported.contains("NEW_VAR"));
    }

    #[test]
    fn test_export_builtin_export_existing() {
        let cmd = ShellCommand {
            args: vec!["export".to_string(), "EXISTING_VAR".to_string()],
            redirections: Vec::new(),
            compound: None,
        };
        let mut shell_state = ShellState::new();
        shell_state.set_var("EXISTING_VAR", "existing_value".to_string());
        let builtin = ExportBuiltin;
        let mut output = Vec::new();
        let exit_code = builtin.run(&cmd, &mut shell_state, &mut output);
        assert_eq!(exit_code, 0);
        assert!(shell_state.exported.contains("EXISTING_VAR"));
    }
}