...

src/rhai/nerdctl.rs

@@ -95,6 +95,79 @@ pub fn container_with_health_check(mut container: Container, cmd: &str) -> Conta
     container.with_health_check(cmd)
 }
 
+/// Add multiple port mappings to a Container
+pub fn container_with_ports(mut container: Container, ports: Array) -> Container {
+    for port in ports.iter() {
+        if port.is_string() {
+            let port_str = port.clone().cast::<String>();
+            container = container.with_port(&port_str);
+        }
+    }
+    container
+}
+
+/// Add multiple volume mounts to a Container
+pub fn container_with_volumes(mut container: Container, volumes: Array) -> Container {
+    for volume in volumes.iter() {
+        if volume.is_string() {
+            let volume_str = volume.clone().cast::<String>();
+            container = container.with_volume(&volume_str);
+        }
+    }
+    container
+}
+
+/// Add multiple environment variables to a Container
+pub fn container_with_envs(mut container: Container, env_map: Map) -> Container {
+    for (key, value) in env_map.iter() {
+        if value.is_string() {
+            let value_str = value.clone().cast::<String>();
+            container = container.with_env(&key, &value_str);
+        }
+    }
+    container
+}
+
+/// Add multiple network aliases to a Container
+pub fn container_with_network_aliases(mut container: Container, aliases: Array) -> Container {
+    for alias in aliases.iter() {
+        if alias.is_string() {
+            let alias_str = alias.clone().cast::<String>();
+            container = container.with_network_alias(&alias_str);
+        }
+    }
+    container
+}
+
+/// Set memory swap limit for a Container
+pub fn container_with_memory_swap_limit(mut container: Container, memory_swap: &str) -> Container {
+    container.with_memory_swap_limit(memory_swap)
+}
+
+/// Set CPU shares for a Container
+pub fn container_with_cpu_shares(mut container: Container, shares: &str) -> Container {
+    container.with_cpu_shares(shares)
+}
+
+/// Set health check with options for a Container
+pub fn container_with_health_check_options(
+    mut container: Container,
+    cmd: &str,
+    interval: Option<&str>,
+    timeout: Option<&str>,
+    retries: Option<i64>,
+    start_period: Option<&str>
+) -> Container {
+    // Convert i64 to u32 for retries
+    let retries_u32 = retries.map(|r| r as u32);
+    container.with_health_check_options(cmd, interval, timeout, retries_u32, start_period)
+}
+
+/// Set snapshotter for a Container
+pub fn container_with_snapshotter(mut container: Container, snapshotter: &str) -> Container {
+    container.with_snapshotter(snapshotter)
+}
+
 /// Set detach mode for a Container
 pub fn container_with_detach(mut container: Container, detach: bool) -> Container {
     container.with_detach(detach)
@@ -412,6 +485,14 @@ pub fn register_nerdctl_module(engine: &mut Engine) -> Result<(), Box<EvalAltRes
     engine.register_fn("with_memory_limit", container_with_memory_limit);
     engine.register_fn("with_restart_policy", container_with_restart_policy);
     engine.register_fn("with_health_check", container_with_health_check);
+    engine.register_fn("with_ports", container_with_ports);
+    engine.register_fn("with_volumes", container_with_volumes);
+    engine.register_fn("with_envs", container_with_envs);
+    engine.register_fn("with_network_aliases", container_with_network_aliases);
+    engine.register_fn("with_memory_swap_limit", container_with_memory_swap_limit);
+    engine.register_fn("with_cpu_shares", container_with_cpu_shares);
+    engine.register_fn("with_health_check_options", container_with_health_check_options);
+    engine.register_fn("with_snapshotter", container_with_snapshotter);
     engine.register_fn("with_detach", container_with_detach);
     engine.register_fn("build", container_build);
     engine.register_fn("start", container_start);

src/rhai/os.rs

@@ -35,6 +35,10 @@ pub fn register_os_module(engine: &mut Engine) -> Result<(), Box<EvalAltResult>>
     engine.register_fn("file_write", file_write);
     engine.register_fn("file_write_append", file_write_append);
     
+    // Register command check functions
+    engine.register_fn("which", which);
+    engine.register_fn("cmd_ensure_exists", cmd_ensure_exists);
+    
     // Register download functions
     engine.register_fn("download", download);
     engine.register_fn("download_install", download_install);
@@ -176,4 +180,19 @@ pub fn download(url: &str, dest: &str, min_size_kb: i64) -> Result<String, Box<E
 /// Download a file and install it if it's a supported package format.
 pub fn download_install(url: &str, min_size_kb: i64) -> Result<String, Box<EvalAltResult>> {
     os::download_install(url, min_size_kb).to_rhai_error()
+}
+
+/// Wrapper for os::which
+///
+/// Check if a command exists in the system PATH.
+pub fn which(command: &str) -> String {
+    os::which(command)
+}
+
+/// Wrapper for os::cmd_ensure_exists
+///
+/// Ensure that one or more commands exist in the system PATH.
+/// If any command doesn't exist, an error is thrown.
+pub fn cmd_ensure_exists(commands: &str) -> Result<String, Box<EvalAltResult>> {
+    os::cmd_ensure_exists(commands).to_rhai_error()
 }