fusabi-stdlib-ext 0.1.6

Extended standard library modules and domain packs for Fusabi
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

//! Environment module.
//!
//! Provides functions for environment variable access with safety controls.

use std::sync::Arc;

use fusabi_host::ExecutionContext;
use fusabi_host::Value;

use crate::safety::SafetyConfig;

/// Get an environment variable.
pub fn get(
    safety: &Arc<SafetyConfig>,
    args: &[Value],
    _ctx: &ExecutionContext,
) -> fusabi_host::Result<Value> {
    let name = args
        .first()
        .and_then(|v| v.as_str())
        .ok_or_else(|| fusabi_host::Error::host_function("env.get: missing name argument"))?;

    // Check safety
    safety.check_env(name).map_err(|e| {
        fusabi_host::Error::host_function(e.to_string())
    })?;

    match std::env::var(name) {
        Ok(value) => Ok(Value::String(value)),
        Err(_) => Ok(Value::Null),
    }
}

/// Set an environment variable.
pub fn set(
    safety: &Arc<SafetyConfig>,
    args: &[Value],
    _ctx: &ExecutionContext,
) -> fusabi_host::Result<Value> {
    let name = args
        .first()
        .and_then(|v| v.as_str())
        .ok_or_else(|| fusabi_host::Error::host_function("env.set: missing name argument"))?;

    let value = args
        .get(1)
        .and_then(|v| v.as_str())
        .ok_or_else(|| fusabi_host::Error::host_function("env.set: missing value argument"))?;

    // Check safety
    safety.check_env(name).map_err(|e| {
        fusabi_host::Error::host_function(e.to_string())
    })?;

    std::env::set_var(name, value);
    Ok(Value::Null)
}

/// Get the current working directory.
pub fn cwd(
    _args: &[Value],
    _ctx: &ExecutionContext,
) -> fusabi_host::Result<Value> {
    match std::env::current_dir() {
        Ok(path) => Ok(Value::String(path.to_string_lossy().into_owned())),
        Err(e) => Err(fusabi_host::Error::host_function(format!("env.cwd: {}", e))),
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use fusabi_host::Capabilities;
    use fusabi_host::{Sandbox, SandboxConfig};
    use fusabi_host::Limits;

    fn create_test_ctx() -> ExecutionContext {
        let sandbox = Sandbox::new(SandboxConfig::default()).unwrap();
        ExecutionContext::new(1, Capabilities::none(), Limits::default(), sandbox)
    }

    #[test]
    fn test_get_safety_check() {
        let safety = Arc::new(SafetyConfig::strict());
        let ctx = create_test_ctx();

        let result = get(&safety, &[Value::String("PATH".into())], &ctx);
        assert!(result.is_err()); // Should fail - env not allowed
    }

    #[test]
    fn test_get_with_permission() {
        let safety = Arc::new(SafetyConfig::new().with_env_vars(["PATH"]));
        let ctx = create_test_ctx();

        let result = get(&safety, &[Value::String("PATH".into())], &ctx);
        assert!(result.is_ok());
    }

    #[test]
    fn test_cwd() {
        let ctx = create_test_ctx();
        let result = cwd(&[], &ctx);
        assert!(result.is_ok());
        assert!(result.unwrap().as_str().is_some());
    }
}