feat(agent): add rust agent baseline scaffold

2026-06-07 16:08:31 +03:00
parent 00ff5e7ddf
commit 731478411a
16 changed files with 1205 additions and 10 deletions
@@ -0,0 +1,174 @@
+use std::fs;
+use std::path::{Path, PathBuf};
+use std::time::Duration;
+
+use anyhow::{Context, Result};
+use sha2::{Digest, Sha256};
+
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub struct AgentConfig {
+    pub agent_id: String,
+    pub host_id: String,
+    pub platform: String,
+    pub server_url: String,
+    pub spool_dir: PathBuf,
+    pub health_bind: String,
+    pub request_timeout_seconds: u64,
+    pub retry_max_attempts: u32,
+    pub retry_base_backoff_ms: u64,
+}
+
+impl Default for AgentConfig {
+    fn default() -> Self {
+        let hostname = local_hostname();
+        Self {
+            agent_id: uuid_from_seed(&format!("agent:{hostname}")),
+            host_id: uuid_from_seed(&format!("host:{hostname}")),
+            platform: current_platform().to_string(),
+            server_url: "http://127.0.0.1:9/api/agent/telemetry".to_string(),
+            spool_dir: default_spool_dir(),
+            health_bind: "127.0.0.1:8787".to_string(),
+            request_timeout_seconds: 10,
+            retry_max_attempts: 3,
+            retry_base_backoff_ms: 250,
+        }
+    }
+}
+
+impl AgentConfig {
+    pub fn load(path: &Path) -> Result<Self> {
+        let text = fs::read_to_string(path).with_context(|| format!("read {}", path.display()))?;
+        Self::parse_toml_like(&text)
+    }
+
+    pub fn parse_toml_like(text: &str) -> Result<Self> {
+        let mut config = Self::default();
+        for raw in text.lines() {
+            let line = raw.split('#').next().unwrap_or("").trim();
+            if line.is_empty() {
+                continue;
+            }
+            let Some((key, value)) = line.split_once('=') else {
+                continue;
+            };
+            let key = key.trim();
+            let value = value.trim().trim_matches('"');
+            match key {
+                "agent_id" => config.agent_id = value.to_string(),
+                "host_id" => config.host_id = value.to_string(),
+                "platform" => config.platform = value.to_string(),
+                "server_url" => config.server_url = value.to_string(),
+                "spool_dir" => config.spool_dir = PathBuf::from(value),
+                "health_bind" => config.health_bind = value.to_string(),
+                "request_timeout_seconds" => {
+                    config.request_timeout_seconds = value.parse().unwrap_or(10)
+                }
+                "retry_max_attempts" => config.retry_max_attempts = value.parse().unwrap_or(3),
+                "retry_base_backoff_ms" => {
+                    config.retry_base_backoff_ms = value.parse().unwrap_or(250)
+                }
+                _ => {}
+            }
+        }
+        Ok(config)
+    }
+
+    pub fn request_timeout(&self) -> Duration {
+        Duration::from_secs(self.request_timeout_seconds)
+    }
+}
+
+pub fn default_config_path() -> PathBuf {
+    if cfg!(windows) {
+        PathBuf::from(r"C:\ProgramData\AWatch-rus\agent\awatch-agent.toml")
+    } else {
+        PathBuf::from("/etc/awatch-agent/awatch-agent.toml")
+    }
+}
+
+fn default_spool_dir() -> PathBuf {
+    if cfg!(windows) {
+        PathBuf::from(r"C:\ProgramData\AWatch-rus\agent\spool")
+    } else {
+        PathBuf::from("/var/lib/awatch-agent/spool")
+    }
+}
+
+fn current_platform() -> &'static str {
+    if cfg!(windows) {
+        "windows"
+    } else if cfg!(target_os = "macos") {
+        "macos"
+    } else if cfg!(target_os = "freebsd") {
+        "freebsd"
+    } else {
+        "linux"
+    }
+}
+
+fn local_hostname() -> String {
+    std::env::var("COMPUTERNAME")
+        .or_else(|_| std::env::var("HOSTNAME"))
+        .unwrap_or_else(|_| "HOST-EXAMPLE".to_string())
+}
+
+fn uuid_from_seed(seed: &str) -> String {
+    let digest = Sha256::digest(seed.as_bytes());
+    let mut bytes = [0_u8; 16];
+    bytes.copy_from_slice(&digest[..16]);
+    bytes[6] = (bytes[6] & 0x0f) | 0x50;
+    bytes[8] = (bytes[8] & 0x3f) | 0x80;
+    format!(
+        "{:02x}{:02x}{:02x}{:02x}-{:02x}{:02x}-{:02x}{:02x}-{:02x}{:02x}-{:02x}{:02x}{:02x}{:02x}{:02x}{:02x}",
+        bytes[0],
+        bytes[1],
+        bytes[2],
+        bytes[3],
+        bytes[4],
+        bytes[5],
+        bytes[6],
+        bytes[7],
+        bytes[8],
+        bytes[9],
+        bytes[10],
+        bytes[11],
+        bytes[12],
+        bytes[13],
+        bytes[14],
+        bytes[15]
+    )
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn parses_config_without_external_toml_dependency() {
+        let config = AgentConfig::parse_toml_like(
+            r#"
+agent_id = "00000000-0000-5000-8000-000000000001"
+host_id = "00000000-0000-5000-8000-000000000002"
+platform = "windows"
+server_url = "https://awatch.example/api/agent/telemetry"
+spool_dir = "/tmp/awatch-agent-spool"
+health_bind = "127.0.0.1:8788"
+request_timeout_seconds = 2
+retry_max_attempts = 5
+retry_base_backoff_ms = 50
+"#,
+        )
+        .unwrap();
+        assert_eq!(config.platform, "windows");
+        assert_eq!(config.retry_max_attempts, 5);
+        assert_eq!(config.spool_dir, PathBuf::from("/tmp/awatch-agent-spool"));
+    }
+
+    #[test]
+    fn generated_ids_are_uuid_shaped() {
+        let id = uuid_from_seed("HOST-EXAMPLE");
+        assert_eq!(id.len(), 36);
+        assert_eq!(&id[14..15], "5");
+        assert!(matches!(&id[19..20], "8" | "9" | "a" | "b"));
+    }
+}