5194f5245d
- Updated the Gitea host URL in `.roo/mcp.json` and `Cargo.toml` to reflect the change from `git.ourworld.tf` to `git.threefold.info`. - Updated the `zinit-client` dependency in `Cargo.toml` to version `0.3.0`. This ensures compatibility with the updated repository. - Updated file paths in example files to reflect the new repository URL.
79 lines
2.9 KiB
Rust
79 lines
2.9 KiB
Rust
// File: /root/code/git.threefold.info/herocode/sal/src/virt/nerdctl/health_check_script.rs
|
|
|
|
use std::fs;
|
|
use std::path::Path;
|
|
use std::os::unix::fs::PermissionsExt;
|
|
|
|
/// Handles health check scripts for containers
|
|
///
|
|
/// This module provides functionality to create and manage health check scripts
|
|
/// for containers, allowing for more complex health checks than simple commands.
|
|
|
|
/// Converts a health check command or script to a usable command
|
|
///
|
|
/// If the input is a single-line command, it is returned as is.
|
|
/// If the input is a multi-line script, it is written to a file in the
|
|
/// /root/hero/var/containers directory and the path to that file is returned.
|
|
///
|
|
/// # Arguments
|
|
///
|
|
/// * `cmd` - The command or script to convert
|
|
/// * `container_name` - The name of the container, used to create a unique script name
|
|
///
|
|
/// # Returns
|
|
///
|
|
/// * `String` - The command to use for the health check
|
|
pub fn prepare_health_check_command(cmd: &str, container_name: &str) -> String {
|
|
// If the command is a multiline script, write it to a file
|
|
if cmd.contains("\n") {
|
|
// Create the directory if it doesn't exist
|
|
let dir_path = "/root/hero/var/containers";
|
|
if let Err(_) = fs::create_dir_all(dir_path) {
|
|
// If we can't create the directory, just use the command as is
|
|
return cmd.to_string();
|
|
}
|
|
|
|
// Create a unique filename based on container name
|
|
let script_path = format!("{}/healthcheck_{}.sh", dir_path, container_name);
|
|
|
|
// Write the script to the file
|
|
if let Err(_) = fs::write(&script_path, cmd) {
|
|
// If we can't write the file, just use the command as is
|
|
return cmd.to_string();
|
|
}
|
|
|
|
// Make the script executable
|
|
if let Ok(metadata) = fs::metadata(&script_path) {
|
|
let mut perms = metadata.permissions();
|
|
perms.set_mode(0o755);
|
|
if let Err(_) = fs::set_permissions(&script_path, perms) {
|
|
// If we can't set permissions, just use the script path with sh
|
|
return format!("sh {}", script_path);
|
|
}
|
|
} else {
|
|
// If we can't get metadata, just use the script path with sh
|
|
return format!("sh {}", script_path);
|
|
}
|
|
|
|
// Use the script path as the command
|
|
script_path
|
|
} else {
|
|
// If it's a single line command, use it as is
|
|
cmd.to_string()
|
|
}
|
|
}
|
|
|
|
/// Cleans up health check scripts for a container
|
|
///
|
|
/// # Arguments
|
|
///
|
|
/// * `container_name` - The name of the container whose health check scripts should be cleaned up
|
|
pub fn cleanup_health_check_scripts(container_name: &str) {
|
|
let dir_path = "/root/hero/var/containers";
|
|
let script_path = format!("{}/healthcheck_{}.sh", dir_path, container_name);
|
|
|
|
// Try to remove the script file if it exists
|
|
if Path::new(&script_path).exists() {
|
|
let _ = fs::remove_file(script_path);
|
|
}
|
|
} |